X-Git-Url: http://git.archive.openwrt.org/?p=project%2Fmdnsd.git;a=blobdiff_plain;f=main.c;h=c23daf4f2715453c1616e4e3312bdf1c56ae3878;hp=baadda151957a49f10bd549150befef22b1273e5;hb=2c1fe9afacc9379dcfd0db911fc9ee45adfdc0f4;hpb=70ec71015a42e6e4e04b0281b80ec80e5f5f959a diff --git a/main.c b/main.c index baadda1..c23daf4 100644 --- a/main.c +++ b/main.c @@ -28,9 +28,7 @@ #include #include -#include #include -#include #include "dns.h" #include "ubus.h" @@ -40,154 +38,8 @@ #include "announce.h" #include "interface.h" -static struct uloop_timeout reconnect; char *iface_name = "eth0"; -static int -parse_answer(struct interface *iface, uint8_t *buffer, int len, uint8_t **b, int *rlen, int cache) -{ - char *name = dns_consume_name(buffer, len, b, rlen); - struct dns_answer *a; - uint8_t *rdata; - - if (!name) { - fprintf(stderr, "dropping: bad question\n"); - return -1; - } - - a = dns_consume_answer(b, rlen); - if (!a) { - fprintf(stderr, "dropping: bad question\n"); - return -1; - } - - rdata = *b; - if (a->rdlength > *rlen) { - fprintf(stderr, "dropping: bad question\n"); - return -1; - } - - *rlen -= a->rdlength; - *b += a->rdlength; - - if (cache) - cache_answer(iface, buffer, len, name, a, rdata); - - return 0; -} - -static void -parse_question(struct interface *iface, char *name, struct dns_question *q) -{ - char *host; - - DBG(1, "Q -> %s %s\n", dns_type_string(q->type), name); - - switch (q->type) { - case TYPE_ANY: - host = service_name("local"); - if (!strcmp(name, host)) - service_reply(iface, NULL); - break; - - case TYPE_PTR: - service_announce_services(iface, name); - service_reply(iface, name); - break; - - case TYPE_AAAA: - case TYPE_A: - host = strstr(name, ".local"); - if (host) - *host = '\0'; - if (!strcmp(hostname, name)) - service_reply_a(iface, q->type); - break; - }; -} - -static void -read_socket(struct uloop_fd *u, unsigned int events) -{ - struct interface *iface = container_of(u, struct interface, fd); - static uint8_t buffer[8 * 1024]; - uint8_t *b = buffer; - struct dns_header *h; - int len, rlen; - - if (u->eof) { - uloop_fd_delete(u); - close(u->fd); - u->fd = -1; - uloop_timeout_set(&reconnect, 1000); - return; - } - - rlen = len = read(u->fd, buffer, sizeof(buffer)); - if (len < 1) { - fprintf(stderr, "read failed: %s\n", strerror(errno)); - return; - } - - h = dns_consume_header(&b, &rlen); - if (!h) { - fprintf(stderr, "dropping: bad header\n"); - return; - } - - while (h->questions-- > 0) { - char *name = dns_consume_name(buffer, len, &b, &rlen); - struct dns_question *q; - - if (!name) { - fprintf(stderr, "dropping: bad name\n"); - return; - } - - q = dns_consume_question(&b, &rlen); - if (!q) { - fprintf(stderr, "dropping: bad question\n"); - return; - } - - if (!(h->flags & FLAG_RESPONSE)) - parse_question(iface, name, q); - } - - if (!(h->flags & FLAG_RESPONSE)) - return; - - while (h->answers-- > 0) - parse_answer(iface, buffer, len, &b, &rlen, 1); - - while (h->authority-- > 0) - parse_answer(iface, buffer, len, &b, &rlen, 0); - - while (h->additional-- > 0) - parse_answer(iface, buffer, len, &b, &rlen, 1); -} - -static void -reconnect_socket(struct uloop_timeout *timeout) -{ - cur_iface->fd.fd = usock(USOCK_UDP | USOCK_SERVER | USOCK_NONBLOCK, MCAST_ADDR, "5353"); - if (cur_iface->fd.fd < 0) { - fprintf(stderr, "failed to add listener: %s\n", strerror(errno)); - uloop_timeout_set(&reconnect, 1000); - } else { - if (interface_socket_setup(cur_iface)) { - uloop_timeout_set(&reconnect, 1000); - cur_iface->fd.fd = -1; - return; - } - - uloop_fd_add(&cur_iface->fd, ULOOP_READ); - sleep(5); - dns_send_question(cur_iface, "_services._dns-sd._udp.local", TYPE_PTR); - announce_init(); - } -} - int main(int argc, char **argv) { @@ -224,9 +76,6 @@ main(int argc, char **argv) return -1; } - if (!cur_iface) - return -1; - signal_setup(); if (cache_init()) @@ -234,16 +83,13 @@ main(int argc, char **argv) service_init(); - cur_iface->fd.cb = read_socket; - reconnect.cb = reconnect_socket; - - uloop_timeout_set(&reconnect, 100); ubus_startup(); uloop_run(); uloop_done(); cache_cleanup(); service_cleanup(); + vlist_flush(&interfaces); return 0; }