source/tdb/tdbtool.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3 #include <fcntl.h>
   4 #include <unistd.h>
   5 #include <string.h>
   6 #include <fcntl.h>
   7 #include <sys/mman.h>
   8 #include <sys/stat.h>
   9 #include <sys/time.h>
  10 #include "tdb.h"
  11
  12 /* a tdb tool for manipulating a tdb database */
  13
  14 static TDB_CONTEXT *tdb;
  15
  16 static void help(void)
  17 {
  18         printf("
  19 tdbtool:
  20   create    dbname     : create a database
  21   open      dbname     : open an existing database
  22   erase                : erase the database
  23   dump                 : dump the database as strings
  24   insert    key  data  : insert a record
  25   store     key  data  : store a record (replace)
  26   show      key        : show a record by key
  27   delete    key        : delete a record by key
  28 ");
  29 }
  30
  31 static void terror(char *why)
  32 {
  33         printf("%s\n", why);
  34 }
  35
  36 static void create_tdb(void)
  37 {
  38         char *tok = strtok(NULL, " ");
  39         if (!tok) {
  40                 help();
  41                 return;
  42         }
  43         if (tdb) tdb_close(tdb);
  44         tdb = tdb_open(tok, 0, TDB_CLEAR_IF_FIRST,
  45                        O_RDWR | O_CREAT | O_TRUNC, 0600);
  46 }
  47
  48 static void open_tdb(void)
  49 {
  50         char *tok = strtok(NULL, " ");
  51         if (!tok) {
  52                 help();
  53                 return;
  54         }
  55         if (tdb) tdb_close(tdb);
  56         tdb = tdb_open(tok, 0, 0, O_RDWR, 0600);
  57 }
  58
  59 static void insert_tdb(void)
  60 {
  61         char *k = strtok(NULL, " ");
  62         char *d = strtok(NULL, " ");
  63         TDB_DATA key, dbuf;
  64
  65         if (!k || !d) {
  66                 help();
  67                 return;
  68         }
  69
  70         key.dptr = k;
  71         key.dsize = strlen(k);
  72         dbuf.dptr = d;
  73         dbuf.dsize = strlen(d);
  74
  75         if (tdb_store(tdb, key, dbuf, TDB_INSERT) == -1) {
  76                 terror("insert failed");
  77         }
  78 }
  79
  80 static void store_tdb(void)
  81 {
  82         char *k = strtok(NULL, " ");
  83         char *d = strtok(NULL, " ");
  84         TDB_DATA key, dbuf;
  85
  86         if (!k || !d) {
  87                 help();
  88                 return;
  89         }
  90
  91         key.dptr = k;
  92         key.dsize = strlen(k);
  93         dbuf.dptr = d;
  94         dbuf.dsize = strlen(d);
  95
  96         if (tdb_store(tdb, key, dbuf, TDB_REPLACE) == -1) {
  97                 terror("store failed");
  98         }
  99 }
 100
 101 static void show_tdb(void)
 102 {
 103         char *k = strtok(NULL, " ");
 104         TDB_DATA key, dbuf;
 105
 106         if (!k) {
 107                 help();
 108                 return;
 109         }
 110
 111         key.dptr = k;
 112         key.dsize = strlen(k);
 113
 114         dbuf = tdb_fetch(tdb, key);
 115         if (!dbuf.dptr) terror("fetch failed");
 116         printf("%s : %*.*s\n", k, (int)dbuf.dsize, (int)dbuf.dsize, dbuf.dptr);
 117 }
 118
 119 static void delete_tdb(void)
 120 {
 121         char *k = strtok(NULL, " ");
 122         TDB_DATA key;
 123
 124         if (!k) {
 125                 help();
 126                 return;
 127         }
 128
 129         key.dptr = k;
 130         key.dsize = strlen(k);
 131
 132         if (tdb_delete(tdb, key) != 0) {
 133                 terror("delete failed");
 134         }
 135 }
 136
 137 static int print_rec(TDB_CONTEXT *tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
 138 {
 139         printf("%*.*s : %*.*s\n",
 140                (int)key.dsize, (int)key.dsize, key.dptr,
 141                (int)dbuf.dsize, (int)dbuf.dsize, dbuf.dptr);
 142         return 0;
 143 }
 144
 145 static int total_bytes;
 146
 147 static int traverse_fn(TDB_CONTEXT *tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
 148 {
 149         total_bytes += dbuf.dsize;
 150         return 0;
 151 }
 152
 153 static void info_tdb(void)
 154 {
 155         int count;
 156         total_bytes = 0;
 157         count = tdb_traverse(tdb, traverse_fn, NULL);
 158         printf("%d records totalling %d bytes\n", count, total_bytes);
 159 }
 160
 161 static char *getline(char *prompt)
 162 {
 163         static char line[1024];
 164         char *p;
 165         fputs(prompt, stdout);
 166         line[0] = 0;
 167         p = fgets(line, sizeof(line)-1, stdin);
 168         if (p) p = strchr(p, '\n');
 169         if (p) *p = 0;
 170         return p?line:NULL;
 171 }
 172
 173 static int do_delete_fn(TDB_CONTEXT *tdb, TDB_DATA key, TDB_DATA dbuf,
 174                      void *state)
 175 {
 176     return tdb_delete(tdb, key);
 177 }
 178
 179 int main(int argc, char *argv[])
 180 {
 181     char *line;
 182     char *tok;
 183
 184     while ((line = getline("tdb> "))) {
 185
 186         /* Shell command */
 187
 188         if (line[0] == '!') {
 189             system(line + 1);
 190             continue;
 191         }
 192
 193         tok = strtok(line," ");
 194         if (strcmp(tok,"create") == 0) {
 195             create_tdb();
 196             continue;
 197         } else if (strcmp(tok,"open") == 0) {
 198             open_tdb();
 199             continue;
 200         }
 201
 202         /* all the rest require a open database */
 203         if (!tdb) {
 204             terror("database not open");
 205             help();
 206             continue;
 207         }
 208
 209         if (strcmp(tok,"insert") == 0) {
 210             insert_tdb();
 211         } else if (strcmp(tok,"store") == 0) {
 212             store_tdb();
 213         } else if (strcmp(tok,"show") == 0) {
 214             show_tdb();
 215         } else if (strcmp(tok,"erase") == 0) {
 216             tdb_traverse(tdb, do_delete_fn, NULL);
 217         } else if (strcmp(tok,"delete") == 0) {
 218             delete_tdb();
 219         } else if (strcmp(tok,"dump") == 0) {
 220             tdb_traverse(tdb, print_rec, NULL);
 221         } else if (strcmp(tok,"info") == 0) {
 222             info_tdb();
 223         } else {
 224             help();
 225         }
 226     }
 227
 228     if (tdb) tdb_close(tdb);
 229
 230     return 0;
 231 }