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