149 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			149 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| // SPDX-License-Identifier: GPL-2.0
 | |
| 
 | |
| #define _GNU_SOURCE
 | |
| #include <sched.h>
 | |
| 
 | |
| #include <stddef.h>
 | |
| #include <stdio.h>
 | |
| #include <unistd.h>
 | |
| 
 | |
| #include <sys/socket.h>
 | |
| #include <sys/un.h>
 | |
| 
 | |
| #include "../../kselftest_harness.h"
 | |
| 
 | |
| FIXTURE(unix_connect)
 | |
| {
 | |
| 	int server, client;
 | |
| 	int family;
 | |
| };
 | |
| 
 | |
| FIXTURE_VARIANT(unix_connect)
 | |
| {
 | |
| 	int type;
 | |
| 	char sun_path[8];
 | |
| 	int len;
 | |
| 	int flags;
 | |
| 	int err;
 | |
| };
 | |
| 
 | |
| FIXTURE_VARIANT_ADD(unix_connect, stream_pathname)
 | |
| {
 | |
| 	.type = SOCK_STREAM,
 | |
| 	.sun_path = "test",
 | |
| 	.len = 4 + 1,
 | |
| 	.flags = 0,
 | |
| 	.err = 0,
 | |
| };
 | |
| 
 | |
| FIXTURE_VARIANT_ADD(unix_connect, stream_abstract)
 | |
| {
 | |
| 	.type = SOCK_STREAM,
 | |
| 	.sun_path = "\0test",
 | |
| 	.len = 5,
 | |
| 	.flags = 0,
 | |
| 	.err = 0,
 | |
| };
 | |
| 
 | |
| FIXTURE_VARIANT_ADD(unix_connect, stream_pathname_netns)
 | |
| {
 | |
| 	.type = SOCK_STREAM,
 | |
| 	.sun_path = "test",
 | |
| 	.len = 4 + 1,
 | |
| 	.flags = CLONE_NEWNET,
 | |
| 	.err = 0,
 | |
| };
 | |
| 
 | |
| FIXTURE_VARIANT_ADD(unix_connect, stream_abstract_netns)
 | |
| {
 | |
| 	.type = SOCK_STREAM,
 | |
| 	.sun_path = "\0test",
 | |
| 	.len = 5,
 | |
| 	.flags = CLONE_NEWNET,
 | |
| 	.err = ECONNREFUSED,
 | |
| };
 | |
| 
 | |
| FIXTURE_VARIANT_ADD(unix_connect, dgram_pathname)
 | |
| {
 | |
| 	.type = SOCK_DGRAM,
 | |
| 	.sun_path = "test",
 | |
| 	.len = 4 + 1,
 | |
| 	.flags = 0,
 | |
| 	.err = 0,
 | |
| };
 | |
| 
 | |
| FIXTURE_VARIANT_ADD(unix_connect, dgram_abstract)
 | |
| {
 | |
| 	.type = SOCK_DGRAM,
 | |
| 	.sun_path = "\0test",
 | |
| 	.len = 5,
 | |
| 	.flags = 0,
 | |
| 	.err = 0,
 | |
| };
 | |
| 
 | |
| FIXTURE_VARIANT_ADD(unix_connect, dgram_pathname_netns)
 | |
| {
 | |
| 	.type = SOCK_DGRAM,
 | |
| 	.sun_path = "test",
 | |
| 	.len = 4 + 1,
 | |
| 	.flags = CLONE_NEWNET,
 | |
| 	.err = 0,
 | |
| };
 | |
| 
 | |
| FIXTURE_VARIANT_ADD(unix_connect, dgram_abstract_netns)
 | |
| {
 | |
| 	.type = SOCK_DGRAM,
 | |
| 	.sun_path = "\0test",
 | |
| 	.len = 5,
 | |
| 	.flags = CLONE_NEWNET,
 | |
| 	.err = ECONNREFUSED,
 | |
| };
 | |
| 
 | |
| FIXTURE_SETUP(unix_connect)
 | |
| {
 | |
| 	self->family = AF_UNIX;
 | |
| }
 | |
| 
 | |
| FIXTURE_TEARDOWN(unix_connect)
 | |
| {
 | |
| 	close(self->server);
 | |
| 	close(self->client);
 | |
| 
 | |
| 	if (variant->sun_path[0])
 | |
| 		remove("test");
 | |
| }
 | |
| 
 | |
| TEST_F(unix_connect, test)
 | |
| {
 | |
| 	socklen_t addrlen;
 | |
| 	struct sockaddr_un addr = {
 | |
| 		.sun_family = self->family,
 | |
| 	};
 | |
| 	int err;
 | |
| 
 | |
| 	self->server = socket(self->family, variant->type, 0);
 | |
| 	ASSERT_NE(-1, self->server);
 | |
| 
 | |
| 	addrlen = offsetof(struct sockaddr_un, sun_path) + variant->len;
 | |
| 	memcpy(&addr.sun_path, variant->sun_path, variant->len);
 | |
| 
 | |
| 	err = bind(self->server, (struct sockaddr *)&addr, addrlen);
 | |
| 	ASSERT_EQ(0, err);
 | |
| 
 | |
| 	if (variant->type == SOCK_STREAM) {
 | |
| 		err = listen(self->server, 32);
 | |
| 		ASSERT_EQ(0, err);
 | |
| 	}
 | |
| 
 | |
| 	err = unshare(variant->flags);
 | |
| 	ASSERT_EQ(0, err);
 | |
| 
 | |
| 	self->client = socket(self->family, variant->type, 0);
 | |
| 	ASSERT_LT(0, self->client);
 | |
| 
 | |
| 	err = connect(self->client, (struct sockaddr *)&addr, addrlen);
 | |
| 	ASSERT_EQ(variant->err, err == -1 ? errno : 0);
 | |
| }
 | |
| 
 | |
| TEST_HARNESS_MAIN
 |