Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 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 |
Wolfgang Denk | 2ea9103 | 2014-09-30 10:44:01 +0200 | [diff] [blame] | 9 | * SPDX-License-Identifier: GPL-2.0 |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 10 | */ |
| 11 | |
| 12 | #include "ping.h" |
| 13 | #include "arp.h" |
| 14 | |
| 15 | static ushort PingSeqNo; |
| 16 | |
| 17 | /* The ip address to ping */ |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 18 | struct in_addr net_ping_ip; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 19 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 20 | static void set_icmp_header(uchar *pkt, struct in_addr dest) |
Joe Hershberger | 4b11c91 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 21 | { |
| 22 | /* |
| 23 | * Construct an IP and ICMP header. |
| 24 | */ |
| 25 | struct ip_hdr *ip = (struct ip_hdr *)pkt; |
| 26 | struct icmp_hdr *icmp = (struct icmp_hdr *)(pkt + IP_HDR_SIZE); |
| 27 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 28 | net_set_ip_header(pkt, dest, net_ip); |
Joe Hershberger | 4b11c91 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 29 | |
| 30 | ip->ip_len = htons(IP_ICMP_HDR_SIZE); |
| 31 | ip->ip_p = IPPROTO_ICMP; |
Simon Glass | 0da0fcd | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 32 | ip->ip_sum = compute_ip_checksum(ip, IP_HDR_SIZE); |
Joe Hershberger | 4b11c91 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 33 | |
| 34 | icmp->type = ICMP_ECHO_REQUEST; |
| 35 | icmp->code = 0; |
| 36 | icmp->checksum = 0; |
| 37 | icmp->un.echo.id = 0; |
| 38 | icmp->un.echo.sequence = htons(PingSeqNo++); |
Simon Glass | 0da0fcd | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 39 | icmp->checksum = compute_ip_checksum(icmp, ICMP_HDR_SIZE); |
Joe Hershberger | 4b11c91 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 40 | } |
| 41 | |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 42 | static int ping_send(void) |
| 43 | { |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 44 | uchar *pkt; |
Joe Hershberger | 00f3326 | 2012-05-23 07:59:09 +0000 | [diff] [blame] | 45 | int eth_hdr_size; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 46 | |
| 47 | /* XXX always send arp request */ |
| 48 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 49 | debug_cond(DEBUG_DEV_PKT, "sending ARP for %pI4\n", &net_ping_ip); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 50 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 51 | net_arp_wait_packet_ip = net_ping_ip; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 52 | |
Joe Hershberger | 1203fcc | 2015-04-08 01:41:05 -0500 | [diff] [blame^] | 53 | eth_hdr_size = net_set_ether(net_tx_packet, net_null_ethaddr, PROT_IP); |
| 54 | pkt = (uchar *)net_tx_packet + eth_hdr_size; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 55 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 56 | set_icmp_header(pkt, net_ping_ip); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 57 | |
| 58 | /* size of the waiting packet */ |
Joe Hershberger | 00f3326 | 2012-05-23 07:59:09 +0000 | [diff] [blame] | 59 | NetArpWaitTxPacketSize = eth_hdr_size + IP_ICMP_HDR_SIZE; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 60 | |
| 61 | /* and do the ARP request */ |
| 62 | NetArpWaitTry = 1; |
| 63 | NetArpWaitTimerStart = get_timer(0); |
| 64 | ArpRequest(); |
| 65 | return 1; /* waiting */ |
| 66 | } |
| 67 | |
| 68 | static void ping_timeout(void) |
| 69 | { |
| 70 | eth_halt(); |
Joe Hershberger | 22f6e99 | 2012-05-23 07:59:14 +0000 | [diff] [blame] | 71 | net_set_state(NETLOOP_FAIL); /* we did not get the reply */ |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 72 | } |
| 73 | |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 74 | void ping_start(void) |
| 75 | { |
| 76 | printf("Using %s device\n", eth_get_name()); |
| 77 | NetSetTimeout(10000UL, ping_timeout); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 78 | |
| 79 | ping_send(); |
| 80 | } |
| 81 | |
Joe Hershberger | cb487f5 | 2012-05-23 07:58:06 +0000 | [diff] [blame] | 82 | void ping_receive(struct ethernet_hdr *et, struct ip_udp_hdr *ip, int len) |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 83 | { |
Joe Hershberger | e0a6307 | 2012-05-23 07:58:09 +0000 | [diff] [blame] | 84 | struct icmp_hdr *icmph = (struct icmp_hdr *)&ip->udp_src; |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 85 | struct in_addr src_ip; |
Joe Hershberger | e711101 | 2012-05-23 07:59:16 +0000 | [diff] [blame] | 86 | int eth_hdr_size; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 87 | |
| 88 | switch (icmph->type) { |
| 89 | case ICMP_ECHO_REPLY: |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 90 | src_ip = net_read_ip((void *)&ip->ip_src); |
| 91 | if (src_ip.s_addr == net_ping_ip.s_addr) |
Joe Hershberger | 22f6e99 | 2012-05-23 07:59:14 +0000 | [diff] [blame] | 92 | net_set_state(NETLOOP_SUCCESS); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 93 | return; |
| 94 | case ICMP_ECHO_REQUEST: |
Joe Hershberger | e711101 | 2012-05-23 07:59:16 +0000 | [diff] [blame] | 95 | eth_hdr_size = net_update_ether(et, et->et_src, PROT_IP); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 96 | |
Joe Hershberger | 4ef8d53 | 2012-05-23 08:01:04 +0000 | [diff] [blame] | 97 | debug_cond(DEBUG_DEV_PKT, "Got ICMP ECHO REQUEST, return " |
Joe Hershberger | e711101 | 2012-05-23 07:59:16 +0000 | [diff] [blame] | 98 | "%d bytes\n", eth_hdr_size + len); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 99 | |
| 100 | ip->ip_sum = 0; |
| 101 | ip->ip_off = 0; |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 102 | net_copy_ip((void *)&ip->ip_dst, &ip->ip_src); |
| 103 | net_copy_ip((void *)&ip->ip_src, &net_ip); |
Simon Glass | 0da0fcd | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 104 | ip->ip_sum = compute_ip_checksum(ip, IP_HDR_SIZE); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 105 | |
| 106 | icmph->type = ICMP_ECHO_REPLY; |
| 107 | icmph->checksum = 0; |
Simon Glass | 0da0fcd | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 108 | icmph->checksum = compute_ip_checksum(icmph, len - IP_HDR_SIZE); |
Joe Hershberger | 1203fcc | 2015-04-08 01:41:05 -0500 | [diff] [blame^] | 109 | net_send_packet((uchar *)et, eth_hdr_size + len); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 110 | return; |
| 111 | /* default: |
| 112 | return;*/ |
| 113 | } |
| 114 | } |