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