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