util.c

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