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