Joe Hershberger | c21bf37 | 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 | db06592 | 2014-09-30 10:44:01 +0200 | [diff] [blame] | 9 | * SPDX-License-Identifier: GPL-2.0 |
Joe Hershberger | c21bf37 | 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 */ |
| 18 | IPaddr_t NetPingIP; |
| 19 | |
Joe Hershberger | 2ed5b49 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 20 | static void set_icmp_header(uchar *pkt, IPaddr_t dest) |
| 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 | |
| 28 | net_set_ip_header(pkt, dest, NetOurIP); |
| 29 | |
| 30 | ip->ip_len = htons(IP_ICMP_HDR_SIZE); |
| 31 | ip->ip_p = IPPROTO_ICMP; |
Simon Glass | dfcdcee | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 32 | ip->ip_sum = compute_ip_checksum(ip, IP_HDR_SIZE); |
Joe Hershberger | 2ed5b49 | 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 | dfcdcee | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 39 | icmp->checksum = compute_ip_checksum(icmp, ICMP_HDR_SIZE); |
Joe Hershberger | 2ed5b49 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 40 | } |
| 41 | |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 42 | static int ping_send(void) |
| 43 | { |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 44 | uchar *pkt; |
Joe Hershberger | f7ea805 | 2012-05-23 07:59:09 +0000 | [diff] [blame] | 45 | int eth_hdr_size; |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 46 | |
| 47 | /* XXX always send arp request */ |
| 48 | |
Joe Hershberger | 05a377b | 2012-05-23 08:01:04 +0000 | [diff] [blame] | 49 | debug_cond(DEBUG_DEV_PKT, "sending ARP for %pI4\n", &NetPingIP); |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 50 | |
| 51 | NetArpWaitPacketIP = NetPingIP; |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 52 | |
Joe Hershberger | de8205a | 2012-05-23 07:59:24 +0000 | [diff] [blame] | 53 | eth_hdr_size = NetSetEther(NetTxPacket, NetEtherNullAddr, PROT_IP); |
| 54 | pkt = (uchar *)NetTxPacket + eth_hdr_size; |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 55 | |
Joe Hershberger | 2ed5b49 | 2012-05-23 07:59:07 +0000 | [diff] [blame] | 56 | set_icmp_header(pkt, NetPingIP); |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 57 | |
| 58 | /* size of the waiting packet */ |
Joe Hershberger | f7ea805 | 2012-05-23 07:59:09 +0000 | [diff] [blame] | 59 | NetArpWaitTxPacketSize = eth_hdr_size + IP_ICMP_HDR_SIZE; |
Joe Hershberger | c21bf37 | 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 | d4bb76a | 2012-05-23 07:59:14 +0000 | [diff] [blame] | 71 | net_set_state(NETLOOP_FAIL); /* we did not get the reply */ |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 72 | } |
| 73 | |
Joe Hershberger | c21bf37 | 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 | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 78 | |
| 79 | ping_send(); |
| 80 | } |
| 81 | |
Joe Hershberger | 1178f41 | 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 | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 83 | { |
Joe Hershberger | 7849561 | 2012-05-23 07:58:09 +0000 | [diff] [blame] | 84 | struct icmp_hdr *icmph = (struct icmp_hdr *)&ip->udp_src; |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 85 | IPaddr_t src_ip; |
Joe Hershberger | 530cd6b | 2012-05-23 07:59:16 +0000 | [diff] [blame] | 86 | int eth_hdr_size; |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 87 | |
| 88 | switch (icmph->type) { |
| 89 | case ICMP_ECHO_REPLY: |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 90 | src_ip = NetReadIP((void *)&ip->ip_src); |
Joe Hershberger | 86dba5b | 2012-05-23 07:59:12 +0000 | [diff] [blame] | 91 | if (src_ip == NetPingIP) |
Joe Hershberger | d4bb76a | 2012-05-23 07:59:14 +0000 | [diff] [blame] | 92 | net_set_state(NETLOOP_SUCCESS); |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 93 | return; |
| 94 | case ICMP_ECHO_REQUEST: |
Joe Hershberger | 530cd6b | 2012-05-23 07:59:16 +0000 | [diff] [blame] | 95 | eth_hdr_size = net_update_ether(et, et->et_src, PROT_IP); |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 96 | |
Joe Hershberger | 05a377b | 2012-05-23 08:01:04 +0000 | [diff] [blame] | 97 | debug_cond(DEBUG_DEV_PKT, "Got ICMP ECHO REQUEST, return " |
Joe Hershberger | 530cd6b | 2012-05-23 07:59:16 +0000 | [diff] [blame] | 98 | "%d bytes\n", eth_hdr_size + len); |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 99 | |
| 100 | ip->ip_sum = 0; |
| 101 | ip->ip_off = 0; |
| 102 | NetCopyIP((void *)&ip->ip_dst, &ip->ip_src); |
| 103 | NetCopyIP((void *)&ip->ip_src, &NetOurIP); |
Simon Glass | dfcdcee | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 104 | ip->ip_sum = compute_ip_checksum(ip, IP_HDR_SIZE); |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 105 | |
| 106 | icmph->type = ICMP_ECHO_REPLY; |
| 107 | icmph->checksum = 0; |
Simon Glass | dfcdcee | 2015-01-19 22:16:08 -0700 | [diff] [blame] | 108 | icmph->checksum = compute_ip_checksum(icmph, len - IP_HDR_SIZE); |
Joe Hershberger | 530cd6b | 2012-05-23 07:59:16 +0000 | [diff] [blame] | 109 | NetSendPacket((uchar *)et, eth_hdr_size + len); |
Joe Hershberger | c21bf37 | 2012-05-23 07:58:02 +0000 | [diff] [blame] | 110 | return; |
| 111 | /* default: |
| 112 | return;*/ |
| 113 | } |
| 114 | } |