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