lib/utils_devpath.c

   1 /*
   2  * devname - search for device name
   3  *
   4  * Copyright (C) 2004, Jana Saout <jana@saout.de>
   5  * Copyright (C) 2004-2007, Clemens Fruhwirth <clemens@endorphin.org>
   6  * Copyright (C) 2009-2012, Red Hat, Inc. All rights reserved.
   7  * Copyright (C) 2009-2013, Milan Broz
   8  *
   9  * This program is free software; you can redistribute it and/or
  10  * modify it under the terms of the GNU General Public License
  11  * as published by the Free Software Foundation; either version 2
  12  * of the License, or (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  22  */
  23
  24 #include <string.h>
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <unistd.h>
  28 #include <dirent.h>
  29 #include <fcntl.h>
  30 #include <errno.h>
  31 #include <limits.h>
  32 #include <sys/stat.h>
  33 #include <sys/types.h>
  34 #ifdef HAVE_SYS_SYSMACROS_H
  35 #include <sys/sysmacros.h>     /* for major, minor */
  36 #endif
  37 #include "internal.h"
  38
  39 static char *__lookup_dev(char *path, dev_t dev, int dir_level, const int max_level)
  40 {
  41         struct dirent *entry;
  42         struct stat st;
  43         char *ptr;
  44         char *result = NULL;
  45         DIR *dir;
  46         int space;
  47
  48         /* Ignore strange nested directories */
  49         if (dir_level > max_level)
  50                 return NULL;
  51
  52         path[PATH_MAX - 1] = '\0';
  53         ptr = path + strlen(path);
  54         *ptr++ = '/';
  55         *ptr = '\0';
  56         space = PATH_MAX - (ptr - path);
  57
  58         dir = opendir(path);
  59         if (!dir)
  60                 return NULL;
  61
  62         while((entry = readdir(dir))) {
  63                 if (entry->d_name[0] == '.' ||
  64                     !strncmp(entry->d_name, "..", 2))
  65                         continue;
  66
  67                 if (dir_level == 0 &&
  68                     (!strcmp(entry->d_name, "shm") ||
  69                      !strcmp(entry->d_name, "fd") ||
  70                      !strcmp(entry->d_name, "char") ||
  71                      !strcmp(entry->d_name, "pts")))
  72                         continue;
  73
  74                 strncpy(ptr, entry->d_name, space);
  75                 if (stat(path, &st) < 0)
  76                         continue;
  77
  78                 if (S_ISDIR(st.st_mode)) {
  79                         result = __lookup_dev(path, dev, dir_level + 1, max_level);
  80                         if (result)
  81                                 break;
  82                 } else if (S_ISBLK(st.st_mode)) {
  83                         /* workaround: ignore dm-X devices, these are internal kernel names */
  84                         if (dir_level == 0 && dm_is_dm_kernel_name(entry->d_name))
  85                                 continue;
  86                         if (st.st_rdev == dev) {
  87                                 result = strdup(path);
  88                                 break;
  89                         }
  90                 }
  91         }
  92
  93         closedir(dir);
  94         return result;
  95 }
  96
  97 /*
  98  * Non-udev systemd need to scan for device here.
  99  */
 100 static char *lookup_dev_old(int major, int minor)
 101 {
 102         dev_t dev;
 103         char *result = NULL, buf[PATH_MAX + 1];
 104
 105         dev = makedev(major, minor);
 106         strncpy(buf, "/dev", PATH_MAX);
 107         buf[PATH_MAX] = '\0';
 108
 109         /* First try low level device */
 110         if ((result = __lookup_dev(buf, dev, 0, 0)))
 111                 return result;
 112
 113         /* If it is dm, try DM dir  */
 114         if (dm_is_dm_device(major, minor)) {
 115                 strncpy(buf, dm_get_dir(), PATH_MAX);
 116                 if ((result = __lookup_dev(buf, dev, 0, 0)))
 117                         return result;
 118         }
 119
 120         strncpy(buf, "/dev", PATH_MAX);
 121         return  __lookup_dev(buf, dev, 0, 4);
 122 }
 123
 124 /*
 125  * Returns string pointing to device in /dev according to "major:minor" dev_id
 126  */
 127 char *crypt_lookup_dev(const char *dev_id)
 128 {
 129         int major, minor;
 130         char link[PATH_MAX], path[PATH_MAX], *devname, *devpath = NULL;
 131         struct stat st;
 132         ssize_t len;
 133
 134         if (sscanf(dev_id, "%d:%d", &major, &minor) != 2)
 135                 return NULL;
 136
 137         if (snprintf(path, sizeof(path), "/sys/dev/block/%s", dev_id) < 0)
 138                 return NULL;
 139
 140         len = readlink(path, link, sizeof(link) - 1);
 141         if (len < 0) {
 142                 /* Without /sys use old scan */
 143                 if (stat("/sys/dev/block", &st) < 0)
 144                         return lookup_dev_old(major, minor);
 145                 return NULL;
 146         }
 147
 148         link[len] = '\0';
 149         devname = strrchr(link, '/');
 150         if (!devname)
 151                 return NULL;
 152         devname++;
 153
 154         if (dm_is_dm_kernel_name(devname))
 155                 devpath = dm_device_path("/dev/mapper/", major, minor);
 156         else if (snprintf(path, sizeof(path), "/dev/%s", devname) > 0)
 157                 devpath = strdup(path);
 158
 159         /*
 160          * Check that path is correct.
 161          */
 162         if (devpath && ((stat(devpath, &st) < 0) ||
 163             !S_ISBLK(st.st_mode) ||
 164             (st.st_rdev != makedev(major, minor)))) {
 165                 free(devpath);
 166                 /* Should never happen unless user mangles with dev nodes. */
 167                 return lookup_dev_old(major, minor);
 168         }
 169
 170         return devpath;
 171 }
 172
 173 static int _read_uint64(const char *sysfs_path, uint64_t *value)
 174 {
 175         char tmp[64] = {0};
 176         int fd, r;
 177
 178         if ((fd = open(sysfs_path, O_RDONLY)) < 0)
 179                 return 0;
 180         r = read(fd, tmp, sizeof(tmp));
 181         close(fd);
 182
 183         if (r <= 0)
 184                 return 0;
 185
 186         if (sscanf(tmp, "%" PRIu64, value) != 1)
 187                 return 0;
 188
 189         return 1;
 190 }
 191
 192 static int _sysfs_get_uint64(int major, int minor, uint64_t *value, const char *attr)
 193 {
 194         char path[PATH_MAX];
 195
 196         if (snprintf(path, sizeof(path), "/sys/dev/block/%d:%d/%s",
 197                      major, minor, attr) < 0)
 198                 return 0;
 199
 200         return _read_uint64(path, value);
 201 }
 202
 203 static int _path_get_uint64(const char *sysfs_path, uint64_t *value, const char *attr)
 204 {
 205         char path[PATH_MAX];
 206
 207         if (snprintf(path, sizeof(path), "%s/%s",
 208                      sysfs_path, attr) < 0)
 209                 return 0;
 210
 211         return _read_uint64(path, value);
 212 }
 213
 214 int crypt_dev_is_rotational(int major, int minor)
 215 {
 216         uint64_t val;
 217
 218         if (!_sysfs_get_uint64(major, minor, &val, "queue/rotational"))
 219                 return 1; /* if failed, expect rotational disk */
 220
 221         return val ? 1 : 0;
 222 }
 223
 224 int crypt_dev_is_partition(const char *dev_path)
 225 {
 226         uint64_t val;
 227         struct stat st;
 228
 229         if (stat(dev_path, &st) < 0)
 230                 return 0;
 231
 232         if (!S_ISBLK(st.st_mode))
 233                 return 0;
 234
 235         if (!_sysfs_get_uint64(major(st.st_rdev), minor(st.st_rdev),
 236                               &val, "partition"))
 237                 return 0;
 238
 239         return val ? 1 : 0;
 240 }
 241
 242 uint64_t crypt_dev_partition_offset(const char *dev_path)
 243 {
 244         uint64_t val;
 245         struct stat st;
 246
 247         if (!crypt_dev_is_partition(dev_path))
 248                 return 0;
 249
 250         if (stat(dev_path, &st) < 0)
 251                 return 0;
 252
 253         if (!_sysfs_get_uint64(major(st.st_rdev), minor(st.st_rdev),
 254                               &val, "start"))
 255                 return 0;
 256
 257         return val;
 258 }
 259
 260 /* Try to find partition which match offset and size on top level device */
 261 char *crypt_get_partition_device(const char *dev_path, uint64_t offset, uint64_t size)
 262 {
 263         char link[PATH_MAX], path[PATH_MAX], part_path[PATH_MAX], *devname;
 264         char *result = NULL;
 265         struct stat st;
 266         size_t devname_len;
 267         ssize_t len;
 268         struct dirent *entry;
 269         DIR *dir;
 270         uint64_t part_offset, part_size;
 271
 272         if (stat(dev_path, &st) < 0)
 273                 return NULL;
 274
 275         if (!S_ISBLK(st.st_mode))
 276                 return NULL;
 277
 278         if (snprintf(path, sizeof(path), "/sys/dev/block/%d:%d",
 279                 major(st.st_rdev), minor(st.st_rdev)) < 0)
 280                 return NULL;
 281
 282         len = readlink(path, link, sizeof(link) - 1);
 283         if (len < 0)
 284                 return NULL;
 285
 286         /* Get top level disk name for sysfs search */
 287         link[len] = '\0';
 288         devname = strrchr(link, '/');
 289         if (!devname)
 290                 return NULL;
 291         devname++;
 292
 293         /* DM devices do not use kernel partitions. */
 294         if (dm_is_dm_kernel_name(devname))
 295                 return NULL;
 296
 297         dir = opendir(path);
 298         if (!dir)
 299                 return NULL;
 300
 301         devname_len = strlen(devname);
 302         while((entry = readdir(dir))) {
 303                 if (strncmp(entry->d_name, devname, devname_len))
 304                         continue;
 305
 306                 if (snprintf(part_path, sizeof(part_path), "%s/%s",
 307                     path, entry->d_name) < 0)
 308                         continue;
 309
 310                 if (stat(part_path, &st) < 0)
 311                         continue;
 312
 313                 if (S_ISDIR(st.st_mode)) {
 314                         if (!_path_get_uint64(part_path, &part_offset, "start") ||
 315                             !_path_get_uint64(part_path, &part_size, "size"))
 316                                 continue;
 317                         if (part_offset == offset && part_size == size &&
 318                             snprintf(part_path, sizeof(part_path), "/dev/%s",
 319                                      entry->d_name) > 0) {
 320                                 result = strdup(part_path);
 321                                 break;
 322                         }
 323                 }
 324         }
 325         closedir(dir);
 326
 327         return result;
 328 }
 329
 330 /* Try to find base device from partition */
 331 char *crypt_get_base_device(const char *dev_path)
 332 {
 333         char link[PATH_MAX], path[PATH_MAX], part_path[PATH_MAX], *devname;
 334         struct stat st;
 335         ssize_t len;
 336
 337         if (!crypt_dev_is_partition(dev_path))
 338                 return NULL;
 339
 340         if (stat(dev_path, &st) < 0)
 341                 return NULL;
 342
 343         if (snprintf(path, sizeof(path), "/sys/dev/block/%d:%d",
 344                 major(st.st_rdev), minor(st.st_rdev)) < 0)
 345                 return NULL;
 346
 347         len = readlink(path, link, sizeof(link) - 1);
 348         if (len < 0)
 349                 return NULL;
 350
 351         /* Get top level disk name for sysfs search */
 352         link[len] = '\0';
 353         devname = strrchr(link, '/');
 354         if (!devname)
 355                 return NULL;
 356         *devname = '\0';
 357         devname = strrchr(link, '/');
 358         if (!devname)
 359                 return NULL;
 360         devname++;
 361
 362         if (dm_is_dm_kernel_name(devname))
 363                 return NULL;
 364
 365         snprintf(part_path, sizeof(part_path), "/dev/%s", devname);
 366         return strdup(part_path);
 367 }