1
2
3
4
5
6
7
8
9
10
11
12#include "ping.h"
13#include "arp.h"
14#include <log.h>
15#include <net.h>
16
17static ushort ping_seq_number;
18
19
20struct in_addr net_ping_ip;
21
22static void set_icmp_header(uchar *pkt, struct in_addr dest)
23{
24
25
26
27 struct icmp_hdr *icmp = (struct icmp_hdr *)(pkt + IP_HDR_SIZE);
28
29 net_set_ip_header(pkt, dest, net_ip, IP_ICMP_HDR_SIZE, IPPROTO_ICMP);
30
31 icmp->type = ICMP_ECHO_REQUEST;
32 icmp->code = 0;
33 icmp->checksum = 0;
34 icmp->un.echo.id = 0;
35 icmp->un.echo.sequence = htons(ping_seq_number++);
36 icmp->checksum = compute_ip_checksum(icmp, ICMP_HDR_SIZE);
37}
38
39static int ping_send(void)
40{
41 uchar *pkt;
42 int eth_hdr_size;
43
44
45
46 debug_cond(DEBUG_DEV_PKT, "sending ARP for %pI4\n", &net_ping_ip);
47
48 net_arp_wait_packet_ip = net_ping_ip;
49
50 eth_hdr_size = net_set_ether(net_tx_packet, net_null_ethaddr, PROT_IP);
51 pkt = (uchar *)net_tx_packet + eth_hdr_size;
52
53 set_icmp_header(pkt, net_ping_ip);
54
55
56 arp_wait_tx_packet_size = eth_hdr_size + IP_ICMP_HDR_SIZE;
57
58
59 arp_wait_try = 1;
60 arp_wait_timer_start = get_timer(0);
61 arp_request();
62 return 1;
63}
64
65static void ping_timeout_handler(void)
66{
67 eth_halt();
68 net_set_state(NETLOOP_FAIL);
69}
70
71void ping_start(void)
72{
73 printf("Using %s device\n", eth_get_name());
74 net_set_timeout_handler(10000UL, ping_timeout_handler);
75
76 ping_send();
77}
78
79void ping_receive(struct ethernet_hdr *et, struct ip_udp_hdr *ip, int len)
80{
81 struct icmp_hdr *icmph = (struct icmp_hdr *)&ip->udp_src;
82 struct in_addr src_ip;
83 int eth_hdr_size;
84 uchar *tx_packet;
85
86 switch (icmph->type) {
87 case ICMP_ECHO_REPLY:
88 src_ip = net_read_ip((void *)&ip->ip_src);
89 if (src_ip.s_addr == net_ping_ip.s_addr)
90 net_set_state(NETLOOP_SUCCESS);
91 return;
92 case ICMP_ECHO_REQUEST:
93 if (net_ip.s_addr == 0)
94 return;
95
96 eth_hdr_size = net_update_ether(et, et->et_src, PROT_IP);
97
98 debug_cond(DEBUG_DEV_PKT,
99 "Got ICMP ECHO REQUEST, return %d bytes\n",
100 eth_hdr_size + len);
101
102 ip->ip_sum = 0;
103 ip->ip_off = 0;
104 net_copy_ip((void *)&ip->ip_dst, &ip->ip_src);
105 net_copy_ip((void *)&ip->ip_src, &net_ip);
106 ip->ip_sum = compute_ip_checksum(ip, IP_HDR_SIZE);
107
108 icmph->type = ICMP_ECHO_REPLY;
109 icmph->checksum = 0;
110 icmph->checksum = compute_ip_checksum(icmph, len - IP_HDR_SIZE);
111
112 tx_packet = net_get_async_tx_pkt_buf();
113 memcpy(tx_packet, et, eth_hdr_size + len);
114 net_send_packet(tx_packet, eth_hdr_size + len);
115 return;
116
117
118 }
119}
120