net/net.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  *      Copied from Linux Monitor (LiMon) - Networking.
   4  *
   5  *      Copyright 1994 - 2000 Neil Russell.
   6  *      (See License)
   7  *      Copyright 2000 Roland Borde
   8  *      Copyright 2000 Paolo Scaffardi
   9  *      Copyright 2000-2002 Wolfgang Denk, [email protected]
  10  */
  11
  12 /*
  13  * General Desription:
  14  *
  15  * The user interface supports commands for BOOTP, RARP, and TFTP.
  16  * Also, we support ARP internally. Depending on available data,
  17  * these interact as follows:
  18  *
  19  * BOOTP:
  20  *
  21  *      Prerequisites:  - own ethernet address
  22  *      We want:        - own IP address
  23  *                      - TFTP server IP address
  24  *                      - name of bootfile
  25  *      Next step:      ARP
  26  *
  27  * LINKLOCAL:
  28  *
  29  *      Prerequisites:  - own ethernet address
  30  *      We want:        - own IP address
  31  *      Next step:      ARP
  32  *
  33  * RARP:
  34  *
  35  *      Prerequisites:  - own ethernet address
  36  *      We want:        - own IP address
  37  *                      - TFTP server IP address
  38  *      Next step:      ARP
  39  *
  40  * ARP:
  41  *
  42  *      Prerequisites:  - own ethernet address
  43  *                      - own IP address
  44  *                      - TFTP server IP address
  45  *      We want:        - TFTP server ethernet address
  46  *      Next step:      TFTP
  47  *
  48  * DHCP:
  49  *
  50  *     Prerequisites:   - own ethernet address
  51  *     We want:         - IP, Netmask, ServerIP, Gateway IP
  52  *                      - bootfilename, lease time
  53  *     Next step:       - TFTP
  54  *
  55  * TFTP:
  56  *
  57  *      Prerequisites:  - own ethernet address
  58  *                      - own IP address
  59  *                      - TFTP server IP address
  60  *                      - TFTP server ethernet address
  61  *                      - name of bootfile (if unknown, we use a default name
  62  *                        derived from our own IP address)
  63  *      We want:        - load the boot file
  64  *      Next step:      none
  65  *
  66  * NFS:
  67  *
  68  *      Prerequisites:  - own ethernet address
  69  *                      - own IP address
  70  *                      - name of bootfile (if unknown, we use a default name
  71  *                        derived from our own IP address)
  72  *      We want:        - load the boot file
  73  *      Next step:      none
  74  *
  75  *
  76  * WOL:
  77  *
  78  *      Prerequisites:  - own ethernet address
  79  *      We want:        - magic packet or timeout
  80  *      Next step:      none
  81  */
  82
  83
  84 #include <common.h>
  85 #include <bootstage.h>
  86 #include <command.h>
  87 #include <console.h>
  88 #include <env.h>
  89 #include <env_internal.h>
  90 #include <errno.h>
  91 #include <image.h>
  92 #include <log.h>
  93 #include <net.h>
  94 #include <net6.h>
  95 #include <ndisc.h>
  96 #include <net/fastboot_udp.h>
  97 #include <net/fastboot_tcp.h>
  98 #include <net/tftp.h>
  99 #include <net/ncsi.h>
 100 #if defined(CONFIG_CMD_PCAP)
 101 #include <net/pcap.h>
 102 #endif
 103 #include <net/udp.h>
 104 #if defined(CONFIG_LED_STATUS)
 105 #include <miiphy.h>
 106 #include <status_led.h>
 107 #endif
 108 #include <watchdog.h>
 109 #include <linux/compiler.h>
 110 #include <test/test.h>
 111 #include <net/tcp.h>
 112 #include <net/wget.h>
 113 #include "arp.h"
 114 #include "bootp.h"
 115 #include "cdp.h"
 116 #if defined(CONFIG_CMD_DNS)
 117 #include "dns.h"
 118 #endif
 119 #include "link_local.h"
 120 #include "nfs.h"
 121 #include "ping.h"
 122 #include "rarp.h"
 123 #if defined(CONFIG_CMD_WOL)
 124 #include "wol.h"
 125 #endif
 126 #include "dhcpv6.h"
 127 #include "net_rand.h"
 128
 129 /** BOOTP EXTENTIONS **/
 130
 131 /* Our subnet mask (0=unknown) */
 132 struct in_addr net_netmask;
 133 /* Our gateways IP address */
 134 struct in_addr net_gateway;
 135 /* Our DNS IP address */
 136 struct in_addr net_dns_server;
 137 #if defined(CONFIG_BOOTP_DNS2)
 138 /* Our 2nd DNS IP address */
 139 struct in_addr net_dns_server2;
 140 #endif
 141 /* Indicates whether the pxe path prefix / config file was specified in dhcp option */
 142 char *pxelinux_configfile;
 143
 144 /** END OF BOOTP EXTENTIONS **/
 145
 146 /* Our ethernet address */
 147 u8 net_ethaddr[6];
 148 /* Boot server enet address */
 149 u8 net_server_ethaddr[6];
 150 /* Our IP addr (0 = unknown) */
 151 struct in_addr  net_ip;
 152 /* Server IP addr (0 = unknown) */
 153 struct in_addr  net_server_ip;
 154 /* Current receive packet */
 155 uchar *net_rx_packet;
 156 /* Current rx packet length */
 157 int             net_rx_packet_len;
 158 /* IP packet ID */
 159 static unsigned net_ip_id;
 160 /* Ethernet bcast address */
 161 const u8 net_bcast_ethaddr[6] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff };
 162 const u8 net_null_ethaddr[6];
 163 #if defined(CONFIG_API) || defined(CONFIG_EFI_LOADER)
 164 void (*push_packet)(void *, int len) = 0;
 165 #endif
 166 /* Network loop state */
 167 enum net_loop_state net_state;
 168 /* Tried all network devices */
 169 int             net_restart_wrap;
 170 /* Network loop restarted */
 171 static int      net_restarted;
 172 /* At least one device configured */
 173 static int      net_dev_exists;
 174
 175 /* XXX in both little & big endian machines 0xFFFF == ntohs(-1) */
 176 /* default is without VLAN */
 177 ushort          net_our_vlan = 0xFFFF;
 178 /* ditto */
 179 ushort          net_native_vlan = 0xFFFF;
 180
 181 /* Boot File name */
 182 char net_boot_file_name[1024];
 183 /* Indicates whether the file name was specified on the command line */
 184 bool net_boot_file_name_explicit;
 185 /* The actual transferred size of the bootfile (in bytes) */
 186 u32 net_boot_file_size;
 187 /* Boot file size in blocks as reported by the DHCP server */
 188 u32 net_boot_file_expected_size_in_blocks;
 189
 190 static uchar net_pkt_buf[(PKTBUFSRX+1) * PKTSIZE_ALIGN + PKTALIGN];
 191 /* Receive packets */
 192 uchar *net_rx_packets[PKTBUFSRX];
 193 /* Current UDP RX packet handler */
 194 static rxhand_f *udp_packet_handler;
 195 /* Current ARP RX packet handler */
 196 static rxhand_f *arp_packet_handler;
 197 #ifdef CONFIG_CMD_TFTPPUT
 198 /* Current ICMP rx handler */
 199 static rxhand_icmp_f *packet_icmp_handler;
 200 #endif
 201 /* Current timeout handler */
 202 static thand_f *time_handler;
 203 /* Time base value */
 204 static ulong    time_start;
 205 /* Current timeout value */
 206 static ulong    time_delta;
 207 /* THE transmit packet */
 208 uchar *net_tx_packet;
 209
 210 static int net_check_prereq(enum proto_t protocol);
 211
 212 static int net_try_count;
 213
 214 int __maybe_unused net_busy_flag;
 215
 216 /**********************************************************************/
 217
 218 static int on_ipaddr(const char *name, const char *value, enum env_op op,
 219         int flags)
 220 {
 221         if (flags & H_PROGRAMMATIC)
 222                 return 0;
 223
 224         net_ip = string_to_ip(value);
 225
 226         return 0;
 227 }
 228 U_BOOT_ENV_CALLBACK(ipaddr, on_ipaddr);
 229
 230 static int on_gatewayip(const char *name, const char *value, enum env_op op,
 231         int flags)
 232 {
 233         if (flags & H_PROGRAMMATIC)
 234                 return 0;
 235
 236         net_gateway = string_to_ip(value);
 237
 238         return 0;
 239 }
 240 U_BOOT_ENV_CALLBACK(gatewayip, on_gatewayip);
 241
 242 static int on_netmask(const char *name, const char *value, enum env_op op,
 243         int flags)
 244 {
 245         if (flags & H_PROGRAMMATIC)
 246                 return 0;
 247
 248         net_netmask = string_to_ip(value);
 249
 250         return 0;
 251 }
 252 U_BOOT_ENV_CALLBACK(netmask, on_netmask);
 253
 254 static int on_serverip(const char *name, const char *value, enum env_op op,
 255         int flags)
 256 {
 257         if (flags & H_PROGRAMMATIC)
 258                 return 0;
 259
 260         net_server_ip = string_to_ip(value);
 261
 262         return 0;
 263 }
 264 U_BOOT_ENV_CALLBACK(serverip, on_serverip);
 265
 266 static int on_nvlan(const char *name, const char *value, enum env_op op,
 267         int flags)
 268 {
 269         if (flags & H_PROGRAMMATIC)
 270                 return 0;
 271
 272         net_native_vlan = string_to_vlan(value);
 273
 274         return 0;
 275 }
 276 U_BOOT_ENV_CALLBACK(nvlan, on_nvlan);
 277
 278 static int on_vlan(const char *name, const char *value, enum env_op op,
 279         int flags)
 280 {
 281         if (flags & H_PROGRAMMATIC)
 282                 return 0;
 283
 284         net_our_vlan = string_to_vlan(value);
 285
 286         return 0;
 287 }
 288 U_BOOT_ENV_CALLBACK(vlan, on_vlan);
 289
 290 #if defined(CONFIG_CMD_DNS)
 291 static int on_dnsip(const char *name, const char *value, enum env_op op,
 292         int flags)
 293 {
 294         if (flags & H_PROGRAMMATIC)
 295                 return 0;
 296
 297         net_dns_server = string_to_ip(value);
 298
 299         return 0;
 300 }
 301 U_BOOT_ENV_CALLBACK(dnsip, on_dnsip);
 302 #endif
 303
 304 /*
 305  * Check if autoload is enabled. If so, use either NFS or TFTP to download
 306  * the boot file.
 307  */
 308 void net_auto_load(void)
 309 {
 310 #if defined(CONFIG_CMD_NFS) && !defined(CONFIG_SPL_BUILD)
 311         const char *s = env_get("autoload");
 312
 313         if (s != NULL && strcmp(s, "NFS") == 0) {
 314                 if (net_check_prereq(NFS)) {
 315 /* We aren't expecting to get a serverip, so just accept the assigned IP */
 316                         if (IS_ENABLED(CONFIG_BOOTP_SERVERIP)) {
 317                                 net_set_state(NETLOOP_SUCCESS);
 318                         } else {
 319                                 printf("Cannot autoload with NFS\n");
 320                                 net_set_state(NETLOOP_FAIL);
 321                         }
 322                         return;
 323                 }
 324                 /*
 325                  * Use NFS to load the bootfile.
 326                  */
 327                 nfs_start();
 328                 return;
 329         }
 330 #endif
 331         if (env_get_yesno("autoload") == 0) {
 332                 /*
 333                  * Just use BOOTP/RARP to configure system;
 334                  * Do not use TFTP to load the bootfile.
 335                  */
 336                 net_set_state(NETLOOP_SUCCESS);
 337                 return;
 338         }
 339         if (net_check_prereq(TFTPGET)) {
 340 /* We aren't expecting to get a serverip, so just accept the assigned IP */
 341                 if (IS_ENABLED(CONFIG_BOOTP_SERVERIP)) {
 342                         net_set_state(NETLOOP_SUCCESS);
 343                 } else {
 344                         printf("Cannot autoload with TFTPGET\n");
 345                         net_set_state(NETLOOP_FAIL);
 346                 }
 347                 return;
 348         }
 349         tftp_start(TFTPGET);
 350 }
 351
 352 static int net_init_loop(void)
 353 {
 354         static bool first_call = true;
 355
 356         if (eth_get_dev()) {
 357                 memcpy(net_ethaddr, eth_get_ethaddr(), 6);
 358
 359                 if (IS_ENABLED(CONFIG_IPV6)) {
 360                         ip6_make_lladdr(&net_link_local_ip6, net_ethaddr);
 361                         if (!memcmp(&net_ip6, &net_null_addr_ip6,
 362                                     sizeof(struct in6_addr)))
 363                                 memcpy(&net_ip6, &net_link_local_ip6,
 364                                        sizeof(struct in6_addr));
 365                 }
 366         }
 367         else
 368                 /*
 369                  * Not ideal, but there's no way to get the actual error, and I
 370                  * don't feel like fixing all the users of eth_get_dev to deal
 371                  * with errors.
 372                  */
 373                 return -ENONET;
 374
 375         if (IS_ENABLED(CONFIG_IPV6_ROUTER_DISCOVERY))
 376                 if (first_call && use_ip6) {
 377                         first_call = false;
 378                         srand_mac(); /* This is for rand used in ip6_send_rs. */
 379                         net_loop(RS);
 380                 }
 381         return 0;
 382 }
 383
 384 static void net_clear_handlers(void)
 385 {
 386         net_set_udp_handler(NULL);
 387         net_set_arp_handler(NULL);
 388         net_set_timeout_handler(0, NULL);
 389 }
 390
 391 static void net_cleanup_loop(void)
 392 {
 393         net_clear_handlers();
 394 }
 395
 396 int net_init(void)
 397 {
 398         static int first_call = 1;
 399
 400         if (first_call) {
 401                 /*
 402                  *      Setup packet buffers, aligned correctly.
 403                  */
 404                 int i;
 405
 406                 net_tx_packet = &net_pkt_buf[0] + (PKTALIGN - 1);
 407                 net_tx_packet -= (ulong)net_tx_packet % PKTALIGN;
 408                 for (i = 0; i < PKTBUFSRX; i++) {
 409                         net_rx_packets[i] = net_tx_packet +
 410                                 (i + 1) * PKTSIZE_ALIGN;
 411                 }
 412                 arp_init();
 413                 ndisc_init();
 414                 net_clear_handlers();
 415
 416                 /* Only need to setup buffer pointers once. */
 417                 first_call = 0;
 418                 if (IS_ENABLED(CONFIG_PROT_TCP))
 419                         tcp_set_tcp_state(TCP_CLOSED);
 420         }
 421
 422         return net_init_loop();
 423 }
 424
 425 /**********************************************************************/
 426 /*
 427  *      Main network processing loop.
 428  */
 429
 430 int net_loop(enum proto_t protocol)
 431 {
 432         int ret = -EINVAL;
 433         enum net_loop_state prev_net_state = net_state;
 434
 435 #if defined(CONFIG_CMD_PING)
 436         if (protocol != PING)
 437                 net_ping_ip.s_addr = 0;
 438 #endif
 439         net_restarted = 0;
 440         net_dev_exists = 0;
 441         net_try_count = 1;
 442         debug_cond(DEBUG_INT_STATE, "--- net_loop Entry\n");
 443
 444 #ifdef CONFIG_PHY_NCSI
 445         if (phy_interface_is_ncsi() && protocol != NCSI && !ncsi_active()) {
 446                 printf("%s: configuring NCSI first\n", __func__);
 447                 if (net_loop(NCSI) < 0)
 448                         return ret;
 449                 eth_init_state_only();
 450                 goto restart;
 451         }
 452 #endif
 453
 454         bootstage_mark_name(BOOTSTAGE_ID_ETH_START, "eth_start");
 455         net_init();
 456         if (eth_is_on_demand_init()) {
 457                 eth_halt();
 458                 eth_set_current();
 459                 ret = eth_init();
 460                 if (ret < 0) {
 461                         eth_halt();
 462                         return ret;
 463                 }
 464         } else {
 465                 eth_init_state_only();
 466         }
 467
 468 restart:
 469 #ifdef CONFIG_USB_KEYBOARD
 470         net_busy_flag = 0;
 471 #endif
 472         net_set_state(NETLOOP_CONTINUE);
 473
 474         /*
 475          *      Start the ball rolling with the given start function.  From
 476          *      here on, this code is a state machine driven by received
 477          *      packets and timer events.
 478          */
 479         debug_cond(DEBUG_INT_STATE, "--- net_loop Init\n");
 480         net_init_loop();
 481
 482         if (!test_eth_enabled())
 483                 return 0;
 484
 485         switch (net_check_prereq(protocol)) {
 486         case 1:
 487                 /* network not configured */
 488                 eth_halt();
 489                 net_set_state(prev_net_state);
 490                 return -ENODEV;
 491
 492         case 2:
 493                 /* network device not configured */
 494                 break;
 495
 496         case 0:
 497                 net_dev_exists = 1;
 498                 net_boot_file_size = 0;
 499                 switch (protocol) {
 500 #ifdef CONFIG_CMD_TFTPBOOT
 501                 case TFTPGET:
 502 #ifdef CONFIG_CMD_TFTPPUT
 503                 case TFTPPUT:
 504 #endif
 505                         /* always use ARP to get server ethernet address */
 506                         tftp_start(protocol);
 507                         break;
 508 #endif
 509 #ifdef CONFIG_CMD_TFTPSRV
 510                 case TFTPSRV:
 511                         tftp_start_server();
 512                         break;
 513 #endif
 514 #if defined(CONFIG_UDP_FUNCTION_FASTBOOT)
 515                 case FASTBOOT_UDP:
 516                         fastboot_udp_start_server();
 517                         break;
 518 #endif
 519 #if defined(CONFIG_TCP_FUNCTION_FASTBOOT)
 520                 case FASTBOOT_TCP:
 521                         fastboot_tcp_start_server();
 522                         break;
 523 #endif
 524 #if defined(CONFIG_CMD_DHCP)
 525                 case DHCP:
 526                         bootp_reset();
 527                         net_ip.s_addr = 0;
 528                         dhcp_request();         /* Basically same as BOOTP */
 529                         break;
 530 #endif
 531                 case DHCP6:
 532                         if (IS_ENABLED(CONFIG_CMD_DHCP6))
 533                                 dhcp6_start();
 534                         break;
 535 #if defined(CONFIG_CMD_BOOTP)
 536                 case BOOTP:
 537                         bootp_reset();
 538                         net_ip.s_addr = 0;
 539                         bootp_request();
 540                         break;
 541 #endif
 542 #if defined(CONFIG_CMD_RARP)
 543                 case RARP:
 544                         rarp_try = 0;
 545                         net_ip.s_addr = 0;
 546                         rarp_request();
 547                         break;
 548 #endif
 549 #if defined(CONFIG_CMD_PING)
 550                 case PING:
 551                         ping_start();
 552                         break;
 553 #endif
 554 #if defined(CONFIG_CMD_PING6)
 555                 case PING6:
 556                         ping6_start();
 557                         break;
 558 #endif
 559 #if defined(CONFIG_CMD_NFS) && !defined(CONFIG_SPL_BUILD)
 560                 case NFS:
 561                         nfs_start();
 562                         break;
 563 #endif
 564 #if defined(CONFIG_CMD_WGET)
 565                 case WGET:
 566                         wget_start();
 567                         break;
 568 #endif
 569 #if defined(CONFIG_CMD_CDP)
 570                 case CDP:
 571                         cdp_start();
 572                         break;
 573 #endif
 574 #if defined(CONFIG_NETCONSOLE) && !defined(CONFIG_SPL_BUILD)
 575                 case NETCONS:
 576                         nc_start();
 577                         break;
 578 #endif
 579 #if defined(CONFIG_CMD_DNS)
 580                 case DNS:
 581                         dns_start();
 582                         break;
 583 #endif
 584 #if defined(CONFIG_CMD_LINK_LOCAL)
 585                 case LINKLOCAL:
 586                         link_local_start();
 587                         break;
 588 #endif
 589 #if defined(CONFIG_CMD_WOL)
 590                 case WOL:
 591                         wol_start();
 592                         break;
 593 #endif
 594 #if defined(CONFIG_PHY_NCSI)
 595                 case NCSI:
 596                         ncsi_probe_packages();
 597                         break;
 598 #endif
 599                 case RS:
 600                         if (IS_ENABLED(CONFIG_IPV6_ROUTER_DISCOVERY))
 601                                 ip6_send_rs();
 602                         break;
 603                 default:
 604                         break;
 605                 }
 606
 607                 if (IS_ENABLED(CONFIG_PROT_UDP) && protocol == UDP)
 608                         udp_start();
 609
 610                 break;
 611         }
 612
 613 #if defined(CONFIG_MII) || defined(CONFIG_CMD_MII)
 614 #if     defined(CONFIG_SYS_FAULT_ECHO_LINK_DOWN)        && \
 615         defined(CONFIG_LED_STATUS)                      && \
 616         defined(CONFIG_LED_STATUS_RED)
 617         /*
 618          * Echo the inverted link state to the fault LED.
 619          */
 620         if (miiphy_link(eth_get_dev()->name, CONFIG_SYS_FAULT_MII_ADDR))
 621                 status_led_set(CONFIG_LED_STATUS_RED, CONFIG_LED_STATUS_OFF);
 622         else
 623                 status_led_set(CONFIG_LED_STATUS_RED, CONFIG_LED_STATUS_ON);
 624 #endif /* CONFIG_SYS_FAULT_ECHO_LINK_DOWN, ... */
 625 #endif /* CONFIG_MII, ... */
 626 #ifdef CONFIG_USB_KEYBOARD
 627         net_busy_flag = 1;
 628 #endif
 629
 630         /*
 631          *      Main packet reception loop.  Loop receiving packets until
 632          *      someone sets `net_state' to a state that terminates.
 633          */
 634         for (;;) {
 635                 schedule();
 636                 if (arp_timeout_check() > 0)
 637                         time_start = get_timer(0);
 638
 639                 if (IS_ENABLED(CONFIG_IPV6)) {
 640                         if (use_ip6 && (ndisc_timeout_check() > 0))
 641                                 time_start = get_timer(0);
 642                 }
 643
 644                 /*
 645                  *      Check the ethernet for a new packet.  The ethernet
 646                  *      receive routine will process it.
 647                  *      Most drivers return the most recent packet size, but not
 648                  *      errors that may have happened.
 649                  */
 650                 eth_rx();
 651
 652                 /*
 653                  *      Abort if ctrl-c was pressed.
 654                  */
 655                 if (ctrlc()) {
 656                         /* cancel any ARP that may not have completed */
 657                         net_arp_wait_packet_ip.s_addr = 0;
 658
 659                         net_cleanup_loop();
 660                         eth_halt();
 661                         /* Invalidate the last protocol */
 662                         eth_set_last_protocol(BOOTP);
 663
 664                         puts("\nAbort\n");
 665                         /* include a debug print as well incase the debug
 666                            messages are directed to stderr */
 667                         debug_cond(DEBUG_INT_STATE, "--- net_loop Abort!\n");
 668                         ret = -EINTR;
 669                         goto done;
 670                 }
 671
 672                 /*
 673                  *      Check for a timeout, and run the timeout handler
 674                  *      if we have one.
 675                  */
 676                 if (time_handler &&
 677                     ((get_timer(0) - time_start) > time_delta)) {
 678                         thand_f *x;
 679
 680 #if defined(CONFIG_MII) || defined(CONFIG_CMD_MII)
 681 #if     defined(CONFIG_SYS_FAULT_ECHO_LINK_DOWN)        && \
 682         defined(CONFIG_LED_STATUS)                      && \
 683         defined(CONFIG_LED_STATUS_RED)
 684                         /*
 685                          * Echo the inverted link state to the fault LED.
 686                          */
 687                         if (miiphy_link(eth_get_dev()->name,
 688                                         CONFIG_SYS_FAULT_MII_ADDR))
 689                                 status_led_set(CONFIG_LED_STATUS_RED,
 690                                                CONFIG_LED_STATUS_OFF);
 691                         else
 692                                 status_led_set(CONFIG_LED_STATUS_RED,
 693                                                CONFIG_LED_STATUS_ON);
 694 #endif /* CONFIG_SYS_FAULT_ECHO_LINK_DOWN, ... */
 695 #endif /* CONFIG_MII, ... */
 696                         debug_cond(DEBUG_INT_STATE, "--- net_loop timeout\n");
 697                         x = time_handler;
 698                         time_handler = (thand_f *)0;
 699                         (*x)();
 700                 } else if (IS_ENABLED(CONFIG_IPV6_ROUTER_DISCOVERY))
 701                         if (time_handler && protocol == RS)
 702                                 if (!ip6_is_unspecified_addr(&net_gateway6) &&
 703                                     net_prefix_length != 0) {
 704                                         net_set_state(NETLOOP_SUCCESS);
 705                                         net_set_timeout_handler(0, NULL);
 706                                 }
 707
 708                 if (net_state == NETLOOP_FAIL)
 709                         ret = net_start_again();
 710
 711                 switch (net_state) {
 712                 case NETLOOP_RESTART:
 713                         net_restarted = 1;
 714                         goto restart;
 715
 716                 case NETLOOP_SUCCESS:
 717                         net_cleanup_loop();
 718                         if (net_boot_file_size > 0) {
 719                                 printf("Bytes transferred = %d (%x hex)\n",
 720                                        net_boot_file_size, net_boot_file_size);
 721                                 env_set_hex("filesize", net_boot_file_size);
 722                                 env_set_hex("fileaddr", image_load_addr);
 723                         }
 724                         if (protocol != NETCONS && protocol != NCSI)
 725                                 eth_halt();
 726                         else
 727                                 eth_halt_state_only();
 728
 729                         eth_set_last_protocol(protocol);
 730
 731                         ret = net_boot_file_size;
 732                         debug_cond(DEBUG_INT_STATE, "--- net_loop Success!\n");
 733                         goto done;
 734
 735                 case NETLOOP_FAIL:
 736                         net_cleanup_loop();
 737                         /* Invalidate the last protocol */
 738                         eth_set_last_protocol(BOOTP);
 739                         debug_cond(DEBUG_INT_STATE, "--- net_loop Fail!\n");
 740                         ret = -ENONET;
 741                         goto done;
 742
 743                 case NETLOOP_CONTINUE:
 744                         continue;
 745                 }
 746         }
 747
 748 done:
 749 #ifdef CONFIG_USB_KEYBOARD
 750         net_busy_flag = 0;
 751 #endif
 752 #ifdef CONFIG_CMD_TFTPPUT
 753         /* Clear out the handlers */
 754         net_set_udp_handler(NULL);
 755         net_set_icmp_handler(NULL);
 756 #endif
 757         net_set_state(prev_net_state);
 758
 759 #if defined(CONFIG_CMD_PCAP)
 760         if (pcap_active())
 761                 pcap_print_status();
 762 #endif
 763         return ret;
 764 }
 765
 766 /**********************************************************************/
 767
 768 static void start_again_timeout_handler(void)
 769 {
 770         net_set_state(NETLOOP_RESTART);
 771 }
 772
 773 int net_start_again(void)
 774 {
 775         char *nretry;
 776         int retry_forever = 0;
 777         unsigned long retrycnt = 0;
 778         int ret;
 779
 780         nretry = env_get("netretry");
 781         if (nretry) {
 782                 if (!strcmp(nretry, "yes"))
 783                         retry_forever = 1;
 784                 else if (!strcmp(nretry, "no"))
 785                         retrycnt = 0;
 786                 else if (!strcmp(nretry, "once"))
 787                         retrycnt = 1;
 788                 else
 789                         retrycnt = simple_strtoul(nretry, NULL, 0);
 790         } else {
 791                 retrycnt = 0;
 792                 retry_forever = 0;
 793         }
 794
 795         if ((!retry_forever) && (net_try_count > retrycnt)) {
 796                 eth_halt();
 797                 net_set_state(NETLOOP_FAIL);
 798                 /*
 799                  * We don't provide a way for the protocol to return an error,
 800                  * but this is almost always the reason.
 801                  */
 802                 return -ETIMEDOUT;
 803         }
 804
 805         net_try_count++;
 806
 807         eth_halt();
 808 #if !defined(CONFIG_NET_DO_NOT_TRY_ANOTHER)
 809         eth_try_another(!net_restarted);
 810 #endif
 811         ret = eth_init();
 812         if (net_restart_wrap) {
 813                 net_restart_wrap = 0;
 814                 if (net_dev_exists) {
 815                         net_set_timeout_handler(10000UL,
 816                                                 start_again_timeout_handler);
 817                         net_set_udp_handler(NULL);
 818                 } else {
 819                         net_set_state(NETLOOP_FAIL);
 820                 }
 821         } else {
 822                 net_set_state(NETLOOP_RESTART);
 823         }
 824         return ret;
 825 }
 826
 827 /**********************************************************************/
 828 /*
 829  *      Miscelaneous bits.
 830  */
 831
 832 static void dummy_handler(uchar *pkt, unsigned dport,
 833                         struct in_addr sip, unsigned sport,
 834                         unsigned len)
 835 {
 836 }
 837
 838 rxhand_f *net_get_udp_handler(void)
 839 {
 840         return udp_packet_handler;
 841 }
 842
 843 void net_set_udp_handler(rxhand_f *f)
 844 {
 845         debug_cond(DEBUG_INT_STATE, "--- net_loop UDP handler set (%p)\n", f);
 846         if (f == NULL)
 847                 udp_packet_handler = dummy_handler;
 848         else
 849                 udp_packet_handler = f;
 850 }
 851
 852 rxhand_f *net_get_arp_handler(void)
 853 {
 854         return arp_packet_handler;
 855 }
 856
 857 void net_set_arp_handler(rxhand_f *f)
 858 {
 859         debug_cond(DEBUG_INT_STATE, "--- net_loop ARP handler set (%p)\n", f);
 860         if (f == NULL)
 861                 arp_packet_handler = dummy_handler;
 862         else
 863                 arp_packet_handler = f;
 864 }
 865
 866 #ifdef CONFIG_CMD_TFTPPUT
 867 void net_set_icmp_handler(rxhand_icmp_f *f)
 868 {
 869         packet_icmp_handler = f;
 870 }
 871 #endif
 872
 873 void net_set_timeout_handler(ulong iv, thand_f *f)
 874 {
 875         if (iv == 0) {
 876                 debug_cond(DEBUG_INT_STATE,
 877                            "--- net_loop timeout handler cancelled\n");
 878                 time_handler = (thand_f *)0;
 879         } else {
 880                 debug_cond(DEBUG_INT_STATE,
 881                            "--- net_loop timeout handler set (%p)\n", f);
 882                 time_handler = f;
 883                 time_start = get_timer(0);
 884                 time_delta = iv * CONFIG_SYS_HZ / 1000;
 885         }
 886 }
 887
 888 uchar *net_get_async_tx_pkt_buf(void)
 889 {
 890         if (arp_is_waiting())
 891                 return arp_tx_packet; /* If we are waiting, we already sent */
 892         else
 893                 return net_tx_packet;
 894 }
 895
 896 int net_send_udp_packet(uchar *ether, struct in_addr dest, int dport, int sport,
 897                 int payload_len)
 898 {
 899         return net_send_ip_packet(ether, dest, dport, sport, payload_len,
 900                                   IPPROTO_UDP, 0, 0, 0);
 901 }
 902
 903 #if defined(CONFIG_PROT_TCP)
 904 int net_send_tcp_packet(int payload_len, int dport, int sport, u8 action,
 905                         u32 tcp_seq_num, u32 tcp_ack_num)
 906 {
 907         return net_send_ip_packet(net_server_ethaddr, net_server_ip, dport,
 908                                   sport, payload_len, IPPROTO_TCP, action,
 909                                   tcp_seq_num, tcp_ack_num);
 910 }
 911 #endif
 912
 913 int net_send_ip_packet(uchar *ether, struct in_addr dest, int dport, int sport,
 914                        int payload_len, int proto, u8 action, u32 tcp_seq_num,
 915                        u32 tcp_ack_num)
 916 {
 917         uchar *pkt;
 918         int eth_hdr_size;
 919         int pkt_hdr_size;
 920
 921         /* make sure the net_tx_packet is initialized (net_init() was called) */
 922         assert(net_tx_packet != NULL);
 923         if (net_tx_packet == NULL)
 924                 return -1;
 925
 926         /* convert to new style broadcast */
 927         if (dest.s_addr == 0)
 928                 dest.s_addr = 0xFFFFFFFF;
 929
 930         /* if broadcast, make the ether address a broadcast and don't do ARP */
 931         if (dest.s_addr == 0xFFFFFFFF)
 932                 ether = (uchar *)net_bcast_ethaddr;
 933
 934         pkt = (uchar *)net_tx_packet;
 935
 936         eth_hdr_size = net_set_ether(pkt, ether, PROT_IP);
 937
 938         switch (proto) {
 939         case IPPROTO_UDP:
 940                 net_set_udp_header(pkt + eth_hdr_size, dest, dport, sport,
 941                                    payload_len);
 942                 pkt_hdr_size = eth_hdr_size + IP_UDP_HDR_SIZE;
 943                 break;
 944 #if defined(CONFIG_PROT_TCP)
 945         case IPPROTO_TCP:
 946                 pkt_hdr_size = eth_hdr_size
 947                         + tcp_set_tcp_header(pkt + eth_hdr_size, dport, sport,
 948                                              payload_len, action, tcp_seq_num,
 949                                              tcp_ack_num);
 950                 break;
 951 #endif
 952         default:
 953                 return -EINVAL;
 954         }
 955
 956         /* if MAC address was not discovered yet, do an ARP request */
 957         if (memcmp(ether, net_null_ethaddr, 6) == 0) {
 958                 debug_cond(DEBUG_DEV_PKT, "sending ARP for %pI4\n", &dest);
 959
 960                 /* save the ip and eth addr for the packet to send after arp */
 961                 net_arp_wait_packet_ip = dest;
 962                 arp_wait_packet_ethaddr = ether;
 963
 964                 /* size of the waiting packet */
 965                 arp_wait_tx_packet_size = pkt_hdr_size + payload_len;
 966
 967                 /* and do the ARP request */
 968                 arp_wait_try = 1;
 969                 arp_wait_timer_start = get_timer(0);
 970                 arp_request();
 971                 return 1;       /* waiting */
 972         } else {
 973                 debug_cond(DEBUG_DEV_PKT, "sending UDP to %pI4/%pM\n",
 974                            &dest, ether);
 975                 net_send_packet(net_tx_packet, pkt_hdr_size + payload_len);
 976                 return 0;       /* transmitted */
 977         }
 978 }
 979
 980 #ifdef CONFIG_IP_DEFRAG
 981 /*
 982  * This function collects fragments in a single packet, according
 983  * to the algorithm in RFC815. It returns NULL or the pointer to
 984  * a complete packet, in static storage
 985  */
 986 #define IP_PKTSIZE (CONFIG_NET_MAXDEFRAG)
 987
 988 #define IP_MAXUDP (IP_PKTSIZE - IP_HDR_SIZE)
 989
 990 /*
 991  * this is the packet being assembled, either data or frag control.
 992  * Fragments go by 8 bytes, so this union must be 8 bytes long
 993  */
 994 struct hole {
 995         /* first_byte is address of this structure */
 996         u16 last_byte;  /* last byte in this hole + 1 (begin of next hole) */
 997         u16 next_hole;  /* index of next (in 8-b blocks), 0 == none */
 998         u16 prev_hole;  /* index of prev, 0 == none */
 999         u16 unused;
1000 };
1001
1002 static struct ip_udp_hdr *__net_defragment(struct ip_udp_hdr *ip, int *lenp)
1003 {
1004         static uchar pkt_buff[IP_PKTSIZE] __aligned(PKTALIGN);
1005         static u16 first_hole, total_len;
1006         struct hole *payload, *thisfrag, *h, *newh;
1007         struct ip_udp_hdr *localip = (struct ip_udp_hdr *)pkt_buff;
1008         uchar *indata = (uchar *)ip;
1009         int offset8, start, len, done = 0;
1010         u16 ip_off = ntohs(ip->ip_off);
1011
1012         /*
1013          * Calling code already rejected <, but we don't have to deal
1014          * with an IP fragment with no payload.
1015          */
1016         if (ntohs(ip->ip_len) <= IP_HDR_SIZE)
1017                 return NULL;
1018
1019         /* payload starts after IP header, this fragment is in there */
1020         payload = (struct hole *)(pkt_buff + IP_HDR_SIZE);
1021         offset8 =  (ip_off & IP_OFFS);
1022         thisfrag = payload + offset8;
1023         start = offset8 * 8;
1024         len = ntohs(ip->ip_len) - IP_HDR_SIZE;
1025
1026         /* All but last fragment must have a multiple-of-8 payload. */
1027         if ((len & 7) && (ip_off & IP_FLAGS_MFRAG))
1028                 return NULL;
1029
1030         if (start + len > IP_MAXUDP) /* fragment extends too far */
1031                 return NULL;
1032
1033         if (!total_len || localip->ip_id != ip->ip_id) {
1034                 /* new (or different) packet, reset structs */
1035                 total_len = 0xffff;
1036                 payload[0].last_byte = ~0;
1037                 payload[0].next_hole = 0;
1038                 payload[0].prev_hole = 0;
1039                 first_hole = 0;
1040                 /* any IP header will work, copy the first we received */
1041                 memcpy(localip, ip, IP_HDR_SIZE);
1042         }
1043
1044         /*
1045          * What follows is the reassembly algorithm. We use the payload
1046          * array as a linked list of hole descriptors, as each hole starts
1047          * at a multiple of 8 bytes. However, last byte can be whatever value,
1048          * so it is represented as byte count, not as 8-byte blocks.
1049          */
1050
1051         h = payload + first_hole;
1052         while (h->last_byte < start) {
1053                 if (!h->next_hole) {
1054                         /* no hole that far away */
1055                         return NULL;
1056                 }
1057                 h = payload + h->next_hole;
1058         }
1059
1060         /* last fragment may be 1..7 bytes, the "+7" forces acceptance */
1061         if (offset8 + ((len + 7) / 8) <= h - payload) {
1062                 /* no overlap with holes (dup fragment?) */
1063                 return NULL;
1064         }
1065
1066         if (!(ip_off & IP_FLAGS_MFRAG)) {
1067                 /* no more fragmentss: truncate this (last) hole */
1068                 total_len = start + len;
1069                 h->last_byte = start + len;
1070         }
1071
1072         /*
1073          * There is some overlap: fix the hole list. This code deals
1074          * with a fragment that overlaps with two different holes
1075          * (thus being a superset of a previously-received fragment)
1076          * by only using the part of the fragment that fits in the
1077          * first hole.
1078          */
1079         if (h->last_byte < start + len)
1080                 len = h->last_byte - start;
1081
1082         if ((h >= thisfrag) && (h->last_byte <= start + len)) {
1083                 /* complete overlap with hole: remove hole */
1084                 if (!h->prev_hole && !h->next_hole) {
1085                         /* last remaining hole */
1086                         done = 1;
1087                 } else if (!h->prev_hole) {
1088                         /* first hole */
1089                         first_hole = h->next_hole;
1090                         payload[h->next_hole].prev_hole = 0;
1091                 } else if (!h->next_hole) {
1092                         /* last hole */
1093                         payload[h->prev_hole].next_hole = 0;
1094                 } else {
1095                         /* in the middle of the list */
1096                         payload[h->next_hole].prev_hole = h->prev_hole;
1097                         payload[h->prev_hole].next_hole = h->next_hole;
1098                 }
1099
1100         } else if (h->last_byte <= start + len) {
1101                 /* overlaps with final part of the hole: shorten this hole */
1102                 h->last_byte = start;
1103
1104         } else if (h >= thisfrag) {
1105                 /* overlaps with initial part of the hole: move this hole */
1106                 newh = thisfrag + (len / 8);
1107                 *newh = *h;
1108                 h = newh;
1109                 if (h->next_hole)
1110                         payload[h->next_hole].prev_hole = (h - payload);
1111                 if (h->prev_hole)
1112                         payload[h->prev_hole].next_hole = (h - payload);
1113                 else
1114                         first_hole = (h - payload);
1115
1116         } else {
1117                 /* fragment sits in the middle: split the hole */
1118                 newh = thisfrag + (len / 8);
1119                 *newh = *h;
1120                 h->last_byte = start;
1121                 h->next_hole = (newh - payload);
1122                 newh->prev_hole = (h - payload);
1123                 if (newh->next_hole)
1124                         payload[newh->next_hole].prev_hole = (newh - payload);
1125         }
1126
1127         /* finally copy this fragment and possibly return whole packet */
1128         memcpy((uchar *)thisfrag, indata + IP_HDR_SIZE, len);
1129         if (!done)
1130                 return NULL;
1131
1132         *lenp = total_len + IP_HDR_SIZE;
1133         localip->ip_len = htons(*lenp);
1134         return localip;
1135 }
1136
1137 static inline struct ip_udp_hdr *net_defragment(struct ip_udp_hdr *ip,
1138         int *lenp)
1139 {
1140         u16 ip_off = ntohs(ip->ip_off);
1141         if (!(ip_off & (IP_OFFS | IP_FLAGS_MFRAG)))
1142                 return ip; /* not a fragment */
1143         return __net_defragment(ip, lenp);
1144 }
1145
1146 #else /* !CONFIG_IP_DEFRAG */
1147
1148 static inline struct ip_udp_hdr *net_defragment(struct ip_udp_hdr *ip,
1149         int *lenp)
1150 {
1151         u16 ip_off = ntohs(ip->ip_off);
1152         if (!(ip_off & (IP_OFFS | IP_FLAGS_MFRAG)))
1153                 return ip; /* not a fragment */
1154         return NULL;
1155 }
1156 #endif
1157
1158 /**
1159  * Receive an ICMP packet. We deal with REDIRECT and PING here, and silently
1160  * drop others.
1161  *
1162  * @parma ip    IP packet containing the ICMP
1163  */
1164 static void receive_icmp(struct ip_udp_hdr *ip, int len,
1165                         struct in_addr src_ip, struct ethernet_hdr *et)
1166 {
1167         struct icmp_hdr *icmph = (struct icmp_hdr *)&ip->udp_src;
1168
1169         switch (icmph->type) {
1170         case ICMP_REDIRECT:
1171                 if (icmph->code != ICMP_REDIR_HOST)
1172                         return;
1173                 printf(" ICMP Host Redirect to %pI4 ",
1174                        &icmph->un.gateway);
1175                 break;
1176         default:
1177 #if defined(CONFIG_CMD_PING)
1178                 ping_receive(et, ip, len);
1179 #endif
1180 #ifdef CONFIG_CMD_TFTPPUT
1181                 if (packet_icmp_handler)
1182                         packet_icmp_handler(icmph->type, icmph->code,
1183                                             ntohs(ip->udp_dst), src_ip,
1184                                             ntohs(ip->udp_src), icmph->un.data,
1185                                             ntohs(ip->udp_len));
1186 #endif
1187                 break;
1188         }
1189 }
1190
1191 void net_process_received_packet(uchar *in_packet, int len)
1192 {
1193         struct ethernet_hdr *et;
1194         struct ip_udp_hdr *ip;
1195         struct in_addr dst_ip;
1196         struct in_addr src_ip;
1197         int eth_proto;
1198 #if defined(CONFIG_CMD_CDP)
1199         int iscdp;
1200 #endif
1201         ushort cti = 0, vlanid = VLAN_NONE, myvlanid, mynvlanid;
1202
1203         debug_cond(DEBUG_NET_PKT, "packet received\n");
1204
1205 #if defined(CONFIG_CMD_PCAP)
1206         pcap_post(in_packet, len, false);
1207 #endif
1208         net_rx_packet = in_packet;
1209         net_rx_packet_len = len;
1210         et = (struct ethernet_hdr *)in_packet;
1211
1212         /* too small packet? */
1213         if (len < ETHER_HDR_SIZE)
1214                 return;
1215
1216 #if defined(CONFIG_API) || defined(CONFIG_EFI_LOADER)
1217         if (push_packet) {
1218                 (*push_packet)(in_packet, len);
1219                 return;
1220         }
1221 #endif
1222
1223 #if defined(CONFIG_CMD_CDP)
1224         /* keep track if packet is CDP */
1225         iscdp = is_cdp_packet(et->et_dest);
1226 #endif
1227
1228         myvlanid = ntohs(net_our_vlan);
1229         if (myvlanid == (ushort)-1)
1230                 myvlanid = VLAN_NONE;
1231         mynvlanid = ntohs(net_native_vlan);
1232         if (mynvlanid == (ushort)-1)
1233                 mynvlanid = VLAN_NONE;
1234
1235         eth_proto = ntohs(et->et_protlen);
1236
1237         if (eth_proto < 1514) {
1238                 struct e802_hdr *et802 = (struct e802_hdr *)et;
1239                 /*
1240                  *      Got a 802.2 packet.  Check the other protocol field.
1241                  *      XXX VLAN over 802.2+SNAP not implemented!
1242                  */
1243                 eth_proto = ntohs(et802->et_prot);
1244
1245                 ip = (struct ip_udp_hdr *)(in_packet + E802_HDR_SIZE);
1246                 len -= E802_HDR_SIZE;
1247
1248         } else if (eth_proto != PROT_VLAN) {    /* normal packet */
1249                 ip = (struct ip_udp_hdr *)(in_packet + ETHER_HDR_SIZE);
1250                 len -= ETHER_HDR_SIZE;
1251
1252         } else {                        /* VLAN packet */
1253                 struct vlan_ethernet_hdr *vet =
1254                         (struct vlan_ethernet_hdr *)et;
1255
1256                 debug_cond(DEBUG_NET_PKT, "VLAN packet received\n");
1257
1258                 /* too small packet? */
1259                 if (len < VLAN_ETHER_HDR_SIZE)
1260                         return;
1261
1262                 /* if no VLAN active */
1263                 if ((ntohs(net_our_vlan) & VLAN_IDMASK) == VLAN_NONE
1264 #if defined(CONFIG_CMD_CDP)
1265                                 && iscdp == 0
1266 #endif
1267                                 )
1268                         return;
1269
1270                 cti = ntohs(vet->vet_tag);
1271                 vlanid = cti & VLAN_IDMASK;
1272                 eth_proto = ntohs(vet->vet_type);
1273
1274                 ip = (struct ip_udp_hdr *)(in_packet + VLAN_ETHER_HDR_SIZE);
1275                 len -= VLAN_ETHER_HDR_SIZE;
1276         }
1277
1278         debug_cond(DEBUG_NET_PKT, "Receive from protocol 0x%x\n", eth_proto);
1279
1280 #if defined(CONFIG_CMD_CDP)
1281         if (iscdp) {
1282                 cdp_receive((uchar *)ip, len);
1283                 return;
1284         }
1285 #endif
1286
1287         if ((myvlanid & VLAN_IDMASK) != VLAN_NONE) {
1288                 if (vlanid == VLAN_NONE)
1289                         vlanid = (mynvlanid & VLAN_IDMASK);
1290                 /* not matched? */
1291                 if (vlanid != (myvlanid & VLAN_IDMASK))
1292                         return;
1293         }
1294
1295         switch (eth_proto) {
1296         case PROT_ARP:
1297                 arp_receive(et, ip, len);
1298                 break;
1299
1300 #ifdef CONFIG_CMD_RARP
1301         case PROT_RARP:
1302                 rarp_receive(ip, len);
1303                 break;
1304 #endif
1305 #if IS_ENABLED(CONFIG_IPV6)
1306         case PROT_IP6:
1307                 net_ip6_handler(et, (struct ip6_hdr *)ip, len);
1308                 break;
1309 #endif
1310         case PROT_IP:
1311                 debug_cond(DEBUG_NET_PKT, "Got IP\n");
1312                 /* Before we start poking the header, make sure it is there */
1313                 if (len < IP_HDR_SIZE) {
1314                         debug("len bad %d < %lu\n", len,
1315                               (ulong)IP_HDR_SIZE);
1316                         return;
1317                 }
1318                 /* Check the packet length */
1319                 if (len < ntohs(ip->ip_len)) {
1320                         debug("len bad %d < %d\n", len, ntohs(ip->ip_len));
1321                         return;
1322                 }
1323                 len = ntohs(ip->ip_len);
1324                 if (len < IP_HDR_SIZE) {
1325                         debug("bad ip->ip_len %d < %d\n", len, (int)IP_HDR_SIZE);
1326                         return;
1327                 }
1328                 debug_cond(DEBUG_NET_PKT, "len=%d, v=%02x\n",
1329                            len, ip->ip_hl_v & 0xff);
1330
1331                 /* Can't deal with anything except IPv4 */
1332                 if ((ip->ip_hl_v & 0xf0) != 0x40)
1333                         return;
1334                 /* Can't deal with IP options (headers != 20 bytes) */
1335                 if ((ip->ip_hl_v & 0x0f) != 0x05)
1336                         return;
1337                 /* Check the Checksum of the header */
1338                 if (!ip_checksum_ok((uchar *)ip, IP_HDR_SIZE)) {
1339                         debug("checksum bad\n");
1340                         return;
1341                 }
1342                 /* If it is not for us, ignore it */
1343                 dst_ip = net_read_ip(&ip->ip_dst);
1344                 if (net_ip.s_addr && dst_ip.s_addr != net_ip.s_addr &&
1345                     dst_ip.s_addr != 0xFFFFFFFF) {
1346                                 return;
1347                 }
1348                 /* Read source IP address for later use */
1349                 src_ip = net_read_ip(&ip->ip_src);
1350                 /*
1351                  * The function returns the unchanged packet if it's not
1352                  * a fragment, and either the complete packet or NULL if
1353                  * it is a fragment (if !CONFIG_IP_DEFRAG, it returns NULL)
1354                  */
1355                 ip = net_defragment(ip, &len);
1356                 if (!ip)
1357                         return;
1358                 /*
1359                  * watch for ICMP host redirects
1360                  *
1361                  * There is no real handler code (yet). We just watch
1362                  * for ICMP host redirect messages. In case anybody
1363                  * sees these messages: please contact me
1364                  * ([email protected]), or - even better - send me the
1365                  * necessary fixes :-)
1366                  *
1367                  * Note: in all cases where I have seen this so far
1368                  * it was a problem with the router configuration,
1369                  * for instance when a router was configured in the
1370                  * BOOTP reply, but the TFTP server was on the same
1371                  * subnet. So this is probably a warning that your
1372                  * configuration might be wrong. But I'm not really
1373                  * sure if there aren't any other situations.
1374                  *
1375                  * Simon Glass <[email protected]>: We get an ICMP when
1376                  * we send a tftp packet to a dead connection, or when
1377                  * there is no server at the other end.
1378                  */
1379                 if (ip->ip_p == IPPROTO_ICMP) {
1380                         receive_icmp(ip, len, src_ip, et);
1381                         return;
1382 #if defined(CONFIG_PROT_TCP)
1383                 } else if (ip->ip_p == IPPROTO_TCP) {
1384                         debug_cond(DEBUG_DEV_PKT,
1385                                    "TCP PH (to=%pI4, from=%pI4, len=%d)\n",
1386                                    &dst_ip, &src_ip, len);
1387
1388                         rxhand_tcp_f((union tcp_build_pkt *)ip, len);
1389                         return;
1390 #endif
1391                 } else if (ip->ip_p != IPPROTO_UDP) {   /* Only UDP packets */
1392                         return;
1393                 }
1394
1395                 if (ntohs(ip->udp_len) < UDP_HDR_SIZE || ntohs(ip->udp_len) > len - IP_HDR_SIZE)
1396                         return;
1397
1398                 debug_cond(DEBUG_DEV_PKT,
1399                            "received UDP (to=%pI4, from=%pI4, len=%d)\n",
1400                            &dst_ip, &src_ip, len);
1401
1402                 if (IS_ENABLED(CONFIG_UDP_CHECKSUM) && ip->udp_xsum != 0) {
1403                         ulong   xsum;
1404                         u8 *sumptr;
1405                         ushort  sumlen;
1406
1407                         xsum  = ip->ip_p;
1408                         xsum += (ntohs(ip->udp_len));
1409                         xsum += (ntohl(ip->ip_src.s_addr) >> 16) & 0x0000ffff;
1410                         xsum += (ntohl(ip->ip_src.s_addr) >>  0) & 0x0000ffff;
1411                         xsum += (ntohl(ip->ip_dst.s_addr) >> 16) & 0x0000ffff;
1412                         xsum += (ntohl(ip->ip_dst.s_addr) >>  0) & 0x0000ffff;
1413
1414                         sumlen = ntohs(ip->udp_len);
1415                         sumptr = (u8 *)&ip->udp_src;
1416
1417                         while (sumlen > 1) {
1418                                 /* inlined ntohs() to avoid alignment errors */
1419                                 xsum += (sumptr[0] << 8) + sumptr[1];
1420                                 sumptr += 2;
1421                                 sumlen -= 2;
1422                         }
1423                         if (sumlen > 0)
1424                                 xsum += (sumptr[0] << 8) + sumptr[0];
1425                         while ((xsum >> 16) != 0) {
1426                                 xsum = (xsum & 0x0000ffff) +
1427                                        ((xsum >> 16) & 0x0000ffff);
1428                         }
1429                         if ((xsum != 0x00000000) && (xsum != 0x0000ffff)) {
1430                                 printf(" UDP wrong checksum %08lx %08x\n",
1431                                        xsum, ntohs(ip->udp_xsum));
1432                                 return;
1433                         }
1434                 }
1435
1436 #if defined(CONFIG_NETCONSOLE) && !defined(CONFIG_SPL_BUILD)
1437                 nc_input_packet((uchar *)ip + IP_UDP_HDR_SIZE,
1438                                 src_ip,
1439                                 ntohs(ip->udp_dst),
1440                                 ntohs(ip->udp_src),
1441                                 ntohs(ip->udp_len) - UDP_HDR_SIZE);
1442 #endif
1443                 /*
1444                  * IP header OK.  Pass the packet to the current handler.
1445                  */
1446                 (*udp_packet_handler)((uchar *)ip + IP_UDP_HDR_SIZE,
1447                                       ntohs(ip->udp_dst),
1448                                       src_ip,
1449                                       ntohs(ip->udp_src),
1450                                       ntohs(ip->udp_len) - UDP_HDR_SIZE);
1451                 break;
1452 #ifdef CONFIG_CMD_WOL
1453         case PROT_WOL:
1454                 wol_receive(ip, len);
1455                 break;
1456 #endif
1457 #ifdef CONFIG_PHY_NCSI
1458         case PROT_NCSI:
1459                 ncsi_receive(et, ip, len);
1460                 break;
1461 #endif
1462         }
1463 }
1464
1465 /**********************************************************************/
1466
1467 static int net_check_prereq(enum proto_t protocol)
1468 {
1469         switch (protocol) {
1470                 /* Fall through */
1471 #if defined(CONFIG_CMD_PING)
1472         case PING:
1473                 if (net_ping_ip.s_addr == 0) {
1474                         puts("*** ERROR: ping address not given\n");
1475                         return 1;
1476                 }
1477                 goto common;
1478 #endif
1479 #if defined(CONFIG_CMD_PING6)
1480         case PING6:
1481                 if (ip6_is_unspecified_addr(&net_ping_ip6)) {
1482                         puts("*** ERROR: ping address not given\n");
1483                         return 1;
1484                 }
1485                 goto common;
1486 #endif
1487 #if defined(CONFIG_CMD_DNS)
1488         case DNS:
1489                 if (net_dns_server.s_addr == 0) {
1490                         puts("*** ERROR: DNS server address not given\n");
1491                         return 1;
1492                 }
1493                 goto common;
1494 #endif
1495 #if defined(CONFIG_PROT_UDP)
1496         case UDP:
1497                 if (udp_prereq())
1498                         return 1;
1499                 goto common;
1500 #endif
1501
1502 #if defined(CONFIG_CMD_NFS)
1503         case NFS:
1504 #endif
1505                 /* Fall through */
1506         case TFTPGET:
1507         case TFTPPUT:
1508                 if (IS_ENABLED(CONFIG_IPV6) && use_ip6) {
1509                         if (!memcmp(&net_server_ip6, &net_null_addr_ip6,
1510                                     sizeof(struct in6_addr)) &&
1511                                     !strchr(net_boot_file_name, '[')) {
1512                                 puts("*** ERROR: `serverip6' not set\n");
1513                                 return 1;
1514                         }
1515                 } else if (net_server_ip.s_addr == 0 && !is_serverip_in_cmd()) {
1516                         puts("*** ERROR: `serverip' not set\n");
1517                         return 1;
1518                 }
1519 #if     defined(CONFIG_CMD_PING) || \
1520         defined(CONFIG_CMD_DNS) || defined(CONFIG_PROT_UDP)
1521 common:
1522 #endif
1523                 /* Fall through */
1524
1525         case NETCONS:
1526         case FASTBOOT_UDP:
1527         case FASTBOOT_TCP:
1528         case TFTPSRV:
1529                 if (IS_ENABLED(CONFIG_IPV6) && use_ip6) {
1530                         if (!memcmp(&net_link_local_ip6, &net_null_addr_ip6,
1531                                     sizeof(struct in6_addr))) {
1532                                 puts("*** ERROR: `ip6addr` not set\n");
1533                                 return 1;
1534                         }
1535                 } else if (net_ip.s_addr == 0) {
1536                         puts("*** ERROR: `ipaddr' not set\n");
1537                         return 1;
1538                 }
1539                 /* Fall through */
1540
1541 #ifdef CONFIG_CMD_RARP
1542         case RARP:
1543 #endif
1544 #ifdef CONFIG_PHY_NCSI
1545         case NCSI:
1546 #endif
1547         case BOOTP:
1548         case CDP:
1549         case DHCP:
1550         case LINKLOCAL:
1551                 if (memcmp(net_ethaddr, "\0\0\0\0\0\0", 6) == 0) {
1552                         int num = eth_get_dev_index();
1553
1554                         switch (num) {
1555                         case -1:
1556                                 puts("*** ERROR: No ethernet found.\n");
1557                                 return 1;
1558                         case 0:
1559                                 puts("*** ERROR: `ethaddr' not set\n");
1560                                 break;
1561                         default:
1562                                 printf("*** ERROR: `eth%daddr' not set\n",
1563                                        num);
1564                                 break;
1565                         }
1566
1567                         net_start_again();
1568                         return 2;
1569                 }
1570                 /* Fall through */
1571         default:
1572                 return 0;
1573         }
1574         return 0;               /* OK */
1575 }
1576 /**********************************************************************/
1577
1578 int
1579 net_eth_hdr_size(void)
1580 {
1581         ushort myvlanid;
1582
1583         myvlanid = ntohs(net_our_vlan);
1584         if (myvlanid == (ushort)-1)
1585                 myvlanid = VLAN_NONE;
1586
1587         return ((myvlanid & VLAN_IDMASK) == VLAN_NONE) ? ETHER_HDR_SIZE :
1588                 VLAN_ETHER_HDR_SIZE;
1589 }
1590
1591 int net_set_ether(uchar *xet, const uchar *dest_ethaddr, uint prot)
1592 {
1593         struct ethernet_hdr *et = (struct ethernet_hdr *)xet;
1594         ushort myvlanid;
1595
1596         myvlanid = ntohs(net_our_vlan);
1597         if (myvlanid == (ushort)-1)
1598                 myvlanid = VLAN_NONE;
1599
1600         memcpy(et->et_dest, dest_ethaddr, 6);
1601         memcpy(et->et_src, net_ethaddr, 6);
1602         if ((myvlanid & VLAN_IDMASK) == VLAN_NONE) {
1603                 et->et_protlen = htons(prot);
1604                 return ETHER_HDR_SIZE;
1605         } else {
1606                 struct vlan_ethernet_hdr *vet =
1607                         (struct vlan_ethernet_hdr *)xet;
1608
1609                 vet->vet_vlan_type = htons(PROT_VLAN);
1610                 vet->vet_tag = htons((0 << 5) | (myvlanid & VLAN_IDMASK));
1611                 vet->vet_type = htons(prot);
1612                 return VLAN_ETHER_HDR_SIZE;
1613         }
1614 }
1615
1616 int net_update_ether(struct ethernet_hdr *et, uchar *addr, uint prot)
1617 {
1618         ushort protlen;
1619
1620         memcpy(et->et_dest, addr, 6);
1621         memcpy(et->et_src, net_ethaddr, 6);
1622         protlen = ntohs(et->et_protlen);
1623         if (protlen == PROT_VLAN) {
1624                 struct vlan_ethernet_hdr *vet =
1625                         (struct vlan_ethernet_hdr *)et;
1626                 vet->vet_type = htons(prot);
1627                 return VLAN_ETHER_HDR_SIZE;
1628         } else if (protlen > 1514) {
1629                 et->et_protlen = htons(prot);
1630                 return ETHER_HDR_SIZE;
1631         } else {
1632                 /* 802.2 + SNAP */
1633                 struct e802_hdr *et802 = (struct e802_hdr *)et;
1634                 et802->et_prot = htons(prot);
1635                 return E802_HDR_SIZE;
1636         }
1637 }
1638
1639 void net_set_ip_header(uchar *pkt, struct in_addr dest, struct in_addr source,
1640                        u16 pkt_len, u8 proto)
1641 {
1642         struct ip_udp_hdr *ip = (struct ip_udp_hdr *)pkt;
1643
1644         /*
1645          *      Construct an IP header.
1646          */
1647         /* IP_HDR_SIZE / 4 (not including UDP) */
1648         ip->ip_hl_v  = 0x45;
1649         ip->ip_tos   = 0;
1650         ip->ip_len   = htons(pkt_len);
1651         ip->ip_p     = proto;
1652         ip->ip_id    = htons(net_ip_id++);
1653         ip->ip_off   = htons(IP_FLAGS_DFRAG);   /* Don't fragment */
1654         ip->ip_ttl   = 255;
1655         ip->ip_sum   = 0;
1656         /* already in network byte order */
1657         net_copy_ip((void *)&ip->ip_src, &source);
1658         /* already in network byte order */
1659         net_copy_ip((void *)&ip->ip_dst, &dest);
1660
1661         ip->ip_sum   = compute_ip_checksum(ip, IP_HDR_SIZE);
1662 }
1663
1664 void net_set_udp_header(uchar *pkt, struct in_addr dest, int dport, int sport,
1665                         int len)
1666 {
1667         struct ip_udp_hdr *ip = (struct ip_udp_hdr *)pkt;
1668
1669         /*
1670          *      If the data is an odd number of bytes, zero the
1671          *      byte after the last byte so that the checksum
1672          *      will work.
1673          */
1674         if (len & 1)
1675                 pkt[IP_UDP_HDR_SIZE + len] = 0;
1676
1677         net_set_ip_header(pkt, dest, net_ip, IP_UDP_HDR_SIZE + len,
1678                           IPPROTO_UDP);
1679
1680         ip->udp_src  = htons(sport);
1681         ip->udp_dst  = htons(dport);
1682         ip->udp_len  = htons(UDP_HDR_SIZE + len);
1683         ip->udp_xsum = 0;
1684 }
1685
1686 void copy_filename(char *dst, const char *src, int size)
1687 {
1688         if (src && *src && (*src == '"')) {
1689                 ++src;
1690                 --size;
1691         }
1692
1693         while ((--size > 0) && src && *src && (*src != '"'))
1694                 *dst++ = *src++;
1695         *dst = '\0';
1696 }
1697
1698 int is_serverip_in_cmd(void)
1699 {
1700         return !!strchr(net_boot_file_name, ':');
1701 }
1702
1703 int net_parse_bootfile(struct in_addr *ipaddr, char *filename, int max_len)
1704 {
1705         char *colon;
1706         struct in_addr ip;
1707         ip.s_addr = 0;
1708
1709         if (net_boot_file_name[0] == '\0')
1710                 return 0;
1711
1712         colon = strchr(net_boot_file_name, ':');
1713         if (colon) {
1714                 ip = string_to_ip(net_boot_file_name);
1715                 if (ipaddr && ip.s_addr)
1716                         *ipaddr = ip;
1717         }
1718         if (ip.s_addr) {
1719                 strncpy(filename, colon + 1, max_len);
1720         } else {
1721                 strncpy(filename, net_boot_file_name, max_len);
1722         }
1723         filename[max_len - 1] = '\0';
1724
1725         return 1;
1726 }
1727
1728 void ip_to_string(struct in_addr x, char *s)
1729 {
1730         x.s_addr = ntohl(x.s_addr);
1731         sprintf(s, "%d.%d.%d.%d",
1732                 (int) ((x.s_addr >> 24) & 0xff),
1733                 (int) ((x.s_addr >> 16) & 0xff),
1734                 (int) ((x.s_addr >> 8) & 0xff),
1735                 (int) ((x.s_addr >> 0) & 0xff)
1736         );
1737 }
1738
1739 void vlan_to_string(ushort x, char *s)
1740 {
1741         x = ntohs(x);
1742
1743         if (x == (ushort)-1)
1744                 x = VLAN_NONE;
1745
1746         if (x == VLAN_NONE)
1747                 strcpy(s, "none");
1748         else
1749                 sprintf(s, "%d", x & VLAN_IDMASK);
1750 }
1751
1752 ushort string_to_vlan(const char *s)
1753 {
1754         ushort id;
1755
1756         if (s == NULL)
1757                 return htons(VLAN_NONE);
1758
1759         if (*s < '0' || *s > '9')
1760                 id = VLAN_NONE;
1761         else
1762                 id = (ushort)dectoul(s, NULL);
1763
1764         return htons(id);
1765 }
1766
1767 ushort env_get_vlan(char *var)
1768 {
1769         return string_to_vlan(env_get(var));
1770 }