blob: 0e33660f6c5faf8db51940826491c256d7c685ba [file] [log] [blame]
Tom Rinif739fcd2018-05-07 17:02:21 -04001// SPDX-License-Identifier: GPL-2.0
Joe Hershbergera36b12f2012-05-23 07:58:02 +00002/*
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 Glassf7ae49f2020-05-10 11:40:05 -060014#include <log.h>
Simon Glass90526e92020-05-10 11:39:56 -060015#include <net.h>
Joe Hershbergera36b12f2012-05-23 07:58:02 +000016
Joe Hershberger331db5a2015-04-08 01:41:13 -050017static ushort ping_seq_number;
Joe Hershbergera36b12f2012-05-23 07:58:02 +000018
19/* The ip address to ping */
Joe Hershberger049a95a2015-04-08 01:41:01 -050020struct in_addr net_ping_ip;
Joe Hershbergera36b12f2012-05-23 07:58:02 +000021
Joe Hershberger049a95a2015-04-08 01:41:01 -050022static void set_icmp_header(uchar *pkt, struct in_addr dest)
Joe Hershberger4b11c912012-05-23 07:59:07 +000023{
24 /*
25 * Construct an IP and ICMP header.
26 */
Joe Hershberger4b11c912012-05-23 07:59:07 +000027 struct icmp_hdr *icmp = (struct icmp_hdr *)(pkt + IP_HDR_SIZE);
28
Duncan Hare5d457ec2018-06-24 15:40:41 -070029 net_set_ip_header(pkt, dest, net_ip, IP_ICMP_HDR_SIZE, IPPROTO_ICMP);
Joe Hershberger4b11c912012-05-23 07:59:07 +000030
31 icmp->type = ICMP_ECHO_REQUEST;
32 icmp->code = 0;
33 icmp->checksum = 0;
34 icmp->un.echo.id = 0;
Joe Hershberger331db5a2015-04-08 01:41:13 -050035 icmp->un.echo.sequence = htons(ping_seq_number++);
Simon Glass0da0fcd2015-01-19 22:16:08 -070036 icmp->checksum = compute_ip_checksum(icmp, ICMP_HDR_SIZE);
Joe Hershberger4b11c912012-05-23 07:59:07 +000037}
38
Joe Hershbergera36b12f2012-05-23 07:58:02 +000039static int ping_send(void)
40{
Joe Hershbergera36b12f2012-05-23 07:58:02 +000041 uchar *pkt;
Joe Hershberger00f33262012-05-23 07:59:09 +000042 int eth_hdr_size;
Joe Hershbergera36b12f2012-05-23 07:58:02 +000043
44 /* XXX always send arp request */
45
Joe Hershberger049a95a2015-04-08 01:41:01 -050046 debug_cond(DEBUG_DEV_PKT, "sending ARP for %pI4\n", &net_ping_ip);
Joe Hershbergera36b12f2012-05-23 07:58:02 +000047
Joe Hershberger049a95a2015-04-08 01:41:01 -050048 net_arp_wait_packet_ip = net_ping_ip;
Joe Hershbergera36b12f2012-05-23 07:58:02 +000049
Joe Hershberger1203fcc2015-04-08 01:41:05 -050050 eth_hdr_size = net_set_ether(net_tx_packet, net_null_ethaddr, PROT_IP);
51 pkt = (uchar *)net_tx_packet + eth_hdr_size;
Joe Hershbergera36b12f2012-05-23 07:58:02 +000052
Joe Hershberger049a95a2015-04-08 01:41:01 -050053 set_icmp_header(pkt, net_ping_ip);
Joe Hershbergera36b12f2012-05-23 07:58:02 +000054
55 /* size of the waiting packet */
Joe Hershberger85d25e02015-04-08 01:41:08 -050056 arp_wait_tx_packet_size = eth_hdr_size + IP_ICMP_HDR_SIZE;
Joe Hershbergera36b12f2012-05-23 07:58:02 +000057
58 /* and do the ARP request */
Joe Hershberger85d25e02015-04-08 01:41:08 -050059 arp_wait_try = 1;
60 arp_wait_timer_start = get_timer(0);
61 arp_request();
Joe Hershbergera36b12f2012-05-23 07:58:02 +000062 return 1; /* waiting */
63}
64
Joe Hershberger331db5a2015-04-08 01:41:13 -050065static void ping_timeout_handler(void)
Joe Hershbergera36b12f2012-05-23 07:58:02 +000066{
67 eth_halt();
Joe Hershberger22f6e992012-05-23 07:59:14 +000068 net_set_state(NETLOOP_FAIL); /* we did not get the reply */
Joe Hershbergera36b12f2012-05-23 07:58:02 +000069}
70
Joe Hershbergera36b12f2012-05-23 07:58:02 +000071void ping_start(void)
72{
73 printf("Using %s device\n", eth_get_name());
Joe Hershbergerbc0571f2015-04-08 01:41:21 -050074 net_set_timeout_handler(10000UL, ping_timeout_handler);
Joe Hershbergera36b12f2012-05-23 07:58:02 +000075
76 ping_send();
77}
78
Joe Hershbergercb487f52012-05-23 07:58:06 +000079void ping_receive(struct ethernet_hdr *et, struct ip_udp_hdr *ip, int len)
Joe Hershbergera36b12f2012-05-23 07:58:02 +000080{
Joe Hershbergere0a63072012-05-23 07:58:09 +000081 struct icmp_hdr *icmph = (struct icmp_hdr *)&ip->udp_src;
Joe Hershberger049a95a2015-04-08 01:41:01 -050082 struct in_addr src_ip;
Joe Hershbergere7111012012-05-23 07:59:16 +000083 int eth_hdr_size;
Joe Hershbergerac3f26c2018-09-26 16:49:02 -050084 uchar *tx_packet;
Joe Hershbergera36b12f2012-05-23 07:58:02 +000085
86 switch (icmph->type) {
87 case ICMP_ECHO_REPLY:
Joe Hershberger049a95a2015-04-08 01:41:01 -050088 src_ip = net_read_ip((void *)&ip->ip_src);
89 if (src_ip.s_addr == net_ping_ip.s_addr)
Joe Hershberger22f6e992012-05-23 07:59:14 +000090 net_set_state(NETLOOP_SUCCESS);
Joe Hershbergera36b12f2012-05-23 07:58:02 +000091 return;
92 case ICMP_ECHO_REQUEST:
Joe Hershbergere7111012012-05-23 07:59:16 +000093 eth_hdr_size = net_update_ether(et, et->et_src, PROT_IP);
Joe Hershbergera36b12f2012-05-23 07:58:02 +000094
Joe Hershberger331db5a2015-04-08 01:41:13 -050095 debug_cond(DEBUG_DEV_PKT,
96 "Got ICMP ECHO REQUEST, return %d bytes\n",
97 eth_hdr_size + len);
Joe Hershbergera36b12f2012-05-23 07:58:02 +000098
99 ip->ip_sum = 0;
100 ip->ip_off = 0;
Joe Hershberger049a95a2015-04-08 01:41:01 -0500101 net_copy_ip((void *)&ip->ip_dst, &ip->ip_src);
102 net_copy_ip((void *)&ip->ip_src, &net_ip);
Simon Glass0da0fcd2015-01-19 22:16:08 -0700103 ip->ip_sum = compute_ip_checksum(ip, IP_HDR_SIZE);
Joe Hershbergera36b12f2012-05-23 07:58:02 +0000104
105 icmph->type = ICMP_ECHO_REPLY;
106 icmph->checksum = 0;
Simon Glass0da0fcd2015-01-19 22:16:08 -0700107 icmph->checksum = compute_ip_checksum(icmph, len - IP_HDR_SIZE);
Joe Hershbergerac3f26c2018-09-26 16:49:02 -0500108
109 tx_packet = net_get_async_tx_pkt_buf();
110 memcpy(tx_packet, et, eth_hdr_size + len);
111 net_send_packet(tx_packet, eth_hdr_size + len);
Joe Hershbergera36b12f2012-05-23 07:58:02 +0000112 return;
113/* default:
114 return;*/
115 }
116}