1// SPDX-License-Identifier: GPL-2.0+
  2#include <errno.h>
  3#include <fcntl.h>
  4#include <stdio.h>
  5#include <string.h>
  6#include <unistd.h>
  7#include <sys/socket.h>
  8#include <sys/stat.h>
  9#include <sys/sysmacros.h>
 10#include <sys/types.h>
 11
 12#include "../kselftest_harness.h"
 13
 14/* Remove a file, ignoring the result if it didn't exist. */
 15void rm(struct __test_metadata *_metadata, const char *pathname,
 16	int is_dir)
 17{
 18	int rc;
 19
 20	if (is_dir)
 21		rc = rmdir(pathname);
 22	else
 23		rc = unlink(pathname);
 24
 25	if (rc < 0) {
 26		ASSERT_EQ(errno, ENOENT) {
 27			TH_LOG("Not ENOENT: %s", pathname);
 28		}
 29	} else {
 30		ASSERT_EQ(rc, 0) {
 31			TH_LOG("Failed to remove: %s", pathname);
 32		}
 33	}
 34}
 35
 36FIXTURE(file) {
 37	char *pathname;
 38	int is_dir;
 39};
 40
 41FIXTURE_VARIANT(file)
 42{
 43	const char *name;
 44	int expected;
 45	int is_dir;
 46	void (*setup)(struct __test_metadata *_metadata,
 47		      FIXTURE_DATA(file) *self,
 48		      const FIXTURE_VARIANT(file) *variant);
 49	int major, minor, mode; /* for mknod() */
 50};
 51
 52void setup_link(struct __test_metadata *_metadata,
 53		FIXTURE_DATA(file) *self,
 54		const FIXTURE_VARIANT(file) *variant)
 55{
 56	const char * const paths[] = {
 57		"/bin/true",
 58		"/usr/bin/true",
 59	};
 60	int i;
 61
 62	for (i = 0; i < ARRAY_SIZE(paths); i++) {
 63		if (access(paths[i], X_OK) == 0) {
 64			ASSERT_EQ(symlink(paths[i], self->pathname), 0);
 65			return;
 66		}
 67	}
 68	ASSERT_EQ(1, 0) {
 69		TH_LOG("Could not find viable 'true' binary");
 70	}
 71}
 72
 73FIXTURE_VARIANT_ADD(file, S_IFLNK)
 74{
 75	.name = "S_IFLNK",
 76	.expected = ELOOP,
 77	.setup = setup_link,
 78};
 79
 80void setup_dir(struct __test_metadata *_metadata,
 81	       FIXTURE_DATA(file) *self,
 82	       const FIXTURE_VARIANT(file) *variant)
 83{
 84	ASSERT_EQ(mkdir(self->pathname, 0755), 0);
 85}
 86
 87FIXTURE_VARIANT_ADD(file, S_IFDIR)
 88{
 89	.name = "S_IFDIR",
 90	.is_dir = 1,
 91	.expected = EACCES,
 92	.setup = setup_dir,
 93};
 94
 95void setup_node(struct __test_metadata *_metadata,
 96		FIXTURE_DATA(file) *self,
 97		const FIXTURE_VARIANT(file) *variant)
 98{
 99	dev_t dev;
100	int rc;
101
102	dev = makedev(variant->major, variant->minor);
103	rc = mknod(self->pathname, 0755 | variant->mode, dev);
104	ASSERT_EQ(rc, 0) {
105		if (errno == EPERM)
106			SKIP(return, "Please run as root; cannot mknod(%s)",
107				variant->name);
108	}
109}
110
111FIXTURE_VARIANT_ADD(file, S_IFBLK)
112{
113	.name = "S_IFBLK",
114	.expected = EACCES,
115	.setup = setup_node,
116	/* /dev/loop0 */
117	.major = 7,
118	.minor = 0,
119	.mode = S_IFBLK,
120};
121
122FIXTURE_VARIANT_ADD(file, S_IFCHR)
123{
124	.name = "S_IFCHR",
125	.expected = EACCES,
126	.setup = setup_node,
127	/* /dev/zero */
128	.major = 1,
129	.minor = 5,
130	.mode = S_IFCHR,
131};
132
133void setup_fifo(struct __test_metadata *_metadata,
134		FIXTURE_DATA(file) *self,
135		const FIXTURE_VARIANT(file) *variant)
136{
137	ASSERT_EQ(mkfifo(self->pathname, 0755), 0);
138}
139
140FIXTURE_VARIANT_ADD(file, S_IFIFO)
141{
142	.name = "S_IFIFO",
143	.expected = EACCES,
144	.setup = setup_fifo,
145};
146
147FIXTURE_SETUP(file)
148{
149	ASSERT_GT(asprintf(&self->pathname, "%s.test", variant->name), 6);
150	self->is_dir = variant->is_dir;
151
152	rm(_metadata, self->pathname, variant->is_dir);
153	variant->setup(_metadata, self, variant);
154}
155
156FIXTURE_TEARDOWN(file)
157{
158	rm(_metadata, self->pathname, self->is_dir);
159}
160
161TEST_F(file, exec_errno)
162{
163	char * const argv[2] = { (char * const)self->pathname, NULL };
164
165	EXPECT_LT(execv(argv[0], argv), 0);
166	EXPECT_EQ(errno, variant->expected);
167}
168
169/* S_IFSOCK */
170FIXTURE(sock)
171{
172	int fd;
173};
174
175FIXTURE_SETUP(sock)
176{
177	self->fd = socket(AF_INET, SOCK_STREAM, 0);
178	ASSERT_GE(self->fd, 0);
179}
180
181FIXTURE_TEARDOWN(sock)
182{
183	if (self->fd >= 0)
184		ASSERT_EQ(close(self->fd), 0);
185}
186
187TEST_F(sock, exec_errno)
188{
189	char * const argv[2] = { " magic socket ", NULL };
190	char * const envp[1] = { NULL };
191
192	EXPECT_LT(fexecve(self->fd, argv, envp), 0);
193	EXPECT_EQ(errno, EACCES);
194}
195
196TEST_HARNESS_MAIN