X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=extras%2Fvolume_id%2Fvol_id.c;h=3e44c3b069794b81a1e61338c98eef0fa7671bf3;hp=0a1b756fa3174c62d557234844220faed798ac7b;hb=86d9bf4bece6d42d2ed91e99bf2773c6b5c30ec3;hpb=5c51e31e0e794d8fcdc1225c65985571c9820949 diff --git a/extras/volume_id/vol_id.c b/extras/volume_id/vol_id.c index 0a1b756fa..3e44c3b06 100644 --- a/extras/volume_id/vol_id.c +++ b/extras/volume_id/vol_id.c @@ -1,21 +1,20 @@ /* * vol_id - read filesystem label and uuid * - * Copyright (C) 2005-2006 Kay Sievers + * Copyright (C) 2005-2008 Kay Sievers * - * This program is free software; you can redistribute it and/or modify it - * under the terms of the GNU General Public License as published by the - * Free Software Foundation version 2 of the License. - * - * This program is distributed in the hope that it will be useful, but - * WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU - * General Public License for more details. - * - * 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., - * 675 Mass Ave, Cambridge, MA 02139, USA. + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License 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 + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . */ #ifndef _GNU_SOURCE @@ -30,288 +29,300 @@ #include #include #include +#include +#include +#include +#include #include -#include "../../udev.h" +#include "../../udev/udev.h" #include "lib/libvolume_id.h" #define BLKGETSIZE64 _IOR(0x12,114,size_t) -#ifdef USE_LOG -void log_message(int priority, const char *format, ...) -{ - va_list args; - static int udev_log = -1; - - if (udev_log == -1) { - const char *value; +static int debug; +struct udev *udev_ctx; - value = getenv("UDEV_LOG"); - if (value) - udev_log = log_priority(value); - else - udev_log = LOG_ERR; +static void log_fn(struct udev *udev, int priority, + const char *file, int line, const char *fn, + const char *format, va_list args) +{ + if (debug) { + fprintf(stderr, "%s: ", fn != NULL ? fn : file); + vfprintf(stderr, format, args); + } else { + vsyslog(priority, format, args); } - - if (priority > udev_log) - return; - - va_start(args, format); - vsyslog(priority, format, args); - va_end(args); } -#endif static void vid_log(int priority, const char *file, int line, const char *format, ...) { -#ifdef USE_LOG - char log_str[1024]; va_list args; + if (priority > udev_get_log_priority(udev_ctx)) + return; va_start(args, format); - vsnprintf(log_str, sizeof(log_str), format, args); - log_str[sizeof(log_str)-1] = '\0'; - log_message(priority, "%s:%i %s", file, line, log_str); + log_fn(udev_ctx, priority, file, line, NULL, format, args); va_end(args); -#endif return; } -static void set_str(char *to, const char *from, size_t count) +static int all_probers(volume_id_probe_fn_t probe_fn, + struct volume_id *id, uint64_t off, uint64_t size, + void *data) { - size_t i, j, len; - - /* strip trailing whitespace */ - len = strnlen(from, count); - while (len && isspace(from[len-1])) - len--; - - /* strip leading whitespace */ - i = 0; - while (isspace(from[i]) && (i < len)) - i++; - - j = 0; - while (i < len) { - /* substitute multiple whitespace */ - if (isspace(from[i])) { - while (isspace(from[i])) - i++; - to[j++] = '_'; - } - /* skip chars */ - if (from[i] == '/') { - i++; - continue; - } - to[j++] = from[i++]; - } - to[j] = '\0'; + const char *type; + + if (probe_fn(id, off, size) == 0) + if (volume_id_get_type(id, &type)) + printf("%s\n", type); + + return 0; } int main(int argc, char *argv[]) { - const char help[] = "Usage: vol_id [options] \n" - " --export export key/value pairs\n" - " -t filesystem type\n" - " -l filesystem label\n" - " -u filesystem uuid\n" - " --skip-raid don't probe for raid\n" - " --probe-all find possibly conflicting signatures\n" - " --help\n" - "\n"; + static const struct option options[] = { + { "label", no_argument, NULL, 'l' }, + { "label-raw", no_argument, NULL, 'L' }, + { "uuid", no_argument, NULL, 'u' }, + { "type", no_argument, NULL, 't' }, + { "export", no_argument, NULL, 'x' }, + { "skip-raid", no_argument, NULL, 's' }, + { "size", required_argument, NULL, 'S' }, + { "probe-all", no_argument, NULL, 'a' }, + { "offset", optional_argument, NULL, 'o' }, + { "debug", no_argument, NULL, 'd' }, + { "help", no_argument, NULL, 'h' }, + {} + }; + enum print_type { PRINT_EXPORT, PRINT_TYPE, PRINT_LABEL, PRINT_UUID, + PRINT_LABEL_RAW, } print = PRINT_EXPORT; + struct volume_id *vid = NULL; - static char name[VOLUME_ID_LABEL_SIZE]; - int i; - uint64_t size; + char label_safe[256]; + char label_enc[256]; + char uuid_safe[256]; + char uuid_enc[256]; + char uuid_sub_enc[256]; + char type_enc[256]; + char type_version_enc[256]; + uint64_t size = 0; int skip_raid = 0; int probe_all = 0; - const char *node = NULL; - struct passwd *pw; + uint64_t offset = 0; + const char *node; + int fd; + const char *label, *uuid, *uuid_sub, *type, *type_version, *usage; int retval; int rc = 0; + udev_ctx = udev_new(); + if (udev_ctx == NULL) + goto exit; logging_init("vol_id"); + udev_set_log_fn(udev_ctx, log_fn); /* hook in our debug into libvolume_id */ volume_id_log_fn = vid_log; - for (i = 1 ; i < argc; i++) { - char *arg = argv[i]; + while (1) { + int option; - if (strcmp(arg, "--export") == 0) { - print = PRINT_EXPORT; - } else if (strcmp(arg, "-t") == 0) { - print = PRINT_TYPE; - } else if (strcmp(arg, "-l") == 0) { + option = getopt_long(argc, argv, "lLutxsS:aodh", options, NULL); + if (option == -1) + break; + + switch (option) { + case 'd': + debug = 1; + if (udev_get_log_priority(udev_ctx) < LOG_INFO) + udev_set_log_priority(udev_ctx, LOG_INFO); + break; + case 'l': print = PRINT_LABEL; - } else if (strcmp(arg, "-u") == 0) { + break; + case 'L': + print = PRINT_LABEL_RAW; + break; + case 'u': print = PRINT_UUID; - } else if (strcmp(arg, "--skip-raid") == 0) { + break; + case 't': + print = PRINT_TYPE; + break; + case 'x': + print = PRINT_EXPORT; + break; + case 's': skip_raid = 1; - } else if (strcmp(arg, "--probe-all") == 0) { + break; + case 'a': probe_all = 1; - } else if (strcmp(arg, "--help") == 0 || strcmp(arg, "-h") == 0) { - printf(help); + break; + case 'S': + if (optarg[0] != '\0') + size = strtoull(optarg, NULL, 0); + break; + case 'o': + if (optarg[0] != '\0') + offset = strtoull(optarg, NULL, 0); + break; + case 'h': + printf("Usage: vol_id [options] \n" + " --export export key/value pairs\n" + " --type filesystem type\n" + " --label filesystem label\n" + " --label-raw raw label\n" + " --uuid filesystem uuid\n" + " --skip-raid don't probe for raid\n" + " --probe-all find possibly conflicting signatures\n" + " --offset= probe at the given offset\n" + " --size= overwrite device size\n" + " --debug print debug output to stderr\n" + " --help\n\n"); goto exit; - } else - node = arg; + default: + retval = 1; + goto exit; + } } + + node = argv[optind]; if (!node) { - err("no node specified"); - fprintf(stderr, help); + err(udev_ctx, "no device\n"); + fprintf(stderr, "no device\n"); rc = 1; goto exit; } - vid = volume_id_open_node(node); + fd = open(node, O_RDONLY); + if (fd < 0) { + fprintf(stderr, "%s: error opening volume\n", node); + rc = 2; + goto exit; + } + + vid = volume_id_open_fd(fd); if (vid == NULL) { - fprintf(stderr, "%s: error open volume\n", node); rc = 2; goto exit; } - if (ioctl(vid->fd, BLKGETSIZE64, &size) != 0) - size = 0; - dbg("BLKGETSIZE64=%llu", size); + if (size == 0) { + if (ioctl(fd, BLKGETSIZE64, &size) == 0) { + info(udev_ctx, "BLKGETSIZE64=%" PRIu64 " (%" PRIu64 "GB)\n", size, size >> 30); + } else { + struct stat statbuf; + + if (fstat(fd, &statbuf) == 0 && S_ISREG(statbuf.st_mode)) + size = statbuf.st_size; + else + size = 0; + info(udev_ctx, "stat=%" PRIu64 " (%" PRIu64 "GB)\n", size, size >> 30); + } + } /* try to drop all privileges before reading disk content */ - pw = getpwnam ("nobody"); - if (pw != NULL && pw->pw_uid > 0 && pw->pw_gid > 0) { - dbg("dropping privileges to %u:%u", (unsigned int)pw->pw_uid, (unsigned int)pw->pw_gid); - if (setgroups(0, NULL) != 0 || - setgid(pw->pw_gid) != 0 || - setuid(pw->pw_uid) != 0) { - fprintf(stderr, "error dropping privileges: %s\n", strerror(errno)); - rc = 3; - goto exit; + if (getuid() == 0) { + struct passwd *pw; + + pw = getpwnam("nobody"); + if (pw != NULL && pw->pw_uid > 0 && pw->pw_gid > 0) { + if (setgroups(0, NULL) != 0 || + setgid(pw->pw_gid) != 0 || + setuid(pw->pw_uid) != 0) + info(udev_ctx, "unable to drop privileges: %s\n\n", strerror(errno)); } } if (probe_all) { - if (volume_id_probe_linux_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_intel_software_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_lsi_mega_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_via_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_silicon_medley_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_nvidia_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_promise_fasttrack_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_highpoint_45x_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_adaptec_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_jmicron_raid(vid, 0, size) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_vfat(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_linux_swap(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_luks(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_xfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_ext(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_reiserfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_jfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_udf(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_iso9660(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_hfs_hfsplus(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_ufs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_ntfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_cramfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_romfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_hpfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_sysv(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_minix(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_ocfs1(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_ocfs2(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_vxfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_squashfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_netware(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_gfs(vid, 0, 0) == 0) - printf("%s\n", vid->type); - if (volume_id_probe_gfs2(vid, 0, 0) == 0) - printf("%s\n", vid->type); - + volume_id_all_probers(all_probers, vid, offset, size, NULL); goto exit; } if (skip_raid) - retval = volume_id_probe_filesystem(vid, 0, size); + retval = volume_id_probe_filesystem(vid, offset, size); else - retval = volume_id_probe_all(vid, 0, size); + retval = volume_id_probe_all(vid, offset, size); if (retval != 0) { - fprintf(stderr, "%s: unknown volume type\n", node); + fprintf(stderr, "unknown or non-unique volume type " + "(--probe-all lists possibly conflicting types)\n"); rc = 4; goto exit; } - set_str(name, vid->label, sizeof(vid->label)); - replace_untrusted_chars(name); + if (!volume_id_get_label(vid, &label) || + !volume_id_get_usage(vid, &usage) || + !volume_id_get_type(vid, &type) || + !volume_id_get_type_version(vid, &type_version) || + !volume_id_get_uuid(vid, &uuid) || + !volume_id_get_uuid_sub(vid, &uuid_sub)) { + rc = 4; + goto exit; + } + + udev_util_replace_whitespace(label, label_safe, sizeof(label_safe)); + udev_util_replace_chars(label_safe, UDEV_ALLOWED_CHARS_INPUT); + volume_id_encode_string(label, label_enc, sizeof(label_enc)); + + udev_util_replace_whitespace(uuid, uuid_safe, sizeof(uuid_safe)); + udev_util_replace_chars(uuid_safe, UDEV_ALLOWED_CHARS_INPUT); + volume_id_encode_string(uuid, uuid_enc, sizeof(uuid_enc)); + + volume_id_encode_string(uuid_sub, uuid_sub_enc, sizeof(uuid_sub_enc)); + + volume_id_encode_string(type, type_enc, sizeof(type_enc)); + volume_id_encode_string(type_version, type_version_enc, sizeof(type_version_enc)); switch (print) { case PRINT_EXPORT: - printf("ID_FS_USAGE=%s\n", vid->usage); - printf("ID_FS_TYPE=%s\n", vid->type); - printf("ID_FS_VERSION=%s\n", vid->type_version); - printf("ID_FS_UUID=%s\n", vid->uuid); - printf("ID_FS_LABEL=%s\n", vid->label); - printf("ID_FS_LABEL_SAFE=%s\n", name); + printf("ID_FS_USAGE=%s\n", usage); + printf("ID_FS_TYPE=%s\n", type_enc); + printf("ID_FS_VERSION=%s\n", type_version_enc); + printf("ID_FS_UUID=%s\n", uuid_safe); + printf("ID_FS_UUID_ENC=%s\n", uuid_enc); + if (uuid_sub_enc[0] != '\0') + printf("ID_FS_UUID_SUB_ENC=%s\n", uuid_sub_enc); + printf("ID_FS_LABEL=%s\n", label_safe); + printf("ID_FS_LABEL_ENC=%s\n", label_enc); break; case PRINT_TYPE: - printf("%s\n", vid->type); + printf("%s\n", type); break; case PRINT_LABEL: - if (name[0] == '\0' || vid->usage_id == VOLUME_ID_RAID) { + if (label_safe[0] == '\0' || strcmp(usage, "raid") == 0) { rc = 3; goto exit; } - printf("%s\n", name); + printf("%s\n", label_safe); break; case PRINT_UUID: - if (vid->uuid[0] == '\0' || vid->usage_id == VOLUME_ID_RAID) { + if (uuid_enc[0] == '\0' || strcmp(usage, "raid") == 0) { rc = 4; goto exit; } - printf("%s\n", vid->uuid); + printf("%s\n", uuid_enc); + break; + case PRINT_LABEL_RAW: + if (label[0] == '\0' || strcmp(usage, "raid") == 0) { + rc = 3; + goto exit; + } + printf("%s\n", label); break; } exit: if (vid != NULL) volume_id_close(vid); - + udev_unref(udev_ctx); logging_close(); return rc; }