1 /*
2 ** 2006 June 14
3 **
4 ** The author disclaims copyright to this source code. In place of
5 ** a legal notice, here is a blessing:
6 **
7 ** May you do good and not evil.
8 ** May you find forgiveness for yourself and forgive others.
9 ** May you share freely, never taking more than you give.
10 **
11 *************************************************************************
12 ** Test extension for testing the sqlite3_load_extension() function.
13 */
14 #include <string.h>
15 #include "sqlite3ext.h"
16 SQLITE_EXTENSION_INIT1
17
18 /*
19 ** The half() SQL function returns half of its input value.
20 */
halfFunc(sqlite3_context * context,int argc,sqlite3_value ** argv)21 static void halfFunc(
22 sqlite3_context *context,
23 int argc,
24 sqlite3_value **argv
25 ){
26 sqlite3_result_double(context, 0.5*sqlite3_value_double(argv[0]));
27 }
28
29 /*
30 ** SQL functions to call the sqlite3_status function and return results.
31 */
statusFunc(sqlite3_context * context,int argc,sqlite3_value ** argv)32 static void statusFunc(
33 sqlite3_context *context,
34 int argc,
35 sqlite3_value **argv
36 ){
37 int op, mx, cur, resetFlag, rc;
38 if( sqlite3_value_type(argv[0])==SQLITE_INTEGER ){
39 op = sqlite3_value_int(argv[0]);
40 }else if( sqlite3_value_type(argv[0])==SQLITE_TEXT ){
41 int i;
42 const char *zName;
43 static const struct {
44 const char *zName;
45 int op;
46 } aOp[] = {
47 { "MEMORY_USED", SQLITE_STATUS_MEMORY_USED },
48 { "PAGECACHE_USED", SQLITE_STATUS_PAGECACHE_USED },
49 { "PAGECACHE_OVERFLOW", SQLITE_STATUS_PAGECACHE_OVERFLOW },
50 { "SCRATCH_USED", SQLITE_STATUS_SCRATCH_USED },
51 { "SCRATCH_OVERFLOW", SQLITE_STATUS_SCRATCH_OVERFLOW },
52 { "MALLOC_SIZE", SQLITE_STATUS_MALLOC_SIZE },
53 };
54 int nOp = sizeof(aOp)/sizeof(aOp[0]);
55 zName = (const char*)sqlite3_value_text(argv[0]);
56 for(i=0; i<nOp; i++){
57 if( strcmp(aOp[i].zName, zName)==0 ){
58 op = aOp[i].op;
59 break;
60 }
61 }
62 if( i>=nOp ){
63 char *zMsg = sqlite3_mprintf("unknown status property: %s", zName);
64 sqlite3_result_error(context, zMsg, -1);
65 sqlite3_free(zMsg);
66 return;
67 }
68 }else{
69 sqlite3_result_error(context, "unknown status type", -1);
70 return;
71 }
72 if( argc==2 ){
73 resetFlag = sqlite3_value_int(argv[1]);
74 }else{
75 resetFlag = 0;
76 }
77 rc = sqlite3_status(op, &cur, &mx, resetFlag);
78 if( rc!=SQLITE_OK ){
79 char *zMsg = sqlite3_mprintf("sqlite3_status(%d,...) returns %d", op, rc);
80 sqlite3_result_error(context, zMsg, -1);
81 sqlite3_free(zMsg);
82 return;
83 }
84 if( argc==2 ){
85 sqlite3_result_int(context, mx);
86 }else{
87 sqlite3_result_int(context, cur);
88 }
89 }
90
91 /*
92 ** Extension load function.
93 */
testloadext_init(sqlite3 * db,char ** pzErrMsg,const sqlite3_api_routines * pApi)94 int testloadext_init(
95 sqlite3 *db,
96 char **pzErrMsg,
97 const sqlite3_api_routines *pApi
98 ){
99 int nErr = 0;
100 SQLITE_EXTENSION_INIT2(pApi);
101 nErr |= sqlite3_create_function(db, "half", 1, SQLITE_ANY, 0, halfFunc, 0, 0);
102 nErr |= sqlite3_create_function(db, "sqlite3_status", 1, SQLITE_ANY, 0,
103 statusFunc, 0, 0);
104 nErr |= sqlite3_create_function(db, "sqlite3_status", 2, SQLITE_ANY, 0,
105 statusFunc, 0, 0);
106 return nErr ? SQLITE_ERROR : SQLITE_OK;
107 }
108
109 /*
110 ** Another extension entry point. This one always fails.
111 */
testbrokenext_init(sqlite3 * db,char ** pzErrMsg,const sqlite3_api_routines * pApi)112 int testbrokenext_init(
113 sqlite3 *db,
114 char **pzErrMsg,
115 const sqlite3_api_routines *pApi
116 ){
117 char *zErr;
118 SQLITE_EXTENSION_INIT2(pApi);
119 zErr = sqlite3_mprintf("broken!");
120 *pzErrMsg = zErr;
121 return 1;
122 }
123