diff options
Diffstat (limited to 'lib/libsqlite3/src/test_vfs.c')
| -rw-r--r-- | lib/libsqlite3/src/test_vfs.c | 1542 |
1 files changed, 0 insertions, 1542 deletions
diff --git a/lib/libsqlite3/src/test_vfs.c b/lib/libsqlite3/src/test_vfs.c deleted file mode 100644 index a8c6ae7bf50..00000000000 --- a/lib/libsqlite3/src/test_vfs.c +++ /dev/null @@ -1,1542 +0,0 @@ -/* -** 2010 May 05 -** -** 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. -** -****************************************************************************** -** -** This file contains the implementation of the Tcl [testvfs] command, -** used to create SQLite VFS implementations with various properties and -** instrumentation to support testing SQLite. -** -** testvfs VFSNAME ?OPTIONS? -** -** Available options are: -** -** -noshm BOOLEAN (True to omit shm methods. Default false) -** -default BOOLEAN (True to make the vfs default. Default false) -** -szosfile INTEGER (Value for sqlite3_vfs.szOsFile) -** -mxpathname INTEGER (Value for sqlite3_vfs.mxPathname) -** -iversion INTEGER (Value for sqlite3_vfs.iVersion) -*/ -#if SQLITE_TEST /* This file is used for testing only */ - -#include "sqlite3.h" -#include "sqliteInt.h" -#include <tcl.h> - -typedef struct Testvfs Testvfs; -typedef struct TestvfsShm TestvfsShm; -typedef struct TestvfsBuffer TestvfsBuffer; -typedef struct TestvfsFile TestvfsFile; -typedef struct TestvfsFd TestvfsFd; - -/* -** An open file handle. -*/ -struct TestvfsFile { - sqlite3_file base; /* Base class. Must be first */ - TestvfsFd *pFd; /* File data */ -}; -#define tvfsGetFd(pFile) (((TestvfsFile *)pFile)->pFd) - -struct TestvfsFd { - sqlite3_vfs *pVfs; /* The VFS */ - const char *zFilename; /* Filename as passed to xOpen() */ - sqlite3_file *pReal; /* The real, underlying file descriptor */ - Tcl_Obj *pShmId; /* Shared memory id for Tcl callbacks */ - - TestvfsBuffer *pShm; /* Shared memory buffer */ - u32 excllock; /* Mask of exclusive locks */ - u32 sharedlock; /* Mask of shared locks */ - TestvfsFd *pNext; /* Next handle opened on the same file */ -}; - - -#define FAULT_INJECT_NONE 0 -#define FAULT_INJECT_TRANSIENT 1 -#define FAULT_INJECT_PERSISTENT 2 - -typedef struct TestFaultInject TestFaultInject; -struct TestFaultInject { - int iCnt; /* Remaining calls before fault injection */ - int eFault; /* A FAULT_INJECT_* value */ - int nFail; /* Number of faults injected */ -}; - -/* -** An instance of this structure is allocated for each VFS created. The -** sqlite3_vfs.pAppData field of the VFS structure registered with SQLite -** is set to point to it. -*/ -struct Testvfs { - char *zName; /* Name of this VFS */ - sqlite3_vfs *pParent; /* The VFS to use for file IO */ - sqlite3_vfs *pVfs; /* The testvfs registered with SQLite */ - Tcl_Interp *interp; /* Interpreter to run script in */ - Tcl_Obj *pScript; /* Script to execute */ - TestvfsBuffer *pBuffer; /* List of shared buffers */ - int isNoshm; - int isFullshm; - - int mask; /* Mask controlling [script] and [ioerr] */ - - TestFaultInject ioerr_err; - TestFaultInject full_err; - TestFaultInject cantopen_err; - -#if 0 - int iIoerrCnt; - int ioerr; - int nIoerrFail; - int iFullCnt; - int fullerr; - int nFullFail; -#endif - - int iDevchar; - int iSectorsize; -}; - -/* -** The Testvfs.mask variable is set to a combination of the following. -** If a bit is clear in Testvfs.mask, then calls made by SQLite to the -** corresponding VFS method is ignored for purposes of: -** -** + Simulating IO errors, and -** + Invoking the Tcl callback script. -*/ -#define TESTVFS_SHMOPEN_MASK 0x00000001 -#define TESTVFS_SHMLOCK_MASK 0x00000010 -#define TESTVFS_SHMMAP_MASK 0x00000020 -#define TESTVFS_SHMBARRIER_MASK 0x00000040 -#define TESTVFS_SHMCLOSE_MASK 0x00000080 - -#define TESTVFS_OPEN_MASK 0x00000100 -#define TESTVFS_SYNC_MASK 0x00000200 -#define TESTVFS_DELETE_MASK 0x00000400 -#define TESTVFS_CLOSE_MASK 0x00000800 -#define TESTVFS_WRITE_MASK 0x00001000 -#define TESTVFS_TRUNCATE_MASK 0x00002000 -#define TESTVFS_ACCESS_MASK 0x00004000 -#define TESTVFS_FULLPATHNAME_MASK 0x00008000 -#define TESTVFS_READ_MASK 0x00010000 -#define TESTVFS_UNLOCK_MASK 0x00020000 -#define TESTVFS_LOCK_MASK 0x00040000 -#define TESTVFS_CKLOCK_MASK 0x00080000 - -#define TESTVFS_ALL_MASK 0x000FFFFF - - -#define TESTVFS_MAX_PAGES 1024 - -/* -** A shared-memory buffer. There is one of these objects for each shared -** memory region opened by clients. If two clients open the same file, -** there are two TestvfsFile structures but only one TestvfsBuffer structure. -*/ -struct TestvfsBuffer { - char *zFile; /* Associated file name */ - int pgsz; /* Page size */ - u8 *aPage[TESTVFS_MAX_PAGES]; /* Array of ckalloc'd pages */ - TestvfsFd *pFile; /* List of open handles */ - TestvfsBuffer *pNext; /* Next in linked list of all buffers */ -}; - - -#define PARENTVFS(x) (((Testvfs *)((x)->pAppData))->pParent) - -#define TESTVFS_MAX_ARGS 12 - - -/* -** Method declarations for TestvfsFile. -*/ -static int tvfsClose(sqlite3_file*); -static int tvfsRead(sqlite3_file*, void*, int iAmt, sqlite3_int64 iOfst); -static int tvfsWrite(sqlite3_file*,const void*,int iAmt, sqlite3_int64 iOfst); -static int tvfsTruncate(sqlite3_file*, sqlite3_int64 size); -static int tvfsSync(sqlite3_file*, int flags); -static int tvfsFileSize(sqlite3_file*, sqlite3_int64 *pSize); -static int tvfsLock(sqlite3_file*, int); -static int tvfsUnlock(sqlite3_file*, int); -static int tvfsCheckReservedLock(sqlite3_file*, int *); -static int tvfsFileControl(sqlite3_file*, int op, void *pArg); -static int tvfsSectorSize(sqlite3_file*); -static int tvfsDeviceCharacteristics(sqlite3_file*); - -/* -** Method declarations for tvfs_vfs. -*/ -static int tvfsOpen(sqlite3_vfs*, const char *, sqlite3_file*, int , int *); -static int tvfsDelete(sqlite3_vfs*, const char *zName, int syncDir); -static int tvfsAccess(sqlite3_vfs*, const char *zName, int flags, int *); -static int tvfsFullPathname(sqlite3_vfs*, const char *zName, int, char *zOut); -#ifndef SQLITE_OMIT_LOAD_EXTENSION -static void *tvfsDlOpen(sqlite3_vfs*, const char *zFilename); -static void tvfsDlError(sqlite3_vfs*, int nByte, char *zErrMsg); -static void (*tvfsDlSym(sqlite3_vfs*,void*, const char *zSymbol))(void); -static void tvfsDlClose(sqlite3_vfs*, void*); -#endif /* SQLITE_OMIT_LOAD_EXTENSION */ -static int tvfsRandomness(sqlite3_vfs*, int nByte, char *zOut); -static int tvfsSleep(sqlite3_vfs*, int microseconds); -static int tvfsCurrentTime(sqlite3_vfs*, double*); - -static int tvfsShmOpen(sqlite3_file*); -static int tvfsShmLock(sqlite3_file*, int , int, int); -static int tvfsShmMap(sqlite3_file*,int,int,int, void volatile **); -static void tvfsShmBarrier(sqlite3_file*); -static int tvfsShmUnmap(sqlite3_file*, int); - -static int tvfsFetch(sqlite3_file*, sqlite3_int64, int, void**); -static int tvfsUnfetch(sqlite3_file*, sqlite3_int64, void*); - -static sqlite3_io_methods tvfs_io_methods = { - 3, /* iVersion */ - tvfsClose, /* xClose */ - tvfsRead, /* xRead */ - tvfsWrite, /* xWrite */ - tvfsTruncate, /* xTruncate */ - tvfsSync, /* xSync */ - tvfsFileSize, /* xFileSize */ - tvfsLock, /* xLock */ - tvfsUnlock, /* xUnlock */ - tvfsCheckReservedLock, /* xCheckReservedLock */ - tvfsFileControl, /* xFileControl */ - tvfsSectorSize, /* xSectorSize */ - tvfsDeviceCharacteristics, /* xDeviceCharacteristics */ - tvfsShmMap, /* xShmMap */ - tvfsShmLock, /* xShmLock */ - tvfsShmBarrier, /* xShmBarrier */ - tvfsShmUnmap, /* xShmUnmap */ - tvfsFetch, - tvfsUnfetch -}; - -static int tvfsResultCode(Testvfs *p, int *pRc){ - struct errcode { - int eCode; - const char *zCode; - } aCode[] = { - { SQLITE_OK, "SQLITE_OK" }, - { SQLITE_ERROR, "SQLITE_ERROR" }, - { SQLITE_IOERR, "SQLITE_IOERR" }, - { SQLITE_LOCKED, "SQLITE_LOCKED" }, - { SQLITE_BUSY, "SQLITE_BUSY" }, - }; - - const char *z; - int i; - - z = Tcl_GetStringResult(p->interp); - for(i=0; i<ArraySize(aCode); i++){ - if( 0==strcmp(z, aCode[i].zCode) ){ - *pRc = aCode[i].eCode; - return 1; - } - } - - return 0; -} - -static int tvfsInjectFault(TestFaultInject *p){ - int ret = 0; - if( p->eFault ){ - p->iCnt--; - if( p->iCnt==0 || (p->iCnt<0 && p->eFault==FAULT_INJECT_PERSISTENT ) ){ - ret = 1; - p->nFail++; - } - } - return ret; -} - - -static int tvfsInjectIoerr(Testvfs *p){ - return tvfsInjectFault(&p->ioerr_err); -} - -static int tvfsInjectFullerr(Testvfs *p){ - return tvfsInjectFault(&p->full_err); -} -static int tvfsInjectCantopenerr(Testvfs *p){ - return tvfsInjectFault(&p->cantopen_err); -} - - -static void tvfsExecTcl( - Testvfs *p, - const char *zMethod, - Tcl_Obj *arg1, - Tcl_Obj *arg2, - Tcl_Obj *arg3, - Tcl_Obj *arg4 -){ - int rc; /* Return code from Tcl_EvalObj() */ - Tcl_Obj *pEval; - assert( p->pScript ); - - assert( zMethod ); - assert( p ); - assert( arg2==0 || arg1!=0 ); - assert( arg3==0 || arg2!=0 ); - - pEval = Tcl_DuplicateObj(p->pScript); - Tcl_IncrRefCount(p->pScript); - Tcl_ListObjAppendElement(p->interp, pEval, Tcl_NewStringObj(zMethod, -1)); - if( arg1 ) Tcl_ListObjAppendElement(p->interp, pEval, arg1); - if( arg2 ) Tcl_ListObjAppendElement(p->interp, pEval, arg2); - if( arg3 ) Tcl_ListObjAppendElement(p->interp, pEval, arg3); - if( arg4 ) Tcl_ListObjAppendElement(p->interp, pEval, arg4); - - rc = Tcl_EvalObjEx(p->interp, pEval, TCL_EVAL_GLOBAL); - if( rc!=TCL_OK ){ - Tcl_BackgroundError(p->interp); - Tcl_ResetResult(p->interp); - } -} - - -/* -** Close an tvfs-file. -*/ -static int tvfsClose(sqlite3_file *pFile){ - int rc; - TestvfsFile *pTestfile = (TestvfsFile *)pFile; - TestvfsFd *pFd = pTestfile->pFd; - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - - if( p->pScript && p->mask&TESTVFS_CLOSE_MASK ){ - tvfsExecTcl(p, "xClose", - Tcl_NewStringObj(pFd->zFilename, -1), pFd->pShmId, 0, 0 - ); - } - - if( pFd->pShmId ){ - Tcl_DecrRefCount(pFd->pShmId); - pFd->pShmId = 0; - } - if( pFile->pMethods ){ - ckfree((char *)pFile->pMethods); - } - rc = sqlite3OsClose(pFd->pReal); - ckfree((char *)pFd); - pTestfile->pFd = 0; - return rc; -} - -/* -** Read data from an tvfs-file. -*/ -static int tvfsRead( - sqlite3_file *pFile, - void *zBuf, - int iAmt, - sqlite_int64 iOfst -){ - int rc = SQLITE_OK; - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - if( p->pScript && p->mask&TESTVFS_READ_MASK ){ - tvfsExecTcl(p, "xRead", - Tcl_NewStringObj(pFd->zFilename, -1), pFd->pShmId, 0, 0 - ); - tvfsResultCode(p, &rc); - } - if( rc==SQLITE_OK && p->mask&TESTVFS_READ_MASK && tvfsInjectIoerr(p) ){ - rc = SQLITE_IOERR; - } - if( rc==SQLITE_OK ){ - rc = sqlite3OsRead(pFd->pReal, zBuf, iAmt, iOfst); - } - return rc; -} - -/* -** Write data to an tvfs-file. -*/ -static int tvfsWrite( - sqlite3_file *pFile, - const void *zBuf, - int iAmt, - sqlite_int64 iOfst -){ - int rc = SQLITE_OK; - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - - if( p->pScript && p->mask&TESTVFS_WRITE_MASK ){ - tvfsExecTcl(p, "xWrite", - Tcl_NewStringObj(pFd->zFilename, -1), pFd->pShmId, - Tcl_NewWideIntObj(iOfst), Tcl_NewIntObj(iAmt) - ); - tvfsResultCode(p, &rc); - } - - if( rc==SQLITE_OK && tvfsInjectFullerr(p) ){ - rc = SQLITE_FULL; - } - if( rc==SQLITE_OK && p->mask&TESTVFS_WRITE_MASK && tvfsInjectIoerr(p) ){ - rc = SQLITE_IOERR; - } - - if( rc==SQLITE_OK ){ - rc = sqlite3OsWrite(pFd->pReal, zBuf, iAmt, iOfst); - } - return rc; -} - -/* -** Truncate an tvfs-file. -*/ -static int tvfsTruncate(sqlite3_file *pFile, sqlite_int64 size){ - int rc = SQLITE_OK; - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - - if( p->pScript && p->mask&TESTVFS_TRUNCATE_MASK ){ - tvfsExecTcl(p, "xTruncate", - Tcl_NewStringObj(pFd->zFilename, -1), pFd->pShmId, 0, 0 - ); - tvfsResultCode(p, &rc); - } - - if( rc==SQLITE_OK ){ - rc = sqlite3OsTruncate(pFd->pReal, size); - } - return rc; -} - -/* -** Sync an tvfs-file. -*/ -static int tvfsSync(sqlite3_file *pFile, int flags){ - int rc = SQLITE_OK; - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - - if( p->pScript && p->mask&TESTVFS_SYNC_MASK ){ - char *zFlags = 0; - - switch( flags ){ - case SQLITE_SYNC_NORMAL: - zFlags = "normal"; - break; - case SQLITE_SYNC_FULL: - zFlags = "full"; - break; - case SQLITE_SYNC_NORMAL|SQLITE_SYNC_DATAONLY: - zFlags = "normal|dataonly"; - break; - case SQLITE_SYNC_FULL|SQLITE_SYNC_DATAONLY: - zFlags = "full|dataonly"; - break; - default: - assert(0); - } - - tvfsExecTcl(p, "xSync", - Tcl_NewStringObj(pFd->zFilename, -1), pFd->pShmId, - Tcl_NewStringObj(zFlags, -1), 0 - ); - tvfsResultCode(p, &rc); - } - - if( rc==SQLITE_OK && tvfsInjectFullerr(p) ) rc = SQLITE_FULL; - - if( rc==SQLITE_OK ){ - rc = sqlite3OsSync(pFd->pReal, flags); - } - - return rc; -} - -/* -** Return the current file-size of an tvfs-file. -*/ -static int tvfsFileSize(sqlite3_file *pFile, sqlite_int64 *pSize){ - TestvfsFd *p = tvfsGetFd(pFile); - return sqlite3OsFileSize(p->pReal, pSize); -} - -/* -** Lock an tvfs-file. -*/ -static int tvfsLock(sqlite3_file *pFile, int eLock){ - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - if( p->pScript && p->mask&TESTVFS_LOCK_MASK ){ - char zLock[30]; - sqlite3_snprintf(sizeof(zLock),zLock,"%d",eLock); - tvfsExecTcl(p, "xLock", Tcl_NewStringObj(pFd->zFilename, -1), - Tcl_NewStringObj(zLock, -1), 0, 0); - } - return sqlite3OsLock(pFd->pReal, eLock); -} - -/* -** Unlock an tvfs-file. -*/ -static int tvfsUnlock(sqlite3_file *pFile, int eLock){ - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - if( p->pScript && p->mask&TESTVFS_UNLOCK_MASK ){ - char zLock[30]; - sqlite3_snprintf(sizeof(zLock),zLock,"%d",eLock); - tvfsExecTcl(p, "xUnlock", Tcl_NewStringObj(pFd->zFilename, -1), - Tcl_NewStringObj(zLock, -1), 0, 0); - } - if( p->mask&TESTVFS_WRITE_MASK && tvfsInjectIoerr(p) ){ - return SQLITE_IOERR_UNLOCK; - } - return sqlite3OsUnlock(pFd->pReal, eLock); -} - -/* -** Check if another file-handle holds a RESERVED lock on an tvfs-file. -*/ -static int tvfsCheckReservedLock(sqlite3_file *pFile, int *pResOut){ - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - if( p->pScript && p->mask&TESTVFS_CKLOCK_MASK ){ - tvfsExecTcl(p, "xCheckReservedLock", Tcl_NewStringObj(pFd->zFilename, -1), - 0, 0, 0); - } - return sqlite3OsCheckReservedLock(pFd->pReal, pResOut); -} - -/* -** File control method. For custom operations on an tvfs-file. -*/ -static int tvfsFileControl(sqlite3_file *pFile, int op, void *pArg){ - TestvfsFd *p = tvfsGetFd(pFile); - if( op==SQLITE_FCNTL_PRAGMA ){ - char **argv = (char**)pArg; - if( sqlite3_stricmp(argv[1],"error")==0 ){ - int rc = SQLITE_ERROR; - if( argv[2] ){ - const char *z = argv[2]; - int x = atoi(z); - if( x ){ - rc = x; - while( sqlite3Isdigit(z[0]) ){ z++; } - while( sqlite3Isspace(z[0]) ){ z++; } - } - if( z[0] ) argv[0] = sqlite3_mprintf("%s", z); - } - return rc; - } - if( sqlite3_stricmp(argv[1], "filename")==0 ){ - argv[0] = sqlite3_mprintf("%s", p->zFilename); - return SQLITE_OK; - } - } - return sqlite3OsFileControl(p->pReal, op, pArg); -} - -/* -** Return the sector-size in bytes for an tvfs-file. -*/ -static int tvfsSectorSize(sqlite3_file *pFile){ - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - if( p->iSectorsize>=0 ){ - return p->iSectorsize; - } - return sqlite3OsSectorSize(pFd->pReal); -} - -/* -** Return the device characteristic flags supported by an tvfs-file. -*/ -static int tvfsDeviceCharacteristics(sqlite3_file *pFile){ - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)pFd->pVfs->pAppData; - if( p->iDevchar>=0 ){ - return p->iDevchar; - } - return sqlite3OsDeviceCharacteristics(pFd->pReal); -} - -/* -** Open an tvfs file handle. -*/ -static int tvfsOpen( - sqlite3_vfs *pVfs, - const char *zName, - sqlite3_file *pFile, - int flags, - int *pOutFlags -){ - int rc; - TestvfsFile *pTestfile = (TestvfsFile *)pFile; - TestvfsFd *pFd; - Tcl_Obj *pId = 0; - Testvfs *p = (Testvfs *)pVfs->pAppData; - - pFd = (TestvfsFd *)ckalloc(sizeof(TestvfsFd) + PARENTVFS(pVfs)->szOsFile); - memset(pFd, 0, sizeof(TestvfsFd) + PARENTVFS(pVfs)->szOsFile); - pFd->pShm = 0; - pFd->pShmId = 0; - pFd->zFilename = zName; - pFd->pVfs = pVfs; - pFd->pReal = (sqlite3_file *)&pFd[1]; - memset(pTestfile, 0, sizeof(TestvfsFile)); - pTestfile->pFd = pFd; - - /* Evaluate the Tcl script: - ** - ** SCRIPT xOpen FILENAME KEY-VALUE-ARGS - ** - ** If the script returns an SQLite error code other than SQLITE_OK, an - ** error is returned to the caller. If it returns SQLITE_OK, the new - ** connection is named "anon". Otherwise, the value returned by the - ** script is used as the connection name. - */ - Tcl_ResetResult(p->interp); - if( p->pScript && p->mask&TESTVFS_OPEN_MASK ){ - Tcl_Obj *pArg = Tcl_NewObj(); - Tcl_IncrRefCount(pArg); - if( flags&SQLITE_OPEN_MAIN_DB ){ - const char *z = &zName[strlen(zName)+1]; - while( *z ){ - Tcl_ListObjAppendElement(0, pArg, Tcl_NewStringObj(z, -1)); - z += strlen(z) + 1; - Tcl_ListObjAppendElement(0, pArg, Tcl_NewStringObj(z, -1)); - z += strlen(z) + 1; - } - } - tvfsExecTcl(p, "xOpen", Tcl_NewStringObj(pFd->zFilename, -1), pArg, 0, 0); - Tcl_DecrRefCount(pArg); - if( tvfsResultCode(p, &rc) ){ - if( rc!=SQLITE_OK ) return rc; - }else{ - pId = Tcl_GetObjResult(p->interp); - } - } - - if( (p->mask&TESTVFS_OPEN_MASK) && tvfsInjectIoerr(p) ) return SQLITE_IOERR; - if( tvfsInjectCantopenerr(p) ) return SQLITE_CANTOPEN; - if( tvfsInjectFullerr(p) ) return SQLITE_FULL; - - if( !pId ){ - pId = Tcl_NewStringObj("anon", -1); - } - Tcl_IncrRefCount(pId); - pFd->pShmId = pId; - Tcl_ResetResult(p->interp); - - rc = sqlite3OsOpen(PARENTVFS(pVfs), zName, pFd->pReal, flags, pOutFlags); - if( pFd->pReal->pMethods ){ - sqlite3_io_methods *pMethods; - int nByte; - - if( pVfs->iVersion>1 ){ - nByte = sizeof(sqlite3_io_methods); - }else{ - nByte = offsetof(sqlite3_io_methods, xShmMap); - } - - pMethods = (sqlite3_io_methods *)ckalloc(nByte); - memcpy(pMethods, &tvfs_io_methods, nByte); - pMethods->iVersion = pFd->pReal->pMethods->iVersion; - if( pMethods->iVersion>pVfs->iVersion ){ - pMethods->iVersion = pVfs->iVersion; - } - if( pVfs->iVersion>1 && ((Testvfs *)pVfs->pAppData)->isNoshm ){ - pMethods->xShmUnmap = 0; - pMethods->xShmLock = 0; - pMethods->xShmBarrier = 0; - pMethods->xShmMap = 0; - } - pFile->pMethods = pMethods; - } - - return rc; -} - -/* -** Delete the file located at zPath. If the dirSync argument is true, -** ensure the file-system modifications are synced to disk before -** returning. -*/ -static int tvfsDelete(sqlite3_vfs *pVfs, const char *zPath, int dirSync){ - int rc = SQLITE_OK; - Testvfs *p = (Testvfs *)pVfs->pAppData; - - if( p->pScript && p->mask&TESTVFS_DELETE_MASK ){ - tvfsExecTcl(p, "xDelete", - Tcl_NewStringObj(zPath, -1), Tcl_NewIntObj(dirSync), 0, 0 - ); - tvfsResultCode(p, &rc); - } - if( rc==SQLITE_OK ){ - rc = sqlite3OsDelete(PARENTVFS(pVfs), zPath, dirSync); - } - return rc; -} - -/* -** Test for access permissions. Return true if the requested permission -** is available, or false otherwise. -*/ -static int tvfsAccess( - sqlite3_vfs *pVfs, - const char *zPath, - int flags, - int *pResOut -){ - Testvfs *p = (Testvfs *)pVfs->pAppData; - if( p->pScript && p->mask&TESTVFS_ACCESS_MASK ){ - int rc; - char *zArg = 0; - if( flags==SQLITE_ACCESS_EXISTS ) zArg = "SQLITE_ACCESS_EXISTS"; - if( flags==SQLITE_ACCESS_READWRITE ) zArg = "SQLITE_ACCESS_READWRITE"; - if( flags==SQLITE_ACCESS_READ ) zArg = "SQLITE_ACCESS_READ"; - tvfsExecTcl(p, "xAccess", - Tcl_NewStringObj(zPath, -1), Tcl_NewStringObj(zArg, -1), 0, 0 - ); - if( tvfsResultCode(p, &rc) ){ - if( rc!=SQLITE_OK ) return rc; - }else{ - Tcl_Interp *interp = p->interp; - if( TCL_OK==Tcl_GetBooleanFromObj(0, Tcl_GetObjResult(interp), pResOut) ){ - return SQLITE_OK; - } - } - } - return sqlite3OsAccess(PARENTVFS(pVfs), zPath, flags, pResOut); -} - -/* -** Populate buffer zOut with the full canonical pathname corresponding -** to the pathname in zPath. zOut is guaranteed to point to a buffer -** of at least (DEVSYM_MAX_PATHNAME+1) bytes. -*/ -static int tvfsFullPathname( - sqlite3_vfs *pVfs, - const char *zPath, - int nOut, - char *zOut -){ - Testvfs *p = (Testvfs *)pVfs->pAppData; - if( p->pScript && p->mask&TESTVFS_FULLPATHNAME_MASK ){ - int rc; - tvfsExecTcl(p, "xFullPathname", Tcl_NewStringObj(zPath, -1), 0, 0, 0); - if( tvfsResultCode(p, &rc) ){ - if( rc!=SQLITE_OK ) return rc; - } - } - return sqlite3OsFullPathname(PARENTVFS(pVfs), zPath, nOut, zOut); -} - -#ifndef SQLITE_OMIT_LOAD_EXTENSION -/* -** Open the dynamic library located at zPath and return a handle. -*/ -static void *tvfsDlOpen(sqlite3_vfs *pVfs, const char *zPath){ - return sqlite3OsDlOpen(PARENTVFS(pVfs), zPath); -} - -/* -** Populate the buffer zErrMsg (size nByte bytes) with a human readable -** utf-8 string describing the most recent error encountered associated -** with dynamic libraries. -*/ -static void tvfsDlError(sqlite3_vfs *pVfs, int nByte, char *zErrMsg){ - sqlite3OsDlError(PARENTVFS(pVfs), nByte, zErrMsg); -} - -/* -** Return a pointer to the symbol zSymbol in the dynamic library pHandle. -*/ -static void (*tvfsDlSym(sqlite3_vfs *pVfs, void *p, const char *zSym))(void){ - return sqlite3OsDlSym(PARENTVFS(pVfs), p, zSym); -} - -/* -** Close the dynamic library handle pHandle. -*/ -static void tvfsDlClose(sqlite3_vfs *pVfs, void *pHandle){ - sqlite3OsDlClose(PARENTVFS(pVfs), pHandle); -} -#endif /* SQLITE_OMIT_LOAD_EXTENSION */ - -/* -** Populate the buffer pointed to by zBufOut with nByte bytes of -** random data. -*/ -static int tvfsRandomness(sqlite3_vfs *pVfs, int nByte, char *zBufOut){ - return sqlite3OsRandomness(PARENTVFS(pVfs), nByte, zBufOut); -} - -/* -** Sleep for nMicro microseconds. Return the number of microseconds -** actually slept. -*/ -static int tvfsSleep(sqlite3_vfs *pVfs, int nMicro){ - return sqlite3OsSleep(PARENTVFS(pVfs), nMicro); -} - -/* -** Return the current time as a Julian Day number in *pTimeOut. -*/ -static int tvfsCurrentTime(sqlite3_vfs *pVfs, double *pTimeOut){ - return PARENTVFS(pVfs)->xCurrentTime(PARENTVFS(pVfs), pTimeOut); -} - -static int tvfsShmOpen(sqlite3_file *pFile){ - Testvfs *p; - int rc = SQLITE_OK; /* Return code */ - TestvfsBuffer *pBuffer; /* Buffer to open connection to */ - TestvfsFd *pFd; /* The testvfs file structure */ - - pFd = tvfsGetFd(pFile); - p = (Testvfs *)pFd->pVfs->pAppData; - assert( 0==p->isFullshm ); - assert( pFd->pShmId && pFd->pShm==0 && pFd->pNext==0 ); - - /* Evaluate the Tcl script: - ** - ** SCRIPT xShmOpen FILENAME - */ - Tcl_ResetResult(p->interp); - if( p->pScript && p->mask&TESTVFS_SHMOPEN_MASK ){ - tvfsExecTcl(p, "xShmOpen", Tcl_NewStringObj(pFd->zFilename, -1), 0, 0, 0); - if( tvfsResultCode(p, &rc) ){ - if( rc!=SQLITE_OK ) return rc; - } - } - - assert( rc==SQLITE_OK ); - if( p->mask&TESTVFS_SHMOPEN_MASK && tvfsInjectIoerr(p) ){ - return SQLITE_IOERR; - } - - /* Search for a TestvfsBuffer. Create a new one if required. */ - for(pBuffer=p->pBuffer; pBuffer; pBuffer=pBuffer->pNext){ - if( 0==strcmp(pFd->zFilename, pBuffer->zFile) ) break; - } - if( !pBuffer ){ - int szName = (int)strlen(pFd->zFilename); - int nByte = sizeof(TestvfsBuffer) + szName + 1; - pBuffer = (TestvfsBuffer *)ckalloc(nByte); - memset(pBuffer, 0, nByte); - pBuffer->zFile = (char *)&pBuffer[1]; - memcpy(pBuffer->zFile, pFd->zFilename, szName+1); - pBuffer->pNext = p->pBuffer; - p->pBuffer = pBuffer; - } - - /* Connect the TestvfsBuffer to the new TestvfsShm handle and return. */ - pFd->pNext = pBuffer->pFile; - pBuffer->pFile = pFd; - pFd->pShm = pBuffer; - return SQLITE_OK; -} - -static void tvfsAllocPage(TestvfsBuffer *p, int iPage, int pgsz){ - assert( iPage<TESTVFS_MAX_PAGES ); - if( p->aPage[iPage]==0 ){ - p->aPage[iPage] = (u8 *)ckalloc(pgsz); - memset(p->aPage[iPage], 0, pgsz); - p->pgsz = pgsz; - } -} - -static int tvfsShmMap( - sqlite3_file *pFile, /* Handle open on database file */ - int iPage, /* Page to retrieve */ - int pgsz, /* Size of pages */ - int isWrite, /* True to extend file if necessary */ - void volatile **pp /* OUT: Mapped memory */ -){ - int rc = SQLITE_OK; - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)(pFd->pVfs->pAppData); - - if( p->isFullshm ){ - return sqlite3OsShmMap(pFd->pReal, iPage, pgsz, isWrite, pp); - } - - if( 0==pFd->pShm ){ - rc = tvfsShmOpen(pFile); - if( rc!=SQLITE_OK ){ - return rc; - } - } - - if( p->pScript && p->mask&TESTVFS_SHMMAP_MASK ){ - Tcl_Obj *pArg = Tcl_NewObj(); - Tcl_IncrRefCount(pArg); - Tcl_ListObjAppendElement(p->interp, pArg, Tcl_NewIntObj(iPage)); - Tcl_ListObjAppendElement(p->interp, pArg, Tcl_NewIntObj(pgsz)); - Tcl_ListObjAppendElement(p->interp, pArg, Tcl_NewIntObj(isWrite)); - tvfsExecTcl(p, "xShmMap", - Tcl_NewStringObj(pFd->pShm->zFile, -1), pFd->pShmId, pArg, 0 - ); - tvfsResultCode(p, &rc); - Tcl_DecrRefCount(pArg); - } - if( rc==SQLITE_OK && p->mask&TESTVFS_SHMMAP_MASK && tvfsInjectIoerr(p) ){ - rc = SQLITE_IOERR; - } - - if( rc==SQLITE_OK && isWrite && !pFd->pShm->aPage[iPage] ){ - tvfsAllocPage(pFd->pShm, iPage, pgsz); - } - *pp = (void volatile *)pFd->pShm->aPage[iPage]; - - return rc; -} - - -static int tvfsShmLock( - sqlite3_file *pFile, - int ofst, - int n, - int flags -){ - int rc = SQLITE_OK; - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)(pFd->pVfs->pAppData); - int nLock; - char zLock[80]; - - if( p->isFullshm ){ - return sqlite3OsShmLock(pFd->pReal, ofst, n, flags); - } - - if( p->pScript && p->mask&TESTVFS_SHMLOCK_MASK ){ - sqlite3_snprintf(sizeof(zLock), zLock, "%d %d", ofst, n); - nLock = (int)strlen(zLock); - if( flags & SQLITE_SHM_LOCK ){ - strcpy(&zLock[nLock], " lock"); - }else{ - strcpy(&zLock[nLock], " unlock"); - } - nLock += (int)strlen(&zLock[nLock]); - if( flags & SQLITE_SHM_SHARED ){ - strcpy(&zLock[nLock], " shared"); - }else{ - strcpy(&zLock[nLock], " exclusive"); - } - tvfsExecTcl(p, "xShmLock", - Tcl_NewStringObj(pFd->pShm->zFile, -1), pFd->pShmId, - Tcl_NewStringObj(zLock, -1), 0 - ); - tvfsResultCode(p, &rc); - } - - if( rc==SQLITE_OK && p->mask&TESTVFS_SHMLOCK_MASK && tvfsInjectIoerr(p) ){ - rc = SQLITE_IOERR; - } - - if( rc==SQLITE_OK ){ - int isLock = (flags & SQLITE_SHM_LOCK); - int isExcl = (flags & SQLITE_SHM_EXCLUSIVE); - u32 mask = (((1<<n)-1) << ofst); - if( isLock ){ - TestvfsFd *p2; - for(p2=pFd->pShm->pFile; p2; p2=p2->pNext){ - if( p2==pFd ) continue; - if( (p2->excllock&mask) || (isExcl && p2->sharedlock&mask) ){ - rc = SQLITE_BUSY; - break; - } - } - if( rc==SQLITE_OK ){ - if( isExcl ) pFd->excllock |= mask; - if( !isExcl ) pFd->sharedlock |= mask; - } - }else{ - if( isExcl ) pFd->excllock &= (~mask); - if( !isExcl ) pFd->sharedlock &= (~mask); - } - } - - return rc; -} - -static void tvfsShmBarrier(sqlite3_file *pFile){ - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)(pFd->pVfs->pAppData); - - if( p->pScript && p->mask&TESTVFS_SHMBARRIER_MASK ){ - const char *z = pFd->pShm ? pFd->pShm->zFile : ""; - tvfsExecTcl(p, "xShmBarrier", Tcl_NewStringObj(z, -1), pFd->pShmId, 0, 0); - } - - if( p->isFullshm ){ - sqlite3OsShmBarrier(pFd->pReal); - return; - } -} - -static int tvfsShmUnmap( - sqlite3_file *pFile, - int deleteFlag -){ - int rc = SQLITE_OK; - TestvfsFd *pFd = tvfsGetFd(pFile); - Testvfs *p = (Testvfs *)(pFd->pVfs->pAppData); - TestvfsBuffer *pBuffer = pFd->pShm; - TestvfsFd **ppFd; - - if( p->isFullshm ){ - return sqlite3OsShmUnmap(pFd->pReal, deleteFlag); - } - - if( !pBuffer ) return SQLITE_OK; - assert( pFd->pShmId && pFd->pShm ); - - if( p->pScript && p->mask&TESTVFS_SHMCLOSE_MASK ){ - tvfsExecTcl(p, "xShmUnmap", - Tcl_NewStringObj(pFd->pShm->zFile, -1), pFd->pShmId, 0, 0 - ); - tvfsResultCode(p, &rc); - } - - for(ppFd=&pBuffer->pFile; *ppFd!=pFd; ppFd=&((*ppFd)->pNext)); - assert( (*ppFd)==pFd ); - *ppFd = pFd->pNext; - pFd->pNext = 0; - - if( pBuffer->pFile==0 ){ - int i; - TestvfsBuffer **pp; - for(pp=&p->pBuffer; *pp!=pBuffer; pp=&((*pp)->pNext)); - *pp = (*pp)->pNext; - for(i=0; pBuffer->aPage[i]; i++){ - ckfree((char *)pBuffer->aPage[i]); - } - ckfree((char *)pBuffer); - } - pFd->pShm = 0; - - return rc; -} - -static int tvfsFetch( - sqlite3_file *pFile, - sqlite3_int64 iOfst, - int iAmt, - void **pp -){ - TestvfsFd *pFd = tvfsGetFd(pFile); - return sqlite3OsFetch(pFd->pReal, iOfst, iAmt, pp); -} - -static int tvfsUnfetch(sqlite3_file *pFile, sqlite3_int64 iOfst, void *p){ - TestvfsFd *pFd = tvfsGetFd(pFile); - return sqlite3OsUnfetch(pFd->pReal, iOfst, p); -} - -static int testvfs_obj_cmd( - ClientData cd, - Tcl_Interp *interp, - int objc, - Tcl_Obj *CONST objv[] -){ - Testvfs *p = (Testvfs *)cd; - - enum DB_enum { - CMD_SHM, CMD_DELETE, CMD_FILTER, CMD_IOERR, CMD_SCRIPT, - CMD_DEVCHAR, CMD_SECTORSIZE, CMD_FULLERR, CMD_CANTOPENERR - }; - struct TestvfsSubcmd { - char *zName; - enum DB_enum eCmd; - } aSubcmd[] = { - { "shm", CMD_SHM }, - { "delete", CMD_DELETE }, - { "filter", CMD_FILTER }, - { "ioerr", CMD_IOERR }, - { "fullerr", CMD_FULLERR }, - { "cantopenerr", CMD_CANTOPENERR }, - { "script", CMD_SCRIPT }, - { "devchar", CMD_DEVCHAR }, - { "sectorsize", CMD_SECTORSIZE }, - { 0, 0 } - }; - int i; - - if( objc<2 ){ - Tcl_WrongNumArgs(interp, 1, objv, "SUBCOMMAND ..."); - return TCL_ERROR; - } - if( Tcl_GetIndexFromObjStruct( - interp, objv[1], aSubcmd, sizeof(aSubcmd[0]), "subcommand", 0, &i) - ){ - return TCL_ERROR; - } - Tcl_ResetResult(interp); - - switch( aSubcmd[i].eCmd ){ - case CMD_SHM: { - Tcl_Obj *pObj; - int rc; - TestvfsBuffer *pBuffer; - char *zName; - if( objc!=3 && objc!=4 ){ - Tcl_WrongNumArgs(interp, 2, objv, "FILE ?VALUE?"); - return TCL_ERROR; - } - zName = ckalloc(p->pParent->mxPathname); - rc = p->pParent->xFullPathname( - p->pParent, Tcl_GetString(objv[2]), - p->pParent->mxPathname, zName - ); - if( rc!=SQLITE_OK ){ - Tcl_AppendResult(interp, "failed to get full path: ", - Tcl_GetString(objv[2]), 0); - ckfree(zName); - return TCL_ERROR; - } - for(pBuffer=p->pBuffer; pBuffer; pBuffer=pBuffer->pNext){ - if( 0==strcmp(pBuffer->zFile, zName) ) break; - } - ckfree(zName); - if( !pBuffer ){ - Tcl_AppendResult(interp, "no such file: ", Tcl_GetString(objv[2]), 0); - return TCL_ERROR; - } - if( objc==4 ){ - int n; - u8 *a = Tcl_GetByteArrayFromObj(objv[3], &n); - int pgsz = pBuffer->pgsz; - if( pgsz==0 ) pgsz = 65536; - for(i=0; i*pgsz<n; i++){ - int nByte = pgsz; - tvfsAllocPage(pBuffer, i, pgsz); - if( n-i*pgsz<pgsz ){ - nByte = n; - } - memcpy(pBuffer->aPage[i], &a[i*pgsz], nByte); - } - } - - pObj = Tcl_NewObj(); - for(i=0; pBuffer->aPage[i]; i++){ - int pgsz = pBuffer->pgsz; - if( pgsz==0 ) pgsz = 65536; - Tcl_AppendObjToObj(pObj, Tcl_NewByteArrayObj(pBuffer->aPage[i], pgsz)); - } - Tcl_SetObjResult(interp, pObj); - break; - } - - /* TESTVFS filter METHOD-LIST - ** - ** Activate special processing for those methods contained in the list - */ - case CMD_FILTER: { - static struct VfsMethod { - char *zName; - int mask; - } vfsmethod [] = { - { "xShmOpen", TESTVFS_SHMOPEN_MASK }, - { "xShmLock", TESTVFS_SHMLOCK_MASK }, - { "xShmBarrier", TESTVFS_SHMBARRIER_MASK }, - { "xShmUnmap", TESTVFS_SHMCLOSE_MASK }, - { "xShmMap", TESTVFS_SHMMAP_MASK }, - { "xSync", TESTVFS_SYNC_MASK }, - { "xDelete", TESTVFS_DELETE_MASK }, - { "xWrite", TESTVFS_WRITE_MASK }, - { "xRead", TESTVFS_READ_MASK }, - { "xTruncate", TESTVFS_TRUNCATE_MASK }, - { "xOpen", TESTVFS_OPEN_MASK }, - { "xClose", TESTVFS_CLOSE_MASK }, - { "xAccess", TESTVFS_ACCESS_MASK }, - { "xFullPathname", TESTVFS_FULLPATHNAME_MASK }, - { "xUnlock", TESTVFS_UNLOCK_MASK }, - { "xLock", TESTVFS_LOCK_MASK }, - { "xCheckReservedLock", TESTVFS_CKLOCK_MASK }, - }; - Tcl_Obj **apElem = 0; - int nElem = 0; - int mask = 0; - if( objc!=3 ){ - Tcl_WrongNumArgs(interp, 2, objv, "LIST"); - return TCL_ERROR; - } - if( Tcl_ListObjGetElements(interp, objv[2], &nElem, &apElem) ){ - return TCL_ERROR; - } - Tcl_ResetResult(interp); - for(i=0; i<nElem; i++){ - int iMethod; - char *zElem = Tcl_GetString(apElem[i]); - for(iMethod=0; iMethod<ArraySize(vfsmethod); iMethod++){ - if( strcmp(zElem, vfsmethod[iMethod].zName)==0 ){ - mask |= vfsmethod[iMethod].mask; - break; - } - } - if( iMethod==ArraySize(vfsmethod) ){ - Tcl_AppendResult(interp, "unknown method: ", zElem, 0); - return TCL_ERROR; - } - } - p->mask = mask; - break; - } - - /* - ** TESTVFS script ?SCRIPT? - ** - ** Query or set the script to be run when filtered VFS events - ** occur. - */ - case CMD_SCRIPT: { - if( objc==3 ){ - int nByte; - if( p->pScript ){ - Tcl_DecrRefCount(p->pScript); - p->pScript = 0; - } - Tcl_GetStringFromObj(objv[2], &nByte); - if( nByte>0 ){ - p->pScript = Tcl_DuplicateObj(objv[2]); - Tcl_IncrRefCount(p->pScript); - } - }else if( objc!=2 ){ - Tcl_WrongNumArgs(interp, 2, objv, "?SCRIPT?"); - return TCL_ERROR; - } - - Tcl_ResetResult(interp); - if( p->pScript ) Tcl_SetObjResult(interp, p->pScript); - - break; - } - - /* - ** TESTVFS ioerr ?IFAIL PERSIST? - ** - ** Where IFAIL is an integer and PERSIST is boolean. - */ - case CMD_CANTOPENERR: - case CMD_IOERR: - case CMD_FULLERR: { - TestFaultInject *pTest = 0; - int iRet; - - switch( aSubcmd[i].eCmd ){ - case CMD_IOERR: pTest = &p->ioerr_err; break; - case CMD_FULLERR: pTest = &p->full_err; break; - case CMD_CANTOPENERR: pTest = &p->cantopen_err; break; - default: assert(0); - } - iRet = pTest->nFail; - pTest->nFail = 0; - pTest->eFault = 0; - pTest->iCnt = 0; - - if( objc==4 ){ - int iCnt, iPersist; - if( TCL_OK!=Tcl_GetIntFromObj(interp, objv[2], &iCnt) - || TCL_OK!=Tcl_GetBooleanFromObj(interp, objv[3], &iPersist) - ){ - return TCL_ERROR; - } - pTest->eFault = iPersist?FAULT_INJECT_PERSISTENT:FAULT_INJECT_TRANSIENT; - pTest->iCnt = iCnt; - }else if( objc!=2 ){ - Tcl_WrongNumArgs(interp, 2, objv, "?CNT PERSIST?"); - return TCL_ERROR; - } - Tcl_SetObjResult(interp, Tcl_NewIntObj(iRet)); - break; - } - - case CMD_DELETE: { - Tcl_DeleteCommand(interp, Tcl_GetString(objv[0])); - break; - } - - case CMD_DEVCHAR: { - struct DeviceFlag { - char *zName; - int iValue; - } aFlag[] = { - { "default", -1 }, - { "atomic", SQLITE_IOCAP_ATOMIC }, - { "atomic512", SQLITE_IOCAP_ATOMIC512 }, - { "atomic1k", SQLITE_IOCAP_ATOMIC1K }, - { "atomic2k", SQLITE_IOCAP_ATOMIC2K }, - { "atomic4k", SQLITE_IOCAP_ATOMIC4K }, - { "atomic8k", SQLITE_IOCAP_ATOMIC8K }, - { "atomic16k", SQLITE_IOCAP_ATOMIC16K }, - { "atomic32k", SQLITE_IOCAP_ATOMIC32K }, - { "atomic64k", SQLITE_IOCAP_ATOMIC64K }, - { "sequential", SQLITE_IOCAP_SEQUENTIAL }, - { "safe_append", SQLITE_IOCAP_SAFE_APPEND }, - { "undeletable_when_open", SQLITE_IOCAP_UNDELETABLE_WHEN_OPEN }, - { "powersafe_overwrite", SQLITE_IOCAP_POWERSAFE_OVERWRITE }, - { "immutable", SQLITE_IOCAP_IMMUTABLE }, - { 0, 0 } - }; - Tcl_Obj *pRet; - int iFlag; - - if( objc>3 ){ - Tcl_WrongNumArgs(interp, 2, objv, "?ATTR-LIST?"); - return TCL_ERROR; - } - if( objc==3 ){ - int j; - int iNew = 0; - Tcl_Obj **flags = 0; - int nFlags = 0; - - if( Tcl_ListObjGetElements(interp, objv[2], &nFlags, &flags) ){ - return TCL_ERROR; - } - - for(j=0; j<nFlags; j++){ - int idx = 0; - if( Tcl_GetIndexFromObjStruct(interp, flags[j], aFlag, - sizeof(aFlag[0]), "flag", 0, &idx) - ){ - return TCL_ERROR; - } - if( aFlag[idx].iValue<0 && nFlags>1 ){ - Tcl_AppendResult(interp, "bad flags: ", Tcl_GetString(objv[2]), 0); - return TCL_ERROR; - } - iNew |= aFlag[idx].iValue; - } - - p->iDevchar = iNew| 0x10000000; - } - - pRet = Tcl_NewObj(); - for(iFlag=0; iFlag<sizeof(aFlag)/sizeof(aFlag[0]); iFlag++){ - if( p->iDevchar & aFlag[iFlag].iValue ){ - Tcl_ListObjAppendElement( - interp, pRet, Tcl_NewStringObj(aFlag[iFlag].zName, -1) - ); - } - } - Tcl_SetObjResult(interp, pRet); - - break; - } - - case CMD_SECTORSIZE: { - if( objc>3 ){ - Tcl_WrongNumArgs(interp, 2, objv, "?VALUE?"); - return TCL_ERROR; - } - if( objc==3 ){ - int iNew = 0; - if( Tcl_GetIntFromObj(interp, objv[2], &iNew) ){ - return TCL_ERROR; - } - p->iSectorsize = iNew; - } - Tcl_SetObjResult(interp, Tcl_NewIntObj(p->iSectorsize)); - break; - } - } - - return TCL_OK; -} - -static void testvfs_obj_del(ClientData cd){ - Testvfs *p = (Testvfs *)cd; - if( p->pScript ) Tcl_DecrRefCount(p->pScript); - sqlite3_vfs_unregister(p->pVfs); - ckfree((char *)p->pVfs); - ckfree((char *)p); -} - -/* -** Usage: testvfs VFSNAME ?SWITCHES? -** -** Switches are: -** -** -noshm BOOLEAN (True to omit shm methods. Default false) -** -default BOOLEAN (True to make the vfs default. Default false) -** -** This command creates two things when it is invoked: an SQLite VFS, and -** a Tcl command. Both are named VFSNAME. The VFS is installed. It is not -** installed as the default VFS. -** -** The VFS passes all file I/O calls through to the underlying VFS. -** -** Whenever the xShmMap method of the VFS -** is invoked, the SCRIPT is executed as follows: -** -** SCRIPT xShmMap FILENAME ID -** -** The value returned by the invocation of SCRIPT above is interpreted as -** an SQLite error code and returned to SQLite. Either a symbolic -** "SQLITE_OK" or numeric "0" value may be returned. -** -** The contents of the shared-memory buffer associated with a given file -** may be read and set using the following command: -** -** VFSNAME shm FILENAME ?NEWVALUE? -** -** When the xShmLock method is invoked by SQLite, the following script is -** run: -** -** SCRIPT xShmLock FILENAME ID LOCK -** -** where LOCK is of the form "OFFSET NBYTE lock/unlock shared/exclusive" -*/ -static int testvfs_cmd( - ClientData cd, - Tcl_Interp *interp, - int objc, - Tcl_Obj *CONST objv[] -){ - static sqlite3_vfs tvfs_vfs = { - 3, /* iVersion */ - 0, /* szOsFile */ - 0, /* mxPathname */ - 0, /* pNext */ - 0, /* zName */ - 0, /* pAppData */ - tvfsOpen, /* xOpen */ - tvfsDelete, /* xDelete */ - tvfsAccess, /* xAccess */ - tvfsFullPathname, /* xFullPathname */ -#ifndef SQLITE_OMIT_LOAD_EXTENSION - tvfsDlOpen, /* xDlOpen */ - tvfsDlError, /* xDlError */ - tvfsDlSym, /* xDlSym */ - tvfsDlClose, /* xDlClose */ -#else - 0, /* xDlOpen */ - 0, /* xDlError */ - 0, /* xDlSym */ - 0, /* xDlClose */ -#endif /* SQLITE_OMIT_LOAD_EXTENSION */ - tvfsRandomness, /* xRandomness */ - tvfsSleep, /* xSleep */ - tvfsCurrentTime, /* xCurrentTime */ - 0, /* xGetLastError */ - 0, /* xCurrentTimeInt64 */ - 0, /* xSetSystemCall */ - 0, /* xGetSystemCall */ - 0, /* xNextSystemCall */ - }; - - Testvfs *p; /* New object */ - sqlite3_vfs *pVfs; /* New VFS */ - char *zVfs; - int nByte; /* Bytes of space to allocate at p */ - - int i; - int isNoshm = 0; /* True if -noshm is passed */ - int isFullshm = 0; /* True if -fullshm is passed */ - int isDefault = 0; /* True if -default is passed */ - int szOsFile = 0; /* Value passed to -szosfile */ - int mxPathname = -1; /* Value passed to -mxpathname */ - int iVersion = 3; /* Value passed to -iversion */ - - if( objc<2 || 0!=(objc%2) ) goto bad_args; - for(i=2; i<objc; i += 2){ - int nSwitch; - char *zSwitch; - zSwitch = Tcl_GetStringFromObj(objv[i], &nSwitch); - - if( nSwitch>2 && 0==strncmp("-noshm", zSwitch, nSwitch) ){ - if( Tcl_GetBooleanFromObj(interp, objv[i+1], &isNoshm) ){ - return TCL_ERROR; - } - if( isNoshm ) isFullshm = 0; - } - else if( nSwitch>2 && 0==strncmp("-default", zSwitch, nSwitch) ){ - if( Tcl_GetBooleanFromObj(interp, objv[i+1], &isDefault) ){ - return TCL_ERROR; - } - } - else if( nSwitch>2 && 0==strncmp("-szosfile", zSwitch, nSwitch) ){ - if( Tcl_GetIntFromObj(interp, objv[i+1], &szOsFile) ){ - return TCL_ERROR; - } - } - else if( nSwitch>2 && 0==strncmp("-mxpathname", zSwitch, nSwitch) ){ - if( Tcl_GetIntFromObj(interp, objv[i+1], &mxPathname) ){ - return TCL_ERROR; - } - } - else if( nSwitch>2 && 0==strncmp("-iversion", zSwitch, nSwitch) ){ - if( Tcl_GetIntFromObj(interp, objv[i+1], &iVersion) ){ - return TCL_ERROR; - } - } - else if( nSwitch>2 && 0==strncmp("-fullshm", zSwitch, nSwitch) ){ - if( Tcl_GetBooleanFromObj(interp, objv[i+1], &isFullshm) ){ - return TCL_ERROR; - } - if( isFullshm ) isNoshm = 0; - } - else{ - goto bad_args; - } - } - - if( szOsFile<sizeof(TestvfsFile) ){ - szOsFile = sizeof(TestvfsFile); - } - - zVfs = Tcl_GetString(objv[1]); - nByte = sizeof(Testvfs) + (int)strlen(zVfs)+1; - p = (Testvfs *)ckalloc(nByte); - memset(p, 0, nByte); - p->iDevchar = -1; - p->iSectorsize = -1; - - /* Create the new object command before querying SQLite for a default VFS - ** to use for 'real' IO operations. This is because creating the new VFS - ** may delete an existing [testvfs] VFS of the same name. If such a VFS - ** is currently the default, the new [testvfs] may end up calling the - ** methods of a deleted object. - */ - Tcl_CreateObjCommand(interp, zVfs, testvfs_obj_cmd, p, testvfs_obj_del); - p->pParent = sqlite3_vfs_find(0); - p->interp = interp; - - p->zName = (char *)&p[1]; - memcpy(p->zName, zVfs, strlen(zVfs)+1); - - pVfs = (sqlite3_vfs *)ckalloc(sizeof(sqlite3_vfs)); - memcpy(pVfs, &tvfs_vfs, sizeof(sqlite3_vfs)); - pVfs->pAppData = (void *)p; - pVfs->iVersion = iVersion; - pVfs->zName = p->zName; - pVfs->mxPathname = p->pParent->mxPathname; - if( mxPathname>=0 && mxPathname<pVfs->mxPathname ){ - pVfs->mxPathname = mxPathname; - } - pVfs->szOsFile = szOsFile; - p->pVfs = pVfs; - p->isNoshm = isNoshm; - p->isFullshm = isFullshm; - p->mask = TESTVFS_ALL_MASK; - - sqlite3_vfs_register(pVfs, isDefault); - - return TCL_OK; - - bad_args: - Tcl_WrongNumArgs(interp, 1, objv, "VFSNAME ?-noshm BOOL? ?-fullshm BOOL? ?-default BOOL? ?-mxpathname INT? ?-szosfile INT? ?-iversion INT?"); - return TCL_ERROR; -} - -int Sqlitetestvfs_Init(Tcl_Interp *interp){ - Tcl_CreateObjCommand(interp, "testvfs", testvfs_cmd, 0, 0); - return TCL_OK; -} - -#endif |