Fix missing headers
[platform/upstream/cryptsetup.git] / lib / utils_devpath.c
index 35b639e..fc5175c 100644 (file)
@@ -3,11 +3,13 @@
  *
  * Copyright (C) 2004, Christophe Saout <christophe@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-2012, 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"
+#include "utils_dm.h"
 
-#define DEVICE_DIR     "/dev"
+char *crypt_lookup_dev(const char *dev_id);
+int crypt_sysfs_get_rotational(int major, int minor, int *rotational);
 
 static char *__lookup_dev(char *path, dev_t dev, int dir_level, const int max_level)
 {
@@ -56,6 +64,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 +81,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 +97,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 +111,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 +151,44 @@ 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;
+}
 
-       if (snprintf(path, sizeof(path), "/dev/%s", devname) < 0)
-               return NULL;
+int crypt_sysfs_get_rotational(int major, int minor, int *rotational)
+{
+       char path[PATH_MAX], tmp[64] = {0};
+       int fd, r;
+
+       if (snprintf(path, sizeof(path), "/sys/dev/block/%d:%d/queue/rotational",
+                    major, minor) < 0)
+               return 0;
+
+       if ((fd = open(path, O_RDONLY)) < 0)
+               return 0;
+       r = read(fd, tmp, sizeof(tmp));
+       close(fd);
+
+       if (r <= 0)
+               return 0;
+
+        if (sscanf(tmp, "%d", rotational) != 1)
+               return 0;
 
-       return strdup(path);
+       return 1;
 }