/***
This file is part of systemd.
Copyright 2012 Kay Sievers <kay@vrfy.org>
systemd is free software; you can redistribute it and/or modify it
under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation; either version 2.1 of the License, or
(at your option) any later version.
systemd 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
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with systemd; If not, see <http://www.gnu.org/licenses/>.
***/
#include <fnmatch.h>
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include "sd-hwdb.h"
#include "alloc-util.h"
#include "hwdb-util.h"
#include "string-util.h"
#include "udev-util.h"
#include "udev.h"
static sd_hwdb *hwdb;
int udev_builtin_hwdb_lookup(struct udev_device *dev,
const char *prefix, const char *modalias,
const char *filter, bool test) {
_cleanup_free_ char *lookup = NULL;
const char *key, *value;
int n = 0;
if (!hwdb)
return -ENOENT;
if (prefix) {
lookup = strjoin(prefix, modalias, NULL);
if (!lookup)
return -ENOMEM;
modalias = lookup;
}
SD_HWDB_FOREACH_PROPERTY(hwdb, modalias, key, value) {
if (filter && fnmatch(filter, key, FNM_NOESCAPE) != 0)
continue;
if (udev_builtin_add_property(dev, test, key, value) < 0)
return -ENOMEM;
n++;
}
return n;
}
static const char *modalias_usb(struct udev_device *dev, char *s, size_t size) {
const char *v, *p;
int vn, pn;
v = udev_device_get_sysattr_value(dev, "idVendor");
if (!v)
return NULL;
p = udev_device_get_sysattr_value(dev, "idProduct");
if (!p)
return NULL;
vn = strtol(v, NULL, 16);
if (vn <= 0)
return NULL;
pn = strtol(p, NULL, 16);
if (pn <= 0)
return NULL;
snprintf(s, size, "usb:v%04Xp%04X*", vn, pn);
return s;
}
static int udev_builtin_hwdb_search(struct udev_device *dev, struct udev_device *srcdev,
const char *subsystem, const char *prefix,
const char *filter, bool test) {
struct udev_device *d;
char s[16];
bool last = false;
int r = 0;
assert(dev);
if (!srcdev)
srcdev = dev;
for (d = srcdev; d && !last; d = udev_device_get_parent(d)) {
const char *dsubsys;
const char *modalias = NULL;
dsubsys = udev_device_get_subsystem(d);
if (!dsubsys)
continue;
/* look only at devices of a specific subsystem */
if (subsystem && !streq(dsubsys, subsystem))
continue;
modalias = udev_device_get_property_value(d, "MODALIAS");
if (streq(dsubsys, "usb") && streq_ptr(udev_device_get_devtype(d), "usb_device")) {
/* if the usb_device does not have a modalias, compose one */
if (!modalias)
modalias = modalias_usb(d, s, sizeof(s));
/* avoid looking at any parent device, they are usually just a USB hub */
last = true;
}
if (!modalias)
continue;
r = udev_builtin_hwdb_lookup(dev, prefix, modalias, filter, test);
if (r > 0)
break;
}
return r;
}
static int builtin_hwdb(struct udev_device *dev, int argc, char *argv[], bool test) {
static const struct option options[] = {
{ "filter", required_argument, NULL, 'f' },
{ "device", required_argument, NULL, 'd' },
{ "subsystem", required_argument, NULL, 's' },
{ "lookup-prefix", required_argument, NULL, 'p' },
{}
};
const char *filter = NULL;
const char *device = NULL;
const char *subsystem = NULL;
const char *prefix = NULL;
_cleanup_udev_device_unref_ struct udev_device *srcdev = NULL;
if (!hwdb)
return EXIT_FAILURE;
for (;;) {
int option;
option = getopt_long(argc, argv, "f:d:s:p:", options, NULL);
if (option == -1)
break;
switch (option) {
case 'f':
filter = optarg;
break;
case 'd':
device = optarg;
break;
case 's':
subsystem = optarg;
break;
case 'p':
prefix = optarg;
break;
}
}
/* query a specific key given as argument */
if (argv[optind]) {
if (udev_builtin_hwdb_lookup(dev, prefix, argv[optind], filter, test) > 0)
return EXIT_SUCCESS;
return EXIT_FAILURE;
}
/* read data from another device than the device we will store the data */
if (device) {
srcdev = udev_device_new_from_device_id(udev_device_get_udev(dev), device);
if (!srcdev)
return EXIT_FAILURE;
}
if (udev_builtin_hwdb_search(dev, srcdev, subsystem, prefix, filter, test) > 0)
return EXIT_SUCCESS;
return EXIT_FAILURE;
}
/* called at udev startup and reload */
static int builtin_hwdb_init(struct udev *udev) {
int r;
if (hwdb)
return 0;
r = sd_hwdb_new(&hwdb);
if (r < 0)
return r;
return 0;
}
/* called on udev shutdown and reload request */
static void builtin_hwdb_exit(struct udev *udev) {
hwdb = sd_hwdb_unref(hwdb);
}
/* called every couple of seconds during event activity; 'true' if config has changed */
static bool builtin_hwdb_validate(struct udev *udev) {
return hwdb_validate(hwdb);
}
const struct udev_builtin udev_builtin_hwdb = {
.name = "hwdb",
.cmd = builtin_hwdb,
.init = builtin_hwdb_init,
.exit = builtin_hwdb_exit,
.validate = builtin_hwdb_validate,
.help = "Hardware database",
};