summaryrefslogtreecommitdiffstats
path: root/src/test_loadext.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/test_loadext.c')
-rw-r--r--src/test_loadext.c128
1 files changed, 128 insertions, 0 deletions
diff --git a/src/test_loadext.c b/src/test_loadext.c
new file mode 100644
index 0000000..6404a69
--- /dev/null
+++ b/src/test_loadext.c
@@ -0,0 +1,128 @@
+/*
+** 2006 June 14
+**
+** 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.
+**
+*************************************************************************
+** Test extension for testing the sqlite3_load_extension() function.
+*/
+#include <string.h>
+#include "sqlite3ext.h"
+SQLITE_EXTENSION_INIT1
+
+/*
+** The half() SQL function returns half of its input value.
+*/
+static void halfFunc(
+ sqlite3_context *context,
+ int argc,
+ sqlite3_value **argv
+){
+ sqlite3_result_double(context, 0.5*sqlite3_value_double(argv[0]));
+}
+
+/*
+** SQL functions to call the sqlite3_status function and return results.
+*/
+static void statusFunc(
+ sqlite3_context *context,
+ int argc,
+ sqlite3_value **argv
+){
+ int op = 0, mx, cur, resetFlag, rc;
+ if( sqlite3_value_type(argv[0])==SQLITE_INTEGER ){
+ op = sqlite3_value_int(argv[0]);
+ }else if( sqlite3_value_type(argv[0])==SQLITE_TEXT ){
+ int i;
+ const char *zName;
+ static const struct {
+ const char *zName;
+ int op;
+ } aOp[] = {
+ { "MEMORY_USED", SQLITE_STATUS_MEMORY_USED },
+ { "PAGECACHE_USED", SQLITE_STATUS_PAGECACHE_USED },
+ { "PAGECACHE_OVERFLOW", SQLITE_STATUS_PAGECACHE_OVERFLOW },
+ { "SCRATCH_USED", SQLITE_STATUS_SCRATCH_USED },
+ { "SCRATCH_OVERFLOW", SQLITE_STATUS_SCRATCH_OVERFLOW },
+ { "MALLOC_SIZE", SQLITE_STATUS_MALLOC_SIZE },
+ };
+ int nOp = sizeof(aOp)/sizeof(aOp[0]);
+ zName = (const char*)sqlite3_value_text(argv[0]);
+ for(i=0; i<nOp; i++){
+ if( strcmp(aOp[i].zName, zName)==0 ){
+ op = aOp[i].op;
+ break;
+ }
+ }
+ if( i>=nOp ){
+ char *zMsg = sqlite3_mprintf("unknown status property: %s", zName);
+ sqlite3_result_error(context, zMsg, -1);
+ sqlite3_free(zMsg);
+ return;
+ }
+ }else{
+ sqlite3_result_error(context, "unknown status type", -1);
+ return;
+ }
+ if( argc==2 ){
+ resetFlag = sqlite3_value_int(argv[1]);
+ }else{
+ resetFlag = 0;
+ }
+ rc = sqlite3_status(op, &cur, &mx, resetFlag);
+ if( rc!=SQLITE_OK ){
+ char *zMsg = sqlite3_mprintf("sqlite3_status(%d,...) returns %d", op, rc);
+ sqlite3_result_error(context, zMsg, -1);
+ sqlite3_free(zMsg);
+ return;
+ }
+ if( argc==2 ){
+ sqlite3_result_int(context, mx);
+ }else{
+ sqlite3_result_int(context, cur);
+ }
+}
+
+/*
+** Extension load function.
+*/
+#ifdef _WIN32
+__declspec(dllexport)
+#endif
+int testloadext_init(
+ sqlite3 *db,
+ char **pzErrMsg,
+ const sqlite3_api_routines *pApi
+){
+ int nErr = 0;
+ SQLITE_EXTENSION_INIT2(pApi);
+ nErr |= sqlite3_create_function(db, "half", 1, SQLITE_ANY, 0, halfFunc, 0, 0);
+ nErr |= sqlite3_create_function(db, "sqlite3_status", 1, SQLITE_ANY, 0,
+ statusFunc, 0, 0);
+ nErr |= sqlite3_create_function(db, "sqlite3_status", 2, SQLITE_ANY, 0,
+ statusFunc, 0, 0);
+ return nErr ? SQLITE_ERROR : SQLITE_OK;
+}
+
+/*
+** Another extension entry point. This one always fails.
+*/
+#ifdef _WIN32
+__declspec(dllexport)
+#endif
+int testbrokenext_init(
+ sqlite3 *db,
+ char **pzErrMsg,
+ const sqlite3_api_routines *pApi
+){
+ char *zErr;
+ SQLITE_EXTENSION_INIT2(pApi);
+ zErr = sqlite3_mprintf("broken!");
+ *pzErrMsg = zErr;
+ return 1;
+}