src/test/test-udev.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2 /***
   3   Copyright © 2003-2004 Greg Kroah-Hartman <greg@kroah.com>
   4 ***/
   5
   6 #include <errno.h>
   7 #include <sched.h>
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include <sys/mount.h>
  11 #include <sys/signalfd.h>
  12 #include <unistd.h>
  13
  14 #include "build.h"
  15 #include "device-private.h"
  16 #include "fs-util.h"
  17 #include "log.h"
  18 #include "main-func.h"
  19 #include "mkdir.h"
  20 #include "selinux-util.h"
  21 #include "signal-util.h"
  22 #include "string-util.h"
  23 #include "tests.h"
  24 #include "udev-event.h"
  25
  26 static int fake_filesystems(void) {
  27         static const struct fakefs {
  28                 const char *src;
  29                 const char *target;
  30                 const char *error;
  31                 bool ignore_mount_error;
  32         } fakefss[] = {
  33                 { "test/tmpfs/sys", "/sys",                    "Failed to mount test /sys",                        false },
  34                 { "test/tmpfs/dev", "/dev",                    "Failed to mount test /dev",                        false },
  35                 { "test/run",       "/run",                    "Failed to mount test /run",                        false },
  36                 { "test/run",       "/etc/udev/rules.d",       "Failed to mount empty /etc/udev/rules.d",          true },
  37                 { "test/run",       UDEVLIBEXECDIR "/rules.d", "Failed to mount empty " UDEVLIBEXECDIR "/rules.d", true },
  38         };
  39         unsigned i;
  40
  41         if (unshare(CLONE_NEWNS) < 0)
  42                 return log_error_errno(errno, "Failed to call unshare(): %m");
  43
  44         if (mount(NULL, "/", NULL, MS_SLAVE|MS_REC, NULL) < 0)
  45                 return log_error_errno(errno, "Failed to mount / as private: %m");
  46
  47         for (i = 0; i < ELEMENTSOF(fakefss); i++)
  48                 if (mount(fakefss[i].src, fakefss[i].target, NULL, MS_BIND, NULL) < 0) {
  49                         log_full_errno(fakefss[i].ignore_mount_error ? LOG_DEBUG : LOG_ERR, errno, "%s: %m", fakefss[i].error);
  50                         if (!fakefss[i].ignore_mount_error)
  51                                 return -errno;
  52                 }
  53
  54         return 0;
  55 }
  56
  57 static int run(int argc, char *argv[]) {
  58         _cleanup_(udev_rules_freep) UdevRules *rules = NULL;
  59         _cleanup_(udev_event_freep) UdevEvent *event = NULL;
  60         _cleanup_(sd_device_unrefp) sd_device *dev = NULL;
  61         const char *devpath, *devname, *action;
  62         int r;
  63
  64         test_setup_logging(LOG_INFO);
  65
  66         if (!IN_SET(argc, 2, 3)) {
  67                 log_error("This program needs one or two arguments, %d given", argc - 1);
  68                 return -EINVAL;
  69         }
  70
  71         r = fake_filesystems();
  72         if (r < 0)
  73                 return r;
  74
  75         if (argc == 2) {
  76                 if (!streq(argv[1], "check")) {
  77                         log_error("Unknown argument: %s", argv[1]);
  78                         return -EINVAL;
  79                 }
  80
  81                 return 0;
  82         }
  83
  84         log_debug("version %s", GIT_VERSION);
  85         mac_selinux_init();
  86
  87         action = argv[1];
  88         devpath = argv[2];
  89
  90         assert_se(udev_rules_load(&rules, RESOLVE_NAME_EARLY) == 0);
  91
  92         const char *syspath = strjoina("/sys", devpath);
  93         r = device_new_from_synthetic_event(&dev, syspath, action);
  94         if (r < 0)
  95                 return log_debug_errno(r, "Failed to open device '%s'", devpath);
  96
  97         assert_se(event = udev_event_new(dev, 0, NULL));
  98
  99         assert_se(sigprocmask_many(SIG_BLOCK, NULL, SIGTERM, SIGINT, SIGHUP, SIGCHLD, -1) >= 0);
 100
 101         /* do what devtmpfs usually provides us */
 102         if (sd_device_get_devname(dev, &devname) >= 0) {
 103                 const char *subsystem;
 104                 mode_t mode = 0600;
 105
 106                 if (sd_device_get_subsystem(dev, &subsystem) >= 0 && streq(subsystem, "block"))
 107                         mode |= S_IFBLK;
 108                 else
 109                         mode |= S_IFCHR;
 110
 111                 if (!streq(action, "remove")) {
 112                         dev_t devnum = makedev(0, 0);
 113
 114                         (void) mkdir_parents_label(devname, 0755);
 115                         (void) sd_device_get_devnum(dev, &devnum);
 116                         if (mknod(devname, mode, devnum) < 0)
 117                                 return log_error_errno(errno, "mknod() failed for '%s': %m", devname);
 118                 } else {
 119                         if (unlink(devname) < 0)
 120                                 return log_error_errno(errno, "unlink('%s') failed: %m", devname);
 121                         (void) rmdir_parents(devname, "/");
 122                 }
 123         }
 124
 125         udev_event_execute_rules(event, 3 * USEC_PER_SEC, SIGKILL, NULL, rules);
 126         udev_event_execute_run(event, 3 * USEC_PER_SEC, SIGKILL);
 127
 128         return 0;
 129 }
 130
 131 DEFINE_MAIN_FUNCTION(run);