odhcpd: display correct default log level in usage text
[project/odhcpd.git] / src / odhcpd.c
index 24ef891d3c5be7059b6a9675ff1b2752f10a5221..9a76e4d638ee54b3a0ffcd4592c6b383479e126e 100644 (file)
@@ -29,6 +29,9 @@
 #include <net/if.h>
 #include <netinet/ip6.h>
 #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>
 #include <sys/wait.h>
 #include <sys/syscall.h>
 
+#include <netlink/msg.h>
+#include <netlink/socket.h>
+#include <netlink/attr.h>
 #include <libubox/uloop.h>
 #include "odhcpd.h"
 
 
 
 static int ioctl_sock;
-static int rtnl_socket = -1;
-static int rtnl_seq = 0;
+static struct nl_sock *rtnl_socket = NULL;
 static int urandom_fd = -1;
-
+static int log_level = LOG_INFO;
 
 static void sighandler(_unused int signal)
 {
        uloop_end();
 }
 
+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_level
+       );
+}
 
-int main()
+int main(int argc, char **argv)
 {
        openlog("odhcpd", LOG_PERROR | LOG_PID, LOG_DAEMON);
-       setlogmask(LOG_UPTO(LOG_WARNING));
+       int opt;
+
+       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);
+                       break;
+               }
+       }
+       setlogmask(LOG_UPTO(log_level));
        uloop_init();
 
        if (getuid() != 0) {
@@ -68,8 +95,8 @@ int main()
 
        ioctl_sock = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0);
 
-       if ((rtnl_socket = odhcpd_open_rtnl()) < 0) {
-               syslog(LOG_ERR, "Unable to open socket: %s", strerror(errno));
+       if (!(rtnl_socket = odhcpd_create_nl_socket(NETLINK_ROUTE))) {
+               syslog(LOG_ERR, "Unable to open nl socket: %s", strerror(errno));
                return 2;
        }
 
@@ -96,28 +123,33 @@ int main()
        return 0;
 }
 
-int odhcpd_open_rtnl(void)
+struct nl_sock *odhcpd_create_nl_socket(int protocol)
 {
-       int sock = socket(AF_NETLINK, SOCK_RAW | SOCK_CLOEXEC, NETLINK_ROUTE);
+       struct nl_sock *nl_sock;
 
-       // Connect to the kernel netlink interface
-       struct sockaddr_nl nl = {.nl_family = AF_NETLINK};
-       if (connect(sock, (struct sockaddr*)&nl, sizeof(nl))) {
-               syslog(LOG_ERR, "Failed to connect to kernel rtnetlink: %s",
-                               strerror(errno));
-               return -1;
-       }
+       nl_sock = nl_socket_alloc();
+       if (!nl_sock)
+               goto err;
+
+       if (nl_connect(nl_sock, protocol) < 0)
+               goto err;
+
+       return nl_sock;
+
+err:
+       if (nl_sock)
+               nl_socket_free(nl_sock);
 
-       return sock;
+       return NULL;
 }
 
 
 // Read IPv6 MTU for interface
-int odhcpd_get_interface_mtu(const char *ifname)
+int odhcpd_get_interface_config(const char *ifname, const char *what)
 {
        char buf[64];
-       const char *sysctl_pattern = "/proc/sys/net/ipv6/conf/%s/mtu";
-       snprintf(buf, sizeof(buf), sysctl_pattern, ifname);
+       const char *sysctl_pattern = "/proc/sys/net/ipv6/conf/%s/%s";
+       snprintf(buf, sizeof(buf), sysctl_pattern, ifname, what);
 
        int fd = open(buf, O_RDONLY);
        ssize_t len = read(fd, buf, sizeof(buf) - 1);
@@ -126,10 +158,8 @@ int odhcpd_get_interface_mtu(const char *ifname)
        if (len < 0)
                return -1;
 
-
        buf[len] = 0;
        return atoi(buf);
-
 }
 
 
@@ -176,12 +206,6 @@ ssize_t odhcpd_send(int socket, struct sockaddr_in6 *dest,
                        || IN6_IS_ADDR_MC_LINKLOCAL(&dest->sin6_addr))
                dest->sin6_scope_id = iface->ifindex;
 
-       // IPV6_PKTINFO doesn't really work for IPv6-raw sockets (bug?)
-       if (dest->sin6_port == 0) {
-               msg.msg_control = NULL;
-               msg.msg_controllen = 0;
-       }
-
        char ipbuf[INET6_ADDRSTRLEN];
        inet_ntop(AF_INET6, &dest->sin6_addr, ipbuf, sizeof(ipbuf));
 
@@ -195,99 +219,209 @@ ssize_t odhcpd_send(int socket, struct sockaddr_in6 *dest,
        return sent;
 }
 
+struct addr_info {
+       int ifindex;
+       struct odhcpd_ipaddr *addrs;
+       size_t addrs_sz;
+       int pending;
+       ssize_t ret;
+};
+
+static int cb_valid_handler(struct nl_msg *msg, void *arg)
+{
+       struct addr_info *ctxt = (struct addr_info *)arg;
+       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)
+               return NL_SKIP;
+
+       ifa = NLMSG_DATA(hdr);
+       if (ifa->ifa_scope != RT_SCOPE_UNIVERSE ||
+                       (ctxt->ifindex && ifa->ifa_index != (unsigned)ctxt->ifindex))
+               return NL_SKIP;
+
+       nlmsg_parse(hdr, sizeof(*ifa), nla, __IFA_MAX - 1, NULL);
+       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;
+
+       nla_memcpy(&ctxt->addrs[ctxt->ret].addr, nla[IFA_ADDRESS],
+                       sizeof(ctxt->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;
+       }
+
+       if (ifa->ifa_flags & IFA_F_DEPRECATED)
+               ctxt->addrs[ctxt->ret].preferred = 0;
+
+       ctxt->ret++;
+
+       return NL_OK;
+}
+
+static int cb_finish_handler(_unused struct nl_msg *msg, void *arg)
+{
+       struct addr_info *ctxt = (struct addr_info *)arg;
+
+       ctxt->pending = 0;
+
+       return NL_STOP;
+}
+
+static int cb_error_handler(_unused struct sockaddr_nl *nla, struct nlmsgerr *err,
+               void *arg)
+{
+       struct addr_info *ctxt = (struct addr_info *)arg;
+
+       ctxt->pending = 0;
+       ctxt->ret = err->error;
+
+       return NL_STOP;
+}
 
 // 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)
 {
-       struct {
-               struct nlmsghdr nhm;
-               struct ifaddrmsg ifa;
-       } req = {{sizeof(req), RTM_GETADDR, NLM_F_REQUEST | NLM_F_DUMP,
-                       ++rtnl_seq, 0}, {AF_INET6, 0, 0, 0, ifindex}};
-       if (send(rtnl_socket, &req, sizeof(req), 0) < (ssize_t)sizeof(req))
-               return 0;
-
-       uint8_t buf[8192];
-       ssize_t len = 0, ret = 0;
-
-       for (struct nlmsghdr *nhm = NULL; ; nhm = NLMSG_NEXT(nhm, len)) {
-               while (len < 0 || !NLMSG_OK(nhm, (size_t)len)) {
-                       len = recv(rtnl_socket, buf, sizeof(buf), 0);
-                       nhm = (struct nlmsghdr*)buf;
-                       if (len < 0 || !NLMSG_OK(nhm, (size_t)len)) {
-                               if (errno == EINTR)
-                                       continue;
-                               else
-                                       return ret;
-                       }
-               }
+       struct nl_msg *msg;
+       struct ifaddrmsg ifa = {
+               .ifa_family = AF_INET6,
+               .ifa_prefixlen = 0,
+               .ifa_flags = 0,
+               .ifa_scope = 0,
+               .ifa_index = ifindex, };
+       struct nl_cb *cb = nl_cb_alloc(NL_CB_DEFAULT);
+       struct addr_info ctxt = {
+               .ifindex = ifindex,
+               .addrs = addrs,
+               .addrs_sz = cnt,
+               .ret = 0,
+               .pending = 1,
+       };
 
-               if (nhm->nlmsg_type != RTM_NEWADDR)
-                       break;
+       if (!cb) {
+               ctxt.ret = -1;
+               goto out;
+       }
 
-               // Skip address but keep clearing socket buffer
-               if (ret >= (ssize_t)cnt)
-                       continue;
+       msg = nlmsg_alloc_simple(RTM_GETADDR, NLM_F_REQUEST | NLM_F_DUMP);
 
-               struct ifaddrmsg *ifa = NLMSG_DATA(nhm);
-               if (ifa->ifa_scope != RT_SCOPE_UNIVERSE ||
-                               (ifindex && ifa->ifa_index != (unsigned)ifindex))
-                       continue;
+       if (!msg) {
+               ctxt.ret = - 1;
+               goto out;
+       }
 
-               struct rtattr *rta = (struct rtattr*)&ifa[1];
-               size_t alen = NLMSG_PAYLOAD(nhm, sizeof(*ifa));
-               memset(&addrs[ret], 0, sizeof(addrs[ret]));
-               addrs[ret].prefix = ifa->ifa_prefixlen;
-
-               while (RTA_OK(rta, alen)) {
-                       if (rta->rta_type == IFA_ADDRESS) {
-                               memcpy(&addrs[ret].addr, RTA_DATA(rta),
-                                               sizeof(struct in6_addr));
-                       } else if (rta->rta_type == IFA_CACHEINFO) {
-                               struct ifa_cacheinfo *ifc = RTA_DATA(rta);
-                               addrs[ret].preferred = ifc->ifa_prefered;
-                               addrs[ret].valid = ifc->ifa_valid;
-                       }
+       nlmsg_append(msg, &ifa, sizeof(ifa), 0);
 
-                       rta = RTA_NEXT(rta, alen);
-               }
+       nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, cb_valid_handler, &ctxt);
+       nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, cb_finish_handler, &ctxt);
+       nl_cb_err(cb, NL_CB_CUSTOM, cb_error_handler, &ctxt);
+
+       nl_send_auto_complete(rtnl_socket, msg);
+       while (ctxt.pending > 0)
+               nl_recvmsgs(rtnl_socket, cb);
+
+       nlmsg_free(msg);
+out:
+       nl_cb_put(cb);
 
-               if (ifa->ifa_flags & IFA_F_DEPRECATED)
-                       addrs[ret].preferred = 0;
-
-               addrs[ret].has_class = false;
-               addrs[ret].class = 0;
-#ifdef WITH_UBUS
-               struct interface *iface = odhcpd_get_interface_by_index(ifindex);
-               if (iface)
-                       addrs[ret].has_class = ubus_get_class(iface->ifname,
-                                       &addrs[ret].addr, &addrs[ret].class);
-#endif
-               ++ret;
+       return ctxt.ret;
+}
+
+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};
+       socklen_t alen = sizeof(addr);
+       int sock = socket(AF_INET6, SOCK_RAW, IPPROTO_ICMPV6);
+
+       if (!connect(sock, (struct sockaddr*)&addr, sizeof(addr)) &&
+                       !getsockname(sock, (struct sockaddr*)&addr, &alen)) {
+               *lladdr = addr.sin6_addr;
+               status = 0;
        }
 
-       return ret;
+       close(sock);
+
+       return status;
 }
 
-int odhcpd_get_preferred_interface_address(int ifindex, struct in6_addr *addr)
+int odhcpd_setup_route(const struct in6_addr *addr, const int prefixlen,
+               const struct interface *iface, const struct in6_addr *gw,
+               const uint32_t metric, const bool add)
 {
-       struct odhcpd_ipaddr ipaddrs[8];
-       ssize_t ip_cnt = odhcpd_get_interface_addresses(ifindex, ipaddrs, ARRAY_SIZE(ipaddrs));
-       uint32_t preferred = 0;
+       struct nl_msg *msg;
+       struct rtmsg rtm = {
+               .rtm_family = AF_INET6,
+               .rtm_dst_len = prefixlen,
+               .rtm_src_len = 0,
+               .rtm_table = RT_TABLE_MAIN,
+               .rtm_protocol = (add ? RTPROT_STATIC : RTPROT_UNSPEC),
+               .rtm_scope = (add ? (gw ? RT_SCOPE_UNIVERSE : RT_SCOPE_LINK) : RT_SCOPE_NOWHERE),
+               .rtm_type = (add ? RTN_UNICAST : RTN_UNSPEC),
+       };
        int ret = 0;
 
-       for (ssize_t i = 0; i < ip_cnt; i++) {
-               struct odhcpd_ipaddr *ipaddr = &ipaddrs[i];
+       msg = nlmsg_alloc_simple(add ? RTM_NEWROUTE : RTM_DELROUTE,
+                                       add ? NLM_F_CREATE | NLM_F_REPLACE : 0);
+       if (!msg)
+               return -1;
 
-               if (ipaddr->preferred > preferred || !preferred) {
-                       preferred = ipaddr->preferred;
-                       *addr = ipaddr->addr;
-                       ret = 1;
-               }
-       }
+       nlmsg_append(msg, &rtm, sizeof(rtm), 0);
+
+       nla_put(msg, RTA_DST, sizeof(*addr), addr);
+       nla_put_u32(msg, RTA_OIF, iface->ifindex);
+       nla_put_u32(msg, RTA_PRIORITY, metric);
+
+       if (gw)
+               nla_put(msg, RTA_GATEWAY, sizeof(*gw), gw);
 
-       return ret;
+       ret = nl_send_auto_complete(rtnl_socket, msg);
+       nlmsg_free(msg);
+
+       if (ret < 0)
+               return ret;
+
+       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)
@@ -336,6 +470,20 @@ static void odhcpd_receive_packets(struct uloop_fd *u, _unused unsigned int even
                struct sockaddr_nl nl;
        } addr;
 
+       if (u->error) {
+               int ret = -1;
+               socklen_t ret_len = sizeof(ret);
+               getsockopt(u->fd, SOL_SOCKET, SO_ERROR, &ret, &ret_len);
+               u->error = false;
+               if (e->handle_error)
+                       e->handle_error(e, ret);
+       }
+
+       if (e->recv_msgs) {
+               e->recv_msgs(e);
+               return;
+       }
+
        while (true) {
                struct iovec iov = {data_buf, sizeof(data_buf)};
                struct msghdr msg = {
@@ -403,7 +551,6 @@ static void odhcpd_receive_packets(struct uloop_fd *u, _unused unsigned int even
                else if (addr.in.sin_family == AF_INET)
                        inet_ntop(AF_INET, &addr.in.sin_addr, ipbuf, sizeof(ipbuf));
 
-               syslog(LOG_DEBUG, "--");
                syslog(LOG_DEBUG, "Received %li Bytes from %s%%%s", (long)len,
                                ipbuf, (iface) ? iface->ifname : "netlink");
 
@@ -415,7 +562,14 @@ static void odhcpd_receive_packets(struct uloop_fd *u, _unused unsigned int even
 int odhcpd_register(struct odhcpd_event *event)
 {
        event->uloop.cb = odhcpd_receive_packets;
-       return uloop_fd_add(&event->uloop, ULOOP_READ);
+       return uloop_fd_add(&event->uloop, ULOOP_READ |
+                       ((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)
@@ -423,9 +577,9 @@ void odhcpd_process(struct odhcpd_event *event)
        odhcpd_receive_packets(&event->uloop, 0);
 }
 
-void odhcpd_urandom(void *data, size_t len)
+int odhcpd_urandom(void *data, size_t len)
 {
-       read(urandom_fd, data, len);
+       return read(urandom_fd, data, len);
 }