[PATCH] let the extras/ programs build "pretty" also

[elogind.git] / udevdb.c

diff --git a/udevdb.c b/udevdb.c
index c4e064fc115d4414ef21cbbbc44403ed4570e7bc..a218b66179319f7f48b0db7f8df941f0f6af51d1 100644 (file)
--- a/udevdb.c
+++ b/udevdb.c
@@ -1,5 +1,5 @@
 /*
- * udevdb.c
+ * udevdb.c - udev database library
  *
  * Userspace devfs
  *
@@ -21,45 +21,49 @@
  *
  */
 
-/*
- * udev database library
- */
 #define _KLIBC_HAS_ARCH_SIG_ATOMIC_T
 #include <stdlib.h>
 #include <stdio.h>
+#include <string.h>
+#include <stddef.h>
 #include <fcntl.h>
 #include <string.h>
 #include <sys/stat.h>
 #include <errno.h>
 #include <signal.h>
 
-#include "udev_version.h"
+#include "libsysfs/sysfs/libsysfs.h"
 #include "udev.h"
+#include "udev_lib.h"
+#include "udev_version.h"
 #include "logging.h"
 #include "namedev.h"
 #include "udevdb.h"
 #include "tdb/tdb.h"
-#include "libsysfs/libsysfs.h"
 
 static TDB_CONTEXT *udevdb;
-
+sig_atomic_t gotalarm;
 
 int udevdb_add_dev(const char *path, const struct udevice *dev)
 {
        TDB_DATA key, data;
        char keystr[SYSFS_PATH_MAX];
 
+       if (udevdb == NULL)
+               return -1;
+
        if ((path == NULL) || (dev == NULL))
                return -ENODEV;
 
-       memset(keystr, 0, NAME_SIZE);
-       strcpy(keystr, path);
+       memset(keystr, 0, SYSFS_PATH_MAX);
+       strfieldcpy(keystr, path);
        key.dptr = keystr;
        key.dsize = strlen(keystr) + 1;
 
        data.dptr = (void *)dev;
-       data.dsize = sizeof(*dev);
-       
+       data.dsize = UDEVICE_LEN;
+       dbg("store key '%s' for device '%s'", path, dev->name);
+
        return tdb_store(udevdb, key, data, TDB_REPLACE); 
 }
 
@@ -67,6 +71,9 @@ int udevdb_get_dev(const char *path, struct udevice *dev)
 {
        TDB_DATA key, data;
 
+       if (udevdb == NULL)
+               return -1;
+
        if (path == NULL)
                return -ENODEV;
 
@@ -77,7 +84,8 @@ int udevdb_get_dev(const char *path, struct udevice *dev)
        if (data.dptr == NULL || data.dsize == 0)
                return -ENODEV;
 
-       memcpy(dev, data.dptr, sizeof(*dev));
+       memset(dev, 0, sizeof(struct udevice));
+       memcpy(dev, data.dptr, UDEVICE_LEN);
        return 0;
 }
 
@@ -86,11 +94,14 @@ int udevdb_delete_dev(const char *path)
        TDB_DATA key;
        char keystr[SYSFS_PATH_MAX];
 
+       if (udevdb == NULL)
+               return -1;
+
        if (path == NULL)
                return -EINVAL;
 
        memset(keystr, 0, sizeof(keystr));
-       strcpy(keystr, path);
+       strfieldcpy(keystr, path);
 
        key.dptr = keystr;
        key.dsize = strlen(keystr) + 1;
@@ -119,6 +130,8 @@ int udevdb_init(int init_flag)
        if (init_flag != UDEVDB_DEFAULT && init_flag != UDEVDB_INTERNAL)
                return -EINVAL;
 
+       tdb_set_lock_alarm(&gotalarm);
+
        udevdb = tdb_open(udev_db_filename, 0, init_flag, O_RDWR | O_CREAT, 0644);
        if (udevdb == NULL) {
                if (init_flag == UDEVDB_INTERNAL)
@@ -158,6 +171,9 @@ int udevdb_call_foreach(int (*user_record_handler) (char *path, struct udevice *
 {
        int retval = 0;
 
+       if (udevdb == NULL)
+               return -1;
+
        if (user_record_handler == NULL) {
                dbg("invalid user record handling function");
                return -EINVAL;
@@ -177,13 +193,29 @@ static int find_found;
 
 static int find_device_by_name(char *path, struct udevice *dev)
 {
+       char *pos;
+       int len;
+
        if (strncmp(dev->name, find_name, sizeof(dev->name)) == 0) {
-               memcpy(find_dev, dev, sizeof(*find_dev));
-               strncpy(find_path, path, NAME_SIZE);
+               memcpy(find_dev, dev, sizeof(struct udevice));
+               strfieldcpymax(find_path, path, NAME_SIZE);
                find_found = 1;
                /* stop search */
                return 1;
        }
+       /* look for matching symlink*/
+       foreach_strpart(dev->symlink, " ", pos, len) {
+               if (strncmp(pos, find_name, len) != 0)
+                       continue;
+
+               if (len != strlen(find_name))
+                       continue;
+
+               memcpy(find_dev, dev, sizeof(struct udevice));
+               strfieldcpymax(find_path, path, NAME_SIZE);
+               find_found = 1;
+               return 1;
+       }
        return 0;
 }