1 /* 2 ** 2007 August 15 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 file contains code used to implement test interfaces to the 14 ** memory allocation subsystem. 15 */ 16 #include "sqliteInt.h" 17 #include "tcl.h" 18 #include <stdlib.h> 19 #include <string.h> 20 #include <assert.h> 21 22 /* 23 ** This structure is used to encapsulate the global state variables used 24 ** by malloc() fault simulation. 25 */ 26 static struct MemFault { 27 int iCountdown; /* Number of pending successes before a failure */ 28 int nRepeat; /* Number of times to repeat the failure */ 29 int nBenign; /* Number of benign failures seen since last config */ 30 int nFail; /* Number of failures seen since last config */ 31 u8 enable; /* True if enabled */ 32 int isInstalled; /* True if the fault simulation layer is installed */ 33 int isBenignMode; /* True if malloc failures are considered benign */ 34 sqlite3_mem_methods m; /* 'Real' malloc implementation */ 35 } memfault; 36 37 /* 38 ** This routine exists as a place to set a breakpoint that will 39 ** fire on any simulated malloc() failure. 40 */ 41 static void sqlite3Fault(void){ 42 static int cnt = 0; 43 cnt++; 44 } 45 46 /* 47 ** Check to see if a fault should be simulated. Return true to simulate 48 ** the fault. Return false if the fault should not be simulated. 49 */ 50 static int faultsimStep(void){ 51 if( likely(!memfault.enable) ){ 52 return 0; 53 } 54 if( memfault.iCountdown>0 ){ 55 memfault.iCountdown--; 56 return 0; 57 } 58 sqlite3Fault(); 59 memfault.nFail++; 60 if( memfault.isBenignMode>0 ){ 61 memfault.nBenign++; 62 } 63 memfault.nRepeat--; 64 if( memfault.nRepeat<=0 ){ 65 memfault.enable = 0; 66 } 67 return 1; 68 } 69 70 /* 71 ** A version of sqlite3_mem_methods.xMalloc() that includes fault simulation 72 ** logic. 73 */ 74 static void *faultsimMalloc(int n){ 75 void *p = 0; 76 if( !faultsimStep() ){ 77 p = memfault.m.xMalloc(n); 78 } 79 return p; 80 } 81 82 83 /* 84 ** A version of sqlite3_mem_methods.xRealloc() that includes fault simulation 85 ** logic. 86 */ 87 static void *faultsimRealloc(void *pOld, int n){ 88 void *p = 0; 89 if( !faultsimStep() ){ 90 p = memfault.m.xRealloc(pOld, n); 91 } 92 return p; 93 } 94 95 /* 96 ** The following method calls are passed directly through to the underlying 97 ** malloc system: 98 ** 99 ** xFree 100 ** xSize 101 ** xRoundup 102 ** xInit 103 ** xShutdown 104 */ 105 static void faultsimFree(void *p){ 106 memfault.m.xFree(p); 107 } 108 static int faultsimSize(void *p){ 109 return memfault.m.xSize(p); 110 } 111 static int faultsimRoundup(int n){ 112 return memfault.m.xRoundup(n); 113 } 114 static int faultsimInit(void *p){ 115 return memfault.m.xInit(memfault.m.pAppData); 116 } 117 static void faultsimShutdown(void *p){ 118 memfault.m.xShutdown(memfault.m.pAppData); 119 } 120 121 /* 122 ** This routine configures the malloc failure simulation. After 123 ** calling this routine, the next nDelay mallocs will succeed, followed 124 ** by a block of nRepeat failures, after which malloc() calls will begin 125 ** to succeed again. 126 */ 127 static void faultsimConfig(int nDelay, int nRepeat){ 128 memfault.iCountdown = nDelay; 129 memfault.nRepeat = nRepeat; 130 memfault.nBenign = 0; 131 memfault.nFail = 0; 132 memfault.enable = nDelay>=0; 133 134 /* Sometimes, when running multi-threaded tests, the isBenignMode 135 ** variable is not properly incremented/decremented so that it is 136 ** 0 when not inside a benign malloc block. This doesn't affect 137 ** the multi-threaded tests, as they do not use this system. But 138 ** it does affect OOM tests run later in the same process. So 139 ** zero the variable here, just to be sure. 140 */ 141 memfault.isBenignMode = 0; 142 } 143 144 /* 145 ** Return the number of faults (both hard and benign faults) that have 146 ** occurred since the injector was last configured. 147 */ 148 static int faultsimFailures(void){ 149 return memfault.nFail; 150 } 151 152 /* 153 ** Return the number of benign faults that have occurred since the 154 ** injector was last configured. 155 */ 156 static int faultsimBenignFailures(void){ 157 return memfault.nBenign; 158 } 159 160 /* 161 ** Return the number of successes that will occur before the next failure. 162 ** If no failures are scheduled, return -1. 163 */ 164 static int faultsimPending(void){ 165 if( memfault.enable ){ 166 return memfault.iCountdown; 167 }else{ 168 return -1; 169 } 170 } 171 172 173 static void faultsimBeginBenign(void){ 174 memfault.isBenignMode++; 175 } 176 static void faultsimEndBenign(void){ 177 memfault.isBenignMode--; 178 } 179 180 /* 181 ** Add or remove the fault-simulation layer using sqlite3_config(). If 182 ** the argument is non-zero, the 183 */ 184 static int faultsimInstall(int install){ 185 static struct sqlite3_mem_methods m = { 186 faultsimMalloc, /* xMalloc */ 187 faultsimFree, /* xFree */ 188 faultsimRealloc, /* xRealloc */ 189 faultsimSize, /* xSize */ 190 faultsimRoundup, /* xRoundup */ 191 faultsimInit, /* xInit */ 192 faultsimShutdown, /* xShutdown */ 193 0 /* pAppData */ 194 }; 195 int rc; 196 197 install = (install ? 1 : 0); 198 assert(memfault.isInstalled==1 || memfault.isInstalled==0); 199 200 if( install==memfault.isInstalled ){ 201 return SQLITE_ERROR; 202 } 203 204 if( install ){ 205 rc = sqlite3_config(SQLITE_CONFIG_GETMALLOC, &memfault.m); 206 assert(memfault.m.xMalloc); 207 if( rc==SQLITE_OK ){ 208 rc = sqlite3_config(SQLITE_CONFIG_MALLOC, &m); 209 } 210 sqlite3_test_control(SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS, 211 faultsimBeginBenign, faultsimEndBenign 212 ); 213 }else{ 214 sqlite3_mem_methods m; 215 assert(memfault.m.xMalloc); 216 217 /* One should be able to reset the default memory allocator by storing 218 ** a zeroed allocator then calling GETMALLOC. */ 219 memset(&m, 0, sizeof(m)); 220 sqlite3_config(SQLITE_CONFIG_MALLOC, &m); 221 sqlite3_config(SQLITE_CONFIG_GETMALLOC, &m); 222 assert( memcmp(&m, &memfault.m, sizeof(m))==0 ); 223 224 rc = sqlite3_config(SQLITE_CONFIG_MALLOC, &memfault.m); 225 sqlite3_test_control(SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS, 0, 0); 226 } 227 228 if( rc==SQLITE_OK ){ 229 memfault.isInstalled = 1; 230 } 231 return rc; 232 } 233 234 #ifdef SQLITE_TEST 235 236 /* 237 ** This function is implemented in main.c. Returns a pointer to a static 238 ** buffer containing the symbolic SQLite error code that corresponds to 239 ** the least-significant 8-bits of the integer passed as an argument. 240 ** For example: 241 ** 242 ** sqlite3ErrName(1) -> "SQLITE_ERROR" 243 */ 244 extern const char *sqlite3ErrName(int); 245 246 /* 247 ** Transform pointers to text and back again 248 */ 249 static void pointerToText(void *p, char *z){ 250 static const char zHex[] = "0123456789abcdef"; 251 int i, k; 252 unsigned int u; 253 sqlite3_uint64 n; 254 if( p==0 ){ 255 strcpy(z, "0"); 256 return; 257 } 258 if( sizeof(n)==sizeof(p) ){ 259 memcpy(&n, &p, sizeof(p)); 260 }else if( sizeof(u)==sizeof(p) ){ 261 memcpy(&u, &p, sizeof(u)); 262 n = u; 263 }else{ 264 assert( 0 ); 265 } 266 for(i=0, k=sizeof(p)*2-1; i<sizeof(p)*2; i++, k--){ 267 z[k] = zHex[n&0xf]; 268 n >>= 4; 269 } 270 z[sizeof(p)*2] = 0; 271 } 272 static int hexToInt(int h){ 273 if( h>='0' && h<='9' ){ 274 return h - '0'; 275 }else if( h>='a' && h<='f' ){ 276 return h - 'a' + 10; 277 }else{ 278 return -1; 279 } 280 } 281 static int textToPointer(const char *z, void **pp){ 282 sqlite3_uint64 n = 0; 283 int i; 284 unsigned int u; 285 for(i=0; i<sizeof(void*)*2 && z[0]; i++){ 286 int v; 287 v = hexToInt(*z++); 288 if( v<0 ) return TCL_ERROR; 289 n = n*16 + v; 290 } 291 if( *z!=0 ) return TCL_ERROR; 292 if( sizeof(n)==sizeof(*pp) ){ 293 memcpy(pp, &n, sizeof(n)); 294 }else if( sizeof(u)==sizeof(*pp) ){ 295 u = (unsigned int)n; 296 memcpy(pp, &u, sizeof(u)); 297 }else{ 298 assert( 0 ); 299 } 300 return TCL_OK; 301 } 302 303 /* 304 ** Usage: sqlite3_malloc NBYTES 305 ** 306 ** Raw test interface for sqlite3_malloc(). 307 */ 308 static int test_malloc( 309 void * clientData, 310 Tcl_Interp *interp, 311 int objc, 312 Tcl_Obj *CONST objv[] 313 ){ 314 int nByte; 315 void *p; 316 char zOut[100]; 317 if( objc!=2 ){ 318 Tcl_WrongNumArgs(interp, 1, objv, "NBYTES"); 319 return TCL_ERROR; 320 } 321 if( Tcl_GetIntFromObj(interp, objv[1], &nByte) ) return TCL_ERROR; 322 p = sqlite3_malloc((unsigned)nByte); 323 pointerToText(p, zOut); 324 Tcl_AppendResult(interp, zOut, NULL); 325 return TCL_OK; 326 } 327 328 /* 329 ** Usage: sqlite3_realloc PRIOR NBYTES 330 ** 331 ** Raw test interface for sqlite3_realloc(). 332 */ 333 static int test_realloc( 334 void * clientData, 335 Tcl_Interp *interp, 336 int objc, 337 Tcl_Obj *CONST objv[] 338 ){ 339 int nByte; 340 void *pPrior, *p; 341 char zOut[100]; 342 if( objc!=3 ){ 343 Tcl_WrongNumArgs(interp, 1, objv, "PRIOR NBYTES"); 344 return TCL_ERROR; 345 } 346 if( Tcl_GetIntFromObj(interp, objv[2], &nByte) ) return TCL_ERROR; 347 if( textToPointer(Tcl_GetString(objv[1]), &pPrior) ){ 348 Tcl_AppendResult(interp, "bad pointer: ", Tcl_GetString(objv[1]), (char*)0); 349 return TCL_ERROR; 350 } 351 p = sqlite3_realloc(pPrior, (unsigned)nByte); 352 pointerToText(p, zOut); 353 Tcl_AppendResult(interp, zOut, NULL); 354 return TCL_OK; 355 } 356 357 /* 358 ** Usage: sqlite3_free PRIOR 359 ** 360 ** Raw test interface for sqlite3_free(). 361 */ 362 static int test_free( 363 void * clientData, 364 Tcl_Interp *interp, 365 int objc, 366 Tcl_Obj *CONST objv[] 367 ){ 368 void *pPrior; 369 if( objc!=2 ){ 370 Tcl_WrongNumArgs(interp, 1, objv, "PRIOR"); 371 return TCL_ERROR; 372 } 373 if( textToPointer(Tcl_GetString(objv[1]), &pPrior) ){ 374 Tcl_AppendResult(interp, "bad pointer: ", Tcl_GetString(objv[1]), (char*)0); 375 return TCL_ERROR; 376 } 377 sqlite3_free(pPrior); 378 return TCL_OK; 379 } 380 381 /* 382 ** These routines are in test_hexio.c 383 */ 384 int sqlite3TestHexToBin(const char *, int, char *); 385 int sqlite3TestBinToHex(char*,int); 386 387 /* 388 ** Usage: memset ADDRESS SIZE HEX 389 ** 390 ** Set a chunk of memory (obtained from malloc, probably) to a 391 ** specified hex pattern. 392 */ 393 static int test_memset( 394 void * clientData, 395 Tcl_Interp *interp, 396 int objc, 397 Tcl_Obj *CONST objv[] 398 ){ 399 void *p; 400 int size, n, i; 401 char *zHex; 402 char *zOut; 403 char zBin[100]; 404 405 if( objc!=4 ){ 406 Tcl_WrongNumArgs(interp, 1, objv, "ADDRESS SIZE HEX"); 407 return TCL_ERROR; 408 } 409 if( textToPointer(Tcl_GetString(objv[1]), &p) ){ 410 Tcl_AppendResult(interp, "bad pointer: ", Tcl_GetString(objv[1]), (char*)0); 411 return TCL_ERROR; 412 } 413 if( Tcl_GetIntFromObj(interp, objv[2], &size) ){ 414 return TCL_ERROR; 415 } 416 if( size<=0 ){ 417 Tcl_AppendResult(interp, "size must be positive", (char*)0); 418 return TCL_ERROR; 419 } 420 zHex = Tcl_GetStringFromObj(objv[3], &n); 421 if( n>sizeof(zBin)*2 ) n = sizeof(zBin)*2; 422 n = sqlite3TestHexToBin(zHex, n, zBin); 423 if( n==0 ){ 424 Tcl_AppendResult(interp, "no data", (char*)0); 425 return TCL_ERROR; 426 } 427 zOut = p; 428 for(i=0; i<size; i++){ 429 zOut[i] = zBin[i%n]; 430 } 431 return TCL_OK; 432 } 433 434 /* 435 ** Usage: memget ADDRESS SIZE 436 ** 437 ** Return memory as hexadecimal text. 438 */ 439 static int test_memget( 440 void * clientData, 441 Tcl_Interp *interp, 442 int objc, 443 Tcl_Obj *CONST objv[] 444 ){ 445 void *p; 446 int size, n; 447 char *zBin; 448 char zHex[100]; 449 450 if( objc!=3 ){ 451 Tcl_WrongNumArgs(interp, 1, objv, "ADDRESS SIZE"); 452 return TCL_ERROR; 453 } 454 if( textToPointer(Tcl_GetString(objv[1]), &p) ){ 455 Tcl_AppendResult(interp, "bad pointer: ", Tcl_GetString(objv[1]), (char*)0); 456 return TCL_ERROR; 457 } 458 if( Tcl_GetIntFromObj(interp, objv[2], &size) ){ 459 return TCL_ERROR; 460 } 461 if( size<=0 ){ 462 Tcl_AppendResult(interp, "size must be positive", (char*)0); 463 return TCL_ERROR; 464 } 465 zBin = p; 466 while( size>0 ){ 467 if( size>(sizeof(zHex)-1)/2 ){ 468 n = (sizeof(zHex)-1)/2; 469 }else{ 470 n = size; 471 } 472 memcpy(zHex, zBin, n); 473 zBin += n; 474 size -= n; 475 sqlite3TestBinToHex(zHex, n); 476 Tcl_AppendResult(interp, zHex, (char*)0); 477 } 478 return TCL_OK; 479 } 480 481 /* 482 ** Usage: sqlite3_memory_used 483 ** 484 ** Raw test interface for sqlite3_memory_used(). 485 */ 486 static int test_memory_used( 487 void * clientData, 488 Tcl_Interp *interp, 489 int objc, 490 Tcl_Obj *CONST objv[] 491 ){ 492 Tcl_SetObjResult(interp, Tcl_NewWideIntObj(sqlite3_memory_used())); 493 return TCL_OK; 494 } 495 496 /* 497 ** Usage: sqlite3_memory_highwater ?RESETFLAG? 498 ** 499 ** Raw test interface for sqlite3_memory_highwater(). 500 */ 501 static int test_memory_highwater( 502 void * clientData, 503 Tcl_Interp *interp, 504 int objc, 505 Tcl_Obj *CONST objv[] 506 ){ 507 int resetFlag = 0; 508 if( objc!=1 && objc!=2 ){ 509 Tcl_WrongNumArgs(interp, 1, objv, "?RESET?"); 510 return TCL_ERROR; 511 } 512 if( objc==2 ){ 513 if( Tcl_GetBooleanFromObj(interp, objv[1], &resetFlag) ) return TCL_ERROR; 514 } 515 Tcl_SetObjResult(interp, 516 Tcl_NewWideIntObj(sqlite3_memory_highwater(resetFlag))); 517 return TCL_OK; 518 } 519 520 /* 521 ** Usage: sqlite3_memdebug_backtrace DEPTH 522 ** 523 ** Set the depth of backtracing. If SQLITE_MEMDEBUG is not defined 524 ** then this routine is a no-op. 525 */ 526 static int test_memdebug_backtrace( 527 void * clientData, 528 Tcl_Interp *interp, 529 int objc, 530 Tcl_Obj *CONST objv[] 531 ){ 532 int depth; 533 if( objc!=2 ){ 534 Tcl_WrongNumArgs(interp, 1, objv, "DEPT"); 535 return TCL_ERROR; 536 } 537 if( Tcl_GetIntFromObj(interp, objv[1], &depth) ) return TCL_ERROR; 538 #ifdef SQLITE_MEMDEBUG 539 { 540 extern void sqlite3MemdebugBacktrace(int); 541 sqlite3MemdebugBacktrace(depth); 542 } 543 #endif 544 return TCL_OK; 545 } 546 547 /* 548 ** Usage: sqlite3_memdebug_dump FILENAME 549 ** 550 ** Write a summary of unfreed memory to FILENAME. 551 */ 552 static int test_memdebug_dump( 553 void * clientData, 554 Tcl_Interp *interp, 555 int objc, 556 Tcl_Obj *CONST objv[] 557 ){ 558 if( objc!=2 ){ 559 Tcl_WrongNumArgs(interp, 1, objv, "FILENAME"); 560 return TCL_ERROR; 561 } 562 #if defined(SQLITE_MEMDEBUG) || defined(SQLITE_MEMORY_SIZE) \ 563 || defined(SQLITE_POW2_MEMORY_SIZE) 564 { 565 extern void sqlite3MemdebugDump(const char*); 566 sqlite3MemdebugDump(Tcl_GetString(objv[1])); 567 } 568 #endif 569 return TCL_OK; 570 } 571 572 /* 573 ** Usage: sqlite3_memdebug_malloc_count 574 ** 575 ** Return the total number of times malloc() has been called. 576 */ 577 static int test_memdebug_malloc_count( 578 void * clientData, 579 Tcl_Interp *interp, 580 int objc, 581 Tcl_Obj *CONST objv[] 582 ){ 583 int nMalloc = -1; 584 if( objc!=1 ){ 585 Tcl_WrongNumArgs(interp, 1, objv, ""); 586 return TCL_ERROR; 587 } 588 #if defined(SQLITE_MEMDEBUG) 589 { 590 extern int sqlite3MemdebugMallocCount(); 591 nMalloc = sqlite3MemdebugMallocCount(); 592 } 593 #endif 594 Tcl_SetObjResult(interp, Tcl_NewIntObj(nMalloc)); 595 return TCL_OK; 596 } 597 598 599 /* 600 ** Usage: sqlite3_memdebug_fail COUNTER ?OPTIONS? 601 ** 602 ** where options are: 603 ** 604 ** -repeat <count> 605 ** -benigncnt <varname> 606 ** 607 ** Arrange for a simulated malloc() failure after COUNTER successes. 608 ** If a repeat count is specified, the fault is repeated that many 609 ** times. 610 ** 611 ** Each call to this routine overrides the prior counter value. 612 ** This routine returns the number of simulated failures that have 613 ** happened since the previous call to this routine. 614 ** 615 ** To disable simulated failures, use a COUNTER of -1. 616 */ 617 static int test_memdebug_fail( 618 void * clientData, 619 Tcl_Interp *interp, 620 int objc, 621 Tcl_Obj *CONST objv[] 622 ){ 623 int ii; 624 int iFail; 625 int nRepeat = 1; 626 Tcl_Obj *pBenignCnt = 0; 627 int nBenign; 628 int nFail = 0; 629 630 if( objc<2 ){ 631 Tcl_WrongNumArgs(interp, 1, objv, "COUNTER ?OPTIONS?"); 632 return TCL_ERROR; 633 } 634 if( Tcl_GetIntFromObj(interp, objv[1], &iFail) ) return TCL_ERROR; 635 636 for(ii=2; ii<objc; ii+=2){ 637 int nOption; 638 char *zOption = Tcl_GetStringFromObj(objv[ii], &nOption); 639 char *zErr = 0; 640 641 if( nOption>1 && strncmp(zOption, "-repeat", nOption)==0 ){ 642 if( ii==(objc-1) ){ 643 zErr = "option requires an argument: "; 644 }else{ 645 if( Tcl_GetIntFromObj(interp, objv[ii+1], &nRepeat) ){ 646 return TCL_ERROR; 647 } 648 } 649 }else if( nOption>1 && strncmp(zOption, "-benigncnt", nOption)==0 ){ 650 if( ii==(objc-1) ){ 651 zErr = "option requires an argument: "; 652 }else{ 653 pBenignCnt = objv[ii+1]; 654 } 655 }else{ 656 zErr = "unknown option: "; 657 } 658 659 if( zErr ){ 660 Tcl_AppendResult(interp, zErr, zOption, 0); 661 return TCL_ERROR; 662 } 663 } 664 665 nBenign = faultsimBenignFailures(); 666 nFail = faultsimFailures(); 667 faultsimConfig(iFail, nRepeat); 668 669 if( pBenignCnt ){ 670 Tcl_ObjSetVar2(interp, pBenignCnt, 0, Tcl_NewIntObj(nBenign), 0); 671 } 672 Tcl_SetObjResult(interp, Tcl_NewIntObj(nFail)); 673 return TCL_OK; 674 } 675 676 /* 677 ** Usage: sqlite3_memdebug_pending 678 ** 679 ** Return the number of malloc() calls that will succeed before a 680 ** simulated failure occurs. A negative return value indicates that 681 ** no malloc() failure is scheduled. 682 */ 683 static int test_memdebug_pending( 684 void * clientData, 685 Tcl_Interp *interp, 686 int objc, 687 Tcl_Obj *CONST objv[] 688 ){ 689 int nPending; 690 if( objc!=1 ){ 691 Tcl_WrongNumArgs(interp, 1, objv, ""); 692 return TCL_ERROR; 693 } 694 nPending = faultsimPending(); 695 Tcl_SetObjResult(interp, Tcl_NewIntObj(nPending)); 696 return TCL_OK; 697 } 698 699 /* 700 ** The following global variable keeps track of the number of tests 701 ** that have run. This variable is only useful when running in the 702 ** debugger. 703 */ 704 static int sqlite3_memdebug_title_count = 0; 705 706 /* 707 ** Usage: sqlite3_memdebug_settitle TITLE 708 ** 709 ** Set a title string stored with each allocation. The TITLE is 710 ** typically the name of the test that was running when the 711 ** allocation occurred. The TITLE is stored with the allocation 712 ** and can be used to figure out which tests are leaking memory. 713 ** 714 ** Each title overwrite the previous. 715 */ 716 static int test_memdebug_settitle( 717 void * clientData, 718 Tcl_Interp *interp, 719 int objc, 720 Tcl_Obj *CONST objv[] 721 ){ 722 sqlite3_memdebug_title_count++; 723 if( objc!=2 ){ 724 Tcl_WrongNumArgs(interp, 1, objv, "TITLE"); 725 return TCL_ERROR; 726 } 727 #ifdef SQLITE_MEMDEBUG 728 { 729 const char *zTitle; 730 extern int sqlite3MemdebugSettitle(const char*); 731 zTitle = Tcl_GetString(objv[1]); 732 sqlite3MemdebugSettitle(zTitle); 733 } 734 #endif 735 return TCL_OK; 736 } 737 738 #define MALLOC_LOG_FRAMES 10 739 #define MALLOC_LOG_KEYINTS ( \ 740 10 * ((sizeof(int)>=sizeof(void*)) ? 1 : sizeof(void*)/sizeof(int)) \ 741 ) 742 static Tcl_HashTable aMallocLog; 743 static int mallocLogEnabled = 0; 744 745 typedef struct MallocLog MallocLog; 746 struct MallocLog { 747 int nCall; 748 int nByte; 749 }; 750 751 #ifdef SQLITE_MEMDEBUG 752 static void test_memdebug_callback(int nByte, int nFrame, void **aFrame){ 753 if( mallocLogEnabled ){ 754 MallocLog *pLog; 755 Tcl_HashEntry *pEntry; 756 int isNew; 757 758 int aKey[MALLOC_LOG_KEYINTS]; 759 unsigned int nKey = sizeof(int)*MALLOC_LOG_KEYINTS; 760 761 memset(aKey, 0, nKey); 762 if( (sizeof(void*)*nFrame)<nKey ){ 763 nKey = nFrame*sizeof(void*); 764 } 765 memcpy(aKey, aFrame, nKey); 766 767 pEntry = Tcl_CreateHashEntry(&aMallocLog, (const char *)aKey, &isNew); 768 if( isNew ){ 769 pLog = (MallocLog *)Tcl_Alloc(sizeof(MallocLog)); 770 memset(pLog, 0, sizeof(MallocLog)); 771 Tcl_SetHashValue(pEntry, (ClientData)pLog); 772 }else{ 773 pLog = (MallocLog *)Tcl_GetHashValue(pEntry); 774 } 775 776 pLog->nCall++; 777 pLog->nByte += nByte; 778 } 779 } 780 #endif /* SQLITE_MEMDEBUG */ 781 782 static void test_memdebug_log_clear(void){ 783 Tcl_HashSearch search; 784 Tcl_HashEntry *pEntry; 785 for( 786 pEntry=Tcl_FirstHashEntry(&aMallocLog, &search); 787 pEntry; 788 pEntry=Tcl_NextHashEntry(&search) 789 ){ 790 MallocLog *pLog = (MallocLog *)Tcl_GetHashValue(pEntry); 791 Tcl_Free((char *)pLog); 792 } 793 Tcl_DeleteHashTable(&aMallocLog); 794 Tcl_InitHashTable(&aMallocLog, MALLOC_LOG_KEYINTS); 795 } 796 797 static int test_memdebug_log( 798 void * clientData, 799 Tcl_Interp *interp, 800 int objc, 801 Tcl_Obj *CONST objv[] 802 ){ 803 static int isInit = 0; 804 int iSub; 805 806 static const char *MB_strs[] = { "start", "stop", "dump", "clear", "sync" }; 807 enum MB_enum { 808 MB_LOG_START, MB_LOG_STOP, MB_LOG_DUMP, MB_LOG_CLEAR, MB_LOG_SYNC 809 }; 810 811 if( !isInit ){ 812 #ifdef SQLITE_MEMDEBUG 813 extern void sqlite3MemdebugBacktraceCallback( 814 void (*xBacktrace)(int, int, void **)); 815 sqlite3MemdebugBacktraceCallback(test_memdebug_callback); 816 #endif 817 Tcl_InitHashTable(&aMallocLog, MALLOC_LOG_KEYINTS); 818 isInit = 1; 819 } 820 821 if( objc<2 ){ 822 Tcl_WrongNumArgs(interp, 1, objv, "SUB-COMMAND ..."); 823 } 824 if( Tcl_GetIndexFromObj(interp, objv[1], MB_strs, "sub-command", 0, &iSub) ){ 825 return TCL_ERROR; 826 } 827 828 switch( (enum MB_enum)iSub ){ 829 case MB_LOG_START: 830 mallocLogEnabled = 1; 831 break; 832 case MB_LOG_STOP: 833 mallocLogEnabled = 0; 834 break; 835 case MB_LOG_DUMP: { 836 Tcl_HashSearch search; 837 Tcl_HashEntry *pEntry; 838 Tcl_Obj *pRet = Tcl_NewObj(); 839 840 assert(sizeof(Tcl_WideInt)>=sizeof(void*)); 841 842 for( 843 pEntry=Tcl_FirstHashEntry(&aMallocLog, &search); 844 pEntry; 845 pEntry=Tcl_NextHashEntry(&search) 846 ){ 847 Tcl_Obj *apElem[MALLOC_LOG_FRAMES+2]; 848 MallocLog *pLog = (MallocLog *)Tcl_GetHashValue(pEntry); 849 Tcl_WideInt *aKey = (Tcl_WideInt *)Tcl_GetHashKey(&aMallocLog, pEntry); 850 int ii; 851 852 apElem[0] = Tcl_NewIntObj(pLog->nCall); 853 apElem[1] = Tcl_NewIntObj(pLog->nByte); 854 for(ii=0; ii<MALLOC_LOG_FRAMES; ii++){ 855 apElem[ii+2] = Tcl_NewWideIntObj(aKey[ii]); 856 } 857 858 Tcl_ListObjAppendElement(interp, pRet, 859 Tcl_NewListObj(MALLOC_LOG_FRAMES+2, apElem) 860 ); 861 } 862 863 Tcl_SetObjResult(interp, pRet); 864 break; 865 } 866 case MB_LOG_CLEAR: { 867 test_memdebug_log_clear(); 868 break; 869 } 870 871 case MB_LOG_SYNC: { 872 #ifdef SQLITE_MEMDEBUG 873 extern void sqlite3MemdebugSync(); 874 test_memdebug_log_clear(); 875 mallocLogEnabled = 1; 876 sqlite3MemdebugSync(); 877 #endif 878 break; 879 } 880 } 881 882 return TCL_OK; 883 } 884 885 /* 886 ** Usage: sqlite3_config_scratch SIZE N 887 ** 888 ** Set the scratch memory buffer using SQLITE_CONFIG_SCRATCH. 889 ** The buffer is static and is of limited size. N might be 890 ** adjusted downward as needed to accommodate the requested size. 891 ** The revised value of N is returned. 892 ** 893 ** A negative SIZE causes the buffer pointer to be NULL. 894 */ 895 static int test_config_scratch( 896 void * clientData, 897 Tcl_Interp *interp, 898 int objc, 899 Tcl_Obj *CONST objv[] 900 ){ 901 int sz, N, rc; 902 Tcl_Obj *pResult; 903 static char *buf = 0; 904 if( objc!=3 ){ 905 Tcl_WrongNumArgs(interp, 1, objv, "SIZE N"); 906 return TCL_ERROR; 907 } 908 if( Tcl_GetIntFromObj(interp, objv[1], &sz) ) return TCL_ERROR; 909 if( Tcl_GetIntFromObj(interp, objv[2], &N) ) return TCL_ERROR; 910 free(buf); 911 if( sz<0 ){ 912 buf = 0; 913 rc = sqlite3_config(SQLITE_CONFIG_SCRATCH, 0, 0, 0); 914 }else{ 915 buf = malloc( sz*N + 1 ); 916 rc = sqlite3_config(SQLITE_CONFIG_SCRATCH, buf, sz, N); 917 } 918 pResult = Tcl_NewObj(); 919 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(rc)); 920 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(N)); 921 Tcl_SetObjResult(interp, pResult); 922 return TCL_OK; 923 } 924 925 /* 926 ** Usage: sqlite3_config_pagecache SIZE N 927 ** 928 ** Set the page-cache memory buffer using SQLITE_CONFIG_PAGECACHE. 929 ** The buffer is static and is of limited size. N might be 930 ** adjusted downward as needed to accommodate the requested size. 931 ** The revised value of N is returned. 932 ** 933 ** A negative SIZE causes the buffer pointer to be NULL. 934 */ 935 static int test_config_pagecache( 936 void * clientData, 937 Tcl_Interp *interp, 938 int objc, 939 Tcl_Obj *CONST objv[] 940 ){ 941 int sz, N, rc; 942 Tcl_Obj *pResult; 943 static char *buf = 0; 944 if( objc!=3 ){ 945 Tcl_WrongNumArgs(interp, 1, objv, "SIZE N"); 946 return TCL_ERROR; 947 } 948 if( Tcl_GetIntFromObj(interp, objv[1], &sz) ) return TCL_ERROR; 949 if( Tcl_GetIntFromObj(interp, objv[2], &N) ) return TCL_ERROR; 950 free(buf); 951 if( sz<0 ){ 952 buf = 0; 953 rc = sqlite3_config(SQLITE_CONFIG_PAGECACHE, 0, 0, 0); 954 }else{ 955 buf = malloc( sz*N ); 956 rc = sqlite3_config(SQLITE_CONFIG_PAGECACHE, buf, sz, N); 957 } 958 pResult = Tcl_NewObj(); 959 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(rc)); 960 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(N)); 961 Tcl_SetObjResult(interp, pResult); 962 return TCL_OK; 963 } 964 965 /* 966 ** Usage: sqlite3_config_alt_pcache INSTALL_FLAG DISCARD_CHANCE PRNG_SEED 967 ** 968 ** Set up the alternative test page cache. Install if INSTALL_FLAG is 969 ** true and uninstall (reverting to the default page cache) if INSTALL_FLAG 970 ** is false. DISCARD_CHANGE is an integer between 0 and 100 inclusive 971 ** which determines the chance of discarding a page when unpinned. 100 972 ** is certainty. 0 is never. PRNG_SEED is the pseudo-random number generator 973 ** seed. 974 */ 975 static int test_alt_pcache( 976 void * clientData, 977 Tcl_Interp *interp, 978 int objc, 979 Tcl_Obj *CONST objv[] 980 ){ 981 int installFlag; 982 int discardChance = 0; 983 int prngSeed = 0; 984 int highStress = 0; 985 extern void installTestPCache(int,unsigned,unsigned,unsigned); 986 if( objc<2 || objc>5 ){ 987 Tcl_WrongNumArgs(interp, 1, objv, 988 "INSTALLFLAG DISCARDCHANCE PRNGSEEED HIGHSTRESS"); 989 return TCL_ERROR; 990 } 991 if( Tcl_GetIntFromObj(interp, objv[1], &installFlag) ) return TCL_ERROR; 992 if( objc>=3 && Tcl_GetIntFromObj(interp, objv[2], &discardChance) ){ 993 return TCL_ERROR; 994 } 995 if( objc>=4 && Tcl_GetIntFromObj(interp, objv[3], &prngSeed) ){ 996 return TCL_ERROR; 997 } 998 if( objc>=5 && Tcl_GetIntFromObj(interp, objv[4], &highStress) ){ 999 return TCL_ERROR; 1000 } 1001 if( discardChance<0 || discardChance>100 ){ 1002 Tcl_AppendResult(interp, "discard-chance should be between 0 and 100", 1003 (char*)0); 1004 return TCL_ERROR; 1005 } 1006 installTestPCache(installFlag, (unsigned)discardChance, (unsigned)prngSeed, 1007 (unsigned)highStress); 1008 return TCL_OK; 1009 } 1010 1011 /* 1012 ** Usage: sqlite3_config_memstatus BOOLEAN 1013 ** 1014 ** Enable or disable memory status reporting using SQLITE_CONFIG_MEMSTATUS. 1015 */ 1016 static int test_config_memstatus( 1017 void * clientData, 1018 Tcl_Interp *interp, 1019 int objc, 1020 Tcl_Obj *CONST objv[] 1021 ){ 1022 int enable, rc; 1023 if( objc!=2 ){ 1024 Tcl_WrongNumArgs(interp, 1, objv, "BOOLEAN"); 1025 return TCL_ERROR; 1026 } 1027 if( Tcl_GetBooleanFromObj(interp, objv[1], &enable) ) return TCL_ERROR; 1028 rc = sqlite3_config(SQLITE_CONFIG_MEMSTATUS, enable); 1029 Tcl_SetObjResult(interp, Tcl_NewIntObj(rc)); 1030 return TCL_OK; 1031 } 1032 1033 /* 1034 ** Usage: sqlite3_config_lookaside SIZE COUNT 1035 ** 1036 */ 1037 static int test_config_lookaside( 1038 void * clientData, 1039 Tcl_Interp *interp, 1040 int objc, 1041 Tcl_Obj *CONST objv[] 1042 ){ 1043 int sz, cnt; 1044 Tcl_Obj *pRet; 1045 if( objc!=3 ){ 1046 Tcl_WrongNumArgs(interp, 1, objv, "SIZE COUNT"); 1047 return TCL_ERROR; 1048 } 1049 if( Tcl_GetIntFromObj(interp, objv[1], &sz) ) return TCL_ERROR; 1050 if( Tcl_GetIntFromObj(interp, objv[2], &cnt) ) return TCL_ERROR; 1051 pRet = Tcl_NewObj(); 1052 Tcl_ListObjAppendElement( 1053 interp, pRet, Tcl_NewIntObj(sqlite3GlobalConfig.szLookaside) 1054 ); 1055 Tcl_ListObjAppendElement( 1056 interp, pRet, Tcl_NewIntObj(sqlite3GlobalConfig.nLookaside) 1057 ); 1058 sqlite3_config(SQLITE_CONFIG_LOOKASIDE, sz, cnt); 1059 Tcl_SetObjResult(interp, pRet); 1060 return TCL_OK; 1061 } 1062 1063 1064 /* 1065 ** Usage: sqlite3_db_config_lookaside CONNECTION BUFID SIZE COUNT 1066 ** 1067 ** There are two static buffers with BUFID 1 and 2. Each static buffer 1068 ** is 10KB in size. A BUFID of 0 indicates that the buffer should be NULL 1069 ** which will cause sqlite3_db_config() to allocate space on its own. 1070 */ 1071 static int test_db_config_lookaside( 1072 void * clientData, 1073 Tcl_Interp *interp, 1074 int objc, 1075 Tcl_Obj *CONST objv[] 1076 ){ 1077 int rc; 1078 int sz, cnt; 1079 sqlite3 *db; 1080 int bufid; 1081 static char azBuf[2][10000]; 1082 extern int getDbPointer(Tcl_Interp*, const char*, sqlite3**); 1083 if( objc!=5 ){ 1084 Tcl_WrongNumArgs(interp, 1, objv, "BUFID SIZE COUNT"); 1085 return TCL_ERROR; 1086 } 1087 if( getDbPointer(interp, Tcl_GetString(objv[1]), &db) ) return TCL_ERROR; 1088 if( Tcl_GetIntFromObj(interp, objv[2], &bufid) ) return TCL_ERROR; 1089 if( Tcl_GetIntFromObj(interp, objv[3], &sz) ) return TCL_ERROR; 1090 if( Tcl_GetIntFromObj(interp, objv[4], &cnt) ) return TCL_ERROR; 1091 if( bufid==0 ){ 1092 rc = sqlite3_db_config(db, SQLITE_DBCONFIG_LOOKASIDE, 0, sz, cnt); 1093 }else if( bufid>=1 && bufid<=2 && sz*cnt<=sizeof(azBuf[0]) ){ 1094 rc = sqlite3_db_config(db, SQLITE_DBCONFIG_LOOKASIDE, azBuf[bufid], sz,cnt); 1095 }else{ 1096 Tcl_AppendResult(interp, "illegal arguments - see documentation", (char*)0); 1097 return TCL_ERROR; 1098 } 1099 Tcl_SetObjResult(interp, Tcl_NewIntObj(rc)); 1100 return TCL_OK; 1101 } 1102 1103 /* 1104 ** Usage: sqlite3_config_heap NBYTE NMINALLOC 1105 */ 1106 static int test_config_heap( 1107 void * clientData, 1108 Tcl_Interp *interp, 1109 int objc, 1110 Tcl_Obj *CONST objv[] 1111 ){ 1112 static char *zBuf; /* Use this memory */ 1113 int nByte; /* Size of buffer to pass to sqlite3_config() */ 1114 int nMinAlloc; /* Size of minimum allocation */ 1115 int rc; /* Return code of sqlite3_config() */ 1116 1117 Tcl_Obj * CONST *aArg = &objv[1]; 1118 int nArg = objc-1; 1119 1120 if( nArg!=2 ){ 1121 Tcl_WrongNumArgs(interp, 1, objv, "NBYTE NMINALLOC"); 1122 return TCL_ERROR; 1123 } 1124 if( Tcl_GetIntFromObj(interp, aArg[0], &nByte) ) return TCL_ERROR; 1125 if( Tcl_GetIntFromObj(interp, aArg[1], &nMinAlloc) ) return TCL_ERROR; 1126 1127 if( nByte==0 ){ 1128 free( zBuf ); 1129 zBuf = 0; 1130 rc = sqlite3_config(SQLITE_CONFIG_HEAP, (void*)0, 0, 0); 1131 }else{ 1132 zBuf = realloc(zBuf, nByte); 1133 rc = sqlite3_config(SQLITE_CONFIG_HEAP, zBuf, nByte, nMinAlloc); 1134 } 1135 1136 Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_VOLATILE); 1137 return TCL_OK; 1138 } 1139 1140 /* 1141 ** Usage: sqlite3_config_heap_size NBYTE 1142 */ 1143 static int test_config_heap_size( 1144 void * clientData, 1145 Tcl_Interp *interp, 1146 int objc, 1147 Tcl_Obj *CONST objv[] 1148 ){ 1149 int nByte; /* Size to pass to sqlite3_config() */ 1150 int rc; /* Return code of sqlite3_config() */ 1151 1152 Tcl_Obj * CONST *aArg = &objv[1]; 1153 int nArg = objc-1; 1154 1155 if( nArg!=1 ){ 1156 Tcl_WrongNumArgs(interp, 1, objv, "NBYTE"); 1157 return TCL_ERROR; 1158 } 1159 if( Tcl_GetIntFromObj(interp, aArg[0], &nByte) ) return TCL_ERROR; 1160 1161 rc = sqlite3_config(SQLITE_CONFIG_WIN32_HEAPSIZE, nByte); 1162 1163 Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_VOLATILE); 1164 return TCL_OK; 1165 } 1166 1167 /* 1168 ** Usage: sqlite3_config_error [DB] 1169 ** 1170 ** Invoke sqlite3_config() or sqlite3_db_config() with invalid 1171 ** opcodes and verify that they return errors. 1172 */ 1173 static int test_config_error( 1174 void * clientData, 1175 Tcl_Interp *interp, 1176 int objc, 1177 Tcl_Obj *CONST objv[] 1178 ){ 1179 sqlite3 *db; 1180 extern int getDbPointer(Tcl_Interp*, const char*, sqlite3**); 1181 1182 if( objc!=2 && objc!=1 ){ 1183 Tcl_WrongNumArgs(interp, 1, objv, "[DB]"); 1184 return TCL_ERROR; 1185 } 1186 if( objc==2 ){ 1187 if( getDbPointer(interp, Tcl_GetString(objv[1]), &db) ) return TCL_ERROR; 1188 if( sqlite3_db_config(db, 99999)!=SQLITE_ERROR ){ 1189 Tcl_AppendResult(interp, 1190 "sqlite3_db_config(db, 99999) does not return SQLITE_ERROR", 1191 (char*)0); 1192 return TCL_ERROR; 1193 } 1194 }else{ 1195 if( sqlite3_config(99999)!=SQLITE_ERROR ){ 1196 Tcl_AppendResult(interp, 1197 "sqlite3_config(99999) does not return SQLITE_ERROR", 1198 (char*)0); 1199 return TCL_ERROR; 1200 } 1201 } 1202 return TCL_OK; 1203 } 1204 1205 /* 1206 ** Usage: sqlite3_config_uri BOOLEAN 1207 ** 1208 ** Enables or disables interpretation of URI parameters by default using 1209 ** SQLITE_CONFIG_URI. 1210 */ 1211 static int test_config_uri( 1212 void * clientData, 1213 Tcl_Interp *interp, 1214 int objc, 1215 Tcl_Obj *CONST objv[] 1216 ){ 1217 int rc; 1218 int bOpenUri; 1219 1220 if( objc!=2 ){ 1221 Tcl_WrongNumArgs(interp, 1, objv, "BOOL"); 1222 return TCL_ERROR; 1223 } 1224 if( Tcl_GetBooleanFromObj(interp, objv[1], &bOpenUri) ){ 1225 return TCL_ERROR; 1226 } 1227 1228 rc = sqlite3_config(SQLITE_CONFIG_URI, bOpenUri); 1229 Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_VOLATILE); 1230 1231 return TCL_OK; 1232 } 1233 1234 /* 1235 ** Usage: sqlite3_config_cis BOOLEAN 1236 ** 1237 ** Enables or disables the use of the covering-index scan optimization. 1238 ** SQLITE_CONFIG_COVERING_INDEX_SCAN. 1239 */ 1240 static int test_config_cis( 1241 void * clientData, 1242 Tcl_Interp *interp, 1243 int objc, 1244 Tcl_Obj *CONST objv[] 1245 ){ 1246 int rc; 1247 int bUseCis; 1248 1249 if( objc!=2 ){ 1250 Tcl_WrongNumArgs(interp, 1, objv, "BOOL"); 1251 return TCL_ERROR; 1252 } 1253 if( Tcl_GetBooleanFromObj(interp, objv[1], &bUseCis) ){ 1254 return TCL_ERROR; 1255 } 1256 1257 rc = sqlite3_config(SQLITE_CONFIG_COVERING_INDEX_SCAN, bUseCis); 1258 Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_VOLATILE); 1259 1260 return TCL_OK; 1261 } 1262 1263 1264 /* 1265 ** Usage: sqlite3_dump_memsys3 FILENAME 1266 ** sqlite3_dump_memsys5 FILENAME 1267 ** 1268 ** Write a summary of unfreed memsys3 allocations to FILENAME. 1269 */ 1270 static int test_dump_memsys3( 1271 void * clientData, 1272 Tcl_Interp *interp, 1273 int objc, 1274 Tcl_Obj *CONST objv[] 1275 ){ 1276 if( objc!=2 ){ 1277 Tcl_WrongNumArgs(interp, 1, objv, "FILENAME"); 1278 return TCL_ERROR; 1279 } 1280 1281 switch( SQLITE_PTR_TO_INT(clientData) ){ 1282 case 3: { 1283 #ifdef SQLITE_ENABLE_MEMSYS3 1284 extern void sqlite3Memsys3Dump(const char*); 1285 sqlite3Memsys3Dump(Tcl_GetString(objv[1])); 1286 break; 1287 #endif 1288 } 1289 case 5: { 1290 #ifdef SQLITE_ENABLE_MEMSYS5 1291 extern void sqlite3Memsys5Dump(const char*); 1292 sqlite3Memsys5Dump(Tcl_GetString(objv[1])); 1293 break; 1294 #endif 1295 } 1296 } 1297 return TCL_OK; 1298 } 1299 1300 /* 1301 ** Usage: sqlite3_status OPCODE RESETFLAG 1302 ** 1303 ** Return a list of three elements which are the sqlite3_status() return 1304 ** code, the current value, and the high-water mark value. 1305 */ 1306 static int test_status( 1307 void * clientData, 1308 Tcl_Interp *interp, 1309 int objc, 1310 Tcl_Obj *CONST objv[] 1311 ){ 1312 int rc, iValue, mxValue; 1313 int i, op, resetFlag; 1314 const char *zOpName; 1315 static const struct { 1316 const char *zName; 1317 int op; 1318 } aOp[] = { 1319 { "SQLITE_STATUS_MEMORY_USED", SQLITE_STATUS_MEMORY_USED }, 1320 { "SQLITE_STATUS_MALLOC_SIZE", SQLITE_STATUS_MALLOC_SIZE }, 1321 { "SQLITE_STATUS_PAGECACHE_USED", SQLITE_STATUS_PAGECACHE_USED }, 1322 { "SQLITE_STATUS_PAGECACHE_OVERFLOW", SQLITE_STATUS_PAGECACHE_OVERFLOW }, 1323 { "SQLITE_STATUS_PAGECACHE_SIZE", SQLITE_STATUS_PAGECACHE_SIZE }, 1324 { "SQLITE_STATUS_SCRATCH_USED", SQLITE_STATUS_SCRATCH_USED }, 1325 { "SQLITE_STATUS_SCRATCH_OVERFLOW", SQLITE_STATUS_SCRATCH_OVERFLOW }, 1326 { "SQLITE_STATUS_SCRATCH_SIZE", SQLITE_STATUS_SCRATCH_SIZE }, 1327 { "SQLITE_STATUS_PARSER_STACK", SQLITE_STATUS_PARSER_STACK }, 1328 { "SQLITE_STATUS_MALLOC_COUNT", SQLITE_STATUS_MALLOC_COUNT }, 1329 }; 1330 Tcl_Obj *pResult; 1331 if( objc!=3 ){ 1332 Tcl_WrongNumArgs(interp, 1, objv, "PARAMETER RESETFLAG"); 1333 return TCL_ERROR; 1334 } 1335 zOpName = Tcl_GetString(objv[1]); 1336 for(i=0; i<ArraySize(aOp); i++){ 1337 if( strcmp(aOp[i].zName, zOpName)==0 ){ 1338 op = aOp[i].op; 1339 break; 1340 } 1341 } 1342 if( i>=ArraySize(aOp) ){ 1343 if( Tcl_GetIntFromObj(interp, objv[1], &op) ) return TCL_ERROR; 1344 } 1345 if( Tcl_GetBooleanFromObj(interp, objv[2], &resetFlag) ) return TCL_ERROR; 1346 iValue = 0; 1347 mxValue = 0; 1348 rc = sqlite3_status(op, &iValue, &mxValue, resetFlag); 1349 pResult = Tcl_NewObj(); 1350 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(rc)); 1351 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(iValue)); 1352 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(mxValue)); 1353 Tcl_SetObjResult(interp, pResult); 1354 return TCL_OK; 1355 } 1356 1357 /* 1358 ** Usage: sqlite3_db_status DATABASE OPCODE RESETFLAG 1359 ** 1360 ** Return a list of three elements which are the sqlite3_db_status() return 1361 ** code, the current value, and the high-water mark value. 1362 */ 1363 static int test_db_status( 1364 void * clientData, 1365 Tcl_Interp *interp, 1366 int objc, 1367 Tcl_Obj *CONST objv[] 1368 ){ 1369 int rc, iValue, mxValue; 1370 int i, op, resetFlag; 1371 const char *zOpName; 1372 sqlite3 *db; 1373 extern int getDbPointer(Tcl_Interp*, const char*, sqlite3**); 1374 static const struct { 1375 const char *zName; 1376 int op; 1377 } aOp[] = { 1378 { "LOOKASIDE_USED", SQLITE_DBSTATUS_LOOKASIDE_USED }, 1379 { "CACHE_USED", SQLITE_DBSTATUS_CACHE_USED }, 1380 { "SCHEMA_USED", SQLITE_DBSTATUS_SCHEMA_USED }, 1381 { "STMT_USED", SQLITE_DBSTATUS_STMT_USED }, 1382 { "LOOKASIDE_HIT", SQLITE_DBSTATUS_LOOKASIDE_HIT }, 1383 { "LOOKASIDE_MISS_SIZE", SQLITE_DBSTATUS_LOOKASIDE_MISS_SIZE }, 1384 { "LOOKASIDE_MISS_FULL", SQLITE_DBSTATUS_LOOKASIDE_MISS_FULL }, 1385 { "CACHE_HIT", SQLITE_DBSTATUS_CACHE_HIT }, 1386 { "CACHE_MISS", SQLITE_DBSTATUS_CACHE_MISS }, 1387 { "CACHE_WRITE", SQLITE_DBSTATUS_CACHE_WRITE }, 1388 { "DEFERRED_FKS", SQLITE_DBSTATUS_DEFERRED_FKS } 1389 }; 1390 Tcl_Obj *pResult; 1391 if( objc!=4 ){ 1392 Tcl_WrongNumArgs(interp, 1, objv, "DB PARAMETER RESETFLAG"); 1393 return TCL_ERROR; 1394 } 1395 if( getDbPointer(interp, Tcl_GetString(objv[1]), &db) ) return TCL_ERROR; 1396 zOpName = Tcl_GetString(objv[2]); 1397 if( memcmp(zOpName, "SQLITE_", 7)==0 ) zOpName += 7; 1398 if( memcmp(zOpName, "DBSTATUS_", 9)==0 ) zOpName += 9; 1399 for(i=0; i<ArraySize(aOp); i++){ 1400 if( strcmp(aOp[i].zName, zOpName)==0 ){ 1401 op = aOp[i].op; 1402 break; 1403 } 1404 } 1405 if( i>=ArraySize(aOp) ){ 1406 if( Tcl_GetIntFromObj(interp, objv[2], &op) ) return TCL_ERROR; 1407 } 1408 if( Tcl_GetBooleanFromObj(interp, objv[3], &resetFlag) ) return TCL_ERROR; 1409 iValue = 0; 1410 mxValue = 0; 1411 rc = sqlite3_db_status(db, op, &iValue, &mxValue, resetFlag); 1412 pResult = Tcl_NewObj(); 1413 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(rc)); 1414 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(iValue)); 1415 Tcl_ListObjAppendElement(0, pResult, Tcl_NewIntObj(mxValue)); 1416 Tcl_SetObjResult(interp, pResult); 1417 return TCL_OK; 1418 } 1419 1420 /* 1421 ** install_malloc_faultsim BOOLEAN 1422 */ 1423 static int test_install_malloc_faultsim( 1424 void * clientData, 1425 Tcl_Interp *interp, 1426 int objc, 1427 Tcl_Obj *CONST objv[] 1428 ){ 1429 int rc; 1430 int isInstall; 1431 1432 if( objc!=2 ){ 1433 Tcl_WrongNumArgs(interp, 1, objv, "BOOLEAN"); 1434 return TCL_ERROR; 1435 } 1436 if( TCL_OK!=Tcl_GetBooleanFromObj(interp, objv[1], &isInstall) ){ 1437 return TCL_ERROR; 1438 } 1439 rc = faultsimInstall(isInstall); 1440 Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_VOLATILE); 1441 return TCL_OK; 1442 } 1443 1444 /* 1445 ** sqlite3_install_memsys3 1446 */ 1447 static int test_install_memsys3( 1448 void * clientData, 1449 Tcl_Interp *interp, 1450 int objc, 1451 Tcl_Obj *CONST objv[] 1452 ){ 1453 int rc = SQLITE_MISUSE; 1454 #ifdef SQLITE_ENABLE_MEMSYS3 1455 const sqlite3_mem_methods *sqlite3MemGetMemsys3(void); 1456 rc = sqlite3_config(SQLITE_CONFIG_MALLOC, sqlite3MemGetMemsys3()); 1457 #endif 1458 Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_VOLATILE); 1459 return TCL_OK; 1460 } 1461 1462 static int test_vfs_oom_test( 1463 void * clientData, 1464 Tcl_Interp *interp, 1465 int objc, 1466 Tcl_Obj *CONST objv[] 1467 ){ 1468 extern int sqlite3_memdebug_vfs_oom_test; 1469 if( objc>2 ){ 1470 Tcl_WrongNumArgs(interp, 1, objv, "?INTEGER?"); 1471 return TCL_ERROR; 1472 }else if( objc==2 ){ 1473 int iNew; 1474 if( Tcl_GetIntFromObj(interp, objv[1], &iNew) ) return TCL_ERROR; 1475 sqlite3_memdebug_vfs_oom_test = iNew; 1476 } 1477 Tcl_SetObjResult(interp, Tcl_NewIntObj(sqlite3_memdebug_vfs_oom_test)); 1478 return TCL_OK; 1479 } 1480 1481 /* 1482 ** Register commands with the TCL interpreter. 1483 */ 1484 int Sqlitetest_malloc_Init(Tcl_Interp *interp){ 1485 static struct { 1486 char *zName; 1487 Tcl_ObjCmdProc *xProc; 1488 int clientData; 1489 } aObjCmd[] = { 1490 { "sqlite3_malloc", test_malloc ,0 }, 1491 { "sqlite3_realloc", test_realloc ,0 }, 1492 { "sqlite3_free", test_free ,0 }, 1493 { "memset", test_memset ,0 }, 1494 { "memget", test_memget ,0 }, 1495 { "sqlite3_memory_used", test_memory_used ,0 }, 1496 { "sqlite3_memory_highwater", test_memory_highwater ,0 }, 1497 { "sqlite3_memdebug_backtrace", test_memdebug_backtrace ,0 }, 1498 { "sqlite3_memdebug_dump", test_memdebug_dump ,0 }, 1499 { "sqlite3_memdebug_fail", test_memdebug_fail ,0 }, 1500 { "sqlite3_memdebug_pending", test_memdebug_pending ,0 }, 1501 { "sqlite3_memdebug_settitle", test_memdebug_settitle ,0 }, 1502 { "sqlite3_memdebug_malloc_count", test_memdebug_malloc_count ,0 }, 1503 { "sqlite3_memdebug_log", test_memdebug_log ,0 }, 1504 { "sqlite3_config_scratch", test_config_scratch ,0 }, 1505 { "sqlite3_config_pagecache", test_config_pagecache ,0 }, 1506 { "sqlite3_config_alt_pcache", test_alt_pcache ,0 }, 1507 { "sqlite3_status", test_status ,0 }, 1508 { "sqlite3_db_status", test_db_status ,0 }, 1509 { "install_malloc_faultsim", test_install_malloc_faultsim ,0 }, 1510 { "sqlite3_config_heap", test_config_heap ,0 }, 1511 { "sqlite3_config_heap_size", test_config_heap_size ,0 }, 1512 { "sqlite3_config_memstatus", test_config_memstatus ,0 }, 1513 { "sqlite3_config_lookaside", test_config_lookaside ,0 }, 1514 { "sqlite3_config_error", test_config_error ,0 }, 1515 { "sqlite3_config_uri", test_config_uri ,0 }, 1516 { "sqlite3_config_cis", test_config_cis ,0 }, 1517 { "sqlite3_db_config_lookaside",test_db_config_lookaside ,0 }, 1518 { "sqlite3_dump_memsys3", test_dump_memsys3 ,3 }, 1519 { "sqlite3_dump_memsys5", test_dump_memsys3 ,5 }, 1520 { "sqlite3_install_memsys3", test_install_memsys3 ,0 }, 1521 { "sqlite3_memdebug_vfs_oom_test", test_vfs_oom_test ,0 }, 1522 }; 1523 int i; 1524 for(i=0; i<sizeof(aObjCmd)/sizeof(aObjCmd[0]); i++){ 1525 ClientData c = (ClientData)SQLITE_INT_TO_PTR(aObjCmd[i].clientData); 1526 Tcl_CreateObjCommand(interp, aObjCmd[i].zName, aObjCmd[i].xProc, c, 0); 1527 } 1528 return TCL_OK; 1529 } 1530 #endif 1531