util.c

   1 /*
   2  * Copyright (C) 2006-2008 Stig Venaas <venaas@uninett.no>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  */
   8
   9 #include <sys/socket.h>
  10 #include <netinet/in.h>
  11 #include <netdb.h>
  12 #include <stdio.h>
  13 #include <stdlib.h>
  14 #include <string.h>
  15 #include <unistd.h>
  16 #include <stdarg.h>
  17 #include "debug.h"
  18
  19 #if 0
  20 #include <errno.h>
  21 void errx(char *format, ...) {
  22     extern int errno;
  23
  24     va_list ap;
  25     va_start(ap, format);
  26     vfprintf(stderr, format, ap);
  27     va_end(ap);
  28     if (errno) {
  29         fprintf(stderr, ": ");
  30         perror(NULL);
  31         fprintf(stderr, "errno=%d\n", errno);
  32     } else
  33         fprintf(stderr, "\n");
  34     exit(1);
  35 }
  36
  37 void err(char *format, ...) {
  38     extern int errno;
  39
  40     va_list ap;
  41     va_start(ap, format);
  42     vfprintf(stderr, format, ap);
  43     va_end(ap);
  44     if (errno) {
  45         fprintf(stderr, ": ");
  46         perror(NULL);
  47         fprintf(stderr, "errno=%d\n", errno);
  48     } else
  49         fprintf(stderr, "\n");
  50 }
  51 #endif
  52
  53 char *stringcopy(const char *s, int len) {
  54     char *r;
  55     if (!len)
  56         len = strlen(s);
  57     r = malloc(len + 1);
  58     if (!r)
  59         debug(DBG_ERR, "stringcopy: malloc failed");
  60     memcpy(r, s, len);
  61     r[len] = '\0';
  62     return r;
  63 }
  64
  65 void printfchars(char *prefixfmt, char *prefix, char *charfmt, char *chars, int len) {
  66     int i;
  67     unsigned char *s = (unsigned char *)chars;
  68     if (prefix)
  69         printf(prefixfmt ? prefixfmt : "%s: ", prefix);
  70     for (i = 0; i < len; i++)
  71         printf(charfmt ? charfmt : "%c", s[i]);
  72     printf("\n");
  73 }
  74
  75 uint16_t port_get(struct sockaddr *sa) {
  76     switch (sa->sa_family) {
  77     case AF_INET:
  78         return ntohs(((struct sockaddr_in *)sa)->sin_port);
  79     case AF_INET6:
  80         return ntohs(((struct sockaddr_in6 *)sa)->sin6_port);
  81     }
  82     return 0;
  83 }
  84
  85 void port_set(struct sockaddr *sa, uint16_t port) {
  86     switch (sa->sa_family) {
  87     case AF_INET:
  88         ((struct sockaddr_in *)sa)->sin_port = htons(port);
  89         break;
  90     case AF_INET6:
  91         ((struct sockaddr_in6 *)sa)->sin6_port = htons(port);
  92         break;
  93     }
  94 }
  95
  96 int addr_equal(struct sockaddr *a, struct sockaddr *b) {
  97     switch (a->sa_family) {
  98     case AF_INET:
  99         return !memcmp(&((struct sockaddr_in*)a)->sin_addr,
 100                        &((struct sockaddr_in*)b)->sin_addr,
 101                        sizeof(struct in_addr));
 102     case AF_INET6:
 103         return IN6_ARE_ADDR_EQUAL(&((struct sockaddr_in6*)a)->sin6_addr,
 104                                   &((struct sockaddr_in6*)b)->sin6_addr);
 105     default:
 106         /* Must not reach */
 107         return 0;
 108     }
 109 }
 110
 111 struct sockaddr *addr_copy(struct sockaddr *in) {
 112     struct sockaddr *out = NULL;
 113
 114     switch (in->sa_family) {
 115     case AF_INET:
 116         out = malloc(sizeof(struct sockaddr_in));
 117         if (out) {
 118             memset(out, 0, sizeof(struct sockaddr_in));
 119             ((struct sockaddr_in *)out)->sin_addr = ((struct sockaddr_in *)in)->sin_addr;
 120         }
 121         break;
 122     case AF_INET6:
 123         out = malloc(sizeof(struct sockaddr_in6));
 124         if (out) {
 125             memset(out, 0, sizeof(struct sockaddr_in6));
 126             ((struct sockaddr_in6 *)out)->sin6_addr = ((struct sockaddr_in6 *)in)->sin6_addr;
 127         }
 128         break;
 129     }
 130     out->sa_family = in->sa_family;
 131     return out;
 132 }
 133
 134 char *addr2string(struct sockaddr *addr, socklen_t len) {
 135     struct sockaddr_in6 *sa6;
 136     struct sockaddr_in sa4;
 137     static char addr_buf[2][INET6_ADDRSTRLEN];
 138     static int i = 0;
 139     i = !i;
 140     if (addr->sa_family == AF_INET6) {
 141         sa6 = (struct sockaddr_in6 *)addr;
 142         if (IN6_IS_ADDR_V4MAPPED(&sa6->sin6_addr)) {
 143             memset(&sa4, 0, sizeof(sa4));
 144             sa4.sin_family = AF_INET;
 145             sa4.sin_port = sa6->sin6_port;
 146             memcpy(&sa4.sin_addr, &sa6->sin6_addr.s6_addr[12], 4);
 147             addr = (struct sockaddr *)&sa4;
 148         }
 149     }
 150     len = addr->sa_family == AF_INET ? sizeof(struct sockaddr_in) : sizeof(struct sockaddr_in6);
 151
 152     if (getnameinfo(addr, len, addr_buf[i], sizeof(addr_buf[i]),
 153                     NULL, 0, NI_NUMERICHOST)) {
 154         debug(DBG_WARN, "getnameinfo failed");
 155         return "getnameinfo_failed";
 156     }
 157     return addr_buf[i];
 158 }
 159
 160 int connectport(int type, char *host, char *port) {
 161     struct addrinfo hints, *res0, *res;
 162     int s = -1;
 163
 164     memset(&hints, 0, sizeof(hints));
 165     hints.ai_socktype = type;
 166     hints.ai_family = AF_UNSPEC;
 167
 168     if (getaddrinfo(host, port, &hints, &res0) != 0) {
 169         debug(DBG_ERR, "connectport: can't resolve host %s port %s", host, port);
 170         return -1;
 171     }
 172
 173     for (res = res0; res; res = res->ai_next) {
 174         s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
 175         if (s < 0) {
 176             debug(DBG_WARN, "connectport: socket failed");
 177             continue;
 178         }
 179         if (connect(s, res->ai_addr, res->ai_addrlen) == 0)
 180             break;
 181         debug(DBG_WARN, "connectport: connect failed");
 182         close(s);
 183         s = -1;
 184     }
 185     freeaddrinfo(res0);
 186     return s;
 187 }