Index: third_party/sqlite/sqlite-src-3080704/src/test_func.c |
diff --git a/third_party/sqlite/sqlite-src-3080704/src/test_func.c b/third_party/sqlite/sqlite-src-3080704/src/test_func.c |
new file mode 100644 |
index 0000000000000000000000000000000000000000..c7850631d7b213a8ff5346a0a215be1b748b36c6 |
--- /dev/null |
+++ b/third_party/sqlite/sqlite-src-3080704/src/test_func.c |
@@ -0,0 +1,765 @@ |
+/* |
+** 2008 March 19 |
+** |
+** 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 all sorts of SQLite interfaces. This code |
+** implements new SQL functions used by the test scripts. |
+*/ |
+#include "sqlite3.h" |
+#include "tcl.h" |
+#include <stdlib.h> |
+#include <string.h> |
+#include <assert.h> |
+ |
+#include "sqliteInt.h" |
+#include "vdbeInt.h" |
+ |
+ |
+/* |
+** Allocate nByte bytes of space using sqlite3_malloc(). If the |
+** allocation fails, call sqlite3_result_error_nomem() to notify |
+** the database handle that malloc() has failed. |
+*/ |
+static void *testContextMalloc(sqlite3_context *context, int nByte){ |
+ char *z = sqlite3_malloc(nByte); |
+ if( !z && nByte>0 ){ |
+ sqlite3_result_error_nomem(context); |
+ } |
+ return z; |
+} |
+ |
+/* |
+** This function generates a string of random characters. Used for |
+** generating test data. |
+*/ |
+static void randStr(sqlite3_context *context, int argc, sqlite3_value **argv){ |
+ static const unsigned char zSrc[] = |
+ "abcdefghijklmnopqrstuvwxyz" |
+ "ABCDEFGHIJKLMNOPQRSTUVWXYZ" |
+ "0123456789" |
+ ".-!,:*^+=_|?/<> "; |
+ int iMin, iMax, n, r, i; |
+ unsigned char zBuf[1000]; |
+ |
+ /* It used to be possible to call randstr() with any number of arguments, |
+ ** but now it is registered with SQLite as requiring exactly 2. |
+ */ |
+ assert(argc==2); |
+ |
+ iMin = sqlite3_value_int(argv[0]); |
+ if( iMin<0 ) iMin = 0; |
+ if( iMin>=sizeof(zBuf) ) iMin = sizeof(zBuf)-1; |
+ iMax = sqlite3_value_int(argv[1]); |
+ if( iMax<iMin ) iMax = iMin; |
+ if( iMax>=sizeof(zBuf) ) iMax = sizeof(zBuf)-1; |
+ n = iMin; |
+ if( iMax>iMin ){ |
+ sqlite3_randomness(sizeof(r), &r); |
+ r &= 0x7fffffff; |
+ n += r%(iMax + 1 - iMin); |
+ } |
+ assert( n<sizeof(zBuf) ); |
+ sqlite3_randomness(n, zBuf); |
+ for(i=0; i<n; i++){ |
+ zBuf[i] = zSrc[zBuf[i]%(sizeof(zSrc)-1)]; |
+ } |
+ zBuf[n] = 0; |
+ sqlite3_result_text(context, (char*)zBuf, n, SQLITE_TRANSIENT); |
+} |
+ |
+/* |
+** The following two SQL functions are used to test returning a text |
+** result with a destructor. Function 'test_destructor' takes one argument |
+** and returns the same argument interpreted as TEXT. A destructor is |
+** passed with the sqlite3_result_text() call. |
+** |
+** SQL function 'test_destructor_count' returns the number of outstanding |
+** allocations made by 'test_destructor'; |
+** |
+** WARNING: Not threadsafe. |
+*/ |
+static int test_destructor_count_var = 0; |
+static void destructor(void *p){ |
+ char *zVal = (char *)p; |
+ assert(zVal); |
+ zVal--; |
+ sqlite3_free(zVal); |
+ test_destructor_count_var--; |
+} |
+static void test_destructor( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+ char *zVal; |
+ int len; |
+ |
+ test_destructor_count_var++; |
+ assert( nArg==1 ); |
+ if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return; |
+ len = sqlite3_value_bytes(argv[0]); |
+ zVal = testContextMalloc(pCtx, len+3); |
+ if( !zVal ){ |
+ return; |
+ } |
+ zVal[len+1] = 0; |
+ zVal[len+2] = 0; |
+ zVal++; |
+ memcpy(zVal, sqlite3_value_text(argv[0]), len); |
+ sqlite3_result_text(pCtx, zVal, -1, destructor); |
+} |
+#ifndef SQLITE_OMIT_UTF16 |
+static void test_destructor16( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+ char *zVal; |
+ int len; |
+ |
+ test_destructor_count_var++; |
+ assert( nArg==1 ); |
+ if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return; |
+ len = sqlite3_value_bytes16(argv[0]); |
+ zVal = testContextMalloc(pCtx, len+3); |
+ if( !zVal ){ |
+ return; |
+ } |
+ zVal[len+1] = 0; |
+ zVal[len+2] = 0; |
+ zVal++; |
+ memcpy(zVal, sqlite3_value_text16(argv[0]), len); |
+ sqlite3_result_text16(pCtx, zVal, -1, destructor); |
+} |
+#endif |
+static void test_destructor_count( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+ sqlite3_result_int(pCtx, test_destructor_count_var); |
+} |
+ |
+/* |
+** The following aggregate function, test_agg_errmsg16(), takes zero |
+** arguments. It returns the text value returned by the sqlite3_errmsg16() |
+** API function. |
+*/ |
+#ifndef SQLITE_OMIT_BUILTIN_TEST |
+void sqlite3BeginBenignMalloc(void); |
+void sqlite3EndBenignMalloc(void); |
+#else |
+ #define sqlite3BeginBenignMalloc() |
+ #define sqlite3EndBenignMalloc() |
+#endif |
+static void test_agg_errmsg16_step(sqlite3_context *a, int b,sqlite3_value **c){ |
+} |
+static void test_agg_errmsg16_final(sqlite3_context *ctx){ |
+#ifndef SQLITE_OMIT_UTF16 |
+ const void *z; |
+ sqlite3 * db = sqlite3_context_db_handle(ctx); |
+ sqlite3_aggregate_context(ctx, 2048); |
+ sqlite3BeginBenignMalloc(); |
+ z = sqlite3_errmsg16(db); |
+ sqlite3EndBenignMalloc(); |
+ sqlite3_result_text16(ctx, z, -1, SQLITE_TRANSIENT); |
+#endif |
+} |
+ |
+/* |
+** Routines for testing the sqlite3_get_auxdata() and sqlite3_set_auxdata() |
+** interface. |
+** |
+** The test_auxdata() SQL function attempts to register each of its arguments |
+** as auxiliary data. If there are no prior registrations of aux data for |
+** that argument (meaning the argument is not a constant or this is its first |
+** call) then the result for that argument is 0. If there is a prior |
+** registration, the result for that argument is 1. The overall result |
+** is the individual argument results separated by spaces. |
+*/ |
+static void free_test_auxdata(void *p) {sqlite3_free(p);} |
+static void test_auxdata( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+ int i; |
+ char *zRet = testContextMalloc(pCtx, nArg*2); |
+ if( !zRet ) return; |
+ memset(zRet, 0, nArg*2); |
+ for(i=0; i<nArg; i++){ |
+ char const *z = (char*)sqlite3_value_text(argv[i]); |
+ if( z ){ |
+ int n; |
+ char *zAux = sqlite3_get_auxdata(pCtx, i); |
+ if( zAux ){ |
+ zRet[i*2] = '1'; |
+ assert( strcmp(zAux,z)==0 ); |
+ }else { |
+ zRet[i*2] = '0'; |
+ } |
+ n = (int)strlen(z) + 1; |
+ zAux = testContextMalloc(pCtx, n); |
+ if( zAux ){ |
+ memcpy(zAux, z, n); |
+ sqlite3_set_auxdata(pCtx, i, zAux, free_test_auxdata); |
+ } |
+ zRet[i*2+1] = ' '; |
+ } |
+ } |
+ sqlite3_result_text(pCtx, zRet, 2*nArg-1, free_test_auxdata); |
+} |
+ |
+/* |
+** A function to test error reporting from user functions. This function |
+** returns a copy of its first argument as the error message. If the |
+** second argument exists, it becomes the error code. |
+*/ |
+static void test_error( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+ sqlite3_result_error(pCtx, (char*)sqlite3_value_text(argv[0]), -1); |
+ if( nArg==2 ){ |
+ sqlite3_result_error_code(pCtx, sqlite3_value_int(argv[1])); |
+ } |
+} |
+ |
+/* |
+** Implementation of the counter(X) function. If X is an integer |
+** constant, then the first invocation will return X. The second X+1. |
+** and so forth. Can be used (for example) to provide a sequence number |
+** in a result set. |
+*/ |
+static void counterFunc( |
+ sqlite3_context *pCtx, /* Function context */ |
+ int nArg, /* Number of function arguments */ |
+ sqlite3_value **argv /* Values for all function arguments */ |
+){ |
+ int *pCounter = (int*)sqlite3_get_auxdata(pCtx, 0); |
+ if( pCounter==0 ){ |
+ pCounter = sqlite3_malloc( sizeof(*pCounter) ); |
+ if( pCounter==0 ){ |
+ sqlite3_result_error_nomem(pCtx); |
+ return; |
+ } |
+ *pCounter = sqlite3_value_int(argv[0]); |
+ sqlite3_set_auxdata(pCtx, 0, pCounter, sqlite3_free); |
+ }else{ |
+ ++*pCounter; |
+ } |
+ sqlite3_result_int(pCtx, *pCounter); |
+} |
+ |
+ |
+/* |
+** This function takes two arguments. It performance UTF-8/16 type |
+** conversions on the first argument then returns a copy of the second |
+** argument. |
+** |
+** This function is used in cases such as the following: |
+** |
+** SELECT test_isolation(x,x) FROM t1; |
+** |
+** We want to verify that the type conversions that occur on the |
+** first argument do not invalidate the second argument. |
+*/ |
+static void test_isolation( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+#ifndef SQLITE_OMIT_UTF16 |
+ sqlite3_value_text16(argv[0]); |
+ sqlite3_value_text(argv[0]); |
+ sqlite3_value_text16(argv[0]); |
+ sqlite3_value_text(argv[0]); |
+#endif |
+ sqlite3_result_value(pCtx, argv[1]); |
+} |
+ |
+/* |
+** Invoke an SQL statement recursively. The function result is the |
+** first column of the first row of the result set. |
+*/ |
+static void test_eval( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+ sqlite3_stmt *pStmt; |
+ int rc; |
+ sqlite3 *db = sqlite3_context_db_handle(pCtx); |
+ const char *zSql; |
+ |
+ zSql = (char*)sqlite3_value_text(argv[0]); |
+ rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0); |
+ if( rc==SQLITE_OK ){ |
+ rc = sqlite3_step(pStmt); |
+ if( rc==SQLITE_ROW ){ |
+ sqlite3_result_value(pCtx, sqlite3_column_value(pStmt, 0)); |
+ } |
+ rc = sqlite3_finalize(pStmt); |
+ } |
+ if( rc ){ |
+ char *zErr; |
+ assert( pStmt==0 ); |
+ zErr = sqlite3_mprintf("sqlite3_prepare_v2() error: %s",sqlite3_errmsg(db)); |
+ sqlite3_result_text(pCtx, zErr, -1, sqlite3_free); |
+ sqlite3_result_error_code(pCtx, rc); |
+ } |
+} |
+ |
+ |
+/* |
+** convert one character from hex to binary |
+*/ |
+static int testHexChar(char c){ |
+ if( c>='0' && c<='9' ){ |
+ return c - '0'; |
+ }else if( c>='a' && c<='f' ){ |
+ return c - 'a' + 10; |
+ }else if( c>='A' && c<='F' ){ |
+ return c - 'A' + 10; |
+ } |
+ return 0; |
+} |
+ |
+/* |
+** Convert hex to binary. |
+*/ |
+static void testHexToBin(const char *zIn, char *zOut){ |
+ while( zIn[0] && zIn[1] ){ |
+ *(zOut++) = (testHexChar(zIn[0])<<4) + testHexChar(zIn[1]); |
+ zIn += 2; |
+ } |
+} |
+ |
+/* |
+** hex_to_utf16be(HEX) |
+** |
+** Convert the input string from HEX into binary. Then return the |
+** result using sqlite3_result_text16le(). |
+*/ |
+#ifndef SQLITE_OMIT_UTF16 |
+static void testHexToUtf16be( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+ int n; |
+ const char *zIn; |
+ char *zOut; |
+ assert( nArg==1 ); |
+ n = sqlite3_value_bytes(argv[0]); |
+ zIn = (const char*)sqlite3_value_text(argv[0]); |
+ zOut = sqlite3_malloc( n/2 ); |
+ if( zOut==0 ){ |
+ sqlite3_result_error_nomem(pCtx); |
+ }else{ |
+ testHexToBin(zIn, zOut); |
+ sqlite3_result_text16be(pCtx, zOut, n/2, sqlite3_free); |
+ } |
+} |
+#endif |
+ |
+/* |
+** hex_to_utf8(HEX) |
+** |
+** Convert the input string from HEX into binary. Then return the |
+** result using sqlite3_result_text16le(). |
+*/ |
+static void testHexToUtf8( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+ int n; |
+ const char *zIn; |
+ char *zOut; |
+ assert( nArg==1 ); |
+ n = sqlite3_value_bytes(argv[0]); |
+ zIn = (const char*)sqlite3_value_text(argv[0]); |
+ zOut = sqlite3_malloc( n/2 ); |
+ if( zOut==0 ){ |
+ sqlite3_result_error_nomem(pCtx); |
+ }else{ |
+ testHexToBin(zIn, zOut); |
+ sqlite3_result_text(pCtx, zOut, n/2, sqlite3_free); |
+ } |
+} |
+ |
+/* |
+** hex_to_utf16le(HEX) |
+** |
+** Convert the input string from HEX into binary. Then return the |
+** result using sqlite3_result_text16le(). |
+*/ |
+#ifndef SQLITE_OMIT_UTF16 |
+static void testHexToUtf16le( |
+ sqlite3_context *pCtx, |
+ int nArg, |
+ sqlite3_value **argv |
+){ |
+ int n; |
+ const char *zIn; |
+ char *zOut; |
+ assert( nArg==1 ); |
+ n = sqlite3_value_bytes(argv[0]); |
+ zIn = (const char*)sqlite3_value_text(argv[0]); |
+ zOut = sqlite3_malloc( n/2 ); |
+ if( zOut==0 ){ |
+ sqlite3_result_error_nomem(pCtx); |
+ }else{ |
+ testHexToBin(zIn, zOut); |
+ sqlite3_result_text16le(pCtx, zOut, n/2, sqlite3_free); |
+ } |
+} |
+#endif |
+ |
+/* |
+** SQL function: real2hex(X) |
+** |
+** If argument X is a real number, then convert it into a string which is |
+** the big-endian hexadecimal representation of the ieee754 encoding of |
+** that number. If X is not a real number, return NULL. |
+*/ |
+static void real2hex( |
+ sqlite3_context *context, |
+ int argc, |
+ sqlite3_value **argv |
+){ |
+ union { |
+ sqlite3_uint64 i; |
+ double r; |
+ unsigned char x[8]; |
+ } v; |
+ char zOut[20]; |
+ int i; |
+ int bigEndian; |
+ v.i = 1; |
+ bigEndian = v.x[0]==0; |
+ v.r = sqlite3_value_double(argv[0]); |
+ for(i=0; i<8; i++){ |
+ if( bigEndian ){ |
+ zOut[i*2] = "0123456789abcdef"[v.x[i]>>4]; |
+ zOut[i*2+1] = "0123456789abcdef"[v.x[i]&0xf]; |
+ }else{ |
+ zOut[14-i*2] = "0123456789abcdef"[v.x[i]>>4]; |
+ zOut[14-i*2+1] = "0123456789abcdef"[v.x[i]&0xf]; |
+ } |
+ } |
+ zOut[16] = 0; |
+ sqlite3_result_text(context, zOut, -1, SQLITE_TRANSIENT); |
+} |
+ |
+/* |
+** tclcmd: test_extract(record, field) |
+** |
+** This function implements an SQL user-function that accepts a blob |
+** containing a formatted database record as the first argument. The |
+** second argument is the index of the field within that record to |
+** extract and return. |
+*/ |
+static void test_extract( |
+ sqlite3_context *context, |
+ int argc, |
+ sqlite3_value **argv |
+){ |
+ sqlite3 *db = sqlite3_context_db_handle(context); |
+ u8 *pRec; |
+ u8 *pEndHdr; /* Points to one byte past record header */ |
+ u8 *pHdr; /* Current point in record header */ |
+ u8 *pBody; /* Current point in record data */ |
+ u64 nHdr; /* Bytes in record header */ |
+ int iIdx; /* Required field */ |
+ int iCurrent = 0; /* Current field */ |
+ |
+ assert( argc==2 ); |
+ pRec = (u8*)sqlite3_value_blob(argv[0]); |
+ iIdx = sqlite3_value_int(argv[1]); |
+ |
+ pHdr = pRec + sqlite3GetVarint(pRec, &nHdr); |
+ pBody = pEndHdr = &pRec[nHdr]; |
+ |
+ for(iCurrent=0; pHdr<pEndHdr && iCurrent<=iIdx; iCurrent++){ |
+ u64 iSerialType; |
+ Mem mem; |
+ |
+ memset(&mem, 0, sizeof(mem)); |
+ mem.db = db; |
+ mem.enc = ENC(db); |
+ pHdr += sqlite3GetVarint(pHdr, &iSerialType); |
+ pBody += sqlite3VdbeSerialGet(pBody, (u32)iSerialType, &mem); |
+ |
+ if( iCurrent==iIdx ){ |
+ sqlite3_result_value(context, &mem); |
+ } |
+ |
+ if( mem.szMalloc ) sqlite3DbFree(db, mem.zMalloc); |
+ } |
+} |
+ |
+/* |
+** tclcmd: test_decode(record) |
+** |
+** This function implements an SQL user-function that accepts a blob |
+** containing a formatted database record as its only argument. It returns |
+** a tcl list (type SQLITE_TEXT) containing each of the values stored |
+** in the record. |
+*/ |
+static void test_decode( |
+ sqlite3_context *context, |
+ int argc, |
+ sqlite3_value **argv |
+){ |
+ sqlite3 *db = sqlite3_context_db_handle(context); |
+ u8 *pRec; |
+ u8 *pEndHdr; /* Points to one byte past record header */ |
+ u8 *pHdr; /* Current point in record header */ |
+ u8 *pBody; /* Current point in record data */ |
+ u64 nHdr; /* Bytes in record header */ |
+ Tcl_Obj *pRet; /* Return value */ |
+ |
+ pRet = Tcl_NewObj(); |
+ Tcl_IncrRefCount(pRet); |
+ |
+ assert( argc==1 ); |
+ pRec = (u8*)sqlite3_value_blob(argv[0]); |
+ |
+ pHdr = pRec + sqlite3GetVarint(pRec, &nHdr); |
+ pBody = pEndHdr = &pRec[nHdr]; |
+ while( pHdr<pEndHdr ){ |
+ Tcl_Obj *pVal = 0; |
+ u64 iSerialType; |
+ Mem mem; |
+ |
+ memset(&mem, 0, sizeof(mem)); |
+ mem.db = db; |
+ mem.enc = ENC(db); |
+ pHdr += sqlite3GetVarint(pHdr, &iSerialType); |
+ pBody += sqlite3VdbeSerialGet(pBody, (u32)iSerialType, &mem); |
+ |
+ switch( sqlite3_value_type(&mem) ){ |
+ case SQLITE_TEXT: |
+ pVal = Tcl_NewStringObj((const char*)sqlite3_value_text(&mem), -1); |
+ break; |
+ |
+ case SQLITE_BLOB: { |
+ char hexdigit[] = { |
+ '0', '1', '2', '3', '4', '5', '6', '7', |
+ '8', '9', 'a', 'b', 'c', 'd', 'e', 'f' |
+ }; |
+ int n = sqlite3_value_bytes(&mem); |
+ u8 *z = (u8*)sqlite3_value_blob(&mem); |
+ int i; |
+ pVal = Tcl_NewStringObj("x'", -1); |
+ for(i=0; i<n; i++){ |
+ char hex[3]; |
+ hex[0] = hexdigit[((z[i] >> 4) & 0x0F)]; |
+ hex[1] = hexdigit[(z[i] & 0x0F)]; |
+ hex[2] = '\0'; |
+ Tcl_AppendStringsToObj(pVal, hex, 0); |
+ } |
+ Tcl_AppendStringsToObj(pVal, "'", 0); |
+ break; |
+ } |
+ |
+ case SQLITE_FLOAT: |
+ pVal = Tcl_NewDoubleObj(sqlite3_value_double(&mem)); |
+ break; |
+ |
+ case SQLITE_INTEGER: |
+ pVal = Tcl_NewWideIntObj(sqlite3_value_int64(&mem)); |
+ break; |
+ |
+ case SQLITE_NULL: |
+ pVal = Tcl_NewStringObj("NULL", -1); |
+ break; |
+ |
+ default: |
+ assert( 0 ); |
+ } |
+ |
+ Tcl_ListObjAppendElement(0, pRet, pVal); |
+ |
+ if( mem.szMalloc ){ |
+ sqlite3DbFree(db, mem.zMalloc); |
+ } |
+ } |
+ |
+ sqlite3_result_text(context, Tcl_GetString(pRet), -1, SQLITE_TRANSIENT); |
+ Tcl_DecrRefCount(pRet); |
+} |
+ |
+ |
+static int registerTestFunctions(sqlite3 *db){ |
+ static const struct { |
+ char *zName; |
+ signed char nArg; |
+ unsigned char eTextRep; /* 1: UTF-16. 0: UTF-8 */ |
+ void (*xFunc)(sqlite3_context*,int,sqlite3_value **); |
+ } aFuncs[] = { |
+ { "randstr", 2, SQLITE_UTF8, randStr }, |
+ { "test_destructor", 1, SQLITE_UTF8, test_destructor}, |
+#ifndef SQLITE_OMIT_UTF16 |
+ { "test_destructor16", 1, SQLITE_UTF8, test_destructor16}, |
+ { "hex_to_utf16be", 1, SQLITE_UTF8, testHexToUtf16be}, |
+ { "hex_to_utf16le", 1, SQLITE_UTF8, testHexToUtf16le}, |
+#endif |
+ { "hex_to_utf8", 1, SQLITE_UTF8, testHexToUtf8}, |
+ { "test_destructor_count", 0, SQLITE_UTF8, test_destructor_count}, |
+ { "test_auxdata", -1, SQLITE_UTF8, test_auxdata}, |
+ { "test_error", 1, SQLITE_UTF8, test_error}, |
+ { "test_error", 2, SQLITE_UTF8, test_error}, |
+ { "test_eval", 1, SQLITE_UTF8, test_eval}, |
+ { "test_isolation", 2, SQLITE_UTF8, test_isolation}, |
+ { "test_counter", 1, SQLITE_UTF8, counterFunc}, |
+ { "real2hex", 1, SQLITE_UTF8, real2hex}, |
+ { "test_decode", 1, SQLITE_UTF8, test_decode}, |
+ { "test_extract", 2, SQLITE_UTF8, test_extract}, |
+ }; |
+ int i; |
+ |
+ for(i=0; i<sizeof(aFuncs)/sizeof(aFuncs[0]); i++){ |
+ sqlite3_create_function(db, aFuncs[i].zName, aFuncs[i].nArg, |
+ aFuncs[i].eTextRep, 0, aFuncs[i].xFunc, 0, 0); |
+ } |
+ |
+ sqlite3_create_function(db, "test_agg_errmsg16", 0, SQLITE_ANY, 0, 0, |
+ test_agg_errmsg16_step, test_agg_errmsg16_final); |
+ |
+ return SQLITE_OK; |
+} |
+ |
+/* |
+** TCLCMD: autoinstall_test_functions |
+** |
+** Invoke this TCL command to use sqlite3_auto_extension() to cause |
+** the standard set of test functions to be loaded into each new |
+** database connection. |
+*/ |
+static int autoinstall_test_funcs( |
+ void * clientData, |
+ Tcl_Interp *interp, |
+ int objc, |
+ Tcl_Obj *CONST objv[] |
+){ |
+ extern int Md5_Register(sqlite3*); |
+ int rc = sqlite3_auto_extension((void*)registerTestFunctions); |
+ if( rc==SQLITE_OK ){ |
+ rc = sqlite3_auto_extension((void*)Md5_Register); |
+ } |
+ Tcl_SetObjResult(interp, Tcl_NewIntObj(rc)); |
+ return TCL_OK; |
+} |
+ |
+/* |
+** A bogus step function and finalizer function. |
+*/ |
+static void tStep(sqlite3_context *a, int b, sqlite3_value **c){} |
+static void tFinal(sqlite3_context *a){} |
+ |
+ |
+/* |
+** tclcmd: abuse_create_function |
+** |
+** Make various calls to sqlite3_create_function that do not have valid |
+** parameters. Verify that the error condition is detected and reported. |
+*/ |
+static int abuse_create_function( |
+ void * clientData, |
+ Tcl_Interp *interp, |
+ int objc, |
+ Tcl_Obj *CONST objv[] |
+){ |
+ extern int getDbPointer(Tcl_Interp*, const char*, sqlite3**); |
+ sqlite3 *db; |
+ int rc; |
+ int mxArg; |
+ |
+ if( getDbPointer(interp, Tcl_GetString(objv[1]), &db) ) return TCL_ERROR; |
+ |
+ rc = sqlite3_create_function(db, "tx", 1, SQLITE_UTF8, 0, tStep,tStep,tFinal); |
+ if( rc!=SQLITE_MISUSE ) goto abuse_err; |
+ |
+ rc = sqlite3_create_function(db, "tx", 1, SQLITE_UTF8, 0, tStep, tStep, 0); |
+ if( rc!=SQLITE_MISUSE ) goto abuse_err; |
+ |
+ rc = sqlite3_create_function(db, "tx", 1, SQLITE_UTF8, 0, tStep, 0, tFinal); |
+ if( rc!=SQLITE_MISUSE) goto abuse_err; |
+ |
+ rc = sqlite3_create_function(db, "tx", 1, SQLITE_UTF8, 0, 0, 0, tFinal); |
+ if( rc!=SQLITE_MISUSE ) goto abuse_err; |
+ |
+ rc = sqlite3_create_function(db, "tx", 1, SQLITE_UTF8, 0, 0, tStep, 0); |
+ if( rc!=SQLITE_MISUSE ) goto abuse_err; |
+ |
+ rc = sqlite3_create_function(db, "tx", -2, SQLITE_UTF8, 0, tStep, 0, 0); |
+ if( rc!=SQLITE_MISUSE ) goto abuse_err; |
+ |
+ rc = sqlite3_create_function(db, "tx", 128, SQLITE_UTF8, 0, tStep, 0, 0); |
+ if( rc!=SQLITE_MISUSE ) goto abuse_err; |
+ |
+ rc = sqlite3_create_function(db, "funcxx" |
+ "_123456789_123456789_123456789_123456789_123456789" |
+ "_123456789_123456789_123456789_123456789_123456789" |
+ "_123456789_123456789_123456789_123456789_123456789" |
+ "_123456789_123456789_123456789_123456789_123456789" |
+ "_123456789_123456789_123456789_123456789_123456789", |
+ 1, SQLITE_UTF8, 0, tStep, 0, 0); |
+ if( rc!=SQLITE_MISUSE ) goto abuse_err; |
+ |
+ /* This last function registration should actually work. Generate |
+ ** a no-op function (that always returns NULL) and which has the |
+ ** maximum-length function name and the maximum number of parameters. |
+ */ |
+ sqlite3_limit(db, SQLITE_LIMIT_FUNCTION_ARG, 10000); |
+ mxArg = sqlite3_limit(db, SQLITE_LIMIT_FUNCTION_ARG, -1); |
+ rc = sqlite3_create_function(db, "nullx" |
+ "_123456789_123456789_123456789_123456789_123456789" |
+ "_123456789_123456789_123456789_123456789_123456789" |
+ "_123456789_123456789_123456789_123456789_123456789" |
+ "_123456789_123456789_123456789_123456789_123456789" |
+ "_123456789_123456789_123456789_123456789_123456789", |
+ mxArg, SQLITE_UTF8, 0, tStep, 0, 0); |
+ if( rc!=SQLITE_OK ) goto abuse_err; |
+ |
+ return TCL_OK; |
+ |
+abuse_err: |
+ Tcl_AppendResult(interp, "sqlite3_create_function abused test failed", |
+ (char*)0); |
+ return TCL_ERROR; |
+} |
+ |
+/* |
+** Register commands with the TCL interpreter. |
+*/ |
+int Sqlitetest_func_Init(Tcl_Interp *interp){ |
+ static struct { |
+ char *zName; |
+ Tcl_ObjCmdProc *xProc; |
+ } aObjCmd[] = { |
+ { "autoinstall_test_functions", autoinstall_test_funcs }, |
+ { "abuse_create_function", abuse_create_function }, |
+ }; |
+ int i; |
+ extern int Md5_Register(sqlite3*); |
+ |
+ for(i=0; i<sizeof(aObjCmd)/sizeof(aObjCmd[0]); i++){ |
+ Tcl_CreateObjCommand(interp, aObjCmd[i].zName, aObjCmd[i].xProc, 0, 0); |
+ } |
+ sqlite3_initialize(); |
+ sqlite3_auto_extension((void*)registerTestFunctions); |
+ sqlite3_auto_extension((void*)Md5_Register); |
+ return TCL_OK; |
+} |