Loop until ubus connection succeeds
[project/odhcpd.git] / src / config.c
index ebc6642..f008c12 100644 (file)
@@ -1,6 +1,8 @@
+#include <fcntl.h>
 #include <resolv.h>
 #include <signal.h>
 #include <arpa/inet.h>
+#include <unistd.h>
 
 #include <uci.h>
 #include <uci_blob.h>
@@ -8,6 +10,7 @@
 #include "odhcpd.h"
 
 static struct blob_buf b;
+static int reload_pipe[2];
 struct list_head leases = LIST_HEAD_INIT(leases);
 struct list_head interfaces = LIST_HEAD_INIT(interfaces);
 struct config config = {false, NULL, NULL};
@@ -254,34 +257,45 @@ err:
 }
 
 
-int config_parse_interface(struct blob_attr *b, const char *name, bool overwrite)
+int config_parse_interface(void *data, size_t len, const char *name, bool overwrite)
 {
        struct blob_attr *tb[IFACE_ATTR_MAX], *c;
-       blobmsg_parse(iface_attrs, IFACE_ATTR_MAX, tb, blob_data(b), blob_len(b));
+       blobmsg_parse(iface_attrs, IFACE_ATTR_MAX, tb, data, len);
 
        if (tb[IFACE_ATTR_INTERFACE])
-               name = blobmsg_data(tb[IFACE_ATTR_INTERFACE]);
+               name = blobmsg_get_string(tb[IFACE_ATTR_INTERFACE]);
+
+       if (!name)
+               return -1;
 
        struct interface *iface = get_interface(name);
        if (!iface) {
                iface = calloc(1, sizeof(*iface));
                strncpy(iface->name, name, sizeof(iface->name) - 1);
                list_add(&iface->head, &interfaces);
-       } else {
+       } else if (overwrite) {
                clean_interface(iface);
        }
 
        const char *ifname = NULL;
 #ifdef WITH_UBUS
-       if (overwrite)
+       if (overwrite || !iface->ifname[0])
                ifname = ubus_get_ifname(name);
 #endif
-       if ((c = tb[IFACE_ATTR_IFNAME]))
-               ifname = blobmsg_get_string(c);
-       else if ((c = tb[IFACE_ATTR_NETWORKID]))
-               ifname = blobmsg_get_string(c);
 
-       strncpy(iface->ifname, ifname, sizeof(iface->ifname) - 1);
+       if (overwrite) {
+               if ((c = tb[IFACE_ATTR_IFNAME]))
+                       ifname = blobmsg_get_string(c);
+               else if ((c = tb[IFACE_ATTR_NETWORKID]))
+                       ifname = blobmsg_get_string(c);
+       }
+
+       if (!iface->ifname[0] && !ifname)
+               return -1;
+
+       if (ifname)
+               strncpy(iface->ifname, ifname, sizeof(iface->ifname) - 1);
+
        iface->inuse = true;
 
        if (overwrite)
@@ -290,7 +304,7 @@ int config_parse_interface(struct blob_attr *b, const char *name, bool overwrite
        if ((c = tb[IFACE_ATTR_DYNAMICDHCP]))
                iface->no_dynamic_dhcp = !blobmsg_get_bool(c);
 
-       if ((c = tb[IFACE_ATTR_IGNORE]))
+       if (overwrite && (c = tb[IFACE_ATTR_IGNORE]))
                iface->ignore = blobmsg_get_bool(c);
 
        if ((c = tb[IFACE_ATTR_LEASETIME])) {
@@ -329,9 +343,9 @@ int config_parse_interface(struct blob_attr *b, const char *name, bool overwrite
        if ((c = tb[IFACE_ATTR_MASTER]))
                iface->master = blobmsg_get_bool(c);
 
-       if ((c = tb[IFACE_ATTR_UPSTREAM])) {
+       if (overwrite && (c = tb[IFACE_ATTR_UPSTREAM])) {
                struct blob_attr *cur;
-               int rem;
+               unsigned rem;
 
                blobmsg_for_each_attr(cur, c, rem) {
                        if (blobmsg_type(cur) != BLOBMSG_TYPE_STRING || !blobmsg_check_attr(cur, NULL))
@@ -344,25 +358,38 @@ int config_parse_interface(struct blob_attr *b, const char *name, bool overwrite
                }
        }
 
-       if ((c = tb[IFACE_ATTR_RA]))
-               if ((iface->ra = parse_mode(blobmsg_get_string(c))) < 0)
+       int mode;
+       if ((c = tb[IFACE_ATTR_RA])) {
+               if ((mode = parse_mode(blobmsg_get_string(c))) >= 0)
+                       iface->ra = mode;
+               else
                        goto err;
+       }
 
-       if ((c = tb[IFACE_ATTR_DHCPV4]))
-               if ((iface->dhcpv4 = parse_mode(blobmsg_get_string(c))) < 0)
+       if ((c = tb[IFACE_ATTR_DHCPV4])) {
+               if ((mode = parse_mode(blobmsg_get_string(c))) >= 0)
+                       iface->dhcpv4 = mode;
+               else
                        goto err;
+       }
 
-       if ((c = tb[IFACE_ATTR_DHCPV6]))
-               if ((iface->dhcpv6 = parse_mode(blobmsg_get_string(c))) < 0)
+       if ((c = tb[IFACE_ATTR_DHCPV6])) {
+               if ((mode = parse_mode(blobmsg_get_string(c))) >= 0)
+                       iface->dhcpv6 = mode;
+               else
                        goto err;
+       }
 
-       if ((c = tb[IFACE_ATTR_NDP]))
-               if ((iface->ndp = parse_mode(blobmsg_get_string(c))) < 0)
+       if ((c = tb[IFACE_ATTR_NDP])) {
+               if ((mode = parse_mode(blobmsg_get_string(c))) >= 0)
+                       iface->ndp = mode;
+               else
                        goto err;
+       }
 
        if ((c = tb[IFACE_ATTR_DNS])) {
                struct blob_attr *cur;
-               int rem;
+               unsigned rem;
 
                iface->always_rewrite_dns = true;
                blobmsg_for_each_attr(cur, c, rem) {
@@ -387,7 +414,7 @@ int config_parse_interface(struct blob_attr *b, const char *name, bool overwrite
 
        if ((c = tb[IFACE_ATTR_DOMAIN])) {
                struct blob_attr *cur;
-               int rem;
+               unsigned rem;
 
                blobmsg_for_each_attr(cur, c, rem) {
                        if (blobmsg_type(cur) != BLOBMSG_TYPE_STRING || !blobmsg_check_attr(cur, NULL))
@@ -437,7 +464,7 @@ int config_parse_interface(struct blob_attr *b, const char *name, bool overwrite
 
        if ((c = tb[IFACE_ATTR_NDPROXY_STATIC])) {
                struct blob_attr *cur;
-               int rem;
+               unsigned rem;
 
                blobmsg_for_each_attr(cur, c, rem) {
                        if (blobmsg_type(cur) != BLOBMSG_TYPE_STRING || !blobmsg_check_attr(cur, NULL))
@@ -462,110 +489,134 @@ static int set_interface(struct uci_section *s)
 {
        blob_buf_init(&b, 0);
        uci_to_blob(&b, s, &interface_attr_list);
-       return config_parse_interface(b.head, s->e.name, true);
+       return config_parse_interface(blob_data(b.head), blob_len(b.head), s->e.name, true);
 }
 
 
-static volatile bool do_reload = false;
-static void set_stop(int signal)
-{
-       uloop_end();
-       do_reload = (signal == SIGHUP);
-}
-
-void odhcpd_run(void)
+void odhcpd_reload(void)
 {
        struct uci_context *uci = uci_alloc_context();
-       signal(SIGTERM, set_stop);
-       signal(SIGHUP, set_stop);
-       signal(SIGINT, set_stop);
-
-       do {
-               do_reload = false;
-
-               struct lease *l;
-               list_for_each_entry(l, &leases, head) {
-                       list_del(&l->head);
-                       free(l->duid);
-                       free(l);
-               }
+       struct lease *l;
+       list_for_each_entry(l, &leases, head) {
+               list_del(&l->head);
+               free(l->duid);
+               free(l);
+       }
 
-               struct uci_package *dhcp = NULL;
-               if (!uci_load(uci, "dhcp", &dhcp)) {
-                       struct uci_element *e;
-                       uci_foreach_element(&dhcp->sections, e) {
-                               struct uci_section *s = uci_to_section(e);
-                               if (!strcmp(s->type, "lease"))
-                                       set_lease(s);
-                               else if (!strcmp(s->type, "odhcpd"))
-                                       set_config(s);
-                       }
+       struct uci_package *dhcp = NULL;
+       if (!uci_load(uci, "dhcp", &dhcp)) {
+               struct uci_element *e;
+               uci_foreach_element(&dhcp->sections, e) {
+                       struct uci_section *s = uci_to_section(e);
+                       if (!strcmp(s->type, "lease"))
+                               set_lease(s);
+                       else if (!strcmp(s->type, "odhcpd"))
+                               set_config(s);
+               }
 
-                       uci_foreach_element(&dhcp->sections, e) {
-                               struct uci_section *s = uci_to_section(e);
-                               if (!strcmp(s->type, "dhcp"))
-                                       set_interface(s);
-                       }
+               uci_foreach_element(&dhcp->sections, e) {
+                       struct uci_section *s = uci_to_section(e);
+                       if (!strcmp(s->type, "dhcp"))
+                               set_interface(s);
                }
+       }
 
 #ifdef WITH_UBUS
-               ubus_apply_network();
+       ubus_apply_network();
 #endif
 
-               // Evaluate hybrid mode for master
-               struct interface *master = NULL, *i;
-               list_for_each_entry(i, &interfaces, head) {
-                       if (!i->master)
-                               continue;
+       // Evaluate hybrid mode for master
+       struct interface *master = NULL, *i, *n;
+       list_for_each_entry(i, &interfaces, head) {
+               if (!i->master)
+                       continue;
 
-                       enum odhcpd_mode hybrid_mode = RELAYD_DISABLED;
+               enum odhcpd_mode hybrid_mode = RELAYD_DISABLED;
 #ifdef WITH_UBUS
-                       if (ubus_has_prefix(i->name, i->ifname))
-                               hybrid_mode = RELAYD_RELAY;
+               if (ubus_has_prefix(i->name, i->ifname))
+                       hybrid_mode = RELAYD_RELAY;
 #endif
 
+               if (i->dhcpv6 == RELAYD_HYBRID)
+                       i->dhcpv6 = hybrid_mode;
+
+               if (i->ra == RELAYD_HYBRID)
+                       i->ra = hybrid_mode;
+
+               if (i->ndp == RELAYD_HYBRID)
+                       i->ndp = hybrid_mode;
+
+               if (i->dhcpv6 == RELAYD_RELAY || i->ra == RELAYD_RELAY || i->ndp == RELAYD_RELAY)
+                       master = i;
+       }
+
+
+       list_for_each_entry_safe(i, n, &interfaces, head) {
+               if (i->inuse) {
+                       // Resolve hybrid mode
                        if (i->dhcpv6 == RELAYD_HYBRID)
-                               i->dhcpv6 = hybrid_mode;
+                               i->dhcpv6 = (master && master->dhcpv6 == RELAYD_RELAY) ?
+                                               RELAYD_RELAY : RELAYD_SERVER;
 
                        if (i->ra == RELAYD_HYBRID)
-                               i->ra = hybrid_mode;
+                               i->ra = (master && master->ra == RELAYD_RELAY) ?
+                                               RELAYD_RELAY : RELAYD_SERVER;
 
                        if (i->ndp == RELAYD_HYBRID)
-                               i->ndp = hybrid_mode;
-
-                       if (i->dhcpv6 == RELAYD_RELAY || i->ra == RELAYD_RELAY || i->ndp == RELAYD_RELAY)
-                               master = i;
+                               i->ndp = (master && master->ndp == RELAYD_RELAY) ?
+                                               RELAYD_RELAY : RELAYD_SERVER;
+
+                       setup_router_interface(i, true);
+                       setup_dhcpv6_interface(i, true);
+                       setup_ndp_interface(i, true);
+                       setup_dhcpv4_interface(i, true);
+                       i->inuse = false;
+               } else {
+                       close_interface(i);
                }
+       }
+
+       uci_unload(uci, dhcp);
+       uci_free_context(uci);
+}
+
 
+static void handle_signal(int signal)
+{
+       char b[1] = {0};
+       if (signal == SIGHUP)
+               write(reload_pipe[1], b, sizeof(b));
+       else
+               uloop_end();
+}
 
-               list_for_each_entry(i, &interfaces, head) {
-                       if (i->inuse && !i->ignore) {
-                               // Resolve hybrid mode
-                               if (i->dhcpv6 == RELAYD_HYBRID)
-                                       i->dhcpv6 = (master && master->dhcpv6 == RELAYD_RELAY) ?
-                                                       RELAYD_RELAY : RELAYD_SERVER;
 
-                               if (i->ra == RELAYD_HYBRID)
-                                       i->ra = (master && master->ra == RELAYD_RELAY) ?
-                                                       RELAYD_RELAY : RELAYD_SERVER;
 
-                               if (i->ndp == RELAYD_HYBRID)
-                                       i->ndp = (master && master->ndp == RELAYD_RELAY) ?
-                                                       RELAYD_RELAY : RELAYD_SERVER;
+static void reload_cb(struct uloop_fd *u, _unused unsigned int events)
+{
+       char b[512];
+       read(u->fd, b, sizeof(b));
+       odhcpd_reload();
+}
 
-                               setup_router_interface(i, true);
-                               setup_dhcpv6_interface(i, true);
-                               setup_ndp_interface(i, true);
-                               setup_dhcpv4_interface(i, true);
-                       } else {
-                               close_interface(i);
-                       }
-               }
+static struct uloop_fd reload_fd = { .cb = reload_cb };
 
-               uloop_run();
+void odhcpd_run(void)
+{
+       pipe2(reload_pipe, O_NONBLOCK | O_CLOEXEC);
+       reload_fd.fd = reload_pipe[0];
+       uloop_fd_add(&reload_fd, ULOOP_READ);
+
+       signal(SIGTERM, handle_signal);
+       signal(SIGINT, handle_signal);
+       signal(SIGHUP, handle_signal);
+
+#ifdef WITH_UBUS
+       while (init_ubus())
+               sleep(1);
+#endif
 
-               if (dhcp)
-                       uci_unload(uci, dhcp);
-       } while (do_reload);
+       odhcpd_reload();
+       uloop_run();
 }