Imported Upstream version 1.6.7
[platform/upstream/cryptsetup.git] / lib / utils_devpath.c
index 35b639e..963785a 100644 (file)
@@ -1,13 +1,15 @@
 /*
  * devname - search for device name
  *
- * Copyright (C) 2004, Christophe Saout <christophe@saout.de>
+ * Copyright (C) 2004, Jana Saout <jana@saout.de>
  * Copyright (C) 2004-2007, Clemens Fruhwirth <clemens@endorphin.org>
- * Copyright (C) 2009-2011, Red Hat, Inc. All rights reserved.
+ * Copyright (C) 2009-2012, Red Hat, Inc. All rights reserved.
+ * Copyright (C) 2009-2013, Milan Broz
  *
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of the GNU General Public License
- * version 2 as published by the Free Software Foundation.
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
  *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
 #include <string.h>
 #include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
 #include <dirent.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <limits.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 #include "internal.h"
 
-#define DEVICE_DIR     "/dev"
-
 static char *__lookup_dev(char *path, dev_t dev, int dir_level, const int max_level)
 {
        struct dirent *entry;
@@ -56,6 +61,13 @@ static char *__lookup_dev(char *path, dev_t dev, int dir_level, const int max_le
                    !strncmp(entry->d_name, "..", 2))
                        continue;
 
+               if (dir_level == 0 &&
+                   (!strcmp(entry->d_name, "shm") ||
+                    !strcmp(entry->d_name, "fd") ||
+                    !strcmp(entry->d_name, "char") ||
+                    !strcmp(entry->d_name, "pts")))
+                       continue;
+
                strncpy(ptr, entry->d_name, space);
                if (stat(path, &st) < 0)
                        continue;
@@ -66,7 +78,7 @@ static char *__lookup_dev(char *path, dev_t dev, int dir_level, const int max_le
                                break;
                } else if (S_ISBLK(st.st_mode)) {
                        /* workaround: ignore dm-X devices, these are internal kernel names */
-                       if (dir_level == 0 && !strncmp(entry->d_name, "dm-", 3))
+                       if (dir_level == 0 && dm_is_dm_kernel_name(entry->d_name))
                                continue;
                        if (st.st_rdev == dev) {
                                result = strdup(path);
@@ -82,17 +94,13 @@ static char *__lookup_dev(char *path, dev_t dev, int dir_level, const int max_le
 /*
  * Non-udev systemd need to scan for device here.
  */
-static char *lookup_dev_old(const char *dev_id)
+static char *lookup_dev_old(int major, int minor)
 {
-       int major, minor;
        dev_t dev;
        char *result = NULL, buf[PATH_MAX + 1];
 
-       if (sscanf(dev_id, "%d:%d", &major, &minor) != 2)
-               return NULL;
-
        dev = makedev(major, minor);
-       strncpy(buf, DEVICE_DIR, PATH_MAX);
+       strncpy(buf, "/dev", PATH_MAX);
        buf[PATH_MAX] = '\0';
 
        /* First try low level device */
@@ -100,32 +108,37 @@ static char *lookup_dev_old(const char *dev_id)
                return result;
 
        /* If it is dm, try DM dir  */
-       if (dm_is_dm_device(major)) {
+       if (dm_is_dm_device(major, minor)) {
                strncpy(buf, dm_get_dir(), PATH_MAX);
                if ((result = __lookup_dev(buf, dev, 0, 0)))
                        return result;
        }
 
-       strncpy(buf, DEVICE_DIR, PATH_MAX);
-       result = __lookup_dev(buf, dev, 0, 4);
-
-       /* If not found, return NULL */
-       return result;
+       strncpy(buf, "/dev", PATH_MAX);
+       return  __lookup_dev(buf, dev, 0, 4);
 }
 
+/*
+ * Returns string pointing to device in /dev according to "major:minor" dev_id
+ */
 char *crypt_lookup_dev(const char *dev_id)
 {
-       char link[PATH_MAX], path[PATH_MAX], *devname;
+       int major, minor;
+       char link[PATH_MAX], path[PATH_MAX], *devname, *devpath = NULL;
        struct stat st;
        ssize_t len;
 
+       if (sscanf(dev_id, "%d:%d", &major, &minor) != 2)
+               return NULL;
+
        if (snprintf(path, sizeof(path), "/sys/dev/block/%s", dev_id) < 0)
                return NULL;
 
-       len = readlink(path, link, sizeof(link));
+       len = readlink(path, link, sizeof(link) - 1);
        if (len < 0) {
+               /* Without /sys use old scan */
                if (stat("/sys/dev/block", &st) < 0)
-                       return lookup_dev_old(dev_id);
+                       return lookup_dev_old(major, minor);
                return NULL;
        }
 
@@ -135,11 +148,217 @@ char *crypt_lookup_dev(const char *dev_id)
                return NULL;
        devname++;
 
-       if (!strncmp(devname, "dm-", 3))
-               return dm_device_path(dev_id);
+       if (dm_is_dm_kernel_name(devname))
+               devpath = dm_device_path("/dev/mapper/", major, minor);
+       else if (snprintf(path, sizeof(path), "/dev/%s", devname) > 0)
+               devpath = strdup(path);
+
+       /*
+        * Check that path is correct.
+        */
+       if (devpath && ((stat(devpath, &st) < 0) ||
+           !S_ISBLK(st.st_mode) ||
+           (st.st_rdev != makedev(major, minor)))) {
+               free(devpath);
+               /* Should never happen unless user mangles with dev nodes. */
+               return lookup_dev_old(major, minor);
+       }
+
+       return devpath;
+}
+
+static int _read_uint64(const char *sysfs_path, uint64_t *value)
+{
+       char tmp[64] = {0};
+       int fd, r;
+
+       if ((fd = open(sysfs_path, O_RDONLY)) < 0)
+               return 0;
+       r = read(fd, tmp, sizeof(tmp));
+       close(fd);
+
+       if (r <= 0)
+               return 0;
+
+        if (sscanf(tmp, "%" PRIu64, value) != 1)
+               return 0;
+
+       return 1;
+}
+
+static int _sysfs_get_uint64(int major, int minor, uint64_t *value, const char *attr)
+{
+       char path[PATH_MAX];
+
+       if (snprintf(path, sizeof(path), "/sys/dev/block/%d:%d/%s",
+                    major, minor, attr) < 0)
+               return 0;
+
+       return _read_uint64(path, value);
+}
+
+static int _path_get_uint64(const char *sysfs_path, uint64_t *value, const char *attr)
+{
+       char path[PATH_MAX];
+
+       if (snprintf(path, sizeof(path), "%s/%s",
+                    sysfs_path, attr) < 0)
+               return 0;
+
+       return _read_uint64(path, value);
+}
+
+int crypt_dev_is_rotational(int major, int minor)
+{
+       uint64_t val;
+
+       if (!_sysfs_get_uint64(major, minor, &val, "queue/rotational"))
+               return 1; /* if failed, expect rotational disk */
+
+       return val ? 1 : 0;
+}
+
+int crypt_dev_is_partition(const char *dev_path)
+{
+       uint64_t val;
+       struct stat st;
+
+       if (stat(dev_path, &st) < 0)
+               return 0;
+
+       if (!S_ISBLK(st.st_mode))
+               return 0;
+
+       if (!_sysfs_get_uint64(major(st.st_rdev), minor(st.st_rdev),
+                             &val, "partition"))
+               return 0;
+
+       return val ? 1 : 0;
+}
+
+uint64_t crypt_dev_partition_offset(const char *dev_path)
+{
+       uint64_t val;
+       struct stat st;
+
+       if (!crypt_dev_is_partition(dev_path))
+               return 0;
+
+       if (stat(dev_path, &st) < 0)
+               return 0;
+
+       if (!_sysfs_get_uint64(major(st.st_rdev), minor(st.st_rdev),
+                             &val, "start"))
+               return 0;
+
+       return val;
+}
+
+/* Try to find partition which match offset and size on top level device */
+char *crypt_get_partition_device(const char *dev_path, uint64_t offset, uint64_t size)
+{
+       char link[PATH_MAX], path[PATH_MAX], part_path[PATH_MAX], *devname;
+       char *result = NULL;
+       struct stat st;
+       size_t devname_len;
+       ssize_t len;
+       struct dirent *entry;
+       DIR *dir;
+       uint64_t part_offset, part_size;
+
+       if (stat(dev_path, &st) < 0)
+               return NULL;
+
+       if (!S_ISBLK(st.st_mode))
+               return NULL;
+
+       if (snprintf(path, sizeof(path), "/sys/dev/block/%d:%d",
+               major(st.st_rdev), minor(st.st_rdev)) < 0)
+               return NULL;
+
+       len = readlink(path, link, sizeof(link) - 1);
+       if (len < 0)
+               return NULL;
+
+       /* Get top level disk name for sysfs search */
+       link[len] = '\0';
+       devname = strrchr(link, '/');
+       if (!devname)
+               return NULL;
+       devname++;
+
+       /* DM devices do not use kernel partitions. */
+       if (dm_is_dm_kernel_name(devname))
+               return NULL;
+
+       dir = opendir(path);
+       if (!dir)
+               return NULL;
+
+       devname_len = strlen(devname);
+       while((entry = readdir(dir))) {
+               if (strncmp(entry->d_name, devname, devname_len))
+                       continue;
+
+               if (snprintf(part_path, sizeof(part_path), "%s/%s",
+                   path, entry->d_name) < 0)
+                       continue;
+
+               if (stat(part_path, &st) < 0)
+                       continue;
+
+               if (S_ISDIR(st.st_mode)) {
+                       if (!_path_get_uint64(part_path, &part_offset, "start") ||
+                           !_path_get_uint64(part_path, &part_size, "size"))
+                               continue;
+                       if (part_offset == offset && part_size == size &&
+                           snprintf(part_path, sizeof(part_path), "/dev/%s",
+                                    entry->d_name) > 0) {
+                               result = strdup(part_path);
+                               break;
+                       }
+               }
+       }
+       closedir(dir);
+
+       return result;
+}
+
+/* Try to find base device from partition */
+char *crypt_get_base_device(const char *dev_path)
+{
+       char link[PATH_MAX], path[PATH_MAX], part_path[PATH_MAX], *devname;
+       struct stat st;
+       ssize_t len;
+
+       if (!crypt_dev_is_partition(dev_path))
+               return NULL;
+
+       if (stat(dev_path, &st) < 0)
+               return NULL;
+
+       if (snprintf(path, sizeof(path), "/sys/dev/block/%d:%d",
+               major(st.st_rdev), minor(st.st_rdev)) < 0)
+               return NULL;
+
+       len = readlink(path, link, sizeof(link) - 1);
+       if (len < 0)
+               return NULL;
+
+       /* Get top level disk name for sysfs search */
+       link[len] = '\0';
+       devname = strrchr(link, '/');
+       if (!devname)
+               return NULL;
+       *devname = '\0';
+       devname = strrchr(link, '/');
+       if (!devname)
+               return NULL;
+       devname++;
 
-       if (snprintf(path, sizeof(path), "/dev/%s", devname) < 0)
+       if (dm_is_dm_kernel_name(devname))
                return NULL;
 
-       return strdup(path);
+       snprintf(part_path, sizeof(part_path), "/dev/%s", devname);
+       return strdup(part_path);
 }