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