wraith-ircbot/src/adns.c

/*
 * Copyright (C) 1997 Robey Pointer
 * Copyright (C) 1999 - 2002 Eggheads Development Team
 * Copyright (C) 2002 - 2008 Bryan Drewery
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

/* async dns
 *
 */


#include "common.h"
#include "adns.h"
#include "egg_timer.h"
#include "main.h"
#include "net.h"
#include "misc.h"
#include "socket.h"
#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <bdlib/src/Array.h>
#include <bdlib/src/String.h>

typedef struct dns_query {
	struct dns_query *next;
	bd::Array<bd::String>* answer;
	dns_callback_t callback;
	void *client_data;
	time_t expiretime;
	char *query;
	char *ip;
	int id;
//	int timer_id;
	int answers;
	int remaining;
        int lowest_ttl;
} dns_query_t;

/* RFC1035
                                    1  1  1  1  1  1
      0  1  2  3  4  5  6  7  8  9  0  1  2  3  4  5
      1  1  1  1  1  0  9  8  7  6  5  4  3  2  1  0
      5  4  3  2  1  1
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                      ID                       |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |QR|   Opcode  |AA|TC|RD|RA|   Z    |   RCODE   |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                    QDCOUNT                    |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                    ANCOUNT                    |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                    NSCOUNT                    |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                    ARCOUNT                    |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
*/

typedef struct {
	unsigned short id;
	unsigned short flags;
	unsigned short question_count;
	unsigned short answer_count;
	unsigned short ns_count;
	unsigned short ar_count;
} dns_header_t;

#define GET_QR(x)     (((x) >> 15) & BIT0)
#define GET_OPCODE(x) (((x) >> 11) & (BIT3|BIT2|BIT1|BIT0))
#define GET_AA(x)     (((x) >> 10) & BIT0)
#define GET_TC(x)     (((x) >> 9)  & BIT0)
#define GET_RD(x)     (((x) >> 8)  & BIT0)
#define GET_RA(x)     (((x) >> 7)  & BIT0)
#define GET_RCODE(x)  ((x)         & (BIT3|BIT2|BIT1|BIT0))

#define SET_RD(x) (x) |= ((x) | (1 << 8))

#define HEAD_SIZE 12

/* RFC1035
                                    1  1  1  1  1  1
      0  1  2  3  4  5  6  7  8  9  0  1  2  3  4  5
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                                               |
    /                                               /
    /                      NAME                     /
    |                                               |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                      TYPE                     |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                     CLASS                     |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                      TTL                      |
    |                                               |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
    |                   RDLENGTH                    |
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--|
    /                     RDATA                     /
    /                                               /
    +--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
*/

typedef struct {
	/* char name[]; */
	unsigned short type;
	unsigned short dclass;
	int ttl;
	unsigned short rdlength;
	/* char rdata[]; */
} dns_rr_t;

#define RR_SIZE 10

/* Entries from resolv.conf */
typedef struct dns_server {
	char *ip;
	int idx;
} dns_server_t;

/* Entries from hosts */
typedef struct {
	char *host;
        char *ip;
} dns_host_t;

typedef struct {
	bd::Array<bd::String>* answer;
	char *query;
	time_t expiretime;
} dns_cache_t;


static dns_header_t _dns_header = {0, 0, 0, 0, 0, 0};
static dns_query_t *query_head = NULL;
static dns_host_t *hosts = NULL;
static int nhosts = 0;
static dns_cache_t *cache = NULL;
static int ncache = 0;
static dns_server_t *servers = NULL;
static int nservers = 0;
static int cur_server = -1;

static char separators[] = " ,\t\r\n";

int dns_idx = -1;
int dns_sock = -1;
const char *dns_ip = NULL;

static int make_header(char *buf, int id);
static int cut_host(const char *host, char *query);
static int read_resolv(char *fname);
static void read_hosts(char *fname);
static int get_dns_idx();
//static void dns_resend_queries();
static int cache_find(const char *);
//static int dns_on_read(void *client_data, int idx, char *buf, int len);
//static int dns_on_eof(void *client_data, int idx, int err, const char *errmsg);
static void dns_read(int idx, char*, int&, bool blocking = 0);
static void dns_on_read(int idx, char *buf, int atr);
static void dns_on_eof(int idx);
static const char *dns_next_server();
static int parse_reply(char *response, size_t nbytes, const char* server_ip, bool blocking = 0);

interval_t async_lookup_timeout = 10;
interval_t async_server_timeout = 20;
//int resend_on_read = 0;

static void 
dns_display(int idx, char *buf, size_t bufsiz)
{
  simple_snprintf(buf, bufsiz, "named   waited %ds", (int) (now - dcc[idx].timeval));
}

static void
dns_reinit(int idx)
{
        sdprintf("Re-opening dns socket...");
        killsock(dcc[idx].sock);
        lostdcc(idx);
	dns_idx = -1;
        dns_sock = -1;
	dns_ip = NULL;

        if (!get_dns_idx())
          sdprintf("Successfully reopened dns socket");
        else
          sdprintf("Failed to reopen dns socket");
}

static void
dns_timeout(int idx)
{
  sdprintf("DNS socket timed out");
  /*egg_dns_cancel(dcc[idx].u.dns_id, 1);*/
//  resend_on_read = 1;
  dns_reinit(idx);  
//  sleep(2);
//  dns_resend_queries();
}

static struct dcc_table dns_handler = {
  "adns",
  DCT_VALIDIDX,
  dns_on_eof,
  dns_on_read,
  NULL,
  dns_timeout,
  dns_display,
  NULL,
  NULL,
  NULL
};

static void dcc_dnswait(int idx, char *buf, int len)
{
  /* Ignore anything now. */
}

static void eof_dcc_dnswait(int idx)
{
  putlog(LOG_MISC, "*", "Lost connection while resolving hostname [%s/%d]",
         iptostr(htonl(dcc[idx].addr)), dcc[idx].port);
  killsock(dcc[idx].sock);
  lostdcc(idx);
}

static void display_dcc_dnswait(int idx, char *buf, size_t bufsiz)
{
  simple_snprintf(buf, bufsiz, "dns   waited %ds", (int) (now - dcc[idx].timeval));
}

static void kill_dcc_dnswait(int idx, void *x)
{
  struct dns_info *p = (struct dns_info *) x;

  if (p) {
    if (p->cbuf)
      free(p->cbuf);

    if (p->cptr)
      free(p->cptr);
    // free(p) is same thing here.
    free(dcc[idx].u.other);
    dcc[idx].u.other = NULL;
  }
}

struct dcc_table DCC_DNSWAIT = {
 "DNSWAIT",
  DCT_VALIDIDX,
  eof_dcc_dnswait,
  dcc_dnswait,
  NULL,
  NULL,
  display_dcc_dnswait,
  kill_dcc_dnswait,
  NULL,
  NULL
};



/*
static void async_timeout(void *client_data)
{
  int id = (int) client_data;
  sdprintf("%d timed out", id);
  egg_dns_cancel(id, 1);
}
*/

char s1_7[3] = "",s2_3[3] = "",s2_2[3] = "";

static void free_query(dns_query_t*& q) {
	if (q->ip)
		free(q->ip);
	if (q->answer)
		delete q->answer;
	if (q->query)
		free(q->query);
	free(q);
	q = NULL;
}

static int get_dns_id() {
	while (1) {
		int id = randint(65534);
		bool found = 0;
		// Make sure this ID is not already in use, avoid a race condition
		for (dns_query_t *q = query_head; q; q = q->next) {
			if (q->id == id) {
				found = 1;
				break;
			}
		}
		if (found) continue;
		return id;
	}
}

static dns_query_t *alloc_query(void *client_data, dns_callback_t callback, const char *query)
{
	dns_query_t *q = (dns_query_t *) my_calloc(1, sizeof(*q));

	q->id = get_dns_id();
	q->query = strdup(query);
	q->answers = 0;
        q->answer = new bd::Array<bd::String>;
        q->lowest_ttl = 0;
	q->callback = callback;
	q->client_data = client_data;
	q->expiretime = now + async_lookup_timeout;
	q->next = query_head;
	query_head = q;

	return q;
}

static void query_transform_ip(dns_query_t* q, const char* ip) {
	/* We need to transform the ip address into the proper form
	 * for reverse lookup. */
	if (strchr(ip, ':')) {
		char temp[128] = "";

		socket_ipv6_to_dots(ip, temp);
		sdprintf("dots: %s", temp);
		size_t iplen = strlen(temp) + 9 + 1;
		q->ip = (char *) my_calloc(1, iplen);
		//		reverse_ip(temp, q->ip);
		strlcat(q->ip, temp, iplen);
		strlcat(q->ip, "ip6.arpa", iplen);
		sdprintf("reversed ipv6 ip: %s", q->ip);
	}
	else {
		size_t iplen = strlen(ip) + 13 + 1;
		q->ip = (char *) my_calloc(1, iplen);
		reverse_ip(ip, q->ip);
		strlcat(q->ip, ".in-addr.arpa", iplen);
	}
}

static int get_dns_idx()
{
	int i, sock;
       
	sock = -1;
	for (i = 0; i < nservers; i++) {
		if (!dns_ip) dns_ip = dns_next_server();
		sock = socket_create(dns_ip, DNS_PORT, NULL, 0, SOCKET_CLIENT | SOCKET_NONBLOCK | SOCKET_UDP);
		if (sock < 0) {
			/* Try the next server. */
			dns_ip = NULL;
		}
		else break;
	}
	if (i == nservers) return 1;
//	dns_idx = sockbuf_new();
//	sockbuf_set_handler(dns_idx, &dns_handler, NULL);
//	sockbuf_set_sock(dns_idx, sock, 0);
//        allocsock(sock, SOCK_CONNECT);

        if (sock >= 0 && dns_ip) {
          dns_idx = new_dcc(&dns_handler, 0);

          if (dns_idx < 0) {
           putlog(LOG_SERV, "*", "NO MORE DCC CONNECTIONS -- Can't create dns connection.");
           killsock(sock);
           return 1;
          }
          sdprintf("dns_idx: %d", dns_idx);
          dcc[dns_idx].sock = sock;
          dns_sock = sock;
          sdprintf("dns_sock: %d", dcc[dns_idx].sock);
          strlcpy(dcc[dns_idx].host, dns_ip, sizeof(dcc[dns_idx].host));
          strlcpy(dcc[dns_idx].nick, "(adns)", sizeof(dcc[dns_idx].nick));
          sdprintf("dns_ip: %s", dns_ip);
          dcc[dns_idx].timeval = now;
          dns_handler.timeout_val = 0;
          return 0;
        }
	return 1;
}

void egg_dns_send(char *query, int len, bool blocking)
{
        if (dns_idx >= 0 && dcc[dns_idx].sock == -1) {
          lostdcc(dns_idx);
          dns_idx = -1;
        }
	if (dns_idx < 0) {
		if (get_dns_idx()) {
                  sdprintf("get_dns_idx() failed in egg_dns_send");
                  return;
                }
	}
        if (!dns_handler.timeout_val) {
          dns_handler.timeout_val = &async_server_timeout;
          sdprintf("SETTING TIMEOUT to %d", async_server_timeout);
          dcc[dns_idx].timeval = now;
        }

		if (blocking)
			socket_set_nonblock(dcc[dns_idx].sock, 0);
        if (write(dcc[dns_idx].sock, query, len) == -1) {
	  ;
	}
//	sockbuf_write(dns_idx, query, len);
}

dns_query_t *find_query(const char *host)
{
  dns_query_t *q = NULL;

  for (q = query_head; q; q = q->next)
    if (!strcasecmp(q->query, host))
      return q;
  return NULL;
}

#define DEFAULT_DNS_TYPE (DNS_LOOKUP_A|DNS_LOOKUP_AAAA)
static void dns_send_query(dns_query_t *q, int type = DEFAULT_DNS_TYPE, bool blocking = 0)
{
  char buf[512] = "";
  int len;

  q->remaining = 0;

  if (!q->ip) {
    if (type & DNS_LOOKUP_A) {
      /* Send the ipv4 query. */
      q->remaining++;
      len = make_header(buf, q->id);
      len += cut_host(q->query, buf + len);
      buf[len] = 0; len++; buf[len] = DNS_A; len++;
      buf[len] = 0; len++; buf[len] = 1; len++;

      egg_dns_send(buf, len, blocking);
    }

#ifdef USE_IPV6
    if (type & DNS_LOOKUP_AAAA) {
      /* Now send the ipv6 query. */
      q->remaining++;
      len = make_header(buf, q->id);
      len += cut_host(q->query, buf + len);
      buf[len] = 0; len++; buf[len] = DNS_AAAA; len++;
      buf[len] = 0; len++; buf[len] = 1; len++;

      egg_dns_send(buf, len, blocking);
    }
#endif
  } else if (q->ip) {
    q->remaining++;
    len = make_header(buf, q->id);
    len += cut_host(q->ip, buf + len);
    buf[len] = 0; len++; buf[len] = DNS_PTR; len++;
    buf[len] = 0; len++; buf[len] = 1; len++;

    egg_dns_send(buf, len, blocking);
  }
}

static void dns_send_query_blocking(dns_query_t* q, int type = DEFAULT_DNS_TYPE) {
	dns_send_query(q, type, 1);
}


/*
void dns_resend_queries()
{
  dns_query_t *q = NULL;

  for (q = query_head; q; q = q->next) {
    if (now >= q->expiretime) {
sdprintf("RESENDING: %s", q->query);
      dns_send_query(q);
    }
  }
}
*/

/*
void dns_create_timeout_timer(dns_query_t **qm, const char *query, int timeout)
{
	dns_query_t *q = *qm;
	egg_timeval_t howlong;

	howlong.sec = timeout;
	howlong.usec = 0;

	q->timer_id = timer_create_complex(&howlong, query, (Function) async_timeout, (void *) q->id, 0);
}
*/

/* Perform an async dns lookup. This is host -> ip. For ip -> host, use
 * egg_dns_reverse(). We return a dns id that you can use to cancel the
 * lookup. */
int egg_dns_lookup(const char *host, interval_t timeout, dns_callback_t callback, void *client_data, int type)
{
	dns_query_t *q = NULL;
	int i, cache_id;

	if (is_dotted_ip(host)) {
		/* If it's already an ip, we're done. */
                bd::Array<bd::String> answer;
		sdprintf("egg_dns_lookup(%s, %d): Already an ip.", host, timeout);

                answer << host;
		callback(-1, client_data, host, answer);
		return(-1);
	}

	/* Ok, now see if it's in our host cache. */
	for (i = 0; i < nhosts; i++) {
		if (!strcasecmp(host, hosts[i].host)) {
			bd::Array<bd::String> answer;

			sdprintf("egg_dns_lookup(%s, %d): Found in hosts -> %s", host, timeout, hosts[i].ip);
                        answer << hosts[i].ip;
			callback(-1, client_data, host, answer);
			return(-1);
		}
	}

	cache_id = cache_find(host);
	if (cache_id >= 0) {
//		cache[cache_id].answer->shuffle();
		sdprintf("egg_dns_lookup(%s, %d): Found in cache -> %s", host, timeout, cache[cache_id].answer->join(',').c_str());
		callback(-1, client_data, host, *(cache[cache_id].answer));
		return(-1);
	}

	/* check if the query was already made */
        if ((q = find_query(host))) {
	  sdprintf("egg_dns_lookup(%s, %d): Already querying -> %d", host, timeout, q->id);
          return(-2);
	}

	/* Allocate our query struct. */
        q = alloc_query(client_data, callback, host);

	sdprintf("egg_dns_lookup(%s, %d) -> %d", host, timeout, q->id);
        dns_send_query(q, type);

//        /* setup a timer to detect dead ns */
//	dns_create_timeout_timer(&q, host, timeout);

	/* Send the ipv4 query. */

	return(q->id);
}

bd::Array<bd::String> dns_send_blocking(dns_query_t* q, interval_t timeout, int type = DEFAULT_DNS_TYPE) {
	bd::Array<bd::String> answer;
	dns_send_query_blocking(q, type);

	char buf[512] = "";
	int atr;
	bool read_error = 0;

	/* Now the key difference, call read() on the socket to block until it returns */
	if (!setjmp(alarmret)) {
read_more:
		alarm(timeout);

		dns_read(dns_idx, buf, atr, 1);
		read_error = errno ? 1 : 0;

		alarm(0);

		if (!read_error) {
			dcc[dns_idx].timeval = now;
			if (parse_reply(buf, atr, dns_ip, 1)) {
				dns_on_eof(dns_idx);
			} else if (errno == EAGAIN) {
				goto read_more;
			} else if (q->answer) {
				answer = *(q->answer);
			}
		}
	}

	if (q) {
		dns_query_t *prev = NULL, *cur = NULL;
		for (cur = query_head; cur; cur = cur->next) {
			if (cur->id == q->id) break;
			prev = cur;
		}
		if (cur) {
			if (prev) prev->next = cur->next;
			else query_head = cur->next;
		}
		free_query(q);
	}

	// Disable blocking on the dns socket.
	socket_set_nonblock(dcc[dns_idx].sock, 1);

	return answer;
}

bd::Array<bd::String> dns_blocking_loop(const char* what, interval_t timeout, const char* from, int type = DEFAULT_DNS_TYPE, bool is_ip = 0) {
	/* Allocate our query struct. */
	dns_query_t *q = NULL;
	bd::Array<bd::String> answer;
	static int cnt = 0;
try_again:
	q = alloc_query(NULL, NULL, what);

	if (is_ip)
		query_transform_ip(q, what);

	sdprintf("%s(%s, %d) -> %d", from, what, timeout, q->id);
	if (cnt++ < nservers) {
		answer = dns_send_blocking(q, timeout, type);
		// No answers received?? Blocking DNS demands an answer, check the next server.
		if (answer.size() == 0 && cnt < nservers) {
			dns_on_eof(dns_idx);
			goto try_again;
		}
	}
	cnt = 0;
	return answer;
}


bd::Array<bd::String> dns_lookup_block(const char *host, interval_t timeout, int type)
{
	if (is_dotted_ip(host)) {
		bd::Array<bd::String> answer;
		/* If it's already an ip, we're done. */
		sdprintf("dns_lookup_block(%s, %d): Already an ip.", host, timeout);
                answer << host;
		return answer;
	}

	/* Ok, now see if it's in our host cache. */
	for (int i = 0; i < nhosts; i++) {
		if (!strcasecmp(host, hosts[i].host)) {
			bd::Array<bd::String> answer;
			sdprintf("dns_lookup_block(%s, %d): Found in hosts -> %s", host, timeout, hosts[i].ip);
                        answer << hosts[i].ip;
                        return answer;
		}
	}

	int cache_id = cache_find(host);
	if (cache_id >= 0) {
//		cache[cache_id].answer->shuffle();
		sdprintf("dns_lookup_block(%s, %d): Found in cache -> %s", host, timeout, cache[cache_id].answer->join(',').c_str());
		return *(cache[cache_id].answer);
	}

	return dns_blocking_loop(host, timeout, "dns_lookup_block", type);
}

/* Perform a blocking dns reverse lookup. This does ip -> host. For host -> ip
 * use egg_dns_lookup(). */
bd::Array<bd::String> dns_reverse_block(const char *ip, interval_t timeout)
{

	if (!is_dotted_ip(ip)) {
		bd::Array<bd::String> answer;
		/* If it's not a valid ip, don't even make the request. */
		sdprintf("dns_reverse_block(%s, %d): Not an ip.", ip, timeout);
		answer << ip;
		return answer;
	}

	/* Ok, see if we have it in our host cache. */
	for (int i = 0; i < nhosts; i++) {
		if (!strcasecmp(hosts[i].ip, ip)) {
			bd::Array<bd::String> answer;
			sdprintf("dns_reverse_block(%s, %d): Found in hosts -> %s", ip, timeout, hosts[i].host);
			answer << hosts[i].host;
			return answer;
		}
	}

	int cache_id = cache_find(ip);
        if (cache_id >= 0) {
//		cache[cache_id].answer->shuffle();
		sdprintf("dns_reverse_block(%s, %d): Found in cache -> %s", ip, timeout, cache[cache_id].answer->join(',').c_str());
		return *(cache[cache_id].answer);
	}

	return dns_blocking_loop(ip, timeout, "dns_reverse_block", 0, 1);
}

/* Perform an async dns reverse lookup. This does ip -> host. For host -> ip
 * use egg_dns_lookup(). We return a dns id that you can use to cancel the
 * lookup. */
int egg_dns_reverse(const char *ip, interval_t timeout, dns_callback_t callback, void *client_data)
{
	dns_query_t *q;
	int i, cache_id;

	if (!is_dotted_ip(ip)) {
		/* If it's not a valid ip, don't even make the request. */
		sdprintf("egg_dns_reverse(%s, %d): Not an ip.", ip, timeout);
		callback(-1, client_data, ip, NULL);
		return(-1);
	}

	/* Ok, see if we have it in our host cache. */
	for (i = 0; i < nhosts; i++) {
		if (!strcasecmp(hosts[i].ip, ip)) {
			bd::Array<bd::String> answer;

			sdprintf("egg_dns_reverse(%s, %d): Found in hosts -> %s", ip, timeout, hosts[i].host);
			answer << hosts[i].host;
			callback(-1, client_data, ip, answer);
			return(-1);
		}
	}

	cache_id = cache_find(ip);
        if (cache_id >= 0) {
//		cache[cache_id].answer->shuffle();
		sdprintf("egg_dns_reverse(%s, %d): Found in cache -> %s", ip, timeout, cache[cache_id].answer->join(',').c_str());
		callback(-1, client_data, ip, *(cache[cache_id].answer));
		return(-1);
	}

	/* check if the query was already made */
        if ((q = find_query(ip))) {
	  sdprintf("egg_dns_reverse(%s, %d): Already querying -> %d", ip, timeout, q->id);
          return(-1);
	}

	q = alloc_query(client_data, callback, ip);
	sdprintf("egg_dns_reverse(%s, %d) -> %d", ip, timeout, q->id);

	query_transform_ip(q, ip);

        dns_send_query(q);

//	/* setup timer to detect dead ns */
//	dns_create_timeout_timer(&q, ip, timeout);

	return(q->id);
}

static void dns_read(int idx, char* buf, int& atr, bool blocking) {
	atr = read(dcc[idx].sock, buf, 512);

	if (atr == -1) {
		if (errno == EAGAIN) {
			do {
				atr = read(dcc[idx].sock, buf, 512);
			} while (errno == EAGAIN && blocking);
		}
		if (atr == -1) {
			dns_on_eof(idx);
			errno = EPIPE;
			return;
		}
	}
	sdprintf("SETTING TIMEOUT to 0: Received reply.");
	dns_handler.timeout_val = 0;
	errno = 0;
	return;
}

static void dns_on_read(int idx, char* buf, int atr)
{
	dcc[idx].timeval = now;

//	if (resend_on_read) {
//		resend_on_read = 0;
//		dns_resend_queries();
//		return;
//	}

	dns_read(idx, buf, atr);
	if (parse_reply(buf, atr, dns_ip))
		dns_on_eof(idx);
	return;
}


static void dns_on_eof(int idx)
{
        sdprintf("EOF on dns idx: %d sock: %d (%s)", idx, dcc[idx].sock, dcc[idx].host);
        dns_reinit(idx);

	return;
}

/* for .restart
int egg_dns_shutdown(void)
{
	int i;

	if (nservers > 0) {
		for (i = 0; i < nservers; i++) {
			if (servers[i].ip) free(servers[i].ip);
		}
		free(servers); servers = NULL;
		nservers = 0;
	}
	
	if (nhosts > 0) {
		for (i = 0; i < nhosts; i++) {
			if (hosts[i].host) free(hosts[i].host);
			if (hosts[i].ip) free(hosts[i].ip);
		}
		free(hosts); hosts = NULL;
		nhosts = 0;
	}

	return (0);
}
*/
static const char *dns_next_server()
{
	if (!servers || nservers < 1) return("127.0.0.1");
	cur_server++;
	if (cur_server >= nservers) cur_server = 0;
	return(servers[cur_server].ip);
}

static void add_dns_server(char *ip)
{
	servers = (dns_server_t *) my_realloc(servers, (nservers+1)*sizeof(*servers));
	servers[nservers].ip = strdup(ip);
	nservers++;
        sdprintf("Added NS: %s", ip);
}

static void add_host(char *host, char *ip)
{
	hosts = (dns_host_t *) my_realloc(hosts, (nhosts+1)*sizeof(*hosts));
	hosts[nhosts].host = strdup(host);
	hosts[nhosts].ip = strdup(ip);
	nhosts++;
}

static int cache_expired(int id)
{
	if (cache[id].expiretime && (now >= cache[id].expiretime))  return(1);
	return (0);
}

static void cache_del(int id)
{
	delete cache[id].answer;
	free(cache[id].query);
	cache[id].expiretime = 0;

	ncache--;

	if (id < ncache) memcpy(&cache[id], &cache[ncache], sizeof(dns_cache_t));
	else bzero(&cache[id], sizeof(dns_cache_t));

	cache = (dns_cache_t *) my_realloc(cache, (ncache+1)*sizeof(*cache));
}

static void cache_add(const char *query, bd::Array<bd::String> answer, int ttl)
{
	cache = (dns_cache_t *) my_realloc(cache, (ncache+1)*sizeof(*cache));
	bzero(&cache[ncache], sizeof(cache[ncache]));
	cache[ncache].query = strdup(query);
        cache[ncache].answer = new bd::Array<bd::String>;
        *(cache[ncache].answer) = answer;
	cache[ncache].expiretime = now + ttl;
	ncache++;
}

static int cache_find(const char *query)
{
	int i;

	for (i = 0; i < ncache; i++)
		if (!strcasecmp(cache[i].query, query)) return (i);

	return (-1);
}

void dns_cache_flush()
{
  int i = 0;

  for (i = 0; i < ncache; i++) {
    cache_del(i);
    if (i == ncache) break;
    i--;
  }
}

static int read_thing(char *buf, char *ip)
{
	int skip, len;

	skip = strspn(buf, separators);
	buf += skip;
	len = strcspn(buf, separators);
	memcpy(ip, buf, len);
	ip[len] = 0;
	return(skip + len);
}

static int read_resolv(char *fname)
{
	FILE *fp;
	char buf[512], ip[512];
        int count = 0;

	fp = fopen(fname, "r");
	if (!fp) return 0;
	while (fgets(buf, sizeof(buf), fp)) {
		if (!strncasecmp(buf, "nameserver", 10)) {
			read_thing(buf+10, ip);
			if (strlen(ip)) {
				add_dns_server(ip);
                                ++count;
                        }
		}
	}
	fclose(fp);
        return count;
}

static void read_hosts(char *fname)
{
	FILE *fp;
	char buf[512], ip[512], host[512];
	int skip, n;

	fp = fopen(fname, "r");
	if (!fp) return;
	while (fgets(buf, sizeof(buf), fp)) {
		if (strchr(buf, '#')) continue;
		skip = read_thing(buf, ip);
		if (!strlen(ip)) continue;
		while ((n = read_thing(buf+skip, host))) {
			skip += n;
			if (strlen(host)) add_host(host, ip);
		}
	}
	fclose(fp);
}


static int make_header(char *buf, int id)
{
	_dns_header.question_count = htons(1);
//	_dns_header.id = htons(id);
	_dns_header.id = id;
	memcpy(buf, &_dns_header, HEAD_SIZE);
	return(HEAD_SIZE);
}

static int cut_host(const char *host, char *query)
{
	const char *period = NULL, *orig = NULL;
	int len;

	orig = query;
	while ((period = strchr(host, '.'))) {
		len = period - host;
		if (len > 63) return(-1);
		*query++ = len;
		memcpy(query, host, len);
		query += len;
		host = period+1;
	}
	len = strlen(host);
	if (len) {
		*query++ = len;
		memcpy(query, host, len);
		query += len;
	}
	*query++ = 0;
	return(query-orig);
}

int reverse_ip(const char *host, char *reverse)
{
	const char *period = NULL;
	int offset, len;

	period = strchr(host, '.');
	if (!period) {
		len = strlen(host);
		memcpy(reverse, host, len);
		return(len);
	}
	else {
		len = period - host;
		offset = reverse_ip(host+len+1, reverse);
		reverse[offset++] = '.';
		memcpy(reverse+offset, host, len);
		reverse[offset+len] = 0;
		return(offset+len);
	}
}

int egg_dns_cancel(int id, int issue_callback)
{
	dns_query_t *q, *prev = NULL;

	for (q = query_head; q; q = q->next) {
		if (q->id == id) break;
		prev = q;
	}
	if (!q) return(-1);
	if (prev) prev->next = q->next;
	else query_head = q->next;
	sdprintf("Cancelling query: %s", q->query);
	if (issue_callback && q->callback) {
		if (q->answer->size() > 0) {
			cache_add(q->query, *(q->answer), q->lowest_ttl);

			q->callback(q->id, q->client_data, q->query, *(q->answer));
		} else {
			bd::Array<bd::String> empty;
			q->callback(q->id, q->client_data, q->query, empty);
		}
	}
	free_query(q);
	return(0);
}

static int skip_name(unsigned char *ptr)
{
	int len;
	unsigned char *start = ptr;

	while ((len = *ptr++) > 0) {
		if (len > 63) {
			ptr++;
			break;
		}
		else {
			ptr += len;
		}
	}
	return(ptr - start);
}

/*
void print_header(dns_header_t &header)
{
#define dofield(_field)         sdprintf("%s: %d\n", #_field, _field)
	dofield(header.id);
	dofield(header.question_count);
	dofield(header.answer_count);
	dofield(header.ar_count);
	dofield(header.ns_count);
#undef dofield
}

void print_reply(dns_rr_t &reply)
{
#define dofield(_field)         sdprintf("%s: %d\n", #_field, _field)
	dofield(reply.type);
	dofield(reply.dclass);
	dofield(reply.ttl);
	dofield(reply.rdlength);
#undef dofield
}
*/

static int parse_reply(char *response, size_t nbytes, const char* server_ip, bool blocking)
{
	dns_header_t header;
	dns_query_t *q = NULL, *prev = NULL;
	char result[512] = "";
	short rr;
	dns_rr_t reply;
	unsigned const char *eop = (unsigned char *) response + nbytes;
	unsigned char *ptr = (unsigned char *) response;
        int return_code = 0;

	memcpy(&header, ptr, HEAD_SIZE);
	ptr += HEAD_SIZE;

	/* header.id is already in our order, echoed by the server */
	header.flags = ntohs(header.flags);
	header.question_count = ntohs(header.question_count);
	header.answer_count = ntohs(header.answer_count);
	header.ar_count = ntohs(header.ar_count);
	header.ns_count = ntohs(header.ns_count);

//	print_header(header);

	/* Find our copy of the query before proceeding. */
	for (q = query_head; q; q = q->next) {
		if (q->id == header.id) break;
		prev = q;
	}

        sdprintf("Reply(%d) questions: %d answers: %d ar: %d ns: %d from: %s QR: %d OPCODE: %d AA: %d TC: %d RD: %d RA: %d RCODE: %d",
            header.id,
            header.question_count,
            header.answer_count,
            header.ar_count,
            header.ns_count,
            server_ip,
            GET_QR(header.flags),
            GET_OPCODE(header.flags),
            GET_AA(header.flags),
            GET_TC(header.flags),
            GET_RD(header.flags),
            GET_RA(header.flags),
            GET_RCODE(header.flags)
        );

	if (!q) {
          sdprintf("Reply(%d) not found??", header.id);
          return 0;
        }

        /* Did this server give us recursion? */
        if (!GET_RA(header.flags)) {
                sdprintf("Ignoring reply(%d) from %s: no recusion available.", header.id, server_ip);
                return_code = 1;		/* get a new server */
		q->remaining = 0;		/* Force this query to be removed, any further answers are ignored */
				  errno = EACCES;
                goto callback;
        }

        /* Check for errors */
        if (GET_RCODE(header.flags)) {
          switch (GET_RCODE(header.flags)) {
            case 1:   /* Format error */
                  sdprintf("Ignoring reply(%d) from %s: Format error.", header.id, server_ip);
				  errno = EINVAL;
                  break;
            case 2:   /* Server error */
                  sdprintf("Ignoring reply(%d) from %s: Server error.", header.id, server_ip);
				  errno = EINVAL;
                  return_code = 1;		/* get a new server */
		  q->remaining = 0;		/* Force this query to be removed, any further answers are ignored */
                  break;
            case 3:   /* Name error */
                  sdprintf("Ignoring reply(%d) from %s: NXDOMAIN.", header.id, server_ip);
                  /* Ignore the incoming AAAA or A reply as it will still be NXDOMAIN */
		  q->remaining = 0;		/* Force this query to be removed, any further answers are ignored */
				  errno = ENOENT;
                  break;
            case 4:
                  sdprintf("Ignoring reply(%d) from %s: Query not supported", header.id, server_ip);
				  errno = EINVAL;
                  break;
            case 5:
                  sdprintf("Ignoring reply(%d) from %s: REFUSED", header.id, server_ip);
                  return_code = 1;		/* get a new server */
		  q->remaining = 0;		/* Force this query to be removed, any further answers are ignored */
				  errno = EACCES;
                  break;
          }

          goto callback;
	}

//        /* destroy our async timeout */
//        timer_destroy(q->timer_id);

	/* Pass over the questions. */
	for (rr = 0; rr < header.question_count; rr++) {
		ptr += skip_name(ptr);
		ptr += 4;
	}
	/* End of questions. */

//	for (rr = 0; rr < header.answer_count + header.ar_count + header.ns_count; rr++) {


	q->answers += header.answer_count;

	for (rr = 0; rr < header.answer_count; rr++) {
		result[0] = 0;
		/* Read in the answer. */
		ptr += skip_name(ptr);

		memcpy(&reply, ptr, RR_SIZE);
		ptr += RR_SIZE;

		reply.type = ntohs(reply.type);
		reply.dclass = ntohs(reply.dclass);
		reply.rdlength = ntohs(reply.rdlength);
		reply.ttl = ntohl(reply.ttl);
		/* Save the lowest ttl */
		if (reply.ttl && ((!q->lowest_ttl) || (q->lowest_ttl >  reply.ttl))) q->lowest_ttl = reply.ttl;

//		print_reply(reply);

		switch (reply.type) {
		case DNS_A:
			inet_ntop(AF_INET, ptr, result, 512);
			*(q->answer) << result;
			sdprintf("Reply(%d): %s. \t %d \t IN A \t %s", header.id, q->query, reply.ttl, result);
			break;
		case DNS_AAAA:
#ifdef USE_IPV6
			inet_ntop(AF_INET6, ptr, result, 512);
			*(q->answer) << result;
			sdprintf("Reply(%d): %s. \t %d \t IN AAAA \t %s", header.id, q->query, reply.ttl, result);
#endif /* USE_IPV6 */
			break;
		case DNS_PTR:
			if ((my_dn_expand((const unsigned char *) response, eop, ptr, result, sizeof(result)) != -1) && result[0]) {
				*(q->answer) << result;
				sdprintf("Reply(%d): %s. \t %d \t IN PTR \t %s", header.id, q->query, reply.ttl, result);
			}
			break;
		default:
			sdprintf("Unhandled DNS reply type: %d", reply.type);
			break;
		}

		ptr += reply.rdlength;
                if ((size_t) (ptr - (unsigned char*) response) > nbytes) {
                  sdprintf("MALFORMED/TRUNCATED DNS PACKET detected (need TCP).");
		  q->remaining = 0;		/* Force this query to be removed, any further answers are ignored */
                  break;
                }
	}

callback:
	/* Don't continue if we haven't gotten all expected replies. */
	if (--q->remaining > 0) {
		sdprintf("Waiting on query(%d) for more replies.", header.id);
		errno = EAGAIN;
		return 0;
	}

	errno = 0;

	/* Ok, we have, so now issue the callback with the answers. */
	if (prev) prev->next = q->next;
	else query_head = q->next;

        if (q->answer->size() > 0) {
		cache_add(q->query, *(q->answer), q->lowest_ttl);

		if (q->callback)
			q->callback(q->id, q->client_data, q->query, *(q->answer));
        } else if (q->callback) {
		bd::Array<bd::String> empty;
		q->callback(q->id, q->client_data, q->query, empty);
        }

		if (!blocking)
			free_query(q);

	return return_code;
}


void tell_dnsdebug(int idx)
{
	dns_query_t *q = NULL;

	dprintf(idx, "NS: %s\n", dns_ip);

	for (q = query_head; q; q = q->next)
		dprintf(idx, "DNS (%d) (%ds): %s\n", q->id, (int) (q->expiretime - now), q->query);

//	for (i = 0; i < nhosts; i++)
//           dprintf(idx, "HOST #%d: %s/%s\n", i, hosts[i].host, hosts[i].ip);

	for (int i = 0; i < ncache; i++) {
		dprintf(idx, "cache(%d) %s expires in %ds\n", i, cache[i].query, (int) (cache[i].expiretime - now));
		for (size_t n = 0; n < cache[i].answer->size(); n++)
			dprintf(idx, "%zu: %s\n", n, bd::String((*(cache[i].answer))[n]).c_str());
	}
}

static void expire_queries()
{
  dns_query_t *q = NULL, *next = NULL;
  int i = 0;

  /* need to check for expired queries and either:
    a) recheck/change ns
    b) expire due to ttl
    */

  if (query_head) {
    for (q = query_head; q; q = q->next) {
      if (q->expiretime <= now) {		/* set in alloc_query */
        if (q->next)
          next = q->next;
        egg_dns_cancel(q->id, 1);
        if (!next) break;
        q = next;
      }
    }
  }

  for (i = 0; i < ncache; i++) {
    if (cache_expired(i)) {
      cache_del(i);
      if (i == ncache) break;
      i--;
    }
  }

}


/* Read in .hosts and /etc/hosts and .resolv.conf and /etc/resolv.conf */
int egg_dns_init()
{
	/* Set RECURSION DESIRED */
        SET_RD(_dns_header.flags);

        /* Convert flags to network order */
        _dns_header.flags = htons(_dns_header.flags);

	if (!read_resolv(".resolv.conf")) {
		read_resolv("/etc/resolv.conf");
		/* some backup servers, probably will never be used. */
		add_dns_server("4.2.2.2");
		add_dns_server("8.8.8.8");
		add_dns_server("8.8.4.4");
        }

//	read_hosts("/etc/hosts");
	read_hosts(".hosts");
    
/* root servers for future development (tracing down)
	add_dns_server("198.41.0.4");
	add_dns_server("192.228.79.201");
	add_dns_server("192.33.4.12");
	add_dns_server("128.8.10.90");
	add_dns_server("192.203.230.10");
	add_dns_server("192.5.5.241");
	add_dns_server("192.112.36.4");
	add_dns_server("128.63.2.53");
	add_dns_server("192.36.148.17");
	add_dns_server("192.58.128.30");
	add_dns_server("193.0.14.129");
	add_dns_server("198.32.64.12");
	add_dns_server("202.12.27.33");
*/


	timer_create_secs(3, "adns_check_expires", (Function) expire_queries);

	return(0);
}


bool valid_dns_id(int idx, int id)
{
  if (id == -1)
    return 1;
  if (valid_idx(idx) && dcc[idx].dns_id && dcc[idx].dns_id == id)
    return 1;
  sdprintf("dns_id: %d is not associated with dead idx: %d", id, idx);
  return 0;
}

bd::String dns_find_ip(bd::Array<bd::String> ips, int af_type) {
    for (size_t i = 0; i < ips.size(); ++i) {
      if (is_dotted_ip(bd::String(ips[i]).c_str()) == af_type) {
        return ips[i];
      }
    }
	return bd::String();
}
/* vim: set sts=4 sw=4 ts=4 noet: */