SQLite4  Check-in [9638da92b6]

  sqlite4.h


# Source code to the test files.
#
TESTSRC = \
  $(TOP)/test/test_auth.c \
  $(TOP)/test/test_bt.c \
  $(TOP)/test/test_config.c \
  $(TOP)/test/test_func.c \
  $(TOP)/test/test_hexio.c \
  $(TOP)/test/test_kv.c \
  $(TOP)/test/test_kv2.c \
  $(TOP)/test/test_lsm.c \
  $(TOP)/test/test_main.c \

      case BT_LOCK_SHARED:
        if( nExcl ){
          rc = SQLITE4_BUSY;
        }else{
          if( nShared==0 ){
            rc = btLockSharedFile(p, iLock, BT_LOCK_SHARED);
          }
          /* If no error occurred, set the bit in the mask of SHARED locks
          ** held. Either way, clear the bit in the mask of EXCLUSIVE locks.
          ** The idea here is that when there is any uncertainty as to whether
          ** a lock is held, the corresponding bit is cleared.  */
          if( rc==SQLITE4_OK ) p->mSharedLock |= mask;

          p->mExclLock &= ~mask;

        }
        break;

      default:
        assert( eOp==BT_LOCK_EXCL );
        if( nExcl || nShared ){
          rc = SQLITE4_BUSY;

        p->pFd = p->pBtFile->pFd;
      }else{
        p->pBtFile = (BtFile*)sqlite4_malloc(pEnv, sizeof(BtFile));
        if( p->pBtFile ){
          int flags = BT_OPEN_DATABASE;
          p->pBtFile->pNext = 0;
          rc = pVfs->xOpen(pEnv, pVfs, pShared->zName, flags, &p->pFd);
          if( rc==SQLITE4_OK ){
            p->pBtFile->pFd = p->pFd;
          }else{
            sqlite4_free(pEnv, p->pBtFile);
            p->pBtFile = 0;
          }
        }else{
          rc = btErrorBkpt(SQLITE4_NOMEM);
        }
      }
    }

    if( rc==SQLITE4_OK ){

  sqlite4_mutex_enter(pShared->pClientMutex);
  assert( p->pBtFile->pNext==0 );
  p->pBtFile->pNext = pShared->pBtFile;
  pShared->pBtFile = p->pBtFile;
  sqlite4_mutex_leave(pShared->pClientMutex);

  btLockSharedDeref(p->pEnv, p->pVfs, pShared);
  p->pShared = 0;
  return rc;
}

#ifndef NDEBUG
static void assertNoLockedSlots(BtLock *pLock){
  u32 mask = (1 << (BT_LOCK_READER0+BT_NREADER+1)) - (1 << BT_LOCK_READER0);
  assert( (pLock->mExclLock & mask)==0 );

    p->nFile = strlen(p->zFile);
    rc = sqlite4BtLockConnect((BtLock*)p, btRecover);
    if( rc==SQLITE4_OK && p->pLog==0 ){
      rc = sqlite4BtLogOpen(p, 0, &p->pLog);
    }
  }

  if( rc!=SQLITE4_OK ){
    sqlite4BtLockDisconnect((BtLock*)p, btCheckpoint, btCleanup);
    sqlite4BtLogClose(p->pLog, 0);
    p->pLog = 0;
  }

  return rc;
}

/*
** Open a read-transaction.
*/
static int btOpenReadTransaction(BtPager *p){

#

set testdir [file dirname $argv0]
source $testdir/tester.tcl
source $testdir/malloc_common.tcl
set ::testprefix fault1

db close

proc open_test_db {} { 
  sqlite4 db test.db
  btenv testenv
  testenv attach db
}

proc close_test_db {} { 
  db close
  testenv delete
}

do_faultsim_test 2.0 -prep {
  forcedelete test.db test.db-wal
  open_test_db
} -body {
  execsql {
    CREATE TABLE t1(x,y);
    INSERT INTO t1 VALUES('abc', 'def');
    SELECT * FROM t1;
  }
} -test {
  faultsim_test_result {0 {abc def}}
  close_test_db
}

finish_test

]
  
# Transient and persistent IO errors:
#
set FAULTSIM(ioerr-transient) [list        \
  -injectstart   {ioerr_injectstart 0}     \
  -injectstop    ioerr_injectstop          \
  -injecterrlist {{1 {disk I/O error}} {1 {unable to open database file}}} \
]
set FAULTSIM(ioerr-persistent) [list       \
  -injectstart   {ioerr_injectstart 1}     \
  -injectstop    ioerr_injectstop          \
  -injecterrlist {{1 {disk I/O error}} {1 {unable to open database file}}} \












































]



#--------------------------------------------------------------------------
# Usage do_faultsim_test NAME ?OPTIONS...? 
#

  return $nFault
}

# The following procs are used as [do_one_faultsim_test] callbacks when 
# injecting IO error faults into test cases.
#
proc ioerr_injectstart {persist iFail} {
  testenv ioerr $iFail $persist

}
proc ioerr_injectstop {} {







  testenv ioerr 0 0
}

# The following procs are used as [do_one_faultsim_test] callbacks when 
# injecting shared-memory related error faults into test cases.
#
proc shmerr_injectinstall {} {
  testvfs shmfault -default true

# focus of this file is testing the SELECT statement.
#
# $Id: select1.test,v 1.70 2009/05/28 01:00:56 drh Exp $

set testdir [file dirname $argv0]
source $testdir/tester.tcl

btenv BT
BT attach db

# Try to select on a non-existant table.
#
do_test select1-1.1 {
  set v [catch {execsql {SELECT * FROM test1}} msg]
  lappend v $msg
} {1 {no such table: test1}}

sqlite4_mm *test_mm_debug(sqlite4_mm *p);
sqlite4_mm *test_mm_faultsim(sqlite4_mm *p);

/* test_num.c */
int Sqlitetest_num_init(Tcl_Interp *interp);

/* test_bt.c */
int SqlitetestBt_Init(Tcl_Interp *interp);

#endif

/*
** 2014 February 17
**
** The author disclaims copyright to this source code.  In place of
** a legal notice, here is a blessing:
**
**    May you do good and not evil.
**    May you find forgiveness for yourself and forgive others.
**    May you share freely, never taking more than you give.
**
*************************************************************************
*/

#include <tcl.h>
#include "bt.h"
#include "sqlite4.h"
#include "testInt.h"
#include <assert.h>
#include <string.h>

typedef sqlite4_int64 i64;


typedef struct BtTestEnv BtTestEnv;
struct BtTestEnv {
  bt_env base;                    /* The wrapper environment */
  sqlite4_env *pSqlEnv;           /* SQLite environment */
  bt_env *pEnv;                   /* Underlying environment object */

  /* IO error injection parameters */
  int nIoerrCnt;                  /* Error countdown */
  int bIoerrPersist;              /* True for persistent errors */
  int nIoerrInjected;             /* Number of errors injected */
};

typedef struct BtTestFile BtTestFile;
struct BtTestFile {
  BtTestEnv *pTestEnv;            /* Test environment */
  bt_file *pFile;                 /* Underlying file object */
};

/*
** Return true if an IO error should be injected. False otherwise.
*/
static int testInjectIoerr(BtTestEnv *p){
  if( (p->nIoerrCnt==0 && p->bIoerrPersist)
   || (p->nIoerrCnt>0 && (--p->nIoerrCnt)==0)
  ){
    p->nIoerrInjected++;
    return 1;
  }
  return 0;
}

/*
** The bt_env.xOpen method.
*/
static int testBtOsOpen(
  sqlite4_env *pSqlEnv,
  bt_env *pEnv,
  const char *zFile,
  int flags,
  bt_file **ppFile
){
  BtTestEnv *p = (BtTestEnv*)pEnv;
  BtTestFile *pNew = 0;
  int rc;

  if( testInjectIoerr(p) ){
    rc = SQLITE4_CANTOPEN;
  }else{
    pNew = ckalloc(sizeof(BtTestFile));
    memset(pNew, 0, sizeof(BtTestFile));
    pNew->pTestEnv = p;
    rc = p->pEnv->xOpen(pSqlEnv, p->pEnv, zFile, flags, &pNew->pFile);
    if( rc!=SQLITE4_OK ){
      ckfree(pNew);
      pNew = 0;
    }
  }
  *ppFile = (bt_file*)pNew;
  return rc;
}

static int testBtOsSize(bt_file *pFile, i64 *pnByte){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  if( testInjectIoerr(pTestFile->pTestEnv) ){
    rc = SQLITE4_IOERR_FSTAT;
  }else{
    rc = pEnv->xSize(pTestFile->pFile, pnByte);
  }
  return rc;
}

static int testBtOsWrite(
  bt_file *pFile,                 /* File to write to */
  i64 iOff,                       /* Offset to write to */
  void *pData,                    /* Write data from this buffer */
  int nData                       /* Bytes of data to write */
){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  if( testInjectIoerr(pTestFile->pTestEnv) ){
    rc = SQLITE4_IOERR_WRITE;
  }else{
    rc = pEnv->xWrite(pTestFile->pFile, iOff, pData, nData);
  }
  return rc;
}

static int testBtOsTruncate(
  bt_file *pFile,                 /* File to write to */
  i64 nSize                       /* Size to truncate file to */
){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  if( testInjectIoerr(pTestFile->pTestEnv) ){
    rc = SQLITE4_IOERR_TRUNCATE;
  }else{
    rc = pEnv->xTruncate(pTestFile->pFile, nSize);
  }
  return rc;
}

static int testBtOsRead(
  bt_file *pFile,                 /* File to read from */
  i64 iOff,                       /* Offset to read from */
  void *pData,                    /* Read data into this buffer */
  int nData                       /* Bytes of data to read */
){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  if( testInjectIoerr(pTestFile->pTestEnv) ){
    rc = SQLITE4_IOERR_READ;
  }else{
    rc = pEnv->xRead(pTestFile->pFile, iOff, pData, nData);
  }
  return rc;
}

static int testBtOsSync(bt_file *pFile){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  if( testInjectIoerr(pTestFile->pTestEnv) ){
    rc = SQLITE4_IOERR_FSYNC;
  }else{
    rc = pEnv->xSync(pTestFile->pFile);
  }
  return rc;
}

static int testBtOsSectorSize(bt_file *pFile){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int res;

  res = pEnv->xSectorSize(pTestFile->pFile);
  return res;
}

static int testBtOsFullpath(
  sqlite4_env *pSqlEnv,
  bt_env *pEnv,
  const char *zName,
  char **pzOut
){
  BtTestEnv *p = (BtTestEnv*)pEnv;
  int rc;

  if( testInjectIoerr(p) ){
    rc = SQLITE4_IOERR;
  }else{
    rc = p->pEnv->xFullpath(pSqlEnv, p->pEnv, zName, pzOut);
  }
  return rc;
}

static int testBtOsUnlink(
  sqlite4_env *pSqlEnv, 
  bt_env *pEnv, 
  const char *zFile
){
  BtTestEnv *p = (BtTestEnv*)pEnv;
  int rc;

  if( testInjectIoerr(p) ){
    rc = SQLITE4_IOERR_DIR_FSYNC;
  }else{
    rc = p->pEnv->xUnlink(pSqlEnv, p->pEnv, zFile);
  }
  return rc;
}

int testBtOsLock(bt_file *pFile, int iLock, int eType){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  if( testInjectIoerr(pTestFile->pTestEnv) ){
    rc = SQLITE4_IOERR_LOCK;
  }else{
    rc = pEnv->xLock(pTestFile->pFile, iLock, eType);
  }
  return rc;
}

int testBtOsTestLock(bt_file *pFile, int iLock, int nLock, int eType){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  if( testInjectIoerr(pTestFile->pTestEnv) ){
    rc = SQLITE4_IOERR;
  }else{
    rc = pEnv->xTestLock(pTestFile->pFile, iLock, nLock, eType);
  }
  return rc;
}

int testBtOsShmMap(bt_file *pFile, int iChunk, int sz, void **ppShm){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  if( testInjectIoerr(pTestFile->pTestEnv) ){
    rc = SQLITE4_IOERR_SHMMAP;
  }else{
    rc = pEnv->xShmMap(pTestFile->pFile, iChunk, sz, ppShm);
  }
  return rc;
}

void testBtOsShmBarrier(bt_file *pFile){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;

  pEnv->xShmBarrier(pTestFile->pFile);
}

int testBtOsShmUnmap(bt_file *pFile, int bDelete){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  rc = pEnv->xShmUnmap(pTestFile->pFile, bDelete);
  return rc;
}

static int testBtOsClose(bt_file *pFile){
  BtTestFile *pTestFile = (BtTestFile*)pFile;
  bt_env *pEnv = pTestFile->pTestEnv->pEnv;
  int rc;

  rc = pEnv->xClose(pTestFile->pFile);
  ckfree(pTestFile);
  return rc;
}

/*
** Destructor for object created by tcl [btenv] command.
*/
static void test_btenv_del(void *ctx){
  BtTestEnv *p = (BtTestEnv*)ctx;
  ckfree(p);
}

/*
** Tcl command: BTENV method ...
*/
static int test_btenv_cmd(
  void * clientData,
  Tcl_Interp *interp,
  int objc,
  Tcl_Obj *CONST objv[]
){
  BtTestEnv *p = (BtTestEnv*)clientData;

  enum BtenvCmdSymbol {
    BTC_ATTACH,
    BTC_DELETE,
    BTC_IOERR,
  };
  struct BtenvCmd {
    const char *zOpt;
    int eOpt;
    int nArg;
    const char *zErr;
  } aCmd[] = {
    { "attach", BTC_ATTACH, 3, "DBCMD" },
    { "delete", BTC_DELETE, 2, "" },
    { "ioerr",  BTC_IOERR,  4, "COUNT PERSISTENT" },
    { 0, 0 }
  };
  int rc = TCL_OK;
  int iOpt;

  if( objc<2 ){
    Tcl_WrongNumArgs(interp, 1, objv, "sub-command ...");
    return TCL_ERROR;
  }
  rc = Tcl_GetIndexFromObjStruct(
      interp, objv[1], aCmd, sizeof(aCmd[0]), "sub-command", 0, &iOpt
  );
  if( rc!=TCL_OK ) return rc;

  if( objc!=aCmd[iOpt].nArg ){
    Tcl_WrongNumArgs(interp, 2, objv, aCmd[iOpt].zErr);
    return TCL_ERROR;
  }

  switch( aCmd[iOpt].eOpt ){
    case BTC_ATTACH: {
      if( p->pEnv ){
        Tcl_AppendResult(interp, "object has already been attached to db", 0);
        rc = TCL_ERROR;
      }else{
        sqlite4 *db = 0;
        rc = sqlite4TestDbHandle(interp, objv[2], &db);
        if( rc==TCL_OK ){
          void *pArg = (void*)(&p->pEnv);
          rc = sqlite4_kvstore_control(db, "main", BT_CONTROL_GETVFS, pArg);
          if( rc==SQLITE4_NOTFOUND ){
            Tcl_AppendResult(interp, "not a bt database", 0);
            rc = TCL_ERROR;
          }else{
            sqlite4_kvstore_control(db, "main", BT_CONTROL_SETVFS, (void*)p);
          }
        }
      }
      break;
    }

    case BTC_DELETE: {
      Tcl_DeleteCommand(interp, Tcl_GetString(objv[0]));
      break;
    }

    case BTC_IOERR: {
      int ret = p->nIoerrInjected;
      if( Tcl_GetIntFromObj(interp, objv[2], &p->nIoerrCnt)
       || Tcl_GetBooleanFromObj(interp, objv[3], &p->bIoerrPersist)
      ){
        rc = TCL_ERROR;
      }else{
        Tcl_SetObjResult(interp, Tcl_NewIntObj(ret));
      }
      
      break;
    }

    default:
      assert( 0 );
      break;
  }

  return rc;
}

/*
** Tcl command: btenv NAME
*/
static int test_btenv(
  void * clientData,
  Tcl_Interp *interp,
  int objc,
  Tcl_Obj *CONST objv[]
){
  BtTestEnv *p;
  const char *zName;

  if( objc!=2 ){
    Tcl_WrongNumArgs(interp, 1, objv, "NAME");
    return TCL_ERROR;
  }
  zName = Tcl_GetString(objv[1]);

  p = ckalloc(sizeof(BtTestEnv));
  memset(p, 0, sizeof(BtTestEnv));

  p->base.xFullpath = testBtOsFullpath;
  p->base.xOpen = testBtOsOpen;
  p->base.xSize = testBtOsSize;
  p->base.xRead = testBtOsRead;
  p->base.xWrite = testBtOsWrite;
  p->base.xTruncate = testBtOsTruncate;
  p->base.xSync = testBtOsSync;
  p->base.xSectorSize = testBtOsSectorSize;
  p->base.xClose = testBtOsClose;
  p->base.xUnlink = testBtOsUnlink;
  p->base.xLock = testBtOsLock;
  p->base.xTestLock = testBtOsTestLock;
  p->base.xShmMap = testBtOsShmMap;
  p->base.xShmBarrier = testBtOsShmBarrier;
  p->base.xShmUnmap = testBtOsShmUnmap;

  Tcl_CreateObjCommand(interp, zName, test_btenv_cmd, (void*)p, test_btenv_del);
  Tcl_SetObjResult(interp, Tcl_NewStringObj(zName, -1));
  return TCL_OK;
}

int SqlitetestBt_Init(Tcl_Interp *interp){
  struct SyscallCmd {
    const char *zName;
    Tcl_ObjCmdProc *xCmd;
  } aCmd[] = {
    { "btenv",                  test_btenv },
  };
  int i;

  for(i=0; i<sizeof(aCmd)/sizeof(aCmd[0]); i++){
    Tcl_CreateObjCommand(interp, aCmd[i].zName, aCmd[i].xCmd, 0, 0);
  }
  return TCL_OK;
}

  return TCL_OK;
}

void sqlite4TestInit(Tcl_Interp *interp){
  Sqlitetest_auth_init(interp);
  Sqlitetest_num_init(interp);
  Sqlitetest_func_Init(interp);
  SqlitetestBt_Init(interp);
}

/*
** Register commands with the TCL interpreter.
*/
int Sqlitetest1_Init(Tcl_Interp *interp){
  extern int sqlite4_search_count;