chiark / gitweb /
atomically replace existing nodes and symlinks
[elogind.git] / udev_node.c
1 /*
2  * Copyright (C) 2003 Greg Kroah-Hartman <greg@kroah.com>
3  * Copyright (C) 2004-2006 Kay Sievers <kay.sievers@vrfy.org>
4  *
5  *      This program is free software; you can redistribute it and/or modify it
6  *      under the terms of the GNU General Public License as published by the
7  *      Free Software Foundation version 2 of the License.
8  * 
9  *      This program is distributed in the hope that it will be useful, but
10  *      WITHOUT ANY WARRANTY; without even the implied warranty of
11  *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  *      General Public License for more details.
13  * 
14  *      You should have received a copy of the GNU General Public License along
15  *      with this program; if not, write to the Free Software Foundation, Inc.,
16  *      51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
17  *
18  */
19
20 #include <stdlib.h>
21 #include <string.h>
22 #include <stdio.h>
23 #include <stddef.h>
24 #include <fcntl.h>
25 #include <unistd.h>
26 #include <errno.h>
27 #include <grp.h>
28 #include <dirent.h>
29 #include <sys/stat.h>
30 #include <sys/types.h>
31
32 #include "udev.h"
33 #include "udev_rules.h"
34 #include "udev_selinux.h"
35
36 #define TMP_FILE_EXT            ".udev-tmp"
37
38 int udev_node_mknod(struct udevice *udev, const char *file, dev_t devt, mode_t mode, uid_t uid, gid_t gid)
39 {
40         char file_tmp[PATH_SIZE + sizeof(TMP_FILE_EXT)];
41         struct stat stats;
42         int retval = 0;
43
44         if (major(devt) != 0 && strcmp(udev->dev->subsystem, "block") == 0)
45                 mode |= S_IFBLK;
46         else
47                 mode |= S_IFCHR;
48
49         /* preserve node with already correct numbers, to prevent changing the inode number */
50         if (lstat(file, &stats) == 0) {
51                 if ((stats.st_mode & S_IFMT) == (mode & S_IFMT) && (stats.st_rdev == devt)) {
52                         info("preserve file '%s', because it has correct dev_t", file);
53                         selinux_setfilecon(file, udev->dev->kernel, stats.st_mode);
54                         goto perms;
55                 }
56         }
57
58         selinux_setfscreatecon(file, udev->dev->kernel, mode);
59         retval = mknod(file, mode, devt);
60         selinux_resetfscreatecon();
61         if (retval == 0)
62                 goto perms;
63
64         info("atomically replace '%s'", file);
65         strlcpy(file_tmp, file, sizeof(file_tmp));
66         strlcat(file_tmp, TMP_FILE_EXT, sizeof(file_tmp));
67         selinux_setfscreatecon(file_tmp, udev->dev->kernel, mode);
68         retval = mknod(file_tmp, mode, devt);
69         selinux_resetfscreatecon();
70         if (retval != 0) {
71                 err("mknod(%s, %#o, %u, %u) failed: %s",
72                     file_tmp, mode, major(devt), minor(devt), strerror(errno));
73                 goto exit;
74         }
75         retval = rename(file_tmp, file);
76         if (retval != 0) {
77                 err("rename(%s, %s) failed: %s",
78                     file_tmp, file, strerror(errno));
79                 unlink(file_tmp);
80                 goto exit;
81         }
82
83 perms:
84         dbg("chmod(%s, %#o)", file, mode);
85         if (chmod(file, mode) != 0) {
86                 err("chmod(%s, %#o) failed: %s", file, mode, strerror(errno));
87                 goto exit;
88         }
89
90         if (uid != 0 || gid != 0) {
91                 dbg("chown(%s, %u, %u)", file, uid, gid);
92                 if (chown(file, uid, gid) != 0) {
93                         err("chown(%s, %u, %u) failed: %s",
94                             file, uid, gid, strerror(errno));
95                         goto exit;
96                 }
97         }
98 exit:
99         return retval;
100 }
101
102 static int node_symlink(const char *node, const char *slink)
103 {
104         struct stat stats;
105         char target[PATH_SIZE] = "";
106         char slink_tmp[PATH_SIZE + sizeof(TMP_FILE_EXT)];
107         int i = 0;
108         int tail = 0;
109         int len;
110         int retval = 0;
111
112         /* use relative link */
113         while (node[i] && (node[i] == slink[i])) {
114                 if (node[i] == '/')
115                         tail = i+1;
116                 i++;
117         }
118         while (slink[i] != '\0') {
119                 if (slink[i] == '/')
120                         strlcat(target, "../", sizeof(target));
121                 i++;
122         }
123         strlcat(target, &node[tail], sizeof(target));
124
125         /* preserve link with correct target, do not replace node of other device */
126         if (lstat(slink, &stats) == 0) {
127                 if (S_ISBLK(stats.st_mode) || S_ISCHR(stats.st_mode)) {
128                         struct stat stats2;
129
130                         info("found existing node instead of symlink '%s'", slink);
131                         if (lstat(node, &stats2) == 0) {
132                                 if ((stats.st_mode & S_IFMT) == (stats2.st_mode & S_IFMT) &&
133                                     stats.st_rdev == stats2.st_rdev) {
134                                         info("replace device node '%s' with symlink to our node '%s'", slink, node);
135                                 } else {
136                                         err("device node '%s' already exists, link '%s' will not overwrite it", node, slink);
137                                         goto exit;
138                                 }
139                         }
140                 } else if (S_ISLNK(stats.st_mode)) {
141                         char buf[PATH_SIZE];
142
143                         info("found existing symlink '%s'", slink);
144                         len = readlink(slink, buf, sizeof(buf));
145                         if (len > 0) {
146                                 buf[len] = '\0';
147                                 if (strcmp(target, buf) == 0) {
148                                         info("preserve already existing symlink '%s' to '%s'", slink, target);
149                                         selinux_setfilecon(slink, NULL, S_IFLNK);
150                                         goto exit;
151                                 }
152                         }
153                 }
154         }
155
156         info("creating symlink '%s' to '%s'", slink, target);
157         selinux_setfscreatecon(slink, NULL, S_IFLNK);
158         retval = symlink(target, slink);
159         selinux_resetfscreatecon();
160         if (retval == 0)
161                 goto exit;
162
163         info("atomically replace '%s'", slink);
164         strlcpy(slink_tmp, slink, sizeof(slink_tmp));
165         strlcat(slink_tmp, TMP_FILE_EXT, sizeof(slink_tmp));
166         selinux_setfscreatecon(slink_tmp, NULL, S_IFLNK);
167         retval = symlink(target, slink_tmp);
168         selinux_resetfscreatecon();
169         if (retval != 0) {
170                 err("symlink(%s, %s) failed: %s", target, slink_tmp, strerror(errno));
171                 goto exit;
172         }
173         retval = rename(slink_tmp, slink);
174         if (retval != 0) {
175                 err("rename(%s, %s) failed: %s", slink_tmp, slink, strerror(errno));
176                 unlink(slink_tmp);
177                 goto exit;
178         }
179 exit:
180         return retval;
181 }
182
183 static int update_link(struct udevice *udev, const char *name)
184 {
185         LIST_HEAD(name_list);
186         char slink[PATH_SIZE];
187         char node[PATH_SIZE];
188         struct udevice *udev_db;
189         struct name_entry *device;
190         char target[PATH_MAX] = "";
191         int count;
192         int priority = 0;
193         int rc = 0;
194
195         strlcpy(slink, udev_root, sizeof(slink));
196         strlcat(slink, "/", sizeof(slink));
197         strlcat(slink, name, sizeof(slink));
198
199         count = udev_db_get_devices_by_name(name, &name_list);
200         info("found %i devices with name '%s'", count, name);
201
202         /* if we don't have a reference, delete it */
203         if (count <= 0) {
204                 info("no reference left, remove '%s'", name);
205                 if (!udev->test_run) {
206                         unlink(slink);
207                         delete_path(slink);
208                 }
209                 goto out;
210         }
211
212         /* find the device with the highest priority */
213         list_for_each_entry(device, &name_list, node) {
214                 info("found '%s' for '%s'", device->name, name);
215
216                 /* did we find ourself? we win, if we have the same priority */
217                 if (strcmp(udev->dev->devpath, device->name) == 0) {
218                         info("compare (our own) priority of '%s' %i >= %i",
219                              udev->dev->devpath, udev->link_priority, priority);
220                         if (target[0] == '\0' || udev->link_priority >= priority) {
221                                 priority = udev->link_priority;
222                                 strlcpy(target, udev->name, sizeof(target));
223                         }
224                         continue;
225                 }
226
227                 /* or something else, then read priority from database */
228                 udev_db = udev_device_init(NULL);
229                 if (udev_db == NULL)
230                         continue;
231                 if (udev_db_get_device(udev_db, device->name) == 0) {
232                         info("compare priority of '%s' %i > %i",
233                              udev_db->dev->devpath, udev_db->link_priority, priority);
234                         if (target[0] == '\0' || udev_db->link_priority > priority) {
235                                 priority = udev_db->link_priority;
236                                 strlcpy(target, udev_db->name, sizeof(target));
237                         }
238                 }
239                 udev_device_cleanup(udev_db);
240         }
241         name_list_cleanup(&name_list);
242
243         if (target[0] == '\0') {
244                 err("missing target for '%s'", name);
245                 rc = -1;
246                 goto out;
247         }
248
249         /* create symlink to the target with the highest priority */
250         strlcpy(node, udev_root, sizeof(node));
251         strlcat(node, "/", sizeof(node));
252         strlcat(node, target, sizeof(node));
253         info("'%s' with target '%s' has the highest priority %i, create it", name, target, priority);
254         if (!udev->test_run) {
255                 create_path(slink);
256                 node_symlink(node, slink);
257         }
258 out:
259         return rc;
260 }
261
262 void udev_node_update_symlinks(struct udevice *udev, struct udevice *udev_old)
263 {
264         struct name_entry *name_loop;
265         char symlinks[PATH_SIZE] = "";
266
267         list_for_each_entry(name_loop, &udev->symlink_list, node) {
268                 info("update symlink '%s' of '%s'", name_loop->name, udev->dev->devpath);
269                 update_link(udev, name_loop->name);
270                 strlcat(symlinks, udev_root, sizeof(symlinks));
271                 strlcat(symlinks, "/", sizeof(symlinks));
272                 strlcat(symlinks, name_loop->name, sizeof(symlinks));
273                 strlcat(symlinks, " ", sizeof(symlinks));
274         }
275
276         /* export symlinks to environment */
277         remove_trailing_chars(symlinks, ' ');
278         if (symlinks[0] != '\0')
279                 setenv("DEVLINKS", symlinks, 1);
280
281         /* update possible left-over symlinks (device metadata changed) */
282         if (udev_old != NULL) {
283                 struct name_entry *link_loop;
284                 struct name_entry *link_old_loop;
285                 int found;
286
287                 /* remove current symlinks from old list */
288                 list_for_each_entry(link_old_loop, &udev_old->symlink_list, node) {
289                         found = 0;
290                         list_for_each_entry(link_loop, &udev->symlink_list, node) {
291                                 if (strcmp(link_old_loop->name, link_loop->name) == 0) {
292                                         found = 1;
293                                         break;
294                                 }
295                         }
296                         if (!found) {
297                                 /* link does no longer belong to this device */
298                                 info("update old symlink '%s' no longer belonging to '%s'",
299                                      link_old_loop->name, udev->dev->devpath);
300                                 update_link(udev, link_old_loop->name);
301                         }
302                 }
303
304                 /*
305                  * if the node name has changed, delete the node,
306                  * or possibly restore a symlink of another device
307                  */
308                 if (strcmp(udev->name, udev_old->name) != 0)
309                         update_link(udev, udev_old->name);
310         }
311 }
312
313 int udev_node_add(struct udevice *udev)
314 {
315         char filename[PATH_SIZE];
316         uid_t uid;
317         gid_t gid;
318         int i;
319         int retval = 0;
320
321         strlcpy(filename, udev_root, sizeof(filename));
322         strlcat(filename, "/", sizeof(filename));
323         strlcat(filename, udev->name, sizeof(filename));
324         create_path(filename);
325
326         if (strcmp(udev->owner, "root") == 0)
327                 uid = 0;
328         else {
329                 char *endptr;
330                 unsigned long id;
331
332                 id = strtoul(udev->owner, &endptr, 10);
333                 if (endptr[0] == '\0')
334                         uid = (uid_t) id;
335                 else
336                         uid = lookup_user(udev->owner);
337         }
338
339         if (strcmp(udev->group, "root") == 0)
340                 gid = 0;
341         else {
342                 char *endptr;
343                 unsigned long id;
344
345                 id = strtoul(udev->group, &endptr, 10);
346                 if (endptr[0] == '\0')
347                         gid = (gid_t) id;
348                 else
349                         gid = lookup_group(udev->group);
350         }
351
352         info("creating device node '%s', major=%d, minor=%d, mode=%#o, uid=%d, gid=%d",
353              filename, major(udev->devt), minor(udev->devt), udev->mode, uid, gid);
354
355         if (!udev->test_run)
356                 if (udev_node_mknod(udev, filename, udev->devt, udev->mode, uid, gid) != 0) {
357                         retval = -1;
358                         goto exit;
359                 }
360
361         setenv("DEVNAME", filename, 1);
362
363         /* create all_partitions if requested */
364         if (udev->partitions) {
365                 char partitionname[PATH_SIZE];
366                 char *attr;
367                 int range;
368
369                 /* take the maximum registered minor range */
370                 attr = sysfs_attr_get_value(udev->dev->devpath, "range");
371                 if (attr != NULL) {
372                         range = atoi(attr);
373                         if (range > 1)
374                                 udev->partitions = range-1;
375                 }
376                 info("creating device partition nodes '%s[1-%i]'", filename, udev->partitions);
377                 if (!udev->test_run) {
378                         for (i = 1; i <= udev->partitions; i++) {
379                                 dev_t part_devt;
380
381                                 snprintf(partitionname, sizeof(partitionname), "%s%d", filename, i);
382                                 partitionname[sizeof(partitionname)-1] = '\0';
383                                 part_devt = makedev(major(udev->devt), minor(udev->devt) + i);
384                                 udev_node_mknod(udev, partitionname, part_devt, udev->mode, uid, gid);
385                         }
386                 }
387         }
388 exit:
389         return retval;
390 }
391
392 int udev_node_remove(struct udevice *udev)
393 {
394         char filename[PATH_SIZE];
395         char partitionname[PATH_SIZE];
396         struct stat stats;
397         int retval = 0;
398         int num;
399
400         strlcpy(filename, udev_root, sizeof(filename));
401         strlcat(filename, "/", sizeof(filename));
402         strlcat(filename, udev->name, sizeof(filename));
403         if (stat(filename, &stats) != 0) {
404                 dbg("device node '%s' not found", filename);
405                 return -1;
406         }
407         if (udev->devt && stats.st_rdev != udev->devt) {
408                 info("device node '%s' points to a different device, skip removal", filename);
409                 return -1;
410         }
411
412         info("removing device node '%s'", filename);
413         if (!udev->test_run)
414                 retval = unlink_secure(filename);
415         if (retval)
416                 return retval;
417
418         setenv("DEVNAME", filename, 1);
419         num = udev->partitions;
420         if (num > 0) {
421                 int i;
422
423                 info("removing all_partitions '%s[1-%i]'", filename, num);
424                 if (num > 255)
425                         return -1;
426                 for (i = 1; i <= num; i++) {
427                         snprintf(partitionname, sizeof(partitionname), "%s%d", filename, i);
428                         partitionname[sizeof(partitionname)-1] = '\0';
429                         if (!udev->test_run)
430                                 unlink_secure(partitionname);
431                 }
432         }
433         delete_path(filename);
434         return retval;
435 }