4 ** The author disclaims copyright to this source code. In place of
5 ** a legal notice, here is a blessing:
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.
11 *************************************************************************
13 ** The code in this file is used for testing SQLite. It is not part of
14 ** the source code used in production systems.
16 ** Specifically, this file tests the effect of errors while initializing
17 ** the various pluggable sub-systems from within sqlite3_initialize().
18 ** If an error occurs in sqlite3_initialize() the following should be
21 ** 1) An error code is returned to the user, and
22 ** 2) A subsequent call to sqlite3_shutdown() calls the shutdown method
23 ** of those subsystems that were initialized, and
24 ** 3) A subsequent call to sqlite3_initialize() attempts to initialize
25 ** the remaining, uninitialized, subsystems.
28 #include "sqliteInt.h"
32 static struct Wrapped {
33 sqlite3_pcache_methods pcache;
34 sqlite3_mem_methods mem;
35 sqlite3_mutex_methods mutex;
37 int mem_init; /* True if mem subsystem is initalized */
38 int mem_fail; /* True to fail mem subsystem inialization */
39 int mutex_init; /* True if mutex subsystem is initalized */
40 int mutex_fail; /* True to fail mutex subsystem inialization */
41 int pcache_init; /* True if pcache subsystem is initalized */
42 int pcache_fail; /* True to fail pcache subsystem inialization */
45 static int wrMemInit(void *pAppData){
47 if( wrapped.mem_fail ){
50 rc = wrapped.mem.xInit(wrapped.mem.pAppData);
57 static void wrMemShutdown(void *pAppData){
58 wrapped.mem.xShutdown(wrapped.mem.pAppData);
61 static void *wrMemMalloc(int n) {return wrapped.mem.xMalloc(n);}
62 static void wrMemFree(void *p) {wrapped.mem.xFree(p);}
63 static void *wrMemRealloc(void *p, int n) {return wrapped.mem.xRealloc(p, n);}
64 static int wrMemSize(void *p) {return wrapped.mem.xSize(p);}
65 static int wrMemRoundup(int n) {return wrapped.mem.xRoundup(n);}
68 static int wrMutexInit(void){
70 if( wrapped.mutex_fail ){
73 rc = wrapped.mutex.xMutexInit();
76 wrapped.mutex_init = 1;
80 static int wrMutexEnd(void){
81 wrapped.mutex.xMutexEnd();
82 wrapped.mutex_init = 0;
85 static sqlite3_mutex *wrMutexAlloc(int e){
86 return wrapped.mutex.xMutexAlloc(e);
88 static void wrMutexFree(sqlite3_mutex *p){
89 wrapped.mutex.xMutexFree(p);
91 static void wrMutexEnter(sqlite3_mutex *p){
92 wrapped.mutex.xMutexEnter(p);
94 static int wrMutexTry(sqlite3_mutex *p){
95 return wrapped.mutex.xMutexTry(p);
97 static void wrMutexLeave(sqlite3_mutex *p){
98 wrapped.mutex.xMutexLeave(p);
100 static int wrMutexHeld(sqlite3_mutex *p){
101 return wrapped.mutex.xMutexHeld(p);
103 static int wrMutexNotheld(sqlite3_mutex *p){
104 return wrapped.mutex.xMutexNotheld(p);
109 static int wrPCacheInit(void *pArg){
111 if( wrapped.pcache_fail ){
114 rc = wrapped.pcache.xInit(wrapped.pcache.pArg);
117 wrapped.pcache_init = 1;
121 static void wrPCacheShutdown(void *pArg){
122 wrapped.pcache.xShutdown(wrapped.pcache.pArg);
123 wrapped.pcache_init = 0;
126 static sqlite3_pcache *wrPCacheCreate(int a, int b){
127 return wrapped.pcache.xCreate(a, b);
129 static void wrPCacheCachesize(sqlite3_pcache *p, int n){
130 wrapped.pcache.xCachesize(p, n);
132 static int wrPCachePagecount(sqlite3_pcache *p){
133 return wrapped.pcache.xPagecount(p);
135 static void *wrPCacheFetch(sqlite3_pcache *p, unsigned a, int b){
136 return wrapped.pcache.xFetch(p, a, b);
138 static void wrPCacheUnpin(sqlite3_pcache *p, void *a, int b){
139 wrapped.pcache.xUnpin(p, a, b);
141 static void wrPCacheRekey(sqlite3_pcache *p, void *a, unsigned b, unsigned c){
142 wrapped.pcache.xRekey(p, a, b, c);
144 static void wrPCacheTruncate(sqlite3_pcache *p, unsigned a){
145 wrapped.pcache.xTruncate(p, a);
147 static void wrPCacheDestroy(sqlite3_pcache *p){
148 wrapped.pcache.xDestroy(p);
151 static void installInitWrappers(void){
152 sqlite3_mutex_methods mutexmethods = {
153 wrMutexInit, wrMutexEnd, wrMutexAlloc,
154 wrMutexFree, wrMutexEnter, wrMutexTry,
155 wrMutexLeave, wrMutexHeld, wrMutexNotheld
157 sqlite3_pcache_methods pcachemethods = {
159 wrPCacheInit, wrPCacheShutdown, wrPCacheCreate,
160 wrPCacheCachesize, wrPCachePagecount, wrPCacheFetch,
161 wrPCacheUnpin, wrPCacheRekey, wrPCacheTruncate,
164 sqlite3_mem_methods memmethods = {
165 wrMemMalloc, wrMemFree, wrMemRealloc,
166 wrMemSize, wrMemRoundup, wrMemInit,
171 memset(&wrapped, 0, sizeof(wrapped));
174 sqlite3_config(SQLITE_CONFIG_GETMUTEX, &wrapped.mutex);
175 sqlite3_config(SQLITE_CONFIG_GETMALLOC, &wrapped.mem);
176 sqlite3_config(SQLITE_CONFIG_GETPCACHE, &wrapped.pcache);
177 sqlite3_config(SQLITE_CONFIG_MUTEX, &mutexmethods);
178 sqlite3_config(SQLITE_CONFIG_MALLOC, &memmethods);
179 sqlite3_config(SQLITE_CONFIG_PCACHE, &pcachemethods);
182 static int init_wrapper_install(
183 ClientData clientData, /* Unused */
184 Tcl_Interp *interp, /* The TCL interpreter that invoked this command */
185 int objc, /* Number of arguments */
186 Tcl_Obj *CONST objv[] /* Command arguments */
189 installInitWrappers();
190 for(i=1; i<objc; i++){
191 char *z = Tcl_GetString(objv[i]);
192 if( strcmp(z, "mem")==0 ){
193 wrapped.mem_fail = 1;
194 }else if( strcmp(z, "mutex")==0 ){
195 wrapped.mutex_fail = 1;
196 }else if( strcmp(z, "pcache")==0 ){
197 wrapped.pcache_fail = 1;
199 Tcl_AppendResult(interp, "Unknown argument: \"", z, "\"");
206 static int init_wrapper_uninstall(
207 ClientData clientData, /* Unused */
208 Tcl_Interp *interp, /* The TCL interpreter that invoked this command */
209 int objc, /* Number of arguments */
210 Tcl_Obj *CONST objv[] /* Command arguments */
213 Tcl_WrongNumArgs(interp, 1, objv, "");
217 memset(&wrapped, 0, sizeof(&wrapped));
219 sqlite3_config(SQLITE_CONFIG_MUTEX, &wrapped.mutex);
220 sqlite3_config(SQLITE_CONFIG_MALLOC, &wrapped.mem);
221 sqlite3_config(SQLITE_CONFIG_PCACHE, &wrapped.pcache);
225 static int init_wrapper_clear(
226 ClientData clientData, /* Unused */
227 Tcl_Interp *interp, /* The TCL interpreter that invoked this command */
228 int objc, /* Number of arguments */
229 Tcl_Obj *CONST objv[] /* Command arguments */
232 Tcl_WrongNumArgs(interp, 1, objv, "");
236 wrapped.mem_fail = 0;
237 wrapped.mutex_fail = 0;
238 wrapped.pcache_fail = 0;
242 static int init_wrapper_query(
243 ClientData clientData, /* Unused */
244 Tcl_Interp *interp, /* The TCL interpreter that invoked this command */
245 int objc, /* Number of arguments */
246 Tcl_Obj *CONST objv[] /* Command arguments */
251 Tcl_WrongNumArgs(interp, 1, objv, "");
256 if( wrapped.mutex_init ){
257 Tcl_ListObjAppendElement(interp, pRet, Tcl_NewStringObj("mutex", -1));
259 if( wrapped.mem_init ){
260 Tcl_ListObjAppendElement(interp, pRet, Tcl_NewStringObj("mem", -1));
262 if( wrapped.pcache_init ){
263 Tcl_ListObjAppendElement(interp, pRet, Tcl_NewStringObj("pcache", -1));
266 Tcl_SetObjResult(interp, pRet);
270 int Sqlitetest_init_Init(Tcl_Interp *interp){
273 Tcl_ObjCmdProc *xProc;
275 {"init_wrapper_install", init_wrapper_install},
276 {"init_wrapper_query", init_wrapper_query },
277 {"init_wrapper_uninstall", init_wrapper_uninstall},
278 {"init_wrapper_clear", init_wrapper_clear}
282 for(i=0; i<sizeof(aObjCmd)/sizeof(aObjCmd[0]); i++){
283 Tcl_CreateObjCommand(interp, aObjCmd[i].zName, aObjCmd[i].xProc, 0, 0);