extras/volume_id/lib/volume_id.c

   1 /*
   2  * volume_id - reads volume label and uuid
   3  *
   4  * Copyright (C) 2005-2007 Kay Sievers <kay.sievers@vrfy.org>
   5  *
   6  *      This program is free software; you can redistribute it and/or modify it
   7  *      under the terms of the GNU General Public License as published by the
   8  *      Free Software Foundation version 2 of the License.
   9  */
  10
  11 #ifndef _GNU_SOURCE
  12 #define _GNU_SOURCE 1
  13 #endif
  14
  15 #ifdef HAVE_CONFIG_H
  16 #  include <config.h>
  17 #endif
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <unistd.h>
  22 #include <string.h>
  23 #include <errno.h>
  24 #include <ctype.h>
  25 #include <fcntl.h>
  26 #include <sys/stat.h>
  27
  28 #include "libvolume_id.h"
  29 #include "util.h"
  30
  31 #define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0]))
  32
  33 struct prober {
  34         volume_id_probe_fn_t prober;
  35         const char *name[4];
  36 };
  37
  38 static const struct prober prober_raid[] = {
  39         { volume_id_probe_linux_raid, { "linux_raid", } },
  40         { volume_id_probe_ddf_raid, { "ddf_raid", } },
  41         { volume_id_probe_intel_software_raid, { "isw_raid", } },
  42         { volume_id_probe_lsi_mega_raid, { "lsi_mega_raid", } },
  43         { volume_id_probe_via_raid, { "via_raid", } },
  44         { volume_id_probe_silicon_medley_raid, { "silicon_medley_raid", } },
  45         { volume_id_probe_nvidia_raid, { "nvidia_raid", } },
  46         { volume_id_probe_promise_fasttrack_raid, { "promise_fasttrack_raid", } },
  47         { volume_id_probe_highpoint_45x_raid, { "highpoint_raid", } },
  48         { volume_id_probe_adaptec_raid, { "adaptec_raid", } },
  49         { volume_id_probe_jmicron_raid, { "jmicron_raid", } },
  50         { volume_id_probe_lvm1, { "lvm1", } },
  51         { volume_id_probe_lvm2, { "lvm2", } },
  52         { volume_id_probe_highpoint_37x_raid, { "highpoint_raid", } },
  53 };
  54
  55 static const struct prober prober_filesystem[] = {
  56         { volume_id_probe_vfat, { "vfat", } },
  57         { volume_id_probe_linux_swap, { "swap", } },
  58         { volume_id_probe_luks, { "luks", } },
  59         { volume_id_probe_xfs, { "xfs", } },
  60         { volume_id_probe_ext, { "ext2", "ext3", "jbd", } },
  61         { volume_id_probe_reiserfs, { "reiserfs", "reiser4", } },
  62         { volume_id_probe_jfs, { "jfs", } },
  63         { volume_id_probe_udf, { "udf", } },
  64         { volume_id_probe_iso9660, { "iso9660", } },
  65         { volume_id_probe_hfs_hfsplus, { "hfs", "hfsplus", } },
  66         { volume_id_probe_ufs, { "ufs", } },
  67         { volume_id_probe_ntfs, { "ntfs", } },
  68         { volume_id_probe_cramfs, { "cramfs", } },
  69         { volume_id_probe_romfs, { "romfs", } },
  70         { volume_id_probe_hpfs, { "hpfs", } },
  71         { volume_id_probe_sysv, { "sysv", "xenix", } },
  72         { volume_id_probe_minix, { "minix",  } },
  73         { volume_id_probe_ocfs1, { "ocfs1", } },
  74         { volume_id_probe_ocfs2, { "ocfs2", } },
  75         { volume_id_probe_vxfs, { "vxfs", } },
  76         { volume_id_probe_squashfs, { "squashfs", } },
  77         { volume_id_probe_netware, { "netware", } },
  78 };
  79
  80 /* the user can overwrite this log function */
  81 static void default_log(int priority, const char *file, int line, const char *format, ...)
  82 {
  83         return;
  84 }
  85
  86 volume_id_log_fn_t volume_id_log_fn = default_log;
  87
  88 /**
  89  * volume_id_get_label:
  90  * @type: Type string.
  91  *
  92  * Lookup the probing function for a specific type.
  93  *
  94  * Returns: The probing function for the given type, #NULL otherwise.
  95  **/
  96 const volume_id_probe_fn_t *volume_id_get_prober_by_type(const char *type)
  97 {
  98         unsigned int p, n;
  99
 100         if (type == NULL)
 101                 return NULL;
 102
 103         for (p = 0; p < ARRAY_SIZE(prober_raid); p++)
 104                 for (n = 0; prober_raid[p].name[n] !=  NULL; n++)
 105                         if (strcmp(type, prober_raid[p].name[n]) == 0)
 106                                 return &prober_raid[p].prober;
 107         for (p = 0; p < ARRAY_SIZE(prober_filesystem); p++)
 108                 for (n = 0; prober_filesystem[p].name[n] !=  NULL; n++)
 109                         if (strcmp(type, prober_filesystem[p].name[n]) == 0)
 110                                 return &prober_filesystem[p].prober;
 111         return NULL;
 112 }
 113
 114 /**
 115  * volume_id_get_label:
 116  * @id: Probing context.
 117  * @label: Label string. Must not be freed by the caller.
 118  *
 119  * Get the label string after a successful probe. Unicode
 120  * is translated to UTF-8.
 121  *
 122  * Returns: 1 if the value was set, 0 otherwise.
 123  **/
 124 int volume_id_get_label(struct volume_id *id, const char **label)
 125 {
 126         if (id == NULL)
 127                 return 0;
 128         if (label == NULL)
 129                 return 0;
 130         if (id->usage_id == VOLUME_ID_UNUSED)
 131                 return 0;
 132
 133         *label = id->label;
 134         return 1;
 135 }
 136
 137 /**
 138  * volume_id_get_label_raw:
 139  * @id: Probing context.
 140  * @label: Label byte array. Must not be freed by the caller.
 141  * @len: Length of raw label byte array.
 142  *
 143  * Get the raw label byte array after a successful probe. It may
 144  * contain undecoded multibyte character streams.
 145  *
 146  * Returns: 1 if the value was set, 0 otherwise.
 147  **/
 148 int volume_id_get_label_raw(struct volume_id *id, const uint8_t **label, size_t *len)
 149 {
 150         if (id == NULL)
 151                 return 0;
 152         if (label == NULL)
 153                 return 0;
 154         if (len == NULL)
 155                 return 0;
 156         if (id->usage_id == VOLUME_ID_UNUSED)
 157                 return 0;
 158
 159         *label = id->label_raw;
 160         *len = id->label_raw_len;
 161         return 1;
 162 }
 163
 164 /**
 165  * volume_id_get_uuid:
 166  * @id: Probing context.
 167  * @uuid: UUID string. Must not be freed by the caller.
 168  *
 169  * Get the raw UUID string after a successful probe.
 170  *
 171  * Returns: 1 if the value was set, 0 otherwise.
 172  **/
 173 int volume_id_get_uuid(struct volume_id *id, const char **uuid)
 174 {
 175         if (id == NULL)
 176                 return 0;
 177         if (uuid == NULL)
 178                 return 0;
 179         if (id->usage_id == VOLUME_ID_UNUSED)
 180                 return 0;
 181
 182         *uuid = id->uuid;
 183         return 1;
 184 }
 185
 186 /**
 187  * volume_id_get_uuid_raw:
 188  * @id: Probing context.
 189  * @uuid: UUID byte array. Must not be freed by the caller.
 190  * @len: Length of raw UUID byte array.
 191  *
 192  * Get the raw UUID byte array after a successful probe. It may
 193  * contain unconverted endianes values.
 194  *
 195  * Returns: 1 if the value was set, 0 otherwise.
 196  **/
 197 int volume_id_get_uuid_raw(struct volume_id *id, const uint8_t **uuid, size_t *len)
 198 {
 199         if (id == NULL)
 200                 return 0;
 201         if (uuid == NULL)
 202                 return 0;
 203         if (len == NULL)
 204                 return 0;
 205         if (id->usage_id == VOLUME_ID_UNUSED)
 206                 return 0;
 207
 208         *uuid = id->uuid_raw;
 209         *len = id->uuid_raw_len;
 210         return 1;
 211 }
 212
 213 /**
 214  * volume_id_get_usage:
 215  * @id: Probing context.
 216  * @usage: Usage string. Must not be freed by the caller.
 217  *
 218  * Get the usage string after a successful probe.
 219  *
 220  * Returns: 1 if the value was set, 0 otherwise.
 221  **/
 222 int volume_id_get_usage(struct volume_id *id, const char **usage)
 223 {
 224         if (id == NULL)
 225                 return 0;
 226         if (usage == NULL)
 227                 return 0;
 228         if (id->usage_id == VOLUME_ID_UNUSED)
 229                 return 0;
 230
 231         *usage = id->usage;
 232         return 1;
 233 }
 234
 235 /**
 236  * volume_id_get_type:
 237  * @id: Probing context
 238  * @type: Type string. Must not be freed by the caller.
 239  *
 240  * Get the type string after a successful probe.
 241  *
 242  * Returns: 1 if the value was set, 0 otherwise.
 243  **/
 244 int volume_id_get_type(struct volume_id *id, const char **type)
 245 {
 246         if (id == NULL)
 247                 return 0;
 248         if (type == NULL)
 249                 return 0;
 250         if (id->usage_id == VOLUME_ID_UNUSED)
 251                 return 0;
 252
 253         *type = id->type;
 254         return 1;
 255 }
 256
 257 /**
 258  * volume_id_get_type_version:
 259  * @id: Probing context.
 260  * @type_version: Type version string. Must not be freed by the caller.
 261  *
 262  * Get the Type version string after a successful probe.
 263  *
 264  * Returns: 1 if the value was set, 0 otherwise.
 265  **/
 266 int volume_id_get_type_version(struct volume_id *id, const char **type_version)
 267 {
 268         if (id == NULL)
 269                 return 0;
 270         if (type_version == NULL)
 271                 return 0;
 272         if (id->usage_id == VOLUME_ID_UNUSED)
 273                 return 0;
 274
 275         *type_version = id->type_version;
 276         return 1;
 277 }
 278
 279 /**
 280  * volume_id_probe_raid:
 281  * @id: Probing context.
 282  * @off: Probing offset relative to the start of the device.
 283  * @size: Total size of the device.
 284  *
 285  * Probe device for all known raid signatures.
 286  *
 287  * Returns: 0 on successful probe, otherwise negative value.
 288  **/
 289 int volume_id_probe_raid(struct volume_id *id, uint64_t off, uint64_t size)
 290 {
 291         unsigned int i;
 292
 293         if (id == NULL)
 294                 return -EINVAL;
 295
 296         info("probing at offset 0x%llx, size 0x%llx",
 297             (unsigned long long) off, (unsigned long long) size);
 298
 299         for (i = 0; i < ARRAY_SIZE(prober_raid); i++)
 300                 if (prober_raid[i].prober(id, off, size) == 0)
 301                         goto found;
 302         return -1;
 303
 304 found:
 305         /* If recognized, we free the allocated buffers */
 306         volume_id_free_buffer(id);
 307         return 0;
 308 }
 309
 310 /**
 311  * volume_id_probe_filesystem:
 312  * @id: Probing context.
 313  * @off: Probing offset relative to the start of the device.
 314  * @size: Total size of the device.
 315  *
 316  * Probe device for all known filesystem signatures.
 317  *
 318  * Returns: 0 on successful probe, otherwise negative value.
 319  **/
 320 int volume_id_probe_filesystem(struct volume_id *id, uint64_t off, uint64_t size)
 321 {
 322         unsigned int i;
 323
 324         if (id == NULL)
 325                 return -EINVAL;
 326
 327         info("probing at offset 0x%llx, size 0x%llx",
 328             (unsigned long long) off, (unsigned long long) size);
 329
 330         for (i = 0; i < ARRAY_SIZE(prober_filesystem); i++)
 331                 if (prober_filesystem[i].prober(id, off, size) == 0)
 332                         goto found;
 333         return -1;
 334
 335 found:
 336         /* If recognized, we free the allocated buffers */
 337         volume_id_free_buffer(id);
 338         return 0;
 339 }
 340
 341 /**
 342  * volume_id_probe_all:
 343  * @id: Probing context.
 344  * @off: Probing offset relative to the start of the device.
 345  * @size: Total size of the device.
 346  *
 347  * Probe device for all known raid and filesystem signatures.
 348  *
 349  * Returns: 0 on successful probe, otherwise negative value.
 350  **/
 351 int volume_id_probe_all(struct volume_id *id, uint64_t off, uint64_t size)
 352 {
 353         if (id == NULL)
 354                 return -EINVAL;
 355
 356         /* probe for raid first, because fs probes may be successful on raid members */
 357         if (volume_id_probe_raid(id, off, size) == 0)
 358                 return 0;
 359
 360         if (volume_id_probe_filesystem(id, off, size) == 0)
 361                 return 0;
 362
 363         return -1;
 364 }
 365
 366 /**
 367  * volume_id_probe_raid:
 368  * @all_probers_fn: prober function to called for all known probing routines.
 369  * @id: Context passed to prober function.
 370  * @off: Offset value passed to prober function.
 371  * @size: Size value passed to prober function.
 372  * @data: Arbitrary data passed to the prober function.
 373  *
 374  * Run a custom function for all known probing routines.
 375  **/
 376 void volume_id_all_probers(all_probers_fn_t all_probers_fn,
 377                            struct volume_id *id, uint64_t off, uint64_t size,
 378                            void *data)
 379 {
 380         unsigned int i;
 381
 382         if (all_probers_fn == NULL)
 383                 return;
 384
 385         for (i = 0; i < ARRAY_SIZE(prober_raid); i++)
 386                 if (all_probers_fn(prober_raid[i].prober, id, off, size, data) != 0)
 387                         goto out;
 388         for (i = 0; i < ARRAY_SIZE(prober_filesystem); i++)
 389                 if (all_probers_fn(prober_filesystem[i].prober, id, off, size, data) != 0)
 390                         goto out;
 391 out:
 392         return;
 393 }
 394
 395 /**
 396  * volume_id_open_fd:
 397  * @id: Probing context.
 398  * @fd: Open file descriptor of device to read from.
 399  *
 400  * Create the context for probing.
 401  *
 402  * Returns: Probing context, or #NULL on failure.
 403  **/
 404 struct volume_id *volume_id_open_fd(int fd)
 405 {
 406         struct volume_id *id;
 407
 408         id = malloc(sizeof(struct volume_id));
 409         if (id == NULL)
 410                 return NULL;
 411         memset(id, 0x00, sizeof(struct volume_id));
 412
 413         id->fd = fd;
 414
 415         return id;
 416 }
 417
 418 struct volume_id *volume_id_open_node(const char *path)
 419 {
 420         struct volume_id *id;
 421         int fd;
 422
 423         fd = open(path, O_RDONLY);
 424         if (fd < 0) {
 425                 dbg("unable to open '%s'", path);
 426                 return NULL;
 427         }
 428
 429         id = volume_id_open_fd(fd);
 430         if (id == NULL)
 431                 return NULL;
 432
 433         /* close fd on device close */
 434         id->fd_close = 1;
 435
 436         return id;
 437 }
 438
 439 /**
 440  * volume_id_close:
 441  * @id: Probing context.
 442  *
 443  * Release probing context and free all associated data.
 444  */
 445 void volume_id_close(struct volume_id *id)
 446 {
 447         if (id == NULL)
 448                 return;
 449
 450         if (id->fd_close != 0)
 451                 close(id->fd);
 452
 453         volume_id_free_buffer(id);
 454
 455         free(id);
 456 }