treewide: rework code to get rid of fixed IPv6 address arrays
[project/odhcpd.git] / src / odhcpd.c
index f259239..ee628a5 100644 (file)
@@ -24,6 +24,8 @@
 #include <unistd.h>
 #include <signal.h>
 #include <stdbool.h>
+#include <syslog.h>
+#include <alloca.h>
 
 #include <arpa/inet.h>
 #include <net/if.h>
@@ -31,6 +33,7 @@
 #include <netpacket/packet.h>
 #include <linux/netlink.h>
 #include <linux/if_addr.h>
+#include <linux/neighbour.h>
 #include <linux/rtnetlink.h>
 
 #include <sys/socket.h>
@@ -52,7 +55,6 @@ static int ioctl_sock;
 static struct nl_sock *rtnl_socket = NULL;
 static int urandom_fd = -1;
 
-
 static void sighandler(_unused int signal)
 {
        uloop_end();
@@ -62,9 +64,9 @@ static void print_usage(const char *app)
 {
        printf(
        "== %s Usage ==\n\n"
-        "  -h, --help   Print this help\n"
-        "  -l level     Specify log level 0..7 (default %d)\n",
-               app, LOG_WARNING
+       "  -h, --help   Print this help\n"
+       "  -l level     Specify log level 0..7 (default %d)\n",
+               app, config.log_level
        );
 }
 
@@ -72,19 +74,19 @@ int main(int argc, char **argv)
 {
        openlog("odhcpd", LOG_PERROR | LOG_PID, LOG_DAEMON);
        int opt;
-       int log_level = LOG_INFO;
+
        while ((opt = getopt(argc, argv, "hl:")) != -1) {
                switch (opt) {
                case 'h':
                        print_usage(argv[0]);
                        return 0;
                case 'l':
-                       log_level = atoi(optarg);
-                       fprintf(stderr, "Log level set to %d\n", log_level);
+                       config.log_level = (atoi(optarg) & LOG_PRIMASK);
+                       fprintf(stderr, "Log level set to %d\n", config.log_level);
                        break;
                }
        }
-       setlogmask(LOG_UPTO(log_level));
+       setlogmask(LOG_UPTO(config.log_level));
        uloop_init();
 
        if (getuid() != 0) {
@@ -94,7 +96,7 @@ int main(int argc, char **argv)
 
        ioctl_sock = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0);
 
-       if (!(rtnl_socket = odhcpd_create_nl_socket(NETLINK_ROUTE, 0))) {
+       if (!(rtnl_socket = odhcpd_create_nl_socket(NETLINK_ROUTE))) {
                syslog(LOG_ERR, "Unable to open nl socket: %s", strerror(errno));
                return 2;
        }
@@ -122,7 +124,7 @@ int main(int argc, char **argv)
        return 0;
 }
 
-struct nl_sock *odhcpd_create_nl_socket(int protocol, int groups)
+struct nl_sock *odhcpd_create_nl_socket(int protocol)
 {
        struct nl_sock *nl_sock;
 
@@ -130,9 +132,6 @@ struct nl_sock *odhcpd_create_nl_socket(int protocol, int groups)
        if (!nl_sock)
                goto err;
 
-       if (groups)
-               nl_join_groups(nl_sock, groups);
-
        if (nl_connect(nl_sock, protocol) < 0)
                goto err;
 
@@ -223,8 +222,7 @@ ssize_t odhcpd_send(int socket, struct sockaddr_in6 *dest,
 
 struct addr_info {
        int ifindex;
-       struct odhcpd_ipaddr *addrs;
-       size_t addrs_sz;
+       struct odhcpd_ipaddr **addrs;
        int pending;
        ssize_t ret;
 };
@@ -232,11 +230,12 @@ struct addr_info {
 static int cb_valid_handler(struct nl_msg *msg, void *arg)
 {
        struct addr_info *ctxt = (struct addr_info *)arg;
+       struct odhcpd_ipaddr *addrs = *(ctxt->addrs);
        struct nlmsghdr *hdr = nlmsg_hdr(msg);
        struct ifaddrmsg *ifa;
        struct nlattr *nla[__IFA_MAX];
 
-       if (hdr->nlmsg_type != RTM_NEWADDR || ctxt->ret >= (ssize_t)ctxt->addrs_sz)
+       if (hdr->nlmsg_type != RTM_NEWADDR)
                return NL_SKIP;
 
        ifa = NLMSG_DATA(hdr);
@@ -248,23 +247,28 @@ static int cb_valid_handler(struct nl_msg *msg, void *arg)
        if (!nla[IFA_ADDRESS])
                return NL_SKIP;
 
-       memset(&ctxt->addrs[ctxt->ret], 0, sizeof(ctxt->addrs[ctxt->ret]));
-       ctxt->addrs[ctxt->ret].prefix = ifa->ifa_prefixlen;
+       addrs = realloc(addrs, sizeof(*addrs)*(ctxt->ret + 1));
+       if (!addrs)
+               return NL_SKIP;
 
-       nla_memcpy(&ctxt->addrs[ctxt->ret].addr, nla[IFA_ADDRESS],
-                       sizeof(ctxt->addrs[ctxt->ret].addr));
+       memset(&addrs[ctxt->ret], 0, sizeof(addrs[ctxt->ret]));
+       addrs[ctxt->ret].prefix = ifa->ifa_prefixlen;
+
+       nla_memcpy(&addrs[ctxt->ret].addr, nla[IFA_ADDRESS],
+                       sizeof(addrs[ctxt->ret].addr));
 
        if (nla[IFA_CACHEINFO]) {
                struct ifa_cacheinfo *ifc = nla_data(nla[IFA_CACHEINFO]);
 
-               ctxt->addrs[ctxt->ret].preferred = ifc->ifa_prefered;
-               ctxt->addrs[ctxt->ret].valid = ifc->ifa_valid;
+               addrs[ctxt->ret].preferred = ifc->ifa_prefered;
+               addrs[ctxt->ret].valid = ifc->ifa_valid;
        }
 
        if (ifa->ifa_flags & IFA_F_DEPRECATED)
-               ctxt->addrs[ctxt->ret].preferred = 0;
+               addrs[ctxt->ret].preferred = 0;
 
        ctxt->ret++;
+       *(ctxt->addrs) = addrs;
 
        return NL_OK;
 }
@@ -290,8 +294,7 @@ static int cb_error_handler(_unused struct sockaddr_nl *nla, struct nlmsgerr *er
 }
 
 // Detect an IPV6-address currently assigned to the given interface
-ssize_t odhcpd_get_interface_addresses(int ifindex,
-               struct odhcpd_ipaddr *addrs, size_t cnt)
+ssize_t odhcpd_get_interface_addresses(int ifindex, struct odhcpd_ipaddr **addrs)
 {
        struct nl_msg *msg;
        struct ifaddrmsg ifa = {
@@ -304,7 +307,6 @@ ssize_t odhcpd_get_interface_addresses(int ifindex,
        struct addr_info ctxt = {
                .ifindex = ifindex,
                .addrs = addrs,
-               .addrs_sz = cnt,
                .ret = 0,
                .pending = 1,
        };
@@ -338,7 +340,7 @@ out:
        return ctxt.ret;
 }
 
-int odhcpd_get_linklocal_interface_address(int ifindex, struct in6_addr *lladdr)
+static int odhcpd_get_linklocal_interface_address(int ifindex, struct in6_addr *lladdr)
 {
        int status = -1;
        struct sockaddr_in6 addr = {AF_INET6, 0, 0, ALL_IPV6_ROUTERS, ifindex};
@@ -356,9 +358,54 @@ int odhcpd_get_linklocal_interface_address(int ifindex, struct in6_addr *lladdr)
        return status;
 }
 
-int odhcpd_setup_route(const struct in6_addr *addr, int prefixlen,
+/*
+ * DNS address selection criteria order :
+ * - use IPv6 address with valid lifetime if none is yet selected
+ * - use IPv6 address with a preferred lifetime if the already selected IPv6 address is deprecated
+ * - use an IPv6 ULA address if the already selected IPv6 address is not an ULA address
+ * - use the IPv6 address with the longest preferred lifetime
+ */
+int odhcpd_get_interface_dns_addr(const struct interface *iface, struct in6_addr *addr)
+{
+       time_t now = odhcpd_time();
+       ssize_t m = -1;
+
+       for (size_t i = 0; i < iface->ia_addr_len; ++i) {
+               if (iface->ia_addr[i].valid <= (uint32_t)now)
+                       continue;
+
+               if (m < 0) {
+                       m = i;
+                       continue;
+               }
+
+               if (iface->ia_addr[m].preferred >= (uint32_t)now &&
+                               iface->ia_addr[i].preferred < (uint32_t)now)
+                       continue;
+
+               if (IN6_IS_ADDR_ULA(&iface->ia_addr[i].addr)) {
+                       if (!IN6_IS_ADDR_ULA(&iface->ia_addr[m].addr)) {
+                               m = i;
+                               continue;
+                       }
+               } else if (IN6_IS_ADDR_ULA(&iface->ia_addr[m].addr))
+                       continue;
+
+               if (iface->ia_addr[i].preferred > iface->ia_addr[m].preferred)
+                       m = i;
+       }
+
+       if (m >= 0) {
+               *addr = iface->ia_addr[m].addr;
+               return 0;
+       }
+
+       return odhcpd_get_linklocal_interface_address(iface->ifindex, addr);
+}
+
+int odhcpd_setup_route(const struct in6_addr *addr, const int prefixlen,
                const struct interface *iface, const struct in6_addr *gw,
-               uint32_t metric, bool add)
+               const uint32_t metric, const bool add)
 {
        struct nl_msg *msg;
        struct rtmsg rtm = {
@@ -395,6 +442,37 @@ int odhcpd_setup_route(const struct in6_addr *addr, int prefixlen,
        return nl_wait_for_ack(rtnl_socket);
 }
 
+int odhcpd_setup_proxy_neigh(const struct in6_addr *addr,
+               const struct interface *iface, const bool add)
+{
+       struct nl_msg *msg;
+       struct ndmsg ndm = {
+               .ndm_family = AF_INET6,
+               .ndm_flags = NTF_PROXY,
+               .ndm_ifindex = iface->ifindex,
+       };
+       int ret = 0, flags = NLM_F_REQUEST;
+
+       if (add)
+               flags |= NLM_F_REPLACE | NLM_F_CREATE;
+
+       msg = nlmsg_alloc_simple(add ? RTM_NEWNEIGH : RTM_DELNEIGH, flags);
+       if (!msg)
+               return -1;
+
+       nlmsg_append(msg, &ndm, sizeof(ndm), 0);
+
+       nla_put(msg, NDA_DST, sizeof(*addr), addr);
+
+       ret = nl_send_auto_complete(rtnl_socket, msg);
+       nlmsg_free(msg);
+
+       if (ret < 0)
+               return ret;
+
+       return nl_wait_for_ack(rtnl_socket);
+}
+
 struct interface* odhcpd_get_interface_by_index(int ifindex)
 {
        struct interface *iface;
@@ -447,7 +525,12 @@ static void odhcpd_receive_packets(struct uloop_fd *u, _unused unsigned int even
                getsockopt(u->fd, SOL_SOCKET, SO_ERROR, &ret, &ret_len);
                u->error = false;
                if (e->handle_error)
-                       e->handle_error(ret);
+                       e->handle_error(e, ret);
+       }
+
+       if (e->recv_msgs) {
+               e->recv_msgs(e);
+               return;
        }
 
        while (true) {
@@ -532,6 +615,12 @@ int odhcpd_register(struct odhcpd_event *event)
                        ((event->handle_error) ? ULOOP_ERROR_CB : 0));
 }
 
+int odhcpd_deregister(struct odhcpd_event *event)
+{
+       event->uloop.cb = NULL;
+       return uloop_fd_delete(&event->uloop);
+}
+
 void odhcpd_process(struct odhcpd_event *event)
 {
        odhcpd_receive_packets(&event->uloop, 0);