udevdb.c

   1 /*
   2  * udevdb.c - udev database library
   3  *
   4  * Userspace devfs
   5  *
   6  * Copyright (C) 2003 Greg Kroah-Hartman <greg@kroah.com>
   7  * Copyright (C) 2003 IBM Corp.
   8  *
   9  *      This program is free software; you can redistribute it and/or modify it
  10  *      under the terms of the GNU General Public License as published by the
  11  *      Free Software Foundation version 2 of the License.
  12  *
  13  *      This program is distributed in the hope that it will be useful, but
  14  *      WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  *      General Public License for more details.
  17  *
  18  *      You should have received a copy of the GNU General Public License along
  19  *      with this program; if not, write to the Free Software Foundation, Inc.,
  20  *      675 Mass Ave, Cambridge, MA 02139, USA.
  21  *
  22  */
  23
  24 #define _KLIBC_HAS_ARCH_SIG_ATOMIC_T
  25 #include <stdlib.h>
  26 #include <stdio.h>
  27 #include <string.h>
  28 #include <stddef.h>
  29 #include <fcntl.h>
  30 #include <string.h>
  31 #include <sys/stat.h>
  32 #include <errno.h>
  33 #include <signal.h>
  34
  35 #include "libsysfs/sysfs/libsysfs.h"
  36 #include "udev.h"
  37 #include "udev_lib.h"
  38 #include "udev_version.h"
  39 #include "logging.h"
  40 #include "namedev.h"
  41 #include "udevdb.h"
  42 #include "tdb/tdb.h"
  43
  44 static TDB_CONTEXT *udevdb;
  45 sig_atomic_t gotalarm;
  46
  47 int udevdb_add_dev(struct udevice *udev)
  48 {
  49         TDB_DATA key, data;
  50         char keystr[SYSFS_PATH_MAX];
  51
  52         if (udev->test_run)
  53                 return 0;
  54
  55         if (udevdb == NULL)
  56                 return -1;
  57
  58         memset(keystr, 0x00, SYSFS_PATH_MAX);
  59         strfieldcpy(keystr, udev->devpath);
  60         key.dptr = keystr;
  61         key.dsize = strlen(keystr) + 1;
  62
  63         data.dptr = (void *) udev;
  64         data.dsize = UDEVICE_DB_LEN;
  65         dbg("store key '%s' for device '%s'", keystr, udev->name);
  66
  67         return tdb_store(udevdb, key, data, TDB_REPLACE);
  68 }
  69
  70 int udevdb_get_dev(const char *path, struct udevice *udev)
  71 {
  72         TDB_DATA key, data;
  73
  74         if (udevdb == NULL)
  75                 return -1;
  76
  77         if (path == NULL)
  78                 return -ENODEV;
  79
  80         key.dptr = (void *)path;
  81         key.dsize = strlen(path) + 1;
  82
  83         data = tdb_fetch(udevdb, key);
  84         if (data.dptr == NULL || data.dsize == 0)
  85                 return -ENODEV;
  86
  87         memset(udev, 0x00, sizeof(struct udevice));
  88         memcpy(udev, data.dptr, UDEVICE_DB_LEN);
  89
  90         return 0;
  91 }
  92
  93 int udevdb_delete_dev(const char *path)
  94 {
  95         TDB_DATA key;
  96         char keystr[SYSFS_PATH_MAX];
  97
  98         if (udevdb == NULL)
  99                 return -1;
 100
 101         if (path == NULL)
 102                 return -EINVAL;
 103
 104         memset(keystr, 0, sizeof(keystr));
 105         strfieldcpy(keystr, path);
 106
 107         key.dptr = keystr;
 108         key.dsize = strlen(keystr) + 1;
 109
 110         return tdb_delete(udevdb, key);
 111 }
 112
 113 /**
 114  * udevdb_exit: closes database
 115  */
 116 void udevdb_exit(void)
 117 {
 118         if (udevdb != NULL) {
 119                 tdb_close(udevdb);
 120                 udevdb = NULL;
 121         }
 122 }
 123
 124 /**
 125  * udevdb_init: initializes database
 126  * @init_flag: UDEVDB_INTERNAL - database stays in memory
 127  *             UDEVDB_DEFAULT - database is written to a file
 128  */
 129 int udevdb_init(int init_flag)
 130 {
 131         if (init_flag != UDEVDB_DEFAULT && init_flag != UDEVDB_INTERNAL)
 132                 return -EINVAL;
 133
 134         tdb_set_lock_alarm(&gotalarm);
 135
 136         udevdb = tdb_open(udev_db_filename, 0, init_flag, O_RDWR | O_CREAT, 0644);
 137         if (udevdb == NULL) {
 138                 if (init_flag == UDEVDB_INTERNAL)
 139                         dbg("unable to initialize in-memory database");
 140                 else
 141                         dbg("unable to initialize database at '%s'", udev_db_filename);
 142                 return -EACCES;
 143         }
 144         return 0;
 145 }
 146
 147 /**
 148  * udevdb_open_ro: open database for reading
 149  */
 150 int udevdb_open_ro(void)
 151 {
 152         udevdb = tdb_open(udev_db_filename, 0, 0, O_RDONLY, 0);
 153         if (udevdb == NULL) {
 154                 dbg("unable to open database at '%s'", udev_db_filename);
 155                 return -EACCES;
 156         }
 157         return 0;
 158 }
 159
 160 static int (*user_record_callback) (const char *path, struct udevice *dev);
 161
 162 static int traverse_callback(TDB_CONTEXT *tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
 163 {
 164         return user_record_callback((char*) key.dptr, (struct udevice*) dbuf.dptr);
 165 }
 166
 167 /**
 168  * udevdb_call_foreach: dumps whole database by passing record data to user function
 169  * @user_record_handler: user function called for every record in the database
 170  */
 171 int udevdb_call_foreach(int (*user_record_handler) (const char *path, struct udevice *dev))
 172 {
 173         int retval = 0;
 174
 175         if (udevdb == NULL)
 176                 return -1;
 177
 178         if (user_record_handler == NULL) {
 179                 dbg("invalid user record handling function");
 180                 return -EINVAL;
 181         }
 182         user_record_callback = user_record_handler;
 183         retval = tdb_traverse(udevdb, traverse_callback, NULL);
 184         if (retval < 0)
 185                 return -ENODEV;
 186         else
 187                 return 0;
 188 }
 189
 190 static struct udevice *find_dev;
 191 static char *find_path;
 192 static const char *find_name;
 193 static int find_found;
 194
 195 static int find_device_by_name(const char *path, struct udevice *udev)
 196 {
 197         char *pos;
 198         int len;
 199
 200         if (strncmp(udev->name, find_name, sizeof(udev->name)) == 0) {
 201                 memcpy(find_dev, udev, sizeof(struct udevice));
 202                 strfieldcpymax(find_path, path, NAME_SIZE);
 203                 find_found = 1;
 204                 /* stop search */
 205                 return 1;
 206         }
 207         /* look for matching symlink*/
 208         foreach_strpart(udev->symlink, " ", pos, len) {
 209                 if (strncmp(pos, find_name, len) != 0)
 210                         continue;
 211
 212                 if (len != strlen(find_name))
 213                         continue;
 214
 215                 memcpy(find_dev, udev, sizeof(struct udevice));
 216                 strfieldcpymax(find_path, path, NAME_SIZE);
 217                 find_found = 1;
 218                 return 1;
 219         }
 220         return 0;
 221 }
 222
 223 /**
 224  * udevdb_get_dev_byname: search device with given name by traversing the whole database
 225  */
 226 int udevdb_get_dev_byname(const char *name, char *path, struct udevice *dev)
 227 {
 228         find_found = 0;
 229         find_path = path;
 230         find_dev = dev;
 231         find_name = name;
 232         udevdb_call_foreach(find_device_by_name);
 233         if (find_found == 1)
 234                 return 0;
 235         else
 236                 return -1;
 237 }