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 */ 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 */ 32 static void statusFunc( 33 sqlite3_context *context, 34 int argc, 35 sqlite3_value **argv 36 ){ 37 int op = 0, 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 */ 94 #ifdef _WIN32 95 __declspec(dllexport) 96 #endif 97 int testloadext_init( 98 sqlite3 *db, 99 char **pzErrMsg, 100 const sqlite3_api_routines *pApi 101 ){ 102 int nErr = 0; 103 SQLITE_EXTENSION_INIT2(pApi); 104 nErr |= sqlite3_create_function(db, "half", 1, SQLITE_ANY, 0, halfFunc, 0, 0); 105 nErr |= sqlite3_create_function(db, "sqlite3_status", 1, SQLITE_ANY, 0, 106 statusFunc, 0, 0); 107 nErr |= sqlite3_create_function(db, "sqlite3_status", 2, SQLITE_ANY, 0, 108 statusFunc, 0, 0); 109 return nErr ? SQLITE_ERROR : SQLITE_OK; 110 } 111 112 /* 113 ** Another extension entry point. This one always fails. 114 */ 115 #ifdef _WIN32 116 __declspec(dllexport) 117 #endif 118 int testbrokenext_init( 119 sqlite3 *db, 120 char **pzErrMsg, 121 const sqlite3_api_routines *pApi 122 ){ 123 char *zErr; 124 SQLITE_EXTENSION_INIT2(pApi); 125 zErr = sqlite3_mprintf("broken!"); 126 *pzErrMsg = zErr; 127 return 1; 128 } 129