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