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