Annotation of embedaddon/sqlite3/src/test_tclvar.c, revision 1.1.1.1
1.1 misho 1: /*
2: ** 2006 June 13
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: ** Code for testing the virtual table interfaces. This code
13: ** is not included in the SQLite library. It is used for automated
14: ** testing of the SQLite library.
15: **
16: ** The emphasis of this file is a virtual table that provides
17: ** access to TCL variables.
18: */
19: #include "sqliteInt.h"
20: #include "tcl.h"
21: #include <stdlib.h>
22: #include <string.h>
23:
24: #ifndef SQLITE_OMIT_VIRTUALTABLE
25:
26: typedef struct tclvar_vtab tclvar_vtab;
27: typedef struct tclvar_cursor tclvar_cursor;
28:
29: /*
30: ** A tclvar virtual-table object
31: */
32: struct tclvar_vtab {
33: sqlite3_vtab base;
34: Tcl_Interp *interp;
35: };
36:
37: /* A tclvar cursor object */
38: struct tclvar_cursor {
39: sqlite3_vtab_cursor base;
40:
41: Tcl_Obj *pList1; /* Result of [info vars ?pattern?] */
42: Tcl_Obj *pList2; /* Result of [array names [lindex $pList1 $i1]] */
43: int i1; /* Current item in pList1 */
44: int i2; /* Current item (if any) in pList2 */
45: };
46:
47: /* Methods for the tclvar module */
48: static int tclvarConnect(
49: sqlite3 *db,
50: void *pAux,
51: int argc, const char *const*argv,
52: sqlite3_vtab **ppVtab,
53: char **pzErr
54: ){
55: tclvar_vtab *pVtab;
56: static const char zSchema[] =
57: "CREATE TABLE whatever(name TEXT, arrayname TEXT, value TEXT)";
58: pVtab = sqlite3MallocZero( sizeof(*pVtab) );
59: if( pVtab==0 ) return SQLITE_NOMEM;
60: *ppVtab = &pVtab->base;
61: pVtab->interp = (Tcl_Interp *)pAux;
62: sqlite3_declare_vtab(db, zSchema);
63: return SQLITE_OK;
64: }
65: /* Note that for this virtual table, the xCreate and xConnect
66: ** methods are identical. */
67:
68: static int tclvarDisconnect(sqlite3_vtab *pVtab){
69: sqlite3_free(pVtab);
70: return SQLITE_OK;
71: }
72: /* The xDisconnect and xDestroy methods are also the same */
73:
74: /*
75: ** Open a new tclvar cursor.
76: */
77: static int tclvarOpen(sqlite3_vtab *pVTab, sqlite3_vtab_cursor **ppCursor){
78: tclvar_cursor *pCur;
79: pCur = sqlite3MallocZero(sizeof(tclvar_cursor));
80: *ppCursor = &pCur->base;
81: return SQLITE_OK;
82: }
83:
84: /*
85: ** Close a tclvar cursor.
86: */
87: static int tclvarClose(sqlite3_vtab_cursor *cur){
88: tclvar_cursor *pCur = (tclvar_cursor *)cur;
89: if( pCur->pList1 ){
90: Tcl_DecrRefCount(pCur->pList1);
91: }
92: if( pCur->pList2 ){
93: Tcl_DecrRefCount(pCur->pList2);
94: }
95: sqlite3_free(pCur);
96: return SQLITE_OK;
97: }
98:
99: /*
100: ** Returns 1 if data is ready, or 0 if not.
101: */
102: static int next2(Tcl_Interp *interp, tclvar_cursor *pCur, Tcl_Obj *pObj){
103: Tcl_Obj *p;
104:
105: if( pObj ){
106: if( !pCur->pList2 ){
107: p = Tcl_NewStringObj("array names", -1);
108: Tcl_IncrRefCount(p);
109: Tcl_ListObjAppendElement(0, p, pObj);
110: Tcl_EvalObjEx(interp, p, TCL_EVAL_GLOBAL);
111: Tcl_DecrRefCount(p);
112: pCur->pList2 = Tcl_GetObjResult(interp);
113: Tcl_IncrRefCount(pCur->pList2);
114: assert( pCur->i2==0 );
115: }else{
116: int n = 0;
117: pCur->i2++;
118: Tcl_ListObjLength(0, pCur->pList2, &n);
119: if( pCur->i2>=n ){
120: Tcl_DecrRefCount(pCur->pList2);
121: pCur->pList2 = 0;
122: pCur->i2 = 0;
123: return 0;
124: }
125: }
126: }
127:
128: return 1;
129: }
130:
131: static int tclvarNext(sqlite3_vtab_cursor *cur){
132: Tcl_Obj *pObj;
133: int n = 0;
134: int ok = 0;
135:
136: tclvar_cursor *pCur = (tclvar_cursor *)cur;
137: Tcl_Interp *interp = ((tclvar_vtab *)(cur->pVtab))->interp;
138:
139: Tcl_ListObjLength(0, pCur->pList1, &n);
140: while( !ok && pCur->i1<n ){
141: Tcl_ListObjIndex(0, pCur->pList1, pCur->i1, &pObj);
142: ok = next2(interp, pCur, pObj);
143: if( !ok ){
144: pCur->i1++;
145: }
146: }
147:
148: return 0;
149: }
150:
151: static int tclvarFilter(
152: sqlite3_vtab_cursor *pVtabCursor,
153: int idxNum, const char *idxStr,
154: int argc, sqlite3_value **argv
155: ){
156: tclvar_cursor *pCur = (tclvar_cursor *)pVtabCursor;
157: Tcl_Interp *interp = ((tclvar_vtab *)(pVtabCursor->pVtab))->interp;
158:
159: Tcl_Obj *p = Tcl_NewStringObj("info vars", -1);
160: Tcl_IncrRefCount(p);
161:
162: assert( argc==0 || argc==1 );
163: if( argc==1 ){
164: Tcl_Obj *pArg = Tcl_NewStringObj((char*)sqlite3_value_text(argv[0]), -1);
165: Tcl_ListObjAppendElement(0, p, pArg);
166: }
167: Tcl_EvalObjEx(interp, p, TCL_EVAL_GLOBAL);
168: if( pCur->pList1 ){
169: Tcl_DecrRefCount(pCur->pList1);
170: }
171: if( pCur->pList2 ){
172: Tcl_DecrRefCount(pCur->pList2);
173: pCur->pList2 = 0;
174: }
175: pCur->i1 = 0;
176: pCur->i2 = 0;
177: pCur->pList1 = Tcl_GetObjResult(interp);
178: Tcl_IncrRefCount(pCur->pList1);
179: assert( pCur->i1==0 && pCur->i2==0 && pCur->pList2==0 );
180:
181: Tcl_DecrRefCount(p);
182: return tclvarNext(pVtabCursor);
183: }
184:
185: static int tclvarColumn(sqlite3_vtab_cursor *cur, sqlite3_context *ctx, int i){
186: Tcl_Obj *p1;
187: Tcl_Obj *p2;
188: const char *z1;
189: const char *z2 = "";
190: tclvar_cursor *pCur = (tclvar_cursor*)cur;
191: Tcl_Interp *interp = ((tclvar_vtab *)cur->pVtab)->interp;
192:
193: Tcl_ListObjIndex(interp, pCur->pList1, pCur->i1, &p1);
194: Tcl_ListObjIndex(interp, pCur->pList2, pCur->i2, &p2);
195: z1 = Tcl_GetString(p1);
196: if( p2 ){
197: z2 = Tcl_GetString(p2);
198: }
199: switch (i) {
200: case 0: {
201: sqlite3_result_text(ctx, z1, -1, SQLITE_TRANSIENT);
202: break;
203: }
204: case 1: {
205: sqlite3_result_text(ctx, z2, -1, SQLITE_TRANSIENT);
206: break;
207: }
208: case 2: {
209: Tcl_Obj *pVal = Tcl_GetVar2Ex(interp, z1, *z2?z2:0, TCL_GLOBAL_ONLY);
210: sqlite3_result_text(ctx, Tcl_GetString(pVal), -1, SQLITE_TRANSIENT);
211: break;
212: }
213: }
214: return SQLITE_OK;
215: }
216:
217: static int tclvarRowid(sqlite3_vtab_cursor *cur, sqlite_int64 *pRowid){
218: *pRowid = 0;
219: return SQLITE_OK;
220: }
221:
222: static int tclvarEof(sqlite3_vtab_cursor *cur){
223: tclvar_cursor *pCur = (tclvar_cursor*)cur;
224: return (pCur->pList2?0:1);
225: }
226:
227: static int tclvarBestIndex(sqlite3_vtab *tab, sqlite3_index_info *pIdxInfo){
228: int ii;
229:
230: for(ii=0; ii<pIdxInfo->nConstraint; ii++){
231: struct sqlite3_index_constraint const *pCons = &pIdxInfo->aConstraint[ii];
232: if( pCons->iColumn==0 && pCons->usable
233: && pCons->op==SQLITE_INDEX_CONSTRAINT_EQ ){
234: struct sqlite3_index_constraint_usage *pUsage;
235: pUsage = &pIdxInfo->aConstraintUsage[ii];
236: pUsage->omit = 0;
237: pUsage->argvIndex = 1;
238: return SQLITE_OK;
239: }
240: }
241:
242: for(ii=0; ii<pIdxInfo->nConstraint; ii++){
243: struct sqlite3_index_constraint const *pCons = &pIdxInfo->aConstraint[ii];
244: if( pCons->iColumn==0 && pCons->usable
245: && pCons->op==SQLITE_INDEX_CONSTRAINT_MATCH ){
246: struct sqlite3_index_constraint_usage *pUsage;
247: pUsage = &pIdxInfo->aConstraintUsage[ii];
248: pUsage->omit = 1;
249: pUsage->argvIndex = 1;
250: return SQLITE_OK;
251: }
252: }
253:
254: return SQLITE_OK;
255: }
256:
257: /*
258: ** A virtual table module that provides read-only access to a
259: ** Tcl global variable namespace.
260: */
261: static sqlite3_module tclvarModule = {
262: 0, /* iVersion */
263: tclvarConnect,
264: tclvarConnect,
265: tclvarBestIndex,
266: tclvarDisconnect,
267: tclvarDisconnect,
268: tclvarOpen, /* xOpen - open a cursor */
269: tclvarClose, /* xClose - close a cursor */
270: tclvarFilter, /* xFilter - configure scan constraints */
271: tclvarNext, /* xNext - advance a cursor */
272: tclvarEof, /* xEof - check for end of scan */
273: tclvarColumn, /* xColumn - read data */
274: tclvarRowid, /* xRowid - read data */
275: 0, /* xUpdate */
276: 0, /* xBegin */
277: 0, /* xSync */
278: 0, /* xCommit */
279: 0, /* xRollback */
280: 0, /* xFindMethod */
281: 0, /* xRename */
282: };
283:
284: /*
285: ** Decode a pointer to an sqlite3 object.
286: */
287: extern int getDbPointer(Tcl_Interp *interp, const char *zA, sqlite3 **ppDb);
288:
289: /*
290: ** Register the echo virtual table module.
291: */
292: static int register_tclvar_module(
293: ClientData clientData, /* Pointer to sqlite3_enable_XXX function */
294: Tcl_Interp *interp, /* The TCL interpreter that invoked this command */
295: int objc, /* Number of arguments */
296: Tcl_Obj *CONST objv[] /* Command arguments */
297: ){
298: sqlite3 *db;
299: if( objc!=2 ){
300: Tcl_WrongNumArgs(interp, 1, objv, "DB");
301: return TCL_ERROR;
302: }
303: if( getDbPointer(interp, Tcl_GetString(objv[1]), &db) ) return TCL_ERROR;
304: #ifndef SQLITE_OMIT_VIRTUALTABLE
305: sqlite3_create_module(db, "tclvar", &tclvarModule, (void *)interp);
306: #endif
307: return TCL_OK;
308: }
309:
310: #endif
311:
312:
313: /*
314: ** Register commands with the TCL interpreter.
315: */
316: int Sqlitetesttclvar_Init(Tcl_Interp *interp){
317: #ifndef SQLITE_OMIT_VIRTUALTABLE
318: static struct {
319: char *zName;
320: Tcl_ObjCmdProc *xProc;
321: void *clientData;
322: } aObjCmd[] = {
323: { "register_tclvar_module", register_tclvar_module, 0 },
324: };
325: int i;
326: for(i=0; i<sizeof(aObjCmd)/sizeof(aObjCmd[0]); i++){
327: Tcl_CreateObjCommand(interp, aObjCmd[i].zName,
328: aObjCmd[i].xProc, aObjCmd[i].clientData, 0);
329: }
330: #endif
331: return TCL_OK;
332: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>