xref: /openbmc/u-boot/net/bootp.c (revision 7dfb0602)
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 dhcp_state_t dhcp_state = INIT;
45 unsigned long dhcp_leasetime;
46 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 		NetSetState(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 /*
345  *	Initialize BOOTP extension fields in the request.
346  */
347 #if defined(CONFIG_CMD_DHCP)
348 static int DhcpExtended(u8 *e, int message_type, IPaddr_t ServerID,
349 			IPaddr_t RequestedIP)
350 {
351 	u8 *start = e;
352 	u8 *cnt;
353 #if defined(CONFIG_BOOTP_PXE)
354 	char *uuid;
355 	size_t vci_strlen;
356 	u16 clientarch;
357 #endif
358 
359 #if defined(CONFIG_BOOTP_VENDOREX)
360 	u8 *x;
361 #endif
362 #if defined(CONFIG_BOOTP_SEND_HOSTNAME)
363 	char *hostname;
364 #endif
365 
366 	*e++ = 99;		/* RFC1048 Magic Cookie */
367 	*e++ = 130;
368 	*e++ = 83;
369 	*e++ = 99;
370 
371 	*e++ = 53;		/* DHCP Message Type */
372 	*e++ = 1;
373 	*e++ = message_type;
374 
375 	*e++ = 57;		/* Maximum DHCP Message Size */
376 	*e++ = 2;
377 	*e++ = (576 - 312 + OPT_FIELD_SIZE) >> 8;
378 	*e++ = (576 - 312 + OPT_FIELD_SIZE) & 0xff;
379 
380 	if (ServerID) {
381 		int tmp = ntohl(ServerID);
382 
383 		*e++ = 54;	/* ServerID */
384 		*e++ = 4;
385 		*e++ = tmp >> 24;
386 		*e++ = tmp >> 16;
387 		*e++ = tmp >> 8;
388 		*e++ = tmp & 0xff;
389 	}
390 
391 	if (RequestedIP) {
392 		int tmp = ntohl(RequestedIP);
393 
394 		*e++ = 50;	/* Requested IP */
395 		*e++ = 4;
396 		*e++ = tmp >> 24;
397 		*e++ = tmp >> 16;
398 		*e++ = tmp >> 8;
399 		*e++ = tmp & 0xff;
400 	}
401 #if defined(CONFIG_BOOTP_SEND_HOSTNAME)
402 	hostname = getenv("hostname");
403 	if (hostname) {
404 		int hostnamelen = strlen(hostname);
405 
406 		*e++ = 12;	/* Hostname */
407 		*e++ = hostnamelen;
408 		memcpy(e, hostname, hostnamelen);
409 		e += hostnamelen;
410 	}
411 #endif
412 
413 #if defined(CONFIG_BOOTP_PXE)
414 	clientarch = CONFIG_BOOTP_PXE_CLIENTARCH;
415 	*e++ = 93;	/* Client System Architecture */
416 	*e++ = 2;
417 	*e++ = (clientarch >> 8) & 0xff;
418 	*e++ = clientarch & 0xff;
419 
420 	*e++ = 94;	/* Client Network Interface Identifier */
421 	*e++ = 3;
422 	*e++ = 1;	/* type field for UNDI */
423 	*e++ = 0;	/* major revision */
424 	*e++ = 0;	/* minor revision */
425 
426 	uuid = getenv("pxeuuid");
427 
428 	if (uuid) {
429 		if (uuid_str_valid(uuid)) {
430 			*e++ = 97;	/* Client Machine Identifier */
431 			*e++ = 17;
432 			*e++ = 0;	/* type 0 - UUID */
433 
434 			uuid_str_to_bin(uuid, e);
435 			e += 16;
436 		} else {
437 			printf("Invalid pxeuuid: %s\n", uuid);
438 		}
439 	}
440 
441 	*e++ = 60;	/* Vendor Class Identifier */
442 	vci_strlen = strlen(CONFIG_BOOTP_VCI_STRING);
443 	*e++ = vci_strlen;
444 	memcpy(e, CONFIG_BOOTP_VCI_STRING, vci_strlen);
445 	e += vci_strlen;
446 #endif
447 
448 #if defined(CONFIG_BOOTP_VENDOREX)
449 	x = dhcp_vendorex_prep(e);
450 	if (x)
451 		return x - start;
452 #endif
453 
454 	*e++ = 55;		/* Parameter Request List */
455 	 cnt = e++;		/* Pointer to count of requested items */
456 	*cnt = 0;
457 #if defined(CONFIG_BOOTP_SUBNETMASK)
458 	*e++  = 1;		/* Subnet Mask */
459 	*cnt += 1;
460 #endif
461 #if defined(CONFIG_BOOTP_TIMEOFFSET)
462 	*e++  = 2;
463 	*cnt += 1;
464 #endif
465 #if defined(CONFIG_BOOTP_GATEWAY)
466 	*e++  = 3;		/* Router Option */
467 	*cnt += 1;
468 #endif
469 #if defined(CONFIG_BOOTP_DNS)
470 	*e++  = 6;		/* DNS Server(s) */
471 	*cnt += 1;
472 #endif
473 #if defined(CONFIG_BOOTP_HOSTNAME)
474 	*e++  = 12;		/* Hostname */
475 	*cnt += 1;
476 #endif
477 #if defined(CONFIG_BOOTP_BOOTFILESIZE)
478 	*e++  = 13;		/* Boot File Size */
479 	*cnt += 1;
480 #endif
481 #if defined(CONFIG_BOOTP_BOOTPATH)
482 	*e++  = 17;		/* Boot path */
483 	*cnt += 1;
484 #endif
485 #if defined(CONFIG_BOOTP_NISDOMAIN)
486 	*e++  = 40;		/* NIS Domain name request */
487 	*cnt += 1;
488 #endif
489 #if defined(CONFIG_BOOTP_NTPSERVER)
490 	*e++  = 42;
491 	*cnt += 1;
492 #endif
493 	/* no options, so back up to avoid sending an empty request list */
494 	if (*cnt == 0)
495 		e -= 2;
496 
497 	*e++  = 255;		/* End of the list */
498 
499 	/* Pad to minimal length */
500 #ifdef	CONFIG_DHCP_MIN_EXT_LEN
501 	while ((e - start) < CONFIG_DHCP_MIN_EXT_LEN)
502 		*e++ = 0;
503 #endif
504 
505 	return e - start;
506 }
507 
508 #else
509 /*
510  * Warning: no field size check - change CONFIG_BOOTP_* at your own risk!
511  */
512 static int BootpExtended(u8 *e)
513 {
514 	u8 *start = e;
515 
516 	*e++ = 99;		/* RFC1048 Magic Cookie */
517 	*e++ = 130;
518 	*e++ = 83;
519 	*e++ = 99;
520 
521 #if defined(CONFIG_CMD_DHCP)
522 	*e++ = 53;		/* DHCP Message Type */
523 	*e++ = 1;
524 	*e++ = DHCP_DISCOVER;
525 
526 	*e++ = 57;		/* Maximum DHCP Message Size */
527 	*e++ = 2;
528 	*e++ = (576 - 312 + OPT_FIELD_SIZE) >> 16;
529 	*e++ = (576 - 312 + OPT_FIELD_SIZE) & 0xff;
530 #endif
531 
532 #if defined(CONFIG_BOOTP_SUBNETMASK)
533 	*e++ = 1;		/* Subnet mask request */
534 	*e++ = 4;
535 	e   += 4;
536 #endif
537 
538 #if defined(CONFIG_BOOTP_GATEWAY)
539 	*e++ = 3;		/* Default gateway request */
540 	*e++ = 4;
541 	e   += 4;
542 #endif
543 
544 #if defined(CONFIG_BOOTP_DNS)
545 	*e++ = 6;		/* Domain Name Server */
546 	*e++ = 4;
547 	e   += 4;
548 #endif
549 
550 #if defined(CONFIG_BOOTP_HOSTNAME)
551 	*e++ = 12;		/* Host name request */
552 	*e++ = 32;
553 	e   += 32;
554 #endif
555 
556 #if defined(CONFIG_BOOTP_BOOTFILESIZE)
557 	*e++ = 13;		/* Boot file size */
558 	*e++ = 2;
559 	e   += 2;
560 #endif
561 
562 #if defined(CONFIG_BOOTP_BOOTPATH)
563 	*e++ = 17;		/* Boot path */
564 	*e++ = 32;
565 	e   += 32;
566 #endif
567 
568 #if defined(CONFIG_BOOTP_NISDOMAIN)
569 	*e++ = 40;		/* NIS Domain name request */
570 	*e++ = 32;
571 	e   += 32;
572 #endif
573 #if defined(CONFIG_BOOTP_NTPSERVER)
574 	*e++ = 42;
575 	*e++ = 4;
576 	e   += 4;
577 #endif
578 
579 	*e++ = 255;		/* End of the list */
580 
581 	return e - start;
582 }
583 #endif
584 
585 void
586 BootpRequest(void)
587 {
588 	uchar *pkt, *iphdr;
589 	struct Bootp_t *bp;
590 	int extlen, pktlen, iplen;
591 	int eth_hdr_size;
592 #ifdef CONFIG_BOOTP_RANDOM_DELAY
593 	ulong i, rand_ms;
594 #endif
595 
596 	bootstage_mark_name(BOOTSTAGE_ID_BOOTP_START, "bootp_start");
597 #if defined(CONFIG_CMD_DHCP)
598 	dhcp_state = INIT;
599 #endif
600 
601 #ifdef CONFIG_BOOTP_RANDOM_DELAY		/* Random BOOTP delay */
602 	if (BootpTry == 0)
603 		srand_mac();
604 
605 	if (BootpTry <= 2)	/* Start with max 1024 * 1ms */
606 		rand_ms = rand() >> (22 - BootpTry);
607 	else		/* After 3rd BOOTP request max 8192 * 1ms */
608 		rand_ms = rand() >> 19;
609 
610 	printf("Random delay: %ld ms...\n", rand_ms);
611 	for (i = 0; i < rand_ms; i++)
612 		udelay(1000); /*Wait 1ms*/
613 
614 #endif	/* CONFIG_BOOTP_RANDOM_DELAY */
615 
616 	printf("BOOTP broadcast %d\n", ++BootpTry);
617 	pkt = NetTxPacket;
618 	memset((void *)pkt, 0, PKTSIZE);
619 
620 	eth_hdr_size = NetSetEther(pkt, NetBcastAddr, PROT_IP);
621 	pkt += eth_hdr_size;
622 
623 	/*
624 	 * Next line results in incorrect packet size being transmitted,
625 	 * resulting in errors in some DHCP servers, reporting missing bytes.
626 	 * Size must be set in packet header after extension length has been
627 	 * determined.
628 	 * C. Hallinan, DS4.COM, Inc.
629 	 */
630 	/* net_set_udp_header(pkt, 0xFFFFFFFFL, PORT_BOOTPS, PORT_BOOTPC,
631 		sizeof (struct Bootp_t)); */
632 	iphdr = pkt;	/* We need this later for net_set_udp_header() */
633 	pkt += IP_UDP_HDR_SIZE;
634 
635 	bp = (struct Bootp_t *)pkt;
636 	bp->bp_op = OP_BOOTREQUEST;
637 	bp->bp_htype = HWT_ETHER;
638 	bp->bp_hlen = HWL_ETHER;
639 	bp->bp_hops = 0;
640 	bp->bp_secs = htons(get_timer(0) / 1000);
641 	NetWriteIP(&bp->bp_ciaddr, 0);
642 	NetWriteIP(&bp->bp_yiaddr, 0);
643 	NetWriteIP(&bp->bp_siaddr, 0);
644 	NetWriteIP(&bp->bp_giaddr, 0);
645 	memcpy(bp->bp_chaddr, NetOurEther, 6);
646 	copy_filename(bp->bp_file, BootFile, sizeof(bp->bp_file));
647 
648 	/* Request additional information from the BOOTP/DHCP server */
649 #if defined(CONFIG_CMD_DHCP)
650 	extlen = DhcpExtended((u8 *)bp->bp_vend, DHCP_DISCOVER, 0, 0);
651 #else
652 	extlen = BootpExtended((u8 *)bp->bp_vend);
653 #endif
654 
655 	/*
656 	 *	Bootp ID is the lower 4 bytes of our ethernet address
657 	 *	plus the current time in ms.
658 	 */
659 	BootpID = ((ulong)NetOurEther[2] << 24)
660 		| ((ulong)NetOurEther[3] << 16)
661 		| ((ulong)NetOurEther[4] << 8)
662 		| (ulong)NetOurEther[5];
663 	BootpID += get_timer(0);
664 	BootpID	 = htonl(BootpID);
665 	NetCopyLong(&bp->bp_id, &BootpID);
666 
667 	/*
668 	 * Calculate proper packet lengths taking into account the
669 	 * variable size of the options field
670 	 */
671 	iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen;
672 	pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen;
673 	net_set_udp_header(iphdr, 0xFFFFFFFFL, PORT_BOOTPS, PORT_BOOTPC, iplen);
674 	NetSetTimeout(SELECT_TIMEOUT, BootpTimeout);
675 
676 #if defined(CONFIG_CMD_DHCP)
677 	dhcp_state = SELECTING;
678 	net_set_udp_handler(DhcpHandler);
679 #else
680 	net_set_udp_handler(BootpHandler);
681 #endif
682 	NetSendPacket(NetTxPacket, pktlen);
683 }
684 
685 #if defined(CONFIG_CMD_DHCP)
686 static void DhcpOptionsProcess(uchar *popt, struct Bootp_t *bp)
687 {
688 	uchar *end = popt + BOOTP_HDR_SIZE;
689 	int oplen, size;
690 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET)
691 	int *to_ptr;
692 #endif
693 
694 	while (popt < end && *popt != 0xff) {
695 		oplen = *(popt + 1);
696 		switch (*popt) {
697 		case 1:
698 			NetCopyIP(&NetOurSubnetMask, (popt + 2));
699 			break;
700 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET)
701 		case 2:		/* Time offset	*/
702 			to_ptr = &NetTimeOffset;
703 			NetCopyLong((ulong *)to_ptr, (ulong *)(popt + 2));
704 			NetTimeOffset = ntohl(NetTimeOffset);
705 			break;
706 #endif
707 		case 3:
708 			NetCopyIP(&NetOurGatewayIP, (popt + 2));
709 			break;
710 		case 6:
711 			NetCopyIP(&NetOurDNSIP, (popt + 2));
712 #if defined(CONFIG_BOOTP_DNS2)
713 			if (*(popt + 1) > 4)
714 				NetCopyIP(&NetOurDNS2IP, (popt + 2 + 4));
715 #endif
716 			break;
717 		case 12:
718 			size = truncate_sz("Host Name",
719 				sizeof(NetOurHostName), oplen);
720 			memcpy(&NetOurHostName, popt + 2, size);
721 			NetOurHostName[size] = 0;
722 			break;
723 		case 15:	/* Ignore Domain Name Option */
724 			break;
725 		case 17:
726 			size = truncate_sz("Root Path",
727 				sizeof(NetOurRootPath), oplen);
728 			memcpy(&NetOurRootPath, popt + 2, size);
729 			NetOurRootPath[size] = 0;
730 			break;
731 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_NTPSERVER)
732 		case 42:	/* NTP server IP */
733 			NetCopyIP(&NetNtpServerIP, (popt + 2));
734 			break;
735 #endif
736 		case 51:
737 			NetCopyLong(&dhcp_leasetime, (ulong *) (popt + 2));
738 			break;
739 		case 53:	/* Ignore Message Type Option */
740 			break;
741 		case 54:
742 			NetCopyIP(&NetDHCPServerIP, (popt + 2));
743 			break;
744 		case 58:	/* Ignore Renewal Time Option */
745 			break;
746 		case 59:	/* Ignore Rebinding Time Option */
747 			break;
748 		case 66:	/* Ignore TFTP server name */
749 			break;
750 		case 67:	/* vendor opt bootfile */
751 			/*
752 			 * I can't use dhcp_vendorex_proc here because I need
753 			 * to write into the bootp packet - even then I had to
754 			 * pass the bootp packet pointer into here as the
755 			 * second arg
756 			 */
757 			size = truncate_sz("Opt Boot File",
758 					    sizeof(bp->bp_file),
759 					    oplen);
760 			if (bp->bp_file[0] == '\0' && size > 0) {
761 				/*
762 				 * only use vendor boot file if we didn't
763 				 * receive a boot file in the main non-vendor
764 				 * part of the packet - god only knows why
765 				 * some vendors chose not to use this perfectly
766 				 * good spot to store the boot file (join on
767 				 * Tru64 Unix) it seems mind bogglingly crazy
768 				 * to me
769 				 */
770 				printf("*** WARNING: using vendor "
771 					"optional boot file\n");
772 				memcpy(bp->bp_file, popt + 2, size);
773 				bp->bp_file[size] = '\0';
774 			}
775 			break;
776 		default:
777 #if defined(CONFIG_BOOTP_VENDOREX)
778 			if (dhcp_vendorex_proc(popt))
779 				break;
780 #endif
781 			printf("*** Unhandled DHCP Option in OFFER/ACK:"
782 				" %d\n", *popt);
783 			break;
784 		}
785 		popt += oplen + 2;	/* Process next option */
786 	}
787 }
788 
789 static int DhcpMessageType(unsigned char *popt)
790 {
791 	if (NetReadLong((ulong *)popt) != htonl(BOOTP_VENDOR_MAGIC))
792 		return -1;
793 
794 	popt += 4;
795 	while (*popt != 0xff) {
796 		if (*popt == 53)	/* DHCP Message Type */
797 			return *(popt + 2);
798 		popt += *(popt + 1) + 2;	/* Scan through all options */
799 	}
800 	return -1;
801 }
802 
803 static void DhcpSendRequestPkt(struct Bootp_t *bp_offer)
804 {
805 	uchar *pkt, *iphdr;
806 	struct Bootp_t *bp;
807 	int pktlen, iplen, extlen;
808 	int eth_hdr_size;
809 	IPaddr_t OfferedIP;
810 
811 	debug("DhcpSendRequestPkt: Sending DHCPREQUEST\n");
812 	pkt = NetTxPacket;
813 	memset((void *)pkt, 0, PKTSIZE);
814 
815 	eth_hdr_size = NetSetEther(pkt, NetBcastAddr, PROT_IP);
816 	pkt += eth_hdr_size;
817 
818 	iphdr = pkt;	/* We'll need this later to set proper pkt size */
819 	pkt += IP_UDP_HDR_SIZE;
820 
821 	bp = (struct Bootp_t *)pkt;
822 	bp->bp_op = OP_BOOTREQUEST;
823 	bp->bp_htype = HWT_ETHER;
824 	bp->bp_hlen = HWL_ETHER;
825 	bp->bp_hops = 0;
826 	bp->bp_secs = htons(get_timer(0) / 1000);
827 	/* Do not set the client IP, your IP, or server IP yet, since it
828 	 * hasn't been ACK'ed by the server yet */
829 
830 	/*
831 	 * RFC3046 requires Relay Agents to discard packets with
832 	 * nonzero and offered giaddr
833 	 */
834 	NetWriteIP(&bp->bp_giaddr, 0);
835 
836 	memcpy(bp->bp_chaddr, NetOurEther, 6);
837 
838 	/*
839 	 * ID is the id of the OFFER packet
840 	 */
841 
842 	NetCopyLong(&bp->bp_id, &bp_offer->bp_id);
843 
844 	/*
845 	 * Copy options from OFFER packet if present
846 	 */
847 
848 	/* Copy offered IP into the parameters request list */
849 	NetCopyIP(&OfferedIP, &bp_offer->bp_yiaddr);
850 	extlen = DhcpExtended((u8 *)bp->bp_vend, DHCP_REQUEST,
851 		NetDHCPServerIP, OfferedIP);
852 
853 	iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen;
854 	pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen;
855 	net_set_udp_header(iphdr, 0xFFFFFFFFL, PORT_BOOTPS, PORT_BOOTPC, iplen);
856 
857 #ifdef CONFIG_BOOTP_DHCP_REQUEST_DELAY
858 	udelay(CONFIG_BOOTP_DHCP_REQUEST_DELAY);
859 #endif	/* CONFIG_BOOTP_DHCP_REQUEST_DELAY */
860 	debug("Transmitting DHCPREQUEST packet: len = %d\n", pktlen);
861 	NetSendPacket(NetTxPacket, pktlen);
862 }
863 
864 /*
865  *	Handle DHCP received packets.
866  */
867 static void
868 DhcpHandler(uchar *pkt, unsigned dest, IPaddr_t sip, unsigned src,
869 	    unsigned len)
870 {
871 	struct Bootp_t *bp = (struct Bootp_t *)pkt;
872 
873 	debug("DHCPHandler: got packet: (src=%d, dst=%d, len=%d) state: %d\n",
874 		src, dest, len, dhcp_state);
875 
876 	/* Filter out pkts we don't want */
877 	if (BootpCheckPkt(pkt, dest, src, len))
878 		return;
879 
880 	debug("DHCPHandler: got DHCP packet: (src=%d, dst=%d, len=%d) state:"
881 		" %d\n", src, dest, len, dhcp_state);
882 
883 	switch (dhcp_state) {
884 	case SELECTING:
885 		/*
886 		 * Wait an appropriate time for any potential DHCPOFFER packets
887 		 * to arrive.  Then select one, and generate DHCPREQUEST
888 		 * response.  If filename is in format we recognize, assume it
889 		 * is a valid OFFER from a server we want.
890 		 */
891 		debug("DHCP: state=SELECTING bp_file: \"%s\"\n", bp->bp_file);
892 #ifdef CONFIG_SYS_BOOTFILE_PREFIX
893 		if (strncmp(bp->bp_file,
894 			    CONFIG_SYS_BOOTFILE_PREFIX,
895 			    strlen(CONFIG_SYS_BOOTFILE_PREFIX)) == 0) {
896 #endif	/* CONFIG_SYS_BOOTFILE_PREFIX */
897 
898 			debug("TRANSITIONING TO REQUESTING STATE\n");
899 			dhcp_state = REQUESTING;
900 
901 			if (NetReadLong((ulong *)&bp->bp_vend[0]) ==
902 						htonl(BOOTP_VENDOR_MAGIC))
903 				DhcpOptionsProcess((u8 *)&bp->bp_vend[4], bp);
904 
905 			NetSetTimeout(TIMEOUT, BootpTimeout);
906 			DhcpSendRequestPkt(bp);
907 #ifdef CONFIG_SYS_BOOTFILE_PREFIX
908 		}
909 #endif	/* CONFIG_SYS_BOOTFILE_PREFIX */
910 
911 		return;
912 		break;
913 	case REQUESTING:
914 		debug("DHCP State: REQUESTING\n");
915 
916 		if (DhcpMessageType((u8 *)bp->bp_vend) == DHCP_ACK) {
917 			if (NetReadLong((ulong *)&bp->bp_vend[0]) ==
918 						htonl(BOOTP_VENDOR_MAGIC))
919 				DhcpOptionsProcess((u8 *)&bp->bp_vend[4], bp);
920 			/* Store net params from reply */
921 			BootpCopyNetParams(bp);
922 			dhcp_state = BOUND;
923 			printf("DHCP client bound to address %pI4\n",
924 				&NetOurIP);
925 			bootstage_mark_name(BOOTSTAGE_ID_BOOTP_STOP,
926 				"bootp_stop");
927 
928 			net_auto_load();
929 			return;
930 		}
931 		break;
932 	case BOUND:
933 		/* DHCP client bound to address */
934 		break;
935 	default:
936 		puts("DHCP: INVALID STATE\n");
937 		break;
938 	}
939 
940 }
941 
942 void DhcpRequest(void)
943 {
944 	BootpRequest();
945 }
946 #endif	/* CONFIG_CMD_DHCP */
947