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