Quelle  non-regular.c   Sprache: C

// 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