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