| Index: third_party/sqlite/sqlite-src-3100200/src/test_fs.c
|
| diff --git a/third_party/sqlite/sqlite-src-3100200/src/test_fs.c b/third_party/sqlite/sqlite-src-3100200/src/test_fs.c
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..45db0b53b8ae5dffd88d2b7e77bc6894c5a112de
|
| --- /dev/null
|
| +++ b/third_party/sqlite/sqlite-src-3100200/src/test_fs.c
|
| @@ -0,0 +1,916 @@
|
| +/*
|
| +** 2013 Jan 11
|
| +**
|
| +** 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.
|
| +**
|
| +*************************************************************************
|
| +** Code for testing the virtual table interfaces. This code
|
| +** is not included in the SQLite library. It is used for automated
|
| +** testing of the SQLite library.
|
| +**
|
| +** The FS virtual table is created as follows:
|
| +**
|
| +** CREATE VIRTUAL TABLE tbl USING fs(idx);
|
| +**
|
| +** where idx is the name of a table in the db with 2 columns. The virtual
|
| +** table also has two columns - file path and file contents.
|
| +**
|
| +** The first column of table idx must be an IPK, and the second contains file
|
| +** paths. For example:
|
| +**
|
| +** CREATE TABLE idx(id INTEGER PRIMARY KEY, path TEXT);
|
| +** INSERT INTO idx VALUES(4, '/etc/passwd');
|
| +**
|
| +** Adding the row to the idx table automatically creates a row in the
|
| +** virtual table with rowid=4, path=/etc/passwd and a text field that
|
| +** contains data read from file /etc/passwd on disk.
|
| +**
|
| +*************************************************************************
|
| +** Virtual table module "fsdir"
|
| +**
|
| +** This module is designed to be used as a read-only eponymous virtual table.
|
| +** Its schema is as follows:
|
| +**
|
| +** CREATE TABLE fsdir(dir TEXT, name TEXT);
|
| +**
|
| +** When queried, a WHERE term of the form "dir = $dir" must be provided. The
|
| +** virtual table then appears to have one row for each entry in file-system
|
| +** directory $dir. Column dir contains a copy of $dir, and column "name"
|
| +** contains the name of the directory entry.
|
| +**
|
| +** If the specified $dir cannot be opened or is not a directory, it is not
|
| +** an error. The virtual table appears to be empty in this case.
|
| +**
|
| +*************************************************************************
|
| +** Virtual table module "fstree"
|
| +**
|
| +** This module is also a read-only eponymous virtual table with the
|
| +** following schema:
|
| +**
|
| +** CREATE TABLE fstree(path TEXT, size INT, data BLOB);
|
| +**
|
| +** Running a "SELECT * FROM fstree" query on this table returns the entire
|
| +** contents of the file-system, starting at "/". To restrict the search
|
| +** space, the virtual table supports LIKE and GLOB constraints on the
|
| +** 'path' column. For example:
|
| +**
|
| +** SELECT * FROM fstree WHERE path LIKE '/home/dan/sqlite/%'
|
| +*/
|
| +#include "sqliteInt.h"
|
| +#include "tcl.h"
|
| +
|
| +#include <stdlib.h>
|
| +#include <string.h>
|
| +#include <sys/types.h>
|
| +#include <sys/stat.h>
|
| +#include <fcntl.h>
|
| +
|
| +#if SQLITE_OS_UNIX || defined(__MINGW_H)
|
| +# include <unistd.h>
|
| +# include <dirent.h>
|
| +# ifndef DIRENT
|
| +# define DIRENT dirent
|
| +# endif
|
| +#endif
|
| +#if SQLITE_OS_WIN
|
| +# include <io.h>
|
| +# if !defined(__MINGW_H)
|
| +# include "test_windirent.h"
|
| +# endif
|
| +# ifndef S_ISREG
|
| +# define S_ISREG(mode) (((mode) & S_IFMT) == S_IFREG)
|
| +# endif
|
| +#endif
|
| +
|
| +#ifndef SQLITE_OMIT_VIRTUALTABLE
|
| +
|
| +typedef struct fs_vtab fs_vtab;
|
| +typedef struct fs_cursor fs_cursor;
|
| +
|
| +/*
|
| +** A fs virtual-table object
|
| +*/
|
| +struct fs_vtab {
|
| + sqlite3_vtab base;
|
| + sqlite3 *db;
|
| + char *zDb; /* Name of db containing zTbl */
|
| + char *zTbl; /* Name of docid->file map table */
|
| +};
|
| +
|
| +/* A fs cursor object */
|
| +struct fs_cursor {
|
| + sqlite3_vtab_cursor base;
|
| + sqlite3_stmt *pStmt;
|
| + char *zBuf;
|
| + int nBuf;
|
| + int nAlloc;
|
| +};
|
| +
|
| +/*************************************************************************
|
| +** Start of fsdir implementation.
|
| +*/
|
| +typedef struct FsdirVtab FsdirVtab;
|
| +typedef struct FsdirCsr FsdirCsr;
|
| +struct FsdirVtab {
|
| + sqlite3_vtab base;
|
| +};
|
| +
|
| +struct FsdirCsr {
|
| + sqlite3_vtab_cursor base;
|
| + char *zDir; /* Buffer containing directory scanned */
|
| + DIR *pDir; /* Open directory */
|
| + sqlite3_int64 iRowid;
|
| + struct DIRENT entry; /* Current entry */
|
| +};
|
| +
|
| +/*
|
| +** This function is the implementation of both the xConnect and xCreate
|
| +** methods of the fsdir virtual table.
|
| +**
|
| +** The argv[] array contains the following:
|
| +**
|
| +** argv[0] -> module name ("fs")
|
| +** argv[1] -> database name
|
| +** argv[2] -> table name
|
| +** argv[...] -> other module argument fields.
|
| +*/
|
| +static int fsdirConnect(
|
| + sqlite3 *db,
|
| + void *pAux,
|
| + int argc, const char *const*argv,
|
| + sqlite3_vtab **ppVtab,
|
| + char **pzErr
|
| +){
|
| + FsdirVtab *pTab;
|
| +
|
| + if( argc!=3 ){
|
| + *pzErr = sqlite3_mprintf("wrong number of arguments");
|
| + return SQLITE_ERROR;
|
| + }
|
| +
|
| + pTab = (FsdirVtab *)sqlite3_malloc(sizeof(FsdirVtab));
|
| + if( !pTab ) return SQLITE_NOMEM;
|
| + memset(pTab, 0, sizeof(FsdirVtab));
|
| +
|
| + *ppVtab = &pTab->base;
|
| + sqlite3_declare_vtab(db, "CREATE TABLE xyz(dir, name);");
|
| +
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** xDestroy/xDisconnect implementation.
|
| +*/
|
| +static int fsdirDisconnect(sqlite3_vtab *pVtab){
|
| + sqlite3_free(pVtab);
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** xBestIndex implementation. The only constraint supported is:
|
| +**
|
| +** (dir = ?)
|
| +*/
|
| +static int fsdirBestIndex(sqlite3_vtab *tab, sqlite3_index_info *pIdxInfo){
|
| + int ii;
|
| +
|
| + pIdxInfo->estimatedCost = 1000000000.0;
|
| +
|
| + for(ii=0; ii<pIdxInfo->nConstraint; ii++){
|
| + struct sqlite3_index_constraint const *p = &pIdxInfo->aConstraint[ii];
|
| + if( p->iColumn==0 && p->usable && p->op==SQLITE_INDEX_CONSTRAINT_EQ ){
|
| + struct sqlite3_index_constraint_usage *pUsage;
|
| + pUsage = &pIdxInfo->aConstraintUsage[ii];
|
| + pUsage->omit = 1;
|
| + pUsage->argvIndex = 1;
|
| + pIdxInfo->idxNum = 1;
|
| + pIdxInfo->estimatedCost = 1.0;
|
| + break;
|
| + }
|
| + }
|
| +
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** xOpen implementation.
|
| +**
|
| +** Open a new fsdir cursor.
|
| +*/
|
| +static int fsdirOpen(sqlite3_vtab *pVTab, sqlite3_vtab_cursor **ppCursor){
|
| + FsdirCsr *pCur;
|
| + /* Allocate an extra 256 bytes because it is undefined how big dirent.d_name
|
| + ** is and we need enough space. Linux provides plenty already, but
|
| + ** Solaris only provides one byte. */
|
| + pCur = (FsdirCsr*)sqlite3_malloc(sizeof(FsdirCsr)+256);
|
| + if( pCur==0 ) return SQLITE_NOMEM;
|
| + memset(pCur, 0, sizeof(FsdirCsr));
|
| + *ppCursor = &pCur->base;
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** Close a fsdir cursor.
|
| +*/
|
| +static int fsdirClose(sqlite3_vtab_cursor *cur){
|
| + FsdirCsr *pCur = (FsdirCsr*)cur;
|
| + if( pCur->pDir ) closedir(pCur->pDir);
|
| + sqlite3_free(pCur->zDir);
|
| + sqlite3_free(pCur);
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** Skip the cursor to the next entry.
|
| +*/
|
| +static int fsdirNext(sqlite3_vtab_cursor *cur){
|
| + FsdirCsr *pCsr = (FsdirCsr*)cur;
|
| +
|
| + if( pCsr->pDir ){
|
| + struct DIRENT *pRes = 0;
|
| +#if defined(__MINGW_H)
|
| + pRes = readdir(pCsr->pDir);
|
| + if( pRes!=0 ){
|
| + memcpy(&pCsr->entry, pRes, sizeof(struct DIRENT));
|
| + }
|
| +#else
|
| + readdir_r(pCsr->pDir, &pCsr->entry, &pRes);
|
| +#endif
|
| + if( pRes==0 ){
|
| + closedir(pCsr->pDir);
|
| + pCsr->pDir = 0;
|
| + }
|
| + pCsr->iRowid++;
|
| + }
|
| +
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** xFilter method implementation.
|
| +*/
|
| +static int fsdirFilter(
|
| + sqlite3_vtab_cursor *pVtabCursor,
|
| + int idxNum, const char *idxStr,
|
| + int argc, sqlite3_value **argv
|
| +){
|
| + FsdirCsr *pCsr = (FsdirCsr*)pVtabCursor;
|
| + const char *zDir;
|
| + int nDir;
|
| +
|
| +
|
| + if( idxNum!=1 || argc!=1 ){
|
| + return SQLITE_ERROR;
|
| + }
|
| +
|
| + pCsr->iRowid = 0;
|
| + sqlite3_free(pCsr->zDir);
|
| + if( pCsr->pDir ){
|
| + closedir(pCsr->pDir);
|
| + pCsr->pDir = 0;
|
| + }
|
| +
|
| + zDir = (const char*)sqlite3_value_text(argv[0]);
|
| + nDir = sqlite3_value_bytes(argv[0]);
|
| + pCsr->zDir = sqlite3_malloc(nDir+1);
|
| + if( pCsr->zDir==0 ) return SQLITE_NOMEM;
|
| + memcpy(pCsr->zDir, zDir, nDir+1);
|
| +
|
| + pCsr->pDir = opendir(pCsr->zDir);
|
| + return fsdirNext(pVtabCursor);
|
| +}
|
| +
|
| +/*
|
| +** xEof method implementation.
|
| +*/
|
| +static int fsdirEof(sqlite3_vtab_cursor *cur){
|
| + FsdirCsr *pCsr = (FsdirCsr*)cur;
|
| + return pCsr->pDir==0;
|
| +}
|
| +
|
| +/*
|
| +** xColumn method implementation.
|
| +*/
|
| +static int fsdirColumn(sqlite3_vtab_cursor *cur, sqlite3_context *ctx, int i){
|
| + FsdirCsr *pCsr = (FsdirCsr*)cur;
|
| + switch( i ){
|
| + case 0: /* dir */
|
| + sqlite3_result_text(ctx, pCsr->zDir, -1, SQLITE_STATIC);
|
| + break;
|
| +
|
| + case 1: /* name */
|
| + sqlite3_result_text(ctx, pCsr->entry.d_name, -1, SQLITE_TRANSIENT);
|
| + break;
|
| +
|
| + default:
|
| + assert( 0 );
|
| + }
|
| +
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** xRowid method implementation.
|
| +*/
|
| +static int fsdirRowid(sqlite3_vtab_cursor *cur, sqlite_int64 *pRowid){
|
| + FsdirCsr *pCsr = (FsdirCsr*)cur;
|
| + *pRowid = pCsr->iRowid;
|
| + return SQLITE_OK;
|
| +}
|
| +/*
|
| +** End of fsdir implementation.
|
| +*************************************************************************/
|
| +
|
| +/*************************************************************************
|
| +** Start of fstree implementation.
|
| +*/
|
| +typedef struct FstreeVtab FstreeVtab;
|
| +typedef struct FstreeCsr FstreeCsr;
|
| +struct FstreeVtab {
|
| + sqlite3_vtab base;
|
| + sqlite3 *db;
|
| +};
|
| +
|
| +struct FstreeCsr {
|
| + sqlite3_vtab_cursor base;
|
| + sqlite3_stmt *pStmt; /* Statement to list paths */
|
| + int fd; /* File descriptor open on current path */
|
| +};
|
| +
|
| +/*
|
| +** This function is the implementation of both the xConnect and xCreate
|
| +** methods of the fstree virtual table.
|
| +**
|
| +** The argv[] array contains the following:
|
| +**
|
| +** argv[0] -> module name ("fs")
|
| +** argv[1] -> database name
|
| +** argv[2] -> table name
|
| +** argv[...] -> other module argument fields.
|
| +*/
|
| +static int fstreeConnect(
|
| + sqlite3 *db,
|
| + void *pAux,
|
| + int argc, const char *const*argv,
|
| + sqlite3_vtab **ppVtab,
|
| + char **pzErr
|
| +){
|
| + FstreeVtab *pTab;
|
| +
|
| + if( argc!=3 ){
|
| + *pzErr = sqlite3_mprintf("wrong number of arguments");
|
| + return SQLITE_ERROR;
|
| + }
|
| +
|
| + pTab = (FstreeVtab *)sqlite3_malloc(sizeof(FstreeVtab));
|
| + if( !pTab ) return SQLITE_NOMEM;
|
| + memset(pTab, 0, sizeof(FstreeVtab));
|
| + pTab->db = db;
|
| +
|
| + *ppVtab = &pTab->base;
|
| + sqlite3_declare_vtab(db, "CREATE TABLE xyz(path, size, data);");
|
| +
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** xDestroy/xDisconnect implementation.
|
| +*/
|
| +static int fstreeDisconnect(sqlite3_vtab *pVtab){
|
| + sqlite3_free(pVtab);
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** xBestIndex implementation. The only constraint supported is:
|
| +**
|
| +** (dir = ?)
|
| +*/
|
| +static int fstreeBestIndex(sqlite3_vtab *tab, sqlite3_index_info *pIdxInfo){
|
| + int ii;
|
| +
|
| + for(ii=0; ii<pIdxInfo->nConstraint; ii++){
|
| + struct sqlite3_index_constraint const *p = &pIdxInfo->aConstraint[ii];
|
| + if( p->iColumn==0 && p->usable && (
|
| + p->op==SQLITE_INDEX_CONSTRAINT_GLOB
|
| + || p->op==SQLITE_INDEX_CONSTRAINT_LIKE
|
| + || p->op==SQLITE_INDEX_CONSTRAINT_EQ
|
| + )){
|
| + struct sqlite3_index_constraint_usage *pUsage;
|
| + pUsage = &pIdxInfo->aConstraintUsage[ii];
|
| + pIdxInfo->idxNum = p->op;
|
| + pUsage->argvIndex = 1;
|
| + pIdxInfo->estimatedCost = 100000.0;
|
| + return SQLITE_OK;
|
| + }
|
| + }
|
| +
|
| + pIdxInfo->estimatedCost = 1000000000.0;
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** xOpen implementation.
|
| +**
|
| +** Open a new fstree cursor.
|
| +*/
|
| +static int fstreeOpen(sqlite3_vtab *pVTab, sqlite3_vtab_cursor **ppCursor){
|
| + FstreeCsr *pCur;
|
| + pCur = (FstreeCsr*)sqlite3_malloc(sizeof(FstreeCsr));
|
| + if( pCur==0 ) return SQLITE_NOMEM;
|
| + memset(pCur, 0, sizeof(FstreeCsr));
|
| + pCur->fd = -1;
|
| + *ppCursor = &pCur->base;
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +static void fstreeCloseFd(FstreeCsr *pCsr){
|
| + if( pCsr->fd>=0 ){
|
| + close(pCsr->fd);
|
| + pCsr->fd = -1;
|
| + }
|
| +}
|
| +
|
| +/*
|
| +** Close a fstree cursor.
|
| +*/
|
| +static int fstreeClose(sqlite3_vtab_cursor *cur){
|
| + FstreeCsr *pCsr = (FstreeCsr*)cur;
|
| + sqlite3_finalize(pCsr->pStmt);
|
| + fstreeCloseFd(pCsr);
|
| + sqlite3_free(pCsr);
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** Skip the cursor to the next entry.
|
| +*/
|
| +static int fstreeNext(sqlite3_vtab_cursor *cur){
|
| + FstreeCsr *pCsr = (FstreeCsr*)cur;
|
| + int rc;
|
| +
|
| + fstreeCloseFd(pCsr);
|
| + rc = sqlite3_step(pCsr->pStmt);
|
| + if( rc!=SQLITE_ROW ){
|
| + rc = sqlite3_finalize(pCsr->pStmt);
|
| + pCsr->pStmt = 0;
|
| + }else{
|
| + rc = SQLITE_OK;
|
| + pCsr->fd = open((const char*)sqlite3_column_text(pCsr->pStmt, 0), O_RDONLY);
|
| + }
|
| +
|
| + return rc;
|
| +}
|
| +
|
| +/*
|
| +** xFilter method implementation.
|
| +*/
|
| +static int fstreeFilter(
|
| + sqlite3_vtab_cursor *pVtabCursor,
|
| + int idxNum, const char *idxStr,
|
| + int argc, sqlite3_value **argv
|
| +){
|
| + FstreeCsr *pCsr = (FstreeCsr*)pVtabCursor;
|
| + FstreeVtab *pTab = (FstreeVtab*)(pCsr->base.pVtab);
|
| + int rc;
|
| + const char *zSql =
|
| +"WITH r(d) AS ("
|
| +" SELECT CASE WHEN dir=?2 THEN ?3 ELSE dir END || '/' || name "
|
| +" FROM fsdir WHERE dir=?1 AND name NOT LIKE '.%'"
|
| +" UNION ALL"
|
| +" SELECT dir || '/' || name FROM r, fsdir WHERE dir=d AND name NOT LIKE '.%'"
|
| +") SELECT d FROM r;";
|
| +
|
| + char *zRoot;
|
| + int nRoot;
|
| + char *zPrefix;
|
| + int nPrefix;
|
| + const char *zDir;
|
| + int nDir;
|
| + char aWild[2] = { '\0', '\0' };
|
| +
|
| +#if SQLITE_OS_WIN
|
| + zRoot = sqlite3_mprintf("%s%c", getenv("SystemDrive"), '/');
|
| + nRoot = strlen(zRoot);
|
| + zPrefix = sqlite3_mprintf("%s", getenv("SystemDrive"));
|
| + nPrefix = strlen(zPrefix);
|
| +#else
|
| + zRoot = "/";
|
| + nRoot = 1;
|
| + zPrefix = "";
|
| + nPrefix = 0;
|
| +#endif
|
| +
|
| + zDir = zRoot;
|
| + nDir = nRoot;
|
| +
|
| + fstreeCloseFd(pCsr);
|
| + sqlite3_finalize(pCsr->pStmt);
|
| + pCsr->pStmt = 0;
|
| + rc = sqlite3_prepare_v2(pTab->db, zSql, -1, &pCsr->pStmt, 0);
|
| + if( rc!=SQLITE_OK ) return rc;
|
| +
|
| + if( idxNum ){
|
| + const char *zQuery = (const char*)sqlite3_value_text(argv[0]);
|
| + switch( idxNum ){
|
| + case SQLITE_INDEX_CONSTRAINT_GLOB:
|
| + aWild[0] = '*';
|
| + aWild[1] = '?';
|
| + break;
|
| + case SQLITE_INDEX_CONSTRAINT_LIKE:
|
| + aWild[0] = '_';
|
| + aWild[1] = '%';
|
| + break;
|
| + }
|
| +
|
| + if( sqlite3_strnicmp(zQuery, zPrefix, nPrefix)==0 ){
|
| + int i;
|
| + for(i=nPrefix; zQuery[i]; i++){
|
| + if( zQuery[i]==aWild[0] || zQuery[i]==aWild[1] ) break;
|
| + if( zQuery[i]=='/' ) nDir = i;
|
| + }
|
| + zDir = zQuery;
|
| + }
|
| + }
|
| +
|
| + sqlite3_bind_text(pCsr->pStmt, 1, zDir, nDir, SQLITE_TRANSIENT);
|
| + sqlite3_bind_text(pCsr->pStmt, 2, zRoot, nRoot, SQLITE_TRANSIENT);
|
| + sqlite3_bind_text(pCsr->pStmt, 3, zPrefix, nPrefix, SQLITE_TRANSIENT);
|
| +
|
| +#if SQLITE_OS_WIN
|
| + sqlite3_free(zPrefix);
|
| + sqlite3_free(zRoot);
|
| +#endif
|
| +
|
| + return fstreeNext(pVtabCursor);
|
| +}
|
| +
|
| +/*
|
| +** xEof method implementation.
|
| +*/
|
| +static int fstreeEof(sqlite3_vtab_cursor *cur){
|
| + FstreeCsr *pCsr = (FstreeCsr*)cur;
|
| + return pCsr->pStmt==0;
|
| +}
|
| +
|
| +/*
|
| +** xColumn method implementation.
|
| +*/
|
| +static int fstreeColumn(sqlite3_vtab_cursor *cur, sqlite3_context *ctx, int i){
|
| + FstreeCsr *pCsr = (FstreeCsr*)cur;
|
| + if( i==0 ){ /* path */
|
| + sqlite3_result_value(ctx, sqlite3_column_value(pCsr->pStmt, 0));
|
| + }else{
|
| + struct stat sBuf;
|
| + fstat(pCsr->fd, &sBuf);
|
| +
|
| + if( S_ISREG(sBuf.st_mode) ){
|
| + if( i==1 ){
|
| + sqlite3_result_int64(ctx, sBuf.st_size);
|
| + }else{
|
| + int nRead;
|
| + char *aBuf = sqlite3_malloc(sBuf.st_mode+1);
|
| + if( !aBuf ) return SQLITE_NOMEM;
|
| + nRead = read(pCsr->fd, aBuf, sBuf.st_mode);
|
| + if( nRead!=sBuf.st_mode ){
|
| + return SQLITE_IOERR;
|
| + }
|
| + sqlite3_result_blob(ctx, aBuf, nRead, SQLITE_TRANSIENT);
|
| + sqlite3_free(aBuf);
|
| + }
|
| + }
|
| + }
|
| +
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** xRowid method implementation.
|
| +*/
|
| +static int fstreeRowid(sqlite3_vtab_cursor *cur, sqlite_int64 *pRowid){
|
| + *pRowid = 0;
|
| + return SQLITE_OK;
|
| +}
|
| +/*
|
| +** End of fstree implementation.
|
| +*************************************************************************/
|
| +
|
| +
|
| +
|
| +
|
| +/*
|
| +** This function is the implementation of both the xConnect and xCreate
|
| +** methods of the fs virtual table.
|
| +**
|
| +** The argv[] array contains the following:
|
| +**
|
| +** argv[0] -> module name ("fs")
|
| +** argv[1] -> database name
|
| +** argv[2] -> table name
|
| +** argv[...] -> other module argument fields.
|
| +*/
|
| +static int fsConnect(
|
| + sqlite3 *db,
|
| + void *pAux,
|
| + int argc, const char *const*argv,
|
| + sqlite3_vtab **ppVtab,
|
| + char **pzErr
|
| +){
|
| + fs_vtab *pVtab;
|
| + int nByte;
|
| + const char *zTbl;
|
| + const char *zDb = argv[1];
|
| +
|
| + if( argc!=4 ){
|
| + *pzErr = sqlite3_mprintf("wrong number of arguments");
|
| + return SQLITE_ERROR;
|
| + }
|
| + zTbl = argv[3];
|
| +
|
| + nByte = sizeof(fs_vtab) + (int)strlen(zTbl) + 1 + (int)strlen(zDb) + 1;
|
| + pVtab = (fs_vtab *)sqlite3MallocZero( nByte );
|
| + if( !pVtab ) return SQLITE_NOMEM;
|
| +
|
| + pVtab->zTbl = (char *)&pVtab[1];
|
| + pVtab->zDb = &pVtab->zTbl[strlen(zTbl)+1];
|
| + pVtab->db = db;
|
| + memcpy(pVtab->zTbl, zTbl, strlen(zTbl));
|
| + memcpy(pVtab->zDb, zDb, strlen(zDb));
|
| + *ppVtab = &pVtab->base;
|
| + sqlite3_declare_vtab(db, "CREATE TABLE x(path TEXT, data TEXT)");
|
| +
|
| + return SQLITE_OK;
|
| +}
|
| +/* Note that for this virtual table, the xCreate and xConnect
|
| +** methods are identical. */
|
| +
|
| +static int fsDisconnect(sqlite3_vtab *pVtab){
|
| + sqlite3_free(pVtab);
|
| + return SQLITE_OK;
|
| +}
|
| +/* The xDisconnect and xDestroy methods are also the same */
|
| +
|
| +/*
|
| +** Open a new fs cursor.
|
| +*/
|
| +static int fsOpen(sqlite3_vtab *pVTab, sqlite3_vtab_cursor **ppCursor){
|
| + fs_cursor *pCur;
|
| + pCur = sqlite3MallocZero(sizeof(fs_cursor));
|
| + *ppCursor = &pCur->base;
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** Close a fs cursor.
|
| +*/
|
| +static int fsClose(sqlite3_vtab_cursor *cur){
|
| + fs_cursor *pCur = (fs_cursor *)cur;
|
| + sqlite3_finalize(pCur->pStmt);
|
| + sqlite3_free(pCur->zBuf);
|
| + sqlite3_free(pCur);
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +static int fsNext(sqlite3_vtab_cursor *cur){
|
| + fs_cursor *pCur = (fs_cursor *)cur;
|
| + int rc;
|
| +
|
| + rc = sqlite3_step(pCur->pStmt);
|
| + if( rc==SQLITE_ROW || rc==SQLITE_DONE ) rc = SQLITE_OK;
|
| +
|
| + return rc;
|
| +}
|
| +
|
| +static int fsFilter(
|
| + sqlite3_vtab_cursor *pVtabCursor,
|
| + int idxNum, const char *idxStr,
|
| + int argc, sqlite3_value **argv
|
| +){
|
| + int rc;
|
| + fs_cursor *pCur = (fs_cursor *)pVtabCursor;
|
| + fs_vtab *p = (fs_vtab *)(pVtabCursor->pVtab);
|
| +
|
| + assert( (idxNum==0 && argc==0) || (idxNum==1 && argc==1) );
|
| + if( idxNum==1 ){
|
| + char *zStmt = sqlite3_mprintf(
|
| + "SELECT * FROM %Q.%Q WHERE rowid=?", p->zDb, p->zTbl);
|
| + if( !zStmt ) return SQLITE_NOMEM;
|
| + rc = sqlite3_prepare_v2(p->db, zStmt, -1, &pCur->pStmt, 0);
|
| + sqlite3_free(zStmt);
|
| + if( rc==SQLITE_OK ){
|
| + sqlite3_bind_value(pCur->pStmt, 1, argv[0]);
|
| + }
|
| + }else{
|
| + char *zStmt = sqlite3_mprintf("SELECT * FROM %Q.%Q", p->zDb, p->zTbl);
|
| + if( !zStmt ) return SQLITE_NOMEM;
|
| + rc = sqlite3_prepare_v2(p->db, zStmt, -1, &pCur->pStmt, 0);
|
| + sqlite3_free(zStmt);
|
| + }
|
| +
|
| + if( rc==SQLITE_OK ){
|
| + rc = fsNext(pVtabCursor);
|
| + }
|
| + return rc;
|
| +}
|
| +
|
| +static int fsColumn(sqlite3_vtab_cursor *cur, sqlite3_context *ctx, int i){
|
| + fs_cursor *pCur = (fs_cursor*)cur;
|
| +
|
| + assert( i==0 || i==1 || i==2 );
|
| + if( i==0 ){
|
| + sqlite3_result_value(ctx, sqlite3_column_value(pCur->pStmt, 0));
|
| + }else{
|
| + const char *zFile = (const char *)sqlite3_column_text(pCur->pStmt, 1);
|
| + struct stat sbuf;
|
| + int fd;
|
| +
|
| + int n;
|
| + fd = open(zFile, O_RDONLY);
|
| + if( fd<0 ) return SQLITE_IOERR;
|
| + fstat(fd, &sbuf);
|
| +
|
| + if( sbuf.st_size>=pCur->nAlloc ){
|
| + int nNew = sbuf.st_size*2;
|
| + char *zNew;
|
| + if( nNew<1024 ) nNew = 1024;
|
| +
|
| + zNew = sqlite3Realloc(pCur->zBuf, nNew);
|
| + if( zNew==0 ){
|
| + close(fd);
|
| + return SQLITE_NOMEM;
|
| + }
|
| + pCur->zBuf = zNew;
|
| + pCur->nAlloc = nNew;
|
| + }
|
| +
|
| + n = (int)read(fd, pCur->zBuf, sbuf.st_size);
|
| + close(fd);
|
| + if( n!=sbuf.st_size ) return SQLITE_ERROR;
|
| + pCur->nBuf = sbuf.st_size;
|
| + pCur->zBuf[pCur->nBuf] = '\0';
|
| +
|
| + sqlite3_result_text(ctx, pCur->zBuf, -1, SQLITE_TRANSIENT);
|
| + }
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +static int fsRowid(sqlite3_vtab_cursor *cur, sqlite_int64 *pRowid){
|
| + fs_cursor *pCur = (fs_cursor*)cur;
|
| + *pRowid = sqlite3_column_int64(pCur->pStmt, 0);
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +static int fsEof(sqlite3_vtab_cursor *cur){
|
| + fs_cursor *pCur = (fs_cursor*)cur;
|
| + return (sqlite3_data_count(pCur->pStmt)==0);
|
| +}
|
| +
|
| +static int fsBestIndex(sqlite3_vtab *tab, sqlite3_index_info *pIdxInfo){
|
| + int ii;
|
| +
|
| + for(ii=0; ii<pIdxInfo->nConstraint; ii++){
|
| + struct sqlite3_index_constraint const *pCons = &pIdxInfo->aConstraint[ii];
|
| + if( pCons->iColumn<0 && pCons->usable
|
| + && pCons->op==SQLITE_INDEX_CONSTRAINT_EQ ){
|
| + struct sqlite3_index_constraint_usage *pUsage;
|
| + pUsage = &pIdxInfo->aConstraintUsage[ii];
|
| + pUsage->omit = 0;
|
| + pUsage->argvIndex = 1;
|
| + pIdxInfo->idxNum = 1;
|
| + pIdxInfo->estimatedCost = 1.0;
|
| + break;
|
| + }
|
| + }
|
| +
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** A virtual table module that provides read-only access to a
|
| +** Tcl global variable namespace.
|
| +*/
|
| +static sqlite3_module fsModule = {
|
| + 0, /* iVersion */
|
| + fsConnect,
|
| + fsConnect,
|
| + fsBestIndex,
|
| + fsDisconnect,
|
| + fsDisconnect,
|
| + fsOpen, /* xOpen - open a cursor */
|
| + fsClose, /* xClose - close a cursor */
|
| + fsFilter, /* xFilter - configure scan constraints */
|
| + fsNext, /* xNext - advance a cursor */
|
| + fsEof, /* xEof - check for end of scan */
|
| + fsColumn, /* xColumn - read data */
|
| + fsRowid, /* xRowid - read data */
|
| + 0, /* xUpdate */
|
| + 0, /* xBegin */
|
| + 0, /* xSync */
|
| + 0, /* xCommit */
|
| + 0, /* xRollback */
|
| + 0, /* xFindMethod */
|
| + 0, /* xRename */
|
| +};
|
| +
|
| +static sqlite3_module fsdirModule = {
|
| + 0, /* iVersion */
|
| + fsdirConnect, /* xCreate */
|
| + fsdirConnect, /* xConnect */
|
| + fsdirBestIndex, /* xBestIndex */
|
| + fsdirDisconnect, /* xDisconnect */
|
| + fsdirDisconnect, /* xDestroy */
|
| + fsdirOpen, /* xOpen - open a cursor */
|
| + fsdirClose, /* xClose - close a cursor */
|
| + fsdirFilter, /* xFilter - configure scan constraints */
|
| + fsdirNext, /* xNext - advance a cursor */
|
| + fsdirEof, /* xEof - check for end of scan */
|
| + fsdirColumn, /* xColumn - read data */
|
| + fsdirRowid, /* xRowid - read data */
|
| + 0, /* xUpdate */
|
| + 0, /* xBegin */
|
| + 0, /* xSync */
|
| + 0, /* xCommit */
|
| + 0, /* xRollback */
|
| + 0, /* xFindMethod */
|
| + 0, /* xRename */
|
| +};
|
| +
|
| +static sqlite3_module fstreeModule = {
|
| + 0, /* iVersion */
|
| + fstreeConnect, /* xCreate */
|
| + fstreeConnect, /* xConnect */
|
| + fstreeBestIndex, /* xBestIndex */
|
| + fstreeDisconnect, /* xDisconnect */
|
| + fstreeDisconnect, /* xDestroy */
|
| + fstreeOpen, /* xOpen - open a cursor */
|
| + fstreeClose, /* xClose - close a cursor */
|
| + fstreeFilter, /* xFilter - configure scan constraints */
|
| + fstreeNext, /* xNext - advance a cursor */
|
| + fstreeEof, /* xEof - check for end of scan */
|
| + fstreeColumn, /* xColumn - read data */
|
| + fstreeRowid, /* xRowid - read data */
|
| + 0, /* xUpdate */
|
| + 0, /* xBegin */
|
| + 0, /* xSync */
|
| + 0, /* xCommit */
|
| + 0, /* xRollback */
|
| + 0, /* xFindMethod */
|
| + 0, /* xRename */
|
| +};
|
| +
|
| +/*
|
| +** Decode a pointer to an sqlite3 object.
|
| +*/
|
| +extern int getDbPointer(Tcl_Interp *interp, const char *zA, sqlite3 **ppDb);
|
| +
|
| +/*
|
| +** Register the echo virtual table module.
|
| +*/
|
| +static int register_fs_module(
|
| + ClientData clientData, /* Pointer to sqlite3_enable_XXX function */
|
| + Tcl_Interp *interp, /* The TCL interpreter that invoked this command */
|
| + int objc, /* Number of arguments */
|
| + Tcl_Obj *CONST objv[] /* Command arguments */
|
| +){
|
| + sqlite3 *db;
|
| + if( objc!=2 ){
|
| + Tcl_WrongNumArgs(interp, 1, objv, "DB");
|
| + return TCL_ERROR;
|
| + }
|
| + if( getDbPointer(interp, Tcl_GetString(objv[1]), &db) ) return TCL_ERROR;
|
| +#ifndef SQLITE_OMIT_VIRTUALTABLE
|
| + sqlite3_create_module(db, "fs", &fsModule, (void *)interp);
|
| + sqlite3_create_module(db, "fsdir", &fsdirModule, 0);
|
| + sqlite3_create_module(db, "fstree", &fstreeModule, 0);
|
| +#endif
|
| + return TCL_OK;
|
| +}
|
| +
|
| +#endif
|
| +
|
| +
|
| +/*
|
| +** Register commands with the TCL interpreter.
|
| +*/
|
| +int Sqlitetestfs_Init(Tcl_Interp *interp){
|
| +#ifndef SQLITE_OMIT_VIRTUALTABLE
|
| + static struct {
|
| + char *zName;
|
| + Tcl_ObjCmdProc *xProc;
|
| + void *clientData;
|
| + } aObjCmd[] = {
|
| + { "register_fs_module", register_fs_module, 0 },
|
| + };
|
| + int i;
|
| + for(i=0; i<sizeof(aObjCmd)/sizeof(aObjCmd[0]); i++){
|
| + Tcl_CreateObjCommand(interp, aObjCmd[i].zName,
|
| + aObjCmd[i].xProc, aObjCmd[i].clientData, 0);
|
| + }
|
| +#endif
|
| + return TCL_OK;
|
| +}
|
|
|