Tom Rini | f739fcd | 2018-05-07 17:02:21 -0400 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0 |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 2 | /* |
| 3 | * Copied from Linux Monitor (LiMon) - Networking. |
| 4 | * |
| 5 | * Copyright 1994 - 2000 Neil Russell. |
| 6 | * (See License) |
| 7 | * Copyright 2000 Roland Borde |
| 8 | * Copyright 2000 Paolo Scaffardi |
| 9 | * Copyright 2000-2002 Wolfgang Denk, wd@denx.de |
| 10 | */ |
| 11 | |
| 12 | #include "ping.h" |
| 13 | #include "arp.h" |
Simon Glass | 90526e9 | 2020-05-10 11:39:56 -0600 | [diff] [blame^] | 14 | #include <net.h> |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 15 | |
Joe Hershberger | 331db5a | 2015-04-08 01:41:13 -0500 | [diff] [blame] | 16 | static ushort ping_seq_number; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 17 | |
| 18 | /* The ip address to ping */ |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 19 | struct in_addr net_ping_ip; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 20 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 21 | static void set_icmp_header(uchar *pkt, struct in_addr dest) |
Joe Hershberger | 4b11c91 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 22 | { |
| 23 | /* |
| 24 | * Construct an IP and ICMP header. |
| 25 | */ |
Joe Hershberger | 4b11c91 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 26 | struct icmp_hdr *icmp = (struct icmp_hdr *)(pkt + IP_HDR_SIZE); |
| 27 | |
Duncan Hare | 5d457ec | 2018-06-24 15:40:41 -0700 | [diff] [blame] | 28 | net_set_ip_header(pkt, dest, net_ip, IP_ICMP_HDR_SIZE, IPPROTO_ICMP); |
Joe Hershberger | 4b11c91 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 29 | |
| 30 | icmp->type = ICMP_ECHO_REQUEST; |
| 31 | icmp->code = 0; |
| 32 | icmp->checksum = 0; |
| 33 | icmp->un.echo.id = 0; |
Joe Hershberger | 331db5a | 2015-04-08 01:41:13 -0500 | [diff] [blame] | 34 | icmp->un.echo.sequence = htons(ping_seq_number++); |
Simon Glass | 0da0fcd | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 35 | icmp->checksum = compute_ip_checksum(icmp, ICMP_HDR_SIZE); |
Joe Hershberger | 4b11c91 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 36 | } |
| 37 | |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 38 | static int ping_send(void) |
| 39 | { |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 40 | uchar *pkt; |
Joe Hershberger | 00f3326 | 2012-05-23 07:59:09 +0000 | [diff] [blame] | 41 | int eth_hdr_size; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 42 | |
| 43 | /* XXX always send arp request */ |
| 44 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 45 | 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] | 46 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 47 | net_arp_wait_packet_ip = net_ping_ip; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 48 | |
Joe Hershberger | 1203fcc | 2015-04-08 01:41:05 -0500 | [diff] [blame] | 49 | eth_hdr_size = net_set_ether(net_tx_packet, net_null_ethaddr, PROT_IP); |
| 50 | pkt = (uchar *)net_tx_packet + eth_hdr_size; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 51 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 52 | set_icmp_header(pkt, net_ping_ip); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 53 | |
| 54 | /* size of the waiting packet */ |
Joe Hershberger | 85d25e0 | 2015-04-08 01:41:08 -0500 | [diff] [blame] | 55 | arp_wait_tx_packet_size = eth_hdr_size + IP_ICMP_HDR_SIZE; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 56 | |
| 57 | /* and do the ARP request */ |
Joe Hershberger | 85d25e0 | 2015-04-08 01:41:08 -0500 | [diff] [blame] | 58 | arp_wait_try = 1; |
| 59 | arp_wait_timer_start = get_timer(0); |
| 60 | arp_request(); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 61 | return 1; /* waiting */ |
| 62 | } |
| 63 | |
Joe Hershberger | 331db5a | 2015-04-08 01:41:13 -0500 | [diff] [blame] | 64 | static void ping_timeout_handler(void) |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 65 | { |
| 66 | eth_halt(); |
Joe Hershberger | 22f6e99 | 2012-05-23 07:59:14 +0000 | [diff] [blame] | 67 | net_set_state(NETLOOP_FAIL); /* we did not get the reply */ |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 68 | } |
| 69 | |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 70 | void ping_start(void) |
| 71 | { |
| 72 | printf("Using %s device\n", eth_get_name()); |
Joe Hershberger | bc0571f | 2015-04-08 01:41:21 -0500 | [diff] [blame] | 73 | net_set_timeout_handler(10000UL, ping_timeout_handler); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 74 | |
| 75 | ping_send(); |
| 76 | } |
| 77 | |
Joe Hershberger | cb487f5 | 2012-05-23 07:58:06 +0000 | [diff] [blame] | 78 | 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] | 79 | { |
Joe Hershberger | e0a6307 | 2012-05-23 07:58:09 +0000 | [diff] [blame] | 80 | struct icmp_hdr *icmph = (struct icmp_hdr *)&ip->udp_src; |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 81 | struct in_addr src_ip; |
Joe Hershberger | e711101 | 2012-05-23 07:59:16 +0000 | [diff] [blame] | 82 | int eth_hdr_size; |
Joe Hershberger | ac3f26c | 2018-09-26 16:49:02 -0500 | [diff] [blame] | 83 | uchar *tx_packet; |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 84 | |
| 85 | switch (icmph->type) { |
| 86 | case ICMP_ECHO_REPLY: |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 87 | src_ip = net_read_ip((void *)&ip->ip_src); |
| 88 | if (src_ip.s_addr == net_ping_ip.s_addr) |
Joe Hershberger | 22f6e99 | 2012-05-23 07:59:14 +0000 | [diff] [blame] | 89 | net_set_state(NETLOOP_SUCCESS); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 90 | return; |
| 91 | case ICMP_ECHO_REQUEST: |
Joe Hershberger | e711101 | 2012-05-23 07:59:16 +0000 | [diff] [blame] | 92 | eth_hdr_size = net_update_ether(et, et->et_src, PROT_IP); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 93 | |
Joe Hershberger | 331db5a | 2015-04-08 01:41:13 -0500 | [diff] [blame] | 94 | debug_cond(DEBUG_DEV_PKT, |
| 95 | "Got ICMP ECHO REQUEST, return %d bytes\n", |
| 96 | eth_hdr_size + len); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 97 | |
| 98 | ip->ip_sum = 0; |
| 99 | ip->ip_off = 0; |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 100 | net_copy_ip((void *)&ip->ip_dst, &ip->ip_src); |
| 101 | net_copy_ip((void *)&ip->ip_src, &net_ip); |
Simon Glass | 0da0fcd | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 102 | ip->ip_sum = compute_ip_checksum(ip, IP_HDR_SIZE); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 103 | |
| 104 | icmph->type = ICMP_ECHO_REPLY; |
| 105 | icmph->checksum = 0; |
Simon Glass | 0da0fcd | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 106 | icmph->checksum = compute_ip_checksum(icmph, len - IP_HDR_SIZE); |
Joe Hershberger | ac3f26c | 2018-09-26 16:49:02 -0500 | [diff] [blame] | 107 | |
| 108 | tx_packet = net_get_async_tx_pkt_buf(); |
| 109 | memcpy(tx_packet, et, eth_hdr_size + len); |
| 110 | net_send_packet(tx_packet, eth_hdr_size + len); |
Joe Hershberger | a36b12f | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 111 | return; |
| 112 | /* default: |
| 113 | return;*/ |
| 114 | } |
| 115 | } |