tdb: test for readonly locks mode on tdbbackup command
[samba.git] / lib / tdb / tools / tdbtool.c
1 /*
2    Unix SMB/CIFS implementation.
3    Samba database functions
4    Copyright (C) Andrew Tridgell              1999-2000
5    Copyright (C) Paul `Rusty' Russell              2000
6    Copyright (C) Jeremy Allison                    2000
7    Copyright (C) Andrew Esh                        2001
8
9    This program is free software; you can redistribute it and/or modify
10    it under the terms of the GNU General Public License as published by
11    the Free Software Foundation; either version 3 of the License, or
12    (at your option) any later version.
13
14    This program is distributed in the hope that it will be useful,
15    but WITHOUT ANY WARRANTY; without even the implied warranty of
16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17    GNU General Public License for more details.
18
19    You should have received a copy of the GNU General Public License
20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
21 */
22
23 #include "replace.h"
24 #include "system/locale.h"
25 #include "system/time.h"
26 #include "system/filesys.h"
27 #include "system/wait.h"
28 #include "tdb.h"
29
30 static int do_command(void);
31 const char *cmdname;
32 char *arg1, *arg2;
33 size_t arg1len, arg2len;
34 int bIterate = 0;
35 char *line;
36 TDB_DATA iterate_kbuf;
37 char cmdline[1024];
38 static int disable_mmap;
39 static int _disable_lock;
40
41 enum commands {
42         CMD_CREATE_TDB,
43         CMD_OPEN_TDB,
44         CMD_TRANSACTION_START,
45         CMD_TRANSACTION_COMMIT,
46         CMD_TRANSACTION_CANCEL,
47         CMD_ERASE,
48         CMD_DUMP,
49         CMD_INSERT,
50         CMD_MOVE,
51         CMD_STOREHEX,
52         CMD_STORE,
53         CMD_SHOW,
54         CMD_KEYS,
55         CMD_HEXKEYS,
56         CMD_DELETE,
57         CMD_LIST_HASH_FREE,
58         CMD_LIST_FREE,
59         CMD_FREELIST_SIZE,
60         CMD_INFO,
61         CMD_MMAP,
62         CMD_SPEED,
63         CMD_FIRST,
64         CMD_NEXT,
65         CMD_SYSTEM,
66         CMD_CHECK,
67         CMD_REPACK,
68         CMD_QUIT,
69         CMD_HELP
70 };
71
72 typedef struct {
73         const char *name;
74         enum commands cmd;
75 } COMMAND_TABLE;
76
77 COMMAND_TABLE cmd_table[] = {
78         {"create",      CMD_CREATE_TDB},
79         {"open",        CMD_OPEN_TDB},
80         {"transaction_start",   CMD_TRANSACTION_START},
81         {"transaction_commit",  CMD_TRANSACTION_COMMIT},
82         {"transaction_cancel",  CMD_TRANSACTION_CANCEL},
83         {"erase",       CMD_ERASE},
84         {"dump",        CMD_DUMP},
85         {"insert",      CMD_INSERT},
86         {"move",        CMD_MOVE},
87         {"storehex",    CMD_STOREHEX},
88         {"store",       CMD_STORE},
89         {"show",        CMD_SHOW},
90         {"keys",        CMD_KEYS},
91         {"hexkeys",     CMD_HEXKEYS},
92         {"delete",      CMD_DELETE},
93         {"list",        CMD_LIST_HASH_FREE},
94         {"free",        CMD_LIST_FREE},
95         {"freelist_size",       CMD_FREELIST_SIZE},
96         {"info",        CMD_INFO},
97         {"speed",       CMD_SPEED},
98         {"mmap",        CMD_MMAP},
99         {"first",       CMD_FIRST},
100         {"1",           CMD_FIRST},
101         {"next",        CMD_NEXT},
102         {"n",           CMD_NEXT},
103         {"check",       CMD_CHECK},
104         {"quit",        CMD_QUIT},
105         {"q",           CMD_QUIT},
106         {"!",           CMD_SYSTEM},
107         {"repack",      CMD_REPACK},
108         {NULL,          CMD_HELP}
109 };
110
111 struct timeval tp1,tp2;
112
113 static void _start_timer(void)
114 {
115         gettimeofday(&tp1,NULL);
116 }
117
118 static double _end_timer(void)
119 {
120         gettimeofday(&tp2,NULL);
121         return((tp2.tv_sec - tp1.tv_sec) +
122                (tp2.tv_usec - tp1.tv_usec)*1.0e-6);
123 }
124
125 #ifdef PRINTF_ATTRIBUTE
126 static void tdb_log_open(struct tdb_context *tdb, enum tdb_debug_level level,
127                          const char *format, ...) PRINTF_ATTRIBUTE(3,4);
128 #endif
129 static void tdb_log_open(struct tdb_context *tdb, enum tdb_debug_level level,
130                          const char *format, ...)
131 {
132         const char *mutex_msg =
133                 "Can use mutexes only with MUTEX_LOCKING or NOLOCK\n";
134         char *p;
135         va_list ap;
136
137         p = strstr(format, mutex_msg);
138         if (p != NULL) {
139                 /*
140                  * Yes, this is a hack, but we don't want to see this
141                  * message on first open, but we want to see
142                  * everything else.
143                  */
144                 return;
145         }
146
147         va_start(ap, format);
148         vfprintf(stderr, format, ap);
149         va_end(ap);
150 }
151
152 #ifdef PRINTF_ATTRIBUTE
153 static void tdb_log(struct tdb_context *tdb, enum tdb_debug_level level, const char *format, ...) PRINTF_ATTRIBUTE(3,4);
154 #endif
155 static void tdb_log(struct tdb_context *tdb, enum tdb_debug_level level, const char *format, ...)
156 {
157         va_list ap;
158
159         va_start(ap, format);
160         vfprintf(stderr, format, ap);
161         va_end(ap);
162 }
163
164 /* a tdb tool for manipulating a tdb database */
165
166 static TDB_CONTEXT *tdb;
167
168 static int print_rec(TDB_CONTEXT *the_tdb, TDB_DATA key, TDB_DATA dbuf, void *state);
169 static int print_key(TDB_CONTEXT *the_tdb, TDB_DATA key, TDB_DATA dbuf, void *state);
170 static int print_hexkey(TDB_CONTEXT *the_tdb, TDB_DATA key, TDB_DATA dbuf, void *state);
171
172 static void print_asc(const char *buf,int len)
173 {
174         int i;
175
176         /* We're probably printing ASCII strings so don't try to display
177            the trailing NULL character. */
178
179         if (buf[len - 1] == 0)
180                 len--;
181
182         for (i=0;i<len;i++)
183                 printf("%c",isprint(buf[i])?buf[i]:'.');
184 }
185
186 static void print_data(const char *buf,int len)
187 {
188         int i=0;
189         if (len<=0) return;
190         printf("[%03X] ",i);
191         for (i=0;i<len;) {
192                 printf("%02X ",(int)((unsigned char)buf[i]));
193                 i++;
194                 if (i%8 == 0) printf(" ");
195                 if (i%16 == 0) {
196                         print_asc(&buf[i-16],8); printf(" ");
197                         print_asc(&buf[i-8],8); printf("\n");
198                         if (i<len) printf("[%03X] ",i);
199                 }
200         }
201         if (i%16) {
202                 int n;
203
204                 n = 16 - (i%16);
205                 printf(" ");
206                 if (n>8) printf(" ");
207                 while (n--) printf("   ");
208
209                 n = i%16;
210                 if (n > 8) n = 8;
211                 print_asc(&buf[i-(i%16)],n); printf(" ");
212                 n = (i%16) - n;
213                 if (n>0) print_asc(&buf[i-n],n);
214                 printf("\n");
215         }
216 }
217
218 static void help(void)
219 {
220         printf("\n"
221 "tdbtool: \n"
222 "  create    dbname     : create a database\n"
223 "  open      dbname     : open an existing database\n"
224 "  transaction_start    : start a transaction\n"
225 "  transaction_commit   : commit a transaction\n"
226 "  transaction_cancel   : cancel a transaction\n"
227 "  erase                : erase the database\n"
228 "  dump                 : dump the database as strings\n"
229 "  keys                 : dump the database keys as strings\n"
230 "  hexkeys              : dump the database keys as hex values\n"
231 "  info                 : print summary info about the database\n"
232 "  insert    key  data  : insert a record\n"
233 "  move      key  file  : move a record to a destination tdb\n"
234 "  storehex  key  data  : store a record (replace), key/value in hex format\n"
235 "  store     key  data  : store a record (replace)\n"
236 "  show      key        : show a record by key\n"
237 "  delete    key        : delete a record by key\n"
238 "  list                 : print the database hash table and freelist\n"
239 "  free                 : print the database freelist\n"
240 "  freelist_size        : print the number of records in the freelist\n"
241 "  check                : check the integrity of an opened database\n"
242 "  repack               : repack the database\n"
243 "  speed                : perform speed tests on the database\n"
244 "  ! command            : execute system command\n"
245 "  1 | first            : print the first record\n"
246 "  n | next             : print the next record\n"
247 "  q | quit             : terminate\n"
248 "  \\n                   : repeat 'next' command\n"
249 "\n");
250 }
251
252 static void terror(const char *why)
253 {
254         printf("%s\n", why);
255 }
256
257 static void create_tdb(const char *tdbname)
258 {
259         struct tdb_logging_context log_ctx = { NULL, NULL};
260         log_ctx.log_fn = tdb_log;
261
262         if (tdb) tdb_close(tdb);
263         tdb = tdb_open_ex(tdbname, 0,
264                           TDB_CLEAR_IF_FIRST |
265                           (disable_mmap?TDB_NOMMAP:0) |
266                           (_disable_lock?TDB_NOLOCK:0),
267                           O_RDWR | O_CREAT | O_TRUNC, 0600, &log_ctx, NULL);
268         if (!tdb) {
269                 printf("Could not create %s: %s\n", tdbname, strerror(errno));
270         }
271 }
272
273 static void open_tdb(const char *tdbname)
274 {
275         struct tdb_logging_context log_ctx = { NULL, NULL };
276         log_ctx.log_fn = tdb_log_open;
277
278         if (tdb) tdb_close(tdb);
279         tdb = tdb_open_ex(tdbname, 0,
280                           (disable_mmap?TDB_NOMMAP:0) |
281                           (_disable_lock?TDB_NOLOCK:0),
282                           O_RDWR, 0600,
283                           &log_ctx, NULL);
284
285         log_ctx.log_fn = tdb_log;
286         if (tdb != NULL) {
287                 tdb_set_logging_function(tdb, &log_ctx);
288         }
289
290         if ((tdb == NULL) && (errno == EINVAL)) {
291                 /*
292                  * Retry NOLOCK and readonly. There we want to see all
293                  * error messages.
294                  */
295                 tdb = tdb_open_ex(tdbname, 0,
296                                   (disable_mmap?TDB_NOMMAP:0) |TDB_NOLOCK,
297                                   O_RDONLY, 0600,
298                                   &log_ctx, NULL);
299         }
300
301         if (!tdb) {
302                 printf("Could not open %s: %s\n", tdbname, strerror(errno));
303         }
304 }
305
306 static void insert_tdb(char *keyname, size_t keylen, char* data, size_t datalen)
307 {
308         TDB_DATA key, dbuf;
309
310         if ((keyname == NULL) || (keylen == 0)) {
311                 terror("need key");
312                 return;
313         }
314
315         key.dptr = (unsigned char *)keyname;
316         key.dsize = keylen;
317         dbuf.dptr = (unsigned char *)data;
318         dbuf.dsize = datalen;
319
320         if (tdb_store(tdb, key, dbuf, TDB_INSERT) != 0) {
321                 terror("insert failed");
322         }
323 }
324
325 static void store_tdb(char *keyname, size_t keylen, char* data, size_t datalen)
326 {
327         TDB_DATA key, dbuf;
328
329         if ((keyname == NULL) || (keylen == 0)) {
330                 terror("need key");
331                 return;
332         }
333
334         if ((data == NULL) || (datalen == 0)) {
335                 terror("need data");
336                 return;
337         }
338
339         key.dptr = (unsigned char *)keyname;
340         key.dsize = keylen;
341         dbuf.dptr = (unsigned char *)data;
342         dbuf.dsize = datalen;
343
344         printf("Storing key:\n");
345         print_rec(tdb, key, dbuf, NULL);
346
347         if (tdb_store(tdb, key, dbuf, TDB_REPLACE) != 0) {
348                 terror("store failed");
349         }
350 }
351
352 static bool hexchar(char c, uint8_t *v)
353 {
354         if ((c >= '0') && (c <= '9')) {
355                 *v = (c - '0');
356                 return true;
357         }
358         if ((c >= 'A') && (c <= 'F')) {
359                 *v = (c - 'A' + 10);
360                 return true;
361         }
362         if ((c >= 'a') && (c <= 'f')) {
363                 *v = (c - 'a' + 10);
364                 return true;
365         }
366         return false;
367 }
368
369 static bool parse_hex(const char *src, size_t srclen, uint8_t *dst)
370 {
371         size_t i=0;
372
373         if ((srclen % 2) != 0) {
374                 return false;
375         }
376
377         while (i<srclen) {
378                 bool ok;
379                 uint8_t hi,lo;
380
381                 ok = (hexchar(src[i++], &hi) && hexchar(src[i++], &lo));
382                 if (!ok) {
383                         return false;
384                 }
385                 *dst = (hi<<4)|lo;
386                 dst += 1;
387         }
388
389         return true;
390 }
391
392 static void store_hex_tdb(char *keystr, size_t keylen,
393                           char *datastr, size_t datalen)
394 {
395         if ((keystr == NULL) || (keylen == 0)) {
396                 terror("need key");
397                 return;
398         }
399         if ((datastr == NULL) || (datalen == 0)) {
400                 terror("need data");
401                 return;
402         }
403
404         {
405                 uint8_t keybuf[keylen/2];
406                 TDB_DATA key = { .dptr = keybuf, .dsize = sizeof(keybuf) };
407                 uint8_t databuf[datalen/2];
408                 TDB_DATA data = { .dptr = databuf, .dsize = sizeof(databuf) };
409                 bool ok;
410
411                 ok = parse_hex(keystr, keylen, keybuf);
412                 if (!ok) {
413                         terror("need hex key");
414                         return;
415                 }
416                 ok = parse_hex(datastr, datalen, databuf);
417                 if (!ok) {
418                         terror("need hex data");
419                         return;
420                 }
421
422                 printf("storing key/data:\n");
423                 print_data((char *)key.dptr, key.dsize);
424                 print_data((char *)data.dptr, data.dsize);
425
426                 if (tdb_store(tdb, key, data, TDB_REPLACE) != 0) {
427                         terror("store failed");
428                 }
429         }
430 }
431
432 static void show_tdb(char *keyname, size_t keylen)
433 {
434         TDB_DATA key, dbuf;
435
436         if ((keyname == NULL) || (keylen == 0)) {
437                 terror("need key");
438                 return;
439         }
440
441         key.dptr = (unsigned char *)keyname;
442         key.dsize = keylen;
443
444         dbuf = tdb_fetch(tdb, key);
445         if (!dbuf.dptr) {
446             terror("fetch failed");
447             return;
448         }
449
450         print_rec(tdb, key, dbuf, NULL);
451
452         free( dbuf.dptr );
453
454         return;
455 }
456
457 static void delete_tdb(char *keyname, size_t keylen)
458 {
459         TDB_DATA key;
460
461         if ((keyname == NULL) || (keylen == 0)) {
462                 terror("need key");
463                 return;
464         }
465
466         key.dptr = (unsigned char *)keyname;
467         key.dsize = keylen;
468
469         if (tdb_delete(tdb, key) != 0) {
470                 terror("delete failed");
471         }
472 }
473
474 static void move_rec(char *keyname, size_t keylen, char* tdbname)
475 {
476         TDB_DATA key, dbuf;
477         TDB_CONTEXT *dst_tdb;
478
479         if ((keyname == NULL) || (keylen == 0)) {
480                 terror("need key");
481                 return;
482         }
483
484         if ( !tdbname ) {
485                 terror("need destination tdb name");
486                 return;
487         }
488
489         key.dptr = (unsigned char *)keyname;
490         key.dsize = keylen;
491
492         dbuf = tdb_fetch(tdb, key);
493         if (!dbuf.dptr) {
494                 terror("fetch failed");
495                 return;
496         }
497
498         print_rec(tdb, key, dbuf, NULL);
499
500         dst_tdb = tdb_open(tdbname, 0, 0, O_RDWR, 0600);
501         if ( !dst_tdb ) {
502                 terror("unable to open destination tdb");
503                 return;
504         }
505
506         if (tdb_store( dst_tdb, key, dbuf, TDB_REPLACE ) != 0) {
507                 terror("failed to move record");
508         }
509         else
510                 printf("record moved\n");
511
512         tdb_close( dst_tdb );
513
514         return;
515 }
516
517 static int print_rec(TDB_CONTEXT *the_tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
518 {
519         printf("\nkey %d bytes\n", (int)key.dsize);
520         print_asc((const char *)key.dptr, key.dsize);
521         printf("\ndata %d bytes\n", (int)dbuf.dsize);
522         print_data((const char *)dbuf.dptr, dbuf.dsize);
523         return 0;
524 }
525
526 static int print_key(TDB_CONTEXT *the_tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
527 {
528         printf("key %d bytes: ", (int)key.dsize);
529         print_asc((const char *)key.dptr, key.dsize);
530         printf("\n");
531         return 0;
532 }
533
534 static int print_hexkey(TDB_CONTEXT *the_tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
535 {
536         printf("key %d bytes\n", (int)key.dsize);
537         print_data((const char *)key.dptr, key.dsize);
538         printf("\n");
539         return 0;
540 }
541
542 static int total_bytes;
543
544 static int traverse_fn(TDB_CONTEXT *the_tdb, TDB_DATA key, TDB_DATA dbuf, void *state)
545 {
546         total_bytes += dbuf.dsize;
547         return 0;
548 }
549
550 static void info_tdb(void)
551 {
552         char *summary = tdb_summary(tdb);
553
554         if (!summary) {
555                 printf("Error = %s\n", tdb_errorstr(tdb));
556         } else {
557                 printf("%s", summary);
558                 free(summary);
559         }
560 }
561
562 static void speed_tdb(const char *tlimit)
563 {
564         const char *str = "store test", *str2 = "transaction test";
565         unsigned timelimit = tlimit?atoi(tlimit):0;
566         double t;
567         int ops;
568         if (timelimit == 0) timelimit = 5;
569
570         ops = 0;
571         printf("Testing store speed for %u seconds\n", timelimit);
572         _start_timer();
573         do {
574                 long int r = random();
575                 TDB_DATA key, dbuf;
576                 key.dptr = discard_const_p(uint8_t, str);
577                 key.dsize = strlen((char *)key.dptr);
578                 dbuf.dptr = (uint8_t *) &r;
579                 dbuf.dsize = sizeof(r);
580                 tdb_store(tdb, key, dbuf, TDB_REPLACE);
581                 t = _end_timer();
582                 ops++;
583         } while (t < timelimit);
584         printf("%10.3f ops/sec\n", ops/t);
585
586         ops = 0;
587         printf("Testing fetch speed for %u seconds\n", timelimit);
588         _start_timer();
589         do {
590                 TDB_DATA key;
591                 key.dptr = discard_const_p(uint8_t, str);
592                 key.dsize = strlen((char *)key.dptr);
593                 tdb_fetch(tdb, key);
594                 t = _end_timer();
595                 ops++;
596         } while (t < timelimit);
597         printf("%10.3f ops/sec\n", ops/t);
598
599         ops = 0;
600         printf("Testing transaction speed for %u seconds\n", timelimit);
601         _start_timer();
602         do {
603                 long int r = random();
604                 TDB_DATA key, dbuf;
605                 key.dptr = discard_const_p(uint8_t, str2);
606                 key.dsize = strlen((char *)key.dptr);
607                 dbuf.dptr = (uint8_t *) &r;
608                 dbuf.dsize = sizeof(r);
609                 tdb_transaction_start(tdb);
610                 tdb_store(tdb, key, dbuf, TDB_REPLACE);
611                 tdb_transaction_commit(tdb);
612                 t = _end_timer();
613                 ops++;
614         } while (t < timelimit);
615         printf("%10.3f ops/sec\n", ops/t);
616
617         ops = 0;
618         printf("Testing traverse speed for %u seconds\n", timelimit);
619         _start_timer();
620         do {
621                 tdb_traverse(tdb, traverse_fn, NULL);
622                 t = _end_timer();
623                 ops++;
624         } while (t < timelimit);
625         printf("%10.3f ops/sec\n", ops/t);
626 }
627
628 static void toggle_mmap(void)
629 {
630         disable_mmap = !disable_mmap;
631         if (disable_mmap) {
632                 printf("mmap is disabled\n");
633         } else {
634                 printf("mmap is enabled\n");
635         }
636 }
637
638 static char *tdb_getline(const char *prompt)
639 {
640         static char thisline[1024];
641         char *p;
642         fputs(prompt, stdout);
643         thisline[0] = 0;
644         p = fgets(thisline, sizeof(thisline)-1, stdin);
645         if (p) p = strchr(p, '\n');
646         if (p) *p = 0;
647         return p?thisline:NULL;
648 }
649
650 static void first_record(TDB_CONTEXT *the_tdb, TDB_DATA *pkey)
651 {
652         TDB_DATA dbuf;
653         *pkey = tdb_firstkey(the_tdb);
654
655         dbuf = tdb_fetch(the_tdb, *pkey);
656         if (!dbuf.dptr) terror("fetch failed");
657         else {
658                 print_rec(the_tdb, *pkey, dbuf, NULL);
659         }
660 }
661
662 static void next_record(TDB_CONTEXT *the_tdb, TDB_DATA *pkey)
663 {
664         TDB_DATA dbuf;
665         *pkey = tdb_nextkey(the_tdb, *pkey);
666
667         dbuf = tdb_fetch(the_tdb, *pkey);
668         if (!dbuf.dptr)
669                 terror("fetch failed");
670         else
671                 print_rec(the_tdb, *pkey, dbuf, NULL);
672 }
673
674 static int count(TDB_DATA key, TDB_DATA data, void *private_data)
675 {
676         (*(unsigned int *)private_data)++;
677         return 0;
678 }
679
680 static void check_db(TDB_CONTEXT *the_tdb)
681 {
682         int tdbcount = 0;
683         if (!the_tdb)
684                 printf("Error: No database opened!\n");
685         else if (tdb_check(the_tdb, count, &tdbcount) == -1)
686                 printf("Integrity check for the opened database failed.\n");
687         else
688                 printf("Database integrity is OK and has %d records.\n",
689                        tdbcount);
690 }
691
692 static int do_command(void)
693 {
694         COMMAND_TABLE *ctp = cmd_table;
695         enum commands mycmd = CMD_HELP;
696         int cmd_len;
697
698         if (cmdname && strlen(cmdname) == 0) {
699                 mycmd = CMD_NEXT;
700         } else {
701                 while (ctp->name) {
702                         cmd_len = strlen(ctp->name);
703                         if (strncmp(ctp->name,cmdname,cmd_len) == 0) {
704                                 mycmd = ctp->cmd;
705                                 break;
706                         }
707                         ctp++;
708                 }
709         }
710
711         switch (mycmd) {
712         case CMD_CREATE_TDB:
713                 bIterate = 0;
714                 create_tdb(arg1);
715                 return 0;
716         case CMD_OPEN_TDB:
717                 bIterate = 0;
718                 open_tdb(arg1);
719                 return 0;
720         case CMD_SYSTEM:
721                 /* Shell command */
722                 if (system(arg1) == -1) {
723                         terror("system() call failed\n");
724                 }
725                 return 0;
726         case CMD_QUIT:
727                 return 1;
728         default:
729                 /* all the rest require a open database */
730                 if (!tdb) {
731                         bIterate = 0;
732                         terror("database not open");
733                         help();
734                         return 0;
735                 }
736                 switch (mycmd) {
737                 case CMD_TRANSACTION_START:
738                         bIterate = 0;
739                         tdb_transaction_start(tdb);
740                         return 0;
741                 case CMD_TRANSACTION_COMMIT:
742                         bIterate = 0;
743                         tdb_transaction_commit(tdb);
744                         return 0;
745                 case CMD_REPACK:
746                         bIterate = 0;
747                         tdb_repack(tdb);
748                         return 0;
749                 case CMD_TRANSACTION_CANCEL:
750                         bIterate = 0;
751                         tdb_transaction_cancel(tdb);
752                         return 0;
753                 case CMD_ERASE:
754                         bIterate = 0;
755                         tdb_wipe_all(tdb);
756                         return 0;
757                 case CMD_DUMP:
758                         bIterate = 0;
759                         tdb_traverse(tdb, print_rec, NULL);
760                         return 0;
761                 case CMD_INSERT:
762                         bIterate = 0;
763                         insert_tdb(arg1, arg1len,arg2,arg2len);
764                         return 0;
765                 case CMD_MOVE:
766                         bIterate = 0;
767                         move_rec(arg1,arg1len,arg2);
768                         return 0;
769                 case CMD_STORE:
770                         bIterate = 0;
771                         store_tdb(arg1,arg1len,arg2,arg2len);
772                         return 0;
773                 case CMD_STOREHEX:
774                         bIterate = 0;
775                         store_hex_tdb(arg1,arg1len,arg2,arg2len);
776                         return 0;
777                 case CMD_SHOW:
778                         bIterate = 0;
779                         show_tdb(arg1, arg1len);
780                         return 0;
781                 case CMD_KEYS:
782                         tdb_traverse(tdb, print_key, NULL);
783                         return 0;
784                 case CMD_HEXKEYS:
785                         tdb_traverse(tdb, print_hexkey, NULL);
786                         return 0;
787                 case CMD_DELETE:
788                         bIterate = 0;
789                         delete_tdb(arg1,arg1len);
790                         return 0;
791                 case CMD_LIST_HASH_FREE:
792                         tdb_dump_all(tdb);
793                         return 0;
794                 case CMD_LIST_FREE:
795                         tdb_printfreelist(tdb);
796                         return 0;
797                 case CMD_FREELIST_SIZE: {
798                         int size;
799
800                         size = tdb_freelist_size(tdb);
801                         if (size < 0) {
802                                 printf("Error getting freelist size.\n");
803                         } else {
804                                 printf("freelist size: %d\n", size);
805                         }
806
807                         return 0;
808                 }
809                 case CMD_INFO:
810                         info_tdb();
811                         return 0;
812                 case CMD_SPEED:
813                         speed_tdb(arg1);
814                         return 0;
815                 case CMD_MMAP:
816                         toggle_mmap();
817                         return 0;
818                 case CMD_FIRST:
819                         bIterate = 1;
820                         first_record(tdb, &iterate_kbuf);
821                         return 0;
822                 case CMD_NEXT:
823                         if (bIterate)
824                                 next_record(tdb, &iterate_kbuf);
825                         return 0;
826                 case CMD_CHECK:
827                         check_db(tdb);
828                         return 0;
829                 case CMD_HELP:
830                         help();
831                         return 0;
832                 case CMD_CREATE_TDB:
833                 case CMD_OPEN_TDB:
834                 case CMD_SYSTEM:
835                 case CMD_QUIT:
836                         /*
837                          * unhandled commands.  cases included here to avoid compiler
838                          * warnings.
839                          */
840                         return 0;
841                 }
842         }
843
844         return 0;
845 }
846
847 static char *tdb_convert_string(char *instring, size_t *sizep)
848 {
849         size_t length = 0;
850         char *outp, *inp;
851         char temp[3];
852
853         outp = inp = instring;
854
855         while (*inp) {
856                 if (*inp == '\\') {
857                         inp++;
858                         if (*inp && strchr("0123456789abcdefABCDEF",(int)*inp)) {
859                                 temp[0] = *inp++;
860                                 temp[1] = '\0';
861                                 if (*inp && strchr("0123456789abcdefABCDEF",(int)*inp)) {
862                                         temp[1] = *inp++;
863                                         temp[2] = '\0';
864                                 }
865                                 *outp++ = (char)strtol((const char *)temp,NULL,16);
866                         } else {
867                                 *outp++ = *inp++;
868                         }
869                 } else {
870                         *outp++ = *inp++;
871                 }
872                 length++;
873         }
874         *sizep = length;
875         return instring;
876 }
877
878 int main(int argc, char *argv[])
879 {
880         cmdname = "";
881         arg1 = NULL;
882         arg1len = 0;
883         arg2 = NULL;
884         arg2len = 0;
885
886         if (argv[1] && (strcmp(argv[1], "-l") == 0)) {
887                 _disable_lock = 1;
888                 argv[1] = argv[0];
889                 argv += 1;
890                 argc -= 1;
891         }
892
893         if (argv[1]) {
894                 cmdname = "open";
895                 arg1 = argv[1];
896                 do_command();
897                 cmdname =  "";
898                 arg1 = NULL;
899         }
900
901         switch (argc) {
902         case 1:
903         case 2:
904                 /* Interactive mode */
905                 while ((cmdname = tdb_getline("tdb> "))) {
906                         arg2 = arg1 = NULL;
907                         if ((arg1 = strchr((const char *)cmdname,' ')) != NULL) {
908                                 arg1++;
909                                 arg2 = arg1;
910                                 while (*arg2) {
911                                         if (*arg2 == ' ') {
912                                                 *arg2++ = '\0';
913                                                 break;
914                                         }
915                                         if ((*arg2++ == '\\') && (*arg2 == ' ')) {
916                                                 arg2++;
917                                         }
918                                 }
919                         }
920                         if (arg1) arg1 = tdb_convert_string(arg1,&arg1len);
921                         if (arg2) arg2 = tdb_convert_string(arg2,&arg2len);
922                         if (do_command()) break;
923                 }
924                 break;
925         case 5:
926                 arg2 = tdb_convert_string(argv[4],&arg2len);
927                 FALL_THROUGH;
928         case 4:
929                 arg1 = tdb_convert_string(argv[3],&arg1len);
930                 FALL_THROUGH;
931         case 3:
932                 cmdname = argv[2];
933                 FALL_THROUGH;
934         default:
935                 do_command();
936                 break;
937         }
938
939         if (tdb) tdb_close(tdb);
940
941         return 0;
942 }