开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数据库系统 > MySQL数据库 > 查看源码
testDataBuffers.cpp
资源名称：mysql-4.1.16-win-src.zip　[点击查看]
上传用户：romrleung
上传日期：2022-05-23
资源大小：18897k
文件大小：16k
源码类别：

MySQL数据库

开发平台：

Visual C++

testDataBuffers.cpp：源码内容
  1. /* Copyright (C) 2003 MySQL AB
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2 of the License, or
  6.    (at your option) any later version.
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  17. /*
  18.  * testDataBuffers
  19.  *
  20.  * Test getValue() of byte arrays:
  21.  * - using application buffers of different alignments and sizes
  22.  * - using NdbApi allocated small (<32) and big (>=32) buffers
  23.  *
  24.  * Verifies fixes to tickets 189 and 206.
  25.  *
  26.  * Options: see printusage() below.
  27.  *
  28.  * Creates tables TB00 to TB15
  29.  */
  31. #include <ndb_global.h>
  33. #include <NdbMain.h>
  34. #include <NdbOut.hpp>
  35. #include <NdbApi.hpp>
  36. #include <NdbTest.hpp>
  37. #include <NdbSchemaCon.hpp>
  38. // limits
  39. static int const MaxAttr = 64;
  40. static int const MaxOper = 1000;
  41. static int const MaxSize = 10000;
  42. static int const MaxOff = 64; // max offset to add to data buffer
  43. static int const MaxData = MaxSize + MaxOff + 100;
  45. // options
  46. static int attrcnt = 25;
  47. static int existok = 0;
  48. static bool kontinue = false;
  49. static int loopcnt = 1;
  50. static int opercnt = 100; // also does this many scans
  51. static int randomizer = 171317;
  52. static int sizelim = 500;
  53. static int xverbose = 0;
  55. static void printusage() {
  56.     ndbout
  57. << "usage: testDataBuffers options [default/max]"
  58. << endl
  59. << "NOTE: too large combinations result in NDB error"
  60. << endl
  61. << "-a N  number of attributes (including the key) [25/64]"
  62. << endl
  63. << "-e    no error if table exists (assumed to have same structure)"
  64. << endl
  65. << "-k    on error continue with next test case"
  66. << endl
  67. << "-l N  number of loops to run, 0 means infinite [1]"
  68. << endl
  69. << "-o N  number of operations (rows in each table) [100/1000]"
  70. << endl
  71. << "-r N  source of randomness (big number (prime)) [171317]"
  72. << endl
  73. << "-s N  array size limit (rounded up in some tests) [500/10000]"
  74. << endl
  75. << "-x    extremely verbose"
  76. << endl
  77. << "Tables: TB00 .. TB15"
  78. << endl
  79. ;
  80. }
  82. static Ndb* ndb = 0;
  83. static NdbSchemaCon* tcon = 0;
  84. static NdbSchemaOp* top = 0;
  85. static NdbConnection* con = 0;
  86. static NdbOperation* op = 0;
  87. static NdbScanOperation* sop = 0;
  88. static NdbResultSet* rs = 0;
  90. static int
  91. ndberror(char const* fmt, ...)
  92. {
  93.     va_list ap;
  94.     char buf[200];
  95.     va_start(ap, fmt);
  96.     BaseString::vsnprintf(buf, sizeof(buf), fmt, ap);
  97.     va_end(ap);
  98.     ndbout << buf << " --" << endl;
  99.     if (ndb)
  100.       ndbout << "ndb : " << ndb->getNdbError() << endl;
  101.     if (tcon)
  102. ndbout << "tcon: " << tcon->getNdbError() << endl;
  103.     if (top)
  104. ndbout << "top: " << top->getNdbError() << endl;
  105.     if (con)
  106. ndbout << "con : " << con->getNdbError() << endl;
  107.     if (op)
  108. ndbout << "op  : " << op->getNdbError() << endl;
  109.     return -1;
  110. }
  112. static int
  113. chkerror(char const* fmt, ...)
  114. {
  115.     va_list ap;
  116.     char buf[200];
  117.     va_start(ap, fmt);
  118.     BaseString::vsnprintf(buf, sizeof(buf), fmt, ap);
  119.     va_end(ap);
  120.     ndbout << "*** check failed: " << buf << " ***" << endl;
  121.     return -1;
  122. }
  124. // alignment of addresses and data sizes
  126. static bool isAligned(UintPtr x)
  127. {
  128.     return ((x & 3) == 0);
  129. }
  130. static bool isAligned(char* p)
  131. {
  132.     return isAligned(UintPtr(p));
  133. }
  134. static unsigned toAligned(UintPtr x)
  135. {
  136.     while (! isAligned(x))
  137. x++;
  138.     return x;
  139. }
  140. static char* toAligned(char* p)
  141. {
  142.     while (! isAligned(p))
  143. p++;
  144.     return p;
  145. }
  147. // byte value for key k column i byte j
  148. static int byteVal(int k, int i, int j)
  149. {
  150.     return '0' + (k + i + j)  % 10;
  151. }
  153. // tables
  155. static char tab[20] = "";
  157. static struct col {
  158.     char aAttrName[20];
  159.     AttrType aAttrType;
  160.     int aAttrSize;
  161.     int aArraySize;
  162.     KeyType aTupleKey;
  163.     bool nullable;
  164.     NdbRecAttr* aRa;
  165.     char* buf;
  166.     int bufsiz;
  167.     char data[MaxData];
  168. } ccol[MaxAttr];
  170. static int key = 0;
  172. // independent test bits
  173. static bool alignAddr; // align our buffer addresses to 4x
  174. static bool alignSize; // align data sizes to 4x
  175. static bool useBuf; // use our buffers for output
  176. static bool noRandom; // do not randomize sizes and offsets
  177. static int testbits = 4;
  179. static int
  180. makeSize(int i)
  181. {
  182.     int n;
  183.     if (noRandom)
  184. n = i;
  185.     else
  186. n = i * randomizer;
  187.     n %= sizelim;
  188.     if (n <= 0)
  189. n = 1;
  190.     if (alignSize)
  191. n = toAligned(n);
  192.     return n;
  193. }
  195. static int
  196. makeOff(int k)
  197. {
  198.     int n;
  199.     if (alignAddr)
  200. n = 0;
  201.     else if (noRandom)
  202. n = k;
  203.     else
  204. n = k * randomizer;
  205.     n %= MaxOff;
  206.     if (n < 0)
  207. n = -n;
  208.     return n;
  209. }
  211. static int
  212. testcase(int flag)
  213. {
  214.     ndbout << "--- case " << flag << " ---" << endl;
  215.     sprintf(tab, "TB%02d", flag);
  217.     alignAddr = ! (flag & 1);
  218.     ndbout << (alignAddr ? "align addresses" : "mis-align addresses") << endl;
  219.     alignSize = ! (flag & 2);
  220.     ndbout << (alignSize ? "align data sizes" : "mis-align data sizes") << endl;
  221.     useBuf = ! (flag & 4);
  222.     ndbout << (useBuf ? "use our buffers" : "use ndbapi buffers") << endl;
  223.     noRandom = ! (flag & 8);
  224.     ndbout << (noRandom ? "simple sizes" : "randomize sizes") << endl;
  226.     int smax = 0, stot = 0, i;
  227.     if (xverbose)
  228.       ndbout << "- define table " << tab << endl;
  229.     for (i = 0; i < attrcnt; i++) {
  230. col& c = ccol[i];
  231. memset(&c, 0, sizeof(c));
  232. sprintf(c.aAttrName, "C%d", i);
  233. if (i == 0) {
  234.     c.aAttrType = UnSigned;
  235.     c.aAttrSize = 32;
  236.     c.aArraySize = 1;
  237.     c.aTupleKey = TupleKey;
  238.     c.nullable = false;
  239. } else {
  240.     c.aAttrType = String;
  241.     c.aAttrSize = 8;
  242.     c.aArraySize = makeSize(i);
  243.     if (smax < c.aArraySize)
  244. smax = c.aArraySize;
  245.     stot += c.aArraySize;
  246.     c.aTupleKey = NoKey;
  247.     c.nullable = true;
  248.     if (xverbose)
  249.       ndbout << "-- column " << i << " size=" << c.aArraySize << endl;
  250. }
  251. c.buf = toAligned(c.data);
  252. c.bufsiz = sizeof(c.data) - (c.buf - c.data);
  253.     }
  254.     ndbout << "tab=" << tab << " cols=" << attrcnt
  255. << " size max=" << smax << " tot=" << stot << endl;
  257.     ndb = new Ndb("TEST_DB");
  258.     if (ndb->init() != 0)
  259. return ndberror("init");
  260.     if (ndb->waitUntilReady(30) < 0)
  261. return ndberror("waitUntilReady");
  263.     if ((tcon = NdbSchemaCon::startSchemaTrans(ndb)) == 0)
  264. return ndberror("startSchemaTransaction");
  265.     if ((top = tcon->getNdbSchemaOp()) == 0)
  266. return ndberror("getNdbSchemaOp");
  267.     if (top->createTable(tab) < 0)
  268. return ndberror("createTable");
  269.     for (i = 0; i < attrcnt; i++) {
  270. col& c = ccol[i];
  271. if (top->createAttribute(
  272.     c.aAttrName,
  273.     c.aTupleKey,
  274.     c.aAttrSize,
  275.     c.aArraySize,
  276.     c.aAttrType,
  277.     MMBased,
  278.     c.nullable
  279. ) < 0)
  280.     return ndberror("createAttribute col=%d", i);
  281.     }
  282.     if (tcon->execute() < 0) {
  283. if (! (tcon->getNdbError().code == 721 && existok))
  284.     return ndberror("execute");
  285. ndbout << "using " << tab << endl;
  286.     } else {
  287. ndbout << "created " << tab << endl;
  288.     }
  289.     top = 0;
  290.     tcon = 0;
  292.     if (xverbose)
  293.       ndbout << "- delete" << endl;
  294.     int delcnt = 0;
  295.     for (key = 0; key < opercnt; key++) {
  296. if ((con = ndb->startTransaction()) == 0)
  297.     return ndberror("startTransaction key=%d", key);
  298. if ((op = con->getNdbOperation(tab)) == 0)
  299.     return ndberror("getNdbOperation key=%d", key);
  300. if (op->deleteTuple() < 0)
  301.     return ndberror("deleteTuple key=%d", key);
  302. for (i = 0; i < attrcnt; i++) {
  303.     col& c = ccol[i];
  304.     if (i == 0) {
  305. if (op->equal(c.aAttrName, (char*)&key, sizeof(key)) < 0)
  306.     return ndberror("equal key=%d", key);
  307.     } else {
  308.     }
  309. }
  310. if (con->execute(Commit) < 0) {
  311.   if (con->getNdbError().code != 626)
  312.     return ndberror("execute key=%d", key);
  313. } else {
  314.     delcnt++;
  315. }
  316. ndb->closeTransaction(con);
  317.     }
  318.     con = 0;
  319.     op = 0;
  320.     ndbout << "deleted " << delcnt << endl;
  322.     if (xverbose)
  323.       ndbout << "- insert" << endl;
  324.     for (key = 0; key < opercnt; key++) {
  325. int off = makeOff(key);
  326. if ((con = ndb->startTransaction()) == 0)
  327.     return ndberror("startTransaction key=%d", key);
  328. if ((op = con->getNdbOperation(tab)) == 0)
  329.     return ndberror("getNdbOperation key=%d", key);
  330. if (op->insertTuple() < 0)
  331.     return ndberror("insertTuple key=%d", key);
  332. for (i = 0; i < attrcnt; i++) {
  333.     col& c = ccol[i];
  334.     if (i == 0) {
  335. if (op->equal(c.aAttrName, (char*)&key, sizeof(key)) < 0)
  336.     return ndberror("equal key=%d", key);
  337.     } else {
  338. memset(c.buf, 'A', c.bufsiz);
  339. for (int j = 0; j < c.aArraySize; j++)
  340.     c.buf[j + off] = byteVal(key, i, j);
  341. if (op->setValue(c.aAttrName, c.buf + off, c.aArraySize) < 0)
  342.     return ndberror("setValue key=%d col=%d", key, i);
  343.     }
  344. }
  345. if (con->execute(Commit) < 0)
  346.     return ndberror("execute key=%d", key);
  347. ndb->closeTransaction(con);
  348.     }
  349.     con = 0;
  350.     op = 0;
  351.     ndbout << "inserted " << key << endl;
  353.     if (xverbose)
  354.       ndbout << "- select" << endl;
  355.     for (key = 0; key < opercnt; key++) {
  356. int off = makeOff(key);
  357. if (xverbose)
  358.   ndbout << "-- key " << key << " off=" << off << endl;
  359. if ((con = ndb->startTransaction()) == 0)
  360.     return ndberror("startTransaction key=%d", key);
  361. if ((op = con->getNdbOperation(tab)) == 0)
  362.     return ndberror("getNdbOperation key=%d", key);
  363. if (op->readTuple() < 0)
  364.     return ndberror("readTuple key=%d", key);
  365. for (i = 0; i < attrcnt; i++) {
  366.     col& c = ccol[i];
  367.     if (i == 0) {
  368. if (op->equal(c.aAttrName, (char*)&key, sizeof(key)) < 0)
  369.     return ndberror("equal key=%d", key);
  370.     } else {
  371. if (xverbose) {
  372.   char tmp[20];
  373.   if (useBuf)
  374.     sprintf(tmp, "0x%p", c.buf + off);
  375.   else
  376.     strcpy(tmp, "ndbapi");
  377.   ndbout << "--- column " << i << " addr=" << tmp << endl;
  378. }
  379. memset(c.buf, 'B', c.bufsiz);
  380. if (useBuf) {
  381.     if (op->getValue(c.aAttrName, c.buf + off) < 0)
  382. return ndberror("getValue key=%d col=%d", key, i);
  383. } else {
  384.     if ((c.aRa = op->getValue(c.aAttrName)) == 0)
  385. return ndberror("getValue key=%d col=%d", key, i);
  386. }
  387.     }
  388. }
  389. if (con->execute(Commit) != 0)
  390.     return ndberror("execute key=%d", key);
  391. for (i = 0; i < attrcnt; i++) {
  392.     col& c = ccol[i];
  393.     if (i == 0) {
  394.     } else if (useBuf) {
  395.                 int j;
  396. for (j = 0; j < off; j++) {
  397.     if (c.buf[j] != 'B') {
  398. return chkerror("mismatch before key=%d col=%d pos=%d ok=%02x bad=%02x",
  399.     key, i, j, 'B', c.buf[j]);
  400.     }
  401. }
  402. for (j = 0; j < c.aArraySize; j++) {
  403.     if (c.buf[j + off] != byteVal(key, i, j)) {
  404. return chkerror("mismatch key=%d col=%d pos=%d ok=%02x bad=%02x",
  405.     key, i, j, byteVal(key, i, j), c.buf[j]);
  406.     }
  407. }
  408. for (j = c.aArraySize + off; j < c.bufsiz; j++) {
  409.     if (c.buf[j] != 'B') {
  410. return chkerror("mismatch after key=%d col=%d pos=%d ok=%02x bad=%02x",
  411.     key, i, j, 'B', c.buf[j]);
  412.     }
  413. }
  414.     } else {
  415. char* buf = c.aRa->aRef();
  416. if (buf == 0)
  417.     return ndberror("null aRef key=%d col%d", key, i);
  418. for (int j = 0; j < c.aArraySize; j++) {
  419.     if (buf[j] != byteVal(key, i, j)) {
  420. return chkerror("mismatch key=%d col=%d pos=%d ok=%02x bad=%02x",
  421.     key, i, j, byteVal(key, i, j), buf[j]);
  422.     }
  423. }
  424.     }
  425. }
  426. ndb->closeTransaction(con);
  427.     }
  428.     con = 0;
  429.     op = 0;
  430.     ndbout << "selected " << key << endl;
  432.     if (xverbose)
  433.       ndbout << "- scan" << endl;
  434.     char found[MaxOper];
  435.     int k;
  436.     for (k = 0; k < opercnt; k++)
  437. found[k] = 0;
  438.     for (key = 0; key < opercnt; key++) {
  439. int off = makeOff(key);
  440. if (xverbose)
  441.   ndbout << "-- key " << key << " off=" << off << endl;
  442. int newkey = 0;
  443. if ((con = ndb->startTransaction()) == 0)
  444.     return ndberror("startTransaction key=%d", key);
  445. if ((op = sop = con->getNdbScanOperation(tab)) == 0)
  446.     return ndberror("getNdbOperation key=%d", key);
  447. if ((rs = sop->readTuples(1)) == 0)
  448.     return ndberror("openScanRead key=%d", key);
  449. {
  450.     col& c = ccol[0];
  451.     if (op->load_const_u32(1, key) < 0)
  452. return ndberror("load_const_u32");
  453.     if (op->read_attr(c.aAttrName, 2) < 0)
  454. return ndberror("read_attr");
  455.     if (op->branch_eq(1, 2, 0) < 0)
  456. return ndberror("branch_eq");
  457.     if (op->interpret_exit_nok() < 0)
  458. return ndberror("interpret_exit_nok");
  459.     if (op->def_label(0) < 0)
  460. return ndberror("def_label");
  461.     if (op->interpret_exit_ok() < 0)
  462. return ndberror("interpret_exit_ok");
  463. }
  464. for (i = 0; i < attrcnt; i++) {
  465.     col& c = ccol[i];
  466.     if (i == 0) {
  467. if (op->getValue(c.aAttrName, (char*)&newkey) < 0)
  468.     return ndberror("getValue key=%d col=%d", key, i);
  469.     } else {
  470. if (xverbose) {
  471.   char tmp[20];
  472.   if (useBuf)
  473.     sprintf(tmp, "0x%p", c.buf + off);
  474.   else
  475.     strcpy(tmp, "ndbapi");
  476.   ndbout << "--- column " << i << " addr=" << tmp << endl;
  477. }
  478. memset(c.buf, 'C', c.bufsiz);
  479. if (useBuf) {
  480.     if (op->getValue(c.aAttrName, c.buf + off) < 0)
  481. return ndberror("getValue key=%d col=%d", key, i);
  482. } else {
  483.     if ((c.aRa = op->getValue(c.aAttrName)) == 0)
  484. return ndberror("getValue key=%d col=%d", key, i);
  485. }
  486.     }
  487. }
  488. if (con->execute(NoCommit) < 0)
  489.     return ndberror("executeScan key=%d", key);
  490. int ret, cnt = 0;
  491. while ((ret = rs->nextResult()) == 0) {
  492.     if (key != newkey)
  493. return ndberror("unexpected key=%d newkey=%d", key, newkey);
  494.     for (i = 1; i < attrcnt; i++) {
  495. col& c = ccol[i];
  496. if (useBuf) {
  497.                     int j;
  498.     for (j = 0; j < off; j++) {
  499. if (c.buf[j] != 'C') {
  500.     return chkerror("mismatch before key=%d col=%d pos=%d ok=%02x bad=%02x",
  501. key, i, j, 'C', c.buf[j]);
  502. }
  503.     }
  504.     for (j = 0; j < c.aArraySize; j++) {
  505. if (c.buf[j + off] != byteVal(key, i, j)) {
  506.     return chkerror("mismatch key=%d col=%d pos=%d ok=%02x bad=%02x",
  507. key, i, j, byteVal(key, i, j), c.buf[j]);
  508. }
  509.     }
  510.     for (j = c.aArraySize + off; j < c.bufsiz; j++) {
  511. if (c.buf[j] != 'C') {
  512.     return chkerror("mismatch after key=%d col=%d pos=%d ok=%02x bad=%02x",
  513. key, i, j, 'C', c.buf[j]);
  514. }
  515.     }
  516. } else {
  517.     char* buf = c.aRa->aRef();
  518.     if (buf == 0)
  519. return ndberror("null aRef key=%d col%d", key, i);
  520.     for (int j = 0; j < c.aArraySize; j++) {
  521. if (buf[j] != byteVal(key, i, j)) {
  522.     return chkerror("mismatch key=%d col=%d pos=%d ok=%02x bad=%02x",
  523. key, i, j, byteVal(key, i, j), buf[j]);
  524. }
  525.     }
  526. }
  527.     }
  528.     cnt++;
  529. }
  530. if (ret < 0)
  531.     return ndberror("nextScanResult key=%d", key);
  532. if (cnt != 1)
  533.     return ndberror("scan key=%d found %d", key, cnt);
  534. found[key] = 1;
  535. ndb->closeTransaction(con);
  536.     }
  537.     con = 0;
  538.     op = 0;
  539.     for (k = 0; k < opercnt; k++)
  540. if (! found[k])
  541.     return ndberror("key %d not found", k);
  542.     ndbout << "scanned " << key << endl;
  544.     ndb = 0;
  545.     ndbout << "done" << endl;
  546.     return 0;
  547. }
  549. NDB_COMMAND(testDataBuffers, "testDataBuffers", "testDataBuffers", "testDataBuffers", 65535)
  550. {
  551.     int i;
  552.     ndb_init();
  553.     while (++argv, --argc > 0) {
  554. char const* p = argv[0];
  555. if (*p++ != '-' || strlen(p) != 1)
  556.     goto wrongargs;
  557. switch (*p) {
  558. case 'a':
  559.     if (++argv, --argc > 0) {
  560. attrcnt = atoi(argv[0]);
  561. if (1 <= attrcnt && attrcnt <= MaxAttr)
  562.     break;
  563.     }
  564.     goto wrongargs;
  565. case 'e':
  566.     existok = 1;
  567.     break;
  568. case 'k':
  569.     kontinue = true;
  570.     break;
  571. case 'l':
  572.     if (++argv, --argc > 0) {
  573. loopcnt = atoi(argv[0]);
  574. if (0 <= loopcnt)
  575.     break;
  576.     }
  577.     goto wrongargs;
  578. case 'o':
  579.     if (++argv, --argc > 0) {
  580. opercnt = atoi(argv[0]);
  581. if (0 <= opercnt && opercnt <= MaxOper)
  582.     break;
  583.     }
  584.     goto wrongargs;
  585. case 'r':
  586.     if (++argv, --argc > 0) {
  587. randomizer = atoi(argv[0]);
  588. if (1 <= randomizer)
  589.     break;
  590.     }
  591.     goto wrongargs;
  592. case 's':
  593.     if (++argv, --argc > 0) {
  594. sizelim = atoi(argv[0]);
  595. if (1 <= sizelim && sizelim <= MaxSize)
  596.     break;
  597.     }
  598.     goto wrongargs;
  599. case 'x':
  600.     xverbose = 1;
  601.     break;
  602. default:
  603. wrongargs:
  604.     printusage();
  605.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  606. }
  607.     }
  608.     unsigned ok = true;
  609.     for (i = 1; 0 == loopcnt || i <= loopcnt; i++) {
  610. ndbout << "=== loop " << i << " ===" << endl;
  611. for (int flag = 0; flag < (1<<testbits); flag++) {
  612.     if (testcase(flag) < 0) {
  613. ok = false;
  614. if (! kontinue)
  615.     goto out;
  616.     }
  617. }
  618.     }
  619. out:
  620.     return NDBT_ProgramExit(ok ? NDBT_OK : NDBT_FAILED);
  621. }
  623. // vim: set sw=4: