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 "../../udev/udev.h"
  27
  28 #ifdef USE_LOG
  29 void log_message(int priority, const char *format, ...)
  30 {
  31         va_list args;
  32         static int udev_log = -1;
  33
  34         if (udev_log == -1) {
  35                 const char *value;
  36
  37                 value = getenv("UDEV_LOG");
  38                 if (value)
  39                         udev_log = log_priority(value);
  40                 else
  41                         udev_log = LOG_ERR;
  42         }
  43
  44         if (priority > udev_log)
  45                 return;
  46
  47         va_start(args, format);
  48         vsyslog(priority, format, args);
  49         va_end(args);
  50 }
  51 #endif
  52
  53 int main(int argc, char *argv[])
  54 {
  55         const char *node = NULL;
  56         int i;
  57         int export = 0;
  58         uint32_t disk_id;
  59         uint16_t mbr_valid;
  60         struct dirent *dent;
  61         int disk_fd;
  62         int sysfs_fd;
  63         DIR *dir = NULL;
  64         int rc = 1;
  65         char match[NAME_MAX] = "";
  66
  67         logging_init("edd_id");
  68
  69         for (i = 1 ; i < argc; i++) {
  70                 char *arg = argv[i];
  71
  72                 if (strcmp(arg, "--export") == 0) {
  73                         export = 1;
  74                 } else
  75                         node = arg;
  76         }
  77         if (node == NULL) {
  78                 err("no node specified\n");
  79                 fprintf(stderr, "no node specified\n");
  80                 goto exit;
  81         }
  82
  83         /* check for kernel support */
  84         dir = opendir("/sys/firmware/edd");
  85         if (dir == NULL) {
  86                 info("no kernel EDD support\n");
  87                 fprintf(stderr, "no kernel EDD support\n");
  88                 rc = 2;
  89                 goto exit;
  90         }
  91
  92         disk_fd = open(node, O_RDONLY);
  93         if (disk_fd < 0) {
  94                 info("unable to open '%s'\n", node);
  95                 fprintf(stderr, "unable to open '%s'\n", node);
  96                 rc = 3;
  97                 goto closedir;
  98         }
  99
 100         /* check for valid MBR signature */
 101         if (lseek(disk_fd, 510, SEEK_SET) < 0) {
 102                 info("seek to MBR validity failed '%s'\n", node);
 103                 rc = 4;
 104                 goto close;
 105         }
 106         if (read(disk_fd, &mbr_valid, sizeof(mbr_valid)) != sizeof(mbr_valid)) {
 107                 info("read MBR validity failed '%s'\n", node);
 108                 rc = 5;
 109                 goto close;
 110         }
 111         if (mbr_valid != 0xAA55) {
 112                 fprintf(stderr, "no valid MBR signature '%s'\n", node);
 113                 info("no valid MBR signature '%s'\n", node);
 114                 rc=6;
 115                 goto close;
 116         }
 117
 118         /* read EDD signature */
 119         if (lseek(disk_fd, 440, SEEK_SET) < 0) {
 120                 info("seek to signature failed '%s'\n", node);
 121                 rc = 7;
 122                 goto close;
 123         }
 124         if (read(disk_fd, &disk_id, sizeof(disk_id)) != sizeof(disk_id)) {
 125                 info("read signature failed '%s'\n", node);
 126                 rc = 8;
 127                 goto close;
 128         }
 129         /* all zero is invalid */
 130         info("read id 0x%08x from '%s'\n", disk_id, node);
 131         if (disk_id == 0) {
 132                 fprintf(stderr, "no EDD signature '%s'\n", node);
 133                 info("'%s' signature is zero\n", node);
 134                 rc = 9;
 135                 goto close;
 136         }
 137
 138         /* lookup signature in sysfs to determine the name */
 139         for (dent = readdir(dir); dent != NULL; dent = readdir(dir)) {
 140                 char file[PATH_SIZE];
 141                 char sysfs_id_buf[256];
 142                 uint32_t sysfs_id;
 143                 ssize_t size;
 144
 145                 if (dent->d_name[0] == '.')
 146                         continue;
 147
 148                 snprintf(file, sizeof(file), "/sys/firmware/edd/%s/mbr_signature", dent->d_name);
 149                 file[sizeof(file)-1] = '\0';
 150
 151                 sysfs_fd = open(file, O_RDONLY);
 152                 if (sysfs_fd < 0) {
 153                         info("unable to open sysfs '%s'\n", file);
 154                         continue;
 155                 }
 156
 157                 size = read(sysfs_fd, sysfs_id_buf, sizeof(sysfs_id_buf)-1);
 158                 close(sysfs_fd);
 159                 if (size <= 0) {
 160                         info("read sysfs '%s' failed\n", file);
 161                         continue;
 162                 }
 163                 sysfs_id_buf[size] = '\0';
 164                 info("read '%s' from '%s'\n", sysfs_id_buf, file);
 165                 sysfs_id = strtoul(sysfs_id_buf, NULL, 16);
 166
 167                 /* look for matching value, that appears only once */
 168                 if (disk_id == sysfs_id) {
 169                         if (match[0] == '\0') {
 170                                 /* store id */
 171                                 strlcpy(match, dent->d_name, sizeof(match));
 172                         } else {
 173                                 /* error, same signature for another device */
 174                                 info("'%s' does not have a unique signature\n", node);
 175                                 fprintf(stderr, "'%s' does not have a unique signature\n", node);
 176                                 rc = 10;
 177                                 goto exit;
 178                         }
 179                 }
 180         }
 181
 182         if (match[0] != '\0') {
 183                 if (export)
 184                         printf("ID_EDD=%s\n", match);
 185                 else
 186                         printf("%s\n", match);
 187                 rc = 0;
 188         }
 189
 190 close:
 191         close(disk_fd);
 192 closedir:
 193         closedir(dir);
 194 exit:
 195         logging_close();
 196         return rc;
 197 }