nsz Git - musl/blob - src/network/getnameinfo.c

   1 #include <netdb.h>
   2 #include <limits.h>
   3 #include <string.h>
   4 #include <stdio.h>
   5 #include <sys/socket.h>
   6 #include <netinet/in.h>
   7 #include <arpa/inet.h>
   8
   9 int __dns_parse(const unsigned char *, int, int (*)(void *, int, const void *, int, const void *), void *);
  10 int __dn_expand(const unsigned char *, const unsigned char *, const unsigned char *, char *, int);
  11 int __res_mkquery(int, const char *, int, int, const unsigned char *, int, const unsigned char*, unsigned char *, int);
  12 int __res_send(const unsigned char *, int, unsigned char *, int);
  13
  14 #define PTR_MAX (64 + sizeof ".in-addr.arpa")
  15 #define RR_PTR 12
  16
  17 static void mkptr4(char *s, const unsigned char *ip)
  18 {
  19         sprintf(s, "%d.%d.%d.%d.in-addr.arpa",
  20                 ip[3], ip[2], ip[1], ip[0]);
  21 }
  22
  23 static void mkptr6(char *s, const unsigned char *ip)
  24 {
  25         static const char xdigits[] = "0123456789abcdef";
  26         int i;
  27         for (i=15; i>=0; i--) {
  28                 *s++ = xdigits[ip[i]&15]; *s++ = '.';
  29                 *s++ = xdigits[ip[i]>>4]; *s++ = '.';
  30         }
  31         strcpy(s, "ip6.arpa");
  32 }
  33
  34 static int dns_parse_callback(void *c, int rr, const void *data, int len, const void *packet)
  35 {
  36         char tmp[256];
  37         if (rr != RR_PTR) return 0;
  38         if (__dn_expand(packet, (const unsigned char *)packet + 512,
  39             data, tmp, sizeof tmp) > 0)
  40                 strcpy(c, tmp);
  41         return 0;
  42
  43 }
  44
  45 int getnameinfo(const struct sockaddr *restrict sa, socklen_t sl,
  46         char *restrict node, socklen_t nodelen,
  47         char *restrict serv, socklen_t servlen,
  48         int flags)
  49 {
  50         char ptr[PTR_MAX];
  51         char buf[256];
  52         int af = sa->sa_family;
  53         unsigned char *a;
  54
  55         switch (af) {
  56         case AF_INET:
  57                 a = (void *)&((struct sockaddr_in *)sa)->sin_addr;
  58                 if (sl != sizeof(struct sockaddr_in)) return EAI_FAMILY;
  59                 mkptr4(ptr, a);
  60                 break;
  61         case AF_INET6:
  62                 a = (void *)&((struct sockaddr_in6 *)sa)->sin6_addr;
  63                 if (sl != sizeof(struct sockaddr_in6)) return EAI_FAMILY;
  64                 if (memcmp(a, "\0\0\0\0\0\0\0\0\0\0\xff\xff", 12))
  65                         mkptr6(ptr, a);
  66                 else
  67                         mkptr4(ptr, a+12);
  68                 break;
  69         default:
  70                 return EAI_FAMILY;
  71         }
  72
  73         if (node && nodelen) {
  74                 buf[0] = 0;
  75                 if (!(flags & NI_NUMERICHOST)) {
  76                         unsigned char query[18+PTR_MAX], reply[512];
  77                         int qlen = __res_mkquery(0, ptr, 1, RR_PTR,
  78                                 0, 0, 0, query, sizeof query);
  79                         int rlen = __res_send(query, qlen, reply, sizeof reply);
  80                         buf[0] = 0;
  81                         if (rlen > 0)
  82                                 __dns_parse(reply, rlen, dns_parse_callback, buf);
  83                 }
  84                 if (!*buf) {
  85                         if (flags & NI_NAMEREQD) return EAI_NONAME;
  86                         inet_ntop(af, a, buf, sizeof buf);
  87                 }
  88                 if (strlen(buf) >= nodelen) return EAI_OVERFLOW;
  89                 strcpy(node, buf);
  90         }
  91
  92         if (serv && servlen) {
  93                 if (snprintf(buf, sizeof buf, "%d",
  94                         ntohs(((struct sockaddr_in *)sa)->sin_port))>=servlen)
  95                         return EAI_OVERFLOW;
  96                 strcpy(serv, buf);
  97         }
  98
  99         return 0;
 100 }