mirror of https://github.com/lukechilds/node.git
Jérémy Lal
15 years ago
committed by
Ryan Dahl
4 changed files with 196 additions and 0 deletions
@ -0,0 +1,170 @@ |
|||||
|
|
||||
|
/* Copyright 1998 by the Massachusetts Institute of Technology.
|
||||
|
* Copyright (C) 2010 Jeremy Lal <kapouer@melix.org> |
||||
|
* |
||||
|
* Permission to use, copy, modify, and distribute this |
||||
|
* software and its documentation for any purpose and without |
||||
|
* fee is hereby granted, provided that the above copyright |
||||
|
* notice appear in all copies and that both that copyright |
||||
|
* notice and this permission notice appear in supporting |
||||
|
* documentation, and that the name of M.I.T. not be used in |
||||
|
* advertising or publicity pertaining to distribution of the |
||||
|
* software without specific, written prior permission. |
||||
|
* M.I.T. makes no representations about the suitability of |
||||
|
* this software for any purpose. It is provided "as is" |
||||
|
* without express or implied warranty. |
||||
|
*/ |
||||
|
|
||||
|
#include "ares_setup.h" |
||||
|
|
||||
|
#ifdef HAVE_SYS_SOCKET_H |
||||
|
# include <sys/socket.h> |
||||
|
#endif |
||||
|
#ifdef HAVE_NETINET_IN_H |
||||
|
# include <netinet/in.h> |
||||
|
#endif |
||||
|
#ifdef HAVE_NETDB_H |
||||
|
# include <netdb.h> |
||||
|
#endif |
||||
|
#ifdef HAVE_ARPA_INET_H |
||||
|
# include <arpa/inet.h> |
||||
|
#endif |
||||
|
#ifdef HAVE_ARPA_NAMESER_H |
||||
|
# include <arpa/nameser.h> |
||||
|
#else |
||||
|
# include "nameser.h" |
||||
|
#endif |
||||
|
#ifdef HAVE_ARPA_NAMESER_COMPAT_H |
||||
|
# include <arpa/nameser_compat.h> |
||||
|
#endif |
||||
|
|
||||
|
#include <stdlib.h> |
||||
|
#include <string.h> |
||||
|
#include "ares.h" |
||||
|
#include "ares_dns.h" |
||||
|
#include "ares_data.h" |
||||
|
#include "ares_private.h" |
||||
|
|
||||
|
int |
||||
|
ares_parse_mx_reply (const unsigned char *abuf, int alen, |
||||
|
struct ares_mx_reply **mx_out) |
||||
|
{ |
||||
|
unsigned int qdcount, ancount, i; |
||||
|
const unsigned char *aptr, *vptr; |
||||
|
int status, rr_type, rr_class, rr_len; |
||||
|
long len; |
||||
|
char *hostname = NULL, *rr_name = NULL; |
||||
|
struct ares_mx_reply *mx_head = NULL; |
||||
|
struct ares_mx_reply *mx_last = NULL; |
||||
|
struct ares_mx_reply *mx_curr; |
||||
|
|
||||
|
/* Set *mx_out to NULL for all failure cases. */ |
||||
|
*mx_out = NULL; |
||||
|
|
||||
|
/* Give up if abuf doesn't have room for a header. */ |
||||
|
if (alen < HFIXEDSZ) |
||||
|
return ARES_EBADRESP; |
||||
|
|
||||
|
/* Fetch the question and answer count from the header. */ |
||||
|
qdcount = DNS_HEADER_QDCOUNT (abuf); |
||||
|
ancount = DNS_HEADER_ANCOUNT (abuf); |
||||
|
if (qdcount != 1) |
||||
|
return ARES_EBADRESP; |
||||
|
if (ancount == 0) |
||||
|
return ARES_ENODATA; |
||||
|
|
||||
|
/* Expand the name from the question, and skip past the question. */ |
||||
|
aptr = abuf + HFIXEDSZ; |
||||
|
status = ares_expand_name (aptr, abuf, alen, &hostname, &len); |
||||
|
if (status != ARES_SUCCESS) |
||||
|
return status; |
||||
|
|
||||
|
if (aptr + len + QFIXEDSZ > abuf + alen) |
||||
|
{ |
||||
|
free (hostname); |
||||
|
return ARES_EBADRESP; |
||||
|
} |
||||
|
aptr += len + QFIXEDSZ; |
||||
|
|
||||
|
/* Examine each answer resource record (RR) in turn. */ |
||||
|
for (i = 0; i < ancount; i++) |
||||
|
{ |
||||
|
/* Decode the RR up to the data field. */ |
||||
|
status = ares_expand_name (aptr, abuf, alen, &rr_name, &len); |
||||
|
if (status != ARES_SUCCESS) |
||||
|
{ |
||||
|
break; |
||||
|
} |
||||
|
aptr += len; |
||||
|
if (aptr + RRFIXEDSZ > abuf + alen) |
||||
|
{ |
||||
|
status = ARES_EBADRESP; |
||||
|
break; |
||||
|
} |
||||
|
rr_type = DNS_RR_TYPE (aptr); |
||||
|
rr_class = DNS_RR_CLASS (aptr); |
||||
|
rr_len = DNS_RR_LEN (aptr); |
||||
|
aptr += RRFIXEDSZ; |
||||
|
|
||||
|
/* Check if we are really looking at a MX record */ |
||||
|
if (rr_class == C_IN && rr_type == T_MX) |
||||
|
{ |
||||
|
/* parse the MX record itself */ |
||||
|
if (rr_len < 2) |
||||
|
{ |
||||
|
status = ARES_EBADRESP; |
||||
|
break; |
||||
|
} |
||||
|
|
||||
|
/* Allocate storage for this MX answer appending it to the list */ |
||||
|
mx_curr = ares_malloc_data(ARES_DATATYPE_MX_REPLY); |
||||
|
if (!mx_curr) |
||||
|
{ |
||||
|
status = ARES_ENOMEM; |
||||
|
break; |
||||
|
} |
||||
|
if (mx_last) |
||||
|
{ |
||||
|
mx_last->next = mx_curr; |
||||
|
} |
||||
|
else |
||||
|
{ |
||||
|
mx_head = mx_curr; |
||||
|
} |
||||
|
mx_last = mx_curr; |
||||
|
|
||||
|
vptr = aptr; |
||||
|
mx_curr->priority = ntohs (*((unsigned short *)vptr)); |
||||
|
vptr += sizeof(unsigned short); |
||||
|
|
||||
|
status = ares_expand_name (vptr, abuf, alen, &mx_curr->host, &len); |
||||
|
if (status != ARES_SUCCESS) |
||||
|
break; |
||||
|
} |
||||
|
|
||||
|
/* Don't lose memory in the next iteration */ |
||||
|
free (rr_name); |
||||
|
rr_name = NULL; |
||||
|
|
||||
|
/* Move on to the next record */ |
||||
|
aptr += rr_len; |
||||
|
} |
||||
|
|
||||
|
if (hostname) |
||||
|
free (hostname); |
||||
|
if (rr_name) |
||||
|
free (rr_name); |
||||
|
|
||||
|
/* clean up on error */ |
||||
|
if (status != ARES_SUCCESS) |
||||
|
{ |
||||
|
if (mx_head) |
||||
|
ares_free_data (mx_head); |
||||
|
return status; |
||||
|
} |
||||
|
|
||||
|
/* everything looks fine, return the data */ |
||||
|
*mx_out = mx_head; |
||||
|
|
||||
|
return ARES_SUCCESS; |
||||
|
} |
Loading…
Reference in new issue