]> git.kernelconcepts.de Git - karo-tx-uboot.git/blob - net/net.c
net: Don't copy every packet that waits for an ARP
[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                         /* cancel any ARP that may not have completed */
446                         NetArpWaitPacketIP = 0;
447
448                         net_cleanup_loop();
449                         eth_halt();
450                         puts("\nAbort\n");
451                         goto done;
452                 }
453
454                 ArpTimeoutCheck();
455
456                 /*
457                  *      Check for a timeout, and run the timeout handler
458                  *      if we have one.
459                  */
460                 if (timeHandler && ((get_timer(0) - timeStart) > timeDelta)) {
461                         thand_f *x;
462
463 #if defined(CONFIG_MII) || defined(CONFIG_CMD_MII)
464 #if     defined(CONFIG_SYS_FAULT_ECHO_LINK_DOWN)        && \
465         defined(CONFIG_STATUS_LED)                      && \
466         defined(STATUS_LED_RED)
467                         /*
468                          * Echo the inverted link state to the fault LED.
469                          */
470                         if (miiphy_link(eth_get_dev()->name,
471                                        CONFIG_SYS_FAULT_MII_ADDR)) {
472                                 status_led_set(STATUS_LED_RED, STATUS_LED_OFF);
473                         } else {
474                                 status_led_set(STATUS_LED_RED, STATUS_LED_ON);
475                         }
476 #endif /* CONFIG_SYS_FAULT_ECHO_LINK_DOWN, ... */
477 #endif /* CONFIG_MII, ... */
478                         x = timeHandler;
479                         timeHandler = (thand_f *)0;
480                         (*x)();
481                 }
482
483
484                 switch (net_state) {
485
486                 case NETLOOP_RESTART:
487                         NetRestarted = 1;
488                         goto restart;
489
490                 case NETLOOP_SUCCESS:
491                         net_cleanup_loop();
492                         if (NetBootFileXferSize > 0) {
493                                 char buf[20];
494                                 printf("Bytes transferred = %ld (%lx hex)\n",
495                                         NetBootFileXferSize,
496                                         NetBootFileXferSize);
497                                 sprintf(buf, "%lX", NetBootFileXferSize);
498                                 setenv("filesize", buf);
499
500                                 sprintf(buf, "%lX", (unsigned long)load_addr);
501                                 setenv("fileaddr", buf);
502                         }
503                         eth_halt();
504                         ret = NetBootFileXferSize;
505                         goto done;
506
507                 case NETLOOP_FAIL:
508                         net_cleanup_loop();
509                         goto done;
510
511                 case NETLOOP_CONTINUE:
512                         continue;
513                 }
514         }
515
516 done:
517 #ifdef CONFIG_CMD_TFTPPUT
518         /* Clear out the handlers */
519         net_set_udp_handler(NULL);
520         net_set_icmp_handler(NULL);
521 #endif
522         return ret;
523 }
524
525 /**********************************************************************/
526
527 static void
528 startAgainTimeout(void)
529 {
530         net_set_state(NETLOOP_RESTART);
531 }
532
533 void NetStartAgain(void)
534 {
535         char *nretry;
536         int retry_forever = 0;
537         unsigned long retrycnt = 0;
538
539         nretry = getenv("netretry");
540         if (nretry) {
541                 if (!strcmp(nretry, "yes"))
542                         retry_forever = 1;
543                 else if (!strcmp(nretry, "no"))
544                         retrycnt = 0;
545                 else if (!strcmp(nretry, "once"))
546                         retrycnt = 1;
547                 else
548                         retrycnt = simple_strtoul(nretry, NULL, 0);
549         } else
550                 retry_forever = 1;
551
552         if ((!retry_forever) && (NetTryCount >= retrycnt)) {
553                 eth_halt();
554                 net_set_state(NETLOOP_FAIL);
555                 return;
556         }
557
558         NetTryCount++;
559
560         eth_halt();
561 #if !defined(CONFIG_NET_DO_NOT_TRY_ANOTHER)
562         eth_try_another(!NetRestarted);
563 #endif
564         eth_init(gd->bd);
565         if (NetRestartWrap) {
566                 NetRestartWrap = 0;
567                 if (NetDevExists) {
568                         NetSetTimeout(10000UL, startAgainTimeout);
569                         net_set_udp_handler(NULL);
570                 } else {
571                         net_set_state(NETLOOP_FAIL);
572                 }
573         } else {
574                 net_set_state(NETLOOP_RESTART);
575         }
576 }
577
578 /**********************************************************************/
579 /*
580  *      Miscelaneous bits.
581  */
582
583 static void dummy_handler(uchar *pkt, unsigned dport,
584                         IPaddr_t sip, unsigned sport,
585                         unsigned len)
586 {
587 }
588
589 rxhand_f *net_get_udp_handler(void)
590 {
591         return udp_packet_handler;
592 }
593
594 void net_set_udp_handler(rxhand_f *f)
595 {
596         if (f == NULL)
597                 udp_packet_handler = dummy_handler;
598         else
599                 udp_packet_handler = f;
600 }
601
602 rxhand_f *net_get_arp_handler(void)
603 {
604         return arp_packet_handler;
605 }
606
607 void net_set_arp_handler(rxhand_f *f)
608 {
609         if (f == NULL)
610                 arp_packet_handler = dummy_handler;
611         else
612                 arp_packet_handler = f;
613 }
614
615 #ifdef CONFIG_CMD_TFTPPUT
616 void net_set_icmp_handler(rxhand_icmp_f *f)
617 {
618         packet_icmp_handler = f;
619 }
620 #endif
621
622 void
623 NetSetTimeout(ulong iv, thand_f *f)
624 {
625         if (iv == 0) {
626                 timeHandler = (thand_f *)0;
627         } else {
628                 timeHandler = f;
629                 timeStart = get_timer(0);
630                 timeDelta = iv;
631         }
632 }
633
634 int NetSendUDPPacket(uchar *ether, IPaddr_t dest, int dport, int sport,
635                 int payload_len)
636 {
637         uchar *pkt;
638         int eth_hdr_size;
639         int pkt_hdr_size;
640
641         /* make sure the NetTxPacket is initialized (NetInit() was called) */
642         assert(NetTxPacket != NULL);
643         if (NetTxPacket == NULL)
644                 return -1;
645
646         /* convert to new style broadcast */
647         if (dest == 0)
648                 dest = 0xFFFFFFFF;
649
650         /* if broadcast, make the ether address a broadcast and don't do ARP */
651         if (dest == 0xFFFFFFFF)
652                 ether = NetBcastAddr;
653
654         pkt = (uchar *)NetTxPacket;
655
656         eth_hdr_size = NetSetEther(pkt, ether, PROT_IP);
657         pkt += eth_hdr_size;
658         net_set_udp_header(pkt, dest, dport, sport, payload_len);
659         pkt_hdr_size = eth_hdr_size + IP_UDP_HDR_SIZE;
660
661         /* if MAC address was not discovered yet, do an ARP request */
662         if (memcmp(ether, NetEtherNullAddr, 6) == 0) {
663                 debug("sending ARP for %pI4\n", &dest);
664
665                 /* save the ip and eth addr for the packet to send after arp */
666                 NetArpWaitPacketIP = dest;
667                 NetArpWaitPacketMAC = ether;
668
669                 /* size of the waiting packet */
670                 NetArpWaitTxPacketSize = pkt_hdr_size + payload_len;
671
672                 /* and do the ARP request */
673                 NetArpWaitTry = 1;
674                 NetArpWaitTimerStart = get_timer(0);
675                 ArpRequest();
676                 return 1;       /* waiting */
677         } else {
678                 debug("sending UDP to %pI4/%pM\n", &dest, ether);
679                 NetSendPacket(NetTxPacket, pkt_hdr_size + payload_len);
680                 return 0;       /* transmitted */
681         }
682 }
683
684 #ifdef CONFIG_IP_DEFRAG
685 /*
686  * This function collects fragments in a single packet, according
687  * to the algorithm in RFC815. It returns NULL or the pointer to
688  * a complete packet, in static storage
689  */
690 #ifndef CONFIG_NET_MAXDEFRAG
691 #define CONFIG_NET_MAXDEFRAG 16384
692 #endif
693 /*
694  * MAXDEFRAG, above, is chosen in the config file and  is real data
695  * so we need to add the NFS overhead, which is more than TFTP.
696  * To use sizeof in the internal unnamed structures, we need a real
697  * instance (can't do "sizeof(struct rpc_t.u.reply))", unfortunately).
698  * The compiler doesn't complain nor allocates the actual structure
699  */
700 static struct rpc_t rpc_specimen;
701 #define IP_PKTSIZE (CONFIG_NET_MAXDEFRAG + sizeof(rpc_specimen.u.reply))
702
703 #define IP_MAXUDP (IP_PKTSIZE - IP_HDR_SIZE)
704
705 /*
706  * this is the packet being assembled, either data or frag control.
707  * Fragments go by 8 bytes, so this union must be 8 bytes long
708  */
709 struct hole {
710         /* first_byte is address of this structure */
711         u16 last_byte;  /* last byte in this hole + 1 (begin of next hole) */
712         u16 next_hole;  /* index of next (in 8-b blocks), 0 == none */
713         u16 prev_hole;  /* index of prev, 0 == none */
714         u16 unused;
715 };
716
717 static struct ip_udp_hdr *__NetDefragment(struct ip_udp_hdr *ip, int *lenp)
718 {
719         static uchar pkt_buff[IP_PKTSIZE] __aligned(PKTALIGN);
720         static u16 first_hole, total_len;
721         struct hole *payload, *thisfrag, *h, *newh;
722         struct ip_udp_hdr *localip = (struct ip_udp_hdr *)pkt_buff;
723         uchar *indata = (uchar *)ip;
724         int offset8, start, len, done = 0;
725         u16 ip_off = ntohs(ip->ip_off);
726
727         /* payload starts after IP header, this fragment is in there */
728         payload = (struct hole *)(pkt_buff + IP_HDR_SIZE);
729         offset8 =  (ip_off & IP_OFFS);
730         thisfrag = payload + offset8;
731         start = offset8 * 8;
732         len = ntohs(ip->ip_len) - IP_HDR_SIZE;
733
734         if (start + len > IP_MAXUDP) /* fragment extends too far */
735                 return NULL;
736
737         if (!total_len || localip->ip_id != ip->ip_id) {
738                 /* new (or different) packet, reset structs */
739                 total_len = 0xffff;
740                 payload[0].last_byte = ~0;
741                 payload[0].next_hole = 0;
742                 payload[0].prev_hole = 0;
743                 first_hole = 0;
744                 /* any IP header will work, copy the first we received */
745                 memcpy(localip, ip, IP_HDR_SIZE);
746         }
747
748         /*
749          * What follows is the reassembly algorithm. We use the payload
750          * array as a linked list of hole descriptors, as each hole starts
751          * at a multiple of 8 bytes. However, last byte can be whatever value,
752          * so it is represented as byte count, not as 8-byte blocks.
753          */
754
755         h = payload + first_hole;
756         while (h->last_byte < start) {
757                 if (!h->next_hole) {
758                         /* no hole that far away */
759                         return NULL;
760                 }
761                 h = payload + h->next_hole;
762         }
763
764         /* last fragment may be 1..7 bytes, the "+7" forces acceptance */
765         if (offset8 + ((len + 7) / 8) <= h - payload) {
766                 /* no overlap with holes (dup fragment?) */
767                 return NULL;
768         }
769
770         if (!(ip_off & IP_FLAGS_MFRAG)) {
771                 /* no more fragmentss: truncate this (last) hole */
772                 total_len = start + len;
773                 h->last_byte = start + len;
774         }
775
776         /*
777          * There is some overlap: fix the hole list. This code doesn't
778          * deal with a fragment that overlaps with two different holes
779          * (thus being a superset of a previously-received fragment).
780          */
781
782         if ((h >= thisfrag) && (h->last_byte <= start + len)) {
783                 /* complete overlap with hole: remove hole */
784                 if (!h->prev_hole && !h->next_hole) {
785                         /* last remaining hole */
786                         done = 1;
787                 } else if (!h->prev_hole) {
788                         /* first hole */
789                         first_hole = h->next_hole;
790                         payload[h->next_hole].prev_hole = 0;
791                 } else if (!h->next_hole) {
792                         /* last hole */
793                         payload[h->prev_hole].next_hole = 0;
794                 } else {
795                         /* in the middle of the list */
796                         payload[h->next_hole].prev_hole = h->prev_hole;
797                         payload[h->prev_hole].next_hole = h->next_hole;
798                 }
799
800         } else if (h->last_byte <= start + len) {
801                 /* overlaps with final part of the hole: shorten this hole */
802                 h->last_byte = start;
803
804         } else if (h >= thisfrag) {
805                 /* overlaps with initial part of the hole: move this hole */
806                 newh = thisfrag + (len / 8);
807                 *newh = *h;
808                 h = newh;
809                 if (h->next_hole)
810                         payload[h->next_hole].prev_hole = (h - payload);
811                 if (h->prev_hole)
812                         payload[h->prev_hole].next_hole = (h - payload);
813                 else
814                         first_hole = (h - payload);
815
816         } else {
817                 /* fragment sits in the middle: split the hole */
818                 newh = thisfrag + (len / 8);
819                 *newh = *h;
820                 h->last_byte = start;
821                 h->next_hole = (newh - payload);
822                 newh->prev_hole = (h - payload);
823                 if (newh->next_hole)
824                         payload[newh->next_hole].prev_hole = (newh - payload);
825         }
826
827         /* finally copy this fragment and possibly return whole packet */
828         memcpy((uchar *)thisfrag, indata + IP_HDR_SIZE, len);
829         if (!done)
830                 return NULL;
831
832         localip->ip_len = htons(total_len);
833         *lenp = total_len + IP_HDR_SIZE;
834         return localip;
835 }
836
837 static inline struct ip_udp_hdr *NetDefragment(struct ip_udp_hdr *ip, int *lenp)
838 {
839         u16 ip_off = ntohs(ip->ip_off);
840         if (!(ip_off & (IP_OFFS | IP_FLAGS_MFRAG)))
841                 return ip; /* not a fragment */
842         return __NetDefragment(ip, lenp);
843 }
844
845 #else /* !CONFIG_IP_DEFRAG */
846
847 static inline struct ip_udp_hdr *NetDefragment(struct ip_udp_hdr *ip, int *lenp)
848 {
849         u16 ip_off = ntohs(ip->ip_off);
850         if (!(ip_off & (IP_OFFS | IP_FLAGS_MFRAG)))
851                 return ip; /* not a fragment */
852         return NULL;
853 }
854 #endif
855
856 /**
857  * Receive an ICMP packet. We deal with REDIRECT and PING here, and silently
858  * drop others.
859  *
860  * @parma ip    IP packet containing the ICMP
861  */
862 static void receive_icmp(struct ip_udp_hdr *ip, int len,
863                         IPaddr_t src_ip, struct ethernet_hdr *et)
864 {
865         struct icmp_hdr *icmph = (struct icmp_hdr *)&ip->udp_src;
866
867         switch (icmph->type) {
868         case ICMP_REDIRECT:
869                 if (icmph->code != ICMP_REDIR_HOST)
870                         return;
871                 printf(" ICMP Host Redirect to %pI4 ",
872                         &icmph->un.gateway);
873                 break;
874         default:
875 #if defined(CONFIG_CMD_PING)
876                 ping_receive(et, ip, len);
877 #endif
878 #ifdef CONFIG_CMD_TFTPPUT
879                 if (packet_icmp_handler)
880                         packet_icmp_handler(icmph->type, icmph->code,
881                                 ntohs(ip->udp_dst), src_ip, ntohs(ip->udp_src),
882                                 icmph->un.data, ntohs(ip->udp_len));
883 #endif
884                 break;
885         }
886 }
887
888 void
889 NetReceive(uchar *inpkt, int len)
890 {
891         struct ethernet_hdr *et;
892         struct ip_udp_hdr *ip;
893         IPaddr_t dst_ip;
894         IPaddr_t src_ip;
895         int eth_proto;
896 #if defined(CONFIG_CMD_CDP)
897         int iscdp;
898 #endif
899         ushort cti = 0, vlanid = VLAN_NONE, myvlanid, mynvlanid;
900
901         debug("packet received\n");
902
903         NetRxPacket = inpkt;
904         NetRxPacketLen = len;
905         et = (struct ethernet_hdr *)inpkt;
906
907         /* too small packet? */
908         if (len < ETHER_HDR_SIZE)
909                 return;
910
911 #ifdef CONFIG_API
912         if (push_packet) {
913                 (*push_packet)(inpkt, len);
914                 return;
915         }
916 #endif
917
918 #if defined(CONFIG_CMD_CDP)
919         /* keep track if packet is CDP */
920         iscdp = is_cdp_packet(et->et_dest);
921 #endif
922
923         myvlanid = ntohs(NetOurVLAN);
924         if (myvlanid == (ushort)-1)
925                 myvlanid = VLAN_NONE;
926         mynvlanid = ntohs(NetOurNativeVLAN);
927         if (mynvlanid == (ushort)-1)
928                 mynvlanid = VLAN_NONE;
929
930         eth_proto = ntohs(et->et_protlen);
931
932         debug("packet received\n");
933
934         if (eth_proto < 1514) {
935                 struct e802_hdr *et802 = (struct e802_hdr *)et;
936                 /*
937                  *      Got a 802.2 packet.  Check the other protocol field.
938                  *      XXX VLAN over 802.2+SNAP not implemented!
939                  */
940                 eth_proto = ntohs(et802->et_prot);
941
942                 ip = (struct ip_udp_hdr *)(inpkt + E802_HDR_SIZE);
943                 len -= E802_HDR_SIZE;
944
945         } else if (eth_proto != PROT_VLAN) {    /* normal packet */
946                 ip = (struct ip_udp_hdr *)(inpkt + ETHER_HDR_SIZE);
947                 len -= ETHER_HDR_SIZE;
948
949         } else {                        /* VLAN packet */
950                 struct vlan_ethernet_hdr *vet =
951                         (struct vlan_ethernet_hdr *)et;
952
953                 debug("VLAN packet received\n");
954
955                 /* too small packet? */
956                 if (len < VLAN_ETHER_HDR_SIZE)
957                         return;
958
959                 /* if no VLAN active */
960                 if ((ntohs(NetOurVLAN) & VLAN_IDMASK) == VLAN_NONE
961 #if defined(CONFIG_CMD_CDP)
962                                 && iscdp == 0
963 #endif
964                                 )
965                         return;
966
967                 cti = ntohs(vet->vet_tag);
968                 vlanid = cti & VLAN_IDMASK;
969                 eth_proto = ntohs(vet->vet_type);
970
971                 ip = (struct ip_udp_hdr *)(inpkt + VLAN_ETHER_HDR_SIZE);
972                 len -= VLAN_ETHER_HDR_SIZE;
973         }
974
975         debug("Receive from protocol 0x%x\n", eth_proto);
976
977 #if defined(CONFIG_CMD_CDP)
978         if (iscdp) {
979                 cdp_receive((uchar *)ip, len);
980                 return;
981         }
982 #endif
983
984         if ((myvlanid & VLAN_IDMASK) != VLAN_NONE) {
985                 if (vlanid == VLAN_NONE)
986                         vlanid = (mynvlanid & VLAN_IDMASK);
987                 /* not matched? */
988                 if (vlanid != (myvlanid & VLAN_IDMASK))
989                         return;
990         }
991
992         switch (eth_proto) {
993
994         case PROT_ARP:
995                 ArpReceive(et, ip, len);
996                 break;
997
998 #ifdef CONFIG_CMD_RARP
999         case PROT_RARP:
1000                 rarp_receive(ip, len);
1001                 break;
1002 #endif
1003         case PROT_IP:
1004                 debug("Got IP\n");
1005                 /* Before we start poking the header, make sure it is there */
1006                 if (len < IP_UDP_HDR_SIZE) {
1007                         debug("len bad %d < %lu\n", len,
1008                                 (ulong)IP_UDP_HDR_SIZE);
1009                         return;
1010                 }
1011                 /* Check the packet length */
1012                 if (len < ntohs(ip->ip_len)) {
1013                         printf("len bad %d < %d\n", len, ntohs(ip->ip_len));
1014                         return;
1015                 }
1016                 len = ntohs(ip->ip_len);
1017                 debug("len=%d, v=%02x\n", len, ip->ip_hl_v & 0xff);
1018
1019                 /* Can't deal with anything except IPv4 */
1020                 if ((ip->ip_hl_v & 0xf0) != 0x40)
1021                         return;
1022                 /* Can't deal with IP options (headers != 20 bytes) */
1023                 if ((ip->ip_hl_v & 0x0f) > 0x05)
1024                         return;
1025                 /* Check the Checksum of the header */
1026                 if (!NetCksumOk((uchar *)ip, IP_HDR_SIZE / 2)) {
1027                         puts("checksum bad\n");
1028                         return;
1029                 }
1030                 /* If it is not for us, ignore it */
1031                 dst_ip = NetReadIP(&ip->ip_dst);
1032                 if (NetOurIP && dst_ip != NetOurIP && dst_ip != 0xFFFFFFFF) {
1033 #ifdef CONFIG_MCAST_TFTP
1034                         if (Mcast_addr != dst_ip)
1035 #endif
1036                                 return;
1037                 }
1038                 /* Read source IP address for later use */
1039                 src_ip = NetReadIP(&ip->ip_src);
1040                 /*
1041                  * The function returns the unchanged packet if it's not
1042                  * a fragment, and either the complete packet or NULL if
1043                  * it is a fragment (if !CONFIG_IP_DEFRAG, it returns NULL)
1044                  */
1045                 ip = NetDefragment(ip, &len);
1046                 if (!ip)
1047                         return;
1048                 /*
1049                  * watch for ICMP host redirects
1050                  *
1051                  * There is no real handler code (yet). We just watch
1052                  * for ICMP host redirect messages. In case anybody
1053                  * sees these messages: please contact me
1054                  * (wd@denx.de), or - even better - send me the
1055                  * necessary fixes :-)
1056                  *
1057                  * Note: in all cases where I have seen this so far
1058                  * it was a problem with the router configuration,
1059                  * for instance when a router was configured in the
1060                  * BOOTP reply, but the TFTP server was on the same
1061                  * subnet. So this is probably a warning that your
1062                  * configuration might be wrong. But I'm not really
1063                  * sure if there aren't any other situations.
1064                  *
1065                  * Simon Glass <sjg@chromium.org>: We get an ICMP when
1066                  * we send a tftp packet to a dead connection, or when
1067                  * there is no server at the other end.
1068                  */
1069                 if (ip->ip_p == IPPROTO_ICMP) {
1070                         receive_icmp(ip, len, src_ip, et);
1071                         return;
1072                 } else if (ip->ip_p != IPPROTO_UDP) {   /* Only UDP packets */
1073                         return;
1074                 }
1075
1076 #ifdef CONFIG_UDP_CHECKSUM
1077                 if (ip->udp_xsum != 0) {
1078                         ulong   xsum;
1079                         ushort *sumptr;
1080                         ushort  sumlen;
1081
1082                         xsum  = ip->ip_p;
1083                         xsum += (ntohs(ip->udp_len));
1084                         xsum += (ntohl(ip->ip_src) >> 16) & 0x0000ffff;
1085                         xsum += (ntohl(ip->ip_src) >>  0) & 0x0000ffff;
1086                         xsum += (ntohl(ip->ip_dst) >> 16) & 0x0000ffff;
1087                         xsum += (ntohl(ip->ip_dst) >>  0) & 0x0000ffff;
1088
1089                         sumlen = ntohs(ip->udp_len);
1090                         sumptr = (ushort *) &(ip->udp_src);
1091
1092                         while (sumlen > 1) {
1093                                 ushort sumdata;
1094
1095                                 sumdata = *sumptr++;
1096                                 xsum += ntohs(sumdata);
1097                                 sumlen -= 2;
1098                         }
1099                         if (sumlen > 0) {
1100                                 ushort sumdata;
1101
1102                                 sumdata = *(unsigned char *) sumptr;
1103                                 sumdata = (sumdata << 8) & 0xff00;
1104                                 xsum += sumdata;
1105                         }
1106                         while ((xsum >> 16) != 0) {
1107                                 xsum = (xsum & 0x0000ffff) +
1108                                        ((xsum >> 16) & 0x0000ffff);
1109                         }
1110                         if ((xsum != 0x00000000) && (xsum != 0x0000ffff)) {
1111                                 printf(" UDP wrong checksum %08lx %08x\n",
1112                                         xsum, ntohs(ip->udp_xsum));
1113                                 return;
1114                         }
1115                 }
1116 #endif
1117
1118
1119 #ifdef CONFIG_NETCONSOLE
1120                 nc_input_packet((uchar *)ip + IP_UDP_HDR_SIZE,
1121                                         ntohs(ip->udp_dst),
1122                                         ntohs(ip->udp_src),
1123                                         ntohs(ip->udp_len) - UDP_HDR_SIZE);
1124 #endif
1125                 /*
1126                  *      IP header OK.  Pass the packet to the current handler.
1127                  */
1128                 (*udp_packet_handler)((uchar *)ip + IP_UDP_HDR_SIZE,
1129                                 ntohs(ip->udp_dst),
1130                                 src_ip,
1131                                 ntohs(ip->udp_src),
1132                                 ntohs(ip->udp_len) - UDP_HDR_SIZE);
1133                 break;
1134         }
1135 }
1136
1137
1138 /**********************************************************************/
1139
1140 static int net_check_prereq(enum proto_t protocol)
1141 {
1142         switch (protocol) {
1143                 /* Fall through */
1144 #if defined(CONFIG_CMD_PING)
1145         case PING:
1146                 if (NetPingIP == 0) {
1147                         puts("*** ERROR: ping address not given\n");
1148                         return 1;
1149                 }
1150                 goto common;
1151 #endif
1152 #if defined(CONFIG_CMD_SNTP)
1153         case SNTP:
1154                 if (NetNtpServerIP == 0) {
1155                         puts("*** ERROR: NTP server address not given\n");
1156                         return 1;
1157                 }
1158                 goto common;
1159 #endif
1160 #if defined(CONFIG_CMD_DNS)
1161         case DNS:
1162                 if (NetOurDNSIP == 0) {
1163                         puts("*** ERROR: DNS server address not given\n");
1164                         return 1;
1165                 }
1166                 goto common;
1167 #endif
1168 #if defined(CONFIG_CMD_NFS)
1169         case NFS:
1170 #endif
1171         case TFTPGET:
1172         case TFTPPUT:
1173                 if (NetServerIP == 0) {
1174                         puts("*** ERROR: `serverip' not set\n");
1175                         return 1;
1176                 }
1177 #if     defined(CONFIG_CMD_PING) || defined(CONFIG_CMD_SNTP) || \
1178         defined(CONFIG_CMD_DNS)
1179 common:
1180 #endif
1181                 /* Fall through */
1182
1183         case NETCONS:
1184         case TFTPSRV:
1185                 if (NetOurIP == 0) {
1186                         puts("*** ERROR: `ipaddr' not set\n");
1187                         return 1;
1188                 }
1189                 /* Fall through */
1190
1191 #ifdef CONFIG_CMD_RARP
1192         case RARP:
1193 #endif
1194         case BOOTP:
1195         case CDP:
1196         case DHCP:
1197                 if (memcmp(NetOurEther, "\0\0\0\0\0\0", 6) == 0) {
1198                         int num = eth_get_dev_index();
1199
1200                         switch (num) {
1201                         case -1:
1202                                 puts("*** ERROR: No ethernet found.\n");
1203                                 return 1;
1204                         case 0:
1205                                 puts("*** ERROR: `ethaddr' not set\n");
1206                                 break;
1207                         default:
1208                                 printf("*** ERROR: `eth%daddr' not set\n",
1209                                         num);
1210                                 break;
1211                         }
1212
1213                         NetStartAgain();
1214                         return 2;
1215                 }
1216                 /* Fall through */
1217         default:
1218                 return 0;
1219         }
1220         return 0;               /* OK */
1221 }
1222 /**********************************************************************/
1223
1224 int
1225 NetCksumOk(uchar *ptr, int len)
1226 {
1227         return !((NetCksum(ptr, len) + 1) & 0xfffe);
1228 }
1229
1230
1231 unsigned
1232 NetCksum(uchar *ptr, int len)
1233 {
1234         ulong   xsum;
1235         ushort *p = (ushort *)ptr;
1236
1237         xsum = 0;
1238         while (len-- > 0)
1239                 xsum += *p++;
1240         xsum = (xsum & 0xffff) + (xsum >> 16);
1241         xsum = (xsum & 0xffff) + (xsum >> 16);
1242         return xsum & 0xffff;
1243 }
1244
1245 int
1246 NetEthHdrSize(void)
1247 {
1248         ushort myvlanid;
1249
1250         myvlanid = ntohs(NetOurVLAN);
1251         if (myvlanid == (ushort)-1)
1252                 myvlanid = VLAN_NONE;
1253
1254         return ((myvlanid & VLAN_IDMASK) == VLAN_NONE) ? ETHER_HDR_SIZE :
1255                 VLAN_ETHER_HDR_SIZE;
1256 }
1257
1258 int
1259 NetSetEther(uchar *xet, uchar * addr, uint prot)
1260 {
1261         struct ethernet_hdr *et = (struct ethernet_hdr *)xet;
1262         ushort myvlanid;
1263
1264         myvlanid = ntohs(NetOurVLAN);
1265         if (myvlanid == (ushort)-1)
1266                 myvlanid = VLAN_NONE;
1267
1268         memcpy(et->et_dest, addr, 6);
1269         memcpy(et->et_src, NetOurEther, 6);
1270         if ((myvlanid & VLAN_IDMASK) == VLAN_NONE) {
1271                 et->et_protlen = htons(prot);
1272                 return ETHER_HDR_SIZE;
1273         } else {
1274                 struct vlan_ethernet_hdr *vet =
1275                         (struct vlan_ethernet_hdr *)xet;
1276
1277                 vet->vet_vlan_type = htons(PROT_VLAN);
1278                 vet->vet_tag = htons((0 << 5) | (myvlanid & VLAN_IDMASK));
1279                 vet->vet_type = htons(prot);
1280                 return VLAN_ETHER_HDR_SIZE;
1281         }
1282 }
1283
1284 int net_update_ether(struct ethernet_hdr *et, uchar *addr, uint prot)
1285 {
1286         ushort protlen;
1287
1288         memcpy(et->et_dest, addr, 6);
1289         memcpy(et->et_src, NetOurEther, 6);
1290         protlen = ntohs(et->et_protlen);
1291         if (protlen == PROT_VLAN) {
1292                 struct vlan_ethernet_hdr *vet =
1293                         (struct vlan_ethernet_hdr *)et;
1294                 vet->vet_type = htons(prot);
1295                 return VLAN_ETHER_HDR_SIZE;
1296         } else if (protlen > 1514) {
1297                 et->et_protlen = htons(prot);
1298                 return ETHER_HDR_SIZE;
1299         } else {
1300                 /* 802.2 + SNAP */
1301                 struct e802_hdr *et802 = (struct e802_hdr *)et;
1302                 et802->et_prot = htons(prot);
1303                 return E802_HDR_SIZE;
1304         }
1305 }
1306
1307 void net_set_ip_header(uchar *pkt, IPaddr_t dest, IPaddr_t source)
1308 {
1309         struct ip_udp_hdr *ip = (struct ip_udp_hdr *)pkt;
1310
1311         /*
1312          *      Construct an IP header.
1313          */
1314         /* IP_HDR_SIZE / 4 (not including UDP) */
1315         ip->ip_hl_v  = 0x45;
1316         ip->ip_tos   = 0;
1317         ip->ip_len   = htons(IP_HDR_SIZE);
1318         ip->ip_id    = htons(NetIPID++);
1319         ip->ip_off   = htons(IP_FLAGS_DFRAG);   /* Don't fragment */
1320         ip->ip_ttl   = 255;
1321         ip->ip_sum   = 0;
1322         /* already in network byte order */
1323         NetCopyIP((void *)&ip->ip_src, &source);
1324         /* already in network byte order */
1325         NetCopyIP((void *)&ip->ip_dst, &dest);
1326 }
1327
1328 void net_set_udp_header(uchar *pkt, IPaddr_t dest, int dport, int sport,
1329                         int len)
1330 {
1331         struct ip_udp_hdr *ip = (struct ip_udp_hdr *)pkt;
1332
1333         /*
1334          *      If the data is an odd number of bytes, zero the
1335          *      byte after the last byte so that the checksum
1336          *      will work.
1337          */
1338         if (len & 1)
1339                 pkt[IP_UDP_HDR_SIZE + len] = 0;
1340
1341         net_set_ip_header(pkt, dest, NetOurIP);
1342         ip->ip_len   = htons(IP_UDP_HDR_SIZE + len);
1343         ip->ip_p     = IPPROTO_UDP;
1344         ip->ip_sum   = ~NetCksum((uchar *)ip, IP_HDR_SIZE >> 1);
1345
1346         ip->udp_src  = htons(sport);
1347         ip->udp_dst  = htons(dport);
1348         ip->udp_len  = htons(UDP_HDR_SIZE + len);
1349         ip->udp_xsum = 0;
1350 }
1351
1352 void copy_filename(char *dst, const char *src, int size)
1353 {
1354         if (*src && (*src == '"')) {
1355                 ++src;
1356                 --size;
1357         }
1358
1359         while ((--size > 0) && *src && (*src != '"'))
1360                 *dst++ = *src++;
1361         *dst = '\0';
1362 }
1363
1364 #if     defined(CONFIG_CMD_NFS)         || \
1365         defined(CONFIG_CMD_SNTP)        || \
1366         defined(CONFIG_CMD_DNS)
1367 /*
1368  * make port a little random (1024-17407)
1369  * This keeps the math somewhat trivial to compute, and seems to work with
1370  * all supported protocols/clients/servers
1371  */
1372 unsigned int random_port(void)
1373 {
1374         return 1024 + (get_timer(0) % 0x4000);
1375 }
1376 #endif
1377
1378 void ip_to_string(IPaddr_t x, char *s)
1379 {
1380         x = ntohl(x);
1381         sprintf(s, "%d.%d.%d.%d",
1382                 (int) ((x >> 24) & 0xff),
1383                 (int) ((x >> 16) & 0xff),
1384                 (int) ((x >> 8) & 0xff), (int) ((x >> 0) & 0xff)
1385         );
1386 }
1387
1388 void VLAN_to_string(ushort x, char *s)
1389 {
1390         x = ntohs(x);
1391
1392         if (x == (ushort)-1)
1393                 x = VLAN_NONE;
1394
1395         if (x == VLAN_NONE)
1396                 strcpy(s, "none");
1397         else
1398                 sprintf(s, "%d", x & VLAN_IDMASK);
1399 }
1400
1401 ushort string_to_VLAN(const char *s)
1402 {
1403         ushort id;
1404
1405         if (s == NULL)
1406                 return htons(VLAN_NONE);
1407
1408         if (*s < '0' || *s > '9')
1409                 id = VLAN_NONE;
1410         else
1411                 id = (ushort)simple_strtoul(s, NULL, 10);
1412
1413         return htons(id);
1414 }
1415
1416 ushort getenv_VLAN(char *var)
1417 {
1418         return string_to_VLAN(getenv(var));
1419 }