summaryrefslogtreecommitdiff
path: root/tests/net-packet_mreq.c
blob: 16c42e39b346339a80cd73db0d953f98de5260b6 (plain)
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
/*
 * Copyright (c) 2018 The strace developers.
 * All rights reserved.
 *
 * SPDX-License-Identifier: GPL-2.0-or-later
 */

#include "tests.h"
#include <stdio.h>
#include <sys/socket.h>
#include <linux/if_packet.h>

static const char *errstr;

static long
packet_mreq_membership(int optname, void *optval, socklen_t len)
{
	long rc = setsockopt(-1, SOL_PACKET, optname, optval, len);
	errstr = sprintrc(rc);
	return rc;
}

static void
test_packet_mreq(const int optname, const char *const optname_str)
{
	TAIL_ALLOC_OBJECT_CONST_PTR(struct packet_mreq, pmreq);
	socklen_t len = sizeof(struct packet_mreq);

	/* setsockopt with optname unknown */
	packet_mreq_membership(-1, NULL, 0);
	printf("setsockopt(-1, SOL_PACKET, %#x /* PACKET_??? */, NULL, 0) = %s\n",
	       -1, errstr);

	/* setsockopt with mr_type unknown */
	pmreq->mr_ifindex = 0;
	pmreq->mr_alen = ARRAY_SIZE(pmreq->mr_address);
	packet_mreq_membership(optname, pmreq, len);
	printf("setsockopt(-1, SOL_PACKET, %s, {mr_ifindex=%d,"
	       " mr_type=%#x /* PACKET_MR_??? */, mr_alen=%d, mr_address=",
	       optname_str, pmreq->mr_ifindex, pmreq->mr_type, pmreq->mr_alen);
	for (unsigned int i = 0; i < ARRAY_SIZE(pmreq->mr_address); i++) {
		printf("%02x", pmreq->mr_address[i]);
	}
	printf("}, %d) = %s\n", len, errstr);

	/* setsockopt with mr_type unknown and mr_alen > sizeof(mr_address) */
	pmreq->mr_alen = ARRAY_SIZE(pmreq->mr_address) + 1;
	packet_mreq_membership(optname, pmreq, len);
	printf("setsockopt(-1, SOL_PACKET, %s, {mr_ifindex=%d,"
	       " mr_type=%#x /* PACKET_MR_??? */, mr_alen=%d, mr_address=",
	       optname_str, pmreq->mr_ifindex, pmreq->mr_type, pmreq->mr_alen);
	for (unsigned int i = 0; i < ARRAY_SIZE(pmreq->mr_address); i++) {
		printf("%02x", pmreq->mr_address[i]);
	}
	printf("}, %d) = %s\n", len, errstr);

	/* setsockopt with mr_type unknown and mr_alen < sizeof(mr_address) */
	pmreq->mr_alen = ARRAY_SIZE(pmreq->mr_address) - 1;
	packet_mreq_membership(optname, pmreq, len);
	printf("setsockopt(-1, SOL_PACKET, %s, {mr_ifindex=%d,"
	       " mr_type=%#x /* PACKET_MR_??? */, mr_alen=%d, mr_address=",
	       optname_str, pmreq->mr_ifindex, pmreq->mr_type, pmreq->mr_alen);
	for (unsigned int i = 0; i < pmreq->mr_alen; i++) {
		printf("%02x", pmreq->mr_address[i]);
	}
	printf("}, %d) = %s\n", len, errstr);

	/* setsockopt with valid mr_type */
	pmreq->mr_alen = ARRAY_SIZE(pmreq->mr_address);
	static const struct {
		unsigned short type;
		const char *const type_str;
	} a[] = {
		{ ARG_STR(PACKET_MR_MULTICAST) },
		{ ARG_STR(PACKET_MR_PROMISC) },
		{ ARG_STR(PACKET_MR_ALLMULTI) },
#ifdef PACKET_MR_UNICAST
		{ ARG_STR(PACKET_MR_UNICAST) },
#endif
	};

	for (unsigned int i = 0; i < ARRAY_SIZE(a); i++) {
		pmreq->mr_type = a[i].type;
		packet_mreq_membership(optname, pmreq, len);
		printf("setsockopt(-1, SOL_PACKET, %s, {mr_ifindex=%d,"
		       " mr_type=%s, mr_alen=%d, mr_address=",
		       optname_str, pmreq->mr_ifindex, a[i].type_str, pmreq->mr_alen);
		for (unsigned int i = 0; i < pmreq->mr_alen; i++) {
			printf("%02x", pmreq->mr_address[i]);
		}
		printf("}, %d) = %s\n", len, errstr);
	}

	/* setsockopt with optlen larger than usual */
	len = len + 1;
	packet_mreq_membership(optname, pmreq, len);
	printf("setsockopt(-1, SOL_PACKET, %s, %p,"
	       " %d) = %s\n", optname_str, pmreq, len, errstr);
}

int
main(void)
{
	test_packet_mreq(ARG_STR(PACKET_ADD_MEMBERSHIP));
	test_packet_mreq(ARG_STR(PACKET_DROP_MEMBERSHIP));

	puts("+++ exited with 0 +++");
	return 0;
}