OLD | NEW |
| (Empty) |
1 /* | |
2 ** 2009 August 17 | |
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 ** | |
13 ** The code in this file is used for testing SQLite. It is not part of | |
14 ** the source code used in production systems. | |
15 ** | |
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 | |
19 ** true: | |
20 ** | |
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. | |
26 */ | |
27 | |
28 #include "sqliteInt.h" | |
29 #include <string.h> | |
30 #include <tcl.h> | |
31 | |
32 static struct Wrapped { | |
33 sqlite3_pcache_methods2 pcache; | |
34 sqlite3_mem_methods mem; | |
35 sqlite3_mutex_methods mutex; | |
36 | |
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 */ | |
43 } wrapped; | |
44 | |
45 static int wrMemInit(void *pAppData){ | |
46 int rc; | |
47 if( wrapped.mem_fail ){ | |
48 rc = SQLITE_ERROR; | |
49 }else{ | |
50 rc = wrapped.mem.xInit(wrapped.mem.pAppData); | |
51 } | |
52 if( rc==SQLITE_OK ){ | |
53 wrapped.mem_init = 1; | |
54 } | |
55 return rc; | |
56 } | |
57 static void wrMemShutdown(void *pAppData){ | |
58 wrapped.mem.xShutdown(wrapped.mem.pAppData); | |
59 wrapped.mem_init = 0; | |
60 } | |
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);} | |
66 | |
67 | |
68 static int wrMutexInit(void){ | |
69 int rc; | |
70 if( wrapped.mutex_fail ){ | |
71 rc = SQLITE_ERROR; | |
72 }else{ | |
73 rc = wrapped.mutex.xMutexInit(); | |
74 } | |
75 if( rc==SQLITE_OK ){ | |
76 wrapped.mutex_init = 1; | |
77 } | |
78 return rc; | |
79 } | |
80 static int wrMutexEnd(void){ | |
81 wrapped.mutex.xMutexEnd(); | |
82 wrapped.mutex_init = 0; | |
83 return SQLITE_OK; | |
84 } | |
85 static sqlite3_mutex *wrMutexAlloc(int e){ | |
86 return wrapped.mutex.xMutexAlloc(e); | |
87 } | |
88 static void wrMutexFree(sqlite3_mutex *p){ | |
89 wrapped.mutex.xMutexFree(p); | |
90 } | |
91 static void wrMutexEnter(sqlite3_mutex *p){ | |
92 wrapped.mutex.xMutexEnter(p); | |
93 } | |
94 static int wrMutexTry(sqlite3_mutex *p){ | |
95 return wrapped.mutex.xMutexTry(p); | |
96 } | |
97 static void wrMutexLeave(sqlite3_mutex *p){ | |
98 wrapped.mutex.xMutexLeave(p); | |
99 } | |
100 static int wrMutexHeld(sqlite3_mutex *p){ | |
101 return wrapped.mutex.xMutexHeld(p); | |
102 } | |
103 static int wrMutexNotheld(sqlite3_mutex *p){ | |
104 return wrapped.mutex.xMutexNotheld(p); | |
105 } | |
106 | |
107 | |
108 | |
109 static int wrPCacheInit(void *pArg){ | |
110 int rc; | |
111 if( wrapped.pcache_fail ){ | |
112 rc = SQLITE_ERROR; | |
113 }else{ | |
114 rc = wrapped.pcache.xInit(wrapped.pcache.pArg); | |
115 } | |
116 if( rc==SQLITE_OK ){ | |
117 wrapped.pcache_init = 1; | |
118 } | |
119 return rc; | |
120 } | |
121 static void wrPCacheShutdown(void *pArg){ | |
122 wrapped.pcache.xShutdown(wrapped.pcache.pArg); | |
123 wrapped.pcache_init = 0; | |
124 } | |
125 | |
126 static sqlite3_pcache *wrPCacheCreate(int a, int b, int c){ | |
127 return wrapped.pcache.xCreate(a, b, c); | |
128 } | |
129 static void wrPCacheCachesize(sqlite3_pcache *p, int n){ | |
130 wrapped.pcache.xCachesize(p, n); | |
131 } | |
132 static int wrPCachePagecount(sqlite3_pcache *p){ | |
133 return wrapped.pcache.xPagecount(p); | |
134 } | |
135 static sqlite3_pcache_page *wrPCacheFetch(sqlite3_pcache *p, unsigned a, int b){ | |
136 return wrapped.pcache.xFetch(p, a, b); | |
137 } | |
138 static void wrPCacheUnpin(sqlite3_pcache *p, sqlite3_pcache_page *a, int b){ | |
139 wrapped.pcache.xUnpin(p, a, b); | |
140 } | |
141 static void wrPCacheRekey( | |
142 sqlite3_pcache *p, | |
143 sqlite3_pcache_page *a, | |
144 unsigned b, | |
145 unsigned c | |
146 ){ | |
147 wrapped.pcache.xRekey(p, a, b, c); | |
148 } | |
149 static void wrPCacheTruncate(sqlite3_pcache *p, unsigned a){ | |
150 wrapped.pcache.xTruncate(p, a); | |
151 } | |
152 static void wrPCacheDestroy(sqlite3_pcache *p){ | |
153 wrapped.pcache.xDestroy(p); | |
154 } | |
155 | |
156 static void installInitWrappers(void){ | |
157 sqlite3_mutex_methods mutexmethods = { | |
158 wrMutexInit, wrMutexEnd, wrMutexAlloc, | |
159 wrMutexFree, wrMutexEnter, wrMutexTry, | |
160 wrMutexLeave, wrMutexHeld, wrMutexNotheld | |
161 }; | |
162 sqlite3_pcache_methods2 pcachemethods = { | |
163 1, 0, | |
164 wrPCacheInit, wrPCacheShutdown, wrPCacheCreate, | |
165 wrPCacheCachesize, wrPCachePagecount, wrPCacheFetch, | |
166 wrPCacheUnpin, wrPCacheRekey, wrPCacheTruncate, | |
167 wrPCacheDestroy | |
168 }; | |
169 sqlite3_mem_methods memmethods = { | |
170 wrMemMalloc, wrMemFree, wrMemRealloc, | |
171 wrMemSize, wrMemRoundup, wrMemInit, | |
172 wrMemShutdown, | |
173 0 | |
174 }; | |
175 | |
176 memset(&wrapped, 0, sizeof(wrapped)); | |
177 | |
178 sqlite3_shutdown(); | |
179 sqlite3_config(SQLITE_CONFIG_GETMUTEX, &wrapped.mutex); | |
180 sqlite3_config(SQLITE_CONFIG_GETMALLOC, &wrapped.mem); | |
181 sqlite3_config(SQLITE_CONFIG_GETPCACHE2, &wrapped.pcache); | |
182 sqlite3_config(SQLITE_CONFIG_MUTEX, &mutexmethods); | |
183 sqlite3_config(SQLITE_CONFIG_MALLOC, &memmethods); | |
184 sqlite3_config(SQLITE_CONFIG_PCACHE2, &pcachemethods); | |
185 } | |
186 | |
187 static int init_wrapper_install( | |
188 ClientData clientData, /* Unused */ | |
189 Tcl_Interp *interp, /* The TCL interpreter that invoked this command */ | |
190 int objc, /* Number of arguments */ | |
191 Tcl_Obj *CONST objv[] /* Command arguments */ | |
192 ){ | |
193 int i; | |
194 installInitWrappers(); | |
195 for(i=1; i<objc; i++){ | |
196 char *z = Tcl_GetString(objv[i]); | |
197 if( strcmp(z, "mem")==0 ){ | |
198 wrapped.mem_fail = 1; | |
199 }else if( strcmp(z, "mutex")==0 ){ | |
200 wrapped.mutex_fail = 1; | |
201 }else if( strcmp(z, "pcache")==0 ){ | |
202 wrapped.pcache_fail = 1; | |
203 }else{ | |
204 Tcl_AppendResult(interp, "Unknown argument: \"", z, "\""); | |
205 return TCL_ERROR; | |
206 } | |
207 } | |
208 return TCL_OK; | |
209 } | |
210 | |
211 static int init_wrapper_uninstall( | |
212 ClientData clientData, /* Unused */ | |
213 Tcl_Interp *interp, /* The TCL interpreter that invoked this command */ | |
214 int objc, /* Number of arguments */ | |
215 Tcl_Obj *CONST objv[] /* Command arguments */ | |
216 ){ | |
217 if( objc!=1 ){ | |
218 Tcl_WrongNumArgs(interp, 1, objv, ""); | |
219 return TCL_ERROR; | |
220 } | |
221 | |
222 sqlite3_shutdown(); | |
223 sqlite3_config(SQLITE_CONFIG_MUTEX, &wrapped.mutex); | |
224 sqlite3_config(SQLITE_CONFIG_MALLOC, &wrapped.mem); | |
225 sqlite3_config(SQLITE_CONFIG_PCACHE2, &wrapped.pcache); | |
226 return TCL_OK; | |
227 } | |
228 | |
229 static int init_wrapper_clear( | |
230 ClientData clientData, /* Unused */ | |
231 Tcl_Interp *interp, /* The TCL interpreter that invoked this command */ | |
232 int objc, /* Number of arguments */ | |
233 Tcl_Obj *CONST objv[] /* Command arguments */ | |
234 ){ | |
235 if( objc!=1 ){ | |
236 Tcl_WrongNumArgs(interp, 1, objv, ""); | |
237 return TCL_ERROR; | |
238 } | |
239 | |
240 wrapped.mem_fail = 0; | |
241 wrapped.mutex_fail = 0; | |
242 wrapped.pcache_fail = 0; | |
243 return TCL_OK; | |
244 } | |
245 | |
246 static int init_wrapper_query( | |
247 ClientData clientData, /* Unused */ | |
248 Tcl_Interp *interp, /* The TCL interpreter that invoked this command */ | |
249 int objc, /* Number of arguments */ | |
250 Tcl_Obj *CONST objv[] /* Command arguments */ | |
251 ){ | |
252 Tcl_Obj *pRet; | |
253 | |
254 if( objc!=1 ){ | |
255 Tcl_WrongNumArgs(interp, 1, objv, ""); | |
256 return TCL_ERROR; | |
257 } | |
258 | |
259 pRet = Tcl_NewObj(); | |
260 if( wrapped.mutex_init ){ | |
261 Tcl_ListObjAppendElement(interp, pRet, Tcl_NewStringObj("mutex", -1)); | |
262 } | |
263 if( wrapped.mem_init ){ | |
264 Tcl_ListObjAppendElement(interp, pRet, Tcl_NewStringObj("mem", -1)); | |
265 } | |
266 if( wrapped.pcache_init ){ | |
267 Tcl_ListObjAppendElement(interp, pRet, Tcl_NewStringObj("pcache", -1)); | |
268 } | |
269 | |
270 Tcl_SetObjResult(interp, pRet); | |
271 return TCL_OK; | |
272 } | |
273 | |
274 int Sqlitetest_init_Init(Tcl_Interp *interp){ | |
275 static struct { | |
276 char *zName; | |
277 Tcl_ObjCmdProc *xProc; | |
278 } aObjCmd[] = { | |
279 {"init_wrapper_install", init_wrapper_install}, | |
280 {"init_wrapper_query", init_wrapper_query }, | |
281 {"init_wrapper_uninstall", init_wrapper_uninstall}, | |
282 {"init_wrapper_clear", init_wrapper_clear} | |
283 }; | |
284 int i; | |
285 | |
286 for(i=0; i<sizeof(aObjCmd)/sizeof(aObjCmd[0]); i++){ | |
287 Tcl_CreateObjCommand(interp, aObjCmd[i].zName, aObjCmd[i].xProc, 0, 0); | |
288 } | |
289 | |
290 return TCL_OK; | |
291 } | |
OLD | NEW |