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