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