File: socket_icmp.c

package info (click to toggle)
criu 4.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 11,584 kB
  • sloc: ansic: 139,280; python: 7,484; sh: 3,824; java: 2,799; makefile: 2,659; asm: 1,137; perl: 206; xml: 117; exp: 45
file content (128 lines) | stat: -rw-r--r-- 2,796 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#include "zdtmtst.h"

const char *test_doc = "static test for ICMP socket\n";
const char *test_author = "समीर सिंह Sameer Singh <lumarzeli30@gmail.com>\n";

/* Description:
 * Send a ping to localhost using ICMP socket
 */

#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#if defined(ZDTM_IPV6)
#include <netinet/icmp6.h>
#else
#include <netinet/ip_icmp.h>
#endif
#include <arpa/inet.h>
#include <sys/time.h>
#include <netdb.h>

#include "sysctl.h"

#define PACKET_SIZE  64
#define RECV_TIMEOUT 1

static int echo_id = 1234;

#if defined(ZDTM_IPV6)
#define TEST_ICMP_ECHOREPLY ICMP6_ECHOREPLY
#else
#define TEST_ICMP_ECHOREPLY ICMP_ECHOREPLY
#endif
int main(int argc, char **argv)
{
	int ret, sock, seq = 0;
	char packet[PACKET_SIZE], recv_packet[PACKET_SIZE];

	struct timeval tv;
#if defined(ZDTM_IPV6)
	struct sockaddr_in6 addr, recv_addr;
#else
	struct icmphdr icmp_header, *icmp_reply;
#endif
	struct sockaddr_in addr, recv_addr;
	socklen_t addr_len;

	// Allow GIDs 0-58468 to open an unprivileged ICMP socket
	if (sysctl_write_str("/proc/sys/net/ipv4/ping_group_range", "0 58468"))
		return -1;

	test_init(argc, argv);

#if defined(ZDTM_IPV6)
	sock = socket(PF_INET6, SOCK_DGRAM, IPPROTO_ICMPV6);
#else
	sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_ICMP);
#endif
	if (sock < 0) {
		pr_perror("Can't create socket");
		return 1;
	}

	tv.tv_sec = RECV_TIMEOUT;
	tv.tv_usec = 0;
	if (setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)) < 0) {
		pr_perror("Can't set socket option");
		return 1;
	}

	memset(&addr, 0, sizeof(addr));
	memset(&icmp_header, 0, sizeof(icmp_header));
#if defined(ZDTM_IPV6)
	addr.sin6_family = AF_INET6;
	inet_pton(AF_INET6, "::1", &addr.sin6_addr);

	icmp_header.icmp6_type = ICMP6_ECHO_REQUEST;
	icmp_header.icmp6_code = 0;
	icmp_header.icmp6_id = echo_id;
	icmp_header.icmp6_seq = seq;
#else
	addr.sin_family = AF_INET;
	addr.sin_addr.s_addr = inet_addr("127.0.0.1");

	icmp_header.type = ICMP_ECHO;
	icmp_header.code = 0;
	icmp_header.un.echo.id = echo_id;
	icmp_header.un.echo.sequence = seq;
#endif

	memcpy(packet, &icmp_header, sizeof(icmp_header));
	memset(packet + sizeof(icmp_header), 0xa5,
	       PACKET_SIZE - sizeof(icmp_header));

	test_daemon();
	test_waitsig();

	ret = sendto(sock, packet, PACKET_SIZE, 0,
		     (struct sockaddr *)&addr, sizeof(addr));

	if (ret < 0) {
		fail("Can't send");
		return 1;
	}

	addr_len = sizeof(recv_addr);

	ret = recvfrom(sock, recv_packet, sizeof(recv_packet), 0,
		       (struct sockaddr *)&recv_addr, &addr_len);

	if (ret < 0) {
		fail("Can't recv");
		return 1;
	}

	icmp_reply = (struct icmphdr *)recv_packet;

	if (icmp_reply->type != ICMP_ECHOREPLY) {
		fail("Got no ICMP_ECHO_REPLY");
		return 1;
	}

	close(sock);

	pass();
	return 0;
}