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 char *
  51 eat(char *s, int (*p)(int), int r) {
  52         while(s != '\0' && p(*s) == r)
  53                 s++;
  54         return s;
  55 }
  56
  57 static char*
  58 skip(char *s, char c) {
  59         while(*s != c && *s != '\0')
  60                 s++;
  61         if(*s != '\0')
  62                 *s++ = '\0';
  63         return s;
  64 }
  65
  66 static void
  67 trim(char *s) {
  68         char *e;
  69
  70         e = s + strlen(s) - 1;
  71         while(isspace(*e) && e > s)
  72                 e--;
  73         *(e + 1) = '\0';
  74 }