extras/edd_id/edd_id.c

   1 /*
   2  * edd_id - naming of BIOS disk devices via EDD
   3  *
   4  * Copyright (C) 2005 John Hull <John_Hull@Dell.com>
   5  * Copyright (C) 2005 Kay Sievers <kay.sievers@vrfy.org>
   6  *
   7  *      This program is free software; you can redistribute it and/or modify it
   8  *      under the terms of the GNU General Public License as published by the
   9  *      Free Software Foundation version 2 of the License.
  10  */
  11
  12 #ifndef _GNU_SOURCE
  13 #define _GNU_SOURCE 1
  14 #endif
  15
  16 #include <stdio.h>
  17 #include <stdlib.h>
  18 #include <unistd.h>
  19 #include <fcntl.h>
  20 #include <ctype.h>
  21 #include <string.h>
  22 #include <errno.h>
  23 #include <dirent.h>
  24 #include <stdint.h>
  25
  26 #include "libudev.h"
  27 #include "libudev-private.h"
  28
  29 static void log_fn(struct udev *udev, int priority,
  30                    const char *file, int line, const char *fn,
  31                    const char *format, va_list args)
  32 {
  33         vsyslog(priority, format, args);
  34 }
  35
  36 int main(int argc, char *argv[])
  37 {
  38         struct udev *udev;
  39         const char *node = NULL;
  40         int i;
  41         int export = 0;
  42         uint32_t disk_id;
  43         uint16_t mbr_valid;
  44         struct dirent *dent;
  45         int disk_fd;
  46         int sysfs_fd;
  47         DIR *dir = NULL;
  48         int rc = 1;
  49         char filename[UTIL_PATH_SIZE];
  50         char match[UTIL_PATH_SIZE];
  51
  52         udev = udev_new();
  53         if (udev == NULL)
  54                 goto exit;
  55
  56         udev_log_init("edd_id");
  57         udev_set_log_fn(udev, log_fn);
  58
  59         for (i = 1 ; i < argc; i++) {
  60                 char *arg = argv[i];
  61
  62                 if (strcmp(arg, "--export") == 0) {
  63                         export = 1;
  64                 } else
  65                         node = arg;
  66         }
  67         if (node == NULL) {
  68                 err(udev, "no node specified\n");
  69                 fprintf(stderr, "no node specified\n");
  70                 goto exit;
  71         }
  72
  73         /* check for kernel support */
  74         util_strscpyl(filename, sizeof(filename), udev_get_sys_path(udev), "/firmware/edd", NULL);
  75         dir = opendir(filename);
  76         if (dir == NULL) {
  77                 info(udev, "no kernel EDD support\n");
  78                 fprintf(stderr, "no kernel EDD support\n");
  79                 rc = 2;
  80                 goto exit;
  81         }
  82
  83         disk_fd = open(node, O_RDONLY);
  84         if (disk_fd < 0) {
  85                 info(udev, "unable to open '%s'\n", node);
  86                 fprintf(stderr, "unable to open '%s'\n", node);
  87                 rc = 3;
  88                 goto closedir;
  89         }
  90
  91         /* check for valid MBR signature */
  92         if (lseek(disk_fd, 510, SEEK_SET) < 0) {
  93                 info(udev, "seek to MBR validity failed '%s'\n", node);
  94                 rc = 4;
  95                 goto close;
  96         }
  97         if (read(disk_fd, &mbr_valid, sizeof(mbr_valid)) != sizeof(mbr_valid)) {
  98                 info(udev, "read MBR validity failed '%s'\n", node);
  99                 rc = 5;
 100                 goto close;
 101         }
 102         if (mbr_valid != 0xAA55) {
 103                 fprintf(stderr, "no valid MBR signature '%s'\n", node);
 104                 info(udev, "no valid MBR signature '%s'\n", node);
 105                 rc=6;
 106                 goto close;
 107         }
 108
 109         /* read EDD signature */
 110         if (lseek(disk_fd, 440, SEEK_SET) < 0) {
 111                 info(udev, "seek to signature failed '%s'\n", node);
 112                 rc = 7;
 113                 goto close;
 114         }
 115         if (read(disk_fd, &disk_id, sizeof(disk_id)) != sizeof(disk_id)) {
 116                 info(udev, "read signature failed '%s'\n", node);
 117                 rc = 8;
 118                 goto close;
 119         }
 120         /* all zero is invalid */
 121         info(udev, "read id 0x%08x from '%s'\n", disk_id, node);
 122         if (disk_id == 0) {
 123                 fprintf(stderr, "no EDD signature '%s'\n", node);
 124                 info(udev, "'%s' signature is zero\n", node);
 125                 rc = 9;
 126                 goto close;
 127         }
 128
 129         /* lookup signature in sysfs to determine the name */
 130         match[0] = '\0';
 131         for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
 132                 char sysfs_id_buf[256];
 133                 uint32_t sysfs_id;
 134                 ssize_t size;
 135
 136                 if (dent->d_name[0] == '.')
 137                         continue;
 138
 139                 util_strscpyl(filename, sizeof(filename), dent->d_name, "/mbr_signature", NULL);
 140                 sysfs_fd = openat(dirfd(dir), filename, O_RDONLY);
 141                 if (sysfs_fd < 0) {
 142                         info(udev, "unable to open sysfs '%s'\n", filename);
 143                         continue;
 144                 }
 145
 146                 size = read(sysfs_fd, sysfs_id_buf, sizeof(sysfs_id_buf)-1);
 147                 close(sysfs_fd);
 148                 if (size <= 0) {
 149                         info(udev, "read sysfs '%s' failed\n", filename);
 150                         continue;
 151                 }
 152                 sysfs_id_buf[size] = '\0';
 153                 info(udev, "read '%s' from '%s'\n", sysfs_id_buf, filename);
 154                 sysfs_id = strtoul(sysfs_id_buf, NULL, 16);
 155
 156                 /* look for matching value, that appears only once */
 157                 if (disk_id == sysfs_id) {
 158                         if (match[0] == '\0') {
 159                                 /* store id */
 160                                 util_strscpy(match, sizeof(match), dent->d_name);
 161                         } else {
 162                                 /* error, same signature for another device */
 163                                 info(udev, "'%s' does not have a unique signature\n", node);
 164                                 fprintf(stderr, "'%s' does not have a unique signature\n", node);
 165                                 rc = 10;
 166                                 goto exit;
 167                         }
 168                 }
 169         }
 170
 171         if (match[0] != '\0') {
 172                 if (export)
 173                         printf("ID_EDD=%s\n", match);
 174                 else
 175                         printf("%s\n", match);
 176                 rc = 0;
 177         }
 178
 179 close:
 180         close(disk_fd);
 181 closedir:
 182         closedir(dir);
 183 exit:
 184         udev_unref(udev);
 185         udev_log_close();
 186         return rc;
 187 }