chiark / gitweb /
always unlink temporary file before creating new one
[elogind.git] / udev_node.c
index 526d7135169d35f3b58af446eaf8c7648a398494..fcd52d0e45d5d9b861a353267bfbf5c56d64f64c 100644 (file)
 #include "udev_rules.h"
 #include "udev_selinux.h"
 
+#define TMP_FILE_EXT           ".udev-tmp"
 
 int udev_node_mknod(struct udevice *udev, const char *file, dev_t devt, mode_t mode, uid_t uid, gid_t gid)
 {
+       char file_tmp[PATH_SIZE + sizeof(TMP_FILE_EXT)];
        struct stat stats;
        int retval = 0;
 
@@ -44,28 +46,37 @@ int udev_node_mknod(struct udevice *udev, const char *file, dev_t devt, mode_t m
        else
                mode |= S_IFCHR;
 
-       if (stat(file, &stats) != 0)
-               goto create;
-
-       /* preserve node with already correct numbers, to prevent changing the inode number */
-       if ((stats.st_mode & S_IFMT) == (mode & S_IFMT) && (stats.st_rdev == devt)) {
-               info("preserve file '%s', because it has correct dev_t", file);
-               selinux_setfilecon(file, udev->dev->kernel, stats.st_mode);
-               goto perms;
+       if (lstat(file, &stats) == 0) {
+               if ((stats.st_mode & S_IFMT) == (mode & S_IFMT) && (stats.st_rdev == devt)) {
+                       info("preserve file '%s', because it has correct dev_t", file);
+                       selinux_setfilecon(file, udev->dev->kernel, stats.st_mode);
+                       goto perms;
+               }
+       } else {
+               selinux_setfscreatecon(file, udev->dev->kernel, mode);
+               retval = mknod(file, mode, devt);
+               selinux_resetfscreatecon();
+               if (retval == 0)
+                       goto perms;
        }
 
-       if (unlink(file) != 0)
-               err("unlink(%s) failed: %s", file, strerror(errno));
-       else
-               dbg("already present file '%s' unlinked", file);
-
-create:
-       selinux_setfscreatecon(file, udev->dev->kernel, mode);
-       retval = mknod(file, mode, devt);
+       info("atomically replace '%s'", file);
+       strlcpy(file_tmp, file, sizeof(file_tmp));
+       strlcat(file_tmp, TMP_FILE_EXT, sizeof(file_tmp));
+       unlink(file_tmp);
+       selinux_setfscreatecon(file_tmp, udev->dev->kernel, mode);
+       retval = mknod(file_tmp, mode, devt);
        selinux_resetfscreatecon();
        if (retval != 0) {
                err("mknod(%s, %#o, %u, %u) failed: %s",
-                   file, mode, major(devt), minor(devt), strerror(errno));
+                   file_tmp, mode, major(devt), minor(devt), strerror(errno));
+               goto exit;
+       }
+       retval = rename(file_tmp, file);
+       if (retval != 0) {
+               err("rename(%s, %s) failed: %s",
+                   file_tmp, file, strerror(errno));
+               unlink(file_tmp);
                goto exit;
        }
 
@@ -84,18 +95,19 @@ perms:
                        goto exit;
                }
        }
-
 exit:
        return retval;
 }
 
 static int node_symlink(const char *node, const char *slink)
 {
+       struct stat stats;
        char target[PATH_SIZE] = "";
-       char buf[PATH_SIZE];
+       char slink_tmp[PATH_SIZE + sizeof(TMP_FILE_EXT)];
        int i = 0;
        int tail = 0;
        int len;
+       int retval = 0;
 
        /* use relative link */
        while (node[i] && (node[i] == slink[i])) {
@@ -110,28 +122,63 @@ static int node_symlink(const char *node, const char *slink)
        }
        strlcat(target, &node[tail], sizeof(target));
 
-       /* look if symlink already exists */
-       len = readlink(slink, buf, sizeof(buf));
-       if (len > 0) {
-               buf[len] = '\0';
-               if (strcmp(target, buf) == 0) {
-                       info("preserve already existing symlink '%s' to '%s'", slink, target);
-                       selinux_setfilecon(slink, NULL, S_IFLNK);
-                       goto exit;
+       /* preserve link with correct target, do not replace node of other device */
+       if (lstat(slink, &stats) == 0) {
+               if (S_ISBLK(stats.st_mode) || S_ISCHR(stats.st_mode)) {
+                       struct stat stats2;
+
+                       info("found existing node instead of symlink '%s'", slink);
+                       if (lstat(node, &stats2) == 0) {
+                               if ((stats.st_mode & S_IFMT) == (stats2.st_mode & S_IFMT) &&
+                                   stats.st_rdev == stats2.st_rdev) {
+                                       info("replace device node '%s' with symlink to our node '%s'", slink, node);
+                               } else {
+                                       err("device node '%s' already exists, link '%s' will not overwrite it", node, slink);
+                                       goto exit;
+                               }
+                       }
+               } else if (S_ISLNK(stats.st_mode)) {
+                       char buf[PATH_SIZE];
+
+                       info("found existing symlink '%s'", slink);
+                       len = readlink(slink, buf, sizeof(buf));
+                       if (len > 0) {
+                               buf[len] = '\0';
+                               if (strcmp(target, buf) == 0) {
+                                       info("preserve already existing symlink '%s' to '%s'", slink, target);
+                                       selinux_setfilecon(slink, NULL, S_IFLNK);
+                                       goto exit;
+                               }
+                       }
                }
-               info("link '%s' points to different target '%s', delete it", slink, buf);
-               unlink(slink);
+       } else {
+               info("creating symlink '%s' to '%s'", slink, target);
+               selinux_setfscreatecon(slink, NULL, S_IFLNK);
+               retval = symlink(target, slink);
+               selinux_resetfscreatecon();
+               if (retval == 0)
+                       goto exit;
        }
 
-       /* create link */
-       info("creating symlink '%s' to '%s'", slink, target);
-       selinux_setfscreatecon(slink, NULL, S_IFLNK);
-       if (symlink(target, slink) != 0)
-               err("symlink(%s, %s) failed: %s", target, slink, strerror(errno));
+       info("atomically replace '%s'", slink);
+       strlcpy(slink_tmp, slink, sizeof(slink_tmp));
+       strlcat(slink_tmp, TMP_FILE_EXT, sizeof(slink_tmp));
+       unlink(slink_tmp);
+       selinux_setfscreatecon(slink_tmp, NULL, S_IFLNK);
+       retval = symlink(target, slink_tmp);
        selinux_resetfscreatecon();
-
+       if (retval != 0) {
+               err("symlink(%s, %s) failed: %s", target, slink_tmp, strerror(errno));
+               goto exit;
+       }
+       retval = rename(slink_tmp, slink);
+       if (retval != 0) {
+               err("rename(%s, %s) failed: %s", slink_tmp, slink, strerror(errno));
+               unlink(slink_tmp);
+               goto exit;
+       }
 exit:
-       return 0;
+       return retval;
 }
 
 static int update_link(struct udevice *udev, const char *name)
@@ -153,7 +200,7 @@ static int update_link(struct udevice *udev, const char *name)
        count = udev_db_get_devices_by_name(name, &name_list);
        info("found %i devices with name '%s'", count, name);
 
-       /* if we don't have any reference, we can delete the link */
+       /* if we don't have a reference, delete it */
        if (count <= 0) {
                info("no reference left, remove '%s'", name);
                if (!udev->test_run) {
@@ -236,11 +283,10 @@ void udev_node_update_symlinks(struct udevice *udev, struct udevice *udev_old)
        if (udev_old != NULL) {
                struct name_entry *link_loop;
                struct name_entry *link_old_loop;
-               struct name_entry *link_old_tmp_loop;
                int found;
 
                /* remove current symlinks from old list */
-               list_for_each_entry_safe(link_old_loop, link_old_tmp_loop, &udev_old->symlink_list, node) {
+               list_for_each_entry(link_old_loop, &udev_old->symlink_list, node) {
                        found = 0;
                        list_for_each_entry(link_loop, &udev->symlink_list, node) {
                                if (strcmp(link_old_loop->name, link_loop->name) == 0) {
@@ -255,6 +301,13 @@ void udev_node_update_symlinks(struct udevice *udev, struct udevice *udev_old)
                                update_link(udev, link_old_loop->name);
                        }
                }
+
+               /*
+                * if the node name has changed, delete the node,
+                * or possibly restore a symlink of another device
+                */
+               if (strcmp(udev->name, udev_old->name) != 0)
+                       update_link(udev, udev_old->name);
        }
 }
 
@@ -297,7 +350,7 @@ int udev_node_add(struct udevice *udev)
                        gid = lookup_group(udev->group);
        }
 
-       info("creating device node '%s', major = '%d', minor = '%d', " "mode = '%#o', uid = '%d', gid = '%d'",
+       info("creating device node '%s', major=%d, minor=%d, mode=%#o, uid=%d, gid=%d",
             filename, major(udev->devt), minor(udev->devt), udev->mode, uid, gid);
 
        if (!udev->test_run)
@@ -316,7 +369,7 @@ int udev_node_add(struct udevice *udev)
 
                /* take the maximum registered minor range */
                attr = sysfs_attr_get_value(udev->dev->devpath, "range");
-               if (attr) {
+               if (attr != NULL) {
                        range = atoi(attr);
                        if (range > 1)
                                udev->partitions = range-1;
@@ -342,7 +395,7 @@ int udev_node_remove(struct udevice *udev)
        char filename[PATH_SIZE];
        char partitionname[PATH_SIZE];
        struct stat stats;
-       int retval;
+       int retval = 0;
        int num;
 
        strlcpy(filename, udev_root, sizeof(filename));
@@ -358,7 +411,8 @@ int udev_node_remove(struct udevice *udev)
        }
 
        info("removing device node '%s'", filename);
-       retval = unlink_secure(filename);
+       if (!udev->test_run)
+               retval = unlink_secure(filename);
        if (retval)
                return retval;
 
@@ -373,7 +427,8 @@ int udev_node_remove(struct udevice *udev)
                for (i = 1; i <= num; i++) {
                        snprintf(partitionname, sizeof(partitionname), "%s%d", filename, i);
                        partitionname[sizeof(partitionname)-1] = '\0';
-                       unlink_secure(partitionname);
+                       if (!udev->test_run)
+                               unlink_secure(partitionname);
                }
        }
        delete_path(filename);