kris/util.c

   1 #include <netdb.h>
   2 #include <netinet/in.h>
   3 #include <sys/socket.h>
   4
   5 #define va_buf(buf, fmt) {\
   6         va_list ap; \
   7         va_start(ap, fmt); \
   8         vsnprintf(buf, sizeof buf, fmt, ap); \
   9         va_end(ap); \
  10 }
  11
  12 static void
  13 eprint(const char *fmt, ...) {
  14
  15         va_buf(bufout, fmt);
  16         fprintf(stderr, "%s", bufout);
  17
  18         if(fmt[0] && fmt[strlen(fmt)-1] == ':')
  19                 fprintf(stderr, " %s\n", strerror(errno));
  20         exit(1);
  21 }
  22
  23 static int
  24 dial(char *host, char *port) {
  25         static struct addrinfo hints;
  26         struct addrinfo *res, *r;
  27         int srv;
  28
  29         memset(&hints, 0, sizeof hints);
  30         hints.ai_family = AF_UNSPEC;
  31         hints.ai_socktype = SOCK_STREAM;
  32         if(getaddrinfo(host, port, &hints, &res) != 0)
  33                 eprint("error: cannot resolve hostname '%s':", host);
  34         for(r = res; r; r = r->ai_next) {
  35                 if((srv = socket(r->ai_family, r->ai_socktype, r->ai_protocol)) == -1)
  36                         continue;
  37                 if(connect(srv, r->ai_addr, r->ai_addrlen) == 0)
  38                         break;
  39                 close(srv);
  40         }
  41         freeaddrinfo(res);
  42         if(!r)
  43                 eprint("error: cannot connect to host '%s'\n", host);
  44         return srv;
  45 }
  46
  47 #define strlcpy _strlcpy
  48 static void
  49 strlcpy(char *to, const char *from, int l) {
  50         memccpy(to, from, '\0', l);
  51         to[l-1] = '\0';
  52 }
  53
  54 static void
  55 eat(char **s, int (*p)(int), int r) {
  56         char *q;
  57
  58         for(q=*s; *q && p(*q) == r; q++)
  59                 ;
  60         *s = q;
  61 }
  62
  63 static char*
  64 tok(char **s) {
  65         char *p;
  66
  67         eat(s, isspace, 1);
  68         p = *s;
  69         eat(s, isspace, 0);
  70         if(**s) *(*s)++ = '\0';
  71         return p;
  72 }
  73
  74 static char*
  75 ctok(char **s, int c) {
  76         char *p, *q;
  77
  78         q = *s;
  79         for(p=q; *p && *p != c; p++)
  80                 ;
  81         if(*p) *p++ = '\0';
  82         *s = p;
  83         return q;
  84 }
  85
  86