2014-12-16 00:48:24 +01:00
|
|
|
/***
|
|
|
|
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/>.
|
|
|
|
***/
|
|
|
|
|
2015-09-23 03:01:06 +02:00
|
|
|
#include <ctype.h>
|
2014-12-16 00:48:24 +01:00
|
|
|
#include <getopt.h>
|
2015-09-23 03:01:06 +02:00
|
|
|
#include <stdlib.h>
|
2014-12-16 00:48:24 +01:00
|
|
|
#include <string.h>
|
|
|
|
|
2015-10-27 03:01:06 +01:00
|
|
|
#include "alloc-util.h"
|
2014-12-16 00:48:24 +01:00
|
|
|
#include "conf-files.h"
|
2015-10-26 18:05:03 +01:00
|
|
|
#include "fd-util.h"
|
|
|
|
#include "fileio.h"
|
2015-10-26 21:16:26 +01:00
|
|
|
#include "fs-util.h"
|
2014-12-16 00:48:24 +01:00
|
|
|
#include "hwdb-internal.h"
|
|
|
|
#include "hwdb-util.h"
|
2016-06-07 19:47:41 +02:00
|
|
|
#include "label.h"
|
2015-09-23 03:01:06 +02:00
|
|
|
#include "mkdir.h"
|
2016-06-07 19:47:41 +02:00
|
|
|
#include "selinux-util.h"
|
2015-09-23 03:01:06 +02:00
|
|
|
#include "strbuf.h"
|
2015-10-24 22:58:24 +02:00
|
|
|
#include "string-util.h"
|
2015-09-23 03:01:06 +02:00
|
|
|
#include "strv.h"
|
|
|
|
#include "util.h"
|
|
|
|
#include "verbs.h"
|
2014-12-16 00:48:24 +01:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Generic udev properties, key/value database based on modalias strings.
|
|
|
|
* Uses a Patricia/radix trie to index all matches for efficient lookup.
|
|
|
|
*/
|
|
|
|
|
|
|
|
static const char *arg_hwdb_bin_dir = "/etc/udev";
|
|
|
|
static const char *arg_root = "";
|
|
|
|
|
|
|
|
static const char * const conf_file_dirs[] = {
|
|
|
|
"/etc/udev/hwdb.d",
|
|
|
|
UDEVLIBEXECDIR "/hwdb.d",
|
|
|
|
NULL
|
|
|
|
};
|
|
|
|
|
|
|
|
/* in-memory trie objects */
|
|
|
|
struct trie {
|
|
|
|
struct trie_node *root;
|
|
|
|
struct strbuf *strings;
|
|
|
|
|
|
|
|
size_t nodes_count;
|
|
|
|
size_t children_count;
|
|
|
|
size_t values_count;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct trie_node {
|
|
|
|
/* prefix, common part for all children of this node */
|
|
|
|
size_t prefix_off;
|
|
|
|
|
|
|
|
/* sorted array of pointers to children nodes */
|
|
|
|
struct trie_child_entry *children;
|
|
|
|
uint8_t children_count;
|
|
|
|
|
|
|
|
/* sorted array of key/value pairs */
|
|
|
|
struct trie_value_entry *values;
|
|
|
|
size_t values_count;
|
|
|
|
};
|
|
|
|
|
|
|
|
/* children array item with char (0-255) index */
|
|
|
|
struct trie_child_entry {
|
|
|
|
uint8_t c;
|
|
|
|
struct trie_node *child;
|
|
|
|
};
|
|
|
|
|
|
|
|
/* value array item with key/value pairs */
|
|
|
|
struct trie_value_entry {
|
|
|
|
size_t key_off;
|
|
|
|
size_t value_off;
|
2016-09-21 14:22:49 +02:00
|
|
|
size_t filename_off;
|
|
|
|
size_t line_number;
|
2014-12-16 00:48:24 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
static int trie_children_cmp(const void *v1, const void *v2) {
|
|
|
|
const struct trie_child_entry *n1 = v1;
|
|
|
|
const struct trie_child_entry *n2 = v2;
|
|
|
|
|
|
|
|
return n1->c - n2->c;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int node_add_child(struct trie *trie, struct trie_node *node, struct trie_node *node_child, uint8_t c) {
|
|
|
|
struct trie_child_entry *child;
|
|
|
|
|
|
|
|
/* extend array, add new entry, sort for bisection */
|
|
|
|
child = realloc(node->children, (node->children_count + 1) * sizeof(struct trie_child_entry));
|
|
|
|
if (!child)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
node->children = child;
|
|
|
|
trie->children_count++;
|
|
|
|
node->children[node->children_count].c = c;
|
|
|
|
node->children[node->children_count].child = node_child;
|
|
|
|
node->children_count++;
|
|
|
|
qsort(node->children, node->children_count, sizeof(struct trie_child_entry), trie_children_cmp);
|
|
|
|
trie->nodes_count++;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static struct trie_node *node_lookup(const struct trie_node *node, uint8_t c) {
|
|
|
|
struct trie_child_entry *child;
|
|
|
|
struct trie_child_entry search;
|
|
|
|
|
|
|
|
search.c = c;
|
|
|
|
child = bsearch(&search, node->children, node->children_count, sizeof(struct trie_child_entry), trie_children_cmp);
|
|
|
|
if (child)
|
|
|
|
return child->child;
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void trie_node_cleanup(struct trie_node *node) {
|
|
|
|
size_t i;
|
|
|
|
|
|
|
|
for (i = 0; i < node->children_count; i++)
|
|
|
|
trie_node_cleanup(node->children[i].child);
|
|
|
|
free(node->children);
|
|
|
|
free(node->values);
|
|
|
|
free(node);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void trie_free(struct trie *trie) {
|
|
|
|
if (!trie)
|
|
|
|
return;
|
|
|
|
|
|
|
|
if (trie->root)
|
|
|
|
trie_node_cleanup(trie->root);
|
|
|
|
|
|
|
|
strbuf_cleanup(trie->strings);
|
|
|
|
free(trie);
|
|
|
|
}
|
|
|
|
|
|
|
|
DEFINE_TRIVIAL_CLEANUP_FUNC(struct trie*, trie_free);
|
|
|
|
|
|
|
|
static int trie_values_cmp(const void *v1, const void *v2, void *arg) {
|
|
|
|
const struct trie_value_entry *val1 = v1;
|
|
|
|
const struct trie_value_entry *val2 = v2;
|
|
|
|
struct trie *trie = arg;
|
|
|
|
|
|
|
|
return strcmp(trie->strings->buf + val1->key_off,
|
|
|
|
trie->strings->buf + val2->key_off);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int trie_node_add_value(struct trie *trie, struct trie_node *node,
|
2016-09-21 14:22:49 +02:00
|
|
|
const char *key, const char *value,
|
|
|
|
const char *filename, size_t line_number) {
|
|
|
|
ssize_t k, v, fn;
|
2014-12-16 00:48:24 +01:00
|
|
|
struct trie_value_entry *val;
|
2016-09-21 14:22:49 +02:00
|
|
|
int r;
|
2014-12-16 00:48:24 +01:00
|
|
|
|
|
|
|
k = strbuf_add_string(trie->strings, key, strlen(key));
|
|
|
|
if (k < 0)
|
|
|
|
return k;
|
|
|
|
v = strbuf_add_string(trie->strings, value, strlen(value));
|
2016-09-21 14:22:49 +02:00
|
|
|
if (v < 0)
|
|
|
|
return v;
|
|
|
|
fn = strbuf_add_string(trie->strings, filename, strlen(filename));
|
2014-12-16 00:48:24 +01:00
|
|
|
if (v < 0)
|
|
|
|
return v;
|
|
|
|
|
|
|
|
if (node->values_count) {
|
|
|
|
struct trie_value_entry search = {
|
|
|
|
.key_off = k,
|
|
|
|
.value_off = v,
|
|
|
|
};
|
|
|
|
|
|
|
|
val = xbsearch_r(&search, node->values, node->values_count, sizeof(struct trie_value_entry), trie_values_cmp, trie);
|
|
|
|
if (val) {
|
2016-09-21 14:22:49 +02:00
|
|
|
/*
|
|
|
|
* At this point we have 2 identical properties on the same match-string. We
|
|
|
|
* strictly order them by filename+line-number, since we know the dynamic
|
|
|
|
* runtime lookup does the same for multiple matching nodes.
|
|
|
|
*/
|
|
|
|
r = strcmp(filename, trie->strings->buf + val->filename_off);
|
|
|
|
if (r < 0 ||
|
|
|
|
(r == 0 && line_number < val->line_number))
|
|
|
|
return 0;
|
|
|
|
|
2014-12-16 00:48:24 +01:00
|
|
|
/* replace existing earlier key with new value */
|
|
|
|
val->value_off = v;
|
2016-09-21 14:22:49 +02:00
|
|
|
val->filename_off = fn;
|
|
|
|
val->line_number = line_number;
|
2014-12-16 00:48:24 +01:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* extend array, add new entry, sort for bisection */
|
|
|
|
val = realloc(node->values, (node->values_count + 1) * sizeof(struct trie_value_entry));
|
|
|
|
if (!val)
|
|
|
|
return -ENOMEM;
|
|
|
|
trie->values_count++;
|
|
|
|
node->values = val;
|
|
|
|
node->values[node->values_count].key_off = k;
|
|
|
|
node->values[node->values_count].value_off = v;
|
2016-09-21 14:22:49 +02:00
|
|
|
node->values[node->values_count].filename_off = fn;
|
|
|
|
node->values[node->values_count].line_number = line_number;
|
2014-12-16 00:48:24 +01:00
|
|
|
node->values_count++;
|
|
|
|
qsort_r(node->values, node->values_count, sizeof(struct trie_value_entry), trie_values_cmp, trie);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int trie_insert(struct trie *trie, struct trie_node *node, const char *search,
|
2016-09-21 14:22:49 +02:00
|
|
|
const char *key, const char *value,
|
|
|
|
const char *filename, uint64_t line_number) {
|
2014-12-16 00:48:24 +01:00
|
|
|
size_t i = 0;
|
|
|
|
int err = 0;
|
|
|
|
|
|
|
|
for (;;) {
|
|
|
|
size_t p;
|
|
|
|
uint8_t c;
|
|
|
|
struct trie_node *child;
|
|
|
|
|
|
|
|
for (p = 0; (c = trie->strings->buf[node->prefix_off + p]); p++) {
|
|
|
|
_cleanup_free_ char *s = NULL;
|
|
|
|
ssize_t off;
|
|
|
|
_cleanup_free_ struct trie_node *new_child = NULL;
|
|
|
|
|
|
|
|
if (c == search[i + p])
|
|
|
|
continue;
|
|
|
|
|
|
|
|
/* split node */
|
|
|
|
new_child = new0(struct trie_node, 1);
|
|
|
|
if (!new_child)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
/* move values from parent to child */
|
|
|
|
new_child->prefix_off = node->prefix_off + p+1;
|
|
|
|
new_child->children = node->children;
|
|
|
|
new_child->children_count = node->children_count;
|
|
|
|
new_child->values = node->values;
|
|
|
|
new_child->values_count = node->values_count;
|
|
|
|
|
|
|
|
/* update parent; use strdup() because the source gets realloc()d */
|
|
|
|
s = strndup(trie->strings->buf + node->prefix_off, p);
|
|
|
|
if (!s)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
off = strbuf_add_string(trie->strings, s, p);
|
|
|
|
if (off < 0)
|
|
|
|
return off;
|
|
|
|
|
|
|
|
node->prefix_off = off;
|
|
|
|
node->children = NULL;
|
|
|
|
node->children_count = 0;
|
|
|
|
node->values = NULL;
|
|
|
|
node->values_count = 0;
|
|
|
|
err = node_add_child(trie, node, new_child, c);
|
|
|
|
if (err < 0)
|
|
|
|
return err;
|
|
|
|
|
|
|
|
new_child = NULL; /* avoid cleanup */
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
i += p;
|
|
|
|
|
|
|
|
c = search[i];
|
|
|
|
if (c == '\0')
|
2016-09-21 14:22:49 +02:00
|
|
|
return trie_node_add_value(trie, node, key, value, filename, line_number);
|
2014-12-16 00:48:24 +01:00
|
|
|
|
|
|
|
child = node_lookup(node, c);
|
|
|
|
if (!child) {
|
|
|
|
ssize_t off;
|
|
|
|
|
|
|
|
/* new child */
|
|
|
|
child = new0(struct trie_node, 1);
|
|
|
|
if (!child)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
off = strbuf_add_string(trie->strings, search + i+1, strlen(search + i+1));
|
|
|
|
if (off < 0) {
|
|
|
|
free(child);
|
|
|
|
return off;
|
|
|
|
}
|
|
|
|
|
|
|
|
child->prefix_off = off;
|
|
|
|
err = node_add_child(trie, node, child, c);
|
|
|
|
if (err < 0) {
|
|
|
|
free(child);
|
|
|
|
return err;
|
|
|
|
}
|
|
|
|
|
2016-09-21 14:22:49 +02:00
|
|
|
return trie_node_add_value(trie, child, key, value, filename, line_number);
|
2014-12-16 00:48:24 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
node = child;
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
struct trie_f {
|
|
|
|
FILE *f;
|
|
|
|
struct trie *trie;
|
|
|
|
uint64_t strings_off;
|
|
|
|
|
|
|
|
uint64_t nodes_count;
|
|
|
|
uint64_t children_count;
|
|
|
|
uint64_t values_count;
|
|
|
|
};
|
|
|
|
|
|
|
|
/* calculate the storage space for the nodes, children arrays, value arrays */
|
|
|
|
static void trie_store_nodes_size(struct trie_f *trie, struct trie_node *node) {
|
|
|
|
uint64_t i;
|
|
|
|
|
|
|
|
for (i = 0; i < node->children_count; i++)
|
|
|
|
trie_store_nodes_size(trie, node->children[i].child);
|
|
|
|
|
|
|
|
trie->strings_off += sizeof(struct trie_node_f);
|
|
|
|
for (i = 0; i < node->children_count; i++)
|
|
|
|
trie->strings_off += sizeof(struct trie_child_entry_f);
|
|
|
|
for (i = 0; i < node->values_count; i++)
|
2016-09-21 14:22:49 +02:00
|
|
|
trie->strings_off += sizeof(struct trie_value_entry2_f);
|
2014-12-16 00:48:24 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
static int64_t trie_store_nodes(struct trie_f *trie, struct trie_node *node) {
|
|
|
|
uint64_t i;
|
|
|
|
struct trie_node_f n = {
|
|
|
|
.prefix_off = htole64(trie->strings_off + node->prefix_off),
|
|
|
|
.children_count = node->children_count,
|
|
|
|
.values_count = htole64(node->values_count),
|
|
|
|
};
|
|
|
|
struct trie_child_entry_f *children = NULL;
|
|
|
|
int64_t node_off;
|
|
|
|
|
|
|
|
if (node->children_count) {
|
|
|
|
children = new0(struct trie_child_entry_f, node->children_count);
|
|
|
|
if (!children)
|
|
|
|
return -ENOMEM;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* post-order recursion */
|
|
|
|
for (i = 0; i < node->children_count; i++) {
|
|
|
|
int64_t child_off;
|
|
|
|
|
|
|
|
child_off = trie_store_nodes(trie, node->children[i].child);
|
|
|
|
if (child_off < 0) {
|
|
|
|
free(children);
|
|
|
|
return child_off;
|
|
|
|
}
|
|
|
|
children[i].c = node->children[i].c;
|
|
|
|
children[i].child_off = htole64(child_off);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* write node */
|
|
|
|
node_off = ftello(trie->f);
|
|
|
|
fwrite(&n, sizeof(struct trie_node_f), 1, trie->f);
|
|
|
|
trie->nodes_count++;
|
|
|
|
|
|
|
|
/* append children array */
|
|
|
|
if (node->children_count) {
|
|
|
|
fwrite(children, sizeof(struct trie_child_entry_f), node->children_count, trie->f);
|
|
|
|
trie->children_count += node->children_count;
|
|
|
|
free(children);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* append values array */
|
|
|
|
for (i = 0; i < node->values_count; i++) {
|
2016-09-21 14:22:49 +02:00
|
|
|
struct trie_value_entry2_f v = {
|
2014-12-16 00:48:24 +01:00
|
|
|
.key_off = htole64(trie->strings_off + node->values[i].key_off),
|
|
|
|
.value_off = htole64(trie->strings_off + node->values[i].value_off),
|
2016-09-21 14:22:49 +02:00
|
|
|
.filename_off = htole64(trie->strings_off + node->values[i].filename_off),
|
|
|
|
.line_number = htole64(node->values[i].line_number),
|
2014-12-16 00:48:24 +01:00
|
|
|
};
|
|
|
|
|
2016-09-21 14:22:49 +02:00
|
|
|
fwrite(&v, sizeof(struct trie_value_entry2_f), 1, trie->f);
|
2014-12-16 00:48:24 +01:00
|
|
|
trie->values_count++;
|
|
|
|
}
|
|
|
|
|
|
|
|
return node_off;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int trie_store(struct trie *trie, const char *filename) {
|
|
|
|
struct trie_f t = {
|
|
|
|
.trie = trie,
|
|
|
|
};
|
|
|
|
_cleanup_free_ char *filename_tmp = NULL;
|
|
|
|
int64_t pos;
|
|
|
|
int64_t root_off;
|
|
|
|
int64_t size;
|
|
|
|
struct trie_header_f h = {
|
|
|
|
.signature = HWDB_SIG,
|
|
|
|
.tool_version = htole64(atoi(VERSION)),
|
|
|
|
.header_size = htole64(sizeof(struct trie_header_f)),
|
|
|
|
.node_size = htole64(sizeof(struct trie_node_f)),
|
|
|
|
.child_entry_size = htole64(sizeof(struct trie_child_entry_f)),
|
2016-09-21 14:22:49 +02:00
|
|
|
.value_entry_size = htole64(sizeof(struct trie_value_entry2_f)),
|
2014-12-16 00:48:24 +01:00
|
|
|
};
|
|
|
|
int err;
|
|
|
|
|
|
|
|
/* calculate size of header, nodes, children entries, value entries */
|
|
|
|
t.strings_off = sizeof(struct trie_header_f);
|
|
|
|
trie_store_nodes_size(&t, trie->root);
|
|
|
|
|
|
|
|
err = fopen_temporary(filename , &t.f, &filename_tmp);
|
|
|
|
if (err < 0)
|
|
|
|
return err;
|
|
|
|
fchmod(fileno(t.f), 0444);
|
|
|
|
|
|
|
|
/* write nodes */
|
|
|
|
err = fseeko(t.f, sizeof(struct trie_header_f), SEEK_SET);
|
|
|
|
if (err < 0) {
|
|
|
|
fclose(t.f);
|
|
|
|
unlink_noerrno(filename_tmp);
|
|
|
|
return -errno;
|
|
|
|
}
|
|
|
|
root_off = trie_store_nodes(&t, trie->root);
|
|
|
|
h.nodes_root_off = htole64(root_off);
|
|
|
|
pos = ftello(t.f);
|
|
|
|
h.nodes_len = htole64(pos - sizeof(struct trie_header_f));
|
|
|
|
|
|
|
|
/* write string buffer */
|
|
|
|
fwrite(trie->strings->buf, trie->strings->len, 1, t.f);
|
|
|
|
h.strings_len = htole64(trie->strings->len);
|
|
|
|
|
|
|
|
/* write header */
|
|
|
|
size = ftello(t.f);
|
|
|
|
h.file_size = htole64(size);
|
|
|
|
err = fseeko(t.f, 0, SEEK_SET);
|
|
|
|
if (err < 0) {
|
|
|
|
fclose(t.f);
|
|
|
|
unlink_noerrno(filename_tmp);
|
|
|
|
return -errno;
|
|
|
|
}
|
|
|
|
fwrite(&h, sizeof(struct trie_header_f), 1, t.f);
|
|
|
|
err = ferror(t.f);
|
|
|
|
if (err)
|
|
|
|
err = -errno;
|
|
|
|
fclose(t.f);
|
|
|
|
if (err < 0 || rename(filename_tmp, filename) < 0) {
|
|
|
|
unlink_noerrno(filename_tmp);
|
|
|
|
return err < 0 ? err : -errno;
|
|
|
|
}
|
|
|
|
|
|
|
|
log_debug("=== trie on-disk ===");
|
2015-01-21 04:22:15 +01:00
|
|
|
log_debug("size: %8"PRIi64" bytes", size);
|
2014-12-16 00:48:24 +01:00
|
|
|
log_debug("header: %8zu bytes", sizeof(struct trie_header_f));
|
|
|
|
log_debug("nodes: %8"PRIu64" bytes (%8"PRIu64")",
|
|
|
|
t.nodes_count * sizeof(struct trie_node_f), t.nodes_count);
|
|
|
|
log_debug("child pointers: %8"PRIu64" bytes (%8"PRIu64")",
|
|
|
|
t.children_count * sizeof(struct trie_child_entry_f), t.children_count);
|
|
|
|
log_debug("value pointers: %8"PRIu64" bytes (%8"PRIu64")",
|
2016-09-21 14:22:49 +02:00
|
|
|
t.values_count * sizeof(struct trie_value_entry2_f), t.values_count);
|
2014-12-16 00:48:24 +01:00
|
|
|
log_debug("string store: %8zu bytes", trie->strings->len);
|
|
|
|
log_debug("strings start: %8"PRIu64, t.strings_off);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2016-09-21 14:22:49 +02:00
|
|
|
static int insert_data(struct trie *trie, char **match_list, char *line,
|
|
|
|
const char *filename, size_t line_number) {
|
2014-12-16 00:48:24 +01:00
|
|
|
char *value, **entry;
|
|
|
|
|
|
|
|
value = strchr(line, '=');
|
|
|
|
if (!value) {
|
|
|
|
log_error("Error, key/value pair expected but got '%s' in '%s':", line, filename);
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
value[0] = '\0';
|
|
|
|
value++;
|
|
|
|
|
|
|
|
/* libudev requires properties to start with a space */
|
|
|
|
while (isblank(line[0]) && isblank(line[1]))
|
|
|
|
line++;
|
|
|
|
|
|
|
|
if (line[0] == '\0' || value[0] == '\0') {
|
|
|
|
log_error("Error, empty key or value '%s' in '%s':", line, filename);
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
STRV_FOREACH(entry, match_list)
|
2016-09-21 14:22:49 +02:00
|
|
|
trie_insert(trie, trie->root, *entry, line, value, filename, line_number);
|
2014-12-16 00:48:24 +01:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int import_file(struct trie *trie, const char *filename) {
|
|
|
|
enum {
|
|
|
|
HW_NONE,
|
|
|
|
HW_MATCH,
|
|
|
|
HW_DATA,
|
|
|
|
} state = HW_NONE;
|
|
|
|
_cleanup_fclose_ FILE *f = NULL;
|
|
|
|
char line[LINE_MAX];
|
|
|
|
_cleanup_strv_free_ char **match_list = NULL;
|
2016-09-21 14:22:49 +02:00
|
|
|
size_t line_number = 0;
|
2014-12-16 00:48:24 +01:00
|
|
|
char *match = NULL;
|
|
|
|
int r;
|
|
|
|
|
|
|
|
f = fopen(filename, "re");
|
|
|
|
if (!f)
|
|
|
|
return -errno;
|
|
|
|
|
|
|
|
while (fgets(line, sizeof(line), f)) {
|
|
|
|
size_t len;
|
|
|
|
char *pos;
|
|
|
|
|
2016-09-21 14:22:49 +02:00
|
|
|
++line_number;
|
|
|
|
|
2014-12-16 00:48:24 +01:00
|
|
|
/* comment line */
|
|
|
|
if (line[0] == '#')
|
|
|
|
continue;
|
|
|
|
|
|
|
|
/* strip trailing comment */
|
|
|
|
pos = strchr(line, '#');
|
|
|
|
if (pos)
|
|
|
|
pos[0] = '\0';
|
|
|
|
|
|
|
|
/* strip trailing whitespace */
|
|
|
|
len = strlen(line);
|
|
|
|
while (len > 0 && isspace(line[len-1]))
|
|
|
|
len--;
|
|
|
|
line[len] = '\0';
|
|
|
|
|
|
|
|
switch (state) {
|
|
|
|
case HW_NONE:
|
|
|
|
if (len == 0)
|
|
|
|
break;
|
|
|
|
|
|
|
|
if (line[0] == ' ') {
|
|
|
|
log_error("Error, MATCH expected but got '%s' in '%s':", line, filename);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* start of record, first match */
|
|
|
|
state = HW_MATCH;
|
|
|
|
|
|
|
|
match = strdup(line);
|
|
|
|
if (!match)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
r = strv_consume(&match_list, match);
|
|
|
|
if (r < 0)
|
|
|
|
return r;
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case HW_MATCH:
|
|
|
|
if (len == 0) {
|
|
|
|
log_error("Error, DATA expected but got empty line in '%s':", filename);
|
|
|
|
state = HW_NONE;
|
|
|
|
strv_clear(match_list);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* another match */
|
|
|
|
if (line[0] != ' ') {
|
|
|
|
match = strdup(line);
|
|
|
|
if (!match)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
r = strv_consume(&match_list, match);
|
|
|
|
if (r < 0)
|
|
|
|
return r;
|
|
|
|
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* first data */
|
|
|
|
state = HW_DATA;
|
2016-09-21 14:22:49 +02:00
|
|
|
insert_data(trie, match_list, line, filename, line_number);
|
2014-12-16 00:48:24 +01:00
|
|
|
break;
|
|
|
|
|
|
|
|
case HW_DATA:
|
|
|
|
/* end of record */
|
|
|
|
if (len == 0) {
|
|
|
|
state = HW_NONE;
|
|
|
|
strv_clear(match_list);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (line[0] != ' ') {
|
|
|
|
log_error("Error, DATA expected but got '%s' in '%s':", line, filename);
|
|
|
|
state = HW_NONE;
|
|
|
|
strv_clear(match_list);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2016-09-21 14:22:49 +02:00
|
|
|
insert_data(trie, match_list, line, filename, line_number);
|
2014-12-16 00:48:24 +01:00
|
|
|
break;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-12-19 03:01:27 +01:00
|
|
|
static int hwdb_query(int argc, char *argv[], void *userdata) {
|
tree-wide: expose "p"-suffix unref calls in public APIs to make gcc cleanup easy
GLIB has recently started to officially support the gcc cleanup
attribute in its public API, hence let's do the same for our APIs.
With this patch we'll define an xyz_unrefp() call for each public
xyz_unref() call, to make it easy to use inside a
__attribute__((cleanup())) expression. Then, all code is ported over to
make use of this.
The new calls are also documented in the man pages, with examples how to
use them (well, I only added docs where the _unref() call itself already
had docs, and the examples, only cover sd_bus_unrefp() and
sd_event_unrefp()).
This also renames sd_lldp_free() to sd_lldp_unref(), since that's how we
tend to call our destructors these days.
Note that this defines no public macro that wraps gcc's attribute and
makes it easier to use. While I think it's our duty in the library to
make our stuff easy to use, I figure it's not our duty to make gcc's own
features easy to use on its own. Most likely, client code which wants to
make use of this should define its own:
#define _cleanup_(function) __attribute__((cleanup(function)))
Or similar, to make the gcc feature easier to use.
Making this logic public has the benefit that we can remove three header
files whose only purpose was to define these functions internally.
See #2008.
2015-11-27 19:13:45 +01:00
|
|
|
_cleanup_(sd_hwdb_unrefp) sd_hwdb *hwdb = NULL;
|
2014-12-16 00:48:24 +01:00
|
|
|
const char *key, *value;
|
|
|
|
const char *modalias;
|
|
|
|
int r;
|
|
|
|
|
2014-12-19 03:01:27 +01:00
|
|
|
assert(argc >= 2);
|
|
|
|
assert(argv);
|
2014-12-16 00:48:24 +01:00
|
|
|
|
2014-12-19 03:01:27 +01:00
|
|
|
modalias = argv[1];
|
2014-12-16 00:48:24 +01:00
|
|
|
|
|
|
|
r = sd_hwdb_new(&hwdb);
|
|
|
|
if (r < 0)
|
|
|
|
return r;
|
|
|
|
|
|
|
|
SD_HWDB_FOREACH_PROPERTY(hwdb, modalias, key, value)
|
|
|
|
printf("%s=%s\n", key, value);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-12-19 03:01:27 +01:00
|
|
|
static int hwdb_update(int argc, char *argv[], void *userdata) {
|
2014-12-16 00:48:24 +01:00
|
|
|
_cleanup_free_ char *hwdb_bin = NULL;
|
|
|
|
_cleanup_(trie_freep) struct trie *trie = NULL;
|
|
|
|
char **files, **f;
|
|
|
|
int r;
|
|
|
|
|
|
|
|
trie = new0(struct trie, 1);
|
|
|
|
if (!trie)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
/* string store */
|
|
|
|
trie->strings = strbuf_new();
|
|
|
|
if (!trie->strings)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
/* index */
|
|
|
|
trie->root = new0(struct trie_node, 1);
|
|
|
|
if (!trie->root)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
trie->nodes_count++;
|
|
|
|
|
|
|
|
r = conf_files_list_strv(&files, ".hwdb", arg_root, conf_file_dirs);
|
|
|
|
if (r < 0)
|
|
|
|
return log_error_errno(r, "failed to enumerate hwdb files: %m");
|
|
|
|
|
|
|
|
STRV_FOREACH(f, files) {
|
|
|
|
log_debug("reading file '%s'", *f);
|
|
|
|
import_file(trie, *f);
|
|
|
|
}
|
|
|
|
strv_free(files);
|
|
|
|
|
|
|
|
strbuf_complete(trie->strings);
|
|
|
|
|
|
|
|
log_debug("=== trie in-memory ===");
|
|
|
|
log_debug("nodes: %8zu bytes (%8zu)",
|
|
|
|
trie->nodes_count * sizeof(struct trie_node), trie->nodes_count);
|
|
|
|
log_debug("children arrays: %8zu bytes (%8zu)",
|
|
|
|
trie->children_count * sizeof(struct trie_child_entry), trie->children_count);
|
|
|
|
log_debug("values arrays: %8zu bytes (%8zu)",
|
|
|
|
trie->values_count * sizeof(struct trie_value_entry), trie->values_count);
|
|
|
|
log_debug("strings: %8zu bytes",
|
|
|
|
trie->strings->len);
|
|
|
|
log_debug("strings incoming: %8zu bytes (%8zu)",
|
|
|
|
trie->strings->in_len, trie->strings->in_count);
|
|
|
|
log_debug("strings dedup'ed: %8zu bytes (%8zu)",
|
|
|
|
trie->strings->dedup_len, trie->strings->dedup_count);
|
|
|
|
|
2016-10-23 17:43:27 +02:00
|
|
|
hwdb_bin = strjoin(arg_root, "/", arg_hwdb_bin_dir, "/hwdb.bin");
|
2014-12-16 00:48:24 +01:00
|
|
|
if (!hwdb_bin)
|
|
|
|
return -ENOMEM;
|
|
|
|
|
2016-06-07 19:47:41 +02:00
|
|
|
mkdir_parents_label(hwdb_bin, 0755);
|
2014-12-16 00:48:24 +01:00
|
|
|
r = trie_store(trie, hwdb_bin);
|
|
|
|
if (r < 0)
|
|
|
|
return log_error_errno(r, "Failure writing database %s: %m", hwdb_bin);
|
|
|
|
|
2016-06-07 19:47:41 +02:00
|
|
|
return label_fix(hwdb_bin, false, false);
|
2014-12-16 00:48:24 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
static void help(void) {
|
|
|
|
printf("Usage: %s OPTIONS COMMAND\n\n"
|
2015-02-12 13:17:08 +01:00
|
|
|
"Update or query the hardware database.\n\n"
|
|
|
|
" -h --help Show this help\n"
|
|
|
|
" --version Show package version\n"
|
|
|
|
" --usr Generate in " UDEVLIBEXECDIR " instead of /etc/udev\n"
|
|
|
|
" -r --root=PATH Alternative root path in the filesystem\n\n"
|
2014-12-16 00:48:24 +01:00
|
|
|
"Commands:\n"
|
2015-02-12 13:17:08 +01:00
|
|
|
" update Update the hwdb database\n"
|
|
|
|
" query MODALIAS Query database and print result\n",
|
2014-12-16 00:48:24 +01:00
|
|
|
program_invocation_short_name);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int parse_argv(int argc, char *argv[]) {
|
|
|
|
enum {
|
2015-02-12 13:17:08 +01:00
|
|
|
ARG_VERSION = 0x100,
|
|
|
|
ARG_USR,
|
2014-12-16 00:48:24 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
static const struct option options[] = {
|
2015-02-12 13:17:08 +01:00
|
|
|
{ "help", no_argument, NULL, 'h' },
|
|
|
|
{ "version", no_argument, NULL, ARG_VERSION },
|
|
|
|
{ "usr", no_argument, NULL, ARG_USR },
|
|
|
|
{ "root", required_argument, NULL, 'r' },
|
2014-12-16 00:48:24 +01:00
|
|
|
{}
|
|
|
|
};
|
|
|
|
|
|
|
|
int c;
|
|
|
|
|
|
|
|
assert(argc >= 0);
|
|
|
|
assert(argv);
|
|
|
|
|
|
|
|
while ((c = getopt_long(argc, argv, "ut:r:h", options, NULL)) >= 0) {
|
|
|
|
switch(c) {
|
2015-02-12 13:17:08 +01:00
|
|
|
|
|
|
|
case 'h':
|
|
|
|
help();
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
case ARG_VERSION:
|
2015-09-23 03:01:06 +02:00
|
|
|
return version();
|
2015-02-12 13:17:08 +01:00
|
|
|
|
2014-12-16 00:48:24 +01:00
|
|
|
case ARG_USR:
|
|
|
|
arg_hwdb_bin_dir = UDEVLIBEXECDIR;
|
|
|
|
break;
|
2015-02-12 13:17:08 +01:00
|
|
|
|
2014-12-16 00:48:24 +01:00
|
|
|
case 'r':
|
|
|
|
arg_root = optarg;
|
|
|
|
break;
|
2015-02-12 13:17:08 +01:00
|
|
|
|
2014-12-16 00:48:24 +01:00
|
|
|
case '?':
|
|
|
|
return -EINVAL;
|
2015-02-12 13:17:08 +01:00
|
|
|
|
2014-12-16 00:48:24 +01:00
|
|
|
default:
|
|
|
|
assert_not_reached("Unknown option");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int hwdb_main(int argc, char *argv[]) {
|
2014-12-19 03:01:27 +01:00
|
|
|
const Verb verbs[] = {
|
|
|
|
{ "update", 1, 1, 0, hwdb_update },
|
2015-02-12 13:17:08 +01:00
|
|
|
{ "query", 2, 2, 0, hwdb_query },
|
2014-12-19 03:01:27 +01:00
|
|
|
{},
|
2014-12-16 00:48:24 +01:00
|
|
|
};
|
|
|
|
|
2014-12-19 03:01:27 +01:00
|
|
|
return dispatch_verb(argc, argv, verbs, NULL);
|
2014-12-16 00:48:24 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
int main (int argc, char *argv[]) {
|
|
|
|
int r;
|
|
|
|
|
|
|
|
log_parse_environment();
|
|
|
|
log_open();
|
|
|
|
|
|
|
|
r = parse_argv(argc, argv);
|
|
|
|
if (r <= 0)
|
|
|
|
goto finish;
|
|
|
|
|
2016-06-07 19:47:41 +02:00
|
|
|
mac_selinux_init();
|
|
|
|
|
2014-12-16 00:48:24 +01:00
|
|
|
r = hwdb_main(argc, argv);
|
|
|
|
|
|
|
|
finish:
|
|
|
|
return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
|
|
|
|
}
|