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