1 // SPDX-License-Identifier: GPL-2.0
2 
3 #define _GNU_SOURCE
4 #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 
FIXTURE(unix_connect)15 FIXTURE(unix_connect)
16 {
17 	int server, client;
18 	int family;
19 };
20 
FIXTURE_VARIANT(unix_connect)21 FIXTURE_VARIANT(unix_connect)
22 {
23 	int type;
24 	char sun_path[8];
25 	int len;
26 	int flags;
27 	int err;
28 };
29 
FIXTURE_VARIANT_ADD(unix_connect,stream_pathname)30 FIXTURE_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 
FIXTURE_VARIANT_ADD(unix_connect,stream_abstract)39 FIXTURE_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 
FIXTURE_VARIANT_ADD(unix_connect,stream_pathname_netns)48 FIXTURE_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 
FIXTURE_VARIANT_ADD(unix_connect,stream_abstract_netns)57 FIXTURE_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 
FIXTURE_VARIANT_ADD(unix_connect,dgram_pathname)66 FIXTURE_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 
FIXTURE_VARIANT_ADD(unix_connect,dgram_abstract)75 FIXTURE_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 
FIXTURE_VARIANT_ADD(unix_connect,dgram_pathname_netns)84 FIXTURE_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 
FIXTURE_VARIANT_ADD(unix_connect,dgram_abstract_netns)93 FIXTURE_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 
FIXTURE_SETUP(unix_connect)102 FIXTURE_SETUP(unix_connect)
103 {
104 	self->family = AF_UNIX;
105 }
106 
FIXTURE_TEARDOWN(unix_connect)107 FIXTURE_TEARDOWN(unix_connect)
108 {
109 	close(self->server);
110 	close(self->client);
111 
112 	if (variant->sun_path[0])
113 		remove("test");
114 }
115 
TEST_F(unix_connect,test)116 TEST_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 
148 TEST_HARNESS_MAIN
149