odhcpd: fix coding style
[project/odhcpd.git] / src / odhcpd.c
index c411542..b1e89b3 100644 (file)
@@ -29,6 +29,7 @@
 #include <net/if.h>
 #include <netinet/ip6.h>
 #include <netpacket/packet.h>
+#include <linux/netlink.h>
 #include <linux/rtnetlink.h>
 
 #include <sys/socket.h>
@@ -54,11 +55,33 @@ 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_WARNING
+       );
+}
 
-int main()
+int main(int argc, char **argv)
 {
        openlog("odhcpd", LOG_PERROR | LOG_PID, LOG_DAEMON);
-       setlogmask(LOG_UPTO(LOG_WARNING));
+       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);
+                       break;
+               }
+       }
+       setlogmask(LOG_UPTO(log_level));
        uloop_init();
 
        if (getuid() != 0) {
@@ -188,61 +211,6 @@ ssize_t odhcpd_send(int socket, struct sockaddr_in6 *dest,
 }
 
 
-int odhcpd_iterate_interface_neighbors(const struct interface *iface,
-               void(*cb_neigh)(const struct in6_addr *addr,
-                               const struct interface *iface, void *data), void *data)
-{
-       struct {
-               struct nlmsghdr nhm;
-               struct ndmsg ndm;
-       } req = {{sizeof(req), RTM_GETNEIGH, NLM_F_REQUEST | NLM_F_DUMP,
-                       ++rtnl_seq, 0}, {AF_INET6, 0, 0, iface->ifindex, 0, 0, 0}};
-
-       if (send(rtnl_socket, &req, sizeof(req), 0) < (ssize_t)sizeof(req))
-               return -1;
-
-       uint8_t buf[8192];
-       ssize_t len = 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 -1;
-                       }
-               }
-
-               if (nhm->nlmsg_type != RTM_NEWNEIGH)
-                       break;
-
-               struct ndmsg *ndm = NLMSG_DATA(nhm);
-               if (ndm->ndm_ifindex != iface->ifindex ||
-                               !(ndm->ndm_state & (NUD_STALE | NUD_REACHABLE | NUD_PERMANENT)))
-                       continue;
-
-               struct rtattr *rta = (struct rtattr*)&ndm[1];
-               size_t alen = NLMSG_PAYLOAD(nhm, sizeof(*ndm));
-
-               while (RTA_OK(rta, alen)) {
-                       if (rta->rta_type == NDA_DST &&
-                                       RTA_PAYLOAD(rta) == sizeof(struct in6_addr)) {
-                               cb_neigh(RTA_DATA(rta), iface, data);
-                               break;
-                       } else {
-                               rta = RTA_NEXT(rta, alen);
-                       }
-               }
-
-       }
-
-       return 0;
-}
-
-
 // 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)
@@ -252,8 +220,10 @@ ssize_t odhcpd_get_interface_addresses(int ifindex,
                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))
+       if (send(rtnl_socket, &req, sizeof(req), 0) < (ssize_t)sizeof(req)) {
+               syslog(LOG_WARNING, "Request failed to dump IPv6 addresses (%s)", strerror(errno));
                return 0;
+       }
 
        uint8_t buf[8192];
        ssize_t len = 0, ret = 0;
@@ -265,68 +235,77 @@ ssize_t odhcpd_get_interface_addresses(int ifindex,
                        if (len < 0 || !NLMSG_OK(nhm, (size_t)len)) {
                                if (errno == EINTR)
                                        continue;
-                               else
-                                       return ret;
+
+                               syslog(LOG_WARNING, "Failed to receive IPv6 address rtnetlink message (%s)", strerror(errno));
+                               ret = -1;
+                               goto out;
                        }
                }
 
-               if (nhm->nlmsg_type != RTM_NEWADDR)
-                       break;
+               switch (nhm->nlmsg_type) {
+               case RTM_NEWADDR: {
+                       // Skip address but keep clearing socket buffer
+                       if (ret >= (ssize_t)cnt)
+                               continue;
 
-               // Skip address but keep clearing socket buffer
-               if (ret >= (ssize_t)cnt)
-                       continue;
+                       struct ifaddrmsg *ifa = NLMSG_DATA(nhm);
+                       if (ifa->ifa_scope != RT_SCOPE_UNIVERSE ||
+                                       (ifindex && ifa->ifa_index != (unsigned)ifindex))
+                               continue;
 
-               struct ifaddrmsg *ifa = NLMSG_DATA(nhm);
-               if (ifa->ifa_scope != RT_SCOPE_UNIVERSE ||
-                               (ifindex && ifa->ifa_index != (unsigned)ifindex))
-                       continue;
+                       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;
+                               }
 
-               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;
+                               rta = RTA_NEXT(rta, alen);
                        }
 
-                       rta = RTA_NEXT(rta, alen);
-               }
+                       if (ifa->ifa_flags & IFA_F_DEPRECATED)
+                               addrs[ret].preferred = 0;
 
-               if (ifa->ifa_flags & IFA_F_DEPRECATED)
-                       addrs[ret].preferred = 0;
+                       ++ret;
+                       break;
+               }
+               case NLMSG_DONE:
+                       goto out;
+               default:
+                       syslog(LOG_WARNING, "Unexpected rtnetlink message (%d) in response to IPv6 address dump", nhm->nlmsg_type);
+                       ret = -1;
+                       goto out;
+               }
 
-               ++ret;
        }
-
+out:
        return ret;
 }
 
-int odhcpd_get_preferred_interface_address(int ifindex, struct in6_addr *addr)
+int odhcpd_get_linklocal_interface_address(int ifindex, struct in6_addr *lladdr)
 {
-       struct odhcpd_ipaddr ipaddrs[8];
-       ssize_t ip_cnt = odhcpd_get_interface_addresses(ifindex, ipaddrs, ARRAY_SIZE(ipaddrs));
-       uint32_t preferred = 0;
-       int ret = 0;
-
-       for (ssize_t i = 0; i < ip_cnt; i++) {
-               struct odhcpd_ipaddr *ipaddr = &ipaddrs[i];
-
-               if (ipaddr->preferred > preferred || !preferred) {
-                       preferred = ipaddr->preferred;
-                       *addr = ipaddr->addr;
-                       ret = 1;
-               }
+       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;
 }
 
 void odhcpd_setup_route(const struct in6_addr *addr, int prefixlen,
@@ -425,6 +404,15 @@ 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(ret);
+       }
+
        while (true) {
                struct iovec iov = {data_buf, sizeof(data_buf)};
                struct msghdr msg = {
@@ -492,7 +480,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");
 
@@ -504,7 +491,8 @@ 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));
 }
 
 void odhcpd_process(struct odhcpd_event *event)