udhcpc: reduce stack usage by de-inlining routines with on-stack pkt buf
[oweals/busybox.git] / networking / udhcp / dhcpc.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * udhcp client
4  *
5  * Russ Dill <Russ.Dill@asu.edu> July 2001
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20  */
21 #include <syslog.h>
22 /* Override ENABLE_FEATURE_PIDFILE - ifupdown needs our pidfile to always exist */
23 #define WANT_PIDFILE 1
24 #include "common.h"
25 #include "dhcpd.h"
26 #include "dhcpc.h"
27
28 #include <asm/types.h>
29 #if (defined(__GLIBC__) && __GLIBC__ >= 2 && __GLIBC_MINOR__ >= 1) || defined(_NEWLIB_VERSION)
30 # include <netpacket/packet.h>
31 # include <net/ethernet.h>
32 #else
33 # include <linux/if_packet.h>
34 # include <linux/if_ether.h>
35 #endif
36 #include <linux/filter.h>
37
38 /* struct client_config_t client_config is in bb_common_bufsiz1 */
39
40
41 /*** Script execution code ***/
42
43 /* get a rough idea of how long an option will be (rounding up...) */
44 static const uint8_t len_of_option_as_string[] = {
45         [OPTION_IP              ] = sizeof("255.255.255.255 "),
46         [OPTION_IP_PAIR         ] = sizeof("255.255.255.255 ") * 2,
47         [OPTION_STATIC_ROUTES   ] = sizeof("255.255.255.255/32 255.255.255.255 "),
48         [OPTION_STRING          ] = 1,
49 #if ENABLE_FEATURE_UDHCP_RFC3397
50         [OPTION_DNS_STRING      ] = 1, /* unused */
51         /* Hmmm, this severely overestimates size if SIP_SERVERS option
52          * is in domain name form: N-byte option in binary form
53          * mallocs ~16*N bytes. But it is freed almost at once.
54          */
55         [OPTION_SIP_SERVERS     ] = sizeof("255.255.255.255 "),
56 #endif
57 //      [OPTION_BOOLEAN         ] = sizeof("yes "),
58         [OPTION_U8              ] = sizeof("255 "),
59         [OPTION_U16             ] = sizeof("65535 "),
60 //      [OPTION_S16             ] = sizeof("-32768 "),
61         [OPTION_U32             ] = sizeof("4294967295 "),
62         [OPTION_S32             ] = sizeof("-2147483684 "),
63 };
64
65 /* note: ip is a pointer to an IP in network order, possibly misaliged */
66 static int sprint_nip(char *dest, const char *pre, const uint8_t *ip)
67 {
68         return sprintf(dest, "%s%u.%u.%u.%u", pre, ip[0], ip[1], ip[2], ip[3]);
69 }
70
71 /* really simple implementation, just count the bits */
72 static int mton(uint32_t mask)
73 {
74         int i = 0;
75         mask = ntohl(mask); /* 111110000-like bit pattern */
76         while (mask) {
77                 i++;
78                 mask <<= 1;
79         }
80         return i;
81 }
82
83 /* Create "opt_name=opt_value" string */
84 static NOINLINE char *xmalloc_optname_optval(uint8_t *option, const struct dhcp_optflag *optflag, const char *opt_name)
85 {
86         unsigned upper_length;
87         int len, type, optlen;
88         char *dest, *ret;
89
90         /* option points to OPT_DATA, need to go back and get OPT_LEN */
91         len = option[OPT_LEN - OPT_DATA];
92
93         type = optflag->flags & OPTION_TYPE_MASK;
94         optlen = dhcp_option_lengths[type];
95         upper_length = len_of_option_as_string[type] * ((unsigned)len / (unsigned)optlen);
96
97         dest = ret = xmalloc(upper_length + strlen(opt_name) + 2);
98         dest += sprintf(ret, "%s=", opt_name);
99
100         while (len >= optlen) {
101                 unsigned ip_ofs = 0;
102
103                 switch (type) {
104                 case OPTION_IP_PAIR:
105                         dest += sprint_nip(dest, "", option);
106                         *dest++ = '/';
107                         ip_ofs = 4;
108                         /* fall through */
109                 case OPTION_IP:
110                         dest += sprint_nip(dest, "", option + ip_ofs);
111                         break;
112 //              case OPTION_BOOLEAN:
113 //                      dest += sprintf(dest, *option ? "yes" : "no");
114 //                      break;
115                 case OPTION_U8:
116                         dest += sprintf(dest, "%u", *option);
117                         break;
118 //              case OPTION_S16:
119                 case OPTION_U16: {
120                         uint16_t val_u16;
121                         move_from_unaligned16(val_u16, option);
122                         dest += sprintf(dest, "%u", ntohs(val_u16));
123                         break;
124                 }
125                 case OPTION_S32:
126                 case OPTION_U32: {
127                         uint32_t val_u32;
128                         move_from_unaligned32(val_u32, option);
129                         dest += sprintf(dest, type == OPTION_U32 ? "%lu" : "%ld", (unsigned long) ntohl(val_u32));
130                         break;
131                 }
132                 case OPTION_STRING:
133                         memcpy(dest, option, len);
134                         dest[len] = '\0';
135                         return ret;      /* Short circuit this case */
136                 case OPTION_STATIC_ROUTES: {
137                         /* Option binary format:
138                          * mask [one byte, 0..32]
139                          * ip [big endian, 0..4 bytes depending on mask]
140                          * router [big endian, 4 bytes]
141                          * may be repeated
142                          *
143                          * We convert it to a string "IP/MASK ROUTER IP2/MASK2 ROUTER2"
144                          */
145                         const char *pfx = "";
146
147                         while (len >= 1 + 4) { /* mask + 0-byte ip + router */
148                                 uint32_t nip;
149                                 uint8_t *p;
150                                 unsigned mask;
151                                 int bytes;
152
153                                 mask = *option++;
154                                 if (mask > 32)
155                                         break;
156                                 len--;
157
158                                 nip = 0;
159                                 p = (void*) &nip;
160                                 bytes = (mask + 7) / 8; /* 0 -> 0, 1..8 -> 1, 9..16 -> 2 etc */
161                                 while (--bytes >= 0) {
162                                         *p++ = *option++;
163                                         len--;
164                                 }
165                                 if (len < 4)
166                                         break;
167
168                                 /* print ip/mask */
169                                 dest += sprint_nip(dest, pfx, (void*) &nip);
170                                 pfx = " ";
171                                 dest += sprintf(dest, "/%u ", mask);
172                                 /* print router */
173                                 dest += sprint_nip(dest, "", option);
174                                 option += 4;
175                                 len -= 4;
176                         }
177
178                         return ret;
179                 }
180 #if ENABLE_FEATURE_UDHCP_RFC3397
181                 case OPTION_DNS_STRING:
182                         /* unpack option into dest; use ret for prefix (i.e., "optname=") */
183                         dest = dname_dec(option, len, ret);
184                         if (dest) {
185                                 free(ret);
186                                 return dest;
187                         }
188                         /* error. return "optname=" string */
189                         return ret;
190                 case OPTION_SIP_SERVERS:
191                         /* Option binary format:
192                          * type: byte
193                          * type=0: domain names, dns-compressed
194                          * type=1: IP addrs
195                          */
196                         option++;
197                         len--;
198                         if (option[-1] == 0) {
199                                 dest = dname_dec(option, len, ret);
200                                 if (dest) {
201                                         free(ret);
202                                         return dest;
203                                 }
204                         } else
205                         if (option[-1] == 1) {
206                                 const char *pfx = "";
207                                 while (1) {
208                                         len -= 4;
209                                         if (len < 0)
210                                                 break;
211                                         dest += sprint_nip(dest, pfx, option);
212                                         pfx = " ";
213                                         option += 4;
214                                 }
215                         }
216                         return ret;
217 #endif
218                 } /* switch */
219                 option += optlen;
220                 len -= optlen;
221 // TODO: it can be a list only if (optflag->flags & OPTION_LIST).
222 // Should we bail out/warn if we see multi-ip option which is
223 // not allowed to be such (for example, DHCP_BROADCAST)? -
224                 if (len <= 0 /* || !(optflag->flags & OPTION_LIST) */)
225                         break;
226                 *dest++ = ' ';
227                 *dest = '\0';
228         }
229         return ret;
230 }
231
232 /* put all the parameters into the environment */
233 static char **fill_envp(struct dhcp_packet *packet)
234 {
235         int envc;
236         int i;
237         char **envp, **curr;
238         const char *opt_name;
239         uint8_t *temp;
240         uint8_t overload = 0;
241
242         /* We need 6 elements for:
243          * "interface=IFACE"
244          * "ip=N.N.N.N" from packet->yiaddr
245          * "siaddr=IP" from packet->siaddr_nip (unless 0)
246          * "boot_file=FILE" from packet->file (unless overloaded)
247          * "sname=SERVER_HOSTNAME" from packet->sname (unless overloaded)
248          * terminating NULL
249          */
250         envc = 6;
251         /* +1 element for each option, +2 for subnet option: */
252         if (packet) {
253                 for (i = 0; dhcp_optflags[i].code; i++) {
254                         if (udhcp_get_option(packet, dhcp_optflags[i].code)) {
255                                 if (dhcp_optflags[i].code == DHCP_SUBNET)
256                                         envc++; /* for mton */
257                                 envc++;
258                         }
259                 }
260                 temp = udhcp_get_option(packet, DHCP_OPTION_OVERLOAD);
261                 if (temp)
262                         overload = *temp;
263         }
264         curr = envp = xzalloc(sizeof(char *) * envc);
265
266         *curr = xasprintf("interface=%s", client_config.interface);
267         putenv(*curr++);
268
269         if (!packet)
270                 return envp;
271
272         *curr = xmalloc(sizeof("ip=255.255.255.255"));
273         sprint_nip(*curr, "ip=", (uint8_t *) &packet->yiaddr);
274         putenv(*curr++);
275
276         opt_name = dhcp_option_strings;
277         i = 0;
278         while (*opt_name) {
279                 temp = udhcp_get_option(packet, dhcp_optflags[i].code);
280                 if (!temp)
281                         goto next;
282                 *curr = xmalloc_optname_optval(temp, &dhcp_optflags[i], opt_name);
283                 putenv(*curr++);
284                 if (dhcp_optflags[i].code == DHCP_SUBNET) {
285                         /* Subnet option: make things like "$ip/$mask" possible */
286                         uint32_t subnet;
287                         move_from_unaligned32(subnet, temp);
288                         *curr = xasprintf("mask=%d", mton(subnet));
289                         putenv(*curr++);
290                 }
291  next:
292                 opt_name += strlen(opt_name) + 1;
293                 i++;
294         }
295         if (packet->siaddr_nip) {
296                 *curr = xmalloc(sizeof("siaddr=255.255.255.255"));
297                 sprint_nip(*curr, "siaddr=", (uint8_t *) &packet->siaddr_nip);
298                 putenv(*curr++);
299         }
300         if (!(overload & FILE_FIELD) && packet->file[0]) {
301                 /* watch out for invalid packets */
302                 *curr = xasprintf("boot_file=%."DHCP_PKT_FILE_LEN_STR"s", packet->file);
303                 putenv(*curr++);
304         }
305         if (!(overload & SNAME_FIELD) && packet->sname[0]) {
306                 /* watch out for invalid packets */
307                 *curr = xasprintf("sname=%."DHCP_PKT_SNAME_LEN_STR"s", packet->sname);
308                 putenv(*curr++);
309         }
310         return envp;
311 }
312
313 /* Call a script with a par file and env vars */
314 static void udhcp_run_script(struct dhcp_packet *packet, const char *name)
315 {
316         char **envp, **curr;
317         char *argv[3];
318
319         if (client_config.script == NULL)
320                 return;
321
322         envp = fill_envp(packet);
323
324         /* call script */
325         log1("Executing %s %s", client_config.script, name);
326         argv[0] = (char*) client_config.script;
327         argv[1] = (char*) name;
328         argv[2] = NULL;
329         spawn_and_wait(argv);
330
331         for (curr = envp; *curr; curr++) {
332                 log2(" %s", *curr);
333                 bb_unsetenv_and_free(*curr);
334         }
335         free(envp);
336 }
337
338
339 /*** Sending/receiving packets ***/
340
341 static ALWAYS_INLINE uint32_t random_xid(void)
342 {
343         return rand();
344 }
345
346 /* Initialize the packet with the proper defaults */
347 static void init_packet(struct dhcp_packet *packet, char type)
348 {
349         /* Fill in: op, htype, hlen, cookie fields; message type option: */
350         udhcp_init_header(packet, type);
351
352         packet->xid = random_xid();
353
354         memcpy(packet->chaddr, client_config.client_mac, 6);
355         if (client_config.clientid)
356                 udhcp_add_binary_option(packet, client_config.clientid);
357 }
358
359 static void add_client_options(struct dhcp_packet *packet)
360 {
361         uint8_t c;
362         int i, end, len;
363
364         udhcp_add_simple_option(packet, DHCP_MAX_SIZE, htons(IP_UDP_DHCP_SIZE));
365
366         /* Add a "param req" option with the list of options we'd like to have
367          * from stubborn DHCP servers. Pull the data from the struct in common.c.
368          * No bounds checking because it goes towards the head of the packet. */
369         end = udhcp_end_option(packet->options);
370         len = 0;
371         for (i = 0; (c = dhcp_optflags[i].code) != 0; i++) {
372                 if ((   (dhcp_optflags[i].flags & OPTION_REQ)
373                      && !client_config.no_default_options
374                     )
375                  || (client_config.opt_mask[c >> 3] & (1 << (c & 7)))
376                 ) {
377                         packet->options[end + OPT_DATA + len] = c;
378                         len++;
379                 }
380         }
381         if (len) {
382                 packet->options[end + OPT_CODE] = DHCP_PARAM_REQ;
383                 packet->options[end + OPT_LEN] = len;
384                 packet->options[end + OPT_DATA + len] = DHCP_END;
385         }
386
387         if (client_config.vendorclass)
388                 udhcp_add_binary_option(packet, client_config.vendorclass);
389         if (client_config.hostname)
390                 udhcp_add_binary_option(packet, client_config.hostname);
391         if (client_config.fqdn)
392                 udhcp_add_binary_option(packet, client_config.fqdn);
393
394         /* Add -x options if any */
395         {
396                 struct option_set *curr = client_config.options;
397                 while (curr) {
398                         udhcp_add_binary_option(packet, curr->data);
399                         curr = curr->next;
400                 }
401 //              if (client_config.sname)
402 //                      strncpy((char*)packet->sname, client_config.sname, sizeof(packet->sname) - 1);
403 //              if (client_config.boot_file)
404 //                      strncpy((char*)packet->file, client_config.boot_file, sizeof(packet->file) - 1);
405         }
406 }
407
408 /* RFC 2131
409  * 4.4.4 Use of broadcast and unicast
410  *
411  * The DHCP client broadcasts DHCPDISCOVER, DHCPREQUEST and DHCPINFORM
412  * messages, unless the client knows the address of a DHCP server.
413  * The client unicasts DHCPRELEASE messages to the server. Because
414  * the client is declining the use of the IP address supplied by the server,
415  * the client broadcasts DHCPDECLINE messages.
416  *
417  * When the DHCP client knows the address of a DHCP server, in either
418  * INIT or REBOOTING state, the client may use that address
419  * in the DHCPDISCOVER or DHCPREQUEST rather than the IP broadcast address.
420  * The client may also use unicast to send DHCPINFORM messages
421  * to a known DHCP server. If the client receives no response to DHCP
422  * messages sent to the IP address of a known DHCP server, the DHCP
423  * client reverts to using the IP broadcast address.
424  */
425
426 static int raw_bcast_from_client_config_ifindex(struct dhcp_packet *packet)
427 {
428         return udhcp_send_raw_packet(packet,
429                 /*src*/ INADDR_ANY, CLIENT_PORT,
430                 /*dst*/ INADDR_BROADCAST, SERVER_PORT, MAC_BCAST_ADDR,
431                 client_config.ifindex);
432 }
433
434 /* Broadcast a DHCP discover packet to the network, with an optionally requested IP */
435 static NOINLINE int send_discover(uint32_t xid, uint32_t requested)
436 {
437         struct dhcp_packet packet;
438
439         /* Fill in: op, htype, hlen, cookie, chaddr fields,
440          * random xid field (we override it below),
441          * client-id option (unless -C), message type option:
442          */
443         init_packet(&packet, DHCPDISCOVER);
444
445         packet.xid = xid;
446         if (requested)
447                 udhcp_add_simple_option(&packet, DHCP_REQUESTED_IP, requested);
448
449         /* Add options: maxsize,
450          * optionally: hostname, fqdn, vendorclass,
451          * "param req" option according to -O, options specified with -x
452          */
453         add_client_options(&packet);
454
455         bb_info_msg("Sending discover...");
456         return raw_bcast_from_client_config_ifindex(&packet);
457 }
458
459 /* Broadcast a DHCP request message */
460 /* RFC 2131 3.1 paragraph 3:
461  * "The client _broadcasts_ a DHCPREQUEST message..."
462  */
463 static NOINLINE int send_select(uint32_t xid, uint32_t server, uint32_t requested)
464 {
465         struct dhcp_packet packet;
466         struct in_addr addr;
467
468 /*
469  * RFC 2131 4.3.2 DHCPREQUEST message
470  * ...
471  * If the DHCPREQUEST message contains a 'server identifier'
472  * option, the message is in response to a DHCPOFFER message.
473  * Otherwise, the message is a request to verify or extend an
474  * existing lease. If the client uses a 'client identifier'
475  * in a DHCPREQUEST message, it MUST use that same 'client identifier'
476  * in all subsequent messages. If the client included a list
477  * of requested parameters in a DHCPDISCOVER message, it MUST
478  * include that list in all subsequent messages.
479  */
480         /* Fill in: op, htype, hlen, cookie, chaddr fields,
481          * random xid field (we override it below),
482          * client-id option (unless -C), message type option:
483          */
484         init_packet(&packet, DHCPREQUEST);
485
486         packet.xid = xid;
487         udhcp_add_simple_option(&packet, DHCP_REQUESTED_IP, requested);
488
489         udhcp_add_simple_option(&packet, DHCP_SERVER_ID, server);
490
491         /* Add options: maxsize,
492          * optionally: hostname, fqdn, vendorclass,
493          * "param req" option according to -O, and options specified with -x
494          */
495         add_client_options(&packet);
496
497         addr.s_addr = requested;
498         bb_info_msg("Sending select for %s...", inet_ntoa(addr));
499         return raw_bcast_from_client_config_ifindex(&packet);
500 }
501
502 /* Unicast or broadcast a DHCP renew message */
503 static int send_renew(uint32_t xid, uint32_t server, uint32_t ciaddr)
504 {
505         struct dhcp_packet packet;
506
507 /*
508  * RFC 2131 4.3.2 DHCPREQUEST message
509  * ...
510  * DHCPREQUEST generated during RENEWING state:
511  *
512  * 'server identifier' MUST NOT be filled in, 'requested IP address'
513  * option MUST NOT be filled in, 'ciaddr' MUST be filled in with
514  * client's IP address. In this situation, the client is completely
515  * configured, and is trying to extend its lease. This message will
516  * be unicast, so no relay agents will be involved in its
517  * transmission.  Because 'giaddr' is therefore not filled in, the
518  * DHCP server will trust the value in 'ciaddr', and use it when
519  * replying to the client.
520  */
521         /* Fill in: op, htype, hlen, cookie, chaddr fields,
522          * random xid field (we override it below),
523          * client-id option (unless -C), message type option:
524          */
525         init_packet(&packet, DHCPREQUEST);
526
527         packet.xid = xid;
528         packet.ciaddr = ciaddr;
529
530         /* Add options: maxsize,
531          * optionally: hostname, fqdn, vendorclass,
532          * "param req" option according to -O, and options specified with -x
533          */
534         add_client_options(&packet);
535
536         bb_info_msg("Sending renew...");
537         if (server)
538                 return udhcp_send_kernel_packet(&packet,
539                         ciaddr, CLIENT_PORT,
540                         server, SERVER_PORT);
541         return raw_bcast_from_client_config_ifindex(&packet);
542 }
543
544 #if ENABLE_FEATURE_UDHCPC_ARPING
545 /* Broadcast a DHCP decline message */
546 static NOINLINE int send_decline(uint32_t xid, uint32_t server, uint32_t requested)
547 {
548         struct dhcp_packet packet;
549
550         /* Fill in: op, htype, hlen, cookie, chaddr, random xid fields,
551          * client-id option (unless -C), message type option:
552          */
553         init_packet(&packet, DHCPDECLINE);
554
555         /* RFC 2131 says DHCPDECLINE's xid is randomly selected by client,
556          * but in case the server is buggy and wants DHCPDECLINE's xid
557          * to match the xid which started entire handshake,
558          * we use the same xid we used in initial DHCPDISCOVER:
559          */
560         packet.xid = xid;
561         /* DHCPDECLINE uses "requested ip", not ciaddr, to store offered IP */
562         udhcp_add_simple_option(&packet, DHCP_REQUESTED_IP, requested);
563
564         udhcp_add_simple_option(&packet, DHCP_SERVER_ID, server);
565
566         bb_info_msg("Sending decline...");
567         return raw_bcast_from_client_config_ifindex(&packet);
568 }
569 #endif
570
571 /* Unicast a DHCP release message */
572 static int send_release(uint32_t server, uint32_t ciaddr)
573 {
574         struct dhcp_packet packet;
575
576         /* Fill in: op, htype, hlen, cookie, chaddr, random xid fields,
577          * client-id option (unless -C), message type option:
578          */
579         init_packet(&packet, DHCPRELEASE);
580
581         /* DHCPRELEASE uses ciaddr, not "requested ip", to store IP being released */
582         packet.ciaddr = ciaddr;
583
584         udhcp_add_simple_option(&packet, DHCP_SERVER_ID, server);
585
586         bb_info_msg("Sending release...");
587         return udhcp_send_kernel_packet(&packet, ciaddr, CLIENT_PORT, server, SERVER_PORT);
588 }
589
590 /* Returns -1 on errors that are fatal for the socket, -2 for those that aren't */
591 static NOINLINE int udhcp_recv_raw_packet(struct dhcp_packet *dhcp_pkt, int fd)
592 {
593         int bytes;
594         struct ip_udp_dhcp_packet packet;
595         uint16_t check;
596
597         memset(&packet, 0, sizeof(packet));
598         bytes = safe_read(fd, &packet, sizeof(packet));
599         if (bytes < 0) {
600                 log1("Packet read error, ignoring");
601                 /* NB: possible down interface, etc. Caller should pause. */
602                 return bytes; /* returns -1 */
603         }
604
605         if (bytes < (int) (sizeof(packet.ip) + sizeof(packet.udp))) {
606                 log1("Packet is too short, ignoring");
607                 return -2;
608         }
609
610         if (bytes < ntohs(packet.ip.tot_len)) {
611                 /* packet is bigger than sizeof(packet), we did partial read */
612                 log1("Oversized packet, ignoring");
613                 return -2;
614         }
615
616         /* ignore any extra garbage bytes */
617         bytes = ntohs(packet.ip.tot_len);
618
619         /* make sure its the right packet for us, and that it passes sanity checks */
620         if (packet.ip.protocol != IPPROTO_UDP || packet.ip.version != IPVERSION
621          || packet.ip.ihl != (sizeof(packet.ip) >> 2)
622          || packet.udp.dest != htons(CLIENT_PORT)
623         /* || bytes > (int) sizeof(packet) - can't happen */
624          || ntohs(packet.udp.len) != (uint16_t)(bytes - sizeof(packet.ip))
625         ) {
626                 log1("Unrelated/bogus packet, ignoring");
627                 return -2;
628         }
629
630         /* verify IP checksum */
631         check = packet.ip.check;
632         packet.ip.check = 0;
633         if (check != udhcp_checksum(&packet.ip, sizeof(packet.ip))) {
634                 log1("Bad IP header checksum, ignoring");
635                 return -2;
636         }
637
638         /* verify UDP checksum. IP header has to be modified for this */
639         memset(&packet.ip, 0, offsetof(struct iphdr, protocol));
640         /* ip.xx fields which are not memset: protocol, check, saddr, daddr */
641         packet.ip.tot_len = packet.udp.len; /* yes, this is needed */
642         check = packet.udp.check;
643         packet.udp.check = 0;
644         if (check && check != udhcp_checksum(&packet, bytes)) {
645                 log1("Packet with bad UDP checksum received, ignoring");
646                 return -2;
647         }
648
649         memcpy(dhcp_pkt, &packet.data, bytes - (sizeof(packet.ip) + sizeof(packet.udp)));
650
651         if (dhcp_pkt->cookie != htonl(DHCP_MAGIC)) {
652                 bb_info_msg("Packet with bad magic, ignoring");
653                 return -2;
654         }
655         log1("Got valid DHCP packet");
656         udhcp_dump_packet(dhcp_pkt);
657         return bytes - (sizeof(packet.ip) + sizeof(packet.udp));
658 }
659
660
661 /*** Main ***/
662
663 static int sockfd = -1;
664
665 #define LISTEN_NONE   0
666 #define LISTEN_KERNEL 1
667 #define LISTEN_RAW    2
668 static smallint listen_mode;
669
670 /* initial state: (re)start DHCP negotiation */
671 #define INIT_SELECTING  0
672 /* discover was sent, DHCPOFFER reply received */
673 #define REQUESTING      1
674 /* select/renew was sent, DHCPACK reply received */
675 #define BOUND           2
676 /* half of lease passed, want to renew it by sending unicast renew requests */
677 #define RENEWING        3
678 /* renew requests were not answered, lease is almost over, send broadcast renew */
679 #define REBINDING       4
680 /* manually requested renew (SIGUSR1) */
681 #define RENEW_REQUESTED 5
682 /* release, possibly manually requested (SIGUSR2) */
683 #define RELEASED        6
684 static smallint state;
685
686 static int udhcp_raw_socket(int ifindex)
687 {
688         int fd;
689         struct sockaddr_ll sock;
690
691         /*
692          * Comment:
693          *
694          *      I've selected not to see LL header, so BPF doesn't see it, too.
695          *      The filter may also pass non-IP and non-ARP packets, but we do
696          *      a more complete check when receiving the message in userspace.
697          *
698          * and filter shamelessly stolen from:
699          *
700          *      http://www.flamewarmaster.de/software/dhcpclient/
701          *
702          * There are a few other interesting ideas on that page (look under
703          * "Motivation").  Use of netlink events is most interesting.  Think
704          * of various network servers listening for events and reconfiguring.
705          * That would obsolete sending HUP signals and/or make use of restarts.
706          *
707          * Copyright: 2006, 2007 Stefan Rompf <sux@loplof.de>.
708          * License: GPL v2.
709          *
710          * TODO: make conditional?
711          */
712 #define SERVER_AND_CLIENT_PORTS  ((67 << 16) + 68)
713         static const struct sock_filter filter_instr[] = {
714                 /* check for udp */
715                 BPF_STMT(BPF_LD|BPF_B|BPF_ABS, 9),
716                 BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, IPPROTO_UDP, 2, 0),     /* L5, L1, is UDP? */
717                 /* ugly check for arp on ethernet-like and IPv4 */
718                 BPF_STMT(BPF_LD|BPF_W|BPF_ABS, 2),                      /* L1: */
719                 BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, 0x08000604, 3, 4),      /* L3, L4 */
720                 /* skip IP header */
721                 BPF_STMT(BPF_LDX|BPF_B|BPF_MSH, 0),                     /* L5: */
722                 /* check udp source and destination ports */
723                 BPF_STMT(BPF_LD|BPF_W|BPF_IND, 0),
724                 BPF_JUMP(BPF_JMP|BPF_JEQ|BPF_K, SERVER_AND_CLIENT_PORTS, 0, 1), /* L3, L4 */
725                 /* returns */
726                 BPF_STMT(BPF_RET|BPF_K, 0x0fffffff ),                   /* L3: pass */
727                 BPF_STMT(BPF_RET|BPF_K, 0),                             /* L4: reject */
728         };
729         static const struct sock_fprog filter_prog = {
730                 .len = sizeof(filter_instr) / sizeof(filter_instr[0]),
731                 /* casting const away: */
732                 .filter = (struct sock_filter *) filter_instr,
733         };
734
735         log1("Opening raw socket on ifindex %d", ifindex); //log2?
736
737         fd = xsocket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_IP));
738         log1("Got raw socket fd %d", fd); //log2?
739
740         if (SERVER_PORT == 67 && CLIENT_PORT == 68) {
741                 /* Use only if standard ports are in use */
742                 /* Ignoring error (kernel may lack support for this) */
743                 if (setsockopt(fd, SOL_SOCKET, SO_ATTACH_FILTER, &filter_prog,
744                                 sizeof(filter_prog)) >= 0)
745                         log1("Attached filter to raw socket fd %d", fd); // log?
746         }
747
748         sock.sll_family = AF_PACKET;
749         sock.sll_protocol = htons(ETH_P_IP);
750         sock.sll_ifindex = ifindex;
751         xbind(fd, (struct sockaddr *) &sock, sizeof(sock));
752         log1("Created raw socket");
753
754         return fd;
755 }
756
757 static void change_listen_mode(int new_mode)
758 {
759         log1("Entering listen mode: %s",
760                 new_mode != LISTEN_NONE
761                         ? (new_mode == LISTEN_KERNEL ? "kernel" : "raw")
762                         : "none"
763         );
764
765         listen_mode = new_mode;
766         if (sockfd >= 0) {
767                 close(sockfd);
768                 sockfd = -1;
769         }
770         if (new_mode == LISTEN_KERNEL)
771                 sockfd = udhcp_listen_socket(/*INADDR_ANY,*/ CLIENT_PORT, client_config.interface);
772         else if (new_mode != LISTEN_NONE)
773                 sockfd = udhcp_raw_socket(client_config.ifindex);
774         /* else LISTEN_NONE: sockfd stays closed */
775 }
776
777 static void perform_renew(void)
778 {
779         bb_info_msg("Performing a DHCP renew");
780         switch (state) {
781         case BOUND:
782                 change_listen_mode(LISTEN_KERNEL);
783         case RENEWING:
784         case REBINDING:
785                 state = RENEW_REQUESTED;
786                 break;
787         case RENEW_REQUESTED: /* impatient are we? fine, square 1 */
788                 udhcp_run_script(NULL, "deconfig");
789         case REQUESTING:
790         case RELEASED:
791                 change_listen_mode(LISTEN_RAW);
792                 state = INIT_SELECTING;
793                 break;
794         case INIT_SELECTING:
795                 break;
796         }
797 }
798
799 static void perform_release(uint32_t requested_ip, uint32_t server_addr)
800 {
801         char buffer[sizeof("255.255.255.255")];
802         struct in_addr temp_addr;
803
804         /* send release packet */
805         if (state == BOUND || state == RENEWING || state == REBINDING) {
806                 temp_addr.s_addr = server_addr;
807                 strcpy(buffer, inet_ntoa(temp_addr));
808                 temp_addr.s_addr = requested_ip;
809                 bb_info_msg("Unicasting a release of %s to %s",
810                                 inet_ntoa(temp_addr), buffer);
811                 send_release(server_addr, requested_ip); /* unicast */
812                 udhcp_run_script(NULL, "deconfig");
813         }
814         bb_info_msg("Entering released state");
815
816         change_listen_mode(LISTEN_NONE);
817         state = RELEASED;
818 }
819
820 static uint8_t* alloc_dhcp_option(int code, const char *str, int extra)
821 {
822         uint8_t *storage;
823         int len = strnlen(str, 255);
824         storage = xzalloc(len + extra + OPT_DATA);
825         storage[OPT_CODE] = code;
826         storage[OPT_LEN] = len + extra;
827         memcpy(storage + extra + OPT_DATA, str, len);
828         return storage;
829 }
830
831 #if BB_MMU
832 static void client_background(void)
833 {
834         bb_daemonize(0);
835         logmode &= ~LOGMODE_STDIO;
836         /* rewrite pidfile, as our pid is different now */
837         write_pidfile(client_config.pidfile);
838 }
839 #endif
840
841 //usage:#if defined CONFIG_UDHCP_DEBUG && CONFIG_UDHCP_DEBUG >= 1
842 //usage:# define IF_UDHCP_VERBOSE(...) __VA_ARGS__
843 //usage:#else
844 //usage:# define IF_UDHCP_VERBOSE(...)
845 //usage:#endif
846 //usage:#define udhcpc_trivial_usage
847 //usage:       "[-fbnq"IF_UDHCP_VERBOSE("v")"oCR] [-i IFACE] [-r IP] [-s PROG] [-p PIDFILE]\n"
848 //usage:       "        [-H HOSTNAME] [-V VENDOR] [-x OPT:VAL]... [-O OPT]..." IF_FEATURE_UDHCP_PORT(" [-P N]")
849 //usage:#define udhcpc_full_usage "\n"
850 //usage:        IF_LONG_OPTS(
851 //usage:     "\n        -i,--interface IFACE    Interface to use (default eth0)"
852 //usage:     "\n        -p,--pidfile FILE       Create pidfile"
853 //usage:     "\n        -s,--script PROG        Run PROG at DHCP events (default "CONFIG_UDHCPC_DEFAULT_SCRIPT")"
854 //usage:     "\n        -t,--retries N          Send up to N discover packets"
855 //usage:     "\n        -T,--timeout N          Pause between packets (default 3 seconds)"
856 //usage:     "\n        -A,--tryagain N         Wait N seconds after failure (default 20)"
857 //usage:     "\n        -f,--foreground         Run in foreground"
858 //usage:        USE_FOR_MMU(
859 //usage:     "\n        -b,--background         Background if lease is not obtained"
860 //usage:        )
861 //usage:     "\n        -n,--now                Exit if lease is not obtained"
862 //usage:     "\n        -q,--quit               Exit after obtaining lease"
863 //usage:     "\n        -R,--release            Release IP on exit"
864 //usage:     "\n        -S,--syslog             Log to syslog too"
865 //usage:        IF_FEATURE_UDHCP_PORT(
866 //usage:     "\n        -P,--client-port N      Use port N (default 68)"
867 //usage:        )
868 //usage:        IF_FEATURE_UDHCPC_ARPING(
869 //usage:     "\n        -a,--arping             Use arping to validate offered address"
870 //usage:        )
871 //usage:     "\n        -O,--request-option OPT Request option OPT from server (cumulative)"
872 //usage:     "\n        -o,--no-default-options Don't request any options (unless -O is given)"
873 //usage:     "\n        -r,--request IP         Request this IP address"
874 //usage:     "\n        -x OPT:VAL              Include option OPT in sent packets (cumulative)"
875 //usage:     "\n                                Examples of string, numeric, and hex byte opts:"
876 //usage:     "\n                                -x hostname:bbox - option 12"
877 //usage:     "\n                                -x lease:3600 - option 51 (lease time)"
878 //usage:     "\n                                -x 0x3d:0100BEEFC0FFEE - option 61 (client id)"
879 //usage:     "\n        -F,--fqdn NAME          Ask server to update DNS mapping for NAME"
880 //usage:     "\n        -H,-h,--hostname NAME   Send NAME as client hostname (default none)"
881 //usage:     "\n        -V,--vendorclass VENDOR Vendor identifier (default 'udhcp VERSION')"
882 //usage:     "\n        -C,--clientid-none      Don't send MAC as client identifier"
883 //usage:        IF_UDHCP_VERBOSE(
884 //usage:     "\n        -v                      Verbose"
885 //usage:        )
886 //usage:        )
887 //usage:        IF_NOT_LONG_OPTS(
888 //usage:     "\n        -i IFACE        Interface to use (default eth0)"
889 //usage:     "\n        -p FILE         Create pidfile"
890 //usage:     "\n        -s PROG         Run PROG at DHCP events (default "CONFIG_UDHCPC_DEFAULT_SCRIPT")"
891 //usage:     "\n        -t N            Send up to N discover packets"
892 //usage:     "\n        -T N            Pause between packets (default 3 seconds)"
893 //usage:     "\n        -A N            Wait N seconds (default 20) after failure"
894 //usage:     "\n        -f              Run in foreground"
895 //usage:        USE_FOR_MMU(
896 //usage:     "\n        -b              Background if lease is not obtained"
897 //usage:        )
898 //usage:     "\n        -n              Exit if lease is not obtained"
899 //usage:     "\n        -q              Exit after obtaining lease"
900 //usage:     "\n        -R              Release IP on exit"
901 //usage:     "\n        -S              Log to syslog too"
902 //usage:        IF_FEATURE_UDHCP_PORT(
903 //usage:     "\n        -P N            Use port N (default 68)"
904 //usage:        )
905 //usage:        IF_FEATURE_UDHCPC_ARPING(
906 //usage:     "\n        -a              Use arping to validate offered address"
907 //usage:        )
908 //usage:     "\n        -O OPT          Request option OPT from server (cumulative)"
909 //usage:     "\n        -o              Don't request any options (unless -O is given)"
910 //usage:     "\n        -r IP           Request this IP address"
911 //usage:     "\n        -x OPT:VAL      Include option OPT in sent packets (cumulative)"
912 //usage:     "\n                        Examples of string, numeric, and hex byte opts:"
913 //usage:     "\n                        -x hostname:bbox - option 12"
914 //usage:     "\n                        -x lease:3600 - option 51 (lease time)"
915 //usage:     "\n                        -x 0x3d:0100BEEFC0FFEE - option 61 (client id)"
916 //usage:     "\n        -F NAME         Ask server to update DNS mapping for NAME"
917 //usage:     "\n        -H,-h NAME      Send NAME as client hostname (default none)"
918 //usage:     "\n        -V VENDOR       Vendor identifier (default 'udhcp VERSION')"
919 //usage:     "\n        -C              Don't send MAC as client identifier"
920 //usage:        IF_UDHCP_VERBOSE(
921 //usage:     "\n        -v              Verbose"
922 //usage:        )
923 //usage:        )
924
925 int udhcpc_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
926 int udhcpc_main(int argc UNUSED_PARAM, char **argv)
927 {
928         uint8_t *temp, *message;
929         const char *str_V, *str_h, *str_F, *str_r;
930         IF_FEATURE_UDHCP_PORT(char *str_P;)
931         llist_t *list_O = NULL;
932         llist_t *list_x = NULL;
933         int tryagain_timeout = 20;
934         int discover_timeout = 3;
935         int discover_retries = 3;
936         uint32_t server_addr = server_addr; /* for compiler */
937         uint32_t requested_ip = 0;
938         uint32_t xid = 0;
939         uint32_t lease_seconds = 0; /* can be given as 32-bit quantity */
940         int packet_num;
941         int timeout; /* must be signed */
942         unsigned already_waited_sec;
943         unsigned opt;
944         int max_fd;
945         int retval;
946         struct timeval tv;
947         struct dhcp_packet packet;
948         fd_set rfds;
949
950 #if ENABLE_LONG_OPTS
951         static const char udhcpc_longopts[] ALIGN1 =
952                 "clientid-none\0"  No_argument       "C"
953                 "vendorclass\0"    Required_argument "V"
954                 "hostname\0"       Required_argument "H"
955                 "fqdn\0"           Required_argument "F"
956                 "interface\0"      Required_argument "i"
957                 "now\0"            No_argument       "n"
958                 "pidfile\0"        Required_argument "p"
959                 "quit\0"           No_argument       "q"
960                 "release\0"        No_argument       "R"
961                 "request\0"        Required_argument "r"
962                 "script\0"         Required_argument "s"
963                 "timeout\0"        Required_argument "T"
964                 "version\0"        No_argument       "v"
965                 "retries\0"        Required_argument "t"
966                 "tryagain\0"       Required_argument "A"
967                 "syslog\0"         No_argument       "S"
968                 "request-option\0" Required_argument "O"
969                 "no-default-options\0" No_argument   "o"
970                 "foreground\0"     No_argument       "f"
971                 "background\0"     No_argument       "b"
972                 IF_FEATURE_UDHCPC_ARPING("arping\0"     No_argument       "a")
973                 IF_FEATURE_UDHCP_PORT("client-port\0"   Required_argument "P")
974                 ;
975 #endif
976         enum {
977                 OPT_C = 1 << 0,
978                 OPT_V = 1 << 1,
979                 OPT_H = 1 << 2,
980                 OPT_h = 1 << 3,
981                 OPT_F = 1 << 4,
982                 OPT_i = 1 << 5,
983                 OPT_n = 1 << 6,
984                 OPT_p = 1 << 7,
985                 OPT_q = 1 << 8,
986                 OPT_R = 1 << 9,
987                 OPT_r = 1 << 10,
988                 OPT_s = 1 << 11,
989                 OPT_T = 1 << 12,
990                 OPT_t = 1 << 13,
991                 OPT_S = 1 << 14,
992                 OPT_A = 1 << 15,
993                 OPT_O = 1 << 16,
994                 OPT_o = 1 << 17,
995                 OPT_x = 1 << 18,
996                 OPT_f = 1 << 19,
997 /* The rest has variable bit positions, need to be clever */
998                 OPTBIT_f = 19,
999                 USE_FOR_MMU(             OPTBIT_b,)
1000                 IF_FEATURE_UDHCPC_ARPING(OPTBIT_a,)
1001                 IF_FEATURE_UDHCP_PORT(   OPTBIT_P,)
1002                 USE_FOR_MMU(             OPT_b = 1 << OPTBIT_b,)
1003                 IF_FEATURE_UDHCPC_ARPING(OPT_a = 1 << OPTBIT_a,)
1004                 IF_FEATURE_UDHCP_PORT(   OPT_P = 1 << OPTBIT_P,)
1005         };
1006
1007         /* Default options. */
1008         IF_FEATURE_UDHCP_PORT(SERVER_PORT = 67;)
1009         IF_FEATURE_UDHCP_PORT(CLIENT_PORT = 68;)
1010         client_config.interface = "eth0";
1011         client_config.script = CONFIG_UDHCPC_DEFAULT_SCRIPT;
1012         str_V = "udhcp "BB_VER;
1013
1014         /* Parse command line */
1015         /* O,x: list; -T,-t,-A take numeric param */
1016         opt_complementary = "O::x::T+:t+:A+"
1017 #if defined CONFIG_UDHCP_DEBUG && CONFIG_UDHCP_DEBUG >= 1
1018                 ":vv"
1019 #endif
1020                 ;
1021         IF_LONG_OPTS(applet_long_options = udhcpc_longopts;)
1022         opt = getopt32(argv, "CV:H:h:F:i:np:qRr:s:T:t:SA:O:ox:f"
1023                 USE_FOR_MMU("b")
1024                 IF_FEATURE_UDHCPC_ARPING("a")
1025                 IF_FEATURE_UDHCP_PORT("P:")
1026                 "v"
1027                 , &str_V, &str_h, &str_h, &str_F
1028                 , &client_config.interface, &client_config.pidfile, &str_r /* i,p */
1029                 , &client_config.script /* s */
1030                 , &discover_timeout, &discover_retries, &tryagain_timeout /* T,t,A */
1031                 , &list_O
1032                 , &list_x
1033                 IF_FEATURE_UDHCP_PORT(, &str_P)
1034 #if defined CONFIG_UDHCP_DEBUG && CONFIG_UDHCP_DEBUG >= 1
1035                 , &dhcp_verbose
1036 #endif
1037                 );
1038         if (opt & (OPT_h|OPT_H))
1039                 client_config.hostname = alloc_dhcp_option(DHCP_HOST_NAME, str_h, 0);
1040         if (opt & OPT_F) {
1041                 /* FQDN option format: [0x51][len][flags][0][0]<fqdn> */
1042                 client_config.fqdn = alloc_dhcp_option(DHCP_FQDN, str_F, 3);
1043                 /* Flag bits: 0000NEOS
1044                  * S: 1 = Client requests server to update A RR in DNS as well as PTR
1045                  * O: 1 = Server indicates to client that DNS has been updated regardless
1046                  * E: 1 = Name is in DNS format, i.e. <4>host<6>domain<3>com<0>,
1047                  *    not "host.domain.com". Format 0 is obsolete.
1048                  * N: 1 = Client requests server to not update DNS (S must be 0 then)
1049                  * Two [0] bytes which follow are deprecated and must be 0.
1050                  */
1051                 client_config.fqdn[OPT_DATA + 0] = 0x1;
1052                 /*client_config.fqdn[OPT_DATA + 1] = 0; - xzalloc did it */
1053                 /*client_config.fqdn[OPT_DATA + 2] = 0; */
1054         }
1055         if (opt & OPT_r)
1056                 requested_ip = inet_addr(str_r);
1057 #if ENABLE_FEATURE_UDHCP_PORT
1058         if (opt & OPT_P) {
1059                 CLIENT_PORT = xatou16(str_P);
1060                 SERVER_PORT = CLIENT_PORT - 1;
1061         }
1062 #endif
1063         if (opt & OPT_o)
1064                 client_config.no_default_options = 1;
1065         while (list_O) {
1066                 char *optstr = llist_pop(&list_O);
1067                 unsigned n = udhcp_option_idx(optstr);
1068                 n = dhcp_optflags[n].code;
1069                 client_config.opt_mask[n >> 3] |= 1 << (n & 7);
1070         }
1071         while (list_x) {
1072                 char *optstr = llist_pop(&list_x);
1073                 char *colon = strchr(optstr, ':');
1074                 if (colon)
1075                         *colon = ' ';
1076                 /* now it looks similar to udhcpd's config file line:
1077                  * "optname optval", using the common routine: */
1078                 udhcp_str2optset(optstr, &client_config.options);
1079         }
1080
1081         if (udhcp_read_interface(client_config.interface,
1082                         &client_config.ifindex,
1083                         NULL,
1084                         client_config.client_mac)
1085         ) {
1086                 return 1;
1087         }
1088
1089         if (!(opt & OPT_C) && !udhcp_find_option(client_config.options, DHCP_CLIENT_ID)) {
1090                 /* not suppressed and not set, set the default client ID */
1091                 client_config.clientid = alloc_dhcp_option(DHCP_CLIENT_ID, "", 7);
1092                 client_config.clientid[OPT_DATA] = 1; /* type: ethernet */
1093                 memcpy(client_config.clientid + OPT_DATA+1, client_config.client_mac, 6);
1094         }
1095         if (str_V[0] != '\0')
1096                 client_config.vendorclass = alloc_dhcp_option(DHCP_VENDOR, str_V, 0);
1097 #if !BB_MMU
1098         /* on NOMMU reexec (i.e., background) early */
1099         if (!(opt & OPT_f)) {
1100                 bb_daemonize_or_rexec(0 /* flags */, argv);
1101                 logmode = LOGMODE_NONE;
1102         }
1103 #endif
1104         if (opt & OPT_S) {
1105                 openlog(applet_name, LOG_PID, LOG_DAEMON);
1106                 logmode |= LOGMODE_SYSLOG;
1107         }
1108
1109         /* Make sure fd 0,1,2 are open */
1110         bb_sanitize_stdio();
1111         /* Equivalent of doing a fflush after every \n */
1112         setlinebuf(stdout);
1113         /* Create pidfile */
1114         write_pidfile(client_config.pidfile);
1115         /* Goes to stdout (unless NOMMU) and possibly syslog */
1116         bb_info_msg("%s (v"BB_VER") started", applet_name);
1117         /* Set up the signal pipe */
1118         udhcp_sp_setup();
1119         /* We want random_xid to be random... */
1120         srand(monotonic_us());
1121
1122         state = INIT_SELECTING;
1123         udhcp_run_script(NULL, "deconfig");
1124         change_listen_mode(LISTEN_RAW);
1125         packet_num = 0;
1126         timeout = 0;
1127         already_waited_sec = 0;
1128
1129         /* Main event loop. select() waits on signal pipe and possibly
1130          * on sockfd.
1131          * "continue" statements in code below jump to the top of the loop.
1132          */
1133         for (;;) {
1134                 /* silence "uninitialized!" warning */
1135                 unsigned timestamp_before_wait = timestamp_before_wait;
1136
1137                 //bb_error_msg("sockfd:%d, listen_mode:%d", sockfd, listen_mode);
1138
1139                 /* Was opening raw or udp socket here
1140                  * if (listen_mode != LISTEN_NONE && sockfd < 0),
1141                  * but on fast network renew responses return faster
1142                  * than we open sockets. Thus this code is moved
1143                  * to change_listen_mode(). Thus we open listen socket
1144                  * BEFORE we send renew request (see "case BOUND:"). */
1145
1146                 max_fd = udhcp_sp_fd_set(&rfds, sockfd);
1147
1148                 tv.tv_sec = timeout - already_waited_sec;
1149                 tv.tv_usec = 0;
1150                 retval = 0;
1151                 /* If we already timed out, fall through with retval = 0, else... */
1152                 if ((int)tv.tv_sec > 0) {
1153                         timestamp_before_wait = (unsigned)monotonic_sec();
1154                         log1("Waiting on select...");
1155                         retval = select(max_fd + 1, &rfds, NULL, NULL, &tv);
1156                         if (retval < 0) {
1157                                 /* EINTR? A signal was caught, don't panic */
1158                                 if (errno == EINTR) {
1159                                         already_waited_sec += (unsigned)monotonic_sec() - timestamp_before_wait;
1160                                         continue;
1161                                 }
1162                                 /* Else: an error occured, panic! */
1163                                 bb_perror_msg_and_die("select");
1164                         }
1165                 }
1166
1167                 /* If timeout dropped to zero, time to become active:
1168                  * resend discover/renew/whatever
1169                  */
1170                 if (retval == 0) {
1171                         /* When running on a bridge, the ifindex may have changed
1172                          * (e.g. if member interfaces were added/removed
1173                          * or if the status of the bridge changed).
1174                          * Refresh ifindex and client_mac:
1175                          */
1176                         udhcp_read_interface(client_config.interface,
1177                                 &client_config.ifindex,
1178                                 NULL,
1179                                 client_config.client_mac);
1180
1181                         /* We will restart the wait in any case */
1182                         already_waited_sec = 0;
1183
1184                         switch (state) {
1185                         case INIT_SELECTING:
1186                                 if (packet_num < discover_retries) {
1187                                         if (packet_num == 0)
1188                                                 xid = random_xid();
1189                                         /* broadcast */
1190                                         send_discover(xid, requested_ip);
1191                                         timeout = discover_timeout;
1192                                         packet_num++;
1193                                         continue;
1194                                 }
1195  leasefail:
1196                                 udhcp_run_script(NULL, "leasefail");
1197 #if BB_MMU /* -b is not supported on NOMMU */
1198                                 if (opt & OPT_b) { /* background if no lease */
1199                                         bb_info_msg("No lease, forking to background");
1200                                         client_background();
1201                                         /* do not background again! */
1202                                         opt = ((opt & ~OPT_b) | OPT_f);
1203                                 } else
1204 #endif
1205                                 if (opt & OPT_n) { /* abort if no lease */
1206                                         bb_info_msg("No lease, failing");
1207                                         retval = 1;
1208                                         goto ret;
1209                                 }
1210                                 /* wait before trying again */
1211                                 timeout = tryagain_timeout;
1212                                 packet_num = 0;
1213                                 continue;
1214                         case REQUESTING:
1215                                 if (packet_num < discover_retries) {
1216                                         /* send broadcast select packet */
1217                                         send_select(xid, server_addr, requested_ip);
1218                                         timeout = discover_timeout;
1219                                         packet_num++;
1220                                         continue;
1221                                 }
1222                                 /* Timed out, go back to init state.
1223                                  * "discover...select...discover..." loops
1224                                  * were seen in the wild. Treat them similarly
1225                                  * to "no response to discover" case */
1226                                 change_listen_mode(LISTEN_RAW);
1227                                 state = INIT_SELECTING;
1228                                 goto leasefail;
1229                         case BOUND:
1230                                 /* 1/2 lease passed, enter renewing state */
1231                                 state = RENEWING;
1232                                 change_listen_mode(LISTEN_KERNEL);
1233                                 log1("Entering renew state");
1234                                 /* fall right through */
1235                         case RENEW_REQUESTED: /* manual (SIGUSR1) renew */
1236                         case_RENEW_REQUESTED:
1237                         case RENEWING:
1238                                 if (timeout > 60) {
1239                                         /* send an unicast renew request */
1240                         /* Sometimes observed to fail (EADDRNOTAVAIL) to bind
1241                          * a new UDP socket for sending inside send_renew.
1242                          * I hazard to guess existing listening socket
1243                          * is somehow conflicting with it, but why is it
1244                          * not deterministic then?! Strange.
1245                          * Anyway, it does recover by eventually failing through
1246                          * into INIT_SELECTING state.
1247                          */
1248                                         send_renew(xid, server_addr, requested_ip);
1249                                         timeout >>= 1;
1250                                         continue;
1251                                 }
1252                                 /* Timed out, enter rebinding state */
1253                                 log1("Entering rebinding state");
1254                                 state = REBINDING;
1255                                 /* fall right through */
1256                         case REBINDING:
1257                                 /* Switch to bcast receive */
1258                                 change_listen_mode(LISTEN_RAW);
1259                                 /* Lease is *really* about to run out,
1260                                  * try to find DHCP server using broadcast */
1261                                 if (timeout > 0) {
1262                                         /* send a broadcast renew request */
1263                                         send_renew(xid, 0 /*INADDR_ANY*/, requested_ip);
1264                                         timeout >>= 1;
1265                                         continue;
1266                                 }
1267                                 /* Timed out, enter init state */
1268                                 bb_info_msg("Lease lost, entering init state");
1269                                 udhcp_run_script(NULL, "deconfig");
1270                                 state = INIT_SELECTING;
1271                                 /*timeout = 0; - already is */
1272                                 packet_num = 0;
1273                                 continue;
1274                         /* case RELEASED: */
1275                         }
1276                         /* yah, I know, *you* say it would never happen */
1277                         timeout = INT_MAX;
1278                         continue; /* back to main loop */
1279                 } /* if select timed out */
1280
1281                 /* select() didn't timeout, something happened */
1282
1283                 /* Is it a signal? */
1284                 /* note: udhcp_sp_read checks FD_ISSET before reading */
1285                 switch (udhcp_sp_read(&rfds)) {
1286                 case SIGUSR1:
1287                         perform_renew();
1288                         if (state == RENEW_REQUESTED)
1289                                 goto case_RENEW_REQUESTED;
1290                         /* Start things over */
1291                         packet_num = 0;
1292                         /* Kill any timeouts, user wants this to hurry along */
1293                         timeout = 0;
1294                         continue;
1295                 case SIGUSR2:
1296                         perform_release(requested_ip, server_addr);
1297                         timeout = INT_MAX;
1298                         continue;
1299                 case SIGTERM:
1300                         bb_info_msg("Received SIGTERM");
1301                         if (opt & OPT_R) /* release on quit */
1302                                 perform_release(requested_ip, server_addr);
1303                         goto ret0;
1304                 }
1305
1306                 /* Is it a packet? */
1307                 if (listen_mode == LISTEN_NONE || !FD_ISSET(sockfd, &rfds))
1308                         continue; /* no */
1309
1310                 {
1311                         int len;
1312
1313                         /* A packet is ready, read it */
1314                         if (listen_mode == LISTEN_KERNEL)
1315                                 len = udhcp_recv_kernel_packet(&packet, sockfd);
1316                         else
1317                                 len = udhcp_recv_raw_packet(&packet, sockfd);
1318                         if (len == -1) {
1319                                 /* Error is severe, reopen socket */
1320                                 bb_info_msg("Read error: %s, reopening socket", strerror(errno));
1321                                 sleep(discover_timeout); /* 3 seconds by default */
1322                                 change_listen_mode(listen_mode); /* just close and reopen */
1323                         }
1324                         /* If this packet will turn out to be unrelated/bogus,
1325                          * we will go back and wait for next one.
1326                          * Be sure timeout is properly decreased. */
1327                         already_waited_sec += (unsigned)monotonic_sec() - timestamp_before_wait;
1328                         if (len < 0)
1329                                 continue;
1330                 }
1331
1332                 if (packet.xid != xid) {
1333                         log1("xid %x (our is %x), ignoring packet",
1334                                 (unsigned)packet.xid, (unsigned)xid);
1335                         continue;
1336                 }
1337
1338                 /* Ignore packets that aren't for us */
1339                 if (packet.hlen != 6
1340                  || memcmp(packet.chaddr, client_config.client_mac, 6) != 0
1341                 ) {
1342 //FIXME: need to also check that last 10 bytes are zero
1343                         log1("chaddr does not match, ignoring packet"); // log2?
1344                         continue;
1345                 }
1346
1347                 message = udhcp_get_option(&packet, DHCP_MESSAGE_TYPE);
1348                 if (message == NULL) {
1349                         bb_error_msg("no message type option, ignoring packet");
1350                         continue;
1351                 }
1352
1353                 switch (state) {
1354                 case INIT_SELECTING:
1355                         /* Must be a DHCPOFFER to one of our xid's */
1356                         if (*message == DHCPOFFER) {
1357                 /* TODO: why we don't just fetch server's IP from IP header? */
1358                                 temp = udhcp_get_option(&packet, DHCP_SERVER_ID);
1359                                 if (!temp) {
1360                                         bb_error_msg("no server ID, ignoring packet");
1361                                         continue;
1362                                         /* still selecting - this server looks bad */
1363                                 }
1364                                 /* it IS unaligned sometimes, don't "optimize" */
1365                                 move_from_unaligned32(server_addr, temp);
1366                                 /*xid = packet.xid; - already is */
1367                                 requested_ip = packet.yiaddr;
1368
1369                                 /* enter requesting state */
1370                                 state = REQUESTING;
1371                                 timeout = 0;
1372                                 packet_num = 0;
1373                                 already_waited_sec = 0;
1374                         }
1375                         continue;
1376                 case REQUESTING:
1377                 case RENEWING:
1378                 case RENEW_REQUESTED:
1379                 case REBINDING:
1380                         if (*message == DHCPACK) {
1381                                 temp = udhcp_get_option(&packet, DHCP_LEASE_TIME);
1382                                 if (!temp) {
1383                                         bb_error_msg("no lease time with ACK, using 1 hour lease");
1384                                         lease_seconds = 60 * 60;
1385                                 } else {
1386                                         /* it IS unaligned sometimes, don't "optimize" */
1387                                         move_from_unaligned32(lease_seconds, temp);
1388                                         lease_seconds = ntohl(lease_seconds);
1389                                         lease_seconds &= 0x0fffffff; /* paranoia: must not be prone to overflows */
1390                                         if (lease_seconds < 10) /* and not too small */
1391                                                 lease_seconds = 10;
1392                                 }
1393 #if ENABLE_FEATURE_UDHCPC_ARPING
1394                                 if (opt & OPT_a) {
1395 /* RFC 2131 3.1 paragraph 5:
1396  * "The client receives the DHCPACK message with configuration
1397  * parameters. The client SHOULD perform a final check on the
1398  * parameters (e.g., ARP for allocated network address), and notes
1399  * the duration of the lease specified in the DHCPACK message. At this
1400  * point, the client is configured. If the client detects that the
1401  * address is already in use (e.g., through the use of ARP),
1402  * the client MUST send a DHCPDECLINE message to the server and restarts
1403  * the configuration process..." */
1404                                         if (!arpping(packet.yiaddr,
1405                                                         NULL,
1406                                                         (uint32_t) 0,
1407                                                         client_config.client_mac,
1408                                                         client_config.interface)
1409                                         ) {
1410                                                 bb_info_msg("Offered address is in use "
1411                                                         "(got ARP reply), declining");
1412                                                 send_decline(xid, server_addr, packet.yiaddr);
1413
1414                                                 if (state != REQUESTING)
1415                                                         udhcp_run_script(NULL, "deconfig");
1416                                                 change_listen_mode(LISTEN_RAW);
1417                                                 state = INIT_SELECTING;
1418                                                 requested_ip = 0;
1419                                                 timeout = tryagain_timeout;
1420                                                 packet_num = 0;
1421                                                 already_waited_sec = 0;
1422                                                 continue; /* back to main loop */
1423                                         }
1424                                 }
1425 #endif
1426                                 /* enter bound state */
1427                                 timeout = lease_seconds / 2;
1428                                 {
1429                                         struct in_addr temp_addr;
1430                                         temp_addr.s_addr = packet.yiaddr;
1431                                         bb_info_msg("Lease of %s obtained, lease time %u",
1432                                                 inet_ntoa(temp_addr), (unsigned)lease_seconds);
1433                                 }
1434                                 requested_ip = packet.yiaddr;
1435                                 udhcp_run_script(&packet, state == REQUESTING ? "bound" : "renew");
1436
1437                                 state = BOUND;
1438                                 change_listen_mode(LISTEN_NONE);
1439                                 if (opt & OPT_q) { /* quit after lease */
1440                                         if (opt & OPT_R) /* release on quit */
1441                                                 perform_release(requested_ip, server_addr);
1442                                         goto ret0;
1443                                 }
1444                                 /* future renew failures should not exit (JM) */
1445                                 opt &= ~OPT_n;
1446 #if BB_MMU /* NOMMU case backgrounded earlier */
1447                                 if (!(opt & OPT_f)) {
1448                                         client_background();
1449                                         /* do not background again! */
1450                                         opt = ((opt & ~OPT_b) | OPT_f);
1451                                 }
1452 #endif
1453                                 already_waited_sec = 0;
1454                                 continue; /* back to main loop */
1455                         }
1456                         if (*message == DHCPNAK) {
1457                                 /* return to init state */
1458                                 bb_info_msg("Received DHCP NAK");
1459                                 udhcp_run_script(&packet, "nak");
1460                                 if (state != REQUESTING)
1461                                         udhcp_run_script(NULL, "deconfig");
1462                                 change_listen_mode(LISTEN_RAW);
1463                                 sleep(3); /* avoid excessive network traffic */
1464                                 state = INIT_SELECTING;
1465                                 requested_ip = 0;
1466                                 timeout = 0;
1467                                 packet_num = 0;
1468                                 already_waited_sec = 0;
1469                         }
1470                         continue;
1471                 /* case BOUND: - ignore all packets */
1472                 /* case RELEASED: - ignore all packets */
1473                 }
1474                 /* back to main loop */
1475         } /* for (;;) - main loop ends */
1476
1477  ret0:
1478         retval = 0;
1479  ret:
1480         /*if (client_config.pidfile) - remove_pidfile has its own check */
1481                 remove_pidfile(client_config.pidfile);
1482         return retval;
1483 }