chiark / gitweb /
make: do not delete autotools generated file with distclean
[elogind.git] / extras / scsi_id / scsi_id.c
index 4f0233ed0d6a9a7b44a31ec15378dbe57f99b0d8..21b7cbeaec2a36f128efce8e0011199aa25f3e87 100644 (file)
 #include <getopt.h>
 #include <sys/stat.h>
 
-#include "../../udev.h"
+#include "../../udev/udev.h"
 #include "scsi_id.h"
-#include "scsi_id_version.h"
 
 static const struct option options[] = {
-       { "device", 1, NULL, 'd' },
-       { "config", 1, NULL, 'f' },
-       { "page", 1, NULL, 'p' },
-       { "blacklisted", 0, NULL, 'b' },
-       { "whitelisted", 0, NULL, 'g' },
-       { "prefix-bus-id", 0, NULL, 'i' },
-       { "replace-whitespace", 0, NULL, 'u' },
-       { "verbose", 0, NULL, 'v' },
-       { "version", 0, NULL, 'V' },
-       { "export", 0, NULL, 'x' },
-       { "help", 0, NULL, 'h' },
+       { "device", required_argument, NULL, 'd' },
+       { "config", required_argument, NULL, 'f' },
+       { "page", required_argument, NULL, 'p' },
+       { "blacklisted", no_argument, NULL, 'b' },
+       { "whitelisted", no_argument, NULL, 'g' },
+       { "replace-whitespace", no_argument, NULL, 'u' },
+       { "sg-version", required_argument, NULL, 's' },
+       { "verbose", no_argument, NULL, 'v' },
+       { "version", no_argument, NULL, 'V' },
+       { "export", no_argument, NULL, 'x' },
+       { "help", no_argument, NULL, 'h' },
        {}
 };
 
@@ -51,9 +50,9 @@ static const char dev_short_options[] = "bgp:";
 
 static int all_good;
 static int dev_specified;
-static char config_file[MAX_PATH_LEN] = SCSI_ID_CONFIG_FILE;
-static int display_bus_id;
+static char config_file[MAX_PATH_LEN] = SYSCONFDIR "/scsi_id.config";
 static enum page_code default_page_code;
+static int sg_version = 4;
 static int use_stderr;
 static int debug;
 static int reformat_serial;
@@ -63,64 +62,14 @@ static char model_str[64];
 static char revision_str[16];
 static char type_str[16];
 
-#ifdef USE_LOG
-void log_message(int priority, const char *format, ...)
+static void log_fn(struct udev *udev, int priority,
+                  const char *file, int line, const char *fn,
+                  const char *format, va_list args)
 {
-       va_list args;
-       static int udev_log = -1;
-
-       if (udev_log == -1) {
-               const char *value;
-
-               value = getenv("UDEV_LOG");
-               if (value)
-                       udev_log = log_priority(value);
-               else
-                       udev_log = LOG_ERR;
-       }
-
-       if (priority > udev_log)
-               return;
-
-       va_start(args, format);
        vsyslog(priority, format, args);
-       va_end(args);
-}
-#endif
-
-static void set_str(char *to, const char *from, size_t count)
-{
-       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';
 }
 
-static void set_type(char *to, const char *from, size_t len)
+static void set_type(const char *from, char *to, size_t len)
 {
        int type_num;
        char *eptr;
@@ -154,8 +103,7 @@ static void set_type(char *to, const char *from, size_t len)
                        break;
                }
        }
-       strncpy(to, type, len);
-       to[len-1] = '\0';
+       util_strlcpy(to, type, len);
 }
 
 /*
@@ -214,7 +162,8 @@ static int argc_count(char *opts)
  *
  * vendor and model can end in '\n'.
  */
-static int get_file_options(const char *vendor, const char *model,
+static int get_file_options(struct udev *udev,
+                           const char *vendor, const char *model,
                            int *argc, char ***newargv)
 {
        char *buffer;
@@ -226,14 +175,14 @@ static int get_file_options(const char *vendor, const char *model,
        int c;
        int retval = 0;
 
-       dbg("vendor='%s'; model='%s'\n", vendor, model);
+       dbg(udev, "vendor='%s'; model='%s'\n", vendor, model);
        fd = fopen(config_file, "r");
        if (fd == NULL) {
-               dbg("can't open %s\n", config_file);
+               dbg(udev, "can't open %s\n", config_file);
                if (errno == ENOENT) {
                        return 1;
                } else {
-                       err("can't open %s: %s\n", config_file, strerror(errno));
+                       err(udev, "can't open %s: %s\n", config_file, strerror(errno));
                        return -1;
                }
        }
@@ -245,7 +194,7 @@ static int get_file_options(const char *vendor, const char *model,
         */
        buffer = malloc(MAX_BUFFER_LEN);
        if (!buffer) {
-               err("can't allocate memory\n");
+               err(udev, "can't allocate memory\n");
                return -1;
        }
 
@@ -259,7 +208,7 @@ static int get_file_options(const char *vendor, const char *model,
                        break;
                lineno++;
                if (buf[strlen(buffer) - 1] != '\n') {
-                       info("Config file line %d too long\n", lineno);
+                       err(udev, "Config file line %d too long\n", lineno);
                        break;
                }
 
@@ -274,7 +223,7 @@ static int get_file_options(const char *vendor, const char *model,
                if (*buf == '#')
                        continue;
 
-               dbg("lineno %d: '%s'\n", lineno, buf);
+               dbg(udev, "lineno %d: '%s'\n", lineno, buf);
                str1 = strsep(&buf, "=");
                if (str1 && strcasecmp(str1, "VENDOR") == 0) {
                        str1 = get_value(&buf);
@@ -304,20 +253,20 @@ static int get_file_options(const char *vendor, const char *model,
                        }
                        options_in = str1;
                }
-               dbg("config file line %d:\n"
+               dbg(udev, "config file line %d:\n"
                        " vendor '%s'; model '%s'; options '%s'\n",
                        lineno, vendor_in, model_in, options_in);
                /*
                 * Only allow: [vendor=foo[,model=bar]]options=stuff
                 */
                if (!options_in || (!vendor_in && model_in)) {
-                       info("Error parsing config file line %d '%s'\n", lineno, buffer);
+                       err(udev, "Error parsing config file line %d '%s'\n", lineno, buffer);
                        retval = -1;
                        break;
                }
                if (vendor == NULL) {
                        if (vendor_in == NULL) {
-                               dbg("matched global option\n");
+                               dbg(udev, "matched global option\n");
                                break;
                        }
                } else if ((vendor_in && strncmp(vendor, vendor_in,
@@ -331,10 +280,10 @@ static int get_file_options(const char *vendor, const char *model,
                                 * give a partial match (that is FOO
                                 * matches FOOBAR).
                                 */
-                               dbg("matched vendor/model\n");
+                               dbg(udev, "matched vendor/model\n");
                                break;
                } else {
-                       dbg("no match\n");
+                       dbg(udev, "no match\n");
                }
        }
 
@@ -349,7 +298,7 @@ static int get_file_options(const char *vendor, const char *model,
                        c = argc_count(buffer) + 2;
                        *newargv = calloc(c, sizeof(**newargv));
                        if (!*newargv) {
-                               err("can't allocate memory\n");
+                               err(udev, "can't allocate memory\n");
                                retval = -1;
                        } else {
                                *argc = c;
@@ -374,7 +323,8 @@ static int get_file_options(const char *vendor, const char *model,
        return retval;
 }
 
-static int set_options(int argc, char **argv, const char *short_opts,
+static int set_options(struct udev *udev,
+                      int argc, char **argv, const char *short_opts,
                       char *maj_min_dev)
 {
        int option;
@@ -391,9 +341,9 @@ static int set_options(int argc, char **argv, const char *short_opts,
                        break;
 
                if (optarg)
-                       dbg("option '%c' arg '%s'\n", option, optarg);
+                       dbg(udev, "option '%c' arg '%s'\n", option, optarg);
                else
-                       dbg("option '%c'\n", option);
+                       dbg(udev, "option '%c'\n", option);
 
                switch (option) {
                case 'b':
@@ -402,8 +352,7 @@ static int set_options(int argc, char **argv, const char *short_opts,
 
                case 'd':
                        dev_specified = 1;
-                       strncpy(maj_min_dev, optarg, MAX_PATH_LEN);
-                       maj_min_dev[MAX_PATH_LEN-1] = '\0';
+                       util_strlcpy(maj_min_dev, optarg, MAX_PATH_LEN);
                        break;
 
                case 'e':
@@ -411,8 +360,7 @@ static int set_options(int argc, char **argv, const char *short_opts,
                        break;
 
                case 'f':
-                       strncpy(config_file, optarg, MAX_PATH_LEN);
-                       config_file[MAX_PATH_LEN-1] = '\0';
+                       util_strlcpy(config_file, optarg, MAX_PATH_LEN);
                        break;
 
                case 'g':
@@ -421,23 +369,19 @@ static int set_options(int argc, char **argv, const char *short_opts,
 
                case 'h':
                        printf("Usage: scsi_id OPTIONS <device>\n"
-                              "  --device               device node for SG_IO commands\n"
-                              "  --config               location of config file\n"
-                              "  --page                 SCSI page (0x80, 0x83, pre-spc3-83)\n"
-                              "  --blacklisted          threat device as blacklisted\n"
-                              "  --whitelisted          threat device as whitelisted\n"
-                              "  --prefix-bus-id        prefix SCSI bus id\n"
-                              "  --replace-whitespace   replace all whitespaces by underscores\n"
-                              "  --verbose              verbose logging\n"
-                              "  --version              print version\n"
-                              "  --export               print values as environment keys\n"
-                              "  --help                 print this help text\n\n");
+                              "  --device=                     device node for SG_IO commands\n"
+                              "  --config=                     location of config file\n"
+                              "  --page=0x80|0x83|pre-spc3-83  SCSI page (0x80, 0x83, pre-spc3-83)\n"
+                              "  --sg-version=3|4              use SGv3 or SGv4\n"
+                              "  --blacklisted                 threat device as blacklisted\n"
+                              "  --whitelisted                 threat device as whitelisted\n"
+                              "  --replace-whitespace          replace all whitespaces by underscores\n"
+                              "  --verbose                     verbose logging\n"
+                              "  --version                     print version\n"
+                              "  --export                      print values as environment keys\n"
+                              "  --help                        print this help text\n\n");
                        exit(0);
 
-               case 'i':
-                       display_bus_id = 1;
-                       break;
-
                case 'p':
                        if (strcmp(optarg, "0x80") == 0) {
                                default_page_code = PAGE_80;
@@ -446,7 +390,15 @@ static int set_options(int argc, char **argv, const char *short_opts,
                        } else if (strcmp(optarg, "pre-spc3-83") == 0) {
                                default_page_code = PAGE_83_PRE_SPC3; 
                        } else {
-                               info("Unknown page code '%s'\n", optarg);
+                               err(udev, "Unknown page code '%s'\n", optarg);
+                               return -1;
+                       }
+                       break;
+
+               case 's':
+                       sg_version = atoi(optarg);
+                       if (sg_version < 3 || sg_version > 4) {
+                               err(udev, "Unknown SG version '%s'\n", optarg);
                                return -1;
                        }
                        break;
@@ -464,7 +416,7 @@ static int set_options(int argc, char **argv, const char *short_opts,
                        break;
 
                case 'V':
-                       printf("%s\n", SCSI_ID_VERSION);
+                       printf("%s\n", VERSION);
                        exit(0);
                        break;
 
@@ -474,13 +426,13 @@ static int set_options(int argc, char **argv, const char *short_opts,
        }
        if (optind < argc && !dev_specified) {
                dev_specified = 1;
-               strncpy(maj_min_dev, argv[optind], MAX_PATH_LEN);
-               maj_min_dev[MAX_PATH_LEN-1] = '\0';
+               util_strlcpy(maj_min_dev, argv[optind], MAX_PATH_LEN);
        }
        return 0;
 }
 
-static int per_dev_options(struct scsi_id_device *dev_scsi, int *good_bad, int *page_code)
+static int per_dev_options(struct udev *udev,
+                          struct scsi_id_device *dev_scsi, int *good_bad, int *page_code)
 {
        int retval;
        int newargc;
@@ -490,7 +442,7 @@ static int per_dev_options(struct scsi_id_device *dev_scsi, int *good_bad, int *
        *good_bad = all_good;
        *page_code = default_page_code;
 
-       retval = get_file_options(vendor_str, model_str, &newargc, &newargv);
+       retval = get_file_options(udev, vendor_str, model_str, &newargc, &newargv);
 
        optind = 1; /* reset this global extern */
        while (retval == 0) {
@@ -499,9 +451,9 @@ static int per_dev_options(struct scsi_id_device *dev_scsi, int *good_bad, int *
                        break;
 
                if (optarg)
-                       dbg("option '%c' arg '%s'\n", option, optarg);
+                       dbg(udev, "option '%c' arg '%s'\n", option, optarg);
                else
-                       dbg("option '%c'\n", option);
+                       dbg(udev, "option '%c'\n", option);
 
                switch (option) {
                case 'b':
@@ -520,13 +472,13 @@ static int per_dev_options(struct scsi_id_device *dev_scsi, int *good_bad, int *
                        } else if (strcmp(optarg, "pre-spc3-83") == 0) {
                                *page_code = PAGE_83_PRE_SPC3; 
                        } else {
-                               info("Unknown page code '%s'\n", optarg);
+                               err(udev, "Unknown page code '%s'\n", optarg);
                                retval = -1;
                        }
                        break;
 
                default:
-                       info("Unknown or bad option '%c' (0x%x)\n", option, option);
+                       err(udev, "Unknown or bad option '%c' (0x%x)\n", option, option);
                        retval = -1;
                        break;
                }
@@ -539,19 +491,23 @@ static int per_dev_options(struct scsi_id_device *dev_scsi, int *good_bad, int *
        return retval;
 }
 
-static int set_inq_values(struct scsi_id_device *dev_scsi, const char *path)
+static int set_inq_values(struct udev *udev, struct scsi_id_device *dev_scsi, const char *path)
 {
        int retval;
 
-       retval = scsi_std_inquiry(dev_scsi, path);
-       if (retval)
-           return retval;
-
-       set_str(vendor_str, dev_scsi->vendor, 8);
-       set_str(model_str, dev_scsi->model, 16);
-       set_type(type_str, dev_scsi->type, sizeof(type_str) - 1);
-       set_str(revision_str, dev_scsi->revision, sizeof(revision_str) -1);
+       dev_scsi->use_sg = sg_version;
 
+       retval = scsi_std_inquiry(udev, dev_scsi, path);
+       if (retval)
+               return retval;
+
+       udev_util_replace_whitespace(dev_scsi->vendor, vendor_str, sizeof(vendor_str));
+       udev_util_replace_chars(vendor_str, NULL);
+       udev_util_replace_whitespace(dev_scsi->model, model_str, sizeof(model_str));
+       udev_util_replace_chars(model_str, NULL);
+       set_type(dev_scsi->type, type_str, sizeof(type_str));
+       udev_util_replace_whitespace(dev_scsi->revision, revision_str, sizeof(revision_str));
+       udev_util_replace_chars(revision_str, NULL);
        return 0;
 }
 
@@ -582,78 +538,88 @@ static void format_serial(char *serial)
 
 /*
  * scsi_id: try to get an id, if one is found, printf it to stdout.
- * returns a value passed to exit() - 0 if printed an id, else 1. This
- * could be expanded, for example, if we want to report a failure like no
- * memory etc. return 2, and return 1 for expected cases (like broken
- * device found) that do not print an id.
+ * returns a value passed to exit() - 0 if printed an id, else 1.
  */
-static int scsi_id(char *maj_min_dev)
+static int scsi_id(struct udev *udev, char *maj_min_dev)
 {
-       int retval;
        struct scsi_id_device dev_scsi;
        int good_dev;
        int page_code;
-       char serial_short[MAX_SERIAL_LEN] = "";
+       int retval = 0;
 
-       set_inq_values(&dev_scsi, maj_min_dev);
+       memset(&dev_scsi, 0x00, sizeof(struct scsi_id_device));
 
-       /* get per device (vendor + model) options from the config file */
-       retval = per_dev_options(&dev_scsi, &good_dev, &page_code);
-       dbg("per dev options: good %d; page code 0x%x\n", good_dev, page_code);
+       if (set_inq_values(udev, &dev_scsi, maj_min_dev) < 0) {
+               retval = 1;
+               goto out;
+       }
 
+       /* get per device (vendor + model) options from the config file */
+       per_dev_options(udev, &dev_scsi, &good_dev, &page_code);
+       dbg(udev, "per dev options: good %d; page code 0x%x\n", good_dev, page_code);
        if (!good_dev) {
                retval = 1;
-       } else if (scsi_get_serial(&dev_scsi, maj_min_dev, page_code,
-                                  serial_short, MAX_SERIAL_LEN)) {
-               retval = 1;
-       } else {
-               retval = 0;
+               goto out;
        }
-       if (!retval) {
-               if (export) {
-                       char serial_str[MAX_SERIAL_LEN];
-
-                       printf("ID_VENDOR=%s\n", vendor_str);
-                       printf("ID_MODEL=%s\n", model_str);
-                       printf("ID_REVISION=%s\n", revision_str);
-                       set_str(serial_str, dev_scsi.serial, sizeof(serial_str));
+
+       /* read serial number from mode pages (no values for optical drives) */
+       scsi_get_serial(udev, &dev_scsi, maj_min_dev, page_code, MAX_SERIAL_LEN);
+
+       if (export) {
+               char serial_str[MAX_SERIAL_LEN];
+
+               printf("ID_VENDOR=%s\n", vendor_str);
+               printf("ID_MODEL=%s\n", model_str);
+               printf("ID_REVISION=%s\n", revision_str);
+               printf("ID_TYPE=%s\n", type_str);
+               if (dev_scsi.serial[0] != '\0') {
+                       udev_util_replace_whitespace(dev_scsi.serial, serial_str, sizeof(serial_str));
+                       udev_util_replace_chars(serial_str, NULL);
                        printf("ID_SERIAL=%s\n", serial_str);
-                       set_str(serial_str, serial_short, sizeof(serial_str));
+                       udev_util_replace_whitespace(dev_scsi.serial_short, serial_str, sizeof(serial_str));
+                       udev_util_replace_chars(serial_str, NULL);
                        printf("ID_SERIAL_SHORT=%s\n", serial_str);
-                       printf("ID_TYPE=%s\n", type_str);
-               } else {
-                       if (reformat_serial)
-                               format_serial(dev_scsi.serial);
-                       printf("%s\n", dev_scsi.serial);
                }
-               dbg("%s\n", dev_scsi.serial);
-               retval = 0;
+               goto out;
        }
 
+       if (dev_scsi.serial[0] == '\0') {
+               retval = 1;
+               goto out;
+       }
+       if (reformat_serial)
+               format_serial(dev_scsi.serial);
+       printf("%s\n", dev_scsi.serial);
+out:
        return retval;
 }
 
 int main(int argc, char **argv)
 {
+       struct udev *udev;
        int retval = 0;
        char maj_min_dev[MAX_PATH_LEN];
        int newargc;
        char **newargv;
 
+       udev = udev_new();
+       if (udev == NULL)
+               goto exit;
+
        logging_init("scsi_id");
-       dbg("argc is %d\n", argc);
+       udev_set_log_fn(udev, log_fn);
 
        /*
         * Get config file options.
         */
        newargv = NULL;
-       retval = get_file_options(NULL, NULL, &newargc, &newargv);
+       retval = get_file_options(udev, NULL, NULL, &newargc, &newargv);
        if (retval < 0) {
                retval = 1;
                goto exit;
        }
        if (newargv && (retval == 0)) {
-               if (set_options(newargc, newargv, short_options, maj_min_dev) < 0) {
+               if (set_options(udev, newargc, newargv, short_options, maj_min_dev) < 0) {
                        retval = 2;
                        goto exit;
                }
@@ -663,18 +629,19 @@ int main(int argc, char **argv)
        /*
         * Get command line options (overriding any config file settings).
         */
-       if (set_options(argc, argv, short_options, maj_min_dev) < 0)
+       if (set_options(udev, argc, argv, short_options, maj_min_dev) < 0)
                exit(1);
 
        if (!dev_specified) {
-               info("no device specified\n");
+               err(udev, "no device specified\n");
                retval = 1;
                goto exit;
        }
 
-       retval = scsi_id(maj_min_dev);
+       retval = scsi_id(udev, maj_min_dev);
 
 exit:
+       udev_unref(udev);
        logging_close();
        return retval;
 }