libbb/inet_common.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * stolen from net-tools-1.59 and stripped down for busybox by
   4  *                      Erik Andersen <andersen@codepoet.org>
   5  *
   6  * Heavily modified by Manuel Novoa III       Mar 12, 2001
   7  *
   8  * Version:     $Id: inet_common.c,v 1.8 2004/03/10 07:42:38 mjn3 Exp $
   9  *
  10  */
  11
  12 #include "libbb.h"
  13 #include "inet_common.h"
  14
  15 const char bb_INET_default[] = "default";
  16
  17 int INET_resolve(const char *name, struct sockaddr_in *s_in, int hostfirst)
  18 {
  19         struct hostent *hp;
  20         struct netent *np;
  21
  22         /* Grmpf. -FvK */
  23         s_in->sin_family = AF_INET;
  24         s_in->sin_port = 0;
  25
  26         /* Default is special, meaning 0.0.0.0. */
  27         if (!strcmp(name, bb_INET_default)) {
  28                 s_in->sin_addr.s_addr = INADDR_ANY;
  29                 return (1);
  30         }
  31         /* Look to see if it's a dotted quad. */
  32         if (inet_aton(name, &s_in->sin_addr)) {
  33                 return 0;
  34         }
  35         /* If we expect this to be a hostname, try hostname database first */
  36 #ifdef DEBUG
  37         if (hostfirst) {
  38                 bb_error_msg("gethostbyname (%s)", name);
  39         }
  40 #endif
  41         if (hostfirst && (hp = gethostbyname(name)) != (struct hostent *) NULL) {
  42                 memcpy((char *) &s_in->sin_addr, (char *) hp->h_addr_list[0],
  43                            sizeof(struct in_addr));
  44                 return 0;
  45         }
  46         /* Try the NETWORKS database to see if this is a known network. */
  47 #ifdef DEBUG
  48         bb_error_msg("getnetbyname (%s)", name);
  49 #endif
  50         if ((np = getnetbyname(name)) != (struct netent *) NULL) {
  51                 s_in->sin_addr.s_addr = htonl(np->n_net);
  52                 return 1;
  53         }
  54         if (hostfirst) {
  55                 /* Don't try again */
  56                 return -1;
  57         }
  58 #ifdef DEBUG
  59         res_init();
  60         _res.options |= RES_DEBUG;
  61 #endif
  62
  63 #ifdef DEBUG
  64         bb_error_msg("gethostbyname (%s)", name);
  65 #endif
  66         if ((hp = gethostbyname(name)) == (struct hostent *) NULL) {
  67                 return -1;
  68         }
  69         memcpy((char *) &s_in->sin_addr, (char *) hp->h_addr_list[0],
  70                    sizeof(struct in_addr));
  71
  72         return 0;
  73 }
  74
  75 /* cache */
  76 struct addr {
  77         struct sockaddr_in addr;
  78         char *name;
  79         int host;
  80         struct addr *next;
  81 };
  82
  83 static struct addr *INET_nn = NULL;     /* addr-to-name cache           */
  84
  85 /* numeric: & 0x8000: default instead of *,
  86  *          & 0x4000: host instead of net,
  87  *          & 0x0fff: don't resolve
  88  */
  89 int INET_rresolve(char *name, size_t len, struct sockaddr_in *s_in,
  90                                   int numeric, unsigned int netmask)
  91 {
  92         struct hostent *ent;
  93         struct netent *np;
  94         struct addr *pn;
  95         unsigned long ad, host_ad;
  96         int host = 0;
  97
  98         /* Grmpf. -FvK */
  99         if (s_in->sin_family != AF_INET) {
 100 #ifdef DEBUG
 101                 bb_error_msg("rresolve: unsupport address family %d !",
 102                                   s_in->sin_family);
 103 #endif
 104                 errno = EAFNOSUPPORT;
 105                 return (-1);
 106         }
 107         ad = (unsigned long) s_in->sin_addr.s_addr;
 108 #ifdef DEBUG
 109         bb_error_msg("rresolve: %08lx, mask %08x, num %08x", ad, netmask, numeric);
 110 #endif
 111         if (ad == INADDR_ANY) {
 112                 if ((numeric & 0x0FFF) == 0) {
 113                         if (numeric & 0x8000)
 114                                 safe_strncpy(name, bb_INET_default, len);
 115                         else
 116                                 safe_strncpy(name, "*", len);
 117                         return (0);
 118                 }
 119         }
 120         if (numeric & 0x0FFF) {
 121                 safe_strncpy(name, inet_ntoa(s_in->sin_addr), len);
 122                 return (0);
 123         }
 124
 125         if ((ad & (~netmask)) != 0 || (numeric & 0x4000))
 126                 host = 1;
 127         pn = INET_nn;
 128         while (pn != NULL) {
 129                 if (pn->addr.sin_addr.s_addr == ad && pn->host == host) {
 130                         safe_strncpy(name, pn->name, len);
 131 #ifdef DEBUG
 132                         bb_error_msg("rresolve: found %s %08lx in cache",
 133                                           (host ? "host" : "net"), ad);
 134 #endif
 135                         return (0);
 136                 }
 137                 pn = pn->next;
 138         }
 139
 140         host_ad = ntohl(ad);
 141         np = NULL;
 142         ent = NULL;
 143         if (host) {
 144 #ifdef DEBUG
 145                 bb_error_msg("gethostbyaddr (%08lx)", ad);
 146 #endif
 147                 ent = gethostbyaddr((char *) &ad, 4, AF_INET);
 148                 if (ent != NULL) {
 149                         safe_strncpy(name, ent->h_name, len);
 150                 }
 151         } else {
 152 #ifdef DEBUG
 153                 bb_error_msg("getnetbyaddr (%08lx)", host_ad);
 154 #endif
 155                 np = getnetbyaddr(host_ad, AF_INET);
 156                 if (np != NULL) {
 157                         safe_strncpy(name, np->n_name, len);
 158                 }
 159         }
 160         if ((ent == NULL) && (np == NULL)) {
 161                 safe_strncpy(name, inet_ntoa(s_in->sin_addr), len);
 162         }
 163         pn = (struct addr *) xmalloc(sizeof(struct addr));
 164         pn->addr = *s_in;
 165         pn->next = INET_nn;
 166         pn->host = host;
 167         pn->name = xstrdup(name);
 168         INET_nn = pn;
 169
 170         return (0);
 171 }
 172
 173 #ifdef CONFIG_FEATURE_IPV6
 174
 175 int INET6_resolve(const char *name, struct sockaddr_in6 *sin6)
 176 {
 177         struct addrinfo req, *ai;
 178         int s;
 179
 180         memset(&req, '\0', sizeof req);
 181         req.ai_family = AF_INET6;
 182         if ((s = getaddrinfo(name, NULL, &req, &ai))) {
 183                 bb_error_msg("getaddrinfo: %s: %d", name, s);
 184                 return -1;
 185         }
 186         memcpy(sin6, ai->ai_addr, sizeof(struct sockaddr_in6));
 187
 188         freeaddrinfo(ai);
 189
 190         return (0);
 191 }
 192
 193 #ifndef IN6_IS_ADDR_UNSPECIFIED
 194 # define IN6_IS_ADDR_UNSPECIFIED(a) \
 195         (((uint32_t *) (a))[0] == 0 && ((uint32_t *) (a))[1] == 0 && \
 196          ((uint32_t *) (a))[2] == 0 && ((uint32_t *) (a))[3] == 0)
 197 #endif
 198
 199
 200 int INET6_rresolve(char *name, size_t len, struct sockaddr_in6 *sin6,
 201                                    int numeric)
 202 {
 203         int s;
 204
 205         /* Grmpf. -FvK */
 206         if (sin6->sin6_family != AF_INET6) {
 207 #ifdef DEBUG
 208                 bb_error_msg(_("rresolve: unsupport address family %d!"),
 209                                   sin6->sin6_family);
 210 #endif
 211                 errno = EAFNOSUPPORT;
 212                 return (-1);
 213         }
 214         if (numeric & 0x7FFF) {
 215                 inet_ntop(AF_INET6, &sin6->sin6_addr, name, len);
 216                 return (0);
 217         }
 218         if (IN6_IS_ADDR_UNSPECIFIED(&sin6->sin6_addr)) {
 219                 if (numeric & 0x8000) {
 220                         strcpy(name, "default");
 221                 } else {
 222                         strcpy(name, "*");
 223                 }
 224                 return (0);
 225         }
 226
 227         s = getnameinfo((struct sockaddr *) sin6, sizeof(struct sockaddr_in6), name, len, NULL, 0, 0);
 228         if (s) {
 229                 bb_error_msg("getnameinfo failed");
 230                 return -1;
 231         }
 232         return (0);
 233 }
 234
 235 #endif                                                  /* CONFIG_FEATURE_IPV6 */