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