Index: third_party/sqlite/src/src/test_intarray.c |
diff --git a/third_party/sqlite/src/src/test_intarray.c b/third_party/sqlite/src/src/test_intarray.c |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8651d01cfacb012cc49d50a23adf5dbf8b3250bf |
--- /dev/null |
+++ b/third_party/sqlite/src/src/test_intarray.c |
@@ -0,0 +1,381 @@ |
+/* |
+** 2009 November 10 |
+** |
+** 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 implements a read-only VIRTUAL TABLE that contains the |
+** content of a C-language array of integer values. See the corresponding |
+** header file for full details. |
+*/ |
+#include "test_intarray.h" |
+#include <string.h> |
+#include <assert.h> |
+ |
+ |
+/* |
+** Definition of the sqlite3_intarray object. |
+** |
+** The internal representation of an intarray object is subject |
+** to change, is not externally visible, and should be used by |
+** the implementation of intarray only. This object is opaque |
+** to users. |
+*/ |
+struct sqlite3_intarray { |
+ int n; /* Number of elements in the array */ |
+ sqlite3_int64 *a; /* Contents of the array */ |
+ void (*xFree)(void*); /* Function used to free a[] */ |
+}; |
+ |
+/* Objects used internally by the virtual table implementation */ |
+typedef struct intarray_vtab intarray_vtab; |
+typedef struct intarray_cursor intarray_cursor; |
+ |
+/* A intarray table object */ |
+struct intarray_vtab { |
+ sqlite3_vtab base; /* Base class */ |
+ sqlite3_intarray *pContent; /* Content of the integer array */ |
+}; |
+ |
+/* A intarray cursor object */ |
+struct intarray_cursor { |
+ sqlite3_vtab_cursor base; /* Base class */ |
+ int i; /* Current cursor position */ |
+}; |
+ |
+/* |
+** None of this works unless we have virtual tables. |
+*/ |
+#ifndef SQLITE_OMIT_VIRTUALTABLE |
+ |
+/* |
+** Free an sqlite3_intarray object. |
+*/ |
+static void intarrayFree(sqlite3_intarray *p){ |
+ if( p->xFree ){ |
+ p->xFree(p->a); |
+ } |
+ sqlite3_free(p); |
+} |
+ |
+/* |
+** Table destructor for the intarray module. |
+*/ |
+static int intarrayDestroy(sqlite3_vtab *p){ |
+ intarray_vtab *pVtab = (intarray_vtab*)p; |
+ sqlite3_free(pVtab); |
+ return 0; |
+} |
+ |
+/* |
+** Table constructor for the intarray module. |
+*/ |
+static int intarrayCreate( |
+ sqlite3 *db, /* Database where module is created */ |
+ void *pAux, /* clientdata for the module */ |
+ int argc, /* Number of arguments */ |
+ const char *const*argv, /* Value for all arguments */ |
+ sqlite3_vtab **ppVtab, /* Write the new virtual table object here */ |
+ char **pzErr /* Put error message text here */ |
+){ |
+ int rc = SQLITE_NOMEM; |
+ intarray_vtab *pVtab = sqlite3_malloc(sizeof(intarray_vtab)); |
+ |
+ if( pVtab ){ |
+ memset(pVtab, 0, sizeof(intarray_vtab)); |
+ pVtab->pContent = (sqlite3_intarray*)pAux; |
+ rc = sqlite3_declare_vtab(db, "CREATE TABLE x(value INTEGER PRIMARY KEY)"); |
+ } |
+ *ppVtab = (sqlite3_vtab *)pVtab; |
+ return rc; |
+} |
+ |
+/* |
+** Open a new cursor on the intarray table. |
+*/ |
+static int intarrayOpen(sqlite3_vtab *pVTab, sqlite3_vtab_cursor **ppCursor){ |
+ int rc = SQLITE_NOMEM; |
+ intarray_cursor *pCur; |
+ pCur = sqlite3_malloc(sizeof(intarray_cursor)); |
+ if( pCur ){ |
+ memset(pCur, 0, sizeof(intarray_cursor)); |
+ *ppCursor = (sqlite3_vtab_cursor *)pCur; |
+ rc = SQLITE_OK; |
+ } |
+ return rc; |
+} |
+ |
+/* |
+** Close a intarray table cursor. |
+*/ |
+static int intarrayClose(sqlite3_vtab_cursor *cur){ |
+ intarray_cursor *pCur = (intarray_cursor *)cur; |
+ sqlite3_free(pCur); |
+ return SQLITE_OK; |
+} |
+ |
+/* |
+** Retrieve a column of data. |
+*/ |
+static int intarrayColumn(sqlite3_vtab_cursor *cur, sqlite3_context *ctx, int i){ |
+ intarray_cursor *pCur = (intarray_cursor*)cur; |
+ intarray_vtab *pVtab = (intarray_vtab*)cur->pVtab; |
+ if( pCur->i>=0 && pCur->i<pVtab->pContent->n ){ |
+ sqlite3_result_int64(ctx, pVtab->pContent->a[pCur->i]); |
+ } |
+ return SQLITE_OK; |
+} |
+ |
+/* |
+** Retrieve the current rowid. |
+*/ |
+static int intarrayRowid(sqlite3_vtab_cursor *cur, sqlite_int64 *pRowid){ |
+ intarray_cursor *pCur = (intarray_cursor *)cur; |
+ *pRowid = pCur->i; |
+ return SQLITE_OK; |
+} |
+ |
+static int intarrayEof(sqlite3_vtab_cursor *cur){ |
+ intarray_cursor *pCur = (intarray_cursor *)cur; |
+ intarray_vtab *pVtab = (intarray_vtab *)cur->pVtab; |
+ return pCur->i>=pVtab->pContent->n; |
+} |
+ |
+/* |
+** Advance the cursor to the next row. |
+*/ |
+static int intarrayNext(sqlite3_vtab_cursor *cur){ |
+ intarray_cursor *pCur = (intarray_cursor *)cur; |
+ pCur->i++; |
+ return SQLITE_OK; |
+} |
+ |
+/* |
+** Reset a intarray table cursor. |
+*/ |
+static int intarrayFilter( |
+ sqlite3_vtab_cursor *pVtabCursor, |
+ int idxNum, const char *idxStr, |
+ int argc, sqlite3_value **argv |
+){ |
+ intarray_cursor *pCur = (intarray_cursor *)pVtabCursor; |
+ pCur->i = 0; |
+ return SQLITE_OK; |
+} |
+ |
+/* |
+** Analyse the WHERE condition. |
+*/ |
+static int intarrayBestIndex(sqlite3_vtab *tab, sqlite3_index_info *pIdxInfo){ |
+ return SQLITE_OK; |
+} |
+ |
+/* |
+** A virtual table module that merely echos method calls into TCL |
+** variables. |
+*/ |
+static sqlite3_module intarrayModule = { |
+ 0, /* iVersion */ |
+ intarrayCreate, /* xCreate - create a new virtual table */ |
+ intarrayCreate, /* xConnect - connect to an existing vtab */ |
+ intarrayBestIndex, /* xBestIndex - find the best query index */ |
+ intarrayDestroy, /* xDisconnect - disconnect a vtab */ |
+ intarrayDestroy, /* xDestroy - destroy a vtab */ |
+ intarrayOpen, /* xOpen - open a cursor */ |
+ intarrayClose, /* xClose - close a cursor */ |
+ intarrayFilter, /* xFilter - configure scan constraints */ |
+ intarrayNext, /* xNext - advance a cursor */ |
+ intarrayEof, /* xEof */ |
+ intarrayColumn, /* xColumn - read data */ |
+ intarrayRowid, /* xRowid - read data */ |
+ 0, /* xUpdate */ |
+ 0, /* xBegin */ |
+ 0, /* xSync */ |
+ 0, /* xCommit */ |
+ 0, /* xRollback */ |
+ 0, /* xFindMethod */ |
+ 0, /* xRename */ |
+}; |
+ |
+#endif /* !defined(SQLITE_OMIT_VIRTUALTABLE) */ |
+ |
+/* |
+** Invoke this routine to create a specific instance of an intarray object. |
+** The new intarray object is returned by the 3rd parameter. |
+** |
+** Each intarray object corresponds to a virtual table in the TEMP table |
+** with a name of zName. |
+** |
+** Destroy the intarray object by dropping the virtual table. If not done |
+** explicitly by the application, the virtual table will be dropped implicitly |
+** by the system when the database connection is closed. |
+*/ |
+int sqlite3_intarray_create( |
+ sqlite3 *db, |
+ const char *zName, |
+ sqlite3_intarray **ppReturn |
+){ |
+ int rc = SQLITE_OK; |
+#ifndef SQLITE_OMIT_VIRTUALTABLE |
+ sqlite3_intarray *p; |
+ |
+ *ppReturn = p = sqlite3_malloc( sizeof(*p) ); |
+ if( p==0 ){ |
+ return SQLITE_NOMEM; |
+ } |
+ memset(p, 0, sizeof(*p)); |
+ rc = sqlite3_create_module_v2(db, zName, &intarrayModule, p, |
+ (void(*)(void*))intarrayFree); |
+ if( rc==SQLITE_OK ){ |
+ char *zSql; |
+ zSql = sqlite3_mprintf("CREATE VIRTUAL TABLE temp.%Q USING %Q", |
+ zName, zName); |
+ rc = sqlite3_exec(db, zSql, 0, 0, 0); |
+ sqlite3_free(zSql); |
+ } |
+#endif |
+ return rc; |
+} |
+ |
+/* |
+** Bind a new array array of integers to a specific intarray object. |
+** |
+** The array of integers bound must be unchanged for the duration of |
+** any query against the corresponding virtual table. If the integer |
+** array does change or is deallocated undefined behavior will result. |
+*/ |
+int sqlite3_intarray_bind( |
+ sqlite3_intarray *pIntArray, /* The intarray object to bind to */ |
+ int nElements, /* Number of elements in the intarray */ |
+ sqlite3_int64 *aElements, /* Content of the intarray */ |
+ void (*xFree)(void*) /* How to dispose of the intarray when done */ |
+){ |
+ if( pIntArray->xFree ){ |
+ pIntArray->xFree(pIntArray->a); |
+ } |
+ pIntArray->n = nElements; |
+ pIntArray->a = aElements; |
+ pIntArray->xFree = xFree; |
+ return SQLITE_OK; |
+} |
+ |
+ |
+/***************************************************************************** |
+** Everything below is interface for testing this module. |
+*/ |
+#ifdef SQLITE_TEST |
+#include <tcl.h> |
+ |
+/* |
+** Routines to encode and decode pointers |
+*/ |
+extern int getDbPointer(Tcl_Interp *interp, const char *zA, sqlite3 **ppDb); |
+extern void *sqlite3TestTextToPtr(const char*); |
+extern int sqlite3TestMakePointerStr(Tcl_Interp*, char *zPtr, void*); |
+extern const char *sqlite3TestErrorName(int); |
+ |
+/* |
+** sqlite3_intarray_create DB NAME |
+** |
+** Invoke the sqlite3_intarray_create interface. A string that becomes |
+** the first parameter to sqlite3_intarray_bind. |
+*/ |
+static int test_intarray_create( |
+ ClientData clientData, /* Not used */ |
+ Tcl_Interp *interp, /* The TCL interpreter that invoked this command */ |
+ int objc, /* Number of arguments */ |
+ Tcl_Obj *CONST objv[] /* Command arguments */ |
+){ |
+ sqlite3 *db; |
+ const char *zName; |
+ sqlite3_intarray *pArray; |
+ int rc = SQLITE_OK; |
+ char zPtr[100]; |
+ |
+ if( objc!=3 ){ |
+ Tcl_WrongNumArgs(interp, 1, objv, "DB"); |
+ return TCL_ERROR; |
+ } |
+ if( getDbPointer(interp, Tcl_GetString(objv[1]), &db) ) return TCL_ERROR; |
+ zName = Tcl_GetString(objv[2]); |
+#ifndef SQLITE_OMIT_VIRTUALTABLE |
+ rc = sqlite3_intarray_create(db, zName, &pArray); |
+#endif |
+ if( rc!=SQLITE_OK ){ |
+ assert( pArray==0 ); |
+ Tcl_AppendResult(interp, sqlite3TestErrorName(rc), (char*)0); |
+ return TCL_ERROR; |
+ } |
+ sqlite3TestMakePointerStr(interp, zPtr, pArray); |
+ Tcl_AppendResult(interp, zPtr, (char*)0); |
+ return TCL_OK; |
+} |
+ |
+/* |
+** sqlite3_intarray_bind INTARRAY ?VALUE ...? |
+** |
+** Invoke the sqlite3_intarray_bind interface on the given array of integers. |
+*/ |
+static int test_intarray_bind( |
+ ClientData clientData, /* Not used */ |
+ Tcl_Interp *interp, /* The TCL interpreter that invoked this command */ |
+ int objc, /* Number of arguments */ |
+ Tcl_Obj *CONST objv[] /* Command arguments */ |
+){ |
+ sqlite3_intarray *pArray; |
+ int rc = SQLITE_OK; |
+ int i, n; |
+ sqlite3_int64 *a; |
+ |
+ if( objc<2 ){ |
+ Tcl_WrongNumArgs(interp, 1, objv, "INTARRAY"); |
+ return TCL_ERROR; |
+ } |
+ pArray = (sqlite3_intarray*)sqlite3TestTextToPtr(Tcl_GetString(objv[1])); |
+ n = objc - 2; |
+#ifndef SQLITE_OMIT_VIRTUALTABLE |
+ a = sqlite3_malloc( sizeof(a[0])*n ); |
+ if( a==0 ){ |
+ Tcl_AppendResult(interp, "SQLITE_NOMEM", (char*)0); |
+ return TCL_ERROR; |
+ } |
+ for(i=0; i<n; i++){ |
+ a[i] = 0; |
+ Tcl_GetWideIntFromObj(0, objv[i+2], &a[i]); |
+ } |
+ rc = sqlite3_intarray_bind(pArray, n, a, sqlite3_free); |
+ if( rc!=SQLITE_OK ){ |
+ Tcl_AppendResult(interp, sqlite3TestErrorName(rc), (char*)0); |
+ return TCL_ERROR; |
+ } |
+#endif |
+ return TCL_OK; |
+} |
+ |
+/* |
+** Register commands with the TCL interpreter. |
+*/ |
+int Sqlitetestintarray_Init(Tcl_Interp *interp){ |
+ static struct { |
+ char *zName; |
+ Tcl_ObjCmdProc *xProc; |
+ void *clientData; |
+ } aObjCmd[] = { |
+ { "sqlite3_intarray_create", test_intarray_create, 0 }, |
+ { "sqlite3_intarray_bind", test_intarray_bind, 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); |
+ } |
+ return TCL_OK; |
+} |
+ |
+#endif /* SQLITE_TEST */ |