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 #ifdef CONFIG_LIB_UUID 444 char *uuid; 445 #endif 446 int clientarch = -1; 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 #ifdef CONFIG_BOOTP_PXE_CLIENTARCH 503 clientarch = CONFIG_BOOTP_PXE_CLIENTARCH; 504 #endif 505 506 if (getenv("bootp_arch")) 507 clientarch = getenv_ulong("bootp_arch", 16, clientarch); 508 509 if (clientarch > 0) { 510 *e++ = 93; /* Client System Architecture */ 511 *e++ = 2; 512 *e++ = (clientarch >> 8) & 0xff; 513 *e++ = clientarch & 0xff; 514 } 515 516 *e++ = 94; /* Client Network Interface Identifier */ 517 *e++ = 3; 518 *e++ = 1; /* type field for UNDI */ 519 *e++ = 0; /* major revision */ 520 *e++ = 0; /* minor revision */ 521 522 #ifdef CONFIG_LIB_UUID 523 uuid = getenv("pxeuuid"); 524 525 if (uuid) { 526 if (uuid_str_valid(uuid)) { 527 *e++ = 97; /* Client Machine Identifier */ 528 *e++ = 17; 529 *e++ = 0; /* type 0 - UUID */ 530 531 uuid_str_to_bin(uuid, e, UUID_STR_FORMAT_STD); 532 e += 16; 533 } else { 534 printf("Invalid pxeuuid: %s\n", uuid); 535 } 536 } 537 #endif 538 539 e = add_vci(e); 540 541 #if defined(CONFIG_BOOTP_VENDOREX) 542 x = dhcp_vendorex_prep(e); 543 if (x) 544 return x - start; 545 #endif 546 547 *e++ = 55; /* Parameter Request List */ 548 cnt = e++; /* Pointer to count of requested items */ 549 *cnt = 0; 550 #if defined(CONFIG_BOOTP_SUBNETMASK) 551 *e++ = 1; /* Subnet Mask */ 552 *cnt += 1; 553 #endif 554 #if defined(CONFIG_BOOTP_TIMEOFFSET) 555 *e++ = 2; 556 *cnt += 1; 557 #endif 558 #if defined(CONFIG_BOOTP_GATEWAY) 559 *e++ = 3; /* Router Option */ 560 *cnt += 1; 561 #endif 562 #if defined(CONFIG_BOOTP_DNS) 563 *e++ = 6; /* DNS Server(s) */ 564 *cnt += 1; 565 #endif 566 #if defined(CONFIG_BOOTP_HOSTNAME) 567 *e++ = 12; /* Hostname */ 568 *cnt += 1; 569 #endif 570 #if defined(CONFIG_BOOTP_BOOTFILESIZE) 571 *e++ = 13; /* Boot File Size */ 572 *cnt += 1; 573 #endif 574 #if defined(CONFIG_BOOTP_BOOTPATH) 575 *e++ = 17; /* Boot path */ 576 *cnt += 1; 577 #endif 578 #if defined(CONFIG_BOOTP_NISDOMAIN) 579 *e++ = 40; /* NIS Domain name request */ 580 *cnt += 1; 581 #endif 582 #if defined(CONFIG_BOOTP_NTPSERVER) 583 *e++ = 42; 584 *cnt += 1; 585 #endif 586 /* no options, so back up to avoid sending an empty request list */ 587 if (*cnt == 0) 588 e -= 2; 589 590 *e++ = 255; /* End of the list */ 591 592 /* Pad to minimal length */ 593 #ifdef CONFIG_DHCP_MIN_EXT_LEN 594 while ((e - start) < CONFIG_DHCP_MIN_EXT_LEN) 595 *e++ = 0; 596 #endif 597 598 return e - start; 599 } 600 601 #else 602 /* 603 * Warning: no field size check - change CONFIG_BOOTP_* at your own risk! 604 */ 605 static int bootp_extended(u8 *e) 606 { 607 u8 *start = e; 608 609 *e++ = 99; /* RFC1048 Magic Cookie */ 610 *e++ = 130; 611 *e++ = 83; 612 *e++ = 99; 613 614 #if defined(CONFIG_CMD_DHCP) 615 *e++ = 53; /* DHCP Message Type */ 616 *e++ = 1; 617 *e++ = DHCP_DISCOVER; 618 619 *e++ = 57; /* Maximum DHCP Message Size */ 620 *e++ = 2; 621 *e++ = (576 - 312 + OPT_FIELD_SIZE) >> 16; 622 *e++ = (576 - 312 + OPT_FIELD_SIZE) & 0xff; 623 #endif 624 625 add_vci(e); 626 627 #if defined(CONFIG_BOOTP_SUBNETMASK) 628 *e++ = 1; /* Subnet mask request */ 629 *e++ = 4; 630 e += 4; 631 #endif 632 633 #if defined(CONFIG_BOOTP_GATEWAY) 634 *e++ = 3; /* Default gateway request */ 635 *e++ = 4; 636 e += 4; 637 #endif 638 639 #if defined(CONFIG_BOOTP_DNS) 640 *e++ = 6; /* Domain Name Server */ 641 *e++ = 4; 642 e += 4; 643 #endif 644 645 #if defined(CONFIG_BOOTP_HOSTNAME) 646 *e++ = 12; /* Host name request */ 647 *e++ = 32; 648 e += 32; 649 #endif 650 651 #if defined(CONFIG_BOOTP_BOOTFILESIZE) 652 *e++ = 13; /* Boot file size */ 653 *e++ = 2; 654 e += 2; 655 #endif 656 657 #if defined(CONFIG_BOOTP_BOOTPATH) 658 *e++ = 17; /* Boot path */ 659 *e++ = 32; 660 e += 32; 661 #endif 662 663 #if defined(CONFIG_BOOTP_NISDOMAIN) 664 *e++ = 40; /* NIS Domain name request */ 665 *e++ = 32; 666 e += 32; 667 #endif 668 #if defined(CONFIG_BOOTP_NTPSERVER) 669 *e++ = 42; 670 *e++ = 4; 671 e += 4; 672 #endif 673 674 *e++ = 255; /* End of the list */ 675 676 /* 677 * If nothing in list, remove it altogether. Some DHCP servers get 678 * upset by this minor faux pas and do not respond at all. 679 */ 680 if (e == start + 3) { 681 printf("*** Warning: no DHCP options requested\n"); 682 e -= 3; 683 } 684 685 return e - start; 686 } 687 #endif 688 689 void bootp_reset(void) 690 { 691 bootp_num_ids = 0; 692 bootp_try = 0; 693 bootp_start = get_timer(0); 694 bootp_timeout = 250; 695 } 696 697 void bootp_request(void) 698 { 699 uchar *pkt, *iphdr; 700 struct bootp_hdr *bp; 701 int extlen, pktlen, iplen; 702 int eth_hdr_size; 703 #ifdef CONFIG_BOOTP_RANDOM_DELAY 704 ulong rand_ms; 705 #endif 706 u32 bootp_id; 707 struct in_addr zero_ip; 708 struct in_addr bcast_ip; 709 char *ep; /* Environment pointer */ 710 711 bootstage_mark_name(BOOTSTAGE_ID_BOOTP_START, "bootp_start"); 712 #if defined(CONFIG_CMD_DHCP) 713 dhcp_state = INIT; 714 #endif 715 716 ep = getenv("bootpretryperiod"); 717 if (ep != NULL) 718 time_taken_max = simple_strtoul(ep, NULL, 10); 719 else 720 time_taken_max = TIMEOUT_MS; 721 722 #ifdef CONFIG_BOOTP_RANDOM_DELAY /* Random BOOTP delay */ 723 if (bootp_try == 0) 724 srand_mac(); 725 726 if (bootp_try <= 2) /* Start with max 1024 * 1ms */ 727 rand_ms = rand() >> (22 - bootp_try); 728 else /* After 3rd BOOTP request max 8192 * 1ms */ 729 rand_ms = rand() >> 19; 730 731 printf("Random delay: %ld ms...\n", rand_ms); 732 mdelay(rand_ms); 733 734 #endif /* CONFIG_BOOTP_RANDOM_DELAY */ 735 736 printf("BOOTP broadcast %d\n", ++bootp_try); 737 pkt = net_tx_packet; 738 memset((void *)pkt, 0, PKTSIZE); 739 740 eth_hdr_size = net_set_ether(pkt, net_bcast_ethaddr, PROT_IP); 741 pkt += eth_hdr_size; 742 743 /* 744 * Next line results in incorrect packet size being transmitted, 745 * resulting in errors in some DHCP servers, reporting missing bytes. 746 * Size must be set in packet header after extension length has been 747 * determined. 748 * C. Hallinan, DS4.COM, Inc. 749 */ 750 /* net_set_udp_header(pkt, 0xFFFFFFFFL, PORT_BOOTPS, PORT_BOOTPC, 751 sizeof (struct bootp_hdr)); */ 752 iphdr = pkt; /* We need this later for net_set_udp_header() */ 753 pkt += IP_UDP_HDR_SIZE; 754 755 bp = (struct bootp_hdr *)pkt; 756 bp->bp_op = OP_BOOTREQUEST; 757 bp->bp_htype = HWT_ETHER; 758 bp->bp_hlen = HWL_ETHER; 759 bp->bp_hops = 0; 760 /* 761 * according to RFC1542, should be 0 on first request, secs since 762 * first request otherwise 763 */ 764 bp->bp_secs = htons(get_timer(bootp_start) / 1000); 765 zero_ip.s_addr = 0; 766 net_write_ip(&bp->bp_ciaddr, zero_ip); 767 net_write_ip(&bp->bp_yiaddr, zero_ip); 768 net_write_ip(&bp->bp_siaddr, zero_ip); 769 net_write_ip(&bp->bp_giaddr, zero_ip); 770 memcpy(bp->bp_chaddr, net_ethaddr, 6); 771 copy_filename(bp->bp_file, net_boot_file_name, sizeof(bp->bp_file)); 772 773 /* Request additional information from the BOOTP/DHCP server */ 774 #if defined(CONFIG_CMD_DHCP) 775 extlen = dhcp_extended((u8 *)bp->bp_vend, DHCP_DISCOVER, zero_ip, 776 zero_ip); 777 #else 778 extlen = bootp_extended((u8 *)bp->bp_vend); 779 #endif 780 781 /* 782 * Bootp ID is the lower 4 bytes of our ethernet address 783 * plus the current time in ms. 784 */ 785 bootp_id = ((u32)net_ethaddr[2] << 24) 786 | ((u32)net_ethaddr[3] << 16) 787 | ((u32)net_ethaddr[4] << 8) 788 | (u32)net_ethaddr[5]; 789 bootp_id += get_timer(0); 790 bootp_id = htonl(bootp_id); 791 bootp_add_id(bootp_id); 792 net_copy_u32(&bp->bp_id, &bootp_id); 793 794 /* 795 * Calculate proper packet lengths taking into account the 796 * variable size of the options field 797 */ 798 iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen; 799 pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen; 800 bcast_ip.s_addr = 0xFFFFFFFFL; 801 net_set_udp_header(iphdr, bcast_ip, PORT_BOOTPS, PORT_BOOTPC, iplen); 802 net_set_timeout_handler(bootp_timeout, bootp_timeout_handler); 803 804 #if defined(CONFIG_CMD_DHCP) 805 dhcp_state = SELECTING; 806 net_set_udp_handler(dhcp_handler); 807 #else 808 net_set_udp_handler(bootp_handler); 809 #endif 810 net_send_packet(net_tx_packet, pktlen); 811 } 812 813 #if defined(CONFIG_CMD_DHCP) 814 static void dhcp_process_options(uchar *popt, uchar *end) 815 { 816 int oplen, size; 817 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET) 818 int *to_ptr; 819 #endif 820 821 while (popt < end && *popt != 0xff) { 822 oplen = *(popt + 1); 823 switch (*popt) { 824 case 0: 825 oplen = -1; /* Pad omits len byte */ 826 break; 827 case 1: 828 net_copy_ip(&net_netmask, (popt + 2)); 829 break; 830 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET) 831 case 2: /* Time offset */ 832 to_ptr = &net_ntp_time_offset; 833 net_copy_u32((u32 *)to_ptr, (u32 *)(popt + 2)); 834 net_ntp_time_offset = ntohl(net_ntp_time_offset); 835 break; 836 #endif 837 case 3: 838 net_copy_ip(&net_gateway, (popt + 2)); 839 break; 840 case 6: 841 net_copy_ip(&net_dns_server, (popt + 2)); 842 #if defined(CONFIG_BOOTP_DNS2) 843 if (*(popt + 1) > 4) 844 net_copy_ip(&net_dns_server2, (popt + 2 + 4)); 845 #endif 846 break; 847 case 12: 848 size = truncate_sz("Host Name", 849 sizeof(net_hostname), oplen); 850 memcpy(&net_hostname, popt + 2, size); 851 net_hostname[size] = 0; 852 break; 853 case 15: /* Ignore Domain Name Option */ 854 break; 855 case 17: 856 size = truncate_sz("Root Path", 857 sizeof(net_root_path), oplen); 858 memcpy(&net_root_path, popt + 2, size); 859 net_root_path[size] = 0; 860 break; 861 case 28: /* Ignore Broadcast Address Option */ 862 break; 863 #if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_NTPSERVER) 864 case 42: /* NTP server IP */ 865 net_copy_ip(&net_ntp_server, (popt + 2)); 866 break; 867 #endif 868 case 51: 869 net_copy_u32(&dhcp_leasetime, (u32 *)(popt + 2)); 870 break; 871 case 52: 872 dhcp_option_overload = popt[2]; 873 break; 874 case 53: /* Ignore Message Type Option */ 875 break; 876 case 54: 877 net_copy_ip(&dhcp_server_ip, (popt + 2)); 878 break; 879 case 58: /* Ignore Renewal Time Option */ 880 break; 881 case 59: /* Ignore Rebinding Time Option */ 882 break; 883 case 66: /* Ignore TFTP server name */ 884 break; 885 case 67: /* Bootfile option */ 886 size = truncate_sz("Bootfile", 887 sizeof(net_boot_file_name), oplen); 888 memcpy(&net_boot_file_name, popt + 2, size); 889 net_boot_file_name[size] = 0; 890 break; 891 default: 892 #if defined(CONFIG_BOOTP_VENDOREX) 893 if (dhcp_vendorex_proc(popt)) 894 break; 895 #endif 896 printf("*** Unhandled DHCP Option in OFFER/ACK:" 897 " %d\n", *popt); 898 break; 899 } 900 popt += oplen + 2; /* Process next option */ 901 } 902 } 903 904 static void dhcp_packet_process_options(struct bootp_hdr *bp) 905 { 906 uchar *popt = (uchar *)&bp->bp_vend[4]; 907 uchar *end = popt + BOOTP_HDR_SIZE; 908 909 if (net_read_u32((u32 *)&bp->bp_vend[0]) != htonl(BOOTP_VENDOR_MAGIC)) 910 return; 911 912 dhcp_option_overload = 0; 913 914 /* 915 * The 'options' field MUST be interpreted first, 'file' next, 916 * 'sname' last. 917 */ 918 dhcp_process_options(popt, end); 919 920 if (dhcp_option_overload & OVERLOAD_FILE) { 921 popt = (uchar *)bp->bp_file; 922 end = popt + sizeof(bp->bp_file); 923 dhcp_process_options(popt, end); 924 } 925 926 if (dhcp_option_overload & OVERLOAD_SNAME) { 927 popt = (uchar *)bp->bp_sname; 928 end = popt + sizeof(bp->bp_sname); 929 dhcp_process_options(popt, end); 930 } 931 } 932 933 static int dhcp_message_type(unsigned char *popt) 934 { 935 if (net_read_u32((u32 *)popt) != htonl(BOOTP_VENDOR_MAGIC)) 936 return -1; 937 938 popt += 4; 939 while (*popt != 0xff) { 940 if (*popt == 53) /* DHCP Message Type */ 941 return *(popt + 2); 942 if (*popt == 0) { 943 /* Pad */ 944 popt += 1; 945 } else { 946 /* Scan through all options */ 947 popt += *(popt + 1) + 2; 948 } 949 } 950 return -1; 951 } 952 953 static void dhcp_send_request_packet(struct bootp_hdr *bp_offer) 954 { 955 uchar *pkt, *iphdr; 956 struct bootp_hdr *bp; 957 int pktlen, iplen, extlen; 958 int eth_hdr_size; 959 struct in_addr offered_ip; 960 struct in_addr zero_ip; 961 struct in_addr bcast_ip; 962 963 debug("dhcp_send_request_packet: Sending DHCPREQUEST\n"); 964 pkt = net_tx_packet; 965 memset((void *)pkt, 0, PKTSIZE); 966 967 eth_hdr_size = net_set_ether(pkt, net_bcast_ethaddr, PROT_IP); 968 pkt += eth_hdr_size; 969 970 iphdr = pkt; /* We'll need this later to set proper pkt size */ 971 pkt += IP_UDP_HDR_SIZE; 972 973 bp = (struct bootp_hdr *)pkt; 974 bp->bp_op = OP_BOOTREQUEST; 975 bp->bp_htype = HWT_ETHER; 976 bp->bp_hlen = HWL_ETHER; 977 bp->bp_hops = 0; 978 bp->bp_secs = htons(get_timer(bootp_start) / 1000); 979 /* Do not set the client IP, your IP, or server IP yet, since it 980 * hasn't been ACK'ed by the server yet */ 981 982 /* 983 * RFC3046 requires Relay Agents to discard packets with 984 * nonzero and offered giaddr 985 */ 986 zero_ip.s_addr = 0; 987 net_write_ip(&bp->bp_giaddr, zero_ip); 988 989 memcpy(bp->bp_chaddr, net_ethaddr, 6); 990 copy_filename(bp->bp_file, net_boot_file_name, sizeof(bp->bp_file)); 991 992 /* 993 * ID is the id of the OFFER packet 994 */ 995 996 net_copy_u32(&bp->bp_id, &bp_offer->bp_id); 997 998 /* 999 * Copy options from OFFER packet if present 1000 */ 1001 1002 /* Copy offered IP into the parameters request list */ 1003 net_copy_ip(&offered_ip, &bp_offer->bp_yiaddr); 1004 extlen = dhcp_extended((u8 *)bp->bp_vend, DHCP_REQUEST, 1005 dhcp_server_ip, offered_ip); 1006 1007 iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen; 1008 pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen; 1009 bcast_ip.s_addr = 0xFFFFFFFFL; 1010 net_set_udp_header(iphdr, bcast_ip, PORT_BOOTPS, PORT_BOOTPC, iplen); 1011 1012 #ifdef CONFIG_BOOTP_DHCP_REQUEST_DELAY 1013 udelay(CONFIG_BOOTP_DHCP_REQUEST_DELAY); 1014 #endif /* CONFIG_BOOTP_DHCP_REQUEST_DELAY */ 1015 debug("Transmitting DHCPREQUEST packet: len = %d\n", pktlen); 1016 net_send_packet(net_tx_packet, pktlen); 1017 } 1018 1019 /* 1020 * Handle DHCP received packets. 1021 */ 1022 static void dhcp_handler(uchar *pkt, unsigned dest, struct in_addr sip, 1023 unsigned src, unsigned len) 1024 { 1025 struct bootp_hdr *bp = (struct bootp_hdr *)pkt; 1026 1027 debug("DHCPHandler: got packet: (src=%d, dst=%d, len=%d) state: %d\n", 1028 src, dest, len, dhcp_state); 1029 1030 /* Filter out pkts we don't want */ 1031 if (check_reply_packet(pkt, dest, src, len)) 1032 return; 1033 1034 debug("DHCPHandler: got DHCP packet: (src=%d, dst=%d, len=%d) state: " 1035 "%d\n", src, dest, len, dhcp_state); 1036 1037 if (net_read_ip(&bp->bp_yiaddr).s_addr == 0) 1038 return; 1039 1040 switch (dhcp_state) { 1041 case SELECTING: 1042 /* 1043 * Wait an appropriate time for any potential DHCPOFFER packets 1044 * to arrive. Then select one, and generate DHCPREQUEST 1045 * response. If filename is in format we recognize, assume it 1046 * is a valid OFFER from a server we want. 1047 */ 1048 debug("DHCP: state=SELECTING bp_file: \"%s\"\n", bp->bp_file); 1049 #ifdef CONFIG_SYS_BOOTFILE_PREFIX 1050 if (strncmp(bp->bp_file, 1051 CONFIG_SYS_BOOTFILE_PREFIX, 1052 strlen(CONFIG_SYS_BOOTFILE_PREFIX)) == 0) { 1053 #endif /* CONFIG_SYS_BOOTFILE_PREFIX */ 1054 dhcp_packet_process_options(bp); 1055 efi_net_set_dhcp_ack(pkt, len); 1056 1057 debug("TRANSITIONING TO REQUESTING STATE\n"); 1058 dhcp_state = REQUESTING; 1059 1060 net_set_timeout_handler(5000, bootp_timeout_handler); 1061 dhcp_send_request_packet(bp); 1062 #ifdef CONFIG_SYS_BOOTFILE_PREFIX 1063 } 1064 #endif /* CONFIG_SYS_BOOTFILE_PREFIX */ 1065 1066 return; 1067 break; 1068 case REQUESTING: 1069 debug("DHCP State: REQUESTING\n"); 1070 1071 if (dhcp_message_type((u8 *)bp->bp_vend) == DHCP_ACK) { 1072 dhcp_packet_process_options(bp); 1073 /* Store net params from reply */ 1074 store_net_params(bp); 1075 dhcp_state = BOUND; 1076 printf("DHCP client bound to address %pI4 (%lu ms)\n", 1077 &net_ip, get_timer(bootp_start)); 1078 net_set_timeout_handler(0, (thand_f *)0); 1079 bootstage_mark_name(BOOTSTAGE_ID_BOOTP_STOP, 1080 "bootp_stop"); 1081 1082 net_auto_load(); 1083 return; 1084 } 1085 break; 1086 case BOUND: 1087 /* DHCP client bound to address */ 1088 break; 1089 default: 1090 puts("DHCP: INVALID STATE\n"); 1091 break; 1092 } 1093 } 1094 1095 void dhcp_request(void) 1096 { 1097 bootp_request(); 1098 } 1099 #endif /* CONFIG_CMD_DHCP */ 1100