adns.c 22 KB

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