kris/util.c

   1 #include <netdb.h>
   2 #include <netinet/in.h>
   3 #include <sys/socket.h>
   4
   5 static void
   6 eprint(const char *fmt, ...) {
   7
   8         va_buf(bufout, fmt);
   9         fprintf(stderr, "%s", bufout);
  10
  11         if(fmt[0] && fmt[strlen(fmt)-1] == ':')
  12                 fprintf(stderr, " %s\n", strerror(errno));
  13         exit(1);
  14 }
  15
  16 static int
  17 dial(char *host, int port) {
  18         struct hostent *hp;
  19         static struct sockaddr_in addr;
  20         int i;
  21
  22         if((i = socket(AF_INET, SOCK_STREAM, 0)) < 0)
  23                 eprint("sic: cannot connect host '%s':", host);
  24         if(nil == (hp = gethostbyname(host)))
  25                 eprint("sic: cannot resolve hostname '%s': %s\n", host, hstrerror(h_errno));
  26         addr.sin_family = AF_INET;
  27         addr.sin_port = htons(port);
  28         memcpy(&addr.sin_addr, hp->h_addr, hp->h_length);
  29         if(connect(i, (struct sockaddr*)&addr, sizeof(struct sockaddr_in)))
  30                 eprint("sic: cannot connect host '%s':", host);
  31         return i;
  32 }
  33
  34 #define strlcpy _strlcpy
  35 static void
  36 strlcpy(char *to, const char *from, int l) {
  37         strncpy(to, from, l-1);
  38         to[l-1] = '\0';
  39 }
  40
  41 static void
  42 eat(char **s, int (*p)(int), int r) {
  43         char *q;
  44
  45         for(q=*s; *q && p(*q) == r; q++)
  46                 ;
  47         *s = q;
  48 }
  49
  50 static char*
  51 tok(char **s) {
  52         char *p;
  53
  54         eat(s, isspace, 1);
  55         p = *s;
  56         eat(s, isspace, 0);
  57         if(**s) *(*s)++ = '\0';
  58         return p;
  59 }
  60
  61 static char*
  62 ctok(char **s, int c) {
  63         char *p, *q;
  64
  65         q = *s;
  66         for(p=q; *p && *p != c; p++)
  67                 ;
  68         if(*p) *p++ = '\0';
  69         *s = p;
  70         return q;
  71 }
  72
  73