2003-10-07 17:54:04 -04:00
|
|
|
/* Copyright 1998 by the Massachusetts Institute of Technology.
|
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
|
2004-07-22 18:18:45 -04:00
|
|
|
#include "setup.h"
|
2003-10-07 17:54:04 -04:00
|
|
|
#include <sys/types.h>
|
|
|
|
|
2004-08-20 09:45:26 -04:00
|
|
|
#if defined(WIN32) && !defined(WATT32)
|
2003-10-07 17:54:04 -04:00
|
|
|
#include "nameser.h"
|
2004-08-20 09:45:26 -04:00
|
|
|
|
2003-10-07 17:54:04 -04:00
|
|
|
#else
|
|
|
|
#include <sys/socket.h>
|
2004-08-20 09:45:26 -04:00
|
|
|
#ifdef HAVE_SYS_UIO_H
|
2003-10-07 17:54:04 -04:00
|
|
|
#include <sys/uio.h>
|
2004-08-20 09:45:26 -04:00
|
|
|
#endif
|
2003-10-07 17:54:04 -04:00
|
|
|
#include <netinet/in.h>
|
|
|
|
#include <netdb.h>
|
|
|
|
#include <arpa/nameser.h>
|
2004-08-20 09:45:26 -04:00
|
|
|
#ifdef HAVE_UNISTD_H
|
2003-10-07 17:54:04 -04:00
|
|
|
#include <unistd.h>
|
|
|
|
#endif
|
2004-08-20 09:45:26 -04:00
|
|
|
#endif
|
2003-10-07 17:54:04 -04:00
|
|
|
|
|
|
|
#include <string.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <fcntl.h>
|
|
|
|
#include <time.h>
|
|
|
|
#include <errno.h>
|
2004-08-20 09:45:26 -04:00
|
|
|
|
2003-10-07 17:54:04 -04:00
|
|
|
#include "ares.h"
|
|
|
|
#include "ares_dns.h"
|
|
|
|
#include "ares_private.h"
|
|
|
|
|
2004-08-20 10:07:11 -04:00
|
|
|
#if (defined(WIN32) || defined(WATT32)) && !defined(MSDOS)
|
2004-07-01 09:54:24 -04:00
|
|
|
#define GET_ERRNO() WSAGetLastError()
|
|
|
|
#else
|
|
|
|
#define GET_ERRNO() errno
|
|
|
|
#endif
|
|
|
|
|
2003-10-07 17:54:04 -04:00
|
|
|
static void write_tcp_data(ares_channel channel, fd_set *write_fds,
|
|
|
|
time_t now);
|
|
|
|
static void read_tcp_data(ares_channel channel, fd_set *read_fds, time_t now);
|
|
|
|
static void read_udp_packets(ares_channel channel, fd_set *read_fds,
|
|
|
|
time_t now);
|
|
|
|
static void process_timeouts(ares_channel channel, time_t now);
|
|
|
|
static void process_answer(ares_channel channel, unsigned char *abuf,
|
|
|
|
int alen, int whichserver, int tcp, int now);
|
|
|
|
static void handle_error(ares_channel channel, int whichserver, time_t now);
|
2004-07-24 17:47:49 -04:00
|
|
|
static struct query *next_server(ares_channel channel, struct query *query, time_t now);
|
2003-10-07 17:54:04 -04:00
|
|
|
static int open_tcp_socket(ares_channel channel, struct server_state *server);
|
|
|
|
static int open_udp_socket(ares_channel channel, struct server_state *server);
|
|
|
|
static int same_questions(const unsigned char *qbuf, int qlen,
|
|
|
|
const unsigned char *abuf, int alen);
|
2004-07-24 17:47:49 -04:00
|
|
|
static struct query *end_query(ares_channel channel, struct query *query, int status,
|
2003-10-07 17:54:04 -04:00
|
|
|
unsigned char *abuf, int alen);
|
|
|
|
|
|
|
|
/* Something interesting happened on the wire, or there was a timeout.
|
|
|
|
* See what's up and respond accordingly.
|
|
|
|
*/
|
|
|
|
void ares_process(ares_channel channel, fd_set *read_fds, fd_set *write_fds)
|
|
|
|
{
|
|
|
|
time_t now;
|
|
|
|
|
|
|
|
time(&now);
|
|
|
|
write_tcp_data(channel, write_fds, now);
|
|
|
|
read_tcp_data(channel, read_fds, now);
|
|
|
|
read_udp_packets(channel, read_fds, now);
|
|
|
|
process_timeouts(channel, now);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* If any TCP sockets select true for writing, write out queued data
|
|
|
|
* we have for them.
|
|
|
|
*/
|
|
|
|
static void write_tcp_data(ares_channel channel, fd_set *write_fds, time_t now)
|
|
|
|
{
|
|
|
|
struct server_state *server;
|
|
|
|
struct send_request *sendreq;
|
|
|
|
struct iovec *vec;
|
2004-07-22 18:18:45 -04:00
|
|
|
int i;
|
2004-07-29 03:24:39 -04:00
|
|
|
ssize_t scount;
|
|
|
|
int wcount;
|
2004-07-22 18:18:45 -04:00
|
|
|
size_t n;
|
2003-10-07 17:54:04 -04:00
|
|
|
|
|
|
|
for (i = 0; i < channel->nservers; i++)
|
|
|
|
{
|
|
|
|
/* Make sure server has data to send and is selected in write_fds. */
|
|
|
|
server = &channel->servers[i];
|
2004-07-22 18:18:45 -04:00
|
|
|
if (!server->qhead || server->tcp_socket == ARES_SOCKET_BAD
|
2003-10-07 17:54:04 -04:00
|
|
|
|| !FD_ISSET(server->tcp_socket, write_fds))
|
|
|
|
continue;
|
|
|
|
|
|
|
|
/* Count the number of send queue items. */
|
|
|
|
n = 0;
|
|
|
|
for (sendreq = server->qhead; sendreq; sendreq = sendreq->next)
|
|
|
|
n++;
|
|
|
|
|
|
|
|
/* Allocate iovecs so we can send all our data at once. */
|
|
|
|
vec = malloc(n * sizeof(struct iovec));
|
|
|
|
if (vec)
|
|
|
|
{
|
|
|
|
/* Fill in the iovecs and send. */
|
|
|
|
n = 0;
|
|
|
|
for (sendreq = server->qhead; sendreq; sendreq = sendreq->next)
|
|
|
|
{
|
|
|
|
vec[n].iov_base = (char *) sendreq->data;
|
|
|
|
vec[n].iov_len = sendreq->len;
|
|
|
|
n++;
|
|
|
|
}
|
2004-07-29 03:24:39 -04:00
|
|
|
wcount = writev(server->tcp_socket, vec, n);
|
2003-10-07 17:54:04 -04:00
|
|
|
free(vec);
|
2004-07-29 03:24:39 -04:00
|
|
|
if (wcount < 0)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
|
|
|
handle_error(channel, i, now);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Advance the send queue by as many bytes as we sent. */
|
2004-07-29 03:24:39 -04:00
|
|
|
while (wcount)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
|
|
|
sendreq = server->qhead;
|
2004-07-29 03:24:39 -04:00
|
|
|
if ((size_t)wcount >= sendreq->len)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
2004-07-29 03:24:39 -04:00
|
|
|
wcount -= sendreq->len;
|
2003-10-07 17:54:04 -04:00
|
|
|
server->qhead = sendreq->next;
|
|
|
|
if (server->qhead == NULL)
|
|
|
|
server->qtail = NULL;
|
|
|
|
free(sendreq);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2004-07-29 03:24:39 -04:00
|
|
|
sendreq->data += wcount;
|
|
|
|
sendreq->len -= wcount;
|
2003-10-07 17:54:04 -04:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* Can't allocate iovecs; just send the first request. */
|
|
|
|
sendreq = server->qhead;
|
|
|
|
|
2004-07-29 03:24:39 -04:00
|
|
|
scount = send(server->tcp_socket, sendreq->data, sendreq->len, 0);
|
2003-10-07 17:54:04 -04:00
|
|
|
|
2004-07-29 03:24:39 -04:00
|
|
|
if (scount < 0)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
|
|
|
handle_error(channel, i, now);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Advance the send queue by as many bytes as we sent. */
|
2004-07-29 03:24:39 -04:00
|
|
|
if ((size_t)scount == sendreq->len)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
|
|
|
server->qhead = sendreq->next;
|
|
|
|
if (server->qhead == NULL)
|
|
|
|
server->qtail = NULL;
|
|
|
|
free(sendreq);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2004-07-29 03:24:39 -04:00
|
|
|
sendreq->data += scount;
|
|
|
|
sendreq->len -= scount;
|
2003-10-07 17:54:04 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* If any TCP socket selects true for reading, read some data,
|
|
|
|
* allocate a buffer if we finish reading the length word, and process
|
|
|
|
* a packet if we finish reading one.
|
|
|
|
*/
|
|
|
|
static void read_tcp_data(ares_channel channel, fd_set *read_fds, time_t now)
|
|
|
|
{
|
|
|
|
struct server_state *server;
|
|
|
|
int i, count;
|
|
|
|
|
|
|
|
for (i = 0; i < channel->nservers; i++)
|
|
|
|
{
|
|
|
|
/* Make sure the server has a socket and is selected in read_fds. */
|
|
|
|
server = &channel->servers[i];
|
2004-07-22 18:18:45 -04:00
|
|
|
if (server->tcp_socket == ARES_SOCKET_BAD ||
|
|
|
|
!FD_ISSET(server->tcp_socket, read_fds))
|
2003-10-07 17:54:04 -04:00
|
|
|
continue;
|
|
|
|
|
|
|
|
if (server->tcp_lenbuf_pos != 2)
|
|
|
|
{
|
|
|
|
/* We haven't yet read a length word, so read that (or
|
|
|
|
* what's left to read of it).
|
|
|
|
*/
|
|
|
|
count = recv(server->tcp_socket,
|
|
|
|
server->tcp_lenbuf + server->tcp_buffer_pos,
|
|
|
|
2 - server->tcp_buffer_pos, 0);
|
|
|
|
if (count <= 0)
|
|
|
|
{
|
|
|
|
handle_error(channel, i, now);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
server->tcp_lenbuf_pos += count;
|
|
|
|
if (server->tcp_lenbuf_pos == 2)
|
|
|
|
{
|
|
|
|
/* We finished reading the length word. Decode the
|
|
|
|
* length and allocate a buffer for the data.
|
|
|
|
*/
|
|
|
|
server->tcp_length = server->tcp_lenbuf[0] << 8
|
|
|
|
| server->tcp_lenbuf[1];
|
|
|
|
server->tcp_buffer = malloc(server->tcp_length);
|
|
|
|
if (!server->tcp_buffer)
|
|
|
|
handle_error(channel, i, now);
|
|
|
|
server->tcp_buffer_pos = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* Read data into the allocated buffer. */
|
|
|
|
count = recv(server->tcp_socket,
|
|
|
|
server->tcp_buffer + server->tcp_buffer_pos,
|
|
|
|
server->tcp_length - server->tcp_buffer_pos, 0);
|
|
|
|
if (count <= 0)
|
|
|
|
{
|
|
|
|
handle_error(channel, i, now);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
server->tcp_buffer_pos += count;
|
|
|
|
if (server->tcp_buffer_pos == server->tcp_length)
|
|
|
|
{
|
|
|
|
/* We finished reading this answer; process it and
|
|
|
|
* prepare to read another length word.
|
|
|
|
*/
|
|
|
|
process_answer(channel, server->tcp_buffer, server->tcp_length,
|
|
|
|
i, 1, now);
|
2004-08-20 09:45:26 -04:00
|
|
|
if (server->tcp_buffer)
|
|
|
|
free(server->tcp_buffer);
|
2003-10-07 17:54:04 -04:00
|
|
|
server->tcp_buffer = NULL;
|
|
|
|
server->tcp_lenbuf_pos = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* If any UDP sockets select true for reading, process them. */
|
|
|
|
static void read_udp_packets(ares_channel channel, fd_set *read_fds,
|
|
|
|
time_t now)
|
|
|
|
{
|
|
|
|
struct server_state *server;
|
|
|
|
int i, count;
|
|
|
|
unsigned char buf[PACKETSZ + 1];
|
|
|
|
|
|
|
|
for (i = 0; i < channel->nservers; i++)
|
|
|
|
{
|
|
|
|
/* Make sure the server has a socket and is selected in read_fds. */
|
|
|
|
server = &channel->servers[i];
|
|
|
|
|
2004-07-22 18:18:45 -04:00
|
|
|
if (server->udp_socket == ARES_SOCKET_BAD ||
|
|
|
|
!FD_ISSET(server->udp_socket, read_fds))
|
2003-10-07 17:54:04 -04:00
|
|
|
continue;
|
|
|
|
|
|
|
|
count = recv(server->udp_socket, buf, sizeof(buf), 0);
|
|
|
|
if (count <= 0)
|
|
|
|
handle_error(channel, i, now);
|
|
|
|
|
|
|
|
process_answer(channel, buf, count, i, 0, now);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* If any queries have timed out, note the timeout and move them on. */
|
|
|
|
static void process_timeouts(ares_channel channel, time_t now)
|
|
|
|
{
|
|
|
|
struct query *query, *next;
|
|
|
|
|
|
|
|
for (query = channel->queries; query; query = next)
|
|
|
|
{
|
|
|
|
next = query->next;
|
|
|
|
if (query->timeout != 0 && now >= query->timeout)
|
|
|
|
{
|
|
|
|
query->error_status = ARES_ETIMEOUT;
|
2004-07-24 17:47:49 -04:00
|
|
|
next = next_server(channel, query, now);
|
2003-10-07 17:54:04 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Handle an answer from a server. */
|
|
|
|
static void process_answer(ares_channel channel, unsigned char *abuf,
|
|
|
|
int alen, int whichserver, int tcp, int now)
|
|
|
|
{
|
|
|
|
int id, tc, rcode;
|
|
|
|
struct query *query;
|
|
|
|
|
|
|
|
/* If there's no room in the answer for a header, we can't do much
|
|
|
|
* with it. */
|
|
|
|
if (alen < HFIXEDSZ)
|
|
|
|
return;
|
|
|
|
|
|
|
|
/* Grab the query ID, truncate bit, and response code from the packet. */
|
|
|
|
id = DNS_HEADER_QID(abuf);
|
|
|
|
tc = DNS_HEADER_TC(abuf);
|
|
|
|
rcode = DNS_HEADER_RCODE(abuf);
|
|
|
|
|
|
|
|
/* Find the query corresponding to this packet. */
|
|
|
|
for (query = channel->queries; query; query = query->next)
|
|
|
|
{
|
|
|
|
if (query->qid == id)
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (!query)
|
|
|
|
return;
|
|
|
|
|
|
|
|
/* If we got a truncated UDP packet and are not ignoring truncation,
|
|
|
|
* don't accept the packet, and switch the query to TCP if we hadn't
|
|
|
|
* done so already.
|
|
|
|
*/
|
|
|
|
if ((tc || alen > PACKETSZ) && !tcp && !(channel->flags & ARES_FLAG_IGNTC))
|
|
|
|
{
|
|
|
|
if (!query->using_tcp)
|
|
|
|
{
|
|
|
|
query->using_tcp = 1;
|
|
|
|
ares__send_query(channel, query, now);
|
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Limit alen to PACKETSZ if we aren't using TCP (only relevant if we
|
|
|
|
* are ignoring truncation.
|
|
|
|
*/
|
|
|
|
if (alen > PACKETSZ && !tcp)
|
|
|
|
alen = PACKETSZ;
|
|
|
|
|
|
|
|
/* If we aren't passing through all error packets, discard packets
|
|
|
|
* with SERVFAIL, NOTIMP, or REFUSED response codes.
|
|
|
|
*/
|
|
|
|
if (!(channel->flags & ARES_FLAG_NOCHECKRESP))
|
|
|
|
{
|
|
|
|
if (rcode == SERVFAIL || rcode == NOTIMP || rcode == REFUSED)
|
|
|
|
{
|
|
|
|
query->skip_server[whichserver] = 1;
|
|
|
|
if (query->server == whichserver)
|
|
|
|
next_server(channel, query, now);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
if (!same_questions(query->qbuf, query->qlen, abuf, alen))
|
|
|
|
{
|
|
|
|
if (query->server == whichserver)
|
|
|
|
next_server(channel, query, now);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
end_query(channel, query, ARES_SUCCESS, abuf, alen);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void handle_error(ares_channel channel, int whichserver, time_t now)
|
|
|
|
{
|
2004-07-24 17:47:49 -04:00
|
|
|
struct query *query, *next;
|
2003-10-07 17:54:04 -04:00
|
|
|
|
|
|
|
/* Reset communications with this server. */
|
|
|
|
ares__close_sockets(&channel->servers[whichserver]);
|
|
|
|
|
|
|
|
/* Tell all queries talking to this server to move on and not try
|
|
|
|
* this server again.
|
|
|
|
*/
|
2004-07-24 17:47:49 -04:00
|
|
|
|
|
|
|
for (query = channel->queries; query; query = next)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
2004-07-24 17:47:49 -04:00
|
|
|
next = query->next;
|
2003-10-07 17:54:04 -04:00
|
|
|
if (query->server == whichserver)
|
|
|
|
{
|
|
|
|
query->skip_server[whichserver] = 1;
|
2004-07-24 17:47:49 -04:00
|
|
|
next = next_server(channel, query, now);
|
2003-10-07 17:54:04 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2004-07-24 17:47:49 -04:00
|
|
|
static struct query *next_server(ares_channel channel, struct query *query, time_t now)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
|
|
|
/* Advance to the next server or try. */
|
|
|
|
query->server++;
|
|
|
|
for (; query->try < channel->tries; query->try++)
|
|
|
|
{
|
|
|
|
for (; query->server < channel->nservers; query->server++)
|
|
|
|
{
|
|
|
|
if (!query->skip_server[query->server])
|
|
|
|
{
|
|
|
|
ares__send_query(channel, query, now);
|
2004-07-24 17:47:49 -04:00
|
|
|
return (query->next);
|
2003-10-07 17:54:04 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
query->server = 0;
|
|
|
|
|
|
|
|
/* Only one try if we're using TCP. */
|
|
|
|
if (query->using_tcp)
|
|
|
|
break;
|
|
|
|
}
|
2004-07-24 17:47:49 -04:00
|
|
|
return end_query(channel, query, query->error_status, NULL, 0);
|
2003-10-07 17:54:04 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
void ares__send_query(ares_channel channel, struct query *query, time_t now)
|
|
|
|
{
|
|
|
|
struct send_request *sendreq;
|
|
|
|
struct server_state *server;
|
|
|
|
|
|
|
|
server = &channel->servers[query->server];
|
|
|
|
if (query->using_tcp)
|
|
|
|
{
|
|
|
|
/* Make sure the TCP socket for this server is set up and queue
|
|
|
|
* a send request.
|
|
|
|
*/
|
2004-07-22 18:18:45 -04:00
|
|
|
if (server->tcp_socket == ARES_SOCKET_BAD)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
|
|
|
if (open_tcp_socket(channel, server) == -1)
|
|
|
|
{
|
|
|
|
query->skip_server[query->server] = 1;
|
|
|
|
next_server(channel, query, now);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
2004-07-24 17:47:49 -04:00
|
|
|
sendreq = calloc(sizeof(struct send_request), 1);
|
2003-10-07 17:54:04 -04:00
|
|
|
if (!sendreq)
|
2004-07-24 17:47:49 -04:00
|
|
|
{
|
2003-10-07 17:54:04 -04:00
|
|
|
end_query(channel, query, ARES_ENOMEM, NULL, 0);
|
2004-07-24 17:47:49 -04:00
|
|
|
return;
|
|
|
|
}
|
2003-10-07 17:54:04 -04:00
|
|
|
sendreq->data = query->tcpbuf;
|
|
|
|
sendreq->len = query->tcplen;
|
|
|
|
sendreq->next = NULL;
|
|
|
|
if (server->qtail)
|
|
|
|
server->qtail->next = sendreq;
|
|
|
|
else
|
|
|
|
server->qhead = sendreq;
|
|
|
|
server->qtail = sendreq;
|
|
|
|
query->timeout = 0;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2004-07-22 18:18:45 -04:00
|
|
|
if (server->udp_socket == ARES_SOCKET_BAD)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
|
|
|
if (open_udp_socket(channel, server) == -1)
|
|
|
|
{
|
|
|
|
query->skip_server[query->server] = 1;
|
|
|
|
next_server(channel, query, now);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (send(server->udp_socket, query->qbuf, query->qlen, 0) == -1)
|
|
|
|
{
|
|
|
|
query->skip_server[query->server] = 1;
|
|
|
|
next_server(channel, query, now);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
query->timeout = now
|
|
|
|
+ ((query->try == 0) ? channel->timeout
|
|
|
|
: channel->timeout << query->try / channel->nservers);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int open_tcp_socket(ares_channel channel, struct server_state *server)
|
|
|
|
{
|
2004-07-22 18:18:45 -04:00
|
|
|
ares_socket_t s;
|
|
|
|
int flags;
|
2004-02-04 05:23:15 -05:00
|
|
|
struct sockaddr_in sockin;
|
2003-10-07 17:54:04 -04:00
|
|
|
|
|
|
|
/* Acquire a socket. */
|
|
|
|
s = socket(AF_INET, SOCK_STREAM, 0);
|
2004-07-22 18:18:45 -04:00
|
|
|
if (s == ARES_SOCKET_BAD)
|
2003-10-07 17:54:04 -04:00
|
|
|
return -1;
|
|
|
|
|
|
|
|
/* Set the socket non-blocking. */
|
|
|
|
|
2004-07-01 02:58:47 -04:00
|
|
|
#if defined(WIN32) || defined(WATT32)
|
2003-10-07 17:54:04 -04:00
|
|
|
flags = 1;
|
|
|
|
ioctlsocket(s, FIONBIO, &flags);
|
|
|
|
#else
|
2004-01-29 06:23:36 -05:00
|
|
|
flags = fcntl(s, F_GETFL, 0);
|
|
|
|
|
|
|
|
if (flags == -1)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
2004-07-22 18:18:45 -04:00
|
|
|
closesocket(s);
|
2003-10-07 17:54:04 -04:00
|
|
|
return -1;
|
|
|
|
}
|
2004-01-29 06:23:36 -05:00
|
|
|
flags |= O_NONBLOCK;
|
2003-10-07 17:54:04 -04:00
|
|
|
if (fcntl(s, F_SETFL, flags) == -1)
|
|
|
|
{
|
2004-07-22 18:18:45 -04:00
|
|
|
closesocket(s);
|
2003-10-07 17:54:04 -04:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
#endif
|
2004-07-01 09:54:24 -04:00
|
|
|
|
2003-10-07 17:54:04 -04:00
|
|
|
/* Connect to the server. */
|
2004-02-04 05:23:15 -05:00
|
|
|
memset(&sockin, 0, sizeof(sockin));
|
|
|
|
sockin.sin_family = AF_INET;
|
|
|
|
sockin.sin_addr = server->addr;
|
|
|
|
sockin.sin_port = channel->tcp_port;
|
2004-07-01 09:54:24 -04:00
|
|
|
if (connect(s, (struct sockaddr *) &sockin, sizeof(sockin)) == -1) {
|
|
|
|
int err = GET_ERRNO();
|
|
|
|
|
|
|
|
if (err != EINPROGRESS && err != EWOULDBLOCK) {
|
2003-10-07 17:54:04 -04:00
|
|
|
closesocket(s);
|
|
|
|
return -1;
|
|
|
|
}
|
2004-07-01 09:54:24 -04:00
|
|
|
}
|
2003-10-07 17:54:04 -04:00
|
|
|
|
2004-07-24 17:47:49 -04:00
|
|
|
server->tcp_buffer_pos = 0;
|
2003-10-07 17:54:04 -04:00
|
|
|
server->tcp_socket = s;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int open_udp_socket(ares_channel channel, struct server_state *server)
|
|
|
|
{
|
2004-07-22 18:18:45 -04:00
|
|
|
ares_socket_t s;
|
2004-02-04 05:23:15 -05:00
|
|
|
struct sockaddr_in sockin;
|
2003-10-07 17:54:04 -04:00
|
|
|
|
|
|
|
/* Acquire a socket. */
|
|
|
|
s = socket(AF_INET, SOCK_DGRAM, 0);
|
2004-07-22 18:18:45 -04:00
|
|
|
if (s == ARES_SOCKET_BAD)
|
2003-10-07 17:54:04 -04:00
|
|
|
return -1;
|
|
|
|
|
|
|
|
/* Connect to the server. */
|
2004-02-04 05:23:15 -05:00
|
|
|
memset(&sockin, 0, sizeof(sockin));
|
|
|
|
sockin.sin_family = AF_INET;
|
|
|
|
sockin.sin_addr = server->addr;
|
|
|
|
sockin.sin_port = channel->udp_port;
|
|
|
|
if (connect(s, (struct sockaddr *) &sockin, sizeof(sockin)) == -1)
|
2003-10-07 17:54:04 -04:00
|
|
|
{
|
|
|
|
closesocket(s);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
server->udp_socket = s;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int same_questions(const unsigned char *qbuf, int qlen,
|
|
|
|
const unsigned char *abuf, int alen)
|
|
|
|
{
|
|
|
|
struct {
|
|
|
|
const unsigned char *p;
|
|
|
|
int qdcount;
|
|
|
|
char *name;
|
2004-02-23 02:52:20 -05:00
|
|
|
long namelen;
|
2003-10-07 17:54:04 -04:00
|
|
|
int type;
|
|
|
|
int dnsclass;
|
|
|
|
} q, a;
|
|
|
|
int i, j;
|
|
|
|
|
|
|
|
if (qlen < HFIXEDSZ || alen < HFIXEDSZ)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
/* Extract qdcount from the request and reply buffers and compare them. */
|
|
|
|
q.qdcount = DNS_HEADER_QDCOUNT(qbuf);
|
|
|
|
a.qdcount = DNS_HEADER_QDCOUNT(abuf);
|
|
|
|
if (q.qdcount != a.qdcount)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
/* For each question in qbuf, find it in abuf. */
|
|
|
|
q.p = qbuf + HFIXEDSZ;
|
|
|
|
for (i = 0; i < q.qdcount; i++)
|
|
|
|
{
|
|
|
|
/* Decode the question in the query. */
|
|
|
|
if (ares_expand_name(q.p, qbuf, qlen, &q.name, &q.namelen)
|
|
|
|
!= ARES_SUCCESS)
|
|
|
|
return 0;
|
|
|
|
q.p += q.namelen;
|
|
|
|
if (q.p + QFIXEDSZ > qbuf + qlen)
|
|
|
|
{
|
|
|
|
free(q.name);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
q.type = DNS_QUESTION_TYPE(q.p);
|
|
|
|
q.dnsclass = DNS_QUESTION_CLASS(q.p);
|
|
|
|
q.p += QFIXEDSZ;
|
|
|
|
|
|
|
|
/* Search for this question in the answer. */
|
|
|
|
a.p = abuf + HFIXEDSZ;
|
|
|
|
for (j = 0; j < a.qdcount; j++)
|
|
|
|
{
|
|
|
|
/* Decode the question in the answer. */
|
|
|
|
if (ares_expand_name(a.p, abuf, alen, &a.name, &a.namelen)
|
|
|
|
!= ARES_SUCCESS)
|
|
|
|
{
|
|
|
|
free(q.name);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
a.p += a.namelen;
|
|
|
|
if (a.p + QFIXEDSZ > abuf + alen)
|
|
|
|
{
|
|
|
|
free(q.name);
|
|
|
|
free(a.name);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
a.type = DNS_QUESTION_TYPE(a.p);
|
|
|
|
a.dnsclass = DNS_QUESTION_CLASS(a.p);
|
|
|
|
a.p += QFIXEDSZ;
|
|
|
|
|
|
|
|
/* Compare the decoded questions. */
|
|
|
|
if (strcasecmp(q.name, a.name) == 0 && q.type == a.type
|
|
|
|
&& q.dnsclass == a.dnsclass)
|
|
|
|
{
|
|
|
|
free(a.name);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
free(a.name);
|
|
|
|
}
|
|
|
|
|
|
|
|
free(q.name);
|
|
|
|
if (j == a.qdcount)
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2004-07-24 17:47:49 -04:00
|
|
|
static struct query *end_query (ares_channel channel, struct query *query, int status,
|
2003-10-07 17:54:04 -04:00
|
|
|
unsigned char *abuf, int alen)
|
|
|
|
{
|
2004-07-24 17:47:49 -04:00
|
|
|
struct query **q, *next;
|
2003-10-07 17:54:04 -04:00
|
|
|
int i;
|
|
|
|
|
|
|
|
query->callback(query->arg, status, abuf, alen);
|
|
|
|
for (q = &channel->queries; *q; q = &(*q)->next)
|
|
|
|
{
|
|
|
|
if (*q == query)
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
*q = query->next;
|
2004-07-24 17:47:49 -04:00
|
|
|
if (*q)
|
|
|
|
next = (*q)->next;
|
|
|
|
else
|
|
|
|
next = NULL;
|
2003-10-07 17:54:04 -04:00
|
|
|
free(query->tcpbuf);
|
|
|
|
free(query->skip_server);
|
|
|
|
free(query);
|
|
|
|
|
|
|
|
/* Simple cleanup policy: if no queries are remaining, close all
|
|
|
|
* network sockets unless STAYOPEN is set.
|
|
|
|
*/
|
|
|
|
if (!channel->queries && !(channel->flags & ARES_FLAG_STAYOPEN))
|
|
|
|
{
|
|
|
|
for (i = 0; i < channel->nservers; i++)
|
|
|
|
ares__close_sockets(&channel->servers[i]);
|
|
|
|
}
|
2004-07-24 17:47:49 -04:00
|
|
|
return (next);
|
2003-10-07 17:54:04 -04:00
|
|
|
}
|