1 /* 2 * Copied from Linux Monitor (LiMon) - Networking. 3 * 4 * Copyright 1994 - 2000 Neil Russell. 5 * (See License) 6 * Copyright 2000 Roland Borde 7 * Copyright 2000 Paolo Scaffardi 8 * Copyright 2000-2002 Wolfgang Denk, wd@denx.de 9 */ 10 11 /* 12 * General Desription: 13 * 14 * The user interface supports commands for BOOTP, RARP, and TFTP. 15 * Also, we support ARP internally. Depending on available data, 16 * these interact as follows: 17 * 18 * BOOTP: 19 * 20 * Prerequisites: - own ethernet address 21 * We want: - own IP address 22 * - TFTP server IP address 23 * - name of bootfile 24 * Next step: ARP 25 * 26 * RARP: 27 * 28 * Prerequisites: - own ethernet address 29 * We want: - own IP address 30 * - TFTP server IP address 31 * Next step: ARP 32 * 33 * ARP: 34 * 35 * Prerequisites: - own ethernet address 36 * - own IP address 37 * - TFTP server IP address 38 * We want: - TFTP server ethernet address 39 * Next step: TFTP 40 * 41 * DHCP: 42 * 43 * Prerequisites: - own ethernet address 44 * We want: - IP, Netmask, ServerIP, Gateway IP 45 * - bootfilename, lease time 46 * Next step: - TFTP 47 * 48 * TFTP: 49 * 50 * Prerequisites: - own ethernet address 51 * - own IP address 52 * - TFTP server IP address 53 * - TFTP server ethernet address 54 * - name of bootfile (if unknown, we use a default name 55 * derived from our own IP address) 56 * We want: - load the boot file 57 * Next step: none 58 * 59 * NFS: 60 * 61 * Prerequisites: - own ethernet address 62 * - own IP address 63 * - name of bootfile (if unknown, we use a default name 64 * derived from our own IP address) 65 * We want: - load the boot file 66 * Next step: none 67 * 68 * SNTP: 69 * 70 * Prerequisites: - own ethernet address 71 * - own IP address 72 * We want: - network time 73 * Next step: none 74 */ 75 76 77 #include <common.h> 78 #include <command.h> 79 #include <net.h> 80 #if defined(CONFIG_STATUS_LED) 81 #include <miiphy.h> 82 #include <status_led.h> 83 #endif 84 #include <watchdog.h> 85 #include <linux/compiler.h> 86 #include "arp.h" 87 #include "bootp.h" 88 #include "cdp.h" 89 #if defined(CONFIG_CMD_DNS) 90 #include "dns.h" 91 #endif 92 #include "nfs.h" 93 #include "ping.h" 94 #include "rarp.h" 95 #if defined(CONFIG_CMD_SNTP) 96 #include "sntp.h" 97 #endif 98 #include "tftp.h" 99 100 DECLARE_GLOBAL_DATA_PTR; 101 102 /** BOOTP EXTENTIONS **/ 103 104 /* Our subnet mask (0=unknown) */ 105 IPaddr_t NetOurSubnetMask; 106 /* Our gateways IP address */ 107 IPaddr_t NetOurGatewayIP; 108 /* Our DNS IP address */ 109 IPaddr_t NetOurDNSIP; 110 #if defined(CONFIG_BOOTP_DNS2) 111 /* Our 2nd DNS IP address */ 112 IPaddr_t NetOurDNS2IP; 113 #endif 114 /* Our NIS domain */ 115 char NetOurNISDomain[32] = {0,}; 116 /* Our hostname */ 117 char NetOurHostName[32] = {0,}; 118 /* Our bootpath */ 119 char NetOurRootPath[64] = {0,}; 120 /* Our bootfile size in blocks */ 121 ushort NetBootFileSize; 122 123 #ifdef CONFIG_MCAST_TFTP /* Multicast TFTP */ 124 IPaddr_t Mcast_addr; 125 #endif 126 127 /** END OF BOOTP EXTENTIONS **/ 128 129 /* The actual transferred size of the bootfile (in bytes) */ 130 ulong NetBootFileXferSize; 131 /* Our ethernet address */ 132 uchar NetOurEther[6]; 133 /* Boot server enet address */ 134 uchar NetServerEther[6]; 135 /* Our IP addr (0 = unknown) */ 136 IPaddr_t NetOurIP; 137 /* Server IP addr (0 = unknown) */ 138 IPaddr_t NetServerIP; 139 /* Current receive packet */ 140 uchar *NetRxPacket; 141 /* Current rx packet length */ 142 int NetRxPacketLen; 143 /* IP packet ID */ 144 unsigned NetIPID; 145 /* Ethernet bcast address */ 146 uchar NetBcastAddr[6] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff }; 147 uchar NetEtherNullAddr[6]; 148 #ifdef CONFIG_API 149 void (*push_packet)(void *, int len) = 0; 150 #endif 151 /* Network loop state */ 152 enum net_loop_state net_state; 153 /* Tried all network devices */ 154 int NetRestartWrap; 155 /* Network loop restarted */ 156 static int NetRestarted; 157 /* At least one device configured */ 158 static int NetDevExists; 159 160 /* XXX in both little & big endian machines 0xFFFF == ntohs(-1) */ 161 /* default is without VLAN */ 162 ushort NetOurVLAN = 0xFFFF; 163 /* ditto */ 164 ushort NetOurNativeVLAN = 0xFFFF; 165 166 /* Boot File name */ 167 char BootFile[128]; 168 169 #if defined(CONFIG_CMD_SNTP) 170 /* NTP server IP address */ 171 IPaddr_t NetNtpServerIP; 172 /* offset time from UTC */ 173 int NetTimeOffset; 174 #endif 175 176 uchar PktBuf[(PKTBUFSRX+1) * PKTSIZE_ALIGN + PKTALIGN]; 177 178 /* Receive packet */ 179 uchar *NetRxPackets[PKTBUFSRX]; 180 181 /* Current UDP RX packet handler */ 182 static rxhand_f *udp_packet_handler; 183 /* Current ARP RX packet handler */ 184 static rxhand_f *arp_packet_handler; 185 #ifdef CONFIG_CMD_TFTPPUT 186 /* Current ICMP rx handler */ 187 static rxhand_icmp_f *packet_icmp_handler; 188 #endif 189 /* Current timeout handler */ 190 static thand_f *timeHandler; 191 /* Time base value */ 192 static ulong timeStart; 193 /* Current timeout value */ 194 static ulong timeDelta; 195 /* THE transmit packet */ 196 uchar *NetTxPacket; 197 198 static int net_check_prereq(enum proto_t protocol); 199 200 static int NetTryCount; 201 202 /**********************************************************************/ 203 204 /* 205 * Check if autoload is enabled. If so, use either NFS or TFTP to download 206 * the boot file. 207 */ 208 void net_auto_load(void) 209 { 210 const char *s = getenv("autoload"); 211 212 if (s != NULL) { 213 if (*s == 'n') { 214 /* 215 * Just use BOOTP/RARP to configure system; 216 * Do not use TFTP to load the bootfile. 217 */ 218 net_set_state(NETLOOP_SUCCESS); 219 return; 220 } 221 #if defined(CONFIG_CMD_NFS) 222 if (strcmp(s, "NFS") == 0) { 223 /* 224 * Use NFS to load the bootfile. 225 */ 226 NfsStart(); 227 return; 228 } 229 #endif 230 } 231 TftpStart(TFTPGET); 232 } 233 234 static void NetInitLoop(void) 235 { 236 static int env_changed_id; 237 int env_id = get_env_id(); 238 239 /* update only when the environment has changed */ 240 if (env_changed_id != env_id) { 241 NetOurIP = getenv_IPaddr("ipaddr"); 242 NetOurGatewayIP = getenv_IPaddr("gatewayip"); 243 NetOurSubnetMask = getenv_IPaddr("netmask"); 244 NetServerIP = getenv_IPaddr("serverip"); 245 NetOurNativeVLAN = getenv_VLAN("nvlan"); 246 NetOurVLAN = getenv_VLAN("vlan"); 247 #if defined(CONFIG_CMD_DNS) 248 NetOurDNSIP = getenv_IPaddr("dnsip"); 249 #endif 250 env_changed_id = env_id; 251 } 252 253 return; 254 } 255 256 static void net_clear_handlers(void) 257 { 258 net_set_udp_handler(NULL); 259 net_set_arp_handler(NULL); 260 NetSetTimeout(0, NULL); 261 } 262 263 static void net_cleanup_loop(void) 264 { 265 net_clear_handlers(); 266 } 267 268 void net_init(void) 269 { 270 static int first_call = 1; 271 272 if (first_call) { 273 /* 274 * Setup packet buffers, aligned correctly. 275 */ 276 int i; 277 278 NetTxPacket = &PktBuf[0] + (PKTALIGN - 1); 279 NetTxPacket -= (ulong)NetTxPacket % PKTALIGN; 280 for (i = 0; i < PKTBUFSRX; i++) 281 NetRxPackets[i] = NetTxPacket + (i + 1) * PKTSIZE_ALIGN; 282 283 ArpInit(); 284 net_clear_handlers(); 285 286 /* Only need to setup buffer pointers once. */ 287 first_call = 0; 288 } 289 290 NetInitLoop(); 291 } 292 293 /**********************************************************************/ 294 /* 295 * Main network processing loop. 296 */ 297 298 int NetLoop(enum proto_t protocol) 299 { 300 bd_t *bd = gd->bd; 301 int ret = -1; 302 303 NetRestarted = 0; 304 NetDevExists = 0; 305 NetTryCount = 1; 306 307 bootstage_mark_name(BOOTSTAGE_ID_ETH_START, "eth_start"); 308 net_init(); 309 eth_halt(); 310 eth_set_current(); 311 if (eth_init(bd) < 0) { 312 eth_halt(); 313 return -1; 314 } 315 316 restart: 317 memcpy(NetOurEther, eth_get_dev()->enetaddr, 6); 318 319 net_set_state(NETLOOP_CONTINUE); 320 321 /* 322 * Start the ball rolling with the given start function. From 323 * here on, this code is a state machine driven by received 324 * packets and timer events. 325 */ 326 NetInitLoop(); 327 328 switch (net_check_prereq(protocol)) { 329 case 1: 330 /* network not configured */ 331 eth_halt(); 332 return -1; 333 334 case 2: 335 /* network device not configured */ 336 break; 337 338 case 0: 339 NetDevExists = 1; 340 NetBootFileXferSize = 0; 341 switch (protocol) { 342 case TFTPGET: 343 #ifdef CONFIG_CMD_TFTPPUT 344 case TFTPPUT: 345 #endif 346 /* always use ARP to get server ethernet address */ 347 TftpStart(protocol); 348 break; 349 #ifdef CONFIG_CMD_TFTPSRV 350 case TFTPSRV: 351 TftpStartServer(); 352 break; 353 #endif 354 #if defined(CONFIG_CMD_DHCP) 355 case DHCP: 356 BootpTry = 0; 357 NetOurIP = 0; 358 DhcpRequest(); /* Basically same as BOOTP */ 359 break; 360 #endif 361 362 case BOOTP: 363 BootpTry = 0; 364 NetOurIP = 0; 365 BootpRequest(); 366 break; 367 368 #if defined(CONFIG_CMD_RARP) 369 case RARP: 370 RarpTry = 0; 371 NetOurIP = 0; 372 RarpRequest(); 373 break; 374 #endif 375 #if defined(CONFIG_CMD_PING) 376 case PING: 377 ping_start(); 378 break; 379 #endif 380 #if defined(CONFIG_CMD_NFS) 381 case NFS: 382 NfsStart(); 383 break; 384 #endif 385 #if defined(CONFIG_CMD_CDP) 386 case CDP: 387 CDPStart(); 388 break; 389 #endif 390 #ifdef CONFIG_NETCONSOLE 391 case NETCONS: 392 NcStart(); 393 break; 394 #endif 395 #if defined(CONFIG_CMD_SNTP) 396 case SNTP: 397 SntpStart(); 398 break; 399 #endif 400 #if defined(CONFIG_CMD_DNS) 401 case DNS: 402 DnsStart(); 403 break; 404 #endif 405 default: 406 break; 407 } 408 409 break; 410 } 411 412 #if defined(CONFIG_MII) || defined(CONFIG_CMD_MII) 413 #if defined(CONFIG_SYS_FAULT_ECHO_LINK_DOWN) && \ 414 defined(CONFIG_STATUS_LED) && \ 415 defined(STATUS_LED_RED) 416 /* 417 * Echo the inverted link state to the fault LED. 418 */ 419 if (miiphy_link(eth_get_dev()->name, CONFIG_SYS_FAULT_MII_ADDR)) 420 status_led_set(STATUS_LED_RED, STATUS_LED_OFF); 421 else 422 status_led_set(STATUS_LED_RED, STATUS_LED_ON); 423 #endif /* CONFIG_SYS_FAULT_ECHO_LINK_DOWN, ... */ 424 #endif /* CONFIG_MII, ... */ 425 426 /* 427 * Main packet reception loop. Loop receiving packets until 428 * someone sets `net_state' to a state that terminates. 429 */ 430 for (;;) { 431 WATCHDOG_RESET(); 432 #ifdef CONFIG_SHOW_ACTIVITY 433 show_activity(1); 434 #endif 435 /* 436 * Check the ethernet for a new packet. The ethernet 437 * receive routine will process it. 438 */ 439 eth_rx(); 440 441 /* 442 * Abort if ctrl-c was pressed. 443 */ 444 if (ctrlc()) { 445 net_cleanup_loop(); 446 eth_halt(); 447 puts("\nAbort\n"); 448 goto done; 449 } 450 451 ArpTimeoutCheck(); 452 453 /* 454 * Check for a timeout, and run the timeout handler 455 * if we have one. 456 */ 457 if (timeHandler && ((get_timer(0) - timeStart) > timeDelta)) { 458 thand_f *x; 459 460 #if defined(CONFIG_MII) || defined(CONFIG_CMD_MII) 461 #if defined(CONFIG_SYS_FAULT_ECHO_LINK_DOWN) && \ 462 defined(CONFIG_STATUS_LED) && \ 463 defined(STATUS_LED_RED) 464 /* 465 * Echo the inverted link state to the fault LED. 466 */ 467 if (miiphy_link(eth_get_dev()->name, 468 CONFIG_SYS_FAULT_MII_ADDR)) { 469 status_led_set(STATUS_LED_RED, STATUS_LED_OFF); 470 } else { 471 status_led_set(STATUS_LED_RED, STATUS_LED_ON); 472 } 473 #endif /* CONFIG_SYS_FAULT_ECHO_LINK_DOWN, ... */ 474 #endif /* CONFIG_MII, ... */ 475 x = timeHandler; 476 timeHandler = (thand_f *)0; 477 (*x)(); 478 } 479 480 481 switch (net_state) { 482 483 case NETLOOP_RESTART: 484 NetRestarted = 1; 485 goto restart; 486 487 case NETLOOP_SUCCESS: 488 net_cleanup_loop(); 489 if (NetBootFileXferSize > 0) { 490 char buf[20]; 491 printf("Bytes transferred = %ld (%lx hex)\n", 492 NetBootFileXferSize, 493 NetBootFileXferSize); 494 sprintf(buf, "%lX", NetBootFileXferSize); 495 setenv("filesize", buf); 496 497 sprintf(buf, "%lX", (unsigned long)load_addr); 498 setenv("fileaddr", buf); 499 } 500 eth_halt(); 501 ret = NetBootFileXferSize; 502 goto done; 503 504 case NETLOOP_FAIL: 505 net_cleanup_loop(); 506 goto done; 507 508 case NETLOOP_CONTINUE: 509 continue; 510 } 511 } 512 513 done: 514 #ifdef CONFIG_CMD_TFTPPUT 515 /* Clear out the handlers */ 516 net_set_udp_handler(NULL); 517 net_set_icmp_handler(NULL); 518 #endif 519 return ret; 520 } 521 522 /**********************************************************************/ 523 524 static void 525 startAgainTimeout(void) 526 { 527 net_set_state(NETLOOP_RESTART); 528 } 529 530 void NetStartAgain(void) 531 { 532 char *nretry; 533 int retry_forever = 0; 534 unsigned long retrycnt = 0; 535 536 nretry = getenv("netretry"); 537 if (nretry) { 538 if (!strcmp(nretry, "yes")) 539 retry_forever = 1; 540 else if (!strcmp(nretry, "no")) 541 retrycnt = 0; 542 else if (!strcmp(nretry, "once")) 543 retrycnt = 1; 544 else 545 retrycnt = simple_strtoul(nretry, NULL, 0); 546 } else 547 retry_forever = 1; 548 549 if ((!retry_forever) && (NetTryCount >= retrycnt)) { 550 eth_halt(); 551 net_set_state(NETLOOP_FAIL); 552 return; 553 } 554 555 NetTryCount++; 556 557 eth_halt(); 558 #if !defined(CONFIG_NET_DO_NOT_TRY_ANOTHER) 559 eth_try_another(!NetRestarted); 560 #endif 561 eth_init(gd->bd); 562 if (NetRestartWrap) { 563 NetRestartWrap = 0; 564 if (NetDevExists) { 565 NetSetTimeout(10000UL, startAgainTimeout); 566 net_set_udp_handler(NULL); 567 } else { 568 net_set_state(NETLOOP_FAIL); 569 } 570 } else { 571 net_set_state(NETLOOP_RESTART); 572 } 573 } 574 575 /**********************************************************************/ 576 /* 577 * Miscelaneous bits. 578 */ 579 580 static void dummy_handler(uchar *pkt, unsigned dport, 581 IPaddr_t sip, unsigned sport, 582 unsigned len) 583 { 584 } 585 586 rxhand_f *net_get_udp_handler(void) 587 { 588 return udp_packet_handler; 589 } 590 591 void net_set_udp_handler(rxhand_f *f) 592 { 593 if (f == NULL) 594 udp_packet_handler = dummy_handler; 595 else 596 udp_packet_handler = f; 597 } 598 599 rxhand_f *net_get_arp_handler(void) 600 { 601 return arp_packet_handler; 602 } 603 604 void net_set_arp_handler(rxhand_f *f) 605 { 606 if (f == NULL) 607 arp_packet_handler = dummy_handler; 608 else 609 arp_packet_handler = f; 610 } 611 612 #ifdef CONFIG_CMD_TFTPPUT 613 void net_set_icmp_handler(rxhand_icmp_f *f) 614 { 615 packet_icmp_handler = f; 616 } 617 #endif 618 619 void 620 NetSetTimeout(ulong iv, thand_f *f) 621 { 622 if (iv == 0) { 623 timeHandler = (thand_f *)0; 624 } else { 625 timeHandler = f; 626 timeStart = get_timer(0); 627 timeDelta = iv; 628 } 629 } 630 631 int NetSendUDPPacket(uchar *ether, IPaddr_t dest, int dport, int sport, 632 int payload_len) 633 { 634 uchar *pkt; 635 int need_arp = 0; 636 int eth_hdr_size; 637 int pkt_hdr_size; 638 639 /* make sure the NetTxPacket is initialized (NetInit() was called) */ 640 assert(NetTxPacket != NULL); 641 if (NetTxPacket == NULL) 642 return -1; 643 644 /* convert to new style broadcast */ 645 if (dest == 0) 646 dest = 0xFFFFFFFF; 647 648 /* if broadcast, make the ether address a broadcast and don't do ARP */ 649 if (dest == 0xFFFFFFFF) 650 ether = NetBcastAddr; 651 652 /* 653 * if MAC address was not discovered yet, save the packet and do 654 * an ARP request 655 */ 656 if (memcmp(ether, NetEtherNullAddr, 6) == 0) { 657 need_arp = 1; 658 pkt = NetArpWaitTxPacket; 659 } else 660 pkt = (uchar *)NetTxPacket; 661 662 eth_hdr_size = NetSetEther(pkt, ether, PROT_IP); 663 pkt += eth_hdr_size; 664 net_set_udp_header(pkt, dest, dport, sport, payload_len); 665 pkt_hdr_size = eth_hdr_size + IP_UDP_HDR_SIZE; 666 667 if (need_arp) { 668 debug("sending ARP for %pI4\n", &dest); 669 670 /* save the ip and eth addr for the packet to send after arp */ 671 NetArpWaitPacketIP = dest; 672 NetArpWaitPacketMAC = ether; 673 674 /* 675 * Copy the packet data from the NetTxPacket into the 676 * NetArpWaitTxPacket to send after arp 677 */ 678 memcpy(pkt + IP_UDP_HDR_SIZE, (uchar *)NetTxPacket + 679 pkt_hdr_size, payload_len); 680 681 /* size of the waiting packet */ 682 NetArpWaitTxPacketSize = pkt_hdr_size + payload_len; 683 684 /* and do the ARP request */ 685 NetArpWaitTry = 1; 686 NetArpWaitTimerStart = get_timer(0); 687 ArpRequest(); 688 return 1; /* waiting */ 689 } else { 690 debug("sending UDP to %pI4/%pM\n", &dest, ether); 691 NetSendPacket(NetTxPacket, pkt_hdr_size + payload_len); 692 return 0; /* transmitted */ 693 } 694 } 695 696 #ifdef CONFIG_IP_DEFRAG 697 /* 698 * This function collects fragments in a single packet, according 699 * to the algorithm in RFC815. It returns NULL or the pointer to 700 * a complete packet, in static storage 701 */ 702 #ifndef CONFIG_NET_MAXDEFRAG 703 #define CONFIG_NET_MAXDEFRAG 16384 704 #endif 705 /* 706 * MAXDEFRAG, above, is chosen in the config file and is real data 707 * so we need to add the NFS overhead, which is more than TFTP. 708 * To use sizeof in the internal unnamed structures, we need a real 709 * instance (can't do "sizeof(struct rpc_t.u.reply))", unfortunately). 710 * The compiler doesn't complain nor allocates the actual structure 711 */ 712 static struct rpc_t rpc_specimen; 713 #define IP_PKTSIZE (CONFIG_NET_MAXDEFRAG + sizeof(rpc_specimen.u.reply)) 714 715 #define IP_MAXUDP (IP_PKTSIZE - IP_HDR_SIZE) 716 717 /* 718 * this is the packet being assembled, either data or frag control. 719 * Fragments go by 8 bytes, so this union must be 8 bytes long 720 */ 721 struct hole { 722 /* first_byte is address of this structure */ 723 u16 last_byte; /* last byte in this hole + 1 (begin of next hole) */ 724 u16 next_hole; /* index of next (in 8-b blocks), 0 == none */ 725 u16 prev_hole; /* index of prev, 0 == none */ 726 u16 unused; 727 }; 728 729 static struct ip_udp_hdr *__NetDefragment(struct ip_udp_hdr *ip, int *lenp) 730 { 731 static uchar pkt_buff[IP_PKTSIZE] __aligned(PKTALIGN); 732 static u16 first_hole, total_len; 733 struct hole *payload, *thisfrag, *h, *newh; 734 struct ip_udp_hdr *localip = (struct ip_udp_hdr *)pkt_buff; 735 uchar *indata = (uchar *)ip; 736 int offset8, start, len, done = 0; 737 u16 ip_off = ntohs(ip->ip_off); 738 739 /* payload starts after IP header, this fragment is in there */ 740 payload = (struct hole *)(pkt_buff + IP_HDR_SIZE); 741 offset8 = (ip_off & IP_OFFS); 742 thisfrag = payload + offset8; 743 start = offset8 * 8; 744 len = ntohs(ip->ip_len) - IP_HDR_SIZE; 745 746 if (start + len > IP_MAXUDP) /* fragment extends too far */ 747 return NULL; 748 749 if (!total_len || localip->ip_id != ip->ip_id) { 750 /* new (or different) packet, reset structs */ 751 total_len = 0xffff; 752 payload[0].last_byte = ~0; 753 payload[0].next_hole = 0; 754 payload[0].prev_hole = 0; 755 first_hole = 0; 756 /* any IP header will work, copy the first we received */ 757 memcpy(localip, ip, IP_HDR_SIZE); 758 } 759 760 /* 761 * What follows is the reassembly algorithm. We use the payload 762 * array as a linked list of hole descriptors, as each hole starts 763 * at a multiple of 8 bytes. However, last byte can be whatever value, 764 * so it is represented as byte count, not as 8-byte blocks. 765 */ 766 767 h = payload + first_hole; 768 while (h->last_byte < start) { 769 if (!h->next_hole) { 770 /* no hole that far away */ 771 return NULL; 772 } 773 h = payload + h->next_hole; 774 } 775 776 /* last fragment may be 1..7 bytes, the "+7" forces acceptance */ 777 if (offset8 + ((len + 7) / 8) <= h - payload) { 778 /* no overlap with holes (dup fragment?) */ 779 return NULL; 780 } 781 782 if (!(ip_off & IP_FLAGS_MFRAG)) { 783 /* no more fragmentss: truncate this (last) hole */ 784 total_len = start + len; 785 h->last_byte = start + len; 786 } 787 788 /* 789 * There is some overlap: fix the hole list. This code doesn't 790 * deal with a fragment that overlaps with two different holes 791 * (thus being a superset of a previously-received fragment). 792 */ 793 794 if ((h >= thisfrag) && (h->last_byte <= start + len)) { 795 /* complete overlap with hole: remove hole */ 796 if (!h->prev_hole && !h->next_hole) { 797 /* last remaining hole */ 798 done = 1; 799 } else if (!h->prev_hole) { 800 /* first hole */ 801 first_hole = h->next_hole; 802 payload[h->next_hole].prev_hole = 0; 803 } else if (!h->next_hole) { 804 /* last hole */ 805 payload[h->prev_hole].next_hole = 0; 806 } else { 807 /* in the middle of the list */ 808 payload[h->next_hole].prev_hole = h->prev_hole; 809 payload[h->prev_hole].next_hole = h->next_hole; 810 } 811 812 } else if (h->last_byte <= start + len) { 813 /* overlaps with final part of the hole: shorten this hole */ 814 h->last_byte = start; 815 816 } else if (h >= thisfrag) { 817 /* overlaps with initial part of the hole: move this hole */ 818 newh = thisfrag + (len / 8); 819 *newh = *h; 820 h = newh; 821 if (h->next_hole) 822 payload[h->next_hole].prev_hole = (h - payload); 823 if (h->prev_hole) 824 payload[h->prev_hole].next_hole = (h - payload); 825 else 826 first_hole = (h - payload); 827 828 } else { 829 /* fragment sits in the middle: split the hole */ 830 newh = thisfrag + (len / 8); 831 *newh = *h; 832 h->last_byte = start; 833 h->next_hole = (newh - payload); 834 newh->prev_hole = (h - payload); 835 if (newh->next_hole) 836 payload[newh->next_hole].prev_hole = (newh - payload); 837 } 838 839 /* finally copy this fragment and possibly return whole packet */ 840 memcpy((uchar *)thisfrag, indata + IP_HDR_SIZE, len); 841 if (!done) 842 return NULL; 843 844 localip->ip_len = htons(total_len); 845 *lenp = total_len + IP_HDR_SIZE; 846 return localip; 847 } 848 849 static inline struct ip_udp_hdr *NetDefragment(struct ip_udp_hdr *ip, int *lenp) 850 { 851 u16 ip_off = ntohs(ip->ip_off); 852 if (!(ip_off & (IP_OFFS | IP_FLAGS_MFRAG))) 853 return ip; /* not a fragment */ 854 return __NetDefragment(ip, lenp); 855 } 856 857 #else /* !CONFIG_IP_DEFRAG */ 858 859 static inline struct ip_udp_hdr *NetDefragment(struct ip_udp_hdr *ip, int *lenp) 860 { 861 u16 ip_off = ntohs(ip->ip_off); 862 if (!(ip_off & (IP_OFFS | IP_FLAGS_MFRAG))) 863 return ip; /* not a fragment */ 864 return NULL; 865 } 866 #endif 867 868 /** 869 * Receive an ICMP packet. We deal with REDIRECT and PING here, and silently 870 * drop others. 871 * 872 * @parma ip IP packet containing the ICMP 873 */ 874 static void receive_icmp(struct ip_udp_hdr *ip, int len, 875 IPaddr_t src_ip, struct ethernet_hdr *et) 876 { 877 struct icmp_hdr *icmph = (struct icmp_hdr *)&ip->udp_src; 878 879 switch (icmph->type) { 880 case ICMP_REDIRECT: 881 if (icmph->code != ICMP_REDIR_HOST) 882 return; 883 printf(" ICMP Host Redirect to %pI4 ", 884 &icmph->un.gateway); 885 break; 886 default: 887 #if defined(CONFIG_CMD_PING) 888 ping_receive(et, ip, len); 889 #endif 890 #ifdef CONFIG_CMD_TFTPPUT 891 if (packet_icmp_handler) 892 packet_icmp_handler(icmph->type, icmph->code, 893 ntohs(ip->udp_dst), src_ip, ntohs(ip->udp_src), 894 icmph->un.data, ntohs(ip->udp_len)); 895 #endif 896 break; 897 } 898 } 899 900 void 901 NetReceive(uchar *inpkt, int len) 902 { 903 struct ethernet_hdr *et; 904 struct ip_udp_hdr *ip; 905 IPaddr_t dst_ip; 906 IPaddr_t src_ip; 907 int eth_proto; 908 #if defined(CONFIG_CMD_CDP) 909 int iscdp; 910 #endif 911 ushort cti = 0, vlanid = VLAN_NONE, myvlanid, mynvlanid; 912 913 debug("packet received\n"); 914 915 NetRxPacket = inpkt; 916 NetRxPacketLen = len; 917 et = (struct ethernet_hdr *)inpkt; 918 919 /* too small packet? */ 920 if (len < ETHER_HDR_SIZE) 921 return; 922 923 #ifdef CONFIG_API 924 if (push_packet) { 925 (*push_packet)(inpkt, len); 926 return; 927 } 928 #endif 929 930 #if defined(CONFIG_CMD_CDP) 931 /* keep track if packet is CDP */ 932 iscdp = is_cdp_packet(et->et_dest); 933 #endif 934 935 myvlanid = ntohs(NetOurVLAN); 936 if (myvlanid == (ushort)-1) 937 myvlanid = VLAN_NONE; 938 mynvlanid = ntohs(NetOurNativeVLAN); 939 if (mynvlanid == (ushort)-1) 940 mynvlanid = VLAN_NONE; 941 942 eth_proto = ntohs(et->et_protlen); 943 944 debug("packet received\n"); 945 946 if (eth_proto < 1514) { 947 struct e802_hdr *et802 = (struct e802_hdr *)et; 948 /* 949 * Got a 802.2 packet. Check the other protocol field. 950 * XXX VLAN over 802.2+SNAP not implemented! 951 */ 952 eth_proto = ntohs(et802->et_prot); 953 954 ip = (struct ip_udp_hdr *)(inpkt + E802_HDR_SIZE); 955 len -= E802_HDR_SIZE; 956 957 } else if (eth_proto != PROT_VLAN) { /* normal packet */ 958 ip = (struct ip_udp_hdr *)(inpkt + ETHER_HDR_SIZE); 959 len -= ETHER_HDR_SIZE; 960 961 } else { /* VLAN packet */ 962 struct vlan_ethernet_hdr *vet = 963 (struct vlan_ethernet_hdr *)et; 964 965 debug("VLAN packet received\n"); 966 967 /* too small packet? */ 968 if (len < VLAN_ETHER_HDR_SIZE) 969 return; 970 971 /* if no VLAN active */ 972 if ((ntohs(NetOurVLAN) & VLAN_IDMASK) == VLAN_NONE 973 #if defined(CONFIG_CMD_CDP) 974 && iscdp == 0 975 #endif 976 ) 977 return; 978 979 cti = ntohs(vet->vet_tag); 980 vlanid = cti & VLAN_IDMASK; 981 eth_proto = ntohs(vet->vet_type); 982 983 ip = (struct ip_udp_hdr *)(inpkt + VLAN_ETHER_HDR_SIZE); 984 len -= VLAN_ETHER_HDR_SIZE; 985 } 986 987 debug("Receive from protocol 0x%x\n", eth_proto); 988 989 #if defined(CONFIG_CMD_CDP) 990 if (iscdp) { 991 cdp_receive((uchar *)ip, len); 992 return; 993 } 994 #endif 995 996 if ((myvlanid & VLAN_IDMASK) != VLAN_NONE) { 997 if (vlanid == VLAN_NONE) 998 vlanid = (mynvlanid & VLAN_IDMASK); 999 /* not matched? */ 1000 if (vlanid != (myvlanid & VLAN_IDMASK)) 1001 return; 1002 } 1003 1004 switch (eth_proto) { 1005 1006 case PROT_ARP: 1007 ArpReceive(et, ip, len); 1008 break; 1009 1010 #ifdef CONFIG_CMD_RARP 1011 case PROT_RARP: 1012 rarp_receive(ip, len); 1013 break; 1014 #endif 1015 case PROT_IP: 1016 debug("Got IP\n"); 1017 /* Before we start poking the header, make sure it is there */ 1018 if (len < IP_UDP_HDR_SIZE) { 1019 debug("len bad %d < %lu\n", len, 1020 (ulong)IP_UDP_HDR_SIZE); 1021 return; 1022 } 1023 /* Check the packet length */ 1024 if (len < ntohs(ip->ip_len)) { 1025 printf("len bad %d < %d\n", len, ntohs(ip->ip_len)); 1026 return; 1027 } 1028 len = ntohs(ip->ip_len); 1029 debug("len=%d, v=%02x\n", len, ip->ip_hl_v & 0xff); 1030 1031 /* Can't deal with anything except IPv4 */ 1032 if ((ip->ip_hl_v & 0xf0) != 0x40) 1033 return; 1034 /* Can't deal with IP options (headers != 20 bytes) */ 1035 if ((ip->ip_hl_v & 0x0f) > 0x05) 1036 return; 1037 /* Check the Checksum of the header */ 1038 if (!NetCksumOk((uchar *)ip, IP_HDR_SIZE / 2)) { 1039 puts("checksum bad\n"); 1040 return; 1041 } 1042 /* If it is not for us, ignore it */ 1043 dst_ip = NetReadIP(&ip->ip_dst); 1044 if (NetOurIP && dst_ip != NetOurIP && dst_ip != 0xFFFFFFFF) { 1045 #ifdef CONFIG_MCAST_TFTP 1046 if (Mcast_addr != dst_ip) 1047 #endif 1048 return; 1049 } 1050 /* Read source IP address for later use */ 1051 src_ip = NetReadIP(&ip->ip_src); 1052 /* 1053 * The function returns the unchanged packet if it's not 1054 * a fragment, and either the complete packet or NULL if 1055 * it is a fragment (if !CONFIG_IP_DEFRAG, it returns NULL) 1056 */ 1057 ip = NetDefragment(ip, &len); 1058 if (!ip) 1059 return; 1060 /* 1061 * watch for ICMP host redirects 1062 * 1063 * There is no real handler code (yet). We just watch 1064 * for ICMP host redirect messages. In case anybody 1065 * sees these messages: please contact me 1066 * (wd@denx.de), or - even better - send me the 1067 * necessary fixes :-) 1068 * 1069 * Note: in all cases where I have seen this so far 1070 * it was a problem with the router configuration, 1071 * for instance when a router was configured in the 1072 * BOOTP reply, but the TFTP server was on the same 1073 * subnet. So this is probably a warning that your 1074 * configuration might be wrong. But I'm not really 1075 * sure if there aren't any other situations. 1076 * 1077 * Simon Glass <sjg@chromium.org>: We get an ICMP when 1078 * we send a tftp packet to a dead connection, or when 1079 * there is no server at the other end. 1080 */ 1081 if (ip->ip_p == IPPROTO_ICMP) { 1082 receive_icmp(ip, len, src_ip, et); 1083 return; 1084 } else if (ip->ip_p != IPPROTO_UDP) { /* Only UDP packets */ 1085 return; 1086 } 1087 1088 #ifdef CONFIG_UDP_CHECKSUM 1089 if (ip->udp_xsum != 0) { 1090 ulong xsum; 1091 ushort *sumptr; 1092 ushort sumlen; 1093 1094 xsum = ip->ip_p; 1095 xsum += (ntohs(ip->udp_len)); 1096 xsum += (ntohl(ip->ip_src) >> 16) & 0x0000ffff; 1097 xsum += (ntohl(ip->ip_src) >> 0) & 0x0000ffff; 1098 xsum += (ntohl(ip->ip_dst) >> 16) & 0x0000ffff; 1099 xsum += (ntohl(ip->ip_dst) >> 0) & 0x0000ffff; 1100 1101 sumlen = ntohs(ip->udp_len); 1102 sumptr = (ushort *) &(ip->udp_src); 1103 1104 while (sumlen > 1) { 1105 ushort sumdata; 1106 1107 sumdata = *sumptr++; 1108 xsum += ntohs(sumdata); 1109 sumlen -= 2; 1110 } 1111 if (sumlen > 0) { 1112 ushort sumdata; 1113 1114 sumdata = *(unsigned char *) sumptr; 1115 sumdata = (sumdata << 8) & 0xff00; 1116 xsum += sumdata; 1117 } 1118 while ((xsum >> 16) != 0) { 1119 xsum = (xsum & 0x0000ffff) + 1120 ((xsum >> 16) & 0x0000ffff); 1121 } 1122 if ((xsum != 0x00000000) && (xsum != 0x0000ffff)) { 1123 printf(" UDP wrong checksum %08lx %08x\n", 1124 xsum, ntohs(ip->udp_xsum)); 1125 return; 1126 } 1127 } 1128 #endif 1129 1130 1131 #ifdef CONFIG_NETCONSOLE 1132 nc_input_packet((uchar *)ip + IP_UDP_HDR_SIZE, 1133 ntohs(ip->udp_dst), 1134 ntohs(ip->udp_src), 1135 ntohs(ip->udp_len) - UDP_HDR_SIZE); 1136 #endif 1137 /* 1138 * IP header OK. Pass the packet to the current handler. 1139 */ 1140 (*udp_packet_handler)((uchar *)ip + IP_UDP_HDR_SIZE, 1141 ntohs(ip->udp_dst), 1142 src_ip, 1143 ntohs(ip->udp_src), 1144 ntohs(ip->udp_len) - UDP_HDR_SIZE); 1145 break; 1146 } 1147 } 1148 1149 1150 /**********************************************************************/ 1151 1152 static int net_check_prereq(enum proto_t protocol) 1153 { 1154 switch (protocol) { 1155 /* Fall through */ 1156 #if defined(CONFIG_CMD_PING) 1157 case PING: 1158 if (NetPingIP == 0) { 1159 puts("*** ERROR: ping address not given\n"); 1160 return 1; 1161 } 1162 goto common; 1163 #endif 1164 #if defined(CONFIG_CMD_SNTP) 1165 case SNTP: 1166 if (NetNtpServerIP == 0) { 1167 puts("*** ERROR: NTP server address not given\n"); 1168 return 1; 1169 } 1170 goto common; 1171 #endif 1172 #if defined(CONFIG_CMD_DNS) 1173 case DNS: 1174 if (NetOurDNSIP == 0) { 1175 puts("*** ERROR: DNS server address not given\n"); 1176 return 1; 1177 } 1178 goto common; 1179 #endif 1180 #if defined(CONFIG_CMD_NFS) 1181 case NFS: 1182 #endif 1183 case TFTPGET: 1184 case TFTPPUT: 1185 if (NetServerIP == 0) { 1186 puts("*** ERROR: `serverip' not set\n"); 1187 return 1; 1188 } 1189 #if defined(CONFIG_CMD_PING) || defined(CONFIG_CMD_SNTP) || \ 1190 defined(CONFIG_CMD_DNS) 1191 common: 1192 #endif 1193 /* Fall through */ 1194 1195 case NETCONS: 1196 case TFTPSRV: 1197 if (NetOurIP == 0) { 1198 puts("*** ERROR: `ipaddr' not set\n"); 1199 return 1; 1200 } 1201 /* Fall through */ 1202 1203 #ifdef CONFIG_CMD_RARP 1204 case RARP: 1205 #endif 1206 case BOOTP: 1207 case CDP: 1208 case DHCP: 1209 if (memcmp(NetOurEther, "\0\0\0\0\0\0", 6) == 0) { 1210 int num = eth_get_dev_index(); 1211 1212 switch (num) { 1213 case -1: 1214 puts("*** ERROR: No ethernet found.\n"); 1215 return 1; 1216 case 0: 1217 puts("*** ERROR: `ethaddr' not set\n"); 1218 break; 1219 default: 1220 printf("*** ERROR: `eth%daddr' not set\n", 1221 num); 1222 break; 1223 } 1224 1225 NetStartAgain(); 1226 return 2; 1227 } 1228 /* Fall through */ 1229 default: 1230 return 0; 1231 } 1232 return 0; /* OK */ 1233 } 1234 /**********************************************************************/ 1235 1236 int 1237 NetCksumOk(uchar *ptr, int len) 1238 { 1239 return !((NetCksum(ptr, len) + 1) & 0xfffe); 1240 } 1241 1242 1243 unsigned 1244 NetCksum(uchar *ptr, int len) 1245 { 1246 ulong xsum; 1247 ushort *p = (ushort *)ptr; 1248 1249 xsum = 0; 1250 while (len-- > 0) 1251 xsum += *p++; 1252 xsum = (xsum & 0xffff) + (xsum >> 16); 1253 xsum = (xsum & 0xffff) + (xsum >> 16); 1254 return xsum & 0xffff; 1255 } 1256 1257 int 1258 NetEthHdrSize(void) 1259 { 1260 ushort myvlanid; 1261 1262 myvlanid = ntohs(NetOurVLAN); 1263 if (myvlanid == (ushort)-1) 1264 myvlanid = VLAN_NONE; 1265 1266 return ((myvlanid & VLAN_IDMASK) == VLAN_NONE) ? ETHER_HDR_SIZE : 1267 VLAN_ETHER_HDR_SIZE; 1268 } 1269 1270 int 1271 NetSetEther(uchar *xet, uchar * addr, uint prot) 1272 { 1273 struct ethernet_hdr *et = (struct ethernet_hdr *)xet; 1274 ushort myvlanid; 1275 1276 myvlanid = ntohs(NetOurVLAN); 1277 if (myvlanid == (ushort)-1) 1278 myvlanid = VLAN_NONE; 1279 1280 memcpy(et->et_dest, addr, 6); 1281 memcpy(et->et_src, NetOurEther, 6); 1282 if ((myvlanid & VLAN_IDMASK) == VLAN_NONE) { 1283 et->et_protlen = htons(prot); 1284 return ETHER_HDR_SIZE; 1285 } else { 1286 struct vlan_ethernet_hdr *vet = 1287 (struct vlan_ethernet_hdr *)xet; 1288 1289 vet->vet_vlan_type = htons(PROT_VLAN); 1290 vet->vet_tag = htons((0 << 5) | (myvlanid & VLAN_IDMASK)); 1291 vet->vet_type = htons(prot); 1292 return VLAN_ETHER_HDR_SIZE; 1293 } 1294 } 1295 1296 int net_update_ether(struct ethernet_hdr *et, uchar *addr, uint prot) 1297 { 1298 ushort protlen; 1299 1300 memcpy(et->et_dest, addr, 6); 1301 memcpy(et->et_src, NetOurEther, 6); 1302 protlen = ntohs(et->et_protlen); 1303 if (protlen == PROT_VLAN) { 1304 struct vlan_ethernet_hdr *vet = 1305 (struct vlan_ethernet_hdr *)et; 1306 vet->vet_type = htons(prot); 1307 return VLAN_ETHER_HDR_SIZE; 1308 } else if (protlen > 1514) { 1309 et->et_protlen = htons(prot); 1310 return ETHER_HDR_SIZE; 1311 } else { 1312 /* 802.2 + SNAP */ 1313 struct e802_hdr *et802 = (struct e802_hdr *)et; 1314 et802->et_prot = htons(prot); 1315 return E802_HDR_SIZE; 1316 } 1317 } 1318 1319 void net_set_ip_header(uchar *pkt, IPaddr_t dest, IPaddr_t source) 1320 { 1321 struct ip_udp_hdr *ip = (struct ip_udp_hdr *)pkt; 1322 1323 /* 1324 * Construct an IP header. 1325 */ 1326 /* IP_HDR_SIZE / 4 (not including UDP) */ 1327 ip->ip_hl_v = 0x45; 1328 ip->ip_tos = 0; 1329 ip->ip_len = htons(IP_HDR_SIZE); 1330 ip->ip_id = htons(NetIPID++); 1331 ip->ip_off = htons(IP_FLAGS_DFRAG); /* Don't fragment */ 1332 ip->ip_ttl = 255; 1333 ip->ip_sum = 0; 1334 /* already in network byte order */ 1335 NetCopyIP((void *)&ip->ip_src, &source); 1336 /* already in network byte order */ 1337 NetCopyIP((void *)&ip->ip_dst, &dest); 1338 } 1339 1340 void net_set_udp_header(uchar *pkt, IPaddr_t dest, int dport, int sport, 1341 int len) 1342 { 1343 struct ip_udp_hdr *ip = (struct ip_udp_hdr *)pkt; 1344 1345 /* 1346 * If the data is an odd number of bytes, zero the 1347 * byte after the last byte so that the checksum 1348 * will work. 1349 */ 1350 if (len & 1) 1351 pkt[IP_UDP_HDR_SIZE + len] = 0; 1352 1353 net_set_ip_header(pkt, dest, NetOurIP); 1354 ip->ip_len = htons(IP_UDP_HDR_SIZE + len); 1355 ip->ip_p = IPPROTO_UDP; 1356 ip->ip_sum = ~NetCksum((uchar *)ip, IP_HDR_SIZE >> 1); 1357 1358 ip->udp_src = htons(sport); 1359 ip->udp_dst = htons(dport); 1360 ip->udp_len = htons(UDP_HDR_SIZE + len); 1361 ip->udp_xsum = 0; 1362 } 1363 1364 void copy_filename(char *dst, const char *src, int size) 1365 { 1366 if (*src && (*src == '"')) { 1367 ++src; 1368 --size; 1369 } 1370 1371 while ((--size > 0) && *src && (*src != '"')) 1372 *dst++ = *src++; 1373 *dst = '\0'; 1374 } 1375 1376 #if defined(CONFIG_CMD_NFS) || \ 1377 defined(CONFIG_CMD_SNTP) || \ 1378 defined(CONFIG_CMD_DNS) 1379 /* 1380 * make port a little random (1024-17407) 1381 * This keeps the math somewhat trivial to compute, and seems to work with 1382 * all supported protocols/clients/servers 1383 */ 1384 unsigned int random_port(void) 1385 { 1386 return 1024 + (get_timer(0) % 0x4000); 1387 } 1388 #endif 1389 1390 void ip_to_string(IPaddr_t x, char *s) 1391 { 1392 x = ntohl(x); 1393 sprintf(s, "%d.%d.%d.%d", 1394 (int) ((x >> 24) & 0xff), 1395 (int) ((x >> 16) & 0xff), 1396 (int) ((x >> 8) & 0xff), (int) ((x >> 0) & 0xff) 1397 ); 1398 } 1399 1400 void VLAN_to_string(ushort x, char *s) 1401 { 1402 x = ntohs(x); 1403 1404 if (x == (ushort)-1) 1405 x = VLAN_NONE; 1406 1407 if (x == VLAN_NONE) 1408 strcpy(s, "none"); 1409 else 1410 sprintf(s, "%d", x & VLAN_IDMASK); 1411 } 1412 1413 ushort string_to_VLAN(const char *s) 1414 { 1415 ushort id; 1416 1417 if (s == NULL) 1418 return htons(VLAN_NONE); 1419 1420 if (*s < '0' || *s > '9') 1421 id = VLAN_NONE; 1422 else 1423 id = (ushort)simple_strtoul(s, NULL, 10); 1424 1425 return htons(id); 1426 } 1427 1428 ushort getenv_VLAN(char *var) 1429 { 1430 return string_to_VLAN(getenv(var)); 1431 } 1432