brintos

brintos / linux-shallow public Read only

0
0
Text · 2.5 KiB · d799fd8 Raw
149 lines · c
1// SPDX-License-Identifier: GPL-2.02 3#define _GNU_SOURCE4#include <sched.h>5 6#include <stddef.h>7#include <stdio.h>8#include <unistd.h>9 10#include <sys/socket.h>11#include <sys/un.h>12 13#include "../../kselftest_harness.h"14 15FIXTURE(unix_connect)16{17	int server, client;18	int family;19};20 21FIXTURE_VARIANT(unix_connect)22{23	int type;24	char sun_path[8];25	int len;26	int flags;27	int err;28};29 30FIXTURE_VARIANT_ADD(unix_connect, stream_pathname)31{32	.type = SOCK_STREAM,33	.sun_path = "test",34	.len = 4 + 1,35	.flags = 0,36	.err = 0,37};38 39FIXTURE_VARIANT_ADD(unix_connect, stream_abstract)40{41	.type = SOCK_STREAM,42	.sun_path = "\0test",43	.len = 5,44	.flags = 0,45	.err = 0,46};47 48FIXTURE_VARIANT_ADD(unix_connect, stream_pathname_netns)49{50	.type = SOCK_STREAM,51	.sun_path = "test",52	.len = 4 + 1,53	.flags = CLONE_NEWNET,54	.err = 0,55};56 57FIXTURE_VARIANT_ADD(unix_connect, stream_abstract_netns)58{59	.type = SOCK_STREAM,60	.sun_path = "\0test",61	.len = 5,62	.flags = CLONE_NEWNET,63	.err = ECONNREFUSED,64};65 66FIXTURE_VARIANT_ADD(unix_connect, dgram_pathname)67{68	.type = SOCK_DGRAM,69	.sun_path = "test",70	.len = 4 + 1,71	.flags = 0,72	.err = 0,73};74 75FIXTURE_VARIANT_ADD(unix_connect, dgram_abstract)76{77	.type = SOCK_DGRAM,78	.sun_path = "\0test",79	.len = 5,80	.flags = 0,81	.err = 0,82};83 84FIXTURE_VARIANT_ADD(unix_connect, dgram_pathname_netns)85{86	.type = SOCK_DGRAM,87	.sun_path = "test",88	.len = 4 + 1,89	.flags = CLONE_NEWNET,90	.err = 0,91};92 93FIXTURE_VARIANT_ADD(unix_connect, dgram_abstract_netns)94{95	.type = SOCK_DGRAM,96	.sun_path = "\0test",97	.len = 5,98	.flags = CLONE_NEWNET,99	.err = ECONNREFUSED,100};101 102FIXTURE_SETUP(unix_connect)103{104	self->family = AF_UNIX;105}106 107FIXTURE_TEARDOWN(unix_connect)108{109	close(self->server);110	close(self->client);111 112	if (variant->sun_path[0])113		remove("test");114}115 116TEST_F(unix_connect, test)117{118	socklen_t addrlen;119	struct sockaddr_un addr = {120		.sun_family = self->family,121	};122	int err;123 124	self->server = socket(self->family, variant->type, 0);125	ASSERT_NE(-1, self->server);126 127	addrlen = offsetof(struct sockaddr_un, sun_path) + variant->len;128	memcpy(&addr.sun_path, variant->sun_path, variant->len);129 130	err = bind(self->server, (struct sockaddr *)&addr, addrlen);131	ASSERT_EQ(0, err);132 133	if (variant->type == SOCK_STREAM) {134		err = listen(self->server, 32);135		ASSERT_EQ(0, err);136	}137 138	err = unshare(variant->flags);139	ASSERT_EQ(0, err);140 141	self->client = socket(self->family, variant->type, 0);142	ASSERT_LT(0, self->client);143 144	err = connect(self->client, (struct sockaddr *)&addr, addrlen);145	ASSERT_EQ(variant->err, err == -1 ? errno : 0);146}147 148TEST_HARNESS_MAIN149