197 lines · c
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 else23 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_MAIN197