xref: /rk3399_rockchip-uboot/net/bootp.c (revision 4f28c9b169eb3dfa0981f58f9c649a03be39fda0)
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 	/*
715 	 * according to RFC1542, should be 0 on first request, secs since
716 	 * first request otherwise
717 	 */
718 	bp->bp_secs = htons(get_timer(bootp_start) / 1000);
719 	zero_ip.s_addr = 0;
720 	net_write_ip(&bp->bp_ciaddr, zero_ip);
721 	net_write_ip(&bp->bp_yiaddr, zero_ip);
722 	net_write_ip(&bp->bp_siaddr, zero_ip);
723 	net_write_ip(&bp->bp_giaddr, zero_ip);
724 	memcpy(bp->bp_chaddr, net_ethaddr, 6);
725 	copy_filename(bp->bp_file, net_boot_file_name, sizeof(bp->bp_file));
726 
727 	/* Request additional information from the BOOTP/DHCP server */
728 #if defined(CONFIG_CMD_DHCP)
729 	extlen = dhcp_extended((u8 *)bp->bp_vend, DHCP_DISCOVER, zero_ip,
730 			       zero_ip);
731 #else
732 	extlen = bootp_extended((u8 *)bp->bp_vend);
733 #endif
734 
735 	/*
736 	 *	Bootp ID is the lower 4 bytes of our ethernet address
737 	 *	plus the current time in ms.
738 	 */
739 	bootp_id = ((u32)net_ethaddr[2] << 24)
740 		| ((u32)net_ethaddr[3] << 16)
741 		| ((u32)net_ethaddr[4] << 8)
742 		| (u32)net_ethaddr[5];
743 	bootp_id += get_timer(0);
744 	bootp_id = htonl(bootp_id);
745 	bootp_add_id(bootp_id);
746 	net_copy_u32(&bp->bp_id, &bootp_id);
747 
748 	/*
749 	 * Calculate proper packet lengths taking into account the
750 	 * variable size of the options field
751 	 */
752 	iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen;
753 	pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen;
754 	bcast_ip.s_addr = 0xFFFFFFFFL;
755 	net_set_udp_header(iphdr, bcast_ip, PORT_BOOTPS, PORT_BOOTPC, iplen);
756 	net_set_timeout_handler(bootp_timeout, bootp_timeout_handler);
757 
758 #if defined(CONFIG_CMD_DHCP)
759 	dhcp_state = SELECTING;
760 	net_set_udp_handler(dhcp_handler);
761 #else
762 	net_set_udp_handler(bootp_handler);
763 #endif
764 	net_send_packet(net_tx_packet, pktlen);
765 }
766 
767 #if defined(CONFIG_CMD_DHCP)
768 static void dhcp_process_options(uchar *popt, struct bootp_hdr *bp)
769 {
770 	uchar *end = popt + BOOTP_HDR_SIZE;
771 	int oplen, size;
772 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET)
773 	int *to_ptr;
774 #endif
775 
776 	while (popt < end && *popt != 0xff) {
777 		oplen = *(popt + 1);
778 		switch (*popt) {
779 		case 0:
780 			oplen = -1; /* Pad omits len byte */
781 			break;
782 		case 1:
783 			net_copy_ip(&net_netmask, (popt + 2));
784 			break;
785 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET)
786 		case 2:		/* Time offset	*/
787 			to_ptr = &net_ntp_time_offset;
788 			net_copy_u32((u32 *)to_ptr, (u32 *)(popt + 2));
789 			net_ntp_time_offset = ntohl(net_ntp_time_offset);
790 			break;
791 #endif
792 		case 3:
793 			net_copy_ip(&net_gateway, (popt + 2));
794 			break;
795 		case 6:
796 			net_copy_ip(&net_dns_server, (popt + 2));
797 #if defined(CONFIG_BOOTP_DNS2)
798 			if (*(popt + 1) > 4)
799 				net_copy_ip(&net_dns_server2, (popt + 2 + 4));
800 #endif
801 			break;
802 		case 12:
803 			size = truncate_sz("Host Name",
804 				sizeof(net_hostname), oplen);
805 			memcpy(&net_hostname, popt + 2, size);
806 			net_hostname[size] = 0;
807 			break;
808 		case 15:	/* Ignore Domain Name Option */
809 			break;
810 		case 17:
811 			size = truncate_sz("Root Path",
812 				sizeof(net_root_path), oplen);
813 			memcpy(&net_root_path, popt + 2, size);
814 			net_root_path[size] = 0;
815 			break;
816 		case 28:	/* Ignore Broadcast Address Option */
817 			break;
818 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_NTPSERVER)
819 		case 42:	/* NTP server IP */
820 			net_copy_ip(&net_ntp_server, (popt + 2));
821 			break;
822 #endif
823 		case 51:
824 			net_copy_u32(&dhcp_leasetime, (u32 *)(popt + 2));
825 			break;
826 		case 53:	/* Ignore Message Type Option */
827 			break;
828 		case 54:
829 			net_copy_ip(&dhcp_server_ip, (popt + 2));
830 			break;
831 		case 58:	/* Ignore Renewal Time Option */
832 			break;
833 		case 59:	/* Ignore Rebinding Time Option */
834 			break;
835 		case 66:	/* Ignore TFTP server name */
836 			break;
837 		case 67:	/* vendor opt bootfile */
838 			/*
839 			 * I can't use dhcp_vendorex_proc here because I need
840 			 * to write into the bootp packet - even then I had to
841 			 * pass the bootp packet pointer into here as the
842 			 * second arg
843 			 */
844 			size = truncate_sz("Opt Boot File",
845 					    sizeof(bp->bp_file),
846 					    oplen);
847 			if (bp->bp_file[0] == '\0' && size > 0) {
848 				/*
849 				 * only use vendor boot file if we didn't
850 				 * receive a boot file in the main non-vendor
851 				 * part of the packet - god only knows why
852 				 * some vendors chose not to use this perfectly
853 				 * good spot to store the boot file (join on
854 				 * Tru64 Unix) it seems mind bogglingly crazy
855 				 * to me
856 				 */
857 				printf("*** WARNING: using vendor "
858 				       "optional boot file\n");
859 				memcpy(bp->bp_file, popt + 2, size);
860 				bp->bp_file[size] = '\0';
861 			}
862 			break;
863 		default:
864 #if defined(CONFIG_BOOTP_VENDOREX)
865 			if (dhcp_vendorex_proc(popt))
866 				break;
867 #endif
868 			printf("*** Unhandled DHCP Option in OFFER/ACK:"
869 			       " %d\n", *popt);
870 			break;
871 		}
872 		popt += oplen + 2;	/* Process next option */
873 	}
874 }
875 
876 static int dhcp_message_type(unsigned char *popt)
877 {
878 	if (net_read_u32((u32 *)popt) != htonl(BOOTP_VENDOR_MAGIC))
879 		return -1;
880 
881 	popt += 4;
882 	while (*popt != 0xff) {
883 		if (*popt == 53)	/* DHCP Message Type */
884 			return *(popt + 2);
885 		if (*popt == 0)	{
886 			/* Pad */
887 			popt += 1;
888 		} else {
889 			/* Scan through all options */
890 			popt += *(popt + 1) + 2;
891 		}
892 	}
893 	return -1;
894 }
895 
896 static void dhcp_send_request_packet(struct bootp_hdr *bp_offer)
897 {
898 	uchar *pkt, *iphdr;
899 	struct bootp_hdr *bp;
900 	int pktlen, iplen, extlen;
901 	int eth_hdr_size;
902 	struct in_addr offered_ip;
903 	struct in_addr zero_ip;
904 	struct in_addr bcast_ip;
905 
906 	debug("dhcp_send_request_packet: Sending DHCPREQUEST\n");
907 	pkt = net_tx_packet;
908 	memset((void *)pkt, 0, PKTSIZE);
909 
910 	eth_hdr_size = net_set_ether(pkt, net_bcast_ethaddr, PROT_IP);
911 	pkt += eth_hdr_size;
912 
913 	iphdr = pkt;	/* We'll need this later to set proper pkt size */
914 	pkt += IP_UDP_HDR_SIZE;
915 
916 	bp = (struct bootp_hdr *)pkt;
917 	bp->bp_op = OP_BOOTREQUEST;
918 	bp->bp_htype = HWT_ETHER;
919 	bp->bp_hlen = HWL_ETHER;
920 	bp->bp_hops = 0;
921 	bp->bp_secs = htons(get_timer(bootp_start) / 1000);
922 	/* Do not set the client IP, your IP, or server IP yet, since it
923 	 * hasn't been ACK'ed by the server yet */
924 
925 	/*
926 	 * RFC3046 requires Relay Agents to discard packets with
927 	 * nonzero and offered giaddr
928 	 */
929 	zero_ip.s_addr = 0;
930 	net_write_ip(&bp->bp_giaddr, zero_ip);
931 
932 	memcpy(bp->bp_chaddr, net_ethaddr, 6);
933 
934 	/*
935 	 * ID is the id of the OFFER packet
936 	 */
937 
938 	net_copy_u32(&bp->bp_id, &bp_offer->bp_id);
939 
940 	/*
941 	 * Copy options from OFFER packet if present
942 	 */
943 
944 	/* Copy offered IP into the parameters request list */
945 	net_copy_ip(&offered_ip, &bp_offer->bp_yiaddr);
946 	extlen = dhcp_extended((u8 *)bp->bp_vend, DHCP_REQUEST,
947 		dhcp_server_ip, offered_ip);
948 
949 	iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen;
950 	pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen;
951 	bcast_ip.s_addr = 0xFFFFFFFFL;
952 	net_set_udp_header(iphdr, bcast_ip, PORT_BOOTPS, PORT_BOOTPC, iplen);
953 
954 #ifdef CONFIG_BOOTP_DHCP_REQUEST_DELAY
955 	udelay(CONFIG_BOOTP_DHCP_REQUEST_DELAY);
956 #endif	/* CONFIG_BOOTP_DHCP_REQUEST_DELAY */
957 	debug("Transmitting DHCPREQUEST packet: len = %d\n", pktlen);
958 	net_send_packet(net_tx_packet, pktlen);
959 }
960 
961 /*
962  *	Handle DHCP received packets.
963  */
964 static void dhcp_handler(uchar *pkt, unsigned dest, struct in_addr sip,
965 			 unsigned src, unsigned len)
966 {
967 	struct bootp_hdr *bp = (struct bootp_hdr *)pkt;
968 
969 	debug("DHCPHandler: got packet: (src=%d, dst=%d, len=%d) state: %d\n",
970 	      src, dest, len, dhcp_state);
971 
972 	/* Filter out pkts we don't want */
973 	if (check_reply_packet(pkt, dest, src, len))
974 		return;
975 
976 	debug("DHCPHandler: got DHCP packet: (src=%d, dst=%d, len=%d) state: "
977 	      "%d\n", src, dest, len, dhcp_state);
978 
979 	switch (dhcp_state) {
980 	case SELECTING:
981 		/*
982 		 * Wait an appropriate time for any potential DHCPOFFER packets
983 		 * to arrive.  Then select one, and generate DHCPREQUEST
984 		 * response.  If filename is in format we recognize, assume it
985 		 * is a valid OFFER from a server we want.
986 		 */
987 		debug("DHCP: state=SELECTING bp_file: \"%s\"\n", bp->bp_file);
988 #ifdef CONFIG_SYS_BOOTFILE_PREFIX
989 		if (strncmp(bp->bp_file,
990 			    CONFIG_SYS_BOOTFILE_PREFIX,
991 			    strlen(CONFIG_SYS_BOOTFILE_PREFIX)) == 0) {
992 #endif	/* CONFIG_SYS_BOOTFILE_PREFIX */
993 
994 			debug("TRANSITIONING TO REQUESTING STATE\n");
995 			dhcp_state = REQUESTING;
996 
997 			if (net_read_u32((u32 *)&bp->bp_vend[0]) ==
998 						htonl(BOOTP_VENDOR_MAGIC))
999 				dhcp_process_options((u8 *)&bp->bp_vend[4], bp);
1000 
1001 			net_set_timeout_handler(5000, bootp_timeout_handler);
1002 			dhcp_send_request_packet(bp);
1003 #ifdef CONFIG_SYS_BOOTFILE_PREFIX
1004 		}
1005 #endif	/* CONFIG_SYS_BOOTFILE_PREFIX */
1006 
1007 		return;
1008 		break;
1009 	case REQUESTING:
1010 		debug("DHCP State: REQUESTING\n");
1011 
1012 		if (dhcp_message_type((u8 *)bp->bp_vend) == DHCP_ACK) {
1013 			if (net_read_u32((u32 *)&bp->bp_vend[0]) ==
1014 						htonl(BOOTP_VENDOR_MAGIC))
1015 				dhcp_process_options((u8 *)&bp->bp_vend[4], bp);
1016 			/* Store net params from reply */
1017 			store_net_params(bp);
1018 			dhcp_state = BOUND;
1019 			printf("DHCP client bound to address %pI4 (%lu ms)\n",
1020 			       &net_ip, get_timer(bootp_start));
1021 			net_set_timeout_handler(0, (thand_f *)0);
1022 			bootstage_mark_name(BOOTSTAGE_ID_BOOTP_STOP,
1023 					    "bootp_stop");
1024 
1025 			net_auto_load();
1026 			return;
1027 		}
1028 		break;
1029 	case BOUND:
1030 		/* DHCP client bound to address */
1031 		break;
1032 	default:
1033 		puts("DHCP: INVALID STATE\n");
1034 		break;
1035 	}
1036 }
1037 
1038 void dhcp_request(void)
1039 {
1040 	bootp_request();
1041 }
1042 #endif	/* CONFIG_CMD_DHCP */
1043