router: fix invalid router lifetime
[project/odhcpd.git] / src / dhcpv4.c
index 92a1397..a657e13 100644 (file)
@@ -137,8 +137,8 @@ int setup_dhcpv4_interface(struct interface *iface, bool enable)
 
                        if (start && end && start < end &&
                                        start > ntohl(addr.s_addr & ~mask.s_addr) &&
-                                       (start & ntohl(mask.s_addr)) == start &&
-                                       (end & ntohl(mask.s_addr)) == end) {
+                                       (start & ntohl(~mask.s_addr)) == start &&
+                                       (end & ntohl(~mask.s_addr)) == end) {
                                iface->dhcpv4_start.s_addr = htonl(start) |
                                                (addr.s_addr & mask.s_addr);
                                iface->dhcpv4_end.s_addr = htonl(end) |
@@ -222,7 +222,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 +501,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 +634,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
        }