5d8775f63ea01f86541a6117c26b07225c9f6855
[oweals/busybox.git] / networking / udhcp / leases.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * leases.c -- tools to manage DHCP leases
4  * Russ Dill <Russ.Dill@asu.edu> July 2001
5  */
6
7 #include "common.h"
8 #include "dhcpd.h"
9
10
11 /* Find the oldest expired lease, NULL if there are no expired leases */
12 static struct dhcpOfferedAddr *oldest_expired_lease(void)
13 {
14         struct dhcpOfferedAddr *oldest = NULL;
15         unsigned long oldest_lease = time(0);
16         unsigned i;
17
18
19         for (i = 0; i < server_config.max_leases; i++)
20                 if (oldest_lease > leases[i].expires) {
21                         oldest_lease = leases[i].expires;
22                         oldest = &(leases[i]);
23                 }
24         return oldest;
25 }
26
27
28 /* clear every lease out that chaddr OR yiaddr matches and is nonzero */
29 static void clear_lease(const uint8_t *chaddr, uint32_t yiaddr)
30 {
31         unsigned i, j;
32
33         for (j = 0; j < 16 && !chaddr[j]; j++);
34
35         for (i = 0; i < server_config.max_leases; i++)
36                 if ((j != 16 && memcmp(leases[i].chaddr, chaddr, 16) != 0)
37                  || (yiaddr && leases[i].yiaddr == yiaddr)
38                 ) {
39                         memset(&(leases[i]), 0, sizeof(struct dhcpOfferedAddr));
40                 }
41 }
42
43
44 /* add a lease into the table, clearing out any old ones */
45 struct dhcpOfferedAddr *add_lease(const uint8_t *chaddr, uint32_t yiaddr, unsigned long lease)
46 {
47         struct dhcpOfferedAddr *oldest;
48
49         /* clean out any old ones */
50         clear_lease(chaddr, yiaddr);
51
52         oldest = oldest_expired_lease();
53
54         if (oldest) {
55                 memcpy(oldest->chaddr, chaddr, 16);
56                 oldest->yiaddr = yiaddr;
57                 oldest->expires = time(0) + lease;
58         }
59
60         return oldest;
61 }
62
63
64 /* true if a lease has expired */
65 int lease_expired(struct dhcpOfferedAddr *lease)
66 {
67         return (lease->expires < (unsigned long) time(0));
68 }
69
70
71 /* Find the first lease that matches chaddr, NULL if no match */
72 struct dhcpOfferedAddr *find_lease_by_chaddr(const uint8_t *chaddr)
73 {
74         unsigned i;
75
76         for (i = 0; i < server_config.max_leases; i++)
77                 if (!memcmp(leases[i].chaddr, chaddr, 16))
78                         return &(leases[i]);
79
80         return NULL;
81 }
82
83
84 /* Find the first lease that matches yiaddr, NULL is no match */
85 struct dhcpOfferedAddr *find_lease_by_yiaddr(uint32_t yiaddr)
86 {
87         unsigned i;
88
89         for (i = 0; i < server_config.max_leases; i++)
90                 if (leases[i].yiaddr == yiaddr)
91                         return &(leases[i]);
92
93         return NULL;
94 }
95
96
97 /* check is an IP is taken, if it is, add it to the lease table */
98 static int check_ip(uint32_t addr)
99 {
100         static const uint8_t blank_chaddr[16]; /* 16 zero bytes */
101
102         struct in_addr temp;
103
104         if (arpping(addr, server_config.server, server_config.arp, server_config.interface) == 0) {
105                 temp.s_addr = addr;
106                 bb_info_msg("%s belongs to someone, reserving it for %ld seconds",
107                         inet_ntoa(temp), server_config.conflict_time);
108                 add_lease(blank_chaddr, addr, server_config.conflict_time);
109                 return 1;
110         }
111         return 0;
112 }
113
114
115 /* find an assignable address, it check_expired is true, we check all the expired leases as well.
116  * Maybe this should try expired leases by age... */
117 uint32_t find_address(int check_expired)
118 {
119         uint32_t addr, ret;
120         struct dhcpOfferedAddr *lease = NULL;
121
122         addr = ntohl(server_config.start); /* addr is in host order here */
123         for (;addr <= ntohl(server_config.end); addr++) {
124
125                 /* ie, 192.168.55.0 */
126                 if (!(addr & 0xFF)) continue;
127
128                 /* ie, 192.168.55.255 */
129                 if ((addr & 0xFF) == 0xFF) continue;
130
131                 /* Only do if it isn't assigned as a static lease */
132                 if (!reservedIp(server_config.static_leases, htonl(addr))) {
133
134                         /* lease is not taken */
135                         ret = htonl(addr);
136                         lease = find_lease_by_yiaddr(ret);
137
138                         /* no lease or it expired and we are checking for expired leases */
139                         if ((!lease || (check_expired && lease_expired(lease)))
140                          && /* and it isn't on the network */ !check_ip(ret)
141                         ) {
142                                 return ret;
143                         }
144                 }
145         }
146         return 0;
147 }