| /* |
| ** 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, 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. |
| */ |
| 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. |
| */ |
| 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; |
| } |