net: bootp: Add environment variable for timeout period
[platform/kernel/u-boot.git] / net / bootp.c
1 /*
2  *      Based on LiMon - BOOTP.
3  *
4  *      Copyright 1994, 1995, 2000 Neil Russell.
5  *      (See License)
6  *      Copyright 2000 Roland Borde
7  *      Copyright 2000 Paolo Scaffardi
8  *      Copyright 2000-2004 Wolfgang Denk, wd@denx.de
9  */
10
11 #include <common.h>
12 #include <command.h>
13 #include <net.h>
14 #include <net/tftp.h>
15 #include "bootp.h"
16 #include "nfs.h"
17 #ifdef CONFIG_STATUS_LED
18 #include <status_led.h>
19 #endif
20 #ifdef CONFIG_BOOTP_RANDOM_DELAY
21 #include "net_rand.h"
22 #endif
23
24 #define BOOTP_VENDOR_MAGIC      0x63825363      /* RFC1048 Magic Cookie */
25
26 /*
27  * The timeout for the initial BOOTP/DHCP request used to be described by a
28  * counter of fixed-length timeout periods. TIMEOUT_COUNT represents
29  * that counter
30  *
31  * Now that the timeout periods are variable (exponential backoff and retry)
32  * we convert the timeout count to the absolute time it would have take to
33  * execute that many retries, and keep sending retry packets until that time
34  * is reached.
35  */
36 #ifndef CONFIG_NET_RETRY_COUNT
37 # define TIMEOUT_COUNT  5               /* # of timeouts before giving up */
38 #else
39 # define TIMEOUT_COUNT  (CONFIG_NET_RETRY_COUNT)
40 #endif
41 #define TIMEOUT_MS      ((3 + (TIMEOUT_COUNT * 5)) * 1000)
42
43 #define PORT_BOOTPS     67              /* BOOTP server UDP port */
44 #define PORT_BOOTPC     68              /* BOOTP client UDP port */
45
46 #ifndef CONFIG_DHCP_MIN_EXT_LEN         /* minimal length of extension list */
47 #define CONFIG_DHCP_MIN_EXT_LEN 64
48 #endif
49
50 #ifndef CONFIG_BOOTP_ID_CACHE_SIZE
51 #define CONFIG_BOOTP_ID_CACHE_SIZE 4
52 #endif
53
54 u32             bootp_ids[CONFIG_BOOTP_ID_CACHE_SIZE];
55 unsigned int    bootp_num_ids;
56 int             bootp_try;
57 ulong           bootp_start;
58 ulong           bootp_timeout;
59 char net_nis_domain[32] = {0,}; /* Our NIS domain */
60 char net_hostname[32] = {0,}; /* Our hostname */
61 char net_root_path[64] = {0,}; /* Our bootpath */
62
63 static ulong time_taken_max;
64
65 #if defined(CONFIG_CMD_DHCP)
66 static dhcp_state_t dhcp_state = INIT;
67 static u32 dhcp_leasetime;
68 static struct in_addr dhcp_server_ip;
69 static u8 dhcp_option_overload;
70 #define OVERLOAD_FILE 1
71 #define OVERLOAD_SNAME 2
72 static void dhcp_handler(uchar *pkt, unsigned dest, struct in_addr sip,
73                         unsigned src, unsigned len);
74
75 /* For Debug */
76 #if 0
77 static char *dhcpmsg2str(int type)
78 {
79         switch (type) {
80         case 1:  return "DHCPDISCOVER"; break;
81         case 2:  return "DHCPOFFER";    break;
82         case 3:  return "DHCPREQUEST";  break;
83         case 4:  return "DHCPDECLINE";  break;
84         case 5:  return "DHCPACK";      break;
85         case 6:  return "DHCPNACK";     break;
86         case 7:  return "DHCPRELEASE";  break;
87         default: return "UNKNOWN/INVALID MSG TYPE"; break;
88         }
89 }
90 #endif
91 #endif
92
93 static void bootp_add_id(ulong id)
94 {
95         if (bootp_num_ids >= ARRAY_SIZE(bootp_ids)) {
96                 size_t size = sizeof(bootp_ids) - sizeof(id);
97
98                 memmove(bootp_ids, &bootp_ids[1], size);
99                 bootp_ids[bootp_num_ids - 1] = id;
100         } else {
101                 bootp_ids[bootp_num_ids] = id;
102                 bootp_num_ids++;
103         }
104 }
105
106 static bool bootp_match_id(ulong id)
107 {
108         unsigned int i;
109
110         for (i = 0; i < bootp_num_ids; i++)
111                 if (bootp_ids[i] == id)
112                         return true;
113
114         return false;
115 }
116
117 static int check_reply_packet(uchar *pkt, unsigned dest, unsigned src,
118                               unsigned len)
119 {
120         struct bootp_hdr *bp = (struct bootp_hdr *)pkt;
121         int retval = 0;
122
123         if (dest != PORT_BOOTPC || src != PORT_BOOTPS)
124                 retval = -1;
125         else if (len < sizeof(struct bootp_hdr) - OPT_FIELD_SIZE)
126                 retval = -2;
127         else if (bp->bp_op != OP_BOOTREPLY)
128                 retval = -3;
129         else if (bp->bp_htype != HWT_ETHER)
130                 retval = -4;
131         else if (bp->bp_hlen != HWL_ETHER)
132                 retval = -5;
133         else if (!bootp_match_id(net_read_u32(&bp->bp_id)))
134                 retval = -6;
135
136         debug("Filtering pkt = %d\n", retval);
137
138         return retval;
139 }
140
141 /*
142  * Copy parameters of interest from BOOTP_REPLY/DHCP_OFFER packet
143  */
144 static void store_net_params(struct bootp_hdr *bp)
145 {
146 #if !defined(CONFIG_BOOTP_SERVERIP)
147         struct in_addr tmp_ip;
148
149         net_copy_ip(&tmp_ip, &bp->bp_siaddr);
150         if (tmp_ip.s_addr != 0)
151                 net_copy_ip(&net_server_ip, &bp->bp_siaddr);
152         memcpy(net_server_ethaddr,
153                ((struct ethernet_hdr *)net_rx_packet)->et_src, 6);
154         if (
155 #if defined(CONFIG_CMD_DHCP)
156             !(dhcp_option_overload & OVERLOAD_FILE) &&
157 #endif
158             (strlen(bp->bp_file) > 0)) {
159                 copy_filename(net_boot_file_name, bp->bp_file,
160                               sizeof(net_boot_file_name));
161         }
162
163         debug("net_boot_file_name: %s\n", net_boot_file_name);
164
165         /* Propagate to environment:
166          * don't delete exising entry when BOOTP / DHCP reply does
167          * not contain a new value
168          */
169         if (*net_boot_file_name)
170                 setenv("bootfile", net_boot_file_name);
171 #endif
172         net_copy_ip(&net_ip, &bp->bp_yiaddr);
173 }
174
175 static int truncate_sz(const char *name, int maxlen, int curlen)
176 {
177         if (curlen >= maxlen) {
178                 printf("*** WARNING: %s is too long (%d - max: %d)"
179                         " - truncated\n", name, curlen, maxlen);
180                 curlen = maxlen - 1;
181         }
182         return curlen;
183 }
184
185 #if !defined(CONFIG_CMD_DHCP)
186
187 static void bootp_process_vendor_field(u8 *ext)
188 {
189         int size = *(ext + 1);
190
191         debug("[BOOTP] Processing extension %d... (%d bytes)\n", *ext,
192               *(ext + 1));
193
194         net_boot_file_expected_size_in_blocks = 0;
195
196         switch (*ext) {
197                 /* Fixed length fields */
198         case 1:                 /* Subnet mask */
199                 if (net_netmask.s_addr == 0)
200                         net_copy_ip(&net_netmask, (struct in_addr *)(ext + 2));
201                 break;
202         case 2:                 /* Time offset - Not yet supported */
203                 break;
204                 /* Variable length fields */
205         case 3:                 /* Gateways list */
206                 if (net_gateway.s_addr == 0)
207                         net_copy_ip(&net_gateway, (struct in_addr *)(ext + 2));
208                 break;
209         case 4:                 /* Time server - Not yet supported */
210                 break;
211         case 5:                 /* IEN-116 name server - Not yet supported */
212                 break;
213         case 6:
214                 if (net_dns_server.s_addr == 0)
215                         net_copy_ip(&net_dns_server,
216                                     (struct in_addr *)(ext + 2));
217 #if defined(CONFIG_BOOTP_DNS2)
218                 if ((net_dns_server2.s_addr == 0) && (size > 4))
219                         net_copy_ip(&net_dns_server2,
220                                     (struct in_addr *)(ext + 2 + 4));
221 #endif
222                 break;
223         case 7:                 /* Log server - Not yet supported */
224                 break;
225         case 8:                 /* Cookie/Quote server - Not yet supported */
226                 break;
227         case 9:                 /* LPR server - Not yet supported */
228                 break;
229         case 10:                /* Impress server - Not yet supported */
230                 break;
231         case 11:                /* RPL server - Not yet supported */
232                 break;
233         case 12:                /* Host name */
234                 if (net_hostname[0] == 0) {
235                         size = truncate_sz("Host Name",
236                                 sizeof(net_hostname), size);
237                         memcpy(&net_hostname, ext + 2, size);
238                         net_hostname[size] = 0;
239                 }
240                 break;
241         case 13:                /* Boot file size */
242                 if (size == 2)
243                         net_boot_file_expected_size_in_blocks =
244                                 ntohs(*(ushort *)(ext + 2));
245                 else if (size == 4)
246                         net_boot_file_expected_size_in_blocks =
247                                 ntohl(*(ulong *)(ext + 2));
248                 break;
249         case 14:                /* Merit dump file - Not yet supported */
250                 break;
251         case 15:                /* Domain name - Not yet supported */
252                 break;
253         case 16:                /* Swap server - Not yet supported */
254                 break;
255         case 17:                /* Root path */
256                 if (net_root_path[0] == 0) {
257                         size = truncate_sz("Root Path",
258                                 sizeof(net_root_path), size);
259                         memcpy(&net_root_path, ext + 2, size);
260                         net_root_path[size] = 0;
261                 }
262                 break;
263         case 18:                /* Extension path - Not yet supported */
264                 /*
265                  * This can be used to send the information of the
266                  * vendor area in another file that the client can
267                  * access via TFTP.
268                  */
269                 break;
270                 /* IP host layer fields */
271         case 40:                /* NIS Domain name */
272                 if (net_nis_domain[0] == 0) {
273                         size = truncate_sz("NIS Domain Name",
274                                 sizeof(net_nis_domain), size);
275                         memcpy(&net_nis_domain, ext + 2, size);
276                         net_nis_domain[size] = 0;
277                 }
278                 break;
279 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_NTPSERVER)
280         case 42:        /* NTP server IP */
281                 net_copy_ip(&net_ntp_server, (struct in_addr *)(ext + 2));
282                 break;
283 #endif
284                 /* Application layer fields */
285         case 43:                /* Vendor specific info - Not yet supported */
286                 /*
287                  * Binary information to exchange specific
288                  * product information.
289                  */
290                 break;
291                 /* Reserved (custom) fields (128..254) */
292         }
293 }
294
295 static void bootp_process_vendor(u8 *ext, int size)
296 {
297         u8 *end = ext + size;
298
299         debug("[BOOTP] Checking extension (%d bytes)...\n", size);
300
301         while ((ext < end) && (*ext != 0xff)) {
302                 if (*ext == 0) {
303                         ext++;
304                 } else {
305                         u8 *opt = ext;
306
307                         ext += ext[1] + 2;
308                         if (ext <= end)
309                                 bootp_process_vendor_field(opt);
310                 }
311         }
312
313         debug("[BOOTP] Received fields:\n");
314         if (net_netmask.s_addr)
315                 debug("net_netmask : %pI4\n", &net_netmask);
316
317         if (net_gateway.s_addr)
318                 debug("net_gateway      : %pI4", &net_gateway);
319
320         if (net_boot_file_expected_size_in_blocks)
321                 debug("net_boot_file_expected_size_in_blocks : %d\n",
322                       net_boot_file_expected_size_in_blocks);
323
324         if (net_hostname[0])
325                 debug("net_hostname  : %s\n", net_hostname);
326
327         if (net_root_path[0])
328                 debug("net_root_path  : %s\n", net_root_path);
329
330         if (net_nis_domain[0])
331                 debug("net_nis_domain : %s\n", net_nis_domain);
332
333 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_NTPSERVER)
334         if (net_ntp_server)
335                 debug("net_ntp_server : %pI4\n", &net_ntp_server);
336 #endif
337 }
338
339 /*
340  *      Handle a BOOTP received packet.
341  */
342 static void bootp_handler(uchar *pkt, unsigned dest, struct in_addr sip,
343                           unsigned src, unsigned len)
344 {
345         struct bootp_hdr *bp;
346
347         debug("got BOOTP packet (src=%d, dst=%d, len=%d want_len=%zu)\n",
348               src, dest, len, sizeof(struct bootp_hdr));
349
350         bp = (struct bootp_hdr *)pkt;
351
352         /* Filter out pkts we don't want */
353         if (check_reply_packet(pkt, dest, src, len))
354                 return;
355
356         /*
357          *      Got a good BOOTP reply.  Copy the data into our variables.
358          */
359 #if defined(CONFIG_STATUS_LED) && defined(STATUS_LED_BOOT)
360         status_led_set(STATUS_LED_BOOT, STATUS_LED_OFF);
361 #endif
362
363         store_net_params(bp);           /* Store net parameters from reply */
364
365         /* Retrieve extended information (we must parse the vendor area) */
366         if (net_read_u32((u32 *)&bp->bp_vend[0]) == htonl(BOOTP_VENDOR_MAGIC))
367                 bootp_process_vendor((uchar *)&bp->bp_vend[4], len);
368
369         net_set_timeout_handler(0, (thand_f *)0);
370         bootstage_mark_name(BOOTSTAGE_ID_BOOTP_STOP, "bootp_stop");
371
372         debug("Got good BOOTP\n");
373
374         net_auto_load();
375 }
376 #endif
377
378 /*
379  *      Timeout on BOOTP/DHCP request.
380  */
381 static void bootp_timeout_handler(void)
382 {
383         ulong time_taken = get_timer(bootp_start);
384
385         if (time_taken >= time_taken_max) {
386 #ifdef CONFIG_BOOTP_MAY_FAIL
387                 puts("\nRetry time exceeded\n");
388                 net_set_state(NETLOOP_FAIL);
389 #else
390                 puts("\nRetry time exceeded; starting again\n");
391                 net_start_again();
392 #endif
393         } else {
394                 bootp_timeout *= 2;
395                 if (bootp_timeout > 2000)
396                         bootp_timeout = 2000;
397                 net_set_timeout_handler(bootp_timeout, bootp_timeout_handler);
398                 bootp_request();
399         }
400 }
401
402 #define put_vci(e, str)                                         \
403         do {                                                    \
404                 size_t vci_strlen = strlen(str);                \
405                 *e++ = 60;      /* Vendor Class Identifier */   \
406                 *e++ = vci_strlen;                              \
407                 memcpy(e, str, vci_strlen);                     \
408                 e += vci_strlen;                                \
409         } while (0)
410
411 /*
412  *      Initialize BOOTP extension fields in the request.
413  */
414 #if defined(CONFIG_CMD_DHCP)
415 static int dhcp_extended(u8 *e, int message_type, struct in_addr server_ip,
416                         struct in_addr requested_ip)
417 {
418         u8 *start = e;
419         u8 *cnt;
420 #if defined(CONFIG_BOOTP_PXE)
421         char *uuid;
422         u16 clientarch;
423 #endif
424
425 #if defined(CONFIG_BOOTP_VENDOREX)
426         u8 *x;
427 #endif
428 #if defined(CONFIG_BOOTP_SEND_HOSTNAME)
429         char *hostname;
430 #endif
431
432         *e++ = 99;              /* RFC1048 Magic Cookie */
433         *e++ = 130;
434         *e++ = 83;
435         *e++ = 99;
436
437         *e++ = 53;              /* DHCP Message Type */
438         *e++ = 1;
439         *e++ = message_type;
440
441         *e++ = 57;              /* Maximum DHCP Message Size */
442         *e++ = 2;
443         *e++ = (576 - 312 + OPT_FIELD_SIZE) >> 8;
444         *e++ = (576 - 312 + OPT_FIELD_SIZE) & 0xff;
445
446         if (server_ip.s_addr) {
447                 int tmp = ntohl(server_ip.s_addr);
448
449                 *e++ = 54;      /* ServerID */
450                 *e++ = 4;
451                 *e++ = tmp >> 24;
452                 *e++ = tmp >> 16;
453                 *e++ = tmp >> 8;
454                 *e++ = tmp & 0xff;
455         }
456
457         if (requested_ip.s_addr) {
458                 int tmp = ntohl(requested_ip.s_addr);
459
460                 *e++ = 50;      /* Requested IP */
461                 *e++ = 4;
462                 *e++ = tmp >> 24;
463                 *e++ = tmp >> 16;
464                 *e++ = tmp >> 8;
465                 *e++ = tmp & 0xff;
466         }
467 #if defined(CONFIG_BOOTP_SEND_HOSTNAME)
468         hostname = getenv("hostname");
469         if (hostname) {
470                 int hostnamelen = strlen(hostname);
471
472                 *e++ = 12;      /* Hostname */
473                 *e++ = hostnamelen;
474                 memcpy(e, hostname, hostnamelen);
475                 e += hostnamelen;
476         }
477 #endif
478
479 #if defined(CONFIG_BOOTP_PXE)
480         clientarch = CONFIG_BOOTP_PXE_CLIENTARCH;
481         *e++ = 93;      /* Client System Architecture */
482         *e++ = 2;
483         *e++ = (clientarch >> 8) & 0xff;
484         *e++ = clientarch & 0xff;
485
486         *e++ = 94;      /* Client Network Interface Identifier */
487         *e++ = 3;
488         *e++ = 1;       /* type field for UNDI */
489         *e++ = 0;       /* major revision */
490         *e++ = 0;       /* minor revision */
491
492         uuid = getenv("pxeuuid");
493
494         if (uuid) {
495                 if (uuid_str_valid(uuid)) {
496                         *e++ = 97;      /* Client Machine Identifier */
497                         *e++ = 17;
498                         *e++ = 0;       /* type 0 - UUID */
499
500                         uuid_str_to_bin(uuid, e, UUID_STR_FORMAT_STD);
501                         e += 16;
502                 } else {
503                         printf("Invalid pxeuuid: %s\n", uuid);
504                 }
505         }
506 #endif
507
508 #if defined(CONFIG_SPL_BUILD) && defined(CONFIG_SPL_NET_VCI_STRING)
509         put_vci(e, CONFIG_SPL_NET_VCI_STRING);
510 #elif defined(CONFIG_BOOTP_VCI_STRING)
511         put_vci(e, CONFIG_BOOTP_VCI_STRING);
512 #endif
513
514 #if defined(CONFIG_BOOTP_VENDOREX)
515         x = dhcp_vendorex_prep(e);
516         if (x)
517                 return x - start;
518 #endif
519
520         *e++ = 55;              /* Parameter Request List */
521          cnt = e++;             /* Pointer to count of requested items */
522         *cnt = 0;
523 #if defined(CONFIG_BOOTP_SUBNETMASK)
524         *e++  = 1;              /* Subnet Mask */
525         *cnt += 1;
526 #endif
527 #if defined(CONFIG_BOOTP_TIMEOFFSET)
528         *e++  = 2;
529         *cnt += 1;
530 #endif
531 #if defined(CONFIG_BOOTP_GATEWAY)
532         *e++  = 3;              /* Router Option */
533         *cnt += 1;
534 #endif
535 #if defined(CONFIG_BOOTP_DNS)
536         *e++  = 6;              /* DNS Server(s) */
537         *cnt += 1;
538 #endif
539 #if defined(CONFIG_BOOTP_HOSTNAME)
540         *e++  = 12;             /* Hostname */
541         *cnt += 1;
542 #endif
543 #if defined(CONFIG_BOOTP_BOOTFILESIZE)
544         *e++  = 13;             /* Boot File Size */
545         *cnt += 1;
546 #endif
547 #if defined(CONFIG_BOOTP_BOOTPATH)
548         *e++  = 17;             /* Boot path */
549         *cnt += 1;
550 #endif
551 #if defined(CONFIG_BOOTP_NISDOMAIN)
552         *e++  = 40;             /* NIS Domain name request */
553         *cnt += 1;
554 #endif
555 #if defined(CONFIG_BOOTP_NTPSERVER)
556         *e++  = 42;
557         *cnt += 1;
558 #endif
559         /* no options, so back up to avoid sending an empty request list */
560         if (*cnt == 0)
561                 e -= 2;
562
563         *e++  = 255;            /* End of the list */
564
565         /* Pad to minimal length */
566 #ifdef  CONFIG_DHCP_MIN_EXT_LEN
567         while ((e - start) < CONFIG_DHCP_MIN_EXT_LEN)
568                 *e++ = 0;
569 #endif
570
571         return e - start;
572 }
573
574 #else
575 /*
576  * Warning: no field size check - change CONFIG_BOOTP_* at your own risk!
577  */
578 static int bootp_extended(u8 *e)
579 {
580         u8 *start = e;
581
582         *e++ = 99;              /* RFC1048 Magic Cookie */
583         *e++ = 130;
584         *e++ = 83;
585         *e++ = 99;
586
587 #if defined(CONFIG_CMD_DHCP)
588         *e++ = 53;              /* DHCP Message Type */
589         *e++ = 1;
590         *e++ = DHCP_DISCOVER;
591
592         *e++ = 57;              /* Maximum DHCP Message Size */
593         *e++ = 2;
594         *e++ = (576 - 312 + OPT_FIELD_SIZE) >> 16;
595         *e++ = (576 - 312 + OPT_FIELD_SIZE) & 0xff;
596 #endif
597
598 #if defined(CONFIG_BOOTP_VCI_STRING) || \
599         (defined(CONFIG_SPL_BUILD) && defined(CONFIG_SPL_NET_VCI_STRING))
600 #ifdef CONFIG_SPL_BUILD
601         put_vci(e, CONFIG_SPL_NET_VCI_STRING);
602 #else
603         put_vci(e, CONFIG_BOOTP_VCI_STRING);
604 #endif
605 #endif
606
607 #if defined(CONFIG_BOOTP_SUBNETMASK)
608         *e++ = 1;               /* Subnet mask request */
609         *e++ = 4;
610         e   += 4;
611 #endif
612
613 #if defined(CONFIG_BOOTP_GATEWAY)
614         *e++ = 3;               /* Default gateway request */
615         *e++ = 4;
616         e   += 4;
617 #endif
618
619 #if defined(CONFIG_BOOTP_DNS)
620         *e++ = 6;               /* Domain Name Server */
621         *e++ = 4;
622         e   += 4;
623 #endif
624
625 #if defined(CONFIG_BOOTP_HOSTNAME)
626         *e++ = 12;              /* Host name request */
627         *e++ = 32;
628         e   += 32;
629 #endif
630
631 #if defined(CONFIG_BOOTP_BOOTFILESIZE)
632         *e++ = 13;              /* Boot file size */
633         *e++ = 2;
634         e   += 2;
635 #endif
636
637 #if defined(CONFIG_BOOTP_BOOTPATH)
638         *e++ = 17;              /* Boot path */
639         *e++ = 32;
640         e   += 32;
641 #endif
642
643 #if defined(CONFIG_BOOTP_NISDOMAIN)
644         *e++ = 40;              /* NIS Domain name request */
645         *e++ = 32;
646         e   += 32;
647 #endif
648 #if defined(CONFIG_BOOTP_NTPSERVER)
649         *e++ = 42;
650         *e++ = 4;
651         e   += 4;
652 #endif
653
654         *e++ = 255;             /* End of the list */
655
656         return e - start;
657 }
658 #endif
659
660 void bootp_reset(void)
661 {
662         bootp_num_ids = 0;
663         bootp_try = 0;
664         bootp_start = get_timer(0);
665         bootp_timeout = 250;
666 }
667
668 void bootp_request(void)
669 {
670         uchar *pkt, *iphdr;
671         struct bootp_hdr *bp;
672         int extlen, pktlen, iplen;
673         int eth_hdr_size;
674 #ifdef CONFIG_BOOTP_RANDOM_DELAY
675         ulong rand_ms;
676 #endif
677         u32 bootp_id;
678         struct in_addr zero_ip;
679         struct in_addr bcast_ip;
680         char *ep;  /* Environment pointer */
681
682         bootstage_mark_name(BOOTSTAGE_ID_BOOTP_START, "bootp_start");
683 #if defined(CONFIG_CMD_DHCP)
684         dhcp_state = INIT;
685 #endif
686
687         ep = getenv("bootpretryperiod");
688         if (ep != NULL)
689                 time_taken_max = simple_strtoul(ep, NULL, 10);
690         else
691                 time_taken_max = TIMEOUT_MS;
692
693 #ifdef CONFIG_BOOTP_RANDOM_DELAY                /* Random BOOTP delay */
694         if (bootp_try == 0)
695                 srand_mac();
696
697         if (bootp_try <= 2)     /* Start with max 1024 * 1ms */
698                 rand_ms = rand() >> (22 - bootp_try);
699         else            /* After 3rd BOOTP request max 8192 * 1ms */
700                 rand_ms = rand() >> 19;
701
702         printf("Random delay: %ld ms...\n", rand_ms);
703         mdelay(rand_ms);
704
705 #endif  /* CONFIG_BOOTP_RANDOM_DELAY */
706
707         printf("BOOTP broadcast %d\n", ++bootp_try);
708         pkt = net_tx_packet;
709         memset((void *)pkt, 0, PKTSIZE);
710
711         eth_hdr_size = net_set_ether(pkt, net_bcast_ethaddr, PROT_IP);
712         pkt += eth_hdr_size;
713
714         /*
715          * Next line results in incorrect packet size being transmitted,
716          * resulting in errors in some DHCP servers, reporting missing bytes.
717          * Size must be set in packet header after extension length has been
718          * determined.
719          * C. Hallinan, DS4.COM, Inc.
720          */
721         /* net_set_udp_header(pkt, 0xFFFFFFFFL, PORT_BOOTPS, PORT_BOOTPC,
722                 sizeof (struct bootp_hdr)); */
723         iphdr = pkt;    /* We need this later for net_set_udp_header() */
724         pkt += IP_UDP_HDR_SIZE;
725
726         bp = (struct bootp_hdr *)pkt;
727         bp->bp_op = OP_BOOTREQUEST;
728         bp->bp_htype = HWT_ETHER;
729         bp->bp_hlen = HWL_ETHER;
730         bp->bp_hops = 0;
731         /*
732          * according to RFC1542, should be 0 on first request, secs since
733          * first request otherwise
734          */
735         bp->bp_secs = htons(get_timer(bootp_start) / 1000);
736         zero_ip.s_addr = 0;
737         net_write_ip(&bp->bp_ciaddr, zero_ip);
738         net_write_ip(&bp->bp_yiaddr, zero_ip);
739         net_write_ip(&bp->bp_siaddr, zero_ip);
740         net_write_ip(&bp->bp_giaddr, zero_ip);
741         memcpy(bp->bp_chaddr, net_ethaddr, 6);
742         copy_filename(bp->bp_file, net_boot_file_name, sizeof(bp->bp_file));
743
744         /* Request additional information from the BOOTP/DHCP server */
745 #if defined(CONFIG_CMD_DHCP)
746         extlen = dhcp_extended((u8 *)bp->bp_vend, DHCP_DISCOVER, zero_ip,
747                                zero_ip);
748 #else
749         extlen = bootp_extended((u8 *)bp->bp_vend);
750 #endif
751
752         /*
753          *      Bootp ID is the lower 4 bytes of our ethernet address
754          *      plus the current time in ms.
755          */
756         bootp_id = ((u32)net_ethaddr[2] << 24)
757                 | ((u32)net_ethaddr[3] << 16)
758                 | ((u32)net_ethaddr[4] << 8)
759                 | (u32)net_ethaddr[5];
760         bootp_id += get_timer(0);
761         bootp_id = htonl(bootp_id);
762         bootp_add_id(bootp_id);
763         net_copy_u32(&bp->bp_id, &bootp_id);
764
765         /*
766          * Calculate proper packet lengths taking into account the
767          * variable size of the options field
768          */
769         iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen;
770         pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen;
771         bcast_ip.s_addr = 0xFFFFFFFFL;
772         net_set_udp_header(iphdr, bcast_ip, PORT_BOOTPS, PORT_BOOTPC, iplen);
773         net_set_timeout_handler(bootp_timeout, bootp_timeout_handler);
774
775 #if defined(CONFIG_CMD_DHCP)
776         dhcp_state = SELECTING;
777         net_set_udp_handler(dhcp_handler);
778 #else
779         net_set_udp_handler(bootp_handler);
780 #endif
781         net_send_packet(net_tx_packet, pktlen);
782 }
783
784 #if defined(CONFIG_CMD_DHCP)
785 static void dhcp_process_options(uchar *popt, uchar *end)
786 {
787         int oplen, size;
788 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET)
789         int *to_ptr;
790 #endif
791
792         while (popt < end && *popt != 0xff) {
793                 oplen = *(popt + 1);
794                 switch (*popt) {
795                 case 0:
796                         oplen = -1; /* Pad omits len byte */
797                         break;
798                 case 1:
799                         net_copy_ip(&net_netmask, (popt + 2));
800                         break;
801 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET)
802                 case 2:         /* Time offset  */
803                         to_ptr = &net_ntp_time_offset;
804                         net_copy_u32((u32 *)to_ptr, (u32 *)(popt + 2));
805                         net_ntp_time_offset = ntohl(net_ntp_time_offset);
806                         break;
807 #endif
808                 case 3:
809                         net_copy_ip(&net_gateway, (popt + 2));
810                         break;
811                 case 6:
812                         net_copy_ip(&net_dns_server, (popt + 2));
813 #if defined(CONFIG_BOOTP_DNS2)
814                         if (*(popt + 1) > 4)
815                                 net_copy_ip(&net_dns_server2, (popt + 2 + 4));
816 #endif
817                         break;
818                 case 12:
819                         size = truncate_sz("Host Name",
820                                 sizeof(net_hostname), oplen);
821                         memcpy(&net_hostname, popt + 2, size);
822                         net_hostname[size] = 0;
823                         break;
824                 case 15:        /* Ignore Domain Name Option */
825                         break;
826                 case 17:
827                         size = truncate_sz("Root Path",
828                                 sizeof(net_root_path), oplen);
829                         memcpy(&net_root_path, popt + 2, size);
830                         net_root_path[size] = 0;
831                         break;
832                 case 28:        /* Ignore Broadcast Address Option */
833                         break;
834 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_NTPSERVER)
835                 case 42:        /* NTP server IP */
836                         net_copy_ip(&net_ntp_server, (popt + 2));
837                         break;
838 #endif
839                 case 51:
840                         net_copy_u32(&dhcp_leasetime, (u32 *)(popt + 2));
841                         break;
842                 case 52:
843                         dhcp_option_overload = popt[2];
844                         break;
845                 case 53:        /* Ignore Message Type Option */
846                         break;
847                 case 54:
848                         net_copy_ip(&dhcp_server_ip, (popt + 2));
849                         break;
850                 case 58:        /* Ignore Renewal Time Option */
851                         break;
852                 case 59:        /* Ignore Rebinding Time Option */
853                         break;
854                 case 66:        /* Ignore TFTP server name */
855                         break;
856                 case 67:        /* Bootfile option */
857                         size = truncate_sz("Bootfile",
858                                            sizeof(net_boot_file_name), oplen);
859                         memcpy(&net_boot_file_name, popt + 2, size);
860                         net_boot_file_name[size] = 0;
861                         break;
862                 default:
863 #if defined(CONFIG_BOOTP_VENDOREX)
864                         if (dhcp_vendorex_proc(popt))
865                                 break;
866 #endif
867                         printf("*** Unhandled DHCP Option in OFFER/ACK:"
868                                " %d\n", *popt);
869                         break;
870                 }
871                 popt += oplen + 2;      /* Process next option */
872         }
873 }
874
875 static void dhcp_packet_process_options(struct bootp_hdr *bp)
876 {
877         uchar *popt = (uchar *)&bp->bp_vend[4];
878         uchar *end = popt + BOOTP_HDR_SIZE;
879
880         if (net_read_u32((u32 *)&bp->bp_vend[0]) != htonl(BOOTP_VENDOR_MAGIC))
881                 return;
882
883         dhcp_option_overload = 0;
884
885         /*
886          * The 'options' field MUST be interpreted first, 'file' next,
887          * 'sname' last.
888          */
889         dhcp_process_options(popt, end);
890
891         if (dhcp_option_overload & OVERLOAD_FILE) {
892                 popt = (uchar *)bp->bp_file;
893                 end = popt + sizeof(bp->bp_file);
894                 dhcp_process_options(popt, end);
895         }
896
897         if (dhcp_option_overload & OVERLOAD_SNAME) {
898                 popt = (uchar *)bp->bp_sname;
899                 end = popt + sizeof(bp->bp_sname);
900                 dhcp_process_options(popt, end);
901         }
902 }
903
904 static int dhcp_message_type(unsigned char *popt)
905 {
906         if (net_read_u32((u32 *)popt) != htonl(BOOTP_VENDOR_MAGIC))
907                 return -1;
908
909         popt += 4;
910         while (*popt != 0xff) {
911                 if (*popt == 53)        /* DHCP Message Type */
912                         return *(popt + 2);
913                 if (*popt == 0) {
914                         /* Pad */
915                         popt += 1;
916                 } else {
917                         /* Scan through all options */
918                         popt += *(popt + 1) + 2;
919                 }
920         }
921         return -1;
922 }
923
924 static void dhcp_send_request_packet(struct bootp_hdr *bp_offer)
925 {
926         uchar *pkt, *iphdr;
927         struct bootp_hdr *bp;
928         int pktlen, iplen, extlen;
929         int eth_hdr_size;
930         struct in_addr offered_ip;
931         struct in_addr zero_ip;
932         struct in_addr bcast_ip;
933
934         debug("dhcp_send_request_packet: Sending DHCPREQUEST\n");
935         pkt = net_tx_packet;
936         memset((void *)pkt, 0, PKTSIZE);
937
938         eth_hdr_size = net_set_ether(pkt, net_bcast_ethaddr, PROT_IP);
939         pkt += eth_hdr_size;
940
941         iphdr = pkt;    /* We'll need this later to set proper pkt size */
942         pkt += IP_UDP_HDR_SIZE;
943
944         bp = (struct bootp_hdr *)pkt;
945         bp->bp_op = OP_BOOTREQUEST;
946         bp->bp_htype = HWT_ETHER;
947         bp->bp_hlen = HWL_ETHER;
948         bp->bp_hops = 0;
949         bp->bp_secs = htons(get_timer(bootp_start) / 1000);
950         /* Do not set the client IP, your IP, or server IP yet, since it
951          * hasn't been ACK'ed by the server yet */
952
953         /*
954          * RFC3046 requires Relay Agents to discard packets with
955          * nonzero and offered giaddr
956          */
957         zero_ip.s_addr = 0;
958         net_write_ip(&bp->bp_giaddr, zero_ip);
959
960         memcpy(bp->bp_chaddr, net_ethaddr, 6);
961         copy_filename(bp->bp_file, net_boot_file_name, sizeof(bp->bp_file));
962
963         /*
964          * ID is the id of the OFFER packet
965          */
966
967         net_copy_u32(&bp->bp_id, &bp_offer->bp_id);
968
969         /*
970          * Copy options from OFFER packet if present
971          */
972
973         /* Copy offered IP into the parameters request list */
974         net_copy_ip(&offered_ip, &bp_offer->bp_yiaddr);
975         extlen = dhcp_extended((u8 *)bp->bp_vend, DHCP_REQUEST,
976                 dhcp_server_ip, offered_ip);
977
978         iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen;
979         pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen;
980         bcast_ip.s_addr = 0xFFFFFFFFL;
981         net_set_udp_header(iphdr, bcast_ip, PORT_BOOTPS, PORT_BOOTPC, iplen);
982
983 #ifdef CONFIG_BOOTP_DHCP_REQUEST_DELAY
984         udelay(CONFIG_BOOTP_DHCP_REQUEST_DELAY);
985 #endif  /* CONFIG_BOOTP_DHCP_REQUEST_DELAY */
986         debug("Transmitting DHCPREQUEST packet: len = %d\n", pktlen);
987         net_send_packet(net_tx_packet, pktlen);
988 }
989
990 /*
991  *      Handle DHCP received packets.
992  */
993 static void dhcp_handler(uchar *pkt, unsigned dest, struct in_addr sip,
994                          unsigned src, unsigned len)
995 {
996         struct bootp_hdr *bp = (struct bootp_hdr *)pkt;
997
998         debug("DHCPHandler: got packet: (src=%d, dst=%d, len=%d) state: %d\n",
999               src, dest, len, dhcp_state);
1000
1001         /* Filter out pkts we don't want */
1002         if (check_reply_packet(pkt, dest, src, len))
1003                 return;
1004
1005         debug("DHCPHandler: got DHCP packet: (src=%d, dst=%d, len=%d) state: "
1006               "%d\n", src, dest, len, dhcp_state);
1007
1008         if (net_read_ip(&bp->bp_yiaddr).s_addr == 0)
1009                 return;
1010
1011         switch (dhcp_state) {
1012         case SELECTING:
1013                 /*
1014                  * Wait an appropriate time for any potential DHCPOFFER packets
1015                  * to arrive.  Then select one, and generate DHCPREQUEST
1016                  * response.  If filename is in format we recognize, assume it
1017                  * is a valid OFFER from a server we want.
1018                  */
1019                 debug("DHCP: state=SELECTING bp_file: \"%s\"\n", bp->bp_file);
1020 #ifdef CONFIG_SYS_BOOTFILE_PREFIX
1021                 if (strncmp(bp->bp_file,
1022                             CONFIG_SYS_BOOTFILE_PREFIX,
1023                             strlen(CONFIG_SYS_BOOTFILE_PREFIX)) == 0) {
1024 #endif  /* CONFIG_SYS_BOOTFILE_PREFIX */
1025                         dhcp_packet_process_options(bp);
1026
1027                         debug("TRANSITIONING TO REQUESTING STATE\n");
1028                         dhcp_state = REQUESTING;
1029
1030                         net_set_timeout_handler(5000, bootp_timeout_handler);
1031                         dhcp_send_request_packet(bp);
1032 #ifdef CONFIG_SYS_BOOTFILE_PREFIX
1033                 }
1034 #endif  /* CONFIG_SYS_BOOTFILE_PREFIX */
1035
1036                 return;
1037                 break;
1038         case REQUESTING:
1039                 debug("DHCP State: REQUESTING\n");
1040
1041                 if (dhcp_message_type((u8 *)bp->bp_vend) == DHCP_ACK) {
1042                         dhcp_packet_process_options(bp);
1043                         /* Store net params from reply */
1044                         store_net_params(bp);
1045                         dhcp_state = BOUND;
1046                         printf("DHCP client bound to address %pI4 (%lu ms)\n",
1047                                &net_ip, get_timer(bootp_start));
1048                         net_set_timeout_handler(0, (thand_f *)0);
1049                         bootstage_mark_name(BOOTSTAGE_ID_BOOTP_STOP,
1050                                             "bootp_stop");
1051
1052                         net_auto_load();
1053                         return;
1054                 }
1055                 break;
1056         case BOUND:
1057                 /* DHCP client bound to address */
1058                 break;
1059         default:
1060                 puts("DHCP: INVALID STATE\n");
1061                 break;
1062         }
1063 }
1064
1065 void dhcp_request(void)
1066 {
1067         bootp_request();
1068 }
1069 #endif  /* CONFIG_CMD_DHCP */