chiark / gitweb /
nspawn: fix use-after-free and leak in error paths
[elogind.git] / 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 <stdlib.h>
23 #include <errno.h>
24 #include <unistd.h>
25 #include <sched.h>
26 #include <sys/mount.h>
27 #include <sys/signalfd.h>
28
29 #include "missing.h"
30 #include "selinux-util.h"
31 #include "udev.h"
32 #include "udev-util.h"
33
34 static int fake_filesystems(void) {
35         static const struct fakefs {
36                 const char *src;
37                 const char *target;
38                 const char *error;
39         } fakefss[] = {
40                 { "test/sys", "/sys",                   "failed to mount test /sys" },
41                 { "test/dev", "/dev",                   "failed to mount test /dev" },
42                 { "test/run", "/run",                   "failed to mount test /run" },
43                 { "test/run", "/etc/udev/rules.d",      "failed to mount empty /etc/udev/rules.d" },
44                 { "test/run", "/usr/lib/udev/rules.d",  "failed to mount empty /usr/lib/udev/rules.d" },
45         };
46         unsigned int i;
47         int err;
48
49         err = unshare(CLONE_NEWNS);
50         if (err < 0) {
51                 err = -errno;
52                 fprintf(stderr, "failed to call unshare(): %m\n");
53                 goto out;
54         }
55
56         if (mount(NULL, "/", NULL, MS_PRIVATE|MS_REC, NULL) < 0) {
57                 err = -errno;
58                 fprintf(stderr, "failed to mount / as private: %m\n");
59                 goto out;
60         }
61
62         for (i = 0; i < ELEMENTSOF(fakefss); i++) {
63                 err = mount(fakefss[i].src, fakefss[i].target, NULL, MS_BIND, NULL);
64                 if (err < 0) {
65                         err = -errno;
66                         fprintf(stderr, "%s %m", fakefss[i].error);
67                         return err;
68                 }
69         }
70 out:
71         return err;
72 }
73
74 int main(int argc, char *argv[]) {
75         _cleanup_udev_unref_ struct udev *udev = NULL;
76         _cleanup_udev_event_unref_ struct udev_event *event = NULL;
77         _cleanup_udev_device_unref_ struct udev_device *dev = NULL;
78         _cleanup_udev_rules_unref_ struct udev_rules *rules = NULL;
79         char syspath[UTIL_PATH_SIZE];
80         const char *devpath;
81         const char *action;
82         sigset_t mask, sigmask_orig;
83         int err;
84
85         err = fake_filesystems();
86         if (err < 0)
87                 return EXIT_FAILURE;
88
89         udev = udev_new();
90         if (udev == NULL)
91                 return EXIT_FAILURE;
92
93         log_debug("version %s", VERSION);
94         mac_selinux_init("/dev");
95
96         sigprocmask(SIG_SETMASK, NULL, &sigmask_orig);
97
98         action = argv[1];
99         if (action == NULL) {
100                 log_error("action missing");
101                 goto out;
102         }
103
104         devpath = argv[2];
105         if (devpath == NULL) {
106                 log_error("devpath missing");
107                 goto out;
108         }
109
110         rules = udev_rules_new(udev, 1);
111
112         strscpyl(syspath, sizeof(syspath), "/sys", devpath, NULL);
113         dev = udev_device_new_from_syspath(udev, syspath);
114         if (dev == NULL) {
115                 log_debug("unknown device '%s'", devpath);
116                 goto out;
117         }
118
119         udev_device_set_action(dev, action);
120         event = udev_event_new(dev);
121
122         sigfillset(&mask);
123         sigprocmask(SIG_SETMASK, &mask, &sigmask_orig);
124         event->fd_signal = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC);
125         if (event->fd_signal < 0) {
126                 fprintf(stderr, "error creating signalfd\n");
127                 goto out;
128         }
129
130         /* do what devtmpfs usually provides us */
131         if (udev_device_get_devnode(dev) != NULL) {
132                 mode_t mode = 0600;
133
134                 if (streq(udev_device_get_subsystem(dev), "block"))
135                         mode |= S_IFBLK;
136                 else
137                         mode |= S_IFCHR;
138
139                 if (!streq(action, "remove")) {
140                         mkdir_parents_label(udev_device_get_devnode(dev), 0755);
141                         mknod(udev_device_get_devnode(dev), mode, udev_device_get_devnum(dev));
142                 } else {
143                         unlink(udev_device_get_devnode(dev));
144                         rmdir_parents(udev_device_get_devnode(dev), "/");
145                 }
146         }
147
148         udev_event_execute_rules(event,
149                                  3 * USEC_PER_SEC, USEC_PER_SEC,
150                                  NULL,
151                                  rules,
152                                  &sigmask_orig);
153         udev_event_execute_run(event,
154                                3 * USEC_PER_SEC, USEC_PER_SEC,
155                                NULL);
156 out:
157         if (event != NULL && event->fd_signal >= 0)
158                 close(event->fd_signal);
159         mac_selinux_finish();
160
161         return err ? EXIT_FAILURE : EXIT_SUCCESS;
162 }