src/test/test-udev.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2003-2004 Greg Kroah-Hartman <greg@kroah.com>
   5   Copyright 2004-2012 Kay Sievers <kay@vrfy.org>
   6
   7   systemd is free software; you can redistribute it and/or modify it
   8   under the terms of the GNU Lesser General Public License as published by
   9   the Free Software Foundation; either version 2.1 of the License, or
  10   (at your option) any later version.
  11
  12   systemd is distributed in the hope that it will be useful, but
  13   WITHOUT ANY WARRANTY; without even the implied warranty of
  14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15   Lesser General Public License for more details.
  16
  17   You should have received a copy of the GNU Lesser General Public License
  18   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  19 ***/
  20
  21 #include <stdio.h>
  22 #include <stddef.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <fcntl.h>
  26 #include <ctype.h>
  27 #include <errno.h>
  28 #include <unistd.h>
  29 #include <syslog.h>
  30 #include <grp.h>
  31 #include <sched.h>
  32 #include <sys/mount.h>
  33 #include <sys/signalfd.h>
  34
  35 #include "missing.h"
  36 #include "udev.h"
  37 #include "udev-util.h"
  38
  39 void udev_main_log(struct udev *udev, int priority,
  40                    const char *file, int line, const char *fn,
  41                    const char *format, va_list args) {}
  42
  43 static int fake_filesystems(void) {
  44         static const struct fakefs {
  45                 const char *src;
  46                 const char *target;
  47                 const char *error;
  48         } fakefss[] = {
  49                 { "test/sys", "/sys",                   "failed to mount test /sys" },
  50                 { "test/dev", "/dev",                   "failed to mount test /dev" },
  51                 { "test/run", "/run",                   "failed to mount test /run" },
  52                 { "test/run", "/etc/udev/rules.d",      "failed to mount empty /etc/udev/rules.d" },
  53                 { "test/run", "/usr/lib/udev/rules.d",  "failed to mount empty /usr/lib/udev/rules.d" },
  54         };
  55         unsigned int i;
  56         int err;
  57
  58         err = unshare(CLONE_NEWNS);
  59         if (err < 0) {
  60                 err = -errno;
  61                 fprintf(stderr, "failed to call unshare(): %m\n");
  62                 goto out;
  63         }
  64
  65         if (mount(NULL, "/", NULL, MS_PRIVATE|MS_REC, NULL) < 0) {
  66                 err = -errno;
  67                 fprintf(stderr, "failed to mount / as private: %m\n");
  68                 goto out;
  69         }
  70
  71         for (i = 0; i < ELEMENTSOF(fakefss); i++) {
  72                 err = mount(fakefss[i].src, fakefss[i].target, NULL, MS_BIND, NULL);
  73                 if (err < 0) {
  74                         err = -errno;
  75                         fprintf(stderr, "%s %m", fakefss[i].error);
  76                         return err;
  77                 }
  78         }
  79 out:
  80         return err;
  81 }
  82
  83
  84 int main(int argc, char *argv[])
  85 {
  86         _cleanup_udev_unref_ struct udev *udev = NULL;
  87         _cleanup_udev_event_unref_ struct udev_event *event = NULL;
  88         _cleanup_udev_device_unref_ struct udev_device *dev = NULL;
  89         _cleanup_udev_rules_unref_ struct udev_rules *rules = NULL;
  90         char syspath[UTIL_PATH_SIZE];
  91         const char *devpath;
  92         const char *action;
  93         sigset_t mask, sigmask_orig;
  94         int err;
  95
  96         err = fake_filesystems();
  97         if (err < 0)
  98                 return EXIT_FAILURE;
  99
 100         udev = udev_new();
 101         if (udev == NULL)
 102                 return EXIT_FAILURE;
 103
 104         log_debug("version %s\n", VERSION);
 105         label_init("/dev");
 106
 107         sigprocmask(SIG_SETMASK, NULL, &sigmask_orig);
 108
 109         action = argv[1];
 110         if (action == NULL) {
 111                 log_error("action missing\n");
 112                 goto out;
 113         }
 114
 115         devpath = argv[2];
 116         if (devpath == NULL) {
 117                 log_error("devpath missing\n");
 118                 goto out;
 119         }
 120
 121         rules = udev_rules_new(udev, 1);
 122
 123         strscpyl(syspath, sizeof(syspath), "/sys", devpath, NULL);
 124         dev = udev_device_new_from_syspath(udev, syspath);
 125         if (dev == NULL) {
 126                 log_debug("unknown device '%s'\n", devpath);
 127                 goto out;
 128         }
 129
 130         udev_device_set_action(dev, action);
 131         event = udev_event_new(dev);
 132
 133         sigfillset(&mask);
 134         sigprocmask(SIG_SETMASK, &mask, &sigmask_orig);
 135         event->fd_signal = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC);
 136         if (event->fd_signal < 0) {
 137                 fprintf(stderr, "error creating signalfd\n");
 138                 goto out;
 139         }
 140
 141         /* do what devtmpfs usually provides us */
 142         if (udev_device_get_devnode(dev) != NULL) {
 143                 mode_t mode = 0600;
 144
 145                 if (streq(udev_device_get_subsystem(dev), "block"))
 146                         mode |= S_IFBLK;
 147                 else
 148                         mode |= S_IFCHR;
 149
 150                 if (!streq(action, "remove")) {
 151                         mkdir_parents_label(udev_device_get_devnode(dev), 0755);
 152                         mknod(udev_device_get_devnode(dev), mode, udev_device_get_devnum(dev));
 153                 } else {
 154                         unlink(udev_device_get_devnode(dev));
 155                         util_delete_path(udev, udev_device_get_devnode(dev));
 156                 }
 157         }
 158
 159         err = udev_event_execute_rules(event, rules, &sigmask_orig);
 160         if (err == 0)
 161                 udev_event_execute_run(event, NULL);
 162 out:
 163         if (event != NULL && event->fd_signal >= 0)
 164                 close(event->fd_signal);
 165         label_finish();
 166
 167         return err ? EXIT_FAILURE : EXIT_SUCCESS;
 168 }