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