#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/nameser.h>
-#include <asm/byteorder.h>
+#include <udebug.h>
#include <libubus.h>
-#include <libubox/usock.h>
#include <libubox/uloop.h>
-#include <libubox/avl-cmp.h>
#include "dns.h"
#include "ubus.h"
#include "announce.h"
#include "interface.h"
-static struct uloop_timeout reconnect;
-char *iface_name = "eth0";
+int cfg_proto = 0;
+int cfg_no_subnet = 0;
+
+static struct udebug ud;
+static struct udebug_buf udb;
+static bool udebug_enabled;
static void
-read_socket(struct uloop_fd *u, unsigned int events)
+umdns_udebug_vprintf(const char *format, va_list ap)
{
- struct interface *iface = container_of(u, struct interface, fd);
- static uint8_t buffer[8 * 1024];
- int len;
-
- if (u->eof) {
- uloop_fd_delete(u);
- close(u->fd);
- u->fd = -1;
- uloop_timeout_set(&reconnect, 1000);
+ if (!udebug_enabled)
return;
- }
- len = read(u->fd, buffer, sizeof(buffer));
- if (len < 1) {
- fprintf(stderr, "read failed: %s\n", strerror(errno));
+ udebug_entry_init(&udb);
+ udebug_entry_vprintf(&udb, format, ap);
+ udebug_entry_add(&udb);
+}
+
+void umdns_udebug_printf(const char *format, ...)
+{
+ va_list ap;
+
+ va_start(ap, format);
+ umdns_udebug_vprintf(format, ap);
+ va_end(ap);
+}
+
+void umdns_udebug_set_enabled(bool val)
+{
+ static const struct udebug_buf_meta meta = {
+ .name = "umdns_log",
+ .format = UDEBUG_FORMAT_STRING,
+ };
+
+ if (udebug_enabled == val)
+ return;
+
+ udebug_enabled = val;
+ if (!val) {
+ udebug_buf_free(&udb);
+ udebug_free(&ud);
return;
}
- dns_handle_packet(iface, buffer, len);
+ udebug_init(&ud);
+ udebug_auto_connect(&ud, NULL);
+ udebug_buf_init(&udb, 1024, 64 * 1024);
+ udebug_buf_add(&ud, &udb, &meta);
}
static void
-reconnect_socket(struct uloop_timeout *timeout)
+signal_shutdown(int signal)
{
- 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(cur_iface);
- }
+ uloop_end();
}
int
{
int ch, ttl;
- while ((ch = getopt(argc, argv, "h:t:i:d")) != -1) {
+ uloop_init();
+
+ while ((ch = getopt(argc, argv, "t:i:d46n")) != -1) {
switch (ch) {
- case 'h':
- hostname = optarg;
- break;
case 't':
ttl = atoi(optarg);
if (ttl > 0)
debug++;
break;
case 'i':
- iface_name = optarg;
+ interface_add(optarg);
+ break;
+ case '4':
+ cfg_proto = 4;
+ break;
+ case '6':
+ cfg_proto = 6;
+ break;
+ case 'n':
+ cfg_no_subnet = 1;
break;
- }
- }
-
- if (!iface_name)
- return -1;
-
- uloop_init();
- if (interface_add(iface_name)) {
- fprintf(stderr, "Failed to add interface %s\n", iface_name);
- return -1;
+ default:
+ return -1;
+ }
}
- if (!cur_iface)
- return -1;
-
- signal_setup();
+ signal(SIGPIPE, SIG_IGN);
+ signal(SIGTERM, signal_shutdown);
+ signal(SIGKILL, signal_shutdown);
if (cache_init())
return -1;
- service_init();
+ ubus_startup();
- cur_iface->fd.cb = read_socket;
- reconnect.cb = reconnect_socket;
+ service_init(0);
- uloop_timeout_set(&reconnect, 100);
- ubus_startup();
uloop_run();
uloop_done();
- cache_cleanup();
+ interface_shutdown();
+ cache_cleanup(NULL);
service_cleanup();
+ vlist_flush(&interfaces);
return 0;
}