1 /* 2 ** 2015-04-17 3 ** 4 ** The author disclaims copyright to this source code. In place of 5 ** a legal notice, here is a blessing: 6 ** 7 ** May you do good and not evil. 8 ** May you find forgiveness for yourself and forgive others. 9 ** May you share freely, never taking more than you give. 10 ** 11 ************************************************************************* 12 ** 13 ** This is a utility program designed to aid running the SQLite library 14 ** against an external fuzzer, such as American Fuzzy Lop (AFL) 15 ** (http://lcamtuf.coredump.cx/afl/). Basically, this program reads 16 ** SQL text from standard input and passes it through to SQLite for evaluation, 17 ** just like the "sqlite3" command-line shell. Differences from the 18 ** command-line shell: 19 ** 20 ** (1) The complex "dot-command" extensions are omitted. This 21 ** prevents the fuzzer from discovering that it can run things 22 ** like ".shell rm -rf ~" 23 ** 24 ** (2) The database is opened with the SQLITE_OPEN_MEMORY flag so that 25 ** no disk I/O from the database is permitted. The ATTACH command 26 ** with a filename still uses an in-memory database. 27 ** 28 ** (3) The main in-memory database can be initialized from a template 29 ** disk database so that the fuzzer starts with a database containing 30 ** content. 31 ** 32 ** (4) The eval() SQL function is added, allowing the fuzzer to do 33 ** interesting recursive operations. 34 ** 35 ** (5) An error is raised if there is a memory leak. 36 ** 37 ** The input text can be divided into separate test cases using comments 38 ** of the form: 39 ** 40 ** |****<...>****| 41 ** 42 ** where the "..." is arbitrary text. (Except the "|" should really be "/". 43 ** "|" is used here to avoid compiler errors about nested comments.) 44 ** A separate in-memory SQLite database is created to run each test case. 45 ** This feature allows the "queue" of AFL to be captured into a single big 46 ** file using a command like this: 47 ** 48 ** (for i in id:*; do echo '|****<'$i'>****|'; cat $i; done) >~/all-queue.txt 49 ** 50 ** (Once again, change the "|" to "/") Then all elements of the AFL queue 51 ** can be run in a single go (for regression testing, for example) by typing: 52 ** 53 ** fuzzershell -f ~/all-queue.txt 54 ** 55 ** After running each chunk of SQL, the database connection is closed. The 56 ** program aborts if the close fails or if there is any unfreed memory after 57 ** the close. 58 ** 59 ** New test cases can be appended to all-queue.txt at any time. If redundant 60 ** test cases are added, they can be eliminated by running: 61 ** 62 ** fuzzershell -f ~/all-queue.txt --unique-cases ~/unique-cases.txt 63 */ 64 #include <stdio.h> 65 #include <stdlib.h> 66 #include <string.h> 67 #include <stdarg.h> 68 #include <ctype.h> 69 #include "sqlite3.h" 70 71 /* 72 ** All global variables are gathered into the "g" singleton. 73 */ 74 struct GlobalVars { 75 const char *zArgv0; /* Name of program */ 76 sqlite3_mem_methods sOrigMem; /* Original memory methods */ 77 sqlite3_mem_methods sOomMem; /* Memory methods with OOM simulator */ 78 int iOomCntdown; /* Memory fails on 1 to 0 transition */ 79 int nOomFault; /* Increments for each OOM fault */ 80 int bOomOnce; /* Fail just once if true */ 81 int bOomEnable; /* True to enable OOM simulation */ 82 int nOomBrkpt; /* Number of calls to oomFault() */ 83 char zTestName[100]; /* Name of current test */ 84 } g; 85 86 /* 87 ** Maximum number of iterations for an OOM test 88 */ 89 #ifndef OOM_MAX 90 # define OOM_MAX 625 91 #endif 92 93 /* 94 ** This routine is called when a simulated OOM occurs. It exists as a 95 ** convenient place to set a debugger breakpoint. 96 */ 97 static void oomFault(void){ 98 g.nOomBrkpt++; /* Prevent oomFault() from being optimized out */ 99 } 100 101 102 /* Versions of malloc() and realloc() that simulate OOM conditions */ 103 static void *oomMalloc(int nByte){ 104 if( nByte>0 && g.bOomEnable && g.iOomCntdown>0 ){ 105 g.iOomCntdown--; 106 if( g.iOomCntdown==0 ){ 107 if( g.nOomFault==0 ) oomFault(); 108 g.nOomFault++; 109 if( !g.bOomOnce ) g.iOomCntdown = 1; 110 return 0; 111 } 112 } 113 return g.sOrigMem.xMalloc(nByte); 114 } 115 static void *oomRealloc(void *pOld, int nByte){ 116 if( nByte>0 && g.bOomEnable && g.iOomCntdown>0 ){ 117 g.iOomCntdown--; 118 if( g.iOomCntdown==0 ){ 119 if( g.nOomFault==0 ) oomFault(); 120 g.nOomFault++; 121 if( !g.bOomOnce ) g.iOomCntdown = 1; 122 return 0; 123 } 124 } 125 return g.sOrigMem.xRealloc(pOld, nByte); 126 } 127 128 /* 129 ** Print an error message and abort in such a way to indicate to the 130 ** fuzzer that this counts as a crash. 131 */ 132 static void abendError(const char *zFormat, ...){ 133 va_list ap; 134 if( g.zTestName[0] ){ 135 fprintf(stderr, "%s (%s): ", g.zArgv0, g.zTestName); 136 }else{ 137 fprintf(stderr, "%s: ", g.zArgv0); 138 } 139 va_start(ap, zFormat); 140 vfprintf(stderr, zFormat, ap); 141 va_end(ap); 142 fprintf(stderr, "\n"); 143 abort(); 144 } 145 /* 146 ** Print an error message and quit, but not in a way that would look 147 ** like a crash. 148 */ 149 static void fatalError(const char *zFormat, ...){ 150 va_list ap; 151 if( g.zTestName[0] ){ 152 fprintf(stderr, "%s (%s): ", g.zArgv0, g.zTestName); 153 }else{ 154 fprintf(stderr, "%s: ", g.zArgv0); 155 } 156 va_start(ap, zFormat); 157 vfprintf(stderr, zFormat, ap); 158 va_end(ap); 159 fprintf(stderr, "\n"); 160 exit(1); 161 } 162 163 /* 164 ** Evaluate some SQL. Abort if unable. 165 */ 166 static void sqlexec(sqlite3 *db, const char *zFormat, ...){ 167 va_list ap; 168 char *zSql; 169 char *zErrMsg = 0; 170 int rc; 171 va_start(ap, zFormat); 172 zSql = sqlite3_vmprintf(zFormat, ap); 173 va_end(ap); 174 rc = sqlite3_exec(db, zSql, 0, 0, &zErrMsg); 175 if( rc ) abendError("failed sql [%s]: %s", zSql, zErrMsg); 176 sqlite3_free(zSql); 177 } 178 179 /* 180 ** This callback is invoked by sqlite3_log(). 181 */ 182 static void shellLog(void *pNotUsed, int iErrCode, const char *zMsg){ 183 printf("LOG: (%d) %s\n", iErrCode, zMsg); 184 fflush(stdout); 185 } 186 static void shellLogNoop(void *pNotUsed, int iErrCode, const char *zMsg){ 187 return; 188 } 189 190 /* 191 ** This callback is invoked by sqlite3_exec() to return query results. 192 */ 193 static int execCallback(void *NotUsed, int argc, char **argv, char **colv){ 194 int i; 195 static unsigned cnt = 0; 196 printf("ROW #%u:\n", ++cnt); 197 for(i=0; i<argc; i++){ 198 printf(" %s=", colv[i]); 199 if( argv[i] ){ 200 printf("[%s]\n", argv[i]); 201 }else{ 202 printf("NULL\n"); 203 } 204 } 205 fflush(stdout); 206 return 0; 207 } 208 static int execNoop(void *NotUsed, int argc, char **argv, char **colv){ 209 return 0; 210 } 211 212 #ifndef SQLITE_OMIT_TRACE 213 /* 214 ** This callback is invoked by sqlite3_trace() as each SQL statement 215 ** starts. 216 */ 217 static void traceCallback(void *NotUsed, const char *zMsg){ 218 printf("TRACE: %s\n", zMsg); 219 fflush(stdout); 220 } 221 static void traceNoop(void *NotUsed, const char *zMsg){ 222 return; 223 } 224 #endif 225 226 /*************************************************************************** 227 ** eval() implementation copied from ../ext/misc/eval.c 228 */ 229 /* 230 ** Structure used to accumulate the output 231 */ 232 struct EvalResult { 233 char *z; /* Accumulated output */ 234 const char *zSep; /* Separator */ 235 int szSep; /* Size of the separator string */ 236 sqlite3_int64 nAlloc; /* Number of bytes allocated for z[] */ 237 sqlite3_int64 nUsed; /* Number of bytes of z[] actually used */ 238 }; 239 240 /* 241 ** Callback from sqlite_exec() for the eval() function. 242 */ 243 static int callback(void *pCtx, int argc, char **argv, char **colnames){ 244 struct EvalResult *p = (struct EvalResult*)pCtx; 245 int i; 246 for(i=0; i<argc; i++){ 247 const char *z = argv[i] ? argv[i] : ""; 248 size_t sz = strlen(z); 249 if( (sqlite3_int64)sz+p->nUsed+p->szSep+1 > p->nAlloc ){ 250 char *zNew; 251 p->nAlloc = p->nAlloc*2 + sz + p->szSep + 1; 252 /* Using sqlite3_realloc64() would be better, but it is a recent 253 ** addition and will cause a segfault if loaded by an older version 254 ** of SQLite. */ 255 zNew = p->nAlloc<=0x7fffffff ? sqlite3_realloc(p->z, (int)p->nAlloc) : 0; 256 if( zNew==0 ){ 257 sqlite3_free(p->z); 258 memset(p, 0, sizeof(*p)); 259 return 1; 260 } 261 p->z = zNew; 262 } 263 if( p->nUsed>0 ){ 264 memcpy(&p->z[p->nUsed], p->zSep, p->szSep); 265 p->nUsed += p->szSep; 266 } 267 memcpy(&p->z[p->nUsed], z, sz); 268 p->nUsed += sz; 269 } 270 return 0; 271 } 272 273 /* 274 ** Implementation of the eval(X) and eval(X,Y) SQL functions. 275 ** 276 ** Evaluate the SQL text in X. Return the results, using string 277 ** Y as the separator. If Y is omitted, use a single space character. 278 */ 279 static void sqlEvalFunc( 280 sqlite3_context *context, 281 int argc, 282 sqlite3_value **argv 283 ){ 284 const char *zSql; 285 sqlite3 *db; 286 char *zErr = 0; 287 int rc; 288 struct EvalResult x; 289 290 memset(&x, 0, sizeof(x)); 291 x.zSep = " "; 292 zSql = (const char*)sqlite3_value_text(argv[0]); 293 if( zSql==0 ) return; 294 if( argc>1 ){ 295 x.zSep = (const char*)sqlite3_value_text(argv[1]); 296 if( x.zSep==0 ) return; 297 } 298 x.szSep = (int)strlen(x.zSep); 299 db = sqlite3_context_db_handle(context); 300 rc = sqlite3_exec(db, zSql, callback, &x, &zErr); 301 if( rc!=SQLITE_OK ){ 302 sqlite3_result_error(context, zErr, -1); 303 sqlite3_free(zErr); 304 }else if( x.zSep==0 ){ 305 sqlite3_result_error_nomem(context); 306 sqlite3_free(x.z); 307 }else{ 308 sqlite3_result_text(context, x.z, (int)x.nUsed, sqlite3_free); 309 } 310 } 311 /* End of the eval() implementation 312 ******************************************************************************/ 313 314 /* 315 ** Print sketchy documentation for this utility program 316 */ 317 static void showHelp(void){ 318 printf("Usage: %s [options] ?FILE...?\n", g.zArgv0); 319 printf( 320 "Read SQL text from FILE... (or from standard input if FILE... is omitted)\n" 321 "and then evaluate each block of SQL contained therein.\n" 322 "Options:\n" 323 " --autovacuum Enable AUTOVACUUM mode\n" 324 " --database FILE Use database FILE instead of an in-memory database\n" 325 " --heap SZ MIN Memory allocator uses SZ bytes & min allocation MIN\n" 326 " --help Show this help text\n" 327 " --lookaside N SZ Configure lookaside for N slots of SZ bytes each\n" 328 " --oom Run each test multiple times in a simulated OOM loop\n" 329 " --pagesize N Set the page size to N\n" 330 " --pcache N SZ Configure N pages of pagecache each of size SZ bytes\n" 331 " -q Reduced output\n" 332 " --quiet Reduced output\n" 333 " --scratch N SZ Configure scratch memory for N slots of SZ bytes each\n" 334 " --unique-cases FILE Write all unique test cases to FILE\n" 335 " --utf16be Set text encoding to UTF-16BE\n" 336 " --utf16le Set text encoding to UTF-16LE\n" 337 " -v Increased output\n" 338 " --verbose Increased output\n" 339 ); 340 } 341 342 /* 343 ** Return the value of a hexadecimal digit. Return -1 if the input 344 ** is not a hex digit. 345 */ 346 static int hexDigitValue(char c){ 347 if( c>='0' && c<='9' ) return c - '0'; 348 if( c>='a' && c<='f' ) return c - 'a' + 10; 349 if( c>='A' && c<='F' ) return c - 'A' + 10; 350 return -1; 351 } 352 353 /* 354 ** Interpret zArg as an integer value, possibly with suffixes. 355 */ 356 static int integerValue(const char *zArg){ 357 sqlite3_int64 v = 0; 358 static const struct { char *zSuffix; int iMult; } aMult[] = { 359 { "KiB", 1024 }, 360 { "MiB", 1024*1024 }, 361 { "GiB", 1024*1024*1024 }, 362 { "KB", 1000 }, 363 { "MB", 1000000 }, 364 { "GB", 1000000000 }, 365 { "K", 1000 }, 366 { "M", 1000000 }, 367 { "G", 1000000000 }, 368 }; 369 int i; 370 int isNeg = 0; 371 if( zArg[0]=='-' ){ 372 isNeg = 1; 373 zArg++; 374 }else if( zArg[0]=='+' ){ 375 zArg++; 376 } 377 if( zArg[0]=='0' && zArg[1]=='x' ){ 378 int x; 379 zArg += 2; 380 while( (x = hexDigitValue(zArg[0]))>=0 ){ 381 v = (v<<4) + x; 382 zArg++; 383 } 384 }else{ 385 while( isdigit(zArg[0]) ){ 386 v = v*10 + zArg[0] - '0'; 387 zArg++; 388 } 389 } 390 for(i=0; i<sizeof(aMult)/sizeof(aMult[0]); i++){ 391 if( sqlite3_stricmp(aMult[i].zSuffix, zArg)==0 ){ 392 v *= aMult[i].iMult; 393 break; 394 } 395 } 396 if( v>0x7fffffff ) abendError("parameter too large - max 2147483648"); 397 return (int)(isNeg? -v : v); 398 } 399 400 /* Return the current wall-clock time */ 401 static sqlite3_int64 timeOfDay(void){ 402 static sqlite3_vfs *clockVfs = 0; 403 sqlite3_int64 t; 404 if( clockVfs==0 ) clockVfs = sqlite3_vfs_find(0); 405 if( clockVfs->iVersion>=1 && clockVfs->xCurrentTimeInt64!=0 ){ 406 clockVfs->xCurrentTimeInt64(clockVfs, &t); 407 }else{ 408 double r; 409 clockVfs->xCurrentTime(clockVfs, &r); 410 t = (sqlite3_int64)(r*86400000.0); 411 } 412 return t; 413 } 414 415 int main(int argc, char **argv){ 416 char *zIn = 0; /* Input text */ 417 int nAlloc = 0; /* Number of bytes allocated for zIn[] */ 418 int nIn = 0; /* Number of bytes of zIn[] used */ 419 size_t got; /* Bytes read from input */ 420 int rc = SQLITE_OK; /* Result codes from API functions */ 421 int i; /* Loop counter */ 422 int iNext; /* Next block of SQL */ 423 sqlite3 *db; /* Open database */ 424 char *zErrMsg = 0; /* Error message returned from sqlite3_exec() */ 425 const char *zEncoding = 0; /* --utf16be or --utf16le */ 426 int nHeap = 0, mnHeap = 0; /* Heap size from --heap */ 427 int nLook = 0, szLook = 0; /* --lookaside configuration */ 428 int nPCache = 0, szPCache = 0;/* --pcache configuration */ 429 int nScratch = 0, szScratch=0;/* --scratch configuration */ 430 int pageSize = 0; /* Desired page size. 0 means default */ 431 void *pHeap = 0; /* Allocated heap space */ 432 void *pLook = 0; /* Allocated lookaside space */ 433 void *pPCache = 0; /* Allocated storage for pcache */ 434 void *pScratch = 0; /* Allocated storage for scratch */ 435 int doAutovac = 0; /* True for --autovacuum */ 436 char *zSql; /* SQL to run */ 437 char *zToFree = 0; /* Call sqlite3_free() on this afte running zSql */ 438 int verboseFlag = 0; /* --verbose or -v flag */ 439 int quietFlag = 0; /* --quiet or -q flag */ 440 int nTest = 0; /* Number of test cases run */ 441 int multiTest = 0; /* True if there will be multiple test cases */ 442 int lastPct = -1; /* Previous percentage done output */ 443 sqlite3 *dataDb = 0; /* Database holding compacted input data */ 444 sqlite3_stmt *pStmt = 0; /* Statement to insert testcase into dataDb */ 445 const char *zDataOut = 0; /* Write compacted data to this output file */ 446 int nHeader = 0; /* Bytes of header comment text on input file */ 447 int oomFlag = 0; /* --oom */ 448 int oomCnt = 0; /* Counter for the OOM loop */ 449 char zErrBuf[200]; /* Space for the error message */ 450 const char *zFailCode; /* Value of the TEST_FAILURE environment var */ 451 const char *zPrompt; /* Initial prompt when large-file fuzzing */ 452 int nInFile = 0; /* Number of input files to read */ 453 char **azInFile = 0; /* Array of input file names */ 454 int jj; /* Loop counter for azInFile[] */ 455 sqlite3_int64 iBegin; /* Start time for the whole program */ 456 sqlite3_int64 iStart, iEnd; /* Start and end-times for a test case */ 457 const char *zDbName = 0; /* Name of an on-disk database file to open */ 458 459 iBegin = timeOfDay(); 460 zFailCode = getenv("TEST_FAILURE"); 461 g.zArgv0 = argv[0]; 462 zPrompt = "<stdin>"; 463 for(i=1; i<argc; i++){ 464 const char *z = argv[i]; 465 if( z[0]=='-' ){ 466 z++; 467 if( z[0]=='-' ) z++; 468 if( strcmp(z,"autovacuum")==0 ){ 469 doAutovac = 1; 470 }else 471 if( strcmp(z,"database")==0 ){ 472 if( i>=argc-1 ) abendError("missing argument on %s\n", argv[i]); 473 zDbName = argv[i+1]; 474 i += 1; 475 }else 476 if( strcmp(z, "f")==0 && i+1<argc ){ 477 i++; 478 goto addNewInFile; 479 }else 480 if( strcmp(z,"heap")==0 ){ 481 if( i>=argc-2 ) abendError("missing arguments on %s\n", argv[i]); 482 nHeap = integerValue(argv[i+1]); 483 mnHeap = integerValue(argv[i+2]); 484 i += 2; 485 }else 486 if( strcmp(z,"help")==0 ){ 487 showHelp(); 488 return 0; 489 }else 490 if( strcmp(z,"lookaside")==0 ){ 491 if( i>=argc-2 ) abendError("missing arguments on %s", argv[i]); 492 nLook = integerValue(argv[i+1]); 493 szLook = integerValue(argv[i+2]); 494 i += 2; 495 }else 496 if( strcmp(z,"oom")==0 ){ 497 oomFlag = 1; 498 }else 499 if( strcmp(z,"pagesize")==0 ){ 500 if( i>=argc-1 ) abendError("missing argument on %s", argv[i]); 501 pageSize = integerValue(argv[++i]); 502 }else 503 if( strcmp(z,"pcache")==0 ){ 504 if( i>=argc-2 ) abendError("missing arguments on %s", argv[i]); 505 nPCache = integerValue(argv[i+1]); 506 szPCache = integerValue(argv[i+2]); 507 i += 2; 508 }else 509 if( strcmp(z,"quiet")==0 || strcmp(z,"q")==0 ){ 510 quietFlag = 1; 511 verboseFlag = 0; 512 }else 513 if( strcmp(z,"scratch")==0 ){ 514 if( i>=argc-2 ) abendError("missing arguments on %s", argv[i]); 515 nScratch = integerValue(argv[i+1]); 516 szScratch = integerValue(argv[i+2]); 517 i += 2; 518 }else 519 if( strcmp(z, "unique-cases")==0 ){ 520 if( i>=argc-1 ) abendError("missing arguments on %s", argv[i]); 521 if( zDataOut ) abendError("only one --minimize allowed"); 522 zDataOut = argv[++i]; 523 }else 524 if( strcmp(z,"utf16le")==0 ){ 525 zEncoding = "utf16le"; 526 }else 527 if( strcmp(z,"utf16be")==0 ){ 528 zEncoding = "utf16be"; 529 }else 530 if( strcmp(z,"verbose")==0 || strcmp(z,"v")==0 ){ 531 quietFlag = 0; 532 verboseFlag = 1; 533 }else 534 { 535 abendError("unknown option: %s", argv[i]); 536 } 537 }else{ 538 addNewInFile: 539 nInFile++; 540 azInFile = realloc(azInFile, sizeof(azInFile[0])*nInFile); 541 if( azInFile==0 ) abendError("out of memory"); 542 azInFile[nInFile-1] = argv[i]; 543 } 544 } 545 546 /* Do global SQLite initialization */ 547 sqlite3_config(SQLITE_CONFIG_LOG, verboseFlag ? shellLog : shellLogNoop, 0); 548 if( nHeap>0 ){ 549 pHeap = malloc( nHeap ); 550 if( pHeap==0 ) fatalError("cannot allocate %d-byte heap\n", nHeap); 551 rc = sqlite3_config(SQLITE_CONFIG_HEAP, pHeap, nHeap, mnHeap); 552 if( rc ) abendError("heap configuration failed: %d\n", rc); 553 } 554 if( oomFlag ){ 555 sqlite3_config(SQLITE_CONFIG_GETMALLOC, &g.sOrigMem); 556 g.sOomMem = g.sOrigMem; 557 g.sOomMem.xMalloc = oomMalloc; 558 g.sOomMem.xRealloc = oomRealloc; 559 sqlite3_config(SQLITE_CONFIG_MALLOC, &g.sOomMem); 560 } 561 if( nLook>0 ){ 562 sqlite3_config(SQLITE_CONFIG_LOOKASIDE, 0, 0); 563 if( szLook>0 ){ 564 pLook = malloc( nLook*szLook ); 565 if( pLook==0 ) fatalError("out of memory"); 566 } 567 } 568 if( nScratch>0 && szScratch>0 ){ 569 pScratch = malloc( nScratch*(sqlite3_int64)szScratch ); 570 if( pScratch==0 ) fatalError("cannot allocate %lld-byte scratch", 571 nScratch*(sqlite3_int64)szScratch); 572 rc = sqlite3_config(SQLITE_CONFIG_SCRATCH, pScratch, szScratch, nScratch); 573 if( rc ) abendError("scratch configuration failed: %d\n", rc); 574 } 575 if( nPCache>0 && szPCache>0 ){ 576 pPCache = malloc( nPCache*(sqlite3_int64)szPCache ); 577 if( pPCache==0 ) fatalError("cannot allocate %lld-byte pcache", 578 nPCache*(sqlite3_int64)szPCache); 579 rc = sqlite3_config(SQLITE_CONFIG_PAGECACHE, pPCache, szPCache, nPCache); 580 if( rc ) abendError("pcache configuration failed: %d", rc); 581 } 582 583 /* If the --unique-cases option was supplied, open the database that will 584 ** be used to gather unique test cases. 585 */ 586 if( zDataOut ){ 587 rc = sqlite3_open(":memory:", &dataDb); 588 if( rc ) abendError("cannot open :memory: database"); 589 rc = sqlite3_exec(dataDb, 590 "CREATE TABLE testcase(sql BLOB PRIMARY KEY, tm) WITHOUT ROWID;",0,0,0); 591 if( rc ) abendError("%s", sqlite3_errmsg(dataDb)); 592 rc = sqlite3_prepare_v2(dataDb, 593 "INSERT OR IGNORE INTO testcase(sql,tm)VALUES(?1,?2)", 594 -1, &pStmt, 0); 595 if( rc ) abendError("%s", sqlite3_errmsg(dataDb)); 596 } 597 598 /* Initialize the input buffer used to hold SQL text */ 599 if( nInFile==0 ) nInFile = 1; 600 nAlloc = 1000; 601 zIn = malloc(nAlloc); 602 if( zIn==0 ) fatalError("out of memory"); 603 604 /* Loop over all input files */ 605 for(jj=0; jj<nInFile; jj++){ 606 607 /* Read the complete content of the next input file into zIn[] */ 608 FILE *in; 609 if( azInFile ){ 610 int j, k; 611 in = fopen(azInFile[jj],"rb"); 612 if( in==0 ){ 613 abendError("cannot open %s for reading", azInFile[jj]); 614 } 615 zPrompt = azInFile[jj]; 616 for(j=k=0; zPrompt[j]; j++) if( zPrompt[j]=='/' ) k = j+1; 617 zPrompt += k; 618 }else{ 619 in = stdin; 620 zPrompt = "<stdin>"; 621 } 622 while( !feof(in) ){ 623 got = fread(zIn+nIn, 1, nAlloc-nIn-1, in); 624 nIn += (int)got; 625 zIn[nIn] = 0; 626 if( got==0 ) break; 627 if( nAlloc - nIn - 1 < 100 ){ 628 nAlloc += nAlloc+1000; 629 zIn = realloc(zIn, nAlloc); 630 if( zIn==0 ) fatalError("out of memory"); 631 } 632 } 633 if( in!=stdin ) fclose(in); 634 lastPct = -1; 635 636 /* Skip initial lines of the input file that begin with "#" */ 637 for(i=0; i<nIn; i=iNext+1){ 638 if( zIn[i]!='#' ) break; 639 for(iNext=i+1; iNext<nIn && zIn[iNext]!='\n'; iNext++){} 640 } 641 nHeader = i; 642 643 /* Process all test cases contained within the input file. 644 */ 645 for(; i<nIn; i=iNext, nTest++, g.zTestName[0]=0){ 646 char cSaved; 647 if( strncmp(&zIn[i], "/****<",6)==0 ){ 648 char *z = strstr(&zIn[i], ">****/"); 649 if( z ){ 650 z += 6; 651 sqlite3_snprintf(sizeof(g.zTestName), g.zTestName, "%.*s", 652 (int)(z-&zIn[i]) - 12, &zIn[i+6]); 653 if( verboseFlag ){ 654 printf("%.*s\n", (int)(z-&zIn[i]), &zIn[i]); 655 fflush(stdout); 656 } 657 i += (int)(z-&zIn[i]); 658 multiTest = 1; 659 } 660 } 661 for(iNext=i; iNext<nIn && strncmp(&zIn[iNext],"/****<",6)!=0; iNext++){} 662 cSaved = zIn[iNext]; 663 zIn[iNext] = 0; 664 665 666 /* Print out the SQL of the next test case is --verbose is enabled 667 */ 668 zSql = &zIn[i]; 669 if( verboseFlag ){ 670 printf("INPUT (offset: %d, size: %d): [%s]\n", 671 i, (int)strlen(&zIn[i]), &zIn[i]); 672 }else if( multiTest && !quietFlag ){ 673 if( oomFlag ){ 674 printf("%s\n", g.zTestName); 675 }else{ 676 int pct = (10*iNext)/nIn; 677 if( pct!=lastPct ){ 678 if( lastPct<0 ) printf("%s:", zPrompt); 679 printf(" %d%%", pct*10); 680 lastPct = pct; 681 } 682 } 683 }else if( nInFile>1 ){ 684 printf("%s\n", zPrompt); 685 } 686 fflush(stdout); 687 688 /* Run the next test case. Run it multiple times in --oom mode 689 */ 690 if( oomFlag ){ 691 oomCnt = g.iOomCntdown = 1; 692 g.nOomFault = 0; 693 g.bOomOnce = 1; 694 if( verboseFlag ){ 695 printf("Once.%d\n", oomCnt); 696 fflush(stdout); 697 } 698 }else{ 699 oomCnt = 0; 700 } 701 do{ 702 if( zDbName ){ 703 rc = sqlite3_open_v2(zDbName, &db, SQLITE_OPEN_READWRITE, 0); 704 if( rc!=SQLITE_OK ){ 705 abendError("Cannot open database file %s", zDbName); 706 } 707 }else{ 708 rc = sqlite3_open_v2( 709 "main.db", &db, 710 SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE | SQLITE_OPEN_MEMORY, 711 0); 712 if( rc!=SQLITE_OK ){ 713 abendError("Unable to open the in-memory database"); 714 } 715 } 716 if( pLook ){ 717 rc = sqlite3_db_config(db, SQLITE_DBCONFIG_LOOKASIDE,pLook,szLook,nLook); 718 if( rc!=SQLITE_OK ) abendError("lookaside configuration filed: %d", rc); 719 } 720 #ifndef SQLITE_OMIT_TRACE 721 sqlite3_trace(db, verboseFlag ? traceCallback : traceNoop, 0); 722 #endif 723 sqlite3_create_function(db, "eval", 1, SQLITE_UTF8, 0, sqlEvalFunc, 0, 0); 724 sqlite3_create_function(db, "eval", 2, SQLITE_UTF8, 0, sqlEvalFunc, 0, 0); 725 sqlite3_limit(db, SQLITE_LIMIT_LENGTH, 1000000); 726 if( zEncoding ) sqlexec(db, "PRAGMA encoding=%s", zEncoding); 727 if( pageSize ) sqlexec(db, "PRAGMA pagesize=%d", pageSize); 728 if( doAutovac ) sqlexec(db, "PRAGMA auto_vacuum=FULL"); 729 iStart = timeOfDay(); 730 g.bOomEnable = 1; 731 if( verboseFlag ){ 732 zErrMsg = 0; 733 rc = sqlite3_exec(db, zSql, execCallback, 0, &zErrMsg); 734 if( zErrMsg ){ 735 sqlite3_snprintf(sizeof(zErrBuf),zErrBuf,"%z", zErrMsg); 736 zErrMsg = 0; 737 } 738 }else { 739 rc = sqlite3_exec(db, zSql, execNoop, 0, 0); 740 } 741 g.bOomEnable = 0; 742 iEnd = timeOfDay(); 743 rc = sqlite3_close(db); 744 if( rc ){ 745 abendError("sqlite3_close() failed with rc=%d", rc); 746 } 747 if( !zDataOut && sqlite3_memory_used()>0 ){ 748 abendError("memory in use after close: %lld bytes",sqlite3_memory_used()); 749 } 750 if( oomFlag ){ 751 /* Limit the number of iterations of the OOM loop to OOM_MAX. If the 752 ** first pass (single failure) exceeds 2/3rds of OOM_MAX this skip the 753 ** second pass (continuous failure after first) completely. */ 754 if( g.nOomFault==0 || oomCnt>OOM_MAX ){ 755 if( g.bOomOnce && oomCnt<=(OOM_MAX*2/3) ){ 756 oomCnt = g.iOomCntdown = 1; 757 g.bOomOnce = 0; 758 }else{ 759 oomCnt = 0; 760 } 761 }else{ 762 g.iOomCntdown = ++oomCnt; 763 g.nOomFault = 0; 764 } 765 if( oomCnt ){ 766 if( verboseFlag ){ 767 printf("%s.%d\n", g.bOomOnce ? "Once" : "Multi", oomCnt); 768 fflush(stdout); 769 } 770 nTest++; 771 } 772 } 773 }while( oomCnt>0 ); 774 775 /* Store unique test cases in the in the dataDb database if the 776 ** --unique-cases flag is present 777 */ 778 if( zDataOut ){ 779 sqlite3_bind_blob(pStmt, 1, &zIn[i], iNext-i, SQLITE_STATIC); 780 sqlite3_bind_int64(pStmt, 2, iEnd - iStart); 781 rc = sqlite3_step(pStmt); 782 if( rc!=SQLITE_DONE ) abendError("%s", sqlite3_errmsg(dataDb)); 783 sqlite3_reset(pStmt); 784 } 785 786 /* Free the SQL from the current test case 787 */ 788 if( zToFree ){ 789 sqlite3_free(zToFree); 790 zToFree = 0; 791 } 792 zIn[iNext] = cSaved; 793 794 /* Show test-case results in --verbose mode 795 */ 796 if( verboseFlag ){ 797 printf("RESULT-CODE: %d\n", rc); 798 if( zErrMsg ){ 799 printf("ERROR-MSG: [%s]\n", zErrBuf); 800 } 801 fflush(stdout); 802 } 803 804 /* Simulate an error if the TEST_FAILURE environment variable is "5". 805 ** This is used to verify that automated test script really do spot 806 ** errors that occur in this test program. 807 */ 808 if( zFailCode ){ 809 if( zFailCode[0]=='5' && zFailCode[1]==0 ){ 810 abendError("simulated failure"); 811 }else if( zFailCode[0]!=0 ){ 812 /* If TEST_FAILURE is something other than 5, just exit the test 813 ** early */ 814 printf("\nExit early due to TEST_FAILURE being set"); 815 break; 816 } 817 } 818 } 819 if( !verboseFlag && multiTest && !quietFlag && !oomFlag ) printf("\n"); 820 } 821 822 /* Report total number of tests run 823 */ 824 if( nTest>1 && !quietFlag ){ 825 sqlite3_int64 iElapse = timeOfDay() - iBegin; 826 printf("%s: 0 errors out of %d tests in %d.%03d seconds\nSQLite %s %s\n", 827 g.zArgv0, nTest, (int)(iElapse/1000), (int)(iElapse%1000), 828 sqlite3_libversion(), sqlite3_sourceid()); 829 } 830 831 /* Write the unique test cases if the --unique-cases flag was used 832 */ 833 if( zDataOut ){ 834 int n = 0; 835 FILE *out = fopen(zDataOut, "wb"); 836 if( out==0 ) abendError("cannot open %s for writing", zDataOut); 837 if( nHeader>0 ) fwrite(zIn, nHeader, 1, out); 838 sqlite3_finalize(pStmt); 839 rc = sqlite3_prepare_v2(dataDb, "SELECT sql, tm FROM testcase ORDER BY tm, sql", 840 -1, &pStmt, 0); 841 if( rc ) abendError("%s", sqlite3_errmsg(dataDb)); 842 while( sqlite3_step(pStmt)==SQLITE_ROW ){ 843 fprintf(out,"/****<%d:%dms>****/", ++n, sqlite3_column_int(pStmt,1)); 844 fwrite(sqlite3_column_blob(pStmt,0),sqlite3_column_bytes(pStmt,0),1,out); 845 } 846 fclose(out); 847 sqlite3_finalize(pStmt); 848 sqlite3_close(dataDb); 849 } 850 851 /* Clean up and exit. 852 */ 853 free(azInFile); 854 free(zIn); 855 free(pHeap); 856 free(pLook); 857 free(pScratch); 858 free(pPCache); 859 return 0; 860 } 861