| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039 |
- /* 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>
- typedef struct {
- char **list;
- time_t ttl;
- int len;
- } dns_answer_t;
- typedef struct dns_query {
- struct dns_query *next;
- dns_answer_t answer;
- dns_callback_t callback;
- void *client_data;
- time_t expiretime;
- char *query;
- char *ip;
- int id;
- // int timer_id;
- int answers;
- int remaining;
- } dns_query_t;
- 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 HEAD_SIZE 12
- 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 {
- dns_answer_t answer;
- char *query;
- time_t expiretime;
- } dns_cache_t;
- static int query_id = 1;
- 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 reverse_ip(const char *host, char *reverse);
- static void 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_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);
- time_t async_lookup_timeout = 30;
- time_t async_server_timeout = 40;
- //int resend_on_read = 0;
- static void
- dns_display(int idx, char *buf)
- {
- sprintf(buf, "named waited %lis", 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)
- {
- sprintf(buf, "dns waited %lis", 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);
- }
- }
- 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);
- }
- */
- static void answer_init(dns_answer_t *answer)
- {
- memset(answer, 0, sizeof(*answer));
- }
- static void answer_add(dns_answer_t *answer, const char *what)
- {
- answer->list = (char **) my_realloc(answer->list, sizeof(*answer->list) * (answer->len+2));
- answer->list[answer->len] = strdup(what);
- answer->len++;
- answer->list[answer->len] = NULL;
- }
- static void answer_free(dns_answer_t *answer)
- {
- int i;
- for (i = 0; i < answer->len; i++) free(answer->list[i]);
- if (answer->list) free(answer->list);
- }
- 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 = query_id;
- query_id++;
- q->query = strdup(query);
- q->answers = 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 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.");
- return 1;
- }
- sdprintf("dns_idx: %d", dns_idx);
- dcc[dns_idx].sock = sock;
- dns_sock = sock;
- sdprintf("dns_sock: %d", dcc[dns_idx].sock);
- strcpy(dcc[dns_idx].host, dns_ip);
- strcpy(dcc[dns_idx].nick, "(adns)");
- 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)
- {
- 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 %li", async_server_timeout);
- dcc[dns_idx].timeval = now;
- }
- write(dcc[dns_idx].sock, query, len);
- // 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 (!egg_strcasecmp(q->query, host))
- return q;
- return NULL;
- }
- void dns_send_query(dns_query_t *q)
- {
- char buf[512] = "";
- int len;
- if (!q->ip) {
- /* Send the ipv4 query. */
- q->remaining = 1;
- 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);
- #ifdef USE_IPV6
- /* 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);
- #endif
- } else if (q->ip) {
- q->remaining = 1;
- 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);
- }
- }
- /*
- 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, int timeout, dns_callback_t callback, void *client_data)
- {
- dns_query_t *q = NULL;
- int i, cache_id;
- sdprintf("egg_dns_lookup(%s, %d)", host, timeout);
- if (is_dotted_ip(host)) {
- /* If it's already an ip, we're done. */
- dns_answer_t answer;
- answer_init(&answer);
- answer_add(&answer, host);
- callback(-1, client_data, host, answer.list);
- answer_free(&answer);
- return(-1);
- }
- /* Ok, now see if it's in our host cache. */
- for (i = 0; i < nhosts; i++) {
- if (!egg_strcasecmp(host, hosts[i].host)) {
- dns_answer_t answer;
- answer_init(&answer);
- answer_add(&answer, hosts[i].ip);
- callback(-1, client_data, host, answer.list);
- answer_free(&answer);
- return(-1);
- }
- }
- cache_id = cache_find(host);
- if (cache_id >= 0) {
- shuffleArray(cache[cache_id].answer.list, cache[cache_id].answer.len);
- callback(-1, client_data, host, cache[cache_id].answer.list);
- return(-1);
- }
- /* check if the query was already made */
- if (find_query(host))
- return(-1);
- /* Allocate our query struct. */
- q = alloc_query(client_data, callback, host);
- dns_send_query(q);
- // /* setup a timer to detect dead ns */
- // dns_create_timeout_timer(&q, host, timeout);
- /* Send the ipv4 query. */
- return(q->id);
- }
- /* 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, int timeout, dns_callback_t callback, void *client_data)
- {
- dns_query_t *q;
- int i, cache_id;
- sdprintf("egg_dns_reverse(%s, %d)", ip, timeout);
- if (!is_dotted_ip(ip)) {
- /* If it's not a valid ip, don't even make the request. */
- 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 (!egg_strcasecmp(hosts[i].ip, ip)) {
- dns_answer_t answer;
- answer_init(&answer);
- answer_add(&answer, hosts[i].host);
- callback(-1, client_data, ip, answer.list);
- answer_free(&answer);
- return(-1);
- }
- }
- cache_id = cache_find(ip);
- if (cache_id >= 0) {
- shuffleArray(cache[cache_id].answer.list, cache[cache_id].answer.len);
- callback(-1, client_data, ip, cache[cache_id].answer.list);
- return(-1);
- }
- /* check if the query was already made */
- if (find_query(ip))
- return(-1);
- q = alloc_query(client_data, callback, 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);
- q->ip = (char *) my_calloc(1, strlen(temp) + 9 + 1);
- // reverse_ip(temp, q->ip);
- strcat(q->ip, temp);
- strcat(q->ip, "ip6.arpa");
- sdprintf("reversed ipv6 ip: %s", q->ip);
- }
- else {
- q->ip = (char *) my_calloc(1, strlen(ip) + 13 + 1);
- reverse_ip(ip, q->ip);
- strcat(q->ip, ".in-addr.arpa");
- }
- dns_send_query(q);
- // /* setup timer to detect dead ns */
- // dns_create_timeout_timer(&q, ip, timeout);
- return(q->id);
- }
- //static int dns_on_read(void *client_data, int idx, char *buf, int len)
- 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;
- // }
- atr = read(dcc[idx].sock, buf, 512);
- if (atr == -1) {
- if (errno == EAGAIN)
- atr = read(dcc[idx].sock, buf, 512);
- if (atr == -1) {
- dns_on_eof(idx);
- return;
- }
- }
- sdprintf("SETTING TIMEOUT to 0");
- dns_handler.timeout_val = 0;
- if (parse_reply(buf, atr))
- 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)
- {
- answer_free(&cache[id].answer);
- free(cache[id].query);
- cache[id].expiretime = 0;
- ncache--;
- if (id < ncache) egg_memcpy(&cache[id], &cache[ncache], sizeof(dns_cache_t));
- else egg_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, dns_answer_t *answer)
- {
- int i;
- cache = (dns_cache_t *) my_realloc(cache, (ncache+1)*sizeof(*cache));
- egg_bzero(&cache[ncache], sizeof(cache[ncache]));
- cache[ncache].query = strdup(query);
- answer_init(&cache[ncache].answer);
- for (i = 0; i < answer->len; i++)
- answer_add(&cache[ncache].answer, answer->list[i]);
- cache[ncache].expiretime = now + answer->ttl;
- ncache++;
- }
- static int cache_find(const char *query)
- {
- int i;
- for (i = 0; i < ncache; i++)
- if (!egg_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);
- egg_memcpy(ip, buf, len);
- ip[len] = 0;
- return(skip + len);
- }
- static void read_resolv(char *fname)
- {
- FILE *fp;
- char buf[512], ip[512];
- fp = fopen(fname, "r");
- if (!fp) return;
- while (fgets(buf, sizeof(buf), fp)) {
- if (!strncasecmp(buf, "nameserver", 10)) {
- read_thing(buf+10, ip);
- if (strlen(ip)) add_dns_server(ip);
- }
- }
- fclose(fp);
- }
- 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;
- egg_memcpy(buf, &_dns_header, HEAD_SIZE);
- return(HEAD_SIZE);
- }
- static int cut_host(const char *host, char *query)
- {
- char *period, *orig;
- int len;
- orig = query;
- while ((period = strchr(host, '.'))) {
- len = period - host;
- if (len > 63) return(-1);
- *query++ = len;
- egg_memcpy(query, host, len);
- query += len;
- host = period+1;
- }
- len = strlen(host);
- if (len) {
- *query++ = len;
- egg_memcpy(query, host, len);
- query += len;
- }
- *query++ = 0;
- return(query-orig);
- }
- static int reverse_ip(const char *host, char *reverse)
- {
- char *period;
- int offset, len;
- period = strchr(host, '.');
- if (!period) {
- len = strlen(host);
- egg_memcpy(reverse, host, len);
- return(len);
- }
- else {
- len = period - host;
- offset = reverse_ip(host+len+1, reverse);
- reverse[offset++] = '.';
- egg_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;
- 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(q->id, q->client_data, q->query, NULL);
- if (q->ip)
- free(q->ip);
- free(q->query);
- free(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)
- {
- dns_header_t header;
- dns_query_t *q = NULL, *prev = NULL;
- dns_rr_t reply;
- char result[512] = "";
- short rr;
- int r = -1;
- unsigned const char *eop = (unsigned char *) response + nbytes;
- unsigned char *ptr = (unsigned char *) response;
- egg_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;
- }
- if (!q) return 0;
-
- // /* 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);
- egg_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->answer.ttl) || (q->answer.ttl > reply.ttl))) q->answer.ttl = reply.ttl;
- // print_reply(reply);
- switch (reply.type) {
- case DNS_A:
- inet_ntop(AF_INET, ptr, result, 512);
- answer_add(&q->answer, result);
- break;
- case DNS_AAAA:
- #ifdef USE_IPV6
- inet_ntop(AF_INET6, ptr, result, 512);
- answer_add(&q->answer, result);
- #endif /* USE_IPV6 */
- break;
- case DNS_PTR:
- r = my_dn_expand((const unsigned char *) response, eop, ptr, result, sizeof(result));
- if (r != -1 && result[0])
- answer_add(&q->answer, result);
- break;
- default:
- sdprintf("Unhandled DNS reply type: %d", reply.type);
- break;
- }
- ptr += reply.rdlength;
- }
- q->remaining--;
- /* Don't continue if we haven't gotten all expected replies. */
- if (q->remaining > 0) return 0;
- if (!q->answers) {
- sdprintf("Failed to get any answers for query");
- if (prev) prev->next = q->next;
- else query_head = q->next;
- q->callback(q->id, q->client_data, q->query, NULL);
- free(q->query);
- if (q->ip)
- free(q->ip);
- free(q);
- return 1; /* get a new server */
- }
- /* Ok, we have, so now issue the callback with the answers. */
- if (prev) prev->next = q->next;
- else query_head = q->next;
- cache_add(q->query, &q->answer);
- q->callback(q->id, q->client_data, q->query, q->answer.list);
- answer_free(&q->answer);
- free(q->query);
- if (q->ip)
- free(q->ip);
- free(q);
- return(0);
- }
- void tell_dnsdebug(int idx)
- {
- dns_query_t *q = NULL;
- int i, n = 0;
- dprintf(idx, "NS: %s\n", dns_ip);
- for (q = query_head; q; q = q->next)
- dprintf(idx, "DNS (%d) (%lis): %s\n", q->id, 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 (i = 0; i < ncache; i++) {
- dprintf(idx, "cache(%d) %s expires in %lis\n", i, cache[i].query, cache[i].expiretime - now);
- for (n = 0; n < cache[i].answer.len; n++)
- dprintf(idx, "%d: %s\n", n, cache[i].answer.list[n]);
- }
- }
- 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()
- {
- _dns_header.flags = htons(1 << 8 | 1 << 7);
- read_resolv(".resolv.conf");
- read_resolv("/etc/resolv.conf");
- // read_hosts("/etc/hosts");
- read_hosts(".hosts");
-
- /* some backup servers, probably will never be used. */
- add_dns_server("205.152.0.20"); //ns.atl.bellsouth.net
- add_dns_server("203.251.80.133"); //ns.abovenet.net
- add_dns_server("68.2.16.30"); //some cox ns
- add_dns_server("68.6.16.25"); //another cox
- add_dns_server("65.215.220.12"); //staminus
- add_dns_server("72.20.1.2"); //new staminus
- /* 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;
- }
|