adns.c 23 KB

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