1 87a52e04 2005-12-26 devnull #include <u.h>
2 87a52e04 2005-12-26 devnull #include <libc.h>
3 87a52e04 2005-12-26 devnull #include <ip.h>
4 87a52e04 2005-12-26 devnull #include <bio.h>
5 87a52e04 2005-12-26 devnull #include <ndb.h>
6 87a52e04 2005-12-26 devnull #include <ctype.h>
7 87a52e04 2005-12-26 devnull #include "dat.h"
10 87a52e04 2005-12-26 devnull * format of a binding entry:
11 87a52e04 2005-12-26 devnull * char ipaddr[32];
12 87a52e04 2005-12-26 devnull * char id[32];
13 87a52e04 2005-12-26 devnull * char hwa[32];
14 87a52e04 2005-12-26 devnull * char otime[10];
16 87a52e04 2005-12-26 devnull Binding *bcache;
17 87a52e04 2005-12-26 devnull uchar bfirst[IPaddrlen];
18 87a52e04 2005-12-26 devnull char *binddir = nil;
19 87a52e04 2005-12-26 devnull char *xbinddir = "#9/ndb/dhcp";
22 87a52e04 2005-12-26 devnull * convert a byte array to hex
24 87a52e04 2005-12-26 devnull static char
25 87a52e04 2005-12-26 devnull hex(int x)
27 87a52e04 2005-12-26 devnull if(x < 10)
28 87a52e04 2005-12-26 devnull return x + '0';
29 87a52e04 2005-12-26 devnull return x - 10 + 'a';
31 87a52e04 2005-12-26 devnull extern char*
32 87a52e04 2005-12-26 devnull tohex(char *hdr, uchar *p, int len)
34 87a52e04 2005-12-26 devnull char *s, *sp;
35 87a52e04 2005-12-26 devnull int hlen;
37 87a52e04 2005-12-26 devnull hlen = strlen(hdr);
38 87a52e04 2005-12-26 devnull s = malloc(hlen + 2*len + 1);
40 87a52e04 2005-12-26 devnull strcpy(sp, hdr);
41 87a52e04 2005-12-26 devnull sp += hlen;
42 87a52e04 2005-12-26 devnull for(; len > 0; len--){
43 87a52e04 2005-12-26 devnull *sp++ = hex(*p>>4);
44 87a52e04 2005-12-26 devnull *sp++ = hex(*p & 0xf);
48 87a52e04 2005-12-26 devnull return s;
52 87a52e04 2005-12-26 devnull * convert a client id to a string. If it's already
53 87a52e04 2005-12-26 devnull * ascii, leave it be. Otherwise, convert it to hex.
55 87a52e04 2005-12-26 devnull extern char*
56 87a52e04 2005-12-26 devnull toid(uchar *p, int n)
61 87a52e04 2005-12-26 devnull for(i = 0; i < n; i++)
62 87a52e04 2005-12-26 devnull if(!isprint(p[i]))
63 87a52e04 2005-12-26 devnull return tohex("id", p, n);
64 87a52e04 2005-12-26 devnull s = malloc(n + 1);
65 87a52e04 2005-12-26 devnull memmove(s, p, n);
66 87a52e04 2005-12-26 devnull s[n] = 0;
67 87a52e04 2005-12-26 devnull return s;
71 87a52e04 2005-12-26 devnull * increment an ip address
73 87a52e04 2005-12-26 devnull static void
74 87a52e04 2005-12-26 devnull incip(uchar *ip)
76 87a52e04 2005-12-26 devnull int i, x;
78 87a52e04 2005-12-26 devnull for(i = IPaddrlen-1; i >= 0; i--){
79 87a52e04 2005-12-26 devnull x = ip[i];
81 87a52e04 2005-12-26 devnull ip[i] = x;
82 87a52e04 2005-12-26 devnull if((x & 0x100) == 0)
88 87a52e04 2005-12-26 devnull * find a binding for an id or hardware address
90 87a52e04 2005-12-26 devnull static int
91 87a52e04 2005-12-26 devnull lockopen(char *file)
93 87a52e04 2005-12-26 devnull char err[ERRMAX];
94 87a52e04 2005-12-26 devnull int fd, tries;
96 87a52e04 2005-12-26 devnull for(tries = 0; tries < 5; tries++){
97 87a52e04 2005-12-26 devnull fd = open(file, OLOCK|ORDWR);
98 87a52e04 2005-12-26 devnull if(fd >= 0)
99 87a52e04 2005-12-26 devnull return fd;
100 87a52e04 2005-12-26 devnull print("open %s: %r\n", file);
101 87a52e04 2005-12-26 devnull errstr(err, sizeof err);
102 87a52e04 2005-12-26 devnull if(strstr(err, "lock")){
103 87a52e04 2005-12-26 devnull /* wait for other process to let go of lock */
104 87a52e04 2005-12-26 devnull sleep(250);
106 87a52e04 2005-12-26 devnull /* try again */
107 87a52e04 2005-12-26 devnull continue;
109 87a52e04 2005-12-26 devnull if(strstr(err, "exist") || strstr(err, "No such")){
110 87a52e04 2005-12-26 devnull /* no file, create an exclusive access file */
111 87a52e04 2005-12-26 devnull fd = create(file, ORDWR, DMEXCL|0666);
112 87a52e04 2005-12-26 devnull chmod(file, 0666);
113 87a52e04 2005-12-26 devnull if(fd >= 0)
114 87a52e04 2005-12-26 devnull return fd;
117 87a52e04 2005-12-26 devnull return -1;
121 87a52e04 2005-12-26 devnull setbinding(Binding *b, char *id, long t)
123 87a52e04 2005-12-26 devnull if(b->boundto)
124 87a52e04 2005-12-26 devnull free(b->boundto);
126 87a52e04 2005-12-26 devnull b->boundto = strdup(id);
127 87a52e04 2005-12-26 devnull b->lease = t;
130 87a52e04 2005-12-26 devnull static void
131 87a52e04 2005-12-26 devnull parsebinding(Binding *b, char *buf)
134 87a52e04 2005-12-26 devnull char *id, *p;
136 87a52e04 2005-12-26 devnull /* parse */
137 87a52e04 2005-12-26 devnull t = atoi(buf);
138 87a52e04 2005-12-26 devnull id = strchr(buf, '\n');
140 87a52e04 2005-12-26 devnull *id++ = 0;
141 87a52e04 2005-12-26 devnull p = strchr(id, '\n');
145 87a52e04 2005-12-26 devnull id = "";
147 87a52e04 2005-12-26 devnull /* replace any past info */
148 87a52e04 2005-12-26 devnull setbinding(b, id, t);
151 87a52e04 2005-12-26 devnull static int
152 87a52e04 2005-12-26 devnull writebinding(int fd, Binding *b)
156 87a52e04 2005-12-26 devnull seek(fd, 0, 0);
157 87a52e04 2005-12-26 devnull if(fprint(fd, "%ld\n%s\n", b->lease, b->boundto) < 0)
158 87a52e04 2005-12-26 devnull return -1;
159 87a52e04 2005-12-26 devnull d = dirfstat(fd);
160 87a52e04 2005-12-26 devnull if(d == nil)
161 87a52e04 2005-12-26 devnull return -1;
162 87a52e04 2005-12-26 devnull b->q.type = d->qid.type;
163 87a52e04 2005-12-26 devnull b->q.path = d->qid.path;
164 87a52e04 2005-12-26 devnull b->q.vers = d->qid.vers;
165 87a52e04 2005-12-26 devnull free(d);
166 87a52e04 2005-12-26 devnull return 0;
170 87a52e04 2005-12-26 devnull * synchronize cached binding with file. the file always wins.
173 87a52e04 2005-12-26 devnull syncbinding(Binding *b, int returnfd)
175 87a52e04 2005-12-26 devnull char buf[512];
176 87a52e04 2005-12-26 devnull int i, fd;
179 87a52e04 2005-12-26 devnull if(binddir == nil)
180 87a52e04 2005-12-26 devnull binddir = unsharp(xbinddir);
182 87a52e04 2005-12-26 devnull snprint(buf, sizeof(buf), "%s/%I", binddir, b->ip);
183 87a52e04 2005-12-26 devnull fd = lockopen(buf);
184 87a52e04 2005-12-26 devnull if(fd < 0){
185 87a52e04 2005-12-26 devnull /* assume someone else is using it */
186 87a52e04 2005-12-26 devnull b->lease = time(0) + OfferTimeout;
187 87a52e04 2005-12-26 devnull return -1;
190 87a52e04 2005-12-26 devnull /* reread if changed */
191 87a52e04 2005-12-26 devnull d = dirfstat(fd);
192 87a52e04 2005-12-26 devnull if(d != nil) /* BUG? */
193 87a52e04 2005-12-26 devnull if(d->qid.type != b->q.type || d->qid.path != b->q.path || d->qid.vers != b->q.vers){
194 87a52e04 2005-12-26 devnull i = read(fd, buf, sizeof(buf)-1);
195 87a52e04 2005-12-26 devnull if(i < 0)
197 87a52e04 2005-12-26 devnull buf[i] = 0;
198 87a52e04 2005-12-26 devnull parsebinding(b, buf);
199 87a52e04 2005-12-26 devnull b->lasttouched = d->mtime;
200 87a52e04 2005-12-26 devnull b->q.path = d->qid.path;
201 87a52e04 2005-12-26 devnull b->q.vers = d->qid.vers;
204 87a52e04 2005-12-26 devnull free(d);
206 87a52e04 2005-12-26 devnull if(returnfd)
207 87a52e04 2005-12-26 devnull return fd;
209 87a52e04 2005-12-26 devnull close(fd);
210 87a52e04 2005-12-26 devnull return 0;
213 87a52e04 2005-12-26 devnull extern int
214 87a52e04 2005-12-26 devnull samenet(uchar *ip, Info *iip)
216 87a52e04 2005-12-26 devnull uchar x[IPaddrlen];
218 87a52e04 2005-12-26 devnull maskip(iip->ipmask, ip, x);
219 87a52e04 2005-12-26 devnull return ipcmp(x, iip->ipnet) == 0;
223 87a52e04 2005-12-26 devnull * create a record for each binding
225 87a52e04 2005-12-26 devnull extern void
226 87a52e04 2005-12-26 devnull initbinding(uchar *first, int n)
228 87a52e04 2005-12-26 devnull while(n-- > 0){
229 87a52e04 2005-12-26 devnull iptobinding(first, 1);
230 87a52e04 2005-12-26 devnull incip(first);
235 87a52e04 2005-12-26 devnull * find a binding for a specific ip address
237 87a52e04 2005-12-26 devnull extern Binding*
238 87a52e04 2005-12-26 devnull iptobinding(uchar *ip, int mk)
240 87a52e04 2005-12-26 devnull Binding *b;
242 87a52e04 2005-12-26 devnull for(b = bcache; b; b = b->next){
243 87a52e04 2005-12-26 devnull if(ipcmp(b->ip, ip) == 0){
244 87a52e04 2005-12-26 devnull syncbinding(b, 0);
245 87a52e04 2005-12-26 devnull return b;
249 87a52e04 2005-12-26 devnull if(mk == 0)
250 87a52e04 2005-12-26 devnull return 0;
251 87a52e04 2005-12-26 devnull b = malloc(sizeof(*b));
252 87a52e04 2005-12-26 devnull memset(b, 0, sizeof(*b));
253 87a52e04 2005-12-26 devnull ipmove(b->ip, ip);
254 87a52e04 2005-12-26 devnull b->next = bcache;
255 87a52e04 2005-12-26 devnull bcache = b;
256 87a52e04 2005-12-26 devnull syncbinding(b, 0);
257 87a52e04 2005-12-26 devnull return b;
260 87a52e04 2005-12-26 devnull static void
261 87a52e04 2005-12-26 devnull lognolease(Binding *b)
263 87a52e04 2005-12-26 devnull /* renew the old binding, and hope it eventually goes away */
264 87a52e04 2005-12-26 devnull b->offer = 5*60;
265 87a52e04 2005-12-26 devnull commitbinding(b);
267 87a52e04 2005-12-26 devnull /* complain if we haven't in the last 5 minutes */
268 87a52e04 2005-12-26 devnull if(now - b->lastcomplained < 5*60)
270 87a52e04 2005-12-26 devnull syslog(0, blog, "dhcp: lease for %I to %s ended at %ld but still in use\n",
271 87a52e04 2005-12-26 devnull b->ip, b->boundto != nil ? b->boundto : "?", b->lease);
272 87a52e04 2005-12-26 devnull b->lastcomplained = now;
276 87a52e04 2005-12-26 devnull * find a free binding for a hw addr or id on the same network as iip
278 87a52e04 2005-12-26 devnull extern Binding*
279 87a52e04 2005-12-26 devnull idtobinding(char *id, Info *iip, int ping)
281 87a52e04 2005-12-26 devnull Binding *b, *oldest;
282 87a52e04 2005-12-26 devnull int oldesttime;
285 87a52e04 2005-12-26 devnull * first look for an old binding that matches. that way
286 87a52e04 2005-12-26 devnull * clients will tend to keep the same ip addresses.
288 87a52e04 2005-12-26 devnull for(b = bcache; b; b = b->next){
289 87a52e04 2005-12-26 devnull if(b->boundto && strcmp(b->boundto, id) == 0){
290 87a52e04 2005-12-26 devnull if(!samenet(b->ip, iip))
291 87a52e04 2005-12-26 devnull continue;
293 87a52e04 2005-12-26 devnull /* check with the other servers */
294 87a52e04 2005-12-26 devnull syncbinding(b, 0);
295 87a52e04 2005-12-26 devnull if(strcmp(b->boundto, id) == 0)
296 87a52e04 2005-12-26 devnull return b;
300 87a52e04 2005-12-26 devnull print("looking for old for %I\n", iip->ipnet);
303 87a52e04 2005-12-26 devnull * look for oldest binding that we think is unused
305 87a52e04 2005-12-26 devnull for(;;){
306 87a52e04 2005-12-26 devnull oldest = nil;
307 87a52e04 2005-12-26 devnull oldesttime = 0;
308 87a52e04 2005-12-26 devnull for(b = bcache; b; b = b->next){
309 87a52e04 2005-12-26 devnull print("tried %d now %d lease %d exp %d %I\n", b->tried, now, b->lease, b->expoffer, b->ip);
310 87a52e04 2005-12-26 devnull if(b->tried != now)
311 87a52e04 2005-12-26 devnull if(b->lease < now && b->expoffer < now && samenet(b->ip, iip))
312 87a52e04 2005-12-26 devnull if(oldest == nil || b->lasttouched < oldesttime){
313 87a52e04 2005-12-26 devnull /* sync and check again */
314 87a52e04 2005-12-26 devnull syncbinding(b, 0);
315 87a52e04 2005-12-26 devnull if(b->lease < now && b->expoffer < now && samenet(b->ip, iip))
316 87a52e04 2005-12-26 devnull if(oldest == nil || b->lasttouched < oldesttime){
317 87a52e04 2005-12-26 devnull oldest = b;
318 87a52e04 2005-12-26 devnull print("have oldest\n");
319 87a52e04 2005-12-26 devnull oldesttime = b->lasttouched;
323 87a52e04 2005-12-26 devnull if(oldest == nil)
326 87a52e04 2005-12-26 devnull /* make sure noone is still using it */
327 87a52e04 2005-12-26 devnull oldest->tried = now;
328 87a52e04 2005-12-26 devnull print("return oldest\n");
329 87a52e04 2005-12-26 devnull if(ping == 0 || icmpecho(oldest->ip) == 0)
330 87a52e04 2005-12-26 devnull return oldest;
332 87a52e04 2005-12-26 devnull lognolease(oldest); /* sets lastcomplained */
335 87a52e04 2005-12-26 devnull /* try all bindings */
336 87a52e04 2005-12-26 devnull for(b = bcache; b; b = b->next){
337 87a52e04 2005-12-26 devnull syncbinding(b, 0);
338 87a52e04 2005-12-26 devnull if(b->tried != now)
339 87a52e04 2005-12-26 devnull if(b->lease < now && b->expoffer < now && samenet(b->ip, iip)){
340 87a52e04 2005-12-26 devnull b->tried = now;
341 87a52e04 2005-12-26 devnull if(ping == 0 || icmpecho(b->ip) == 0)
342 87a52e04 2005-12-26 devnull return b;
344 87a52e04 2005-12-26 devnull lognolease(b);
348 87a52e04 2005-12-26 devnull /* nothing worked, give up */
349 87a52e04 2005-12-26 devnull return 0;
353 87a52e04 2005-12-26 devnull * create an offer
355 87a52e04 2005-12-26 devnull extern void
356 87a52e04 2005-12-26 devnull mkoffer(Binding *b, char *id, long leasetime)
358 87a52e04 2005-12-26 devnull if(leasetime <= 0){
359 87a52e04 2005-12-26 devnull if(b->lease > now + minlease)
360 87a52e04 2005-12-26 devnull leasetime = b->lease - now;
362 87a52e04 2005-12-26 devnull leasetime = minlease;
364 87a52e04 2005-12-26 devnull if(b->offeredto)
365 87a52e04 2005-12-26 devnull free(b->offeredto);
366 87a52e04 2005-12-26 devnull b->offeredto = strdup(id);
367 87a52e04 2005-12-26 devnull b->offer = leasetime;
368 87a52e04 2005-12-26 devnull b->expoffer = now + OfferTimeout;
372 87a52e04 2005-12-26 devnull * find an offer for this id
374 87a52e04 2005-12-26 devnull extern Binding*
375 87a52e04 2005-12-26 devnull idtooffer(char *id, Info *iip)
377 87a52e04 2005-12-26 devnull Binding *b;
379 87a52e04 2005-12-26 devnull /* look for an offer to this id */
380 87a52e04 2005-12-26 devnull for(b = bcache; b; b = b->next){
381 87a52e04 2005-12-26 devnull print("%I %I ? offeredto %s id %s\n", b->ip, iip->ipnet, b->offeredto, id);
382 87a52e04 2005-12-26 devnull if(b->offeredto && strcmp(b->offeredto, id) == 0 && samenet(b->ip, iip)){
383 87a52e04 2005-12-26 devnull /* make sure some other system hasn't stolen it */
384 87a52e04 2005-12-26 devnull syncbinding(b, 0);
385 87a52e04 2005-12-26 devnull print("b->lease %d now %d boundto %s offered %s\n", b->lease, now, b->boundto, b->offeredto);
386 87a52e04 2005-12-26 devnull if(b->lease < now
387 87a52e04 2005-12-26 devnull || (b->boundto && strcmp(b->boundto, b->offeredto) == 0))
388 87a52e04 2005-12-26 devnull return b;
391 87a52e04 2005-12-26 devnull return 0;
395 87a52e04 2005-12-26 devnull * commit a lease, this could fail
397 87a52e04 2005-12-26 devnull extern int
398 87a52e04 2005-12-26 devnull commitbinding(Binding *b)
401 87a52e04 2005-12-26 devnull long now;
403 87a52e04 2005-12-26 devnull now = time(0);
405 87a52e04 2005-12-26 devnull if(b->offeredto == 0)
406 87a52e04 2005-12-26 devnull return -1;
407 87a52e04 2005-12-26 devnull fd = syncbinding(b, 1);
408 87a52e04 2005-12-26 devnull if(fd < 0)
409 87a52e04 2005-12-26 devnull return -1;
410 87a52e04 2005-12-26 devnull if(b->lease > now && b->boundto && strcmp(b->boundto, b->offeredto) != 0){
411 87a52e04 2005-12-26 devnull close(fd);
412 87a52e04 2005-12-26 devnull return -1;
414 87a52e04 2005-12-26 devnull setbinding(b, b->offeredto, now + b->offer);
415 87a52e04 2005-12-26 devnull b->lasttouched = now;
417 87a52e04 2005-12-26 devnull if(writebinding(fd, b) < 0){
418 87a52e04 2005-12-26 devnull close(fd);
419 87a52e04 2005-12-26 devnull return -1;
421 87a52e04 2005-12-26 devnull close(fd);
422 87a52e04 2005-12-26 devnull return 0;
426 87a52e04 2005-12-26 devnull * commit a lease, this could fail
428 87a52e04 2005-12-26 devnull extern int
429 87a52e04 2005-12-26 devnull releasebinding(Binding *b, char *id)
432 87a52e04 2005-12-26 devnull long now;
434 87a52e04 2005-12-26 devnull now = time(0);
436 87a52e04 2005-12-26 devnull fd = syncbinding(b, 1);
437 87a52e04 2005-12-26 devnull if(fd < 0)
438 87a52e04 2005-12-26 devnull return -1;
439 87a52e04 2005-12-26 devnull if(b->lease > now && b->boundto && strcmp(b->boundto, id) != 0){
440 87a52e04 2005-12-26 devnull close(fd);
441 87a52e04 2005-12-26 devnull return -1;
443 87a52e04 2005-12-26 devnull b->lease = 0;
444 87a52e04 2005-12-26 devnull b->expoffer = 0;
446 87a52e04 2005-12-26 devnull if(writebinding(fd, b) < 0){
447 87a52e04 2005-12-26 devnull close(fd);
448 87a52e04 2005-12-26 devnull return -1;
450 87a52e04 2005-12-26 devnull close(fd);
451 87a52e04 2005-12-26 devnull return 0;