OLD | NEW |
(Empty) | |
| 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, 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 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 */ |
| 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 } |
OLD | NEW |