ares_parse_a_reply.c

   1 /* Copyright 1998 by the Massachusetts Institute of Technology.
   2  *
   3  * Permission to use, copy, modify, and distribute this
   4  * software and its documentation for any purpose and without
   5  * fee is hereby granted, provided that the above copyright
   6  * notice appear in all copies and that both that copyright
   7  * notice and this permission notice appear in supporting
   8  * documentation, and that the name of M.I.T. not be used in
   9  * advertising or publicity pertaining to distribution of the
  10  * software without specific, written prior permission.
  11  * M.I.T. makes no representations about the suitability of
  12  * this software for any purpose.  It is provided "as is"
  13  * without express or implied warranty.
  14  */
  15
  16 #include "setup.h"
  17 #include <sys/types.h>
  18
  19 #if defined(WIN32) && !defined(WATT32)
  20 #include "nameser.h"
  21 #else
  22 #include <sys/socket.h>
  23 #include <netinet/in.h>
  24 #include <arpa/inet.h>
  25 #include <netdb.h>
  26 #include <arpa/nameser.h>
  27 #endif
  28
  29 #include <stdlib.h>
  30 #include <string.h>
  31 #include "ares.h"
  32 #include "ares_dns.h"
  33 #include "ares_private.h"
  34
  35 int ares_parse_a_reply(const unsigned char *abuf, int alen,
  36                        struct hostent **host)
  37 {
  38   unsigned int qdcount, ancount;
  39   int status, i, rr_type, rr_class, rr_len, naddrs;
  40   int naliases;
  41   long len;
  42   const unsigned char *aptr;
  43   char *hostname, *rr_name, *rr_data, **aliases;
  44   struct in_addr *addrs;
  45   struct hostent *hostent;
  46
  47   /* Set *host to NULL for all failure cases. */
  48   *host = NULL;
  49
  50   /* Give up if abuf doesn't have room for a header. */
  51   if (alen < HFIXEDSZ)
  52     return ARES_EBADRESP;
  53
  54   /* Fetch the question and answer count from the header. */
  55   qdcount = DNS_HEADER_QDCOUNT(abuf);
  56   ancount = DNS_HEADER_ANCOUNT(abuf);
  57   if (qdcount != 1)
  58     return ARES_EBADRESP;
  59
  60   /* Expand the name from the question, and skip past the question. */
  61   aptr = abuf + HFIXEDSZ;
  62   status = ares_expand_name(aptr, abuf, alen, &hostname, &len);
  63   if (status != ARES_SUCCESS)
  64     return status;
  65   if (aptr + len + QFIXEDSZ > abuf + alen)
  66     {
  67       free(hostname);
  68       return ARES_EBADRESP;
  69     }
  70   aptr += len + QFIXEDSZ;
  71
  72   /* Allocate addresses and aliases; ancount gives an upper bound for both. */
  73   addrs = malloc(ancount * sizeof(struct in_addr));
  74   if (!addrs)
  75     {
  76       free(hostname);
  77       return ARES_ENOMEM;
  78     }
  79   aliases = malloc((ancount + 1) * sizeof(char *));
  80   if (!aliases)
  81     {
  82       free(hostname);
  83       free(addrs);
  84       return ARES_ENOMEM;
  85     }
  86   naddrs = 0;
  87   naliases = 0;
  88
  89   /* Examine each answer resource record (RR) in turn. */
  90   for (i = 0; i < (int)ancount; i++)
  91     {
  92       /* Decode the RR up to the data field. */
  93       status = ares_expand_name(aptr, abuf, alen, &rr_name, &len);
  94       if (status != ARES_SUCCESS)
  95         break;
  96       aptr += len;
  97       if (aptr + RRFIXEDSZ > abuf + alen)
  98         {
  99           status = ARES_EBADRESP;
 100           break;
 101         }
 102       rr_type = DNS_RR_TYPE(aptr);
 103       rr_class = DNS_RR_CLASS(aptr);
 104       rr_len = DNS_RR_LEN(aptr);
 105       aptr += RRFIXEDSZ;
 106
 107       if (rr_class == C_IN && rr_type == T_A
 108           && rr_len == sizeof(struct in_addr)
 109           && strcasecmp(rr_name, hostname) == 0)
 110         {
 111           memcpy(&addrs[naddrs], aptr, sizeof(struct in_addr));
 112           naddrs++;
 113           status = ARES_SUCCESS;
 114         }
 115
 116       if (rr_class == C_IN && rr_type == T_CNAME)
 117         {
 118           /* Record the RR name as an alias. */
 119           aliases[naliases] = rr_name;
 120           naliases++;
 121
 122           /* Decode the RR data and replace the hostname with it. */
 123           status = ares_expand_name(aptr, abuf, alen, &rr_data, &len);
 124           if (status != ARES_SUCCESS)
 125             break;
 126           free(hostname);
 127           hostname = rr_data;
 128         }
 129       else
 130         free(rr_name);
 131
 132       aptr += rr_len;
 133       if (aptr > abuf + alen)
 134         {
 135           status = ARES_EBADRESP;
 136           break;
 137         }
 138     }
 139
 140   if (status == ARES_SUCCESS && naddrs == 0)
 141     status = ARES_ENODATA;
 142   if (status == ARES_SUCCESS)
 143     {
 144       /* We got our answer.  Allocate memory to build the host entry. */
 145       aliases[naliases] = NULL;
 146       hostent = malloc(sizeof(struct hostent));
 147       if (hostent)
 148         {
 149           hostent->h_addr_list = malloc((naddrs + 1) * sizeof(char *));
 150           if (hostent->h_addr_list)
 151             {
 152               /* Fill in the hostent and return successfully. */
 153               hostent->h_name = hostname;
 154               hostent->h_aliases = aliases;
 155               hostent->h_addrtype = AF_INET;
 156               hostent->h_length = sizeof(struct in_addr);
 157               for (i = 0; i < naddrs; i++)
 158                 hostent->h_addr_list[i] = (char *) &addrs[i];
 159               hostent->h_addr_list[naddrs] = NULL;
 160               *host = hostent;
 161               return ARES_SUCCESS;
 162             }
 163           free(hostent);
 164         }
 165       status = ARES_ENOMEM;
 166     }
 167   for (i = 0; i < naliases; i++)
 168     free(aliases[i]);
 169   free(aliases);
 170   free(addrs);
 171   free(hostname);
 172   return status;
 173 }