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