dhcpv4: add ubus_get_address4 ubus_get_mask4 function

[project/odhcpd.git] / src / dhcpv4.c

diff --git a/src/dhcpv4.c b/src/dhcpv4.c
index 89750b2..99ad658 100644 (file)
--- a/src/dhcpv4.c
+++ b/src/dhcpv4.c
@@ -119,18 +119,13 @@ int setup_dhcpv4_interface(struct interface *iface, bool enable)
                }
 
                // Create a range if not specified
-               struct ifreq ifreq;
-               strncpy(ifreq.ifr_name, iface->ifname, sizeof(ifreq.ifr_name));
-
-               struct sockaddr_in *saddr = (struct sockaddr_in*)&ifreq.ifr_addr;
-               struct sockaddr_in *smask = (struct sockaddr_in*)&ifreq.ifr_netmask;
                if (!(iface->dhcpv4_start.s_addr & htonl(0xffff0000)) &&
-                               !(iface->dhcpv4_end.s_addr & htonl(0xffff0000)) &&
-                               !ioctl(sock, SIOCGIFADDR, &ifreq)) {
-                       struct in_addr addr = saddr->sin_addr;
+                               !(iface->dhcpv4_end.s_addr & htonl(0xffff0000))) {
 
-                       ioctl(sock, SIOCGIFNETMASK, &ifreq);
-                       struct in_addr mask = smask->sin_addr;
+                       struct in_addr *saddr = ubus_get_address4(iface->name);
+                       struct in_addr addr = { .s_addr = saddr->s_addr } ;
+                       struct in_addr *smask = ubus_get_mask4(iface->name);
+                       struct in_addr mask = { .s_addr = smask->s_addr } ;
 
                        uint32_t start = ntohl(iface->dhcpv4_start.s_addr);
                        uint32_t end = ntohl(iface->dhcpv4_end.s_addr);
@@ -202,9 +197,11 @@ int setup_dhcpv4_interface(struct interface *iface, bool enable)
 
                // Clean invalid assignments
                struct dhcpv4_assignment *a, *n;
+               struct in_addr *smask = ubus_get_mask4(iface->name);
+               struct in_addr mask = { .s_addr = smask->s_addr } ;
                list_for_each_entry_safe(a, n, &iface->dhcpv4_assignments, head) {
-                       if ((htonl(a->addr) & smask->sin_addr.s_addr) !=
-                                       (iface->dhcpv4_start.s_addr & smask->sin_addr.s_addr)) {
+                       if ((htonl(a->addr) & mask.s_addr) !=
+                                       (iface->dhcpv4_start.s_addr & mask.s_addr)) {
                                list_del(&a->head);
                                free(a);
                        }
@@ -222,7 +219,6 @@ int setup_dhcpv4_interface(struct interface *iface, bool enable)
                        struct dhcpv4_assignment *a = list_first_entry(&iface->dhcpv4_assignments,
                                        struct dhcpv4_assignment, head);
                        list_del(&a->head);
-                       free(a->hostname);
                        free(a);
                }
 
@@ -502,50 +498,74 @@ static void handle_dhcpv4(void *addr, void *data, size_t len,
                        (struct sockaddr*)&dest, sizeof(dest));
 }
 
+static bool dhcpv4_test(struct interface *iface, uint32_t try)
+{
+       struct dhcpv4_assignment *c;
+       list_for_each_entry(c, &iface->dhcpv4_assignments, head) {
+               if (c->addr == try) {
+                       return false;
+               }
+       }
+       return true;
+}
 
 static bool dhcpv4_assign(struct interface *iface,
                struct dhcpv4_assignment *assign, uint32_t raddr)
 {
-       const unsigned tries = 10;
        uint32_t start = ntohl(iface->dhcpv4_start.s_addr);
        uint32_t end = ntohl(iface->dhcpv4_end.s_addr);
        uint32_t count = end - start + 1;
 
-       // Seed RNG with checksum of DUID
+       // try to assign the IP the client asked for
+       if (start <= raddr && raddr <= end && dhcpv4_test(iface, raddr)) {
+               assign->addr = raddr;
+               list_add(&assign->head, &iface->dhcpv4_assignments);
+               syslog(LOG_DEBUG, "assigning the IP the client asked for: %u.%u.%u.%u",
+                               (assign->addr & 0xff000000) >> 24,
+                               (assign->addr & 0x00ff0000) >> 16,
+                               (assign->addr & 0x0000ff00) >> 8,
+                               (assign->addr & 0x000000ff));
+               return true;
+       }
+
+       // Seed RNG with checksum of hwaddress
        uint32_t seed = 0;
-       for (size_t i = 0; i < sizeof(assign->hwaddr); ++i)
-               seed += assign->hwaddr[i];
+       for (size_t i = 0; i < sizeof(assign->hwaddr); ++i) {
+               // Knuth's multiplicative method
+               uint8_t o = assign->hwaddr[i];
+               seed += (o*2654435761) % UINT32_MAX;
+       }
        srand(seed);
 
-       // Try to assign up to 100x
-       for (unsigned i = 0; i < tries; ++i) {
-               uint32_t try = (((uint32_t)rand()) % count) + start;
-               if (i == 0 && raddr >= start && raddr <= end)
-                       try = raddr;
-               else if (i == tries - 1)
-                       try = start;
+       uint32_t try = (((uint32_t)rand()) % count) + start;
+
+       if (list_empty(&iface->dhcpv4_assignments)) {
+               assign->addr = try;
+               list_add(&assign->head, &iface->dhcpv4_assignments);
+               syslog(LOG_DEBUG, "assigning mapped IP (empty list): %u.%u.%u.%u",
+                               (assign->addr & 0xff000000) >> 24,
+                               (assign->addr & 0x00ff0000) >> 16,
+                               (assign->addr & 0x0000ff00) >> 8,
+                               (assign->addr & 0x000000ff));
+               return true;
+       }
 
-               if (list_empty(&iface->dhcpv4_assignments)) {
+       for (uint32_t i = 0; i < count; ++i) {
+               if (dhcpv4_test(iface, try)) {
+                       /* test was successful: IP address is not assigned, assign it */
                        assign->addr = try;
                        list_add(&assign->head, &iface->dhcpv4_assignments);
+                       syslog(LOG_DEBUG, "assigning mapped IP: %u.%u.%u.%u (try %u of %u)",
+                                       (assign->addr & 0xff000000) >> 24,
+                                       (assign->addr & 0x00ff0000) >> 16,
+                                       (assign->addr & 0x0000ff00) >> 8,
+                                       (assign->addr & 0x000000ff), i, count);
                        return true;
                }
-
-               struct dhcpv4_assignment *c;
-               list_for_each_entry(c, &iface->dhcpv4_assignments, head) {
-                       if (c->addr > try) {
-                               assign->addr = try;
-                               list_add_tail(&assign->head, &c->head);
-                               return true;
-                       } else if (c->addr == try) {
-                               if (i < tries - 1)
-                                       break;
-                               else
-                                       ++try;
-                       }
-               }
+               try = (((try - start) + 1) % count) + start;
        }
 
+       syslog(LOG_DEBUG, "can't assign any IP address -> address space is full");
        return false;
 }
 
@@ -611,10 +631,9 @@ static struct dhcpv4_assignment* dhcpv4_lease(struct interface *iface,
                if (assigned && a)
                        lease = a;
        } else if (msg == DHCPV4_MSG_RELEASE) {
-               if (a) {
+               if (a && a->valid_until != LONG_MAX)
                        a->valid_until = 0;
-               }
-       } else if (msg == DHCPV4_MSG_DECLINE) {
+       } else if (msg == DHCPV4_MSG_DECLINE && a && a->valid_until != LONG_MAX) {
                memset(a->hwaddr, 0, sizeof(a->hwaddr));
                a->valid_until = now + 3600; // Block address for 1h
        }