/************************************************************************************************* * The command line utility 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"/* 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 printerr(TCBDB *bdb);static int printdata(const char *ptr, int size, bool px);static char *mygetline(FILE *ifp);static int mycmpfunc(const char *aptr, int asiz, const char *bptr, int bsiz, void *op);static int runcreate(int argc, char **argv);static int runinform(int argc, char **argv);static int runput(int argc, char **argv);static int runout(int argc, char **argv);static int runget(int argc, char **argv);static int runlist(int argc, char **argv);static int runoptimize(int argc, char **argv);static int runimporttsv(int argc, char **argv);static int runversion(int argc, char **argv);static int proccreate(const char *path, int lmemb, int nmemb,                      int bnum, int apow, int fpow, BDBCMP cmp, int opts);static int procinform(const char *path, int omode);static int procput(const char *path, const char *kbuf, int ksiz, const char *vbuf, int vsiz,                   BDBCMP cmp, int omode, int dmode);static int procout(const char *path, const char *kbuf, int ksiz, BDBCMP cmp, int omode);static int procget(const char *path, const char *kbuf, int ksiz, BDBCMP cmp, int omode,                   bool px, bool pz);static int proclist(const char *path, BDBCMP cmp, int omode, int max, bool pv, bool px, bool bk,                    const char *jstr, const char *bstr, const char *estr, const char *fmstr);static int procoptimize(const char *path, int lmemb, int nmemb,                        int bnum, int apow, int fpow, BDBCMP cmp, int opts, int omode);static int procimporttsv(const char *path, const char *file, int omode, bool sc);static int procversion(void);/* 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);  if(argc < 2) usage();  int rv = 0;  if(!strcmp(argv[1], "create")){    rv = runcreate(argc, argv);  } else if(!strcmp(argv[1], "inform")){    rv = runinform(argc, argv);  } else if(!strcmp(argv[1], "put")){    rv = runput(argc, argv);  } else if(!strcmp(argv[1], "out")){    rv = runout(argc, argv);  } else if(!strcmp(argv[1], "get")){    rv = runget(argc, argv);  } else if(!strcmp(argv[1], "list")){    rv = runlist(argc, argv);  } else if(!strcmp(argv[1], "optimize")){    rv = runoptimize(argc, argv);  } else if(!strcmp(argv[1], "importtsv")){    rv = runimporttsv(argc, argv);  } else if(!strcmp(argv[1], "version") || !strcmp(argv[1], "--version")){    rv = runversion(argc, argv);  } else {    usage();  }  return rv;}/* print the usage and exit */static void usage(void){  fprintf(stderr, "%s: the command line utility of the B+ tree database API\n", g_progname);  fprintf(stderr, "\n");  fprintf(stderr, "usage:\n");  fprintf(stderr, "  %s create [-cd|-ci|-cj] [-tl] [-td|-tb|-tt|-tx] path"          " [lmemb [nmemb [bnum [apow [fpow]]]]]\n", g_progname);  fprintf(stderr, "  %s inform [-nl|-nb] path\n", g_progname);  fprintf(stderr, "  %s put [-cd|-ci|-cj] [-nl|-nb] [-sx] [-dk|-dc|-dd|-db] path"          " key value\n", g_progname);  fprintf(stderr, "  %s out [-cd|-ci|-cj] [-nl|-nb] [-sx] path key\n", g_progname);  fprintf(stderr, "  %s get [-cd|-ci|-cj] [-nl|-nb] [-sx] [-px] [-pz] path key\n", g_progname);  fprintf(stderr, "  %s list [-cd|-ci|-cj] [-nl|-nb] [-m num] [-bk] [-pv] [-px] [-j str]"          " [-rb bkey ekey] [-fm str] path\n", g_progname);  fprintf(stderr, "  %s optimize [-cd|-ci|-cj] [-tl] [-td|-tb|-tt|-tx] [-tz] [-nl|-nb] path"          " [lmemb [nmemb [bnum [apow [fpow]]]]]\n", g_progname);  fprintf(stderr, "  %s importtsv [-nl|-nb] [-sc] path [file]\n", g_progname);  fprintf(stderr, "  %s version\n", g_progname);  fprintf(stderr, "\n");  exit(1);}/* print error information */static void printerr(TCBDB *bdb){  const char *path = tcbdbpath(bdb);  int ecode = tcbdbecode(bdb);  fprintf(stderr, "%s: %s: %d: %s\n", g_progname, path ? path : "-", ecode, tcbdberrmsg(ecode));}/* print record data */static int printdata(const char *ptr, int size, bool px){  int len = 0;  while(size-- > 0){    if(px){      if(len > 0) putchar(' ');      len += printf("%02X", *(unsigned char *)ptr);    } else {      putchar(*ptr);      len++;    }    ptr++;  }  return len;}/* read a line from a file descriptor */static char *mygetline(FILE *ifp){  int len = 0;  int blen = 1024;  char *buf = tcmalloc(blen);  bool end = true;  int c;  while((c = fgetc(ifp)) != EOF){    end = false;    if(c == '\0') continue;    if(blen <= len){      blen *= 2;      buf = tcrealloc(buf, blen + 1);    }    if(c == '\n' || c == '\r') c = '\0';    buf[len++] = c;    if(c == '\0') break;  }  if(end){    tcfree(buf);    return NULL;  }  buf[len] = '\0';  return buf;}/* dummy comparison function */static int mycmpfunc(const char *aptr, int asiz, const char *bptr, int bsiz, void *op){  return 0;}/* parse arguments of create command */static int runcreate(int argc, char **argv){  char *path = NULL;  char *lmstr = NULL;  char *nmstr = NULL;  char *bstr = NULL;  char *astr = NULL;  char *fstr = NULL;  BDBCMP cmp = NULL;  int opts = 0;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      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 {        usage();      }    } else if(!path){      path = 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) 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 = proccreate(path, lmemb, nmemb, bnum, apow, fpow, cmp, opts);  return rv;}/* parse arguments of inform command */static int runinform(int argc, char **argv){  char *path = NULL;  int omode = 0;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      if(!strcmp(argv[i], "-nl")){        omode |= BDBONOLCK;      } else if(!strcmp(argv[i], "-nb")){        omode |= BDBOLCKNB;      } else {        usage();      }    } else if(!path){      path = argv[i];    } else {      usage();    }  }  if(!path) usage();  int rv = procinform(path, omode);  return rv;}/* parse arguments of put command */static int runput(int argc, char **argv){  char *path = NULL;  char *key = NULL;  char *value = NULL;  BDBCMP cmp = NULL;  int omode = 0;  int dmode = 0;  bool sx = false;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      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], "-nl")){        omode |= BDBONOLCK;      } else if(!strcmp(argv[i], "-nb")){        omode |= BDBOLCKNB;      } else if(!strcmp(argv[i], "-dk")){        dmode = -1;      } else if(!strcmp(argv[i], "-dc")){        dmode = 1;      } else if(!strcmp(argv[i], "-dd")){        dmode = 2;      } else if(!strcmp(argv[i], "-db")){        dmode = 3;      } else if(!strcmp(argv[i], "-sx")){        sx = true;      } else {        usage();      }    } else if(!path){      path = argv[i];    } else if(!key){      key = argv[i];    } else if(!value){      value = argv[i];    } else {      usage();    }  }  if(!path || !key || !value) usage();  int ksiz, vsiz;  char *kbuf, *vbuf;  if(sx){    kbuf = tchexdecode(key, &ksiz);    vbuf = tchexdecode(value, &vsiz);  } else {    ksiz = strlen(key);    kbuf = tcmemdup(key, ksiz);    vsiz = strlen(value);    vbuf = tcmemdup(value, vsiz);  }  int rv = procput(path, kbuf, ksiz, vbuf, vsiz, cmp, omode, dmode);  tcfree(vbuf);  tcfree(kbuf);  return rv;}/* parse arguments of out command */static int runout(int argc, char **argv){  char *path = NULL;  char *key = NULL;  BDBCMP cmp = NULL;  int omode = 0;  bool sx = false;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      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], "-nl")){        omode |= BDBONOLCK;      } else if(!strcmp(argv[i], "-nb")){        omode |= BDBOLCKNB;      } else if(!strcmp(argv[i], "-sx")){        sx = true;      } else {        usage();      }    } else if(!path){      path = argv[i];    } else if(!key){      key = argv[i];    } else {      usage();    }  }  if(!path || !key) usage();  int ksiz;  char *kbuf;  if(sx){    kbuf = tchexdecode(key, &ksiz);  } else {    ksiz = strlen(key);    kbuf = tcmemdup(key, ksiz);  }  int rv = procout(path, kbuf, ksiz, cmp, omode);  tcfree(kbuf);  return rv;}/* parse arguments of get command */static int runget(int argc, char **argv){  char *path = NULL;  char *key = NULL;  BDBCMP cmp = NULL;  int omode = 0;  bool sx = false;  bool px = false;  bool pz = false;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      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], "-nl")){        omode |= BDBONOLCK;      } else if(!strcmp(argv[i], "-nb")){        omode |= BDBOLCKNB;      } else if(!strcmp(argv[i], "-sx")){        sx = true;      } else if(!strcmp(argv[i], "-px")){        px = true;      } else if(!strcmp(argv[i], "-pz")){        pz = true;      } else {        usage();      }    } else if(!path){      path = argv[i];    } else if(!key){      key = argv[i];    } else {      usage();    }  }  if(!path || !key) usage();  int ksiz;  char *kbuf;  if(sx){    kbuf = tchexdecode(key, &ksiz);  } else {    ksiz = strlen(key);    kbuf = tcmemdup(key, ksiz);  }  int rv = procget(path, kbuf, ksiz, cmp, omode, px, pz);  tcfree(kbuf);  return rv;}/* parse arguments of list command */static int runlist(int argc, char **argv){  char *path = NULL;  BDBCMP cmp = NULL;  int omode = 0;  int max = -1;  bool pv = false;  bool px = false;  bool bk = false;  char *jstr = NULL;  char *bstr = NULL;  char *estr = NULL;  char *fmstr = NULL;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      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], "-nl")){        omode |= BDBONOLCK;      } else if(!strcmp(argv[i], "-nb")){        omode |= BDBOLCKNB;      } else if(!strcmp(argv[i], "-m")){        if(++i >= argc) usage();        max = tcatoi(argv[i]);      } else if(!strcmp(argv[i], "-bk")){        bk = true;      } else if(!strcmp(argv[i], "-pv")){        pv = true;      } else if(!strcmp(argv[i], "-px")){        px = true;      } else if(!strcmp(argv[i], "-j")){        if(++i >= argc) usage();        jstr = argv[i];      } else if(!strcmp(argv[i], "-rb")){        if(++i >= argc) usage();        bstr = argv[i];        if(++i >= argc) usage();        estr = argv[i];      } else if(!strcmp(argv[i], "-fm")){        if(++i >= argc) usage();        fmstr = argv[i];      } else {        usage();      }    } else if(!path){      path = argv[i];    } else {      usage();    }  }  if(!path) usage();  int rv = proclist(path, cmp, omode, max, pv, px, bk, jstr, bstr, estr, fmstr);  return rv;}/* parse arguments of optimize command */static int runoptimize(int argc, char **argv){  char *path = NULL;  char *lmstr = NULL;  char *nmstr = NULL;  char *bstr = NULL;  char *astr = NULL;  char *fstr = NULL;  BDBCMP cmp = NULL;  int opts = UINT8_MAX;  int omode = 0;  for(int i = 2; i < argc; i++){    if(!path && argv[i][0] == '-'){      if(!strcmp(argv[i], "-cd")){        cmp = tcbdbcmpdecimal;