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 #include "util.h"
  19
  20 char *stringcopy(const char *s, int len) {
  21     char *r;
  22     if (!len)
  23         len = strlen(s);
  24     r = malloc(len + 1);
  25     if (!r)
  26         debug(DBG_ERR, "stringcopy: malloc failed");
  27     memcpy(r, s, len);
  28     r[len] = '\0';
  29     return r;
  30 }
  31
  32 void printfchars(char *prefixfmt, char *prefix, char *charfmt, char *chars, int len) {
  33     int i;
  34     unsigned char *s = (unsigned char *)chars;
  35     if (prefix)
  36         printf(prefixfmt ? prefixfmt : "%s: ", prefix);
  37     for (i = 0; i < len; i++)
  38         printf(charfmt ? charfmt : "%c", s[i]);
  39     printf("\n");
  40 }
  41
  42 uint16_t port_get(struct sockaddr *sa) {
  43     switch (sa->sa_family) {
  44     case AF_INET:
  45         return ntohs(((struct sockaddr_in *)sa)->sin_port);
  46     case AF_INET6:
  47         return ntohs(((struct sockaddr_in6 *)sa)->sin6_port);
  48     }
  49     return 0;
  50 }
  51
  52 void port_set(struct sockaddr *sa, uint16_t port) {
  53     switch (sa->sa_family) {
  54     case AF_INET:
  55         ((struct sockaddr_in *)sa)->sin_port = htons(port);
  56         break;
  57     case AF_INET6:
  58         ((struct sockaddr_in6 *)sa)->sin6_port = htons(port);
  59         break;
  60     }
  61 }
  62
  63 int addr_equal(struct sockaddr *a, struct sockaddr *b) {
  64     switch (a->sa_family) {
  65     case AF_INET:
  66         return !memcmp(&((struct sockaddr_in*)a)->sin_addr,
  67                        &((struct sockaddr_in*)b)->sin_addr,
  68                        sizeof(struct in_addr));
  69     case AF_INET6:
  70         return IN6_ARE_ADDR_EQUAL(&((struct sockaddr_in6*)a)->sin6_addr,
  71                                   &((struct sockaddr_in6*)b)->sin6_addr);
  72     default:
  73         /* Must not reach */
  74         return 0;
  75     }
  76 }
  77
  78 struct sockaddr *addr_copy(struct sockaddr *in) {
  79     struct sockaddr *out = NULL;
  80
  81     switch (in->sa_family) {
  82     case AF_INET:
  83         out = malloc(sizeof(struct sockaddr_in));
  84         if (out) {
  85             memset(out, 0, sizeof(struct sockaddr_in));
  86             ((struct sockaddr_in *)out)->sin_addr = ((struct sockaddr_in *)in)->sin_addr;
  87         }
  88         break;
  89     case AF_INET6:
  90         out = malloc(sizeof(struct sockaddr_in6));
  91         if (out) {
  92             memset(out, 0, sizeof(struct sockaddr_in6));
  93             ((struct sockaddr_in6 *)out)->sin6_addr = ((struct sockaddr_in6 *)in)->sin6_addr;
  94         }
  95         break;
  96     }
  97     out->sa_family = in->sa_family;
  98     return out;
  99 }
 100
 101 char *addr2string(struct sockaddr *addr) {
 102     struct sockaddr_in6 *sa6;
 103     struct sockaddr_in sa4;
 104     static char addr_buf[2][INET6_ADDRSTRLEN];
 105     static int i = 0;
 106     i = !i;
 107     if (addr->sa_family == AF_INET6) {
 108         sa6 = (struct sockaddr_in6 *)addr;
 109         if (IN6_IS_ADDR_V4MAPPED(&sa6->sin6_addr)) {
 110             memset(&sa4, 0, sizeof(sa4));
 111             sa4.sin_family = AF_INET;
 112             sa4.sin_port = sa6->sin6_port;
 113             memcpy(&sa4.sin_addr, &sa6->sin6_addr.s6_addr[12], 4);
 114             addr = (struct sockaddr *)&sa4;
 115         }
 116     }
 117     if (getnameinfo(addr, SOCKADDRP_SIZE(addr), addr_buf[i], sizeof(addr_buf[i]),
 118                     NULL, 0, NI_NUMERICHOST)) {
 119         debug(DBG_WARN, "getnameinfo failed");
 120         return "getnameinfo_failed";
 121     }
 122     return addr_buf[i];
 123 }
 124
 125 int connectport(int type, char *host, char *port) {
 126     struct addrinfo hints, *res0, *res;
 127     int s = -1;
 128
 129     memset(&hints, 0, sizeof(hints));
 130     hints.ai_socktype = type;
 131     hints.ai_family = AF_UNSPEC;
 132
 133     if (getaddrinfo(host, port, &hints, &res0) != 0) {
 134         debug(DBG_ERR, "connectport: can't resolve host %s port %s", host, port);
 135         return -1;
 136     }
 137
 138     for (res = res0; res; res = res->ai_next) {
 139         s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
 140         if (s < 0) {
 141             debug(DBG_WARN, "connectport: socket failed");
 142             continue;
 143         }
 144         if (connect(s, res->ai_addr, res->ai_addrlen) == 0)
 145             break;
 146         debug(DBG_WARN, "connectport: connect failed");
 147         close(s);
 148         s = -1;
 149     }
 150     freeaddrinfo(res0);
 151     return s;
 152 }
 153
 154 int bindtoaddr(struct addrinfo *addrinfo, int family, int reuse, int v6only) {
 155     int s, on = 1;
 156     struct addrinfo *res;
 157
 158     for (res = addrinfo; res; res = res->ai_next) {
 159         if (family != AF_UNSPEC && family != res->ai_family)
 160             continue;
 161         s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
 162         if (s < 0) {
 163             debug(DBG_WARN, "bindtoaddr: socket failed");
 164             continue;
 165         }
 166         if (reuse)
 167             setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
 168         #ifdef IPV6_V6ONLY
 169         if (v6only)
 170             setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on));
 171         #endif
 172         if (!bind(s, res->ai_addr, res->ai_addrlen))
 173             return s;
 174         debug(DBG_WARN, "bindtoaddr: bind failed");
 175         close(s);
 176     }
 177     return -1;
 178 }
 179
 180 int connecttcp(struct addrinfo *addrinfo, struct addrinfo *src) {
 181     int s;
 182     struct addrinfo *res;
 183
 184     s = -1;
 185     for (res = addrinfo; res; res = res->ai_next) {
 186         s = bindtoaddr(src, res->ai_family, 1, 1);
 187         if (s < 0) {
 188             debug(DBG_WARN, "connecttoserver: socket failed");
 189             continue;
 190         }
 191         if (connect(s, res->ai_addr, res->ai_addrlen) == 0)
 192             break;
 193         debug(DBG_WARN, "connecttoserver: connect failed");
 194         close(s);
 195         s = -1;
 196     }
 197     return s;
 198 }