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, int port) {
  25         struct hostent *hp;
  26         static struct sockaddr_in addr;
  27         int i;
  28
  29         if((i = socket(AF_INET, SOCK_STREAM, 0)) < 0)
  30                 eprint("sic: cannot connect host '%s':", host);
  31         if(nil == (hp = gethostbyname(host)))
  32                 eprint("sic: cannot resolve hostname '%s': %s\n", host, hstrerror(h_errno));
  33         addr.sin_family = AF_INET;
  34         addr.sin_port = htons(port);
  35         memcpy(&addr.sin_addr, hp->h_addr, hp->h_length);
  36         if(connect(i, (struct sockaddr*)&addr, sizeof(struct sockaddr_in)))
  37                 eprint("sic: cannot connect host '%s':", host);
  38         return i;
  39 }
  40
  41 #define strlcpy _strlcpy
  42 static void
  43 strlcpy(char *to, const char *from, int l) {
  44         memccpy(to, from, '\0', l);
  45         to[l-1] = '\0';
  46 }
  47
  48 static void
  49 eat(char **s, int (*p)(int), int r) {
  50         char *q;
  51
  52         for(q=*s; *q && p(*q) == r; q++)
  53                 ;
  54         *s = q;
  55 }
  56
  57 static char*
  58 tok(char **s) {
  59         char *p;
  60
  61         eat(s, isspace, 1);
  62         p = *s;
  63         eat(s, isspace, 0);
  64         if(**s) *(*s)++ = '\0';
  65         return p;
  66 }
  67
  68 static char*
  69 ctok(char **s, int c) {
  70         char *p, *q;
  71
  72         q = *s;
  73         for(p=q; *p && *p != c; p++)
  74                 ;
  75         if(*p) *p++ = '\0';
  76         *s = p;
  77         return q;
  78 }
  79
  80