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