adns.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989
  1. /* new async dns
  2. *
  3. */
  4. #include "common.h"
  5. #include "adns.h"
  6. #include "egg_timer.h"
  7. #include "main.h"
  8. #include "net.h"
  9. #include "misc.h"
  10. #include "socket.h"
  11. #include <netinet/in.h>
  12. #include <sys/socket.h>
  13. #include <arpa/inet.h>
  14. typedef struct {
  15. char **list;
  16. time_t ttl;
  17. int len;
  18. } dns_answer_t;
  19. typedef struct dns_query {
  20. struct dns_query *next;
  21. dns_answer_t answer;
  22. dns_callback_t callback;
  23. void *client_data;
  24. time_t expiretime;
  25. char *query;
  26. char *ip;
  27. int id;
  28. // int timer_id;
  29. int remaining;
  30. } dns_query_t;
  31. typedef struct {
  32. unsigned short id;
  33. unsigned short flags;
  34. unsigned short question_count;
  35. unsigned short answer_count;
  36. unsigned short ns_count;
  37. unsigned short ar_count;
  38. } dns_header_t;
  39. #define HEAD_SIZE 12
  40. typedef struct {
  41. /* char name[]; */
  42. unsigned short type;
  43. unsigned short dclass;
  44. int ttl;
  45. unsigned short rdlength;
  46. /* char rdata[]; */
  47. } dns_rr_t;
  48. #define RR_SIZE 10
  49. /* Entries from resolv.conf */
  50. typedef struct dns_server {
  51. char *ip;
  52. int idx;
  53. } dns_server_t;
  54. /* Entries from hosts */
  55. typedef struct {
  56. char *host;
  57. char *ip;
  58. } dns_host_t;
  59. typedef struct {
  60. dns_answer_t answer;
  61. char *query;
  62. time_t expiretime;
  63. } dns_cache_t;
  64. static int query_id = 1;
  65. static dns_header_t _dns_header = {0, 0, 0, 0, 0, 0};
  66. static dns_query_t *query_head = NULL;
  67. static dns_host_t *hosts = NULL;
  68. static int nhosts = 0;
  69. static dns_cache_t *cache = NULL;
  70. static int ncache = 0;
  71. static dns_server_t *servers = NULL;
  72. static int nservers = 0;
  73. static int cur_server = -1;
  74. static char separators[] = " ,\t\r\n";
  75. int dns_idx = -1;
  76. int dns_sock = -1;
  77. const char *dns_ip = NULL;
  78. static int make_header(char *buf, int id);
  79. static int cut_host(const char *host, char *query);
  80. static int reverse_ip(const char *host, char *reverse);
  81. static void read_resolv(char *fname);
  82. static void read_hosts(char *fname);
  83. static int get_dns_idx();
  84. static void dns_resend_queries();
  85. static int cache_find(const char *);
  86. //static int dns_on_read(void *client_data, int idx, char *buf, int len);
  87. //static int dns_on_eof(void *client_data, int idx, int err, const char *errmsg);
  88. static void dns_on_read(int idx, char *buf, int atr);
  89. static void dns_on_eof(int idx);
  90. static const char *dns_next_server();
  91. static void parse_reply(char *response, size_t nbytes);
  92. time_t async_resolve_timeout = 30;
  93. //int resend_on_read = 0;
  94. static void
  95. dns_display(int idx, char *buf)
  96. {
  97. sprintf(buf, "named waited %lis", now - dcc[idx].timeval);
  98. }
  99. static void
  100. dns_reinit(int idx)
  101. {
  102. sdprintf("Re-opening dns socket...");
  103. killsock(dcc[idx].sock);
  104. lostdcc(idx);
  105. dns_idx = -1;
  106. dns_sock = -1;
  107. dns_ip = NULL;
  108. if (!get_dns_idx())
  109. sdprintf("Successfully reopened dns socket");
  110. else
  111. sdprintf("Failed to reopen dns socket");
  112. }
  113. static void
  114. dns_timeout(int idx)
  115. {
  116. sdprintf("DNS socket timed out");
  117. /*egg_dns_cancel(dcc[idx].u.dns_id, 1);*/
  118. // resend_on_read = 1;
  119. dns_reinit(idx);
  120. // sleep(2);
  121. // dns_resend_queries();
  122. }
  123. static struct dcc_table dns_handler = {
  124. "adns",
  125. DCT_VALIDIDX,
  126. dns_on_eof,
  127. dns_on_read,
  128. &async_resolve_timeout,
  129. dns_timeout,
  130. dns_display,
  131. NULL,
  132. NULL,
  133. NULL
  134. };
  135. static void dcc_dnswait(int idx, char *buf, int len)
  136. {
  137. /* Ignore anything now. */
  138. }
  139. static void eof_dcc_dnswait(int idx)
  140. {
  141. putlog(LOG_MISC, "*", "Lost connection while resolving hostname [%s/%d]",
  142. iptostr(htonl(dcc[idx].addr)), dcc[idx].port);
  143. killsock(dcc[idx].sock);
  144. lostdcc(idx);
  145. }
  146. static void display_dcc_dnswait(int idx, char *buf)
  147. {
  148. sprintf(buf, "dns waited %lis", now - dcc[idx].timeval);
  149. }
  150. static void kill_dcc_dnswait(int idx, void *x)
  151. {
  152. struct dns_info *p = (struct dns_info *) x;
  153. if (p) {
  154. if (p->cbuf)
  155. free(p->cbuf);
  156. }
  157. }
  158. struct dcc_table DCC_DNSWAIT = {
  159. "DNSWAIT",
  160. DCT_VALIDIDX,
  161. eof_dcc_dnswait,
  162. dcc_dnswait,
  163. NULL,
  164. NULL,
  165. display_dcc_dnswait,
  166. kill_dcc_dnswait,
  167. NULL,
  168. NULL
  169. };
  170. /*
  171. static void async_timeout(void *client_data)
  172. {
  173. int id = (int) client_data;
  174. sdprintf("%d timed out", id);
  175. egg_dns_cancel(id, 1);
  176. }
  177. */
  178. static void answer_init(dns_answer_t *answer)
  179. {
  180. memset(answer, 0, sizeof(*answer));
  181. }
  182. static void answer_add(dns_answer_t *answer, const char *what)
  183. {
  184. answer->list = (char **) my_realloc(answer->list, sizeof(*answer->list) * (answer->len+2));
  185. answer->list[answer->len] = strdup(what);
  186. answer->len++;
  187. answer->list[answer->len] = NULL;
  188. }
  189. static void answer_free(dns_answer_t *answer)
  190. {
  191. int i;
  192. for (i = 0; i < answer->len; i++) free(answer->list[i]);
  193. if (answer->list) free(answer->list);
  194. }
  195. static dns_query_t *alloc_query(void *client_data, dns_callback_t callback, const char *query)
  196. {
  197. dns_query_t *q = (dns_query_t *) my_calloc(1, sizeof(*q));
  198. q->id = query_id;
  199. query_id++;
  200. q->query = strdup(query);
  201. q->callback = callback;
  202. q->client_data = client_data;
  203. q->expiretime = now + async_resolve_timeout;
  204. q->next = query_head;
  205. query_head = q;
  206. return q;
  207. }
  208. static int get_dns_idx()
  209. {
  210. int i, sock;
  211. sock = -1;
  212. for (i = 0; i < nservers; i++) {
  213. if (!dns_ip) dns_ip = dns_next_server();
  214. sock = socket_create(dns_ip, DNS_PORT, NULL, 0, SOCKET_CLIENT | SOCKET_NONBLOCK | SOCKET_UDP);
  215. if (sock < 0) {
  216. /* Try the next server. */
  217. dns_ip = NULL;
  218. }
  219. else break;
  220. }
  221. if (i == nservers) return 1;
  222. // dns_idx = sockbuf_new();
  223. // sockbuf_set_handler(dns_idx, &dns_handler, NULL);
  224. // sockbuf_set_sock(dns_idx, sock, 0);
  225. // allocsock(sock, SOCK_CONNECT);
  226. dns_idx = new_dcc(&dns_handler, 0);
  227. if (dns_idx < 0) {
  228. putlog(LOG_SERV, "*", "NO MORE DCC CONNECTIONS -- Can't create dns connection.");
  229. return 1;
  230. }
  231. sdprintf("dns_idx: %d", dns_idx);
  232. dcc[dns_idx].sock = sock;
  233. dns_sock = sock;
  234. sdprintf("dns_sock: %d", dcc[dns_idx].sock);
  235. strcpy(dcc[dns_idx].host, dns_ip);
  236. strcpy(dcc[dns_idx].nick, "(new_dns)");
  237. sdprintf("dns_ip: %s", dns_ip);
  238. dcc[dns_idx].timeval = now;
  239. dns_handler.timeout_val = 0;
  240. return 0;
  241. }
  242. void egg_dns_send(char *query, int len)
  243. {
  244. if (dns_idx < 0) {
  245. if (get_dns_idx()) {
  246. sdprintf("get_dns_idx() failed in egg_dns_send");
  247. return;
  248. }
  249. }
  250. dns_handler.timeout_val = &async_resolve_timeout;
  251. dcc[dns_idx].timeval = now;
  252. write(dcc[dns_idx].sock, query, len);
  253. // sockbuf_write(dns_idx, query, len);
  254. }
  255. dns_query_t *find_query(const char *host)
  256. {
  257. dns_query_t *q = NULL;
  258. for (q = query_head; q; q = q->next)
  259. if (!egg_strcasecmp(q->query, host))
  260. return q;
  261. return NULL;
  262. }
  263. void dns_send_query(dns_query_t *q)
  264. {
  265. char buf[512] = "";
  266. int len;
  267. if (!q->ip) {
  268. /* Send the ipv4 query. */
  269. q->remaining = 1;
  270. len = make_header(buf, q->id);
  271. len += cut_host(q->query, buf + len);
  272. buf[len] = 0; len++; buf[len] = DNS_A; len++;
  273. buf[len] = 0; len++; buf[len] = 1; len++;
  274. egg_dns_send(buf, len);
  275. #ifdef USE_IPV6
  276. /* Now send the ipv6 query. */
  277. q->remaining++;
  278. len = make_header(buf, q->id);
  279. len += cut_host(q->query, buf + len);
  280. buf[len] = 0; len++; buf[len] = DNS_AAAA; len++;
  281. buf[len] = 0; len++; buf[len] = 1; len++;
  282. egg_dns_send(buf, len);
  283. #endif
  284. } else if (q->ip) {
  285. q->remaining = 1;
  286. len = make_header(buf, q->id);
  287. len += cut_host(q->ip, buf + len);
  288. buf[len] = 0; len++; buf[len] = DNS_PTR; len++;
  289. buf[len] = 0; len++; buf[len] = 1; len++;
  290. egg_dns_send(buf, len);
  291. }
  292. }
  293. void dns_resend_queries()
  294. {
  295. dns_query_t *q = NULL;
  296. for (q = query_head; q; q = q->next) {
  297. if (now >= q->expiretime) {
  298. sdprintf("RESENDING: %s", q->query);
  299. dns_send_query(q);
  300. }
  301. }
  302. }
  303. /*
  304. void dns_create_timeout_timer(dns_query_t **qm, const char *query, int timeout)
  305. {
  306. dns_query_t *q = *qm;
  307. egg_timeval_t howlong;
  308. howlong.sec = timeout;
  309. howlong.usec = 0;
  310. q->timer_id = timer_create_complex(&howlong, query, (Function) async_timeout, (void *) q->id, 0);
  311. }
  312. */
  313. /* Perform an async dns lookup. This is host -> ip. For ip -> host, use
  314. * egg_dns_reverse(). We return a dns id that you can use to cancel the
  315. * lookup. */
  316. int egg_dns_lookup(const char *host, int timeout, dns_callback_t callback, void *client_data)
  317. {
  318. dns_query_t *q = NULL;
  319. int i, cache_id;
  320. sdprintf("egg_dns_lookup(%s, %d)", host, timeout);
  321. if (is_dotted_ip(host)) {
  322. /* If it's already an ip, we're done. */
  323. dns_answer_t answer;
  324. answer_init(&answer);
  325. answer_add(&answer, host);
  326. callback(-1, client_data, host, answer.list);
  327. answer_free(&answer);
  328. return(-1);
  329. }
  330. /* Ok, now see if it's in our host cache. */
  331. for (i = 0; i < nhosts; i++) {
  332. if (!egg_strcasecmp(host, hosts[i].host)) {
  333. dns_answer_t answer;
  334. answer_init(&answer);
  335. answer_add(&answer, hosts[i].ip);
  336. callback(-1, client_data, host, answer.list);
  337. answer_free(&answer);
  338. return(-1);
  339. }
  340. }
  341. cache_id = cache_find(host);
  342. if (cache_id >= 0) {
  343. shuffleArray(cache[cache_id].answer.list, cache[cache_id].answer.len);
  344. callback(-1, client_data, host, cache[cache_id].answer.list);
  345. return(-1);
  346. }
  347. /* check if the query was already made */
  348. if (find_query(host))
  349. return(-1);
  350. /* Allocate our query struct. */
  351. q = alloc_query(client_data, callback, host);
  352. dns_send_query(q);
  353. // /* setup a timer to detect dead ns */
  354. // dns_create_timeout_timer(&q, host, timeout);
  355. /* Send the ipv4 query. */
  356. return(q->id);
  357. }
  358. /* Perform an async dns reverse lookup. This does ip -> host. For host -> ip
  359. * use egg_dns_lookup(). We return a dns id that you can use to cancel the
  360. * lookup. */
  361. int egg_dns_reverse(const char *ip, int timeout, dns_callback_t callback, void *client_data)
  362. {
  363. dns_query_t *q;
  364. int i, cache_id;
  365. sdprintf("egg_dns_reverse(%s, %d)", ip, timeout);
  366. if (!is_dotted_ip(ip)) {
  367. /* If it's not a valid ip, don't even make the request. */
  368. callback(-1, client_data, ip, NULL);
  369. return(-1);
  370. }
  371. /* Ok, see if we have it in our host cache. */
  372. for (i = 0; i < nhosts; i++) {
  373. if (!egg_strcasecmp(hosts[i].ip, ip)) {
  374. dns_answer_t answer;
  375. answer_init(&answer);
  376. answer_add(&answer, hosts[i].host);
  377. callback(-1, client_data, ip, answer.list);
  378. answer_free(&answer);
  379. return(-1);
  380. }
  381. }
  382. cache_id = cache_find(ip);
  383. if (cache_id >= 0) {
  384. shuffleArray(cache[cache_id].answer.list, cache[cache_id].answer.len);
  385. callback(-1, client_data, ip, cache[cache_id].answer.list);
  386. return(-1);
  387. }
  388. /* check if the query was already made */
  389. if (find_query(ip))
  390. return(-1);
  391. q = alloc_query(client_data, callback, ip);
  392. /* We need to transform the ip address into the proper form
  393. * for reverse lookup. */
  394. if (strchr(ip, ':')) {
  395. char temp[128] = "";
  396. socket_ipv6_to_dots(ip, temp);
  397. sdprintf("dots: %s", temp);
  398. q->ip = (char *) my_calloc(1, strlen(temp) + 9 + 1);
  399. // reverse_ip(temp, q->ip);
  400. strcat(q->ip, temp);
  401. strcat(q->ip, "ip6.arpa");
  402. sdprintf("reversed ipv6 ip: %s", q->ip);
  403. }
  404. else {
  405. q->ip = (char *) my_calloc(1, strlen(ip) + 13 + 1);
  406. reverse_ip(ip, q->ip);
  407. strcat(q->ip, ".in-addr.arpa");
  408. }
  409. dns_send_query(q);
  410. // /* setup timer to detect dead ns */
  411. // dns_create_timeout_timer(&q, ip, timeout);
  412. return(q->id);
  413. }
  414. //static int dns_on_read(void *client_data, int idx, char *buf, int len)
  415. static void dns_on_read(int idx, char *buf, int atr)
  416. {
  417. dcc[idx].timeval = now;
  418. dns_handler.timeout_val = 0;
  419. // if (resend_on_read) {
  420. // resend_on_read = 0;
  421. // dns_resend_queries();
  422. // return;
  423. // }
  424. atr = read(dcc[idx].sock, buf, 512);
  425. parse_reply(buf, atr);
  426. return;
  427. }
  428. //static int dns_on_eof(void *client_data, int idx, int err, const char *errmsg)
  429. static void dns_on_eof(int idx)
  430. {
  431. // sockbuf_delete(idx);
  432. sdprintf("eof on dns idx: %d sock: %d", idx, dcc[idx].sock);
  433. dns_reinit(idx);
  434. return;
  435. }
  436. /* for .restart
  437. int egg_dns_shutdown(void)
  438. {
  439. int i;
  440. if (nservers > 0) {
  441. for (i = 0; i < nservers; i++) {
  442. if (servers[i].ip) free(servers[i].ip);
  443. }
  444. free(servers); servers = NULL;
  445. nservers = 0;
  446. }
  447. if (nhosts > 0) {
  448. for (i = 0; i < nhosts; i++) {
  449. if (hosts[i].host) free(hosts[i].host);
  450. if (hosts[i].ip) free(hosts[i].ip);
  451. }
  452. free(hosts); hosts = NULL;
  453. nhosts = 0;
  454. }
  455. return (0);
  456. }
  457. */
  458. static const char *dns_next_server()
  459. {
  460. if (!servers || nservers < 1) return("127.0.0.1");
  461. cur_server++;
  462. if (cur_server >= nservers) cur_server = 0;
  463. return(servers[cur_server].ip);
  464. }
  465. static void add_server(char *ip)
  466. {
  467. servers = (dns_server_t *) my_realloc(servers, (nservers+1)*sizeof(*servers));
  468. servers[nservers].ip = strdup(ip);
  469. nservers++;
  470. sdprintf("Added NS: %s", ip);
  471. }
  472. static void add_host(char *host, char *ip)
  473. {
  474. hosts = (dns_host_t *) my_realloc(hosts, (nhosts+1)*sizeof(*hosts));
  475. hosts[nhosts].host = strdup(host);
  476. hosts[nhosts].ip = strdup(ip);
  477. nhosts++;
  478. }
  479. static int cache_expired(int id)
  480. {
  481. if (cache[id].expiretime && (now >= cache[id].expiretime)) return(1);
  482. return (0);
  483. }
  484. static void cache_del(int id)
  485. {
  486. answer_free(&cache[id].answer);
  487. free(cache[id].query);
  488. cache[id].expiretime = 0;
  489. ncache--;
  490. if (id < ncache) egg_memcpy(&cache[id], &cache[ncache], sizeof(dns_cache_t));
  491. else egg_bzero(&cache[id], sizeof(dns_cache_t));
  492. cache = (dns_cache_t *) my_realloc(cache, (ncache+1)*sizeof(*cache));
  493. }
  494. static void cache_add(const char *query, dns_answer_t *answer)
  495. {
  496. int i;
  497. cache = (dns_cache_t *) my_realloc(cache, (ncache+1)*sizeof(*cache));
  498. egg_bzero(&cache[ncache], sizeof(cache[ncache]));
  499. cache[ncache].query = strdup(query);
  500. answer_init(&cache[ncache].answer);
  501. for (i = 0; i < answer->len; i++)
  502. answer_add(&cache[ncache].answer, answer->list[i]);
  503. cache[ncache].expiretime = now + answer->ttl;
  504. ncache++;
  505. }
  506. static int cache_find(const char *query)
  507. {
  508. int i;
  509. for (i = 0; i < ncache; i++)
  510. if (!egg_strcasecmp(cache[i].query, query)) return (i);
  511. return (-1);
  512. }
  513. void dns_cache_flush()
  514. {
  515. int i = 0;
  516. for (i = 0; i < ncache; i++) {
  517. cache_del(i);
  518. if (i == ncache) break;
  519. i--;
  520. }
  521. }
  522. static int read_thing(char *buf, char *ip)
  523. {
  524. int skip, len;
  525. skip = strspn(buf, separators);
  526. buf += skip;
  527. len = strcspn(buf, separators);
  528. egg_memcpy(ip, buf, len);
  529. ip[len] = 0;
  530. return(skip + len);
  531. }
  532. static void read_resolv(char *fname)
  533. {
  534. FILE *fp;
  535. char buf[512], ip[512];
  536. fp = fopen(fname, "r");
  537. if (!fp) return;
  538. while (fgets(buf, sizeof(buf), fp)) {
  539. if (!strncasecmp(buf, "nameserver", 10)) {
  540. read_thing(buf+10, ip);
  541. if (strlen(ip)) add_server(ip);
  542. }
  543. }
  544. fclose(fp);
  545. }
  546. static void read_hosts(char *fname)
  547. {
  548. FILE *fp;
  549. char buf[512], ip[512], host[512];
  550. int skip, n;
  551. fp = fopen(fname, "r");
  552. if (!fp) return;
  553. while (fgets(buf, sizeof(buf), fp)) {
  554. if (strchr(buf, '#')) continue;
  555. skip = read_thing(buf, ip);
  556. if (!strlen(ip)) continue;
  557. while ((n = read_thing(buf+skip, host))) {
  558. skip += n;
  559. if (strlen(host)) add_host(host, ip);
  560. }
  561. }
  562. fclose(fp);
  563. }
  564. static int make_header(char *buf, int id)
  565. {
  566. _dns_header.question_count = htons(1);
  567. // _dns_header.id = htons(id);
  568. _dns_header.id = id;
  569. egg_memcpy(buf, &_dns_header, HEAD_SIZE);
  570. return(HEAD_SIZE);
  571. }
  572. static int cut_host(const char *host, char *query)
  573. {
  574. char *period, *orig;
  575. int len;
  576. orig = query;
  577. while ((period = strchr(host, '.'))) {
  578. len = period - host;
  579. if (len > 63) return(-1);
  580. *query++ = len;
  581. egg_memcpy(query, host, len);
  582. query += len;
  583. host = period+1;
  584. }
  585. len = strlen(host);
  586. if (len) {
  587. *query++ = len;
  588. egg_memcpy(query, host, len);
  589. query += len;
  590. }
  591. *query++ = 0;
  592. return(query-orig);
  593. }
  594. static int reverse_ip(const char *host, char *reverse)
  595. {
  596. char *period;
  597. int offset, len;
  598. period = strchr(host, '.');
  599. if (!period) {
  600. len = strlen(host);
  601. egg_memcpy(reverse, host, len);
  602. return(len);
  603. }
  604. else {
  605. len = period - host;
  606. offset = reverse_ip(host+len+1, reverse);
  607. reverse[offset++] = '.';
  608. egg_memcpy(reverse+offset, host, len);
  609. reverse[offset+len] = 0;
  610. return(offset+len);
  611. }
  612. }
  613. int egg_dns_cancel(int id, int issue_callback)
  614. {
  615. dns_query_t *q, *prev;
  616. prev = NULL;
  617. for (q = query_head; q; q = q->next) {
  618. if (q->id == id) break;
  619. prev = q;
  620. }
  621. if (!q) return(-1);
  622. if (prev) prev->next = q->next;
  623. else query_head = q->next;
  624. sdprintf("Cancelling query: %s", q->query);
  625. if (issue_callback) q->callback(q->id, q->client_data, q->query, NULL);
  626. free(q);
  627. return(0);
  628. }
  629. static int skip_name(unsigned char *ptr)
  630. {
  631. int len;
  632. unsigned char *start = ptr;
  633. while ((len = *ptr++) > 0) {
  634. if (len > 63) {
  635. ptr++;
  636. break;
  637. }
  638. else {
  639. ptr += len;
  640. }
  641. }
  642. return(ptr - start);
  643. }
  644. /*
  645. void print_header(dns_header_t &header)
  646. {
  647. #define dofield(_field) sdprintf("%s: %d\n", #_field, _field)
  648. dofield(header.id);
  649. dofield(header.question_count);
  650. dofield(header.answer_count);
  651. dofield(header.ar_count);
  652. dofield(header.ns_count);
  653. #undef dofield
  654. }
  655. void print_reply(dns_rr_t &reply)
  656. {
  657. #define dofield(_field) sdprintf("%s: %d\n", #_field, _field)
  658. dofield(reply.type);
  659. dofield(reply.dclass);
  660. dofield(reply.ttl);
  661. dofield(reply.rdlength);
  662. #undef dofield
  663. }
  664. */
  665. static void parse_reply(char *response, size_t nbytes)
  666. {
  667. dns_header_t header;
  668. dns_query_t *q = NULL, *prev = NULL;
  669. dns_rr_t reply;
  670. char result[512] = "";
  671. short rr;
  672. int r = -1;
  673. unsigned const char *eop = (unsigned char *) response + nbytes;
  674. unsigned char *ptr = (unsigned char *) response;
  675. egg_memcpy(&header, ptr, HEAD_SIZE);
  676. ptr += HEAD_SIZE;
  677. /* header.id is already in our order, echoed by the server */
  678. header.flags = ntohs(header.flags);
  679. header.question_count = ntohs(header.question_count);
  680. header.answer_count = ntohs(header.answer_count);
  681. header.ar_count = ntohs(header.ar_count);
  682. header.ns_count = ntohs(header.ns_count);
  683. // print_header(header);
  684. /* Find our copy of the query before proceeding. */
  685. for (q = query_head; q; q = q->next) {
  686. if (q->id == header.id) break;
  687. prev = q;
  688. }
  689. if (!q) return;
  690. // /* destroy our async timeout */
  691. // timer_destroy(q->timer_id);
  692. /* Pass over the questions. */
  693. for (rr = 0; rr < header.question_count; rr++) {
  694. ptr += skip_name(ptr);
  695. ptr += 4;
  696. }
  697. /* End of questions. */
  698. // for (rr = 0; rr < header.answer_count + header.ar_count + header.ns_count; rr++) {
  699. for (rr = 0; rr < header.answer_count; rr++) {
  700. result[0] = 0;
  701. /* Read in the answer. */
  702. ptr += skip_name(ptr);
  703. egg_memcpy(&reply, ptr, RR_SIZE);
  704. ptr += RR_SIZE;
  705. reply.type = ntohs(reply.type);
  706. reply.dclass = ntohs(reply.dclass);
  707. reply.rdlength = ntohs(reply.rdlength);
  708. reply.ttl = ntohl(reply.ttl);
  709. /* Save the lowest ttl */
  710. if (reply.ttl && ((!q->answer.ttl) || (q->answer.ttl > reply.ttl))) q->answer.ttl = reply.ttl;
  711. // print_reply(reply);
  712. switch (reply.type) {
  713. case DNS_A:
  714. inet_ntop(AF_INET, ptr, result, 512);
  715. answer_add(&q->answer, result);
  716. break;
  717. case DNS_AAAA:
  718. inet_ntop(AF_INET6, ptr, result, 512);
  719. answer_add(&q->answer, result);
  720. break;
  721. case DNS_PTR:
  722. r = my_dn_expand((const unsigned char *) response, eop, ptr, result, sizeof(result));
  723. if (r != -1)
  724. answer_add(&q->answer, result);
  725. break;
  726. default:
  727. sdprintf("Unhandled DNS reply type: %d", reply.type);
  728. break;
  729. }
  730. ptr += reply.rdlength;
  731. }
  732. q->remaining--;
  733. /* Don't continue if we haven't gotten all expected replies. */
  734. if (q->remaining > 0) return;
  735. /* Ok, we have, so now issue the callback with the answers. */
  736. if (prev) prev->next = q->next;
  737. else query_head = q->next;
  738. cache_add(q->query, &q->answer);
  739. q->callback(q->id, q->client_data, q->query, q->answer.list);
  740. answer_free(&q->answer);
  741. free(q->query);
  742. if (q->ip)
  743. free(q->ip);
  744. free(q);
  745. }
  746. void tell_dnsdebug(int idx)
  747. {
  748. dns_query_t *q = NULL;
  749. int i, n = 0;
  750. dprintf(idx, "NS: %s\n", dns_ip);
  751. for (q = query_head; q; q = q->next)
  752. dprintf(idx, "DNS (%d) (%lis): %s\n", q->id, q->expiretime - now, q->query);
  753. // for (i = 0; i < nhosts; i++)
  754. // dprintf(idx, "HOST #%d: %s/%s\n", i, hosts[i].host, hosts[i].ip);
  755. for (i = 0; i < ncache; i++) {
  756. dprintf(idx, "cache(%d) %s expires in %lis\n", i, cache[i].query, cache[i].expiretime - now);
  757. for (n = 0; n < cache[i].answer.len; n++)
  758. dprintf(idx, "%d: %s\n", n, cache[i].answer.list[n]);
  759. }
  760. }
  761. static void expire_queries()
  762. {
  763. dns_query_t *q = NULL, *next = NULL;
  764. int i = 0;
  765. /* need to check for expired queries and either:
  766. a) recheck/change ns
  767. b) expire due to ttl
  768. */
  769. if (query_head) {
  770. for (q = query_head; q; q = q->next) {
  771. if (q->expiretime <= now) {
  772. if (q->next)
  773. next = q->next;
  774. egg_dns_cancel(q->id, 1);
  775. if (!next) break;
  776. q = next;
  777. }
  778. }
  779. }
  780. for (i = 0; i < ncache; i++) {
  781. if (cache_expired(i)) {
  782. cache_del(i);
  783. if (i == ncache) break;
  784. i--;
  785. }
  786. }
  787. }
  788. /* Read in .hosts and /etc/hosts and .resolv.conf and /etc/resolv.conf */
  789. int egg_dns_init()
  790. {
  791. _dns_header.flags = htons(1 << 8 | 1 << 7);
  792. read_resolv("/etc/resolv.conf");
  793. read_resolv(".resolv.conf");
  794. // read_hosts("/etc/hosts");
  795. read_hosts(".hosts");
  796. /* some backup servers, probably will never be used. */
  797. add_server("203.251.80.133"); //ns.abovenet.net
  798. add_server("68.2.16.30"); //some cox ns
  799. add_server("68.6.16.25"); //another cox
  800. add_server("66.254.96.53"); //reflected
  801. add_server("65.215.220.12"); //staminus
  802. add_server("69.50.170.230"); //ns1.qsi
  803. add_server("65.75.162.29"); //ns2.qsi
  804. add_server("69.50.180.62"); //ns3.qsi
  805. /* root servers for future development (tracing down)
  806. add_server("198.41.0.4");
  807. add_server("192.228.79.201");
  808. add_server("192.33.4.12");
  809. add_server("128.8.10.90");
  810. add_server("192.203.230.10");
  811. add_server("192.5.5.241");
  812. add_server("192.112.36.4");
  813. add_server("128.63.2.53");
  814. add_server("192.36.148.17");
  815. add_server("192.58.128.30");
  816. add_server("193.0.14.129");
  817. add_server("198.32.64.12");
  818. add_server("202.12.27.33");
  819. */
  820. timer_create_secs(3, "adns_check_expires", (Function) expire_queries);
  821. return(0);
  822. }
  823. bool valid_dns_id(int idx, int id)
  824. {
  825. if (id == -1)
  826. return 1;
  827. if (valid_idx(idx) && dcc[idx].dns_id && dcc[idx].dns_id == id)
  828. return 1;
  829. sdprintf("dns_id: %d is not associated with dead idx: %d", id, idx);
  830. return 0;
  831. }