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"
16SQLITE_EXTENSION_INIT1
17
18/*
19** The half() SQL function returns half of its input value.
20*/
21static 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*/
32static 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*/
94int 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*/
112int 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