123 lines
3.3 KiB
C
123 lines
3.3 KiB
C
/*
|
|
** 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;
|
|
}
|