nmdb/be-tdb.c

   1
   2 #if BE_ENABLE_TDB
   3
   4 #include <string.h>     /* memcpy() */
   5 #include <stdlib.h>     /* malloc() and friends */
   6
   7 /* tdb.h needs mode_t defined externally, and it is defined in one of these
   8 (which are the ones required for open() */
   9 #include <sys/types.h>
  10 #include <sys/stat.h>
  11 #include <fcntl.h>
  12 #include <tdb.h>
  13
  14 #include "be.h"
  15
  16 /* Local operations names are prepended with an 'x' so they don't collide with
  17  * tdb real functions */
  18
  19 int xtdb_set(struct db_conn *db, const unsigned char *key, size_t ksize,
  20                 unsigned char *val, size_t vsize);
  21 int xtdb_get(struct db_conn *db, const unsigned char *key, size_t ksize,
  22                 unsigned char *val, size_t *vsize);
  23 int xtdb_del(struct db_conn *db, const unsigned char *key, size_t ksize);
  24 int xtdb_firstkey(struct db_conn *db, unsigned char *key, size_t *ksize);
  25 int xtdb_nextkey(struct db_conn *db,
  26                 const unsigned char *key, size_t ksize,
  27                 unsigned char *nextkey, size_t *nksize);
  28 int xtdb_close(struct db_conn *db);
  29
  30
  31 struct db_conn *xtdb_open(const char *name, int flags)
  32 {
  33         struct db_conn *db;
  34         TDB_CONTEXT *tdb_db;
  35
  36         tdb_db = tdb_open(name, 0, 0, O_CREAT | O_RDWR, 0640);
  37         if (tdb_db == NULL)
  38                 return NULL;
  39
  40         db = malloc(sizeof(struct db_conn));
  41         if (db == NULL) {
  42                 tdb_close(tdb_db);
  43                 return NULL;
  44         }
  45
  46         db->conn = tdb_db;
  47         db->set = xtdb_set;
  48         db->get = xtdb_get;
  49         db->del = xtdb_del;
  50         db->firstkey = xtdb_firstkey;
  51         db->nextkey = xtdb_nextkey;
  52         db->close = xtdb_close;
  53
  54         return db;
  55 }
  56
  57 int xtdb_close(struct db_conn *db)
  58 {
  59         int rv;
  60
  61         rv = tdb_close(db->conn);
  62         free(db);
  63         return rv == 0;
  64 }
  65
  66 int xtdb_set(struct db_conn *db, const unsigned char *key, size_t ksize,
  67                 unsigned char *val, size_t vsize)
  68 {
  69         TDB_DATA k, v;
  70
  71         /* we can't maintain "const"ness here because tdb's prototypes; the
  72          * same applies to get and del, so we just cast */
  73         k.dptr = (unsigned char *) key;
  74         k.dsize = ksize;
  75         v.dptr = (unsigned char *) val;
  76         v.dsize = vsize;
  77
  78         return tdb_store(db->conn, k, v, TDB_REPLACE) == 0;
  79 }
  80
  81 int xtdb_get(struct db_conn *db, const unsigned char *key, size_t ksize,
  82                 unsigned char *val, size_t *vsize)
  83 {
  84         TDB_DATA k, v;
  85
  86         k.dptr = (unsigned char *) key;
  87         k.dsize = ksize;
  88
  89         v = tdb_fetch(db->conn, k);
  90         if (v.dptr == NULL)
  91                 return 0;
  92
  93         if (v.dsize > *vsize)
  94                 return 0;
  95
  96         *vsize = v.dsize;
  97         memcpy(val, v.dptr, v.dsize);
  98         free(v.dptr);
  99         return 1;
 100 }
 101
 102 int xtdb_del(struct db_conn *db, const unsigned char *key, size_t ksize)
 103 {
 104         TDB_DATA k;
 105
 106         k.dptr = (unsigned char *) key;
 107         k.dsize = ksize;
 108
 109         return tdb_delete(db->conn, k) == 0;
 110 }
 111
 112 int xtdb_firstkey(struct db_conn *db, unsigned char *key, size_t *ksize)
 113 {
 114         TDB_DATA k;
 115
 116         k = tdb_firstkey(db->conn);
 117         if (k.dptr == NULL)
 118                 return 0;
 119
 120         *ksize = k.dsize;
 121         memcpy(key, k.dptr, k.dsize);
 122         free(k.dptr);
 123         return 1;
 124 }
 125
 126 int xtdb_nextkey(struct db_conn *db,
 127                 const unsigned char *key, size_t ksize,
 128                 unsigned char *nextkey, size_t *nksize)
 129 {
 130         TDB_DATA pk, nk;
 131
 132         pk.dptr = (unsigned char *) key;
 133         pk.dsize = ksize;
 134
 135         nk = tdb_nextkey(db->conn, pk);
 136         if (nk.dptr == NULL)
 137                 return 0;
 138
 139         *nksize = nk.dsize;
 140         memcpy(nextkey, nk.dptr, nk.dsize);
 141         free(nk.dptr);
 142         return 1;
 143 }
 144
 145 #else
 146
 147 #include <stddef.h>     /* NULL */
 148
 149 struct db_conn *xtdb_open(const char *name, int flags)
 150 {
 151         return NULL;
 152 }
 153
 154 #endif
 155