103 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			103 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C
		
	
	
	
#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;
 | 
						|
}
 |