forked from Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
196 lines
3.6 KiB
196 lines
3.6 KiB
// SPDX-License-Identifier: GPL-2.0+ |
|
#include <errno.h> |
|
#include <fcntl.h> |
|
#include <stdio.h> |
|
#include <string.h> |
|
#include <unistd.h> |
|
#include <sys/socket.h> |
|
#include <sys/stat.h> |
|
#include <sys/sysmacros.h> |
|
#include <sys/types.h> |
|
|
|
#include "../kselftest_harness.h" |
|
|
|
/* Remove a file, ignoring the result if it didn't exist. */ |
|
void rm(struct __test_metadata *_metadata, const char *pathname, |
|
int is_dir) |
|
{ |
|
int rc; |
|
|
|
if (is_dir) |
|
rc = rmdir(pathname); |
|
else |
|
rc = unlink(pathname); |
|
|
|
if (rc < 0) { |
|
ASSERT_EQ(errno, ENOENT) { |
|
TH_LOG("Not ENOENT: %s", pathname); |
|
} |
|
} else { |
|
ASSERT_EQ(rc, 0) { |
|
TH_LOG("Failed to remove: %s", pathname); |
|
} |
|
} |
|
} |
|
|
|
FIXTURE(file) { |
|
char *pathname; |
|
int is_dir; |
|
}; |
|
|
|
FIXTURE_VARIANT(file) |
|
{ |
|
const char *name; |
|
int expected; |
|
int is_dir; |
|
void (*setup)(struct __test_metadata *_metadata, |
|
FIXTURE_DATA(file) *self, |
|
const FIXTURE_VARIANT(file) *variant); |
|
int major, minor, mode; /* for mknod() */ |
|
}; |
|
|
|
void setup_link(struct __test_metadata *_metadata, |
|
FIXTURE_DATA(file) *self, |
|
const FIXTURE_VARIANT(file) *variant) |
|
{ |
|
const char * const paths[] = { |
|
"/bin/true", |
|
"/usr/bin/true", |
|
}; |
|
int i; |
|
|
|
for (i = 0; i < ARRAY_SIZE(paths); i++) { |
|
if (access(paths[i], X_OK) == 0) { |
|
ASSERT_EQ(symlink(paths[i], self->pathname), 0); |
|
return; |
|
} |
|
} |
|
ASSERT_EQ(1, 0) { |
|
TH_LOG("Could not find viable 'true' binary"); |
|
} |
|
} |
|
|
|
FIXTURE_VARIANT_ADD(file, S_IFLNK) |
|
{ |
|
.name = "S_IFLNK", |
|
.expected = ELOOP, |
|
.setup = setup_link, |
|
}; |
|
|
|
void setup_dir(struct __test_metadata *_metadata, |
|
FIXTURE_DATA(file) *self, |
|
const FIXTURE_VARIANT(file) *variant) |
|
{ |
|
ASSERT_EQ(mkdir(self->pathname, 0755), 0); |
|
} |
|
|
|
FIXTURE_VARIANT_ADD(file, S_IFDIR) |
|
{ |
|
.name = "S_IFDIR", |
|
.is_dir = 1, |
|
.expected = EACCES, |
|
.setup = setup_dir, |
|
}; |
|
|
|
void setup_node(struct __test_metadata *_metadata, |
|
FIXTURE_DATA(file) *self, |
|
const FIXTURE_VARIANT(file) *variant) |
|
{ |
|
dev_t dev; |
|
int rc; |
|
|
|
dev = makedev(variant->major, variant->minor); |
|
rc = mknod(self->pathname, 0755 | variant->mode, dev); |
|
ASSERT_EQ(rc, 0) { |
|
if (errno == EPERM) |
|
SKIP(return, "Please run as root; cannot mknod(%s)", |
|
variant->name); |
|
} |
|
} |
|
|
|
FIXTURE_VARIANT_ADD(file, S_IFBLK) |
|
{ |
|
.name = "S_IFBLK", |
|
.expected = EACCES, |
|
.setup = setup_node, |
|
/* /dev/loop0 */ |
|
.major = 7, |
|
.minor = 0, |
|
.mode = S_IFBLK, |
|
}; |
|
|
|
FIXTURE_VARIANT_ADD(file, S_IFCHR) |
|
{ |
|
.name = "S_IFCHR", |
|
.expected = EACCES, |
|
.setup = setup_node, |
|
/* /dev/zero */ |
|
.major = 1, |
|
.minor = 5, |
|
.mode = S_IFCHR, |
|
}; |
|
|
|
void setup_fifo(struct __test_metadata *_metadata, |
|
FIXTURE_DATA(file) *self, |
|
const FIXTURE_VARIANT(file) *variant) |
|
{ |
|
ASSERT_EQ(mkfifo(self->pathname, 0755), 0); |
|
} |
|
|
|
FIXTURE_VARIANT_ADD(file, S_IFIFO) |
|
{ |
|
.name = "S_IFIFO", |
|
.expected = EACCES, |
|
.setup = setup_fifo, |
|
}; |
|
|
|
FIXTURE_SETUP(file) |
|
{ |
|
ASSERT_GT(asprintf(&self->pathname, "%s.test", variant->name), 6); |
|
self->is_dir = variant->is_dir; |
|
|
|
rm(_metadata, self->pathname, variant->is_dir); |
|
variant->setup(_metadata, self, variant); |
|
} |
|
|
|
FIXTURE_TEARDOWN(file) |
|
{ |
|
rm(_metadata, self->pathname, self->is_dir); |
|
} |
|
|
|
TEST_F(file, exec_errno) |
|
{ |
|
char * const argv[2] = { (char * const)self->pathname, NULL }; |
|
|
|
EXPECT_LT(execv(argv[0], argv), 0); |
|
EXPECT_EQ(errno, variant->expected); |
|
} |
|
|
|
/* S_IFSOCK */ |
|
FIXTURE(sock) |
|
{ |
|
int fd; |
|
}; |
|
|
|
FIXTURE_SETUP(sock) |
|
{ |
|
self->fd = socket(AF_INET, SOCK_STREAM, 0); |
|
ASSERT_GE(self->fd, 0); |
|
} |
|
|
|
FIXTURE_TEARDOWN(sock) |
|
{ |
|
if (self->fd >= 0) |
|
ASSERT_EQ(close(self->fd), 0); |
|
} |
|
|
|
TEST_F(sock, exec_errno) |
|
{ |
|
char * const argv[2] = { " magic socket ", NULL }; |
|
char * const envp[1] = { NULL }; |
|
|
|
EXPECT_LT(fexecve(self->fd, argv, envp), 0); |
|
EXPECT_EQ(errno, EACCES); |
|
} |
|
|
|
TEST_HARNESS_MAIN
|
|
|