tcutest.c

Tokyo Cabinet的Tokyo Cabinet 是一个DBM的实现。这里的数据库由一系列key-value对的记录构成。key和value都可以是任意长度的字节序列,既可以是二进制也可以是字符

    if(strcmp(buf, "ab C")) err = true;
    if(tcstrcntutf(buf) != 4) err = true;
    tcstrcututf(buf, 2);
    if(strcmp(buf, "ab")) err = true;
    tcfree(buf);
    if(i % 10 == 1){
      int anum = myrand(30);
      uint16_t ary[anum+1];
      for(int j = 0; j < anum; j++){
        ary[j] = myrand(65535) + 1;
      }
      char ustr[anum*3+1];
      tcstrucstoutf(ary, anum, ustr);
      uint16_t dary[anum+1];
      int danum;
      tcstrutftoucs(ustr, dary, &danum);
      if(danum != anum){
        err = true;
      } else {
        for(int j = 0; j < danum; j++){
          if(dary[j] != dary[j]) err = true;
        }
      }
      list = tclistnew3("hop", "step", "jump", "touchdown", NULL);
      if(tclistnum(list) != 4) err = true;
      tclistdel(list);
      map = tcmapnew3("hop", "step", "jump", "touchdown", NULL);
      if(tcmaprnum(map) != 2) err = true;
      tcmapdel(map);
      list = tcstrsplit(",a,b..c,d,", ",.");
      if(tclistnum(list) != 7) err = true;
      buf = tcstrjoin(list, ':');
      if(strcmp(buf, ":a:b::c:d:")) err = true;
      tcfree(buf);
      tclistdel(list);
      char zbuf[RECBUFSIZ];
      memcpy(zbuf, "abc\0def\0ghij\0kl\0m", 17);
      list = tcstrsplit2(zbuf, 17);
      if(tclistnum(list) != 5) err = true;
      buf = tcstrjoin2(list, &bsiz);
      if(bsiz != 17 || memcmp(buf, "abc\0def\0ghij\0kl\0m", 17)) err = true;
      tcfree(buf);
      tclistdel(list);
      map = tcstrsplit3("abc.def,ghij.kl,", ",.");
      if(tcmaprnum(map) != 2) err = true;
      buf = tcstrjoin3(map, ':');
      if(strcmp(buf, "abc:def:ghij:kl")) err = true;
      tcfree(buf);
      tcmapdel(map);
      memcpy(zbuf, "abc\0def\0ghij\0kl\0m", 17);
      map = tcstrsplit4(zbuf, 17);
      if(tcmaprnum(map) != 2) err = true;
      buf = tcstrjoin4(map, &bsiz);
      if(bsiz != 15 || memcmp(buf, "abc\0def\0ghij\0kl", 15)) err = true;
      tcfree(buf);
      tcmapdel(map);
      if(!tcregexmatch("ABCDEFGHI", "*(b)c[d-f]*g(h)")) err = true;
      buf = tcregexreplace("ABCDEFGHI", "*(b)c[d-f]*g(h)", "[\\1][\\2][&]");
      if(strcmp(buf, "A[B][H][BCDEFGH]I")) err = true;
      tcfree(buf);
      buf = tcmalloc(48);
      for(int i = 0; i < 10; i++){
        char kbuf[RECBUFSIZ];
        int ksiz = sprintf(kbuf, "%d", myrand(1000000));
        tcmd5hash(kbuf, ksiz, buf);
      }
      tcfree(buf);
      anum = myrand(30) + 1;
      int tary[anum], qary[anum];
      for(int j = 0; j < anum; j++){
        int val = myrand(anum * 2 + 1);
        tary[j] = val;
        qary[j] = val;
      }
      int tnum = myrand(anum);
      tctopsort(tary, anum, sizeof(*tary), tnum, intcompar);
      qsort(qary, anum, sizeof(*qary), intcompar);
      for(int j = 0; j < tnum; j++){
        if(tary[j] != qary[j]) err = true;
      }
      qsort(tary, anum, sizeof(*tary), intcompar);
      for(int j = 0; j < anum; j++){
        if(tary[j] != qary[j]) err = true;
      }
      TCCHIDX *chidx = tcchidxnew(5);
      for(int i = 0; i < 10; i++){
        char kbuf[RECBUFSIZ];
        int ksiz = sprintf(kbuf, "%d", myrand(1000000));
        tcchidxhash(chidx, kbuf, ksiz);
      }
      tcchidxdel(chidx);
      buf = tczeromap(myrand(1024*256) + 1);
      tczerounmap(buf);
    }
    buf = tcmalloc(48);
    date = myrand(INT_MAX - 1000000);
    jl = 3600 * (myrand(23) - 11);
    tcdatestrwww(date, jl, buf);
    ddate = tcstrmktime(buf);
    if(ddate != date) err = true;
    tcdatestrhttp(date, jl, buf);
    ddate = tcstrmktime(buf);
    if(ddate != date) err = true;
    tcfree(buf);
    if(i % 100 == 1){
      map = myrand(2) == 0 ? tcmapnew() : tcmapnew2(myrand(10));
      tree = tctreenew();
      for(int j = 0; j < 10; j++){
        char kbuf[RECBUFSIZ];
        int ksiz = sprintf(kbuf, "%d", myrand(10));
        tcmapaddint(map, kbuf, ksiz, 1);
        const char *vbuf = tcmapget2(map, kbuf);
        if(*(int *)vbuf < 1) err = true;
        tctreeaddint(tree, kbuf, ksiz, 1);
        vbuf = tctreeget2(tree, kbuf);
        if(*(int *)vbuf < 1) err = true;
      }
      tcmapclear(map);
      tctreeclear(tree);
      for(int j = 0; j < 10; j++){
        char kbuf[RECBUFSIZ];
        int ksiz = sprintf(kbuf, "%d", myrand(10));
        tcmapadddouble(map, kbuf, ksiz, 1.0);
        const char *vbuf = tcmapget2(map, kbuf);
        if(*(double *)vbuf < 1.0) err = true;
        tctreeadddouble(tree, kbuf, ksiz, 1.0);
        vbuf = tctreeget2(tree, kbuf);
        if(*(double *)vbuf < 1.0) err = true;
      }
      tctreedel(tree);
      tcmapdel(map);
    }
    if(i % 100 == 1){
      ptrlist = myrand(2) == 0 ? tcptrlistnew() : tcptrlistnew2(myrand(10));
      for(int j = 0; j < 10; j++){
        tcptrlistpush(ptrlist, tcsprintf("%d", j));
        tcptrlistunshift(ptrlist, tcsprintf("::%d", j));
      }
      for(int j = 0; j < 5; j++){
        tcfree(tcptrlistpop(ptrlist));
        tcfree(tcptrlistshift(ptrlist));
      }
      for(int j = 0; j < tcptrlistnum(ptrlist); j++){
        tcfree(tcptrlistval(ptrlist, j));
      }
      tcptrlistdel(ptrlist);
    }
    buf = tcurlencode(str, slen);
    if(strcmp(buf, "5%252%2B3-1%3D4%20%22Yes%2FNo%22%20%3Ca%26b%3E")) err = true;
    dec = tcurldecode(buf, &dsiz);
    if(dsiz != slen || strcmp(dec, str)) err = true;
    tcfree(dec);
    tcfree(buf);
    if(i % 10 == 1){
      map = tcurlbreak("http://mikio:[e-mail]:1978/foo/bar/baz.cgi?ab=cd&ef=jkl#quux");
      const char *elem;
      if(!(elem = tcmapget2(map, "self")) ||
         strcmp(elem, "http://mikio:[e-mail]:1978/foo/bar/baz.cgi?ab=cd&ef=jkl#quux"))
        err = true;
      if(!(elem = tcmapget2(map, "scheme")) || strcmp(elem, "http")) err = true;
      if(!(elem = tcmapget2(map, "host")) || strcmp(elem, "estraier.net")) err = true;
      if(!(elem = tcmapget2(map, "port")) || strcmp(elem, "1978")) err = true;
      if(!(elem = tcmapget2(map, "authority")) || strcmp(elem, "mikio:oikim")) err = true;
      if(!(elem = tcmapget2(map, "path")) || strcmp(elem, "/foo/bar/baz.cgi")) err = true;
      if(!(elem = tcmapget2(map, "file")) || strcmp(elem, "baz.cgi")) err = true;
      if(!(elem = tcmapget2(map, "query")) || strcmp(elem, "ab=cd&ef=jkl")) err = true;
      if(!(elem = tcmapget2(map, "fragment")) || strcmp(elem, "quux")) err = true;
      tcmapdel(map);
      buf = tcurlresolve("http://a:b@c.d:1/e/f/g.h?i=j#k", "http://A:B@C.D:1/E/F/G.H?I=J#K");
      if(strcmp(buf, "http://A:B@c.d:1/E/F/G.H?I=J#K")) err = true;
      tcfree(buf);
      buf = tcurlresolve("http://a:b@c.d:1/e/f/g.h?i=j#k", "/E/F/G.H?I=J#K");
      if(strcmp(buf, "http://a:b@c.d:1/E/F/G.H?I=J#K")) err = true;
      tcfree(buf);
      buf = tcurlresolve("http://a:b@c.d:1/e/f/g.h?i=j#k", "G.H?I=J#K");
      if(strcmp(buf, "http://a:b@c.d:1/e/f/G.H?I=J#K")) err = true;
      tcfree(buf);
      buf = tcurlresolve("http://a:b@c.d:1/e/f/g.h?i=j#k", "?I=J#K");
      if(strcmp(buf, "http://a:b@c.d:1/e/f/g.h?I=J#K")) err = true;
      tcfree(buf);
      buf = tcurlresolve("http://a:b@c.d:1/e/f/g.h?i=j#k", "#K");
      if(strcmp(buf, "http://a:b@c.d:1/e/f/g.h?i=j#K")) err = true;
      tcfree(buf);
    }
    buf = tcbaseencode(str, slen);
    if(strcmp(buf, "NSUyKzMtMT00ICJZZXMvTm8iIDxhJmI+")) err = true;
    dec = tcbasedecode(buf, &dsiz);
    if(dsiz != slen || strcmp(dec, str)) err = true;
    tcfree(dec);
    tcfree(buf);
    buf = tcquoteencode(str, slen);
    if(strcmp(buf, "5%2+3-1=3D4 \"Yes/No\" <a&b>")) err = true;
    dec = tcquotedecode(buf, &dsiz);
    if(dsiz != slen || strcmp(dec, str)) err = true;
    tcfree(dec);
    tcfree(buf);
    buf = tcmimeencode(str, "UTF-8", true);
    if(strcmp(buf, "=?UTF-8?B?NSUyKzMtMT00ICJZZXMvTm8iIDxhJmI+?=")) err = true;
    char encname[32];
    dec = tcmimedecode(buf, encname);
    if(strcmp(dec, str) || strcmp(encname, "UTF-8")) err = true;
    tcfree(dec);
    tcfree(buf);
    if(i % 10 == 1){
      const char *mstr = "Subject: Hello\r\nContent-Type: multipart/mixed; boundary=____\r\n\r\n"
        "\r\n--____\r\nThis is a pen.\r\n--____\r\nIs this your bag?\r\n--____--\r\n";
      map = tcmapnew2(10);
      char *buf = tcmimebreak(mstr, strlen(mstr), map, &bsiz);
      const char *boundary = tcmapget2(map, "BOUNDARY");
      if(boundary){
        list = tcmimeparts(buf, bsiz, boundary);
        if(tclistnum(list) == 2){
          if(strcmp(tclistval2(list, 0), "This is a pen.")) err = true;
          if(strcmp(tclistval2(list, 1), "Is this your bag?")) err = true;
        } else {
          err = true;
        }
        tclistdel(list);
      } else {
        err = true;
      }
      tcfree(buf);
      tcmapdel(map);
    }
    buf = tcpackencode(str, slen, &bsiz);
    dec = tcpackdecode(buf, bsiz, &dsiz);
    if(dsiz != slen || strcmp(dec, str)) err = true;
    tcfree(dec);
    tcfree(buf);
    buf = tcbsencode(str, slen, &bsiz);
    dec = tcbsdecode(buf, bsiz, &dsiz);
    if(dsiz != slen || strcmp(dec, str)) err = true;
    tcfree(dec);
    tcfree(buf);
    int idx;
    buf = tcbwtencode(str, slen, &idx);
    if(memcmp(buf, "4\"o 5a23s-%+=> 1b/\"<&YNe", slen) || idx != 13) err = true;
    dec = tcbwtdecode(buf, slen, idx);
    if(memcmp(dec, str, slen)) err = true;
    tcfree(dec);
    tcfree(buf);
    if(_tc_deflate){
      if((buf = tcdeflate(str, slen, &bsiz)) != NULL){
        if((dec = tcinflate(buf, bsiz, &dsiz)) != NULL){
          if(slen != dsiz || memcmp(str, dec, dsiz)) err = true;
          tcfree(dec);
        } else {
          err = true;
        }
        tcfree(buf);
      } else {
        err = true;
      }
      if((buf = tcgzipencode(str, slen, &bsiz)) != NULL){
        if((dec = tcgzipdecode(buf, bsiz, &dsiz)) != NULL){
          if(slen != dsiz || memcmp(str, dec, dsiz)) err = true;
          tcfree(dec);
        } else {
          err = true;
        }
        tcfree(buf);
      } else {
        err = true;
      }
      if(tcgetcrc("hoge", 4) % 10000 != 7034) err = true;
    }
    if(_tc_bzcompress){
      if((buf = tcbzipencode(str, slen, &bsiz)) != NULL){
        if((dec = tcbzipdecode(buf, bsiz, &dsiz)) != NULL){
          if(slen != dsiz || memcmp(str, dec, dsiz)) err = true;
          tcfree(dec);
        } else {
          err = true;
        }
        tcfree(buf);
      } else {
        err = true;
      }
    }
    int anum = myrand(50)+1;
    unsigned int ary[anum];
    for(int j = 0; j < anum; j++){
      ary[j] = myrand(INT_MAX);
    }
    buf = tcberencode(ary, anum, &bsiz);
    int dnum;
    unsigned int *dary = tcberdecode(buf, bsiz, &dnum);
    if(anum != dnum || memcmp(ary, dary, sizeof(*dary) * dnum)) err = true;
    tcfree(dary);
    tcfree(buf);
    buf = tcxmlescape(str);
    if(strcmp(buf, "5%2+3-1=4 &quot;Yes/No&quot; &lt;a&amp;b&gt;")) err = true;
    dec = tcxmlunescape(buf);
    if(strcmp(dec, str)) err = true;
    tcfree(dec);
    tcfree(buf);
    if(i % 10 == 1){
      list = tcxmlbreak("<abc de=\"foo&amp;\" gh='&lt;bar&gt;'>xyz<br>\na<!--<mikio>--></abc>");
      for(int j = 0; j < tclistnum(list); j++){
        const char *elem = tclistval2(list, j);
        TCMAP *attrs = tcxmlattrs(elem);
        tcmapdel(attrs);
      }
      tclistdel(list);
    }
    if(i % 10 == 1){
      for(int16_t j = 1; j <= 0x2000; j *= 2){
        for(int16_t num = j - 1; num <= j + 1; num++){
          int16_t nnum = TCHTOIS(num);
          if(num != TCITOHS(nnum)) err = true;
        }
      }
      for(int32_t j = 1; j <= 0x20000000; j *= 2){
        for(int32_t num = j - 1; num <= j + 1; num++){
          int32_t nnum = TCHTOIL(num);
          if(num != TCITOHL(nnum)) err = true;
          char buf[TCNUMBUFSIZ];
          int step, nstep;
          TCSETVNUMBUF(step, buf, num);
          TCREADVNUMBUF(buf, nnum, nstep);
          if(num != nnum || step != nstep) err = true;
        }
      }
      for(int64_t j = 1; j <= 0x2000000000000000; j *= 2){
        for(int64_t num = j - 1; num <= j + 1; num++){
          int64_t nnum = TCHTOILL(num);
          if(num != TCITOHLL(nnum)) err = true;
          char buf[TCNUMBUFSIZ];
          int step, nstep;
          TCSETVNUMBUF64(step, buf, num);
          TCREADVNUMBUF64(buf, nnum, nstep);
          if(num != nnum || step != nstep) err = true;
        }
      }
      char *bitmap = TCBITMAPNEW(100);
      for(int j = 0; j < 100; j++){
        if(j % 3 == 0) TCBITMAPON(bitmap, j);
        if(j % 5 == 0) TCBITMAPOFF(bitmap, j);
      }
      for(int j = 0; j < 100; j++){
        if(j % 5 == 0){
          if(TCBITMAPCHECK(bitmap, j)) err = true;
        } else if(j % 3 == 0){
          if(!TCBITMAPCHECK(bitmap, j)) err = true;
        }
      }
      TCBITMAPDEL(bitmap);
      buf = tcmalloc(i / 8 + 2);
      TCBITSTRM strm;
      TCBITSTRMINITW(strm, buf);
      for(int j = 0; j < i; j++){
        int sign = j % 3 == 0 || j % 7 == 0;
        TCBITSTRMCAT(strm, sign);
      }
      TCBITSTRMSETEND(strm);
      int bnum = TCBITSTRMNUM(strm);
      if(bnum != i) err = true;
      TCBITSTRMINITR(strm, buf, bsiz);
      for(int j = 0; j < i; j++){
        int sign;
        TCBITSTRMREAD(strm, sign);
        if(sign != (j % 3 == 0 || j % 7 == 0)) err = true;
      }
      tcfree(buf);
    }
    if(rnum > 250 && i % (rnum / 250) == 0){
      iputchar('.');
      if(i == rnum || i % (rnum / 10) == 0) iprintf(" (%08d)\n", i);
    }
  }
  iprintf("time: %.3f\n", tctime() - stime);
  if(err){
    iprintf("error\n\n");
    return 1;
  }
  iprintf("ok\n\n");
  return 0;
}


/* perform wicked command */
static int procwicked(int rnum){
  iprintf("<Wicked Writing Test>\n  seed=%u  rnum=%d\n\n", g_randseed, rnum);
  double stime = tctime();
  TCMPOOL *mpool = tcmpoolglobal();
  TCXSTR *xstr = myrand(2) > 0 ? tcxstrnew() : tcxstrnew2("hello world");
  tcmpoolputxstr(mpool, xstr);
  TCLIST *list = myrand(2) > 0 ? tclistnew() : tclistnew2(myrand(rnum) + rnum / 2);
  tcmpoolputlist(mpool, list);
  TCPTRLIST *ptrlist = myrand(2) > 0 ? tcptrlistnew() : tcptrlistnew2(myrand(rnum) + rnum / 2);
  tcmpoolput(mpool, ptrlist, (void (*)(void*))tcptrlistdel);
  TCMAP *map = myrand(2) > 0 ? tcmapnew() : tcmapnew2(myrand(rnum) + rnum / 2);
  tcmpoolputmap(mpool, map);
  TCTREE *tree = myrand(2) > 0 ? tctreenew() : tctreenew2(tccmpdecimal, NULL);
  tcmpoolputtree(mpool, tree);
  TCMDB *mdb = myrand(2) > 0 ? tcmdbnew() : tcmdbnew2(myrand(rnum) + rnum / 2);
  tcmpoolput(mpool, mdb, (void (*)(void*))tcmdbdel);
  TCNDB *ndb = myrand(2) > 0 ? tcndbnew() : tcndbnew2(tccmpdecimal, NULL);
  tcmpoolput(mpool, ndb, (void (*)(void*))tcndbdel);
  for(int i = 1; i <= rnum; i++){
    char kbuf[RECBUFSIZ];
    int ksiz = sprintf(kbuf, "%d", myrand(i));
    char vbuf[RECBUFSIZ];
    int vsiz = sprintf(vbuf, "%d", myrand(i));
    char *tmp;
    switch(myrand(70)){
    case 0:
      iputchar('0');
      tcxstrcat(xstr, kbuf, ksiz);
      break;
    case 1:
      iputchar('1');
      tcxstrcat2(xstr, kbuf);
      break;
    case 2:
      iputchar('2');
      if(myrand(rnum / 100 + 1) == 0) tcxstrclear(xstr);
      break;
    case 3:
      iputchar('3');
      tcxstrprintf(xstr, "[%s:%d:%llu:%b:%llb]\n",
                   kbuf, i, (long long)i * 65521, i, (unsigned long long)i * 65521);
      break;
    case 4:
      iputchar('4');
      tclistpush(list, kbuf, ksiz);