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