udev-builtin-hwdb.c revision 33502ffe2eb7b56cdd018a4fb6830d7828519fad
/*
* usb-db, pci-db - lookup vendor/product database
*
* Copyright (C) 2009 Lennart Poettering <lennart@poettering.net>
* Copyright (C) 2011 Kay Sievers <kay.sievers@vrfy.org>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 2 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <inttypes.h>
#include <ctype.h>
#include <stdlib.h>
#include "udev.h"
static int get_id_attr(
struct udev_device *parent,
const char *name,
uint16_t *value) {
const char *t;
unsigned u;
if (!(t = udev_device_get_sysattr_value(parent, name))) {
fprintf(stderr, "%s lacks %s.\n", udev_device_get_syspath(parent), name);
return -1;
}
if (startswith(t, "0x"))
t += 2;
if (sscanf(t, "%04x", &u) != 1 || u > 0xFFFFU) {
fprintf(stderr, "Failed to parse %s on %s.\n", name, udev_device_get_syspath(parent));
return -1;
}
*value = (uint16_t) u;
return 0;
}
static int get_vid_pid(
struct udev_device *parent,
const char *vendor_attr,
const char *product_attr,
uint16_t *vid,
uint16_t *pid) {
if (get_id_attr(parent, vendor_attr, vid) < 0)
return -1;
else if (*vid <= 0) {
fprintf(stderr, "Invalid vendor id.\n");
return -1;
}
if (get_id_attr(parent, product_attr, pid) < 0)
return -1;
return 0;
}
static void rstrip(char *n) {
size_t i;
for (i = strlen(n); i > 0 && isspace(n[i-1]); i--)
n[i-1] = 0;
}
#define HEXCHARS "0123456789abcdefABCDEF"
#define WHITESPACE " \t\n\r"
static int lookup_vid_pid(const char *database,
uint16_t vid, uint16_t pid,
char **vendor, char **product)
{
FILE *f;
int ret = -1;
int found_vendor = 0;
char *line = NULL;
*vendor = *product = NULL;
if (!(f = fopen(database, "rme"))) {
fprintf(stderr, "Failed to open database file '%s': %s\n", database, strerror(errno));
return -1;
}
for (;;) {
size_t n;
if (getline(&line, &n, f) < 0)
break;
rstrip(line);
if (line[0] == '#' || line[0] == 0)
continue;
if (strspn(line, HEXCHARS) == 4) {
unsigned u;
if (found_vendor)
break;
if (sscanf(line, "%04x", &u) == 1 && u == vid) {
char *t;
t = line+4;
t += strspn(t, WHITESPACE);
if (!(*vendor = strdup(t))) {
fprintf(stderr, "Out of memory.\n");
goto finish;
}
found_vendor = 1;
}
continue;
}
if (found_vendor && line[0] == '\t' && strspn(line+1, HEXCHARS) == 4) {
unsigned u;
if (sscanf(line+1, "%04x", &u) == 1 && u == pid) {
char *t;
t = line+5;
t += strspn(t, WHITESPACE);
if (!(*product = strdup(t))) {
fprintf(stderr, "Out of memory.\n");
goto finish;
}
break;
}
}
}
ret = 0;
finish:
free(line);
fclose(f);
if (ret < 0) {
free(*product);
free(*vendor);
*product = *vendor = NULL;
}
return ret;
}
static struct udev_device *find_device(struct udev_device *dev, const char *subsys, const char *devtype)
{
const char *str;
str = udev_device_get_subsystem(dev);
if (str == NULL)
goto try_parent;
if (strcmp(str, subsys) != 0)
goto try_parent;
if (devtype != NULL) {
str = udev_device_get_devtype(dev);
if (str == NULL)
goto try_parent;
if (strcmp(str, devtype) != 0)
goto try_parent;
}
return dev;
try_parent:
return udev_device_get_parent_with_subsystem_devtype(dev, subsys, devtype);
}
static int builtin_db(struct udev_device *dev, bool test,
const char *database,
const char *vendor_attr, const char *product_attr,
const char *subsys, const char *devtype)
{
struct udev_device *parent;
uint16_t vid = 0, pid = 0;
char *vendor = NULL, *product = NULL;
parent = find_device(dev, subsys, devtype);
if (!parent) {
fprintf(stderr, "Failed to find device.\n");
goto finish;
}
if (get_vid_pid(parent, vendor_attr, product_attr, &vid, &pid) < 0)
goto finish;
if (lookup_vid_pid(database, vid, pid, &vendor, &product) < 0)
goto finish;
if (vendor)
udev_builtin_add_property(dev, test, "ID_VENDOR_FROM_DATABASE", vendor);
if (product)
udev_builtin_add_property(dev, test, "ID_MODEL_FROM_DATABASE", product);
finish:
free(vendor);
free(product);
return 0;
}
static int builtin_usb_db(struct udev_device *dev, int argc, char *argv[], bool test)
{
return builtin_db(dev, test, USB_DATABASE, "idVendor", "idProduct", "usb", "usb_device");
}
static int builtin_pci_db(struct udev_device *dev, int argc, char *argv[], bool test)
{
return builtin_db(dev, test, PCI_DATABASE, "vendor", "device", "pci", NULL);
}
const struct udev_builtin udev_builtin_usb_db = {
.name = "usb-db",
.cmd = builtin_usb_db,
.help = "USB vendor/product database",
.run_once = true,
};
const struct udev_builtin udev_builtin_pci_db = {
.name = "pci-db",
.cmd = builtin_pci_db,
.help = "PCI vendor/product database",
.run_once = true,
};