#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;

}

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;

}

static void builtin_hwdb_exit(struct udev *udev) {

hwdb = sd_hwdb_unref(hwdb);

}

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",

};