/************************************************************************************************* * The test cases of the B+ tree database API *                                                      Copyright (C) 2006-2008 Mikio Hirabayashi * This file is part of Tokyo Cabinet. * Tokyo Cabinet is free software; you can redistribute it and/or modify it under the terms of * the GNU Lesser General Public License as published by the Free Software Foundation; either * version 2.1 of the License or any later version.  Tokyo Cabinet is distributed in the hope * that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public * License for more details. * You should have received a copy of the GNU Lesser General Public License along with Tokyo * Cabinet; if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, * Boston, MA 02111-1307 USA. *************************************************************************************************/#include <tcutil.h>#include <tcbdb.h>#include "myconf.h"#define RECBUFSIZ      32                // buffer for records/* global variables */const char *g_progname;                  // program nameint g_dbgfd;                             // debugging output/* function prototypes */int main(int argc, char **argv);static void usage(void);static void iprintf(const char *format, ...);static void iputchar(int c);static void eprint(TCBDB *bdb, const char *func);static void mprint(TCBDB *bdb);static int myrand(int range);static int runwrite(int argc, char **argv);static int runread(int argc, char **argv);static int runremove(int argc, char **argv);static int runrcat(int argc, char **argv);static int runqueue(int argc, char **argv);static int runmisc(int argc, char **argv);static int runwicked(int argc, char **argv);static int procwrite(const char *path, int rnum, int lmemb, int nmemb, int bnum,                     int apow, int fpow, bool mt, BDBCMP cmp, int opts, int lcnum, int ncnum,                     int xmsiz, int lsmax, int capnum, int omode, bool rnd);static int procread(const char *path, bool mt, BDBCMP cmp, int lcnum, int ncnum, int xmsiz,                    int omode, bool wb, bool rnd);static int procremove(const char *path, bool mt, BDBCMP cmp, int lcnum, int ncnum, int xmsiz,                      int omode, bool rnd);static int procrcat(const char *path, int rnum,                    int lmemb, int nmemb, int bnum, int apow, int fpow,                    bool mt, BDBCMP cmp, int opts, int lcnum, int ncnum, int xmsiz,                    int lsmax, int capnum, int omode, int pnum, bool dai, bool dad, bool rl);static int procqueue(const char *path, int rnum, int lmemb, int nmemb, int bnum,                     int apow, int fpow, bool mt, BDBCMP cmp, int opts,                     int lcnum, int ncnum, int xmsiz, int lsmax, int capnum, int omode);static int procmisc(const char *path, int rnum, bool mt, int opts, int omode);static int procwicked(const char *path, int rnum, bool mt, int opts, int omode);/* main routine */int main(int argc, char **argv){  g_progname = argv[0];  g_dbgfd = -1;  const char *ebuf = getenv("TCDBGFD");  if(ebuf) g_dbgfd = tcatoi(ebuf);  srand((unsigned int)(tctime() * 1000) % UINT_MAX);  if(argc < 2) usage();  int rv = 0;  if(!strcmp(argv[1], "write")){    rv = runwrite(argc, argv);  } else if(!strcmp(argv[1], "read")){    rv = runread(argc, argv);  } else if(!strcmp(argv[1], "remove")){    rv = runremove(argc, argv);  } else if(!strcmp(argv[1], "rcat")){    rv = runrcat(argc, argv);  } else if(!strcmp(argv[1], "queue")){    rv = runqueue(argc, argv);  } else if(!strcmp(argv[1], "misc")){    rv = runmisc(argc, argv);  } else if(!strcmp(argv[1], "wicked")){    rv = runwicked(argc, argv);  } else {    usage();  }  return rv;}/* print the usage and exit */static void usage(void){  fprintf(stderr, "%s: test cases of the B+ tree database API of Tokyo Cabinet\n", g_progname);  fprintf(stderr, "\n");  fprintf(stderr, "usage:\n");  fprintf(stderr, "  %s write [-mt] [-cd|-ci|-cj] [-tl] [-td|-tb|-tt|-tx] [-lc num] [-nc num]"          " [-xm num] [-ls num] [-ca num] [-nl|-nb] [-rnd] path rnum"          " [lmemb [nmemb [bnum [apow [fpow]]]]]\n", g_progname);  fprintf(stderr, "  %s read [-mt] [-cd|-ci|-cj] [-lc num] [-nc num] [-xm num] [-nl|-nb]"          " [-wb] [-rnd] path\n", g_progname);  fprintf(stderr, "  %s remove [-mt] [-cd|-ci|-cj] [-lc num] [-nc num] [-xm num] [-nl|-nb] [-rnd]"          " path\n", g_progname);  fprintf(stderr, "  %s rcat [-mt] [-cd|-ci|-cj] [-tl] [-td|-tb|-tt|-tx] [-lc num] [-nc num]"          " [-xm num] [-ls num] [-ca num] [-nl|-nb] [-pn num] [-dai|-dad|-rl] path rnum"          " [lmemb [nmemb [bnum [apow [fpow]]]]]\n", g_progname);  fprintf(stderr, "  %s queue [-mt] [-cd|-ci|-cj] [-tl] [-td|-tb|-tt|-tx] [-lc num] [-nc num]"          " [-xm num] [-ls num] [-ca num] [-nl|-nb] path rnum"          " [lmemb [nmemb [bnum [apow [fpow]]]]]\n", g_progname);  fprintf(stderr, "  %s misc [-mt] [-tl] [-td|-tb|-tt|-tx] [-nl|-nb] path rnum\n", g_progname);  fprintf(stderr, "  %s wicked [-mt] [-tl] [-td|-tb|-tt|-tx] [-nl|-nb] path rnum\n", g_progname);  fprintf(stderr, "\n");  exit(1);}/* print formatted information string and flush the buffer */static void iprintf(const char *format, ...){  va_list ap;  va_start(ap, format);  vprintf(format, ap);  fflush(stdout);  va_end(ap);}/* print a character and flush the buffer */static void iputchar(int c){  putchar(c);  fflush(stdout);}/* print error message of hash database */static void eprint(TCBDB *bdb, const char *func){  const char *path = tcbdbpath(bdb);  int ecode = tcbdbecode(bdb);  fprintf(stderr, "%s: %s: %s: error: %d: %s\n",          g_progname, path ? path : "-", func, ecode, tcbdberrmsg(ecode));}/* print members of B+ tree database */static void mprint(TCBDB *bdb){  if(bdb->hdb->cnt_writerec < 0) return;  iprintf("max leaf member: %d\n", tcbdblmemb(bdb));  iprintf("max node member: %d\n", tcbdbnmemb(bdb));  iprintf("leaf number: %d\n", tcbdblnum(bdb));  iprintf("node number: %d\n", tcbdbnnum(bdb));  iprintf("bucket number: %lld\n", (long long)tcbdbbnum(bdb));  iprintf("used bucket number: %lld\n", (long long)tcbdbbnumused(bdb));  iprintf("cnt_saveleaf: %lld\n", (long long)bdb->cnt_saveleaf);  iprintf("cnt_loadleaf: %lld\n", (long long)bdb->cnt_loadleaf);  iprintf("cnt_killleaf: %lld\n", (long long)bdb->cnt_killleaf);  iprintf("cnt_adjleafc: %lld\n", (long long)bdb->cnt_adjleafc);  iprintf("cnt_savenode: %lld\n", (long long)bdb->cnt_savenode);  iprintf("cnt_loadnode: %lld\n", (long long)bdb->cnt_loadnode);  iprintf("cnt_adjnodec: %lld\n", (long long)bdb->cnt_adjnodec);  iprintf("cnt_writerec: %lld\n", (long long)bdb->hdb->cnt_writerec);  iprintf("cnt_reuserec: %lld\n", (long long)bdb->hdb->cnt_reuserec);  iprintf("cnt_moverec: %lld\n", (long long)bdb->hdb->cnt_moverec);  iprintf("cnt_readrec: %lld\n", (long long)bdb->hdb->cnt_readrec);  iprintf("cnt_searchfbp: %lld\n", (long long)bdb->hdb->cnt_searchfbp);  iprintf("cnt_insertfbp: %lld\n", (long long)bdb->hdb->cnt_insertfbp);  iprintf("cnt_splicefbp: %lld\n", (long long)bdb->hdb->cnt_splicefbp);  iprintf("cnt_dividefbp: %lld\n", (long long)bdb->hdb->cnt_dividefbp);  iprintf("cnt_mergefbp: %lld\n", (long long)bdb->hdb->cnt_mergefbp);  iprintf("cnt_reducefbp: %lld\n", (long long)bdb->hdb->cnt_reducefbp);  iprintf("cnt_appenddrp: %lld\n", (long long)bdb->hdb->cnt_appenddrp);  iprintf("cnt_deferdrp: %lld\n", (long long)bdb->hdb->cnt_deferdrp);  iprintf("cnt_flushdrp: %lld\n", (long long)bdb->hdb->cnt_flushdrp);  iprintf("cnt_adjrecc: %lld\n", (long long)bdb->hdb->cnt_adjrecc);}/* get a random number */static int myrand(int range){  return (int)((double)range * rand() / (RAND_MAX + 1.0));}/* parse arguments of write command */static int runwrite(int argc, char **argv){  char *path = NULL;  char *rstr = NULL;  char *lmstr = NULL;  char *nmstr = NULL;  char *bstr = NULL;  char *astr = NULL;  char *fstr = NULL;  bool mt = false;  BDBCMP cmp = NULL;  int opts = 0;  int lcnum = 0;  int ncnum = 0;  int xmsiz = 0;  int lsmax = 0;  int capnum = 0;  int omode = 0;  bool rnd = false;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      if(!strcmp(argv[i], "-mt")){        mt = true;      } else if(!strcmp(argv[i], "-cd")){        cmp = tcbdbcmpdecimal;      } else if(!strcmp(argv[i], "-ci")){        cmp = tcbdbcmpint32;      } else if(!strcmp(argv[i], "-cj")){        cmp = tcbdbcmpint64;      } else if(!strcmp(argv[i], "-tl")){        opts |= BDBTLARGE;      } else if(!strcmp(argv[i], "-td")){        opts |= BDBTDEFLATE;      } else if(!strcmp(argv[i], "-tb")){        opts |= BDBTBZIP;      } else if(!strcmp(argv[i], "-tt")){        opts |= BDBTTCBS;      } else if(!strcmp(argv[i], "-tx")){        opts |= BDBTEXCODEC;      } else if(!strcmp(argv[i], "-lc")){        if(++i >= argc) usage();        lcnum = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-nc")){        if(++i >= argc) usage();        ncnum = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-xm")){        if(++i >= argc) usage();        xmsiz = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-ls")){        if(++i >= argc) usage();        lsmax = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-ca")){        if(++i >= argc) usage();        capnum = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-nl")){        omode |= BDBONOLCK;      } else if(!strcmp(argv[i], "-nb")){        omode |= BDBOLCKNB;      } else if(!strcmp(argv[i], "-rnd")){        rnd = true;      } else {        usage();      }    } else if(!path){      path = argv[i];    } else if(!rstr){      rstr = argv[i];    } else if(!lmstr){      lmstr = argv[i];    } else if(!nmstr){      nmstr = argv[i];    } else if(!bstr){      bstr = argv[i];    } else if(!astr){      astr = argv[i];    } else if(!fstr){      fstr = argv[i];    } else {      usage();    }  }  if(!path || !rstr) usage();  int rnum = tcatoi(rstr);  if(rnum < 1) usage();  int lmemb = lmstr ? tcatoi(lmstr) : -1;  int nmemb = nmstr ? tcatoi(nmstr) : -1;  int bnum = bstr ? tcatoi(bstr) : -1;  int apow = astr ? tcatoi(astr) : -1;  int fpow = fstr ? tcatoi(fstr) : -1;  int rv = procwrite(path, rnum, lmemb, nmemb, bnum, apow, fpow,                     mt, cmp, opts, lcnum, ncnum, xmsiz, lsmax, capnum, omode, rnd);  return rv;}/* parse arguments of read command */static int runread(int argc, char **argv){  char *path = NULL;  bool mt = false;  BDBCMP cmp = NULL;  int lcnum = 0;  int ncnum = 0;  int xmsiz = 0;  int omode = 0;  bool wb = false;  bool rnd = false;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      if(!strcmp(argv[i], "-mt")){        mt = true;      } else if(!strcmp(argv[i], "-cd")){        cmp = tcbdbcmpdecimal;      } else if(!strcmp(argv[i], "-ci")){        cmp = tcbdbcmpint32;      } else if(!strcmp(argv[i], "-cj")){        cmp = tcbdbcmpint64;      } else if(!strcmp(argv[i], "-lc")){        if(++i >= argc) usage();        lcnum = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-nc")){        if(++i >= argc) usage();        ncnum = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-xm")){        if(++i >= argc) usage();        xmsiz = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-nl")){        omode |= BDBONOLCK;      } else if(!strcmp(argv[i], "-nb")){        omode |= BDBOLCKNB;      } else if(!strcmp(argv[i], "-wb")){        wb = true;      } else if(!strcmp(argv[i], "-rnd")){        rnd = true;      } else {        usage();      }    } else if(!path){      path = argv[i];    } else {      usage();    }  }  if(!path) usage();  int rv = procread(path, mt, cmp, lcnum, ncnum, xmsiz, omode, wb, rnd);  return rv;}/* parse arguments of remove command */static int runremove(int argc, char **argv){  char *path = NULL;  bool mt = false;  BDBCMP cmp = NULL;  int lcnum = 0;  int ncnum = 0;  int xmsiz = 0;  int omode = 0;  bool rnd = false;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      if(!strcmp(argv[i], "-mt")){        mt = true;      } else if(!strcmp(argv[i], "-cd")){        cmp = tcbdbcmpdecimal;      } else if(!strcmp(argv[i], "-ci")){        cmp = tcbdbcmpint32;      } else if(!strcmp(argv[i], "-cj")){        cmp = tcbdbcmpint64;      } else if(!strcmp(argv[i], "-lc")){        if(++i >= argc) usage();        lcnum = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-nc")){        if(++i >= argc) usage();        ncnum = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-xm")){        if(++i >= argc) usage();        xmsiz = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-nl")){        omode |= BDBONOLCK;      } else if(!strcmp(argv[i], "-nb")){        omode |= BDBOLCKNB;      } else if(!strcmp(argv[i], "-rnd")){        rnd = true;      } else {        usage();      }    } else if(!path){      path = argv[i];    } else {      usage();    }  }  if(!path) usage();  int rv = procremove(path, mt, cmp, lcnum, ncnum, xmsiz, omode, rnd);  return rv;}/* parse arguments of rcat command */static int runrcat(int argc, char **argv){  char *path = NULL;  char *rstr = NULL;  char *lmstr = NULL;  char *nmstr = NULL;  char *bstr = NULL;  char *astr = NULL;  char *fstr = NULL;  bool mt = false;  BDBCMP cmp = NULL;  int opts = 0;  int lcnum = 0;  int ncnum = 0;  int xmsiz = 0;  int lsmax = 0;  int capnum = 0;  int omode = 0;  int pnum = 0;  bool dai = false;  bool dad = false;  bool rl = false;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      if(!strcmp(argv[i], "-mt")){        mt = true;      } else if(!strcmp(argv[i], "-cd")){        cmp = tcbdbcmpdecimal;      } else if(!strcmp(argv[i], "-ci")){        cmp = tcbdbcmpint32;      } else if(!strcmp(argv[i], "-cj")){        cmp = tcbdbcmpint64;      } else if(!strcmp(argv[i], "-tl")){        opts |= BDBTLARGE;      } else if(!strcmp(argv[i], "-td")){        opts |= BDBTDEFLATE;      } else if(!strcmp(argv[i], "-tb")){        opts |= BDBTBZIP;      } else if(!strcmp(argv[i], "-tt")){        opts |= BDBTTCBS;      } else if(!strcmp(argv[i], "-tx")){        opts |= BDBTEXCODEC;      } else if(!strcmp(argv[i], "-lc")){        if(++i >= argc) usage();        lcnum = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-nc")){        if(++i >= argc) usage();        ncnum = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-xm")){        if(++i >= argc) usage();        xmsiz = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-ls")){