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