xref: /sqlite-3.40.0/src/tclsqlite.c (revision 97f2ebc1)
175897234Sdrh /*
2b19a2bc6Sdrh ** 2001 September 15
375897234Sdrh **
4b19a2bc6Sdrh ** The author disclaims copyright to this source code.  In place of
5b19a2bc6Sdrh ** a legal notice, here is a blessing:
675897234Sdrh **
7b19a2bc6Sdrh **    May you do good and not evil.
8b19a2bc6Sdrh **    May you find forgiveness for yourself and forgive others.
9b19a2bc6Sdrh **    May you share freely, never taking more than you give.
1075897234Sdrh **
1175897234Sdrh *************************************************************************
1275897234Sdrh ** A TCL Interface to SQLite
1375897234Sdrh **
14*97f2ebc1Sdrh ** $Id: tclsqlite.c,v 1.136 2005/12/10 21:19:05 drh Exp $
1575897234Sdrh */
166d31316cSdrh #ifndef NO_TCL     /* Omit this whole file if TCL is unavailable */
176d31316cSdrh 
1806b2718aSdrh #include "sqliteInt.h"
19895d7472Sdrh #include "hash.h"
2017a68934Sdrh #include "tcl.h"
2175897234Sdrh #include <stdlib.h>
2275897234Sdrh #include <string.h>
23ce927065Sdrh #include <assert.h>
24d1e4733dSdrh #include <ctype.h>
2575897234Sdrh 
2629bc4615Sdrh /*
2729bc4615Sdrh  * Windows needs to know which symbols to export.  Unix does not.
2829bc4615Sdrh  * BUILD_sqlite should be undefined for Unix.
2929bc4615Sdrh  */
3029bc4615Sdrh 
3129bc4615Sdrh #ifdef BUILD_sqlite
3229bc4615Sdrh #undef TCL_STORAGE_CLASS
3329bc4615Sdrh #define TCL_STORAGE_CLASS DLLEXPORT
3429bc4615Sdrh #endif /* BUILD_sqlite */
3529bc4615Sdrh 
36a21c6b6fSdanielk1977 #define NUM_PREPARED_STMTS 10
37fb7e7651Sdrh #define MAX_PREPARED_STMTS 100
38fb7e7651Sdrh 
3975897234Sdrh /*
4098808babSdrh ** If TCL uses UTF-8 and SQLite is configured to use iso8859, then we
4198808babSdrh ** have to do a translation when going between the two.  Set the
4298808babSdrh ** UTF_TRANSLATION_NEEDED macro to indicate that we need to do
4398808babSdrh ** this translation.
4498808babSdrh */
4598808babSdrh #if defined(TCL_UTF_MAX) && !defined(SQLITE_UTF8)
4698808babSdrh # define UTF_TRANSLATION_NEEDED 1
4798808babSdrh #endif
4898808babSdrh 
4998808babSdrh /*
50cabb0819Sdrh ** New SQL functions can be created as TCL scripts.  Each such function
51cabb0819Sdrh ** is described by an instance of the following structure.
52cabb0819Sdrh */
53cabb0819Sdrh typedef struct SqlFunc SqlFunc;
54cabb0819Sdrh struct SqlFunc {
55cabb0819Sdrh   Tcl_Interp *interp;   /* The TCL interpret to execute the function */
56d1e4733dSdrh   Tcl_Obj *pScript;     /* The Tcl_Obj representation of the script */
57d1e4733dSdrh   int useEvalObjv;      /* True if it is safe to use Tcl_EvalObjv */
58d1e4733dSdrh   char *zName;          /* Name of this function */
59cabb0819Sdrh   SqlFunc *pNext;       /* Next function on the list of them all */
60cabb0819Sdrh };
61cabb0819Sdrh 
62cabb0819Sdrh /*
630202b29eSdanielk1977 ** New collation sequences function can be created as TCL scripts.  Each such
640202b29eSdanielk1977 ** function is described by an instance of the following structure.
650202b29eSdanielk1977 */
660202b29eSdanielk1977 typedef struct SqlCollate SqlCollate;
670202b29eSdanielk1977 struct SqlCollate {
680202b29eSdanielk1977   Tcl_Interp *interp;   /* The TCL interpret to execute the function */
690202b29eSdanielk1977   char *zScript;        /* The script to be run */
700202b29eSdanielk1977   SqlCollate *pNext;    /* Next function on the list of them all */
710202b29eSdanielk1977 };
720202b29eSdanielk1977 
730202b29eSdanielk1977 /*
74fb7e7651Sdrh ** Prepared statements are cached for faster execution.  Each prepared
75fb7e7651Sdrh ** statement is described by an instance of the following structure.
76fb7e7651Sdrh */
77fb7e7651Sdrh typedef struct SqlPreparedStmt SqlPreparedStmt;
78fb7e7651Sdrh struct SqlPreparedStmt {
79fb7e7651Sdrh   SqlPreparedStmt *pNext;  /* Next in linked list */
80fb7e7651Sdrh   SqlPreparedStmt *pPrev;  /* Previous on the list */
81fb7e7651Sdrh   sqlite3_stmt *pStmt;     /* The prepared statement */
82fb7e7651Sdrh   int nSql;                /* chars in zSql[] */
83fb7e7651Sdrh   char zSql[1];            /* Text of the SQL statement */
84fb7e7651Sdrh };
85fb7e7651Sdrh 
86fb7e7651Sdrh /*
87bec3f402Sdrh ** There is one instance of this structure for each SQLite database
88bec3f402Sdrh ** that has been opened by the SQLite TCL interface.
89bec3f402Sdrh */
90bec3f402Sdrh typedef struct SqliteDb SqliteDb;
91bec3f402Sdrh struct SqliteDb {
92895d7472Sdrh   sqlite3 *db;               /* The "real" database structure */
93bec3f402Sdrh   Tcl_Interp *interp;        /* The interpreter used for this database */
946d31316cSdrh   char *zBusy;               /* The busy callback routine */
95aa940eacSdrh   char *zCommit;             /* The commit hook callback routine */
96b5a20d3cSdrh   char *zTrace;              /* The trace callback routine */
9719e2d37fSdrh   char *zProfile;            /* The profile callback routine */
98348bb5d6Sdanielk1977   char *zProgress;           /* The progress callback routine */
99e22a334bSdrh   char *zAuth;               /* The authorization callback routine */
10055c45f2eSdanielk1977   char *zNull;               /* Text to substitute for an SQL NULL value */
101cabb0819Sdrh   SqlFunc *pFunc;            /* List of SQL functions */
1020202b29eSdanielk1977   SqlCollate *pCollate;      /* List of SQL collation functions */
1036f8a503dSdanielk1977   int rc;                    /* Return code of most recent sqlite3_exec() */
1047cedc8d4Sdanielk1977   Tcl_Obj *pCollateNeeded;   /* Collation needed script */
105fb7e7651Sdrh   SqlPreparedStmt *stmtList; /* List of prepared statements*/
106fb7e7651Sdrh   SqlPreparedStmt *stmtLast; /* Last statement in the list */
107fb7e7651Sdrh   int maxStmt;               /* The next maximum number of stmtList */
108fb7e7651Sdrh   int nStmt;                 /* Number of statements in stmtList */
10998808babSdrh };
110297ecf14Sdrh 
1116d31316cSdrh /*
112d1e4733dSdrh ** Look at the script prefix in pCmd.  We will be executing this script
113d1e4733dSdrh ** after first appending one or more arguments.  This routine analyzes
114d1e4733dSdrh ** the script to see if it is safe to use Tcl_EvalObjv() on the script
115d1e4733dSdrh ** rather than the more general Tcl_EvalEx().  Tcl_EvalObjv() is much
116d1e4733dSdrh ** faster.
117d1e4733dSdrh **
118d1e4733dSdrh ** Scripts that are safe to use with Tcl_EvalObjv() consists of a
119d1e4733dSdrh ** command name followed by zero or more arguments with no [...] or $
120d1e4733dSdrh ** or {...} or ; to be seen anywhere.  Most callback scripts consist
121d1e4733dSdrh ** of just a single procedure name and they meet this requirement.
122d1e4733dSdrh */
123d1e4733dSdrh static int safeToUseEvalObjv(Tcl_Interp *interp, Tcl_Obj *pCmd){
124d1e4733dSdrh   /* We could try to do something with Tcl_Parse().  But we will instead
125d1e4733dSdrh   ** just do a search for forbidden characters.  If any of the forbidden
126d1e4733dSdrh   ** characters appear in pCmd, we will report the string as unsafe.
127d1e4733dSdrh   */
128d1e4733dSdrh   const char *z;
129d1e4733dSdrh   int n;
130d1e4733dSdrh   z = Tcl_GetStringFromObj(pCmd, &n);
131d1e4733dSdrh   while( n-- > 0 ){
132d1e4733dSdrh     int c = *(z++);
133d1e4733dSdrh     if( c=='$' || c=='[' || c==';' ) return 0;
134d1e4733dSdrh   }
135d1e4733dSdrh   return 1;
136d1e4733dSdrh }
137d1e4733dSdrh 
138d1e4733dSdrh /*
139d1e4733dSdrh ** Find an SqlFunc structure with the given name.  Or create a new
140d1e4733dSdrh ** one if an existing one cannot be found.  Return a pointer to the
141d1e4733dSdrh ** structure.
142d1e4733dSdrh */
143d1e4733dSdrh static SqlFunc *findSqlFunc(SqliteDb *pDb, const char *zName){
144d1e4733dSdrh   SqlFunc *p, *pNew;
145d1e4733dSdrh   int i;
146d1e4733dSdrh   pNew = (SqlFunc*)Tcl_Alloc( sizeof(*pNew) + strlen(zName) + 1 );
147d1e4733dSdrh   pNew->zName = (char*)&pNew[1];
148d1e4733dSdrh   for(i=0; zName[i]; i++){ pNew->zName[i] = tolower(zName[i]); }
149d1e4733dSdrh   pNew->zName[i] = 0;
150d1e4733dSdrh   for(p=pDb->pFunc; p; p=p->pNext){
151d1e4733dSdrh     if( strcmp(p->zName, pNew->zName)==0 ){
152d1e4733dSdrh       Tcl_Free((char*)pNew);
153d1e4733dSdrh       return p;
154d1e4733dSdrh     }
155d1e4733dSdrh   }
156d1e4733dSdrh   pNew->interp = pDb->interp;
157d1e4733dSdrh   pNew->pScript = 0;
158d1e4733dSdrh   pNew->pNext = pDb->pFunc;
159d1e4733dSdrh   pDb->pFunc = pNew;
160d1e4733dSdrh   return pNew;
161d1e4733dSdrh }
162d1e4733dSdrh 
163d1e4733dSdrh /*
164fb7e7651Sdrh ** Finalize and free a list of prepared statements
165fb7e7651Sdrh */
166fb7e7651Sdrh static void flushStmtCache( SqliteDb *pDb ){
167fb7e7651Sdrh   SqlPreparedStmt *pPreStmt;
168fb7e7651Sdrh 
169fb7e7651Sdrh   while(  pDb->stmtList ){
170fb7e7651Sdrh     sqlite3_finalize( pDb->stmtList->pStmt );
171fb7e7651Sdrh     pPreStmt = pDb->stmtList;
172fb7e7651Sdrh     pDb->stmtList = pDb->stmtList->pNext;
173fb7e7651Sdrh     Tcl_Free( (char*)pPreStmt );
174fb7e7651Sdrh   }
175fb7e7651Sdrh   pDb->nStmt = 0;
176fb7e7651Sdrh   pDb->stmtLast = 0;
177fb7e7651Sdrh }
178fb7e7651Sdrh 
179fb7e7651Sdrh /*
180895d7472Sdrh ** TCL calls this procedure when an sqlite3 database command is
181895d7472Sdrh ** deleted.
18275897234Sdrh */
18375897234Sdrh static void DbDeleteCmd(void *db){
184bec3f402Sdrh   SqliteDb *pDb = (SqliteDb*)db;
185fb7e7651Sdrh   flushStmtCache(pDb);
1866f8a503dSdanielk1977   sqlite3_close(pDb->db);
187cabb0819Sdrh   while( pDb->pFunc ){
188cabb0819Sdrh     SqlFunc *pFunc = pDb->pFunc;
189cabb0819Sdrh     pDb->pFunc = pFunc->pNext;
190d1e4733dSdrh     Tcl_DecrRefCount(pFunc->pScript);
191cabb0819Sdrh     Tcl_Free((char*)pFunc);
192cabb0819Sdrh   }
1930202b29eSdanielk1977   while( pDb->pCollate ){
1940202b29eSdanielk1977     SqlCollate *pCollate = pDb->pCollate;
1950202b29eSdanielk1977     pDb->pCollate = pCollate->pNext;
1960202b29eSdanielk1977     Tcl_Free((char*)pCollate);
1970202b29eSdanielk1977   }
198bec3f402Sdrh   if( pDb->zBusy ){
199bec3f402Sdrh     Tcl_Free(pDb->zBusy);
200bec3f402Sdrh   }
201b5a20d3cSdrh   if( pDb->zTrace ){
202b5a20d3cSdrh     Tcl_Free(pDb->zTrace);
2030d1a643aSdrh   }
20419e2d37fSdrh   if( pDb->zProfile ){
20519e2d37fSdrh     Tcl_Free(pDb->zProfile);
20619e2d37fSdrh   }
207e22a334bSdrh   if( pDb->zAuth ){
208e22a334bSdrh     Tcl_Free(pDb->zAuth);
209e22a334bSdrh   }
21055c45f2eSdanielk1977   if( pDb->zNull ){
21155c45f2eSdanielk1977     Tcl_Free(pDb->zNull);
21255c45f2eSdanielk1977   }
213bec3f402Sdrh   Tcl_Free((char*)pDb);
214bec3f402Sdrh }
215bec3f402Sdrh 
216bec3f402Sdrh /*
217bec3f402Sdrh ** This routine is called when a database file is locked while trying
218bec3f402Sdrh ** to execute SQL.
219bec3f402Sdrh */
2202a764eb0Sdanielk1977 static int DbBusyHandler(void *cd, int nTries){
221bec3f402Sdrh   SqliteDb *pDb = (SqliteDb*)cd;
222bec3f402Sdrh   int rc;
223bec3f402Sdrh   char zVal[30];
224bec3f402Sdrh 
225bec3f402Sdrh   sprintf(zVal, "%d", nTries);
226d1e4733dSdrh   rc = Tcl_VarEval(pDb->interp, pDb->zBusy, " ", zVal, (char*)0);
227bec3f402Sdrh   if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){
228bec3f402Sdrh     return 0;
229bec3f402Sdrh   }
230bec3f402Sdrh   return 1;
23175897234Sdrh }
23275897234Sdrh 
23375897234Sdrh /*
234348bb5d6Sdanielk1977 ** This routine is invoked as the 'progress callback' for the database.
235348bb5d6Sdanielk1977 */
236348bb5d6Sdanielk1977 static int DbProgressHandler(void *cd){
237348bb5d6Sdanielk1977   SqliteDb *pDb = (SqliteDb*)cd;
238348bb5d6Sdanielk1977   int rc;
239348bb5d6Sdanielk1977 
240348bb5d6Sdanielk1977   assert( pDb->zProgress );
241348bb5d6Sdanielk1977   rc = Tcl_Eval(pDb->interp, pDb->zProgress);
242348bb5d6Sdanielk1977   if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){
243348bb5d6Sdanielk1977     return 1;
244348bb5d6Sdanielk1977   }
245348bb5d6Sdanielk1977   return 0;
246348bb5d6Sdanielk1977 }
247348bb5d6Sdanielk1977 
248348bb5d6Sdanielk1977 /*
249b5a20d3cSdrh ** This routine is called by the SQLite trace handler whenever a new
250b5a20d3cSdrh ** block of SQL is executed.  The TCL script in pDb->zTrace is executed.
2510d1a643aSdrh */
252b5a20d3cSdrh static void DbTraceHandler(void *cd, const char *zSql){
2530d1a643aSdrh   SqliteDb *pDb = (SqliteDb*)cd;
254b5a20d3cSdrh   Tcl_DString str;
2550d1a643aSdrh 
256b5a20d3cSdrh   Tcl_DStringInit(&str);
257b5a20d3cSdrh   Tcl_DStringAppend(&str, pDb->zTrace, -1);
258b5a20d3cSdrh   Tcl_DStringAppendElement(&str, zSql);
259b5a20d3cSdrh   Tcl_Eval(pDb->interp, Tcl_DStringValue(&str));
260b5a20d3cSdrh   Tcl_DStringFree(&str);
261b5a20d3cSdrh   Tcl_ResetResult(pDb->interp);
2620d1a643aSdrh }
2630d1a643aSdrh 
2640d1a643aSdrh /*
26519e2d37fSdrh ** This routine is called by the SQLite profile handler after a statement
26619e2d37fSdrh ** SQL has executed.  The TCL script in pDb->zProfile is evaluated.
26719e2d37fSdrh */
26819e2d37fSdrh static void DbProfileHandler(void *cd, const char *zSql, sqlite_uint64 tm){
26919e2d37fSdrh   SqliteDb *pDb = (SqliteDb*)cd;
27019e2d37fSdrh   Tcl_DString str;
27119e2d37fSdrh   char zTm[100];
27219e2d37fSdrh 
27319e2d37fSdrh   sqlite3_snprintf(sizeof(zTm)-1, zTm, "%lld", tm);
27419e2d37fSdrh   Tcl_DStringInit(&str);
27519e2d37fSdrh   Tcl_DStringAppend(&str, pDb->zProfile, -1);
27619e2d37fSdrh   Tcl_DStringAppendElement(&str, zSql);
27719e2d37fSdrh   Tcl_DStringAppendElement(&str, zTm);
27819e2d37fSdrh   Tcl_Eval(pDb->interp, Tcl_DStringValue(&str));
27919e2d37fSdrh   Tcl_DStringFree(&str);
28019e2d37fSdrh   Tcl_ResetResult(pDb->interp);
28119e2d37fSdrh }
28219e2d37fSdrh 
28319e2d37fSdrh /*
284aa940eacSdrh ** This routine is called when a transaction is committed.  The
285aa940eacSdrh ** TCL script in pDb->zCommit is executed.  If it returns non-zero or
286aa940eacSdrh ** if it throws an exception, the transaction is rolled back instead
287aa940eacSdrh ** of being committed.
288aa940eacSdrh */
289aa940eacSdrh static int DbCommitHandler(void *cd){
290aa940eacSdrh   SqliteDb *pDb = (SqliteDb*)cd;
291aa940eacSdrh   int rc;
292aa940eacSdrh 
293aa940eacSdrh   rc = Tcl_Eval(pDb->interp, pDb->zCommit);
294aa940eacSdrh   if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){
295aa940eacSdrh     return 1;
296aa940eacSdrh   }
297aa940eacSdrh   return 0;
298aa940eacSdrh }
299aa940eacSdrh 
3007cedc8d4Sdanielk1977 static void tclCollateNeeded(
3017cedc8d4Sdanielk1977   void *pCtx,
3029bb575fdSdrh   sqlite3 *db,
3037cedc8d4Sdanielk1977   int enc,
3047cedc8d4Sdanielk1977   const char *zName
3057cedc8d4Sdanielk1977 ){
3067cedc8d4Sdanielk1977   SqliteDb *pDb = (SqliteDb *)pCtx;
3077cedc8d4Sdanielk1977   Tcl_Obj *pScript = Tcl_DuplicateObj(pDb->pCollateNeeded);
3087cedc8d4Sdanielk1977   Tcl_IncrRefCount(pScript);
3097cedc8d4Sdanielk1977   Tcl_ListObjAppendElement(0, pScript, Tcl_NewStringObj(zName, -1));
3107cedc8d4Sdanielk1977   Tcl_EvalObjEx(pDb->interp, pScript, 0);
3117cedc8d4Sdanielk1977   Tcl_DecrRefCount(pScript);
3127cedc8d4Sdanielk1977 }
3137cedc8d4Sdanielk1977 
314aa940eacSdrh /*
3150202b29eSdanielk1977 ** This routine is called to evaluate an SQL collation function implemented
3160202b29eSdanielk1977 ** using TCL script.
3170202b29eSdanielk1977 */
3180202b29eSdanielk1977 static int tclSqlCollate(
3190202b29eSdanielk1977   void *pCtx,
3200202b29eSdanielk1977   int nA,
3210202b29eSdanielk1977   const void *zA,
3220202b29eSdanielk1977   int nB,
3230202b29eSdanielk1977   const void *zB
3240202b29eSdanielk1977 ){
3250202b29eSdanielk1977   SqlCollate *p = (SqlCollate *)pCtx;
3260202b29eSdanielk1977   Tcl_Obj *pCmd;
3270202b29eSdanielk1977 
3280202b29eSdanielk1977   pCmd = Tcl_NewStringObj(p->zScript, -1);
3290202b29eSdanielk1977   Tcl_IncrRefCount(pCmd);
3300202b29eSdanielk1977   Tcl_ListObjAppendElement(p->interp, pCmd, Tcl_NewStringObj(zA, nA));
3310202b29eSdanielk1977   Tcl_ListObjAppendElement(p->interp, pCmd, Tcl_NewStringObj(zB, nB));
332d1e4733dSdrh   Tcl_EvalObjEx(p->interp, pCmd, TCL_EVAL_DIRECT);
3330202b29eSdanielk1977   Tcl_DecrRefCount(pCmd);
3340202b29eSdanielk1977   return (atoi(Tcl_GetStringResult(p->interp)));
3350202b29eSdanielk1977 }
3360202b29eSdanielk1977 
3370202b29eSdanielk1977 /*
338cabb0819Sdrh ** This routine is called to evaluate an SQL function implemented
339cabb0819Sdrh ** using TCL script.
340cabb0819Sdrh */
3410ae8b831Sdanielk1977 static void tclSqlFunc(sqlite3_context *context, int argc, sqlite3_value**argv){
3426f8a503dSdanielk1977   SqlFunc *p = sqlite3_user_data(context);
343d1e4733dSdrh   Tcl_Obj *pCmd;
344cabb0819Sdrh   int i;
345cabb0819Sdrh   int rc;
346cabb0819Sdrh 
347d1e4733dSdrh   if( argc==0 ){
348d1e4733dSdrh     /* If there are no arguments to the function, call Tcl_EvalObjEx on the
349d1e4733dSdrh     ** script object directly.  This allows the TCL compiler to generate
350d1e4733dSdrh     ** bytecode for the command on the first invocation and thus make
351d1e4733dSdrh     ** subsequent invocations much faster. */
352d1e4733dSdrh     pCmd = p->pScript;
353d1e4733dSdrh     Tcl_IncrRefCount(pCmd);
354d1e4733dSdrh     rc = Tcl_EvalObjEx(p->interp, pCmd, 0);
355d1e4733dSdrh     Tcl_DecrRefCount(pCmd);
35651ad0ecdSdanielk1977   }else{
357d1e4733dSdrh     /* If there are arguments to the function, make a shallow copy of the
358d1e4733dSdrh     ** script object, lappend the arguments, then evaluate the copy.
359d1e4733dSdrh     **
360d1e4733dSdrh     ** By "shallow" copy, we mean a only the outer list Tcl_Obj is duplicated.
361d1e4733dSdrh     ** The new Tcl_Obj contains pointers to the original list elements.
362d1e4733dSdrh     ** That way, when Tcl_EvalObjv() is run and shimmers the first element
363d1e4733dSdrh     ** of the list to tclCmdNameType, that alternate representation will
364d1e4733dSdrh     ** be preserved and reused on the next invocation.
365d1e4733dSdrh     */
366d1e4733dSdrh     Tcl_Obj **aArg;
367d1e4733dSdrh     int nArg;
368d1e4733dSdrh     if( Tcl_ListObjGetElements(p->interp, p->pScript, &nArg, &aArg) ){
369d1e4733dSdrh       sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1);
370d1e4733dSdrh       return;
371d1e4733dSdrh     }
372d1e4733dSdrh     pCmd = Tcl_NewListObj(nArg, aArg);
373d1e4733dSdrh     Tcl_IncrRefCount(pCmd);
374d1e4733dSdrh     for(i=0; i<argc; i++){
375d1e4733dSdrh       sqlite3_value *pIn = argv[i];
376d1e4733dSdrh       Tcl_Obj *pVal;
377d1e4733dSdrh 
378d1e4733dSdrh       /* Set pVal to contain the i'th column of this row. */
379d1e4733dSdrh       switch( sqlite3_value_type(pIn) ){
380d1e4733dSdrh         case SQLITE_BLOB: {
381d1e4733dSdrh           int bytes = sqlite3_value_bytes(pIn);
382d1e4733dSdrh           pVal = Tcl_NewByteArrayObj(sqlite3_value_blob(pIn), bytes);
383d1e4733dSdrh           break;
384d1e4733dSdrh         }
385d1e4733dSdrh         case SQLITE_INTEGER: {
386d1e4733dSdrh           sqlite_int64 v = sqlite3_value_int64(pIn);
387d1e4733dSdrh           if( v>=-2147483647 && v<=2147483647 ){
388d1e4733dSdrh             pVal = Tcl_NewIntObj(v);
389d1e4733dSdrh           }else{
390d1e4733dSdrh             pVal = Tcl_NewWideIntObj(v);
391d1e4733dSdrh           }
392d1e4733dSdrh           break;
393d1e4733dSdrh         }
394d1e4733dSdrh         case SQLITE_FLOAT: {
395d1e4733dSdrh           double r = sqlite3_value_double(pIn);
396d1e4733dSdrh           pVal = Tcl_NewDoubleObj(r);
397d1e4733dSdrh           break;
398d1e4733dSdrh         }
399d1e4733dSdrh         case SQLITE_NULL: {
400d1e4733dSdrh           pVal = Tcl_NewStringObj("", 0);
401d1e4733dSdrh           break;
402d1e4733dSdrh         }
403d1e4733dSdrh         default: {
404d1e4733dSdrh           int bytes = sqlite3_value_bytes(pIn);
40500fd957bSdanielk1977           pVal = Tcl_NewStringObj((char *)sqlite3_value_text(pIn), bytes);
406d1e4733dSdrh           break;
40751ad0ecdSdanielk1977         }
408cabb0819Sdrh       }
409d1e4733dSdrh       rc = Tcl_ListObjAppendElement(p->interp, pCmd, pVal);
410d1e4733dSdrh       if( rc ){
411d1e4733dSdrh         Tcl_DecrRefCount(pCmd);
412d1e4733dSdrh         sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1);
413d1e4733dSdrh         return;
414d1e4733dSdrh       }
415d1e4733dSdrh     }
416d1e4733dSdrh     if( !p->useEvalObjv ){
417d1e4733dSdrh       /* Tcl_EvalObjEx() will automatically call Tcl_EvalObjv() if pCmd
418d1e4733dSdrh       ** is a list without a string representation.  To prevent this from
419d1e4733dSdrh       ** happening, make sure pCmd has a valid string representation */
420d1e4733dSdrh       Tcl_GetString(pCmd);
421d1e4733dSdrh     }
422d1e4733dSdrh     rc = Tcl_EvalObjEx(p->interp, pCmd, TCL_EVAL_DIRECT);
423d1e4733dSdrh     Tcl_DecrRefCount(pCmd);
424d1e4733dSdrh   }
425562e8d3cSdanielk1977 
426c7f269d5Sdrh   if( rc && rc!=TCL_RETURN ){
4277e18c259Sdanielk1977     sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1);
428cabb0819Sdrh   }else{
429c7f269d5Sdrh     Tcl_Obj *pVar = Tcl_GetObjResult(p->interp);
430c7f269d5Sdrh     int n;
431c7f269d5Sdrh     u8 *data;
432c7f269d5Sdrh     char *zType = pVar->typePtr ? pVar->typePtr->name : "";
433c7f269d5Sdrh     char c = zType[0];
434df0bddaeSdrh     if( c=='b' && strcmp(zType,"bytearray")==0 && pVar->bytes==0 ){
435d1e4733dSdrh       /* Only return a BLOB type if the Tcl variable is a bytearray and
436df0bddaeSdrh       ** has no string representation. */
437c7f269d5Sdrh       data = Tcl_GetByteArrayFromObj(pVar, &n);
438c7f269d5Sdrh       sqlite3_result_blob(context, data, n, SQLITE_TRANSIENT);
439c7f269d5Sdrh     }else if( (c=='b' && strcmp(zType,"boolean")==0) ||
440c7f269d5Sdrh           (c=='i' && strcmp(zType,"int")==0) ){
441c7f269d5Sdrh       Tcl_GetIntFromObj(0, pVar, &n);
442c7f269d5Sdrh       sqlite3_result_int(context, n);
443c7f269d5Sdrh     }else if( c=='d' && strcmp(zType,"double")==0 ){
444c7f269d5Sdrh       double r;
445c7f269d5Sdrh       Tcl_GetDoubleFromObj(0, pVar, &r);
446c7f269d5Sdrh       sqlite3_result_double(context, r);
447df0bddaeSdrh     }else if( c=='w' && strcmp(zType,"wideInt")==0 ){
448df0bddaeSdrh       Tcl_WideInt v;
449df0bddaeSdrh       Tcl_GetWideIntFromObj(0, pVar, &v);
450df0bddaeSdrh       sqlite3_result_int64(context, v);
451c7f269d5Sdrh     }else{
45200fd957bSdanielk1977       data = (unsigned char *)Tcl_GetStringFromObj(pVar, &n);
45300fd957bSdanielk1977       sqlite3_result_text(context, (char *)data, n, SQLITE_TRANSIENT);
454c7f269d5Sdrh     }
455cabb0819Sdrh   }
456cabb0819Sdrh }
457895d7472Sdrh 
458e22a334bSdrh #ifndef SQLITE_OMIT_AUTHORIZATION
459e22a334bSdrh /*
460e22a334bSdrh ** This is the authentication function.  It appends the authentication
461e22a334bSdrh ** type code and the two arguments to zCmd[] then invokes the result
462e22a334bSdrh ** on the interpreter.  The reply is examined to determine if the
463e22a334bSdrh ** authentication fails or succeeds.
464e22a334bSdrh */
465e22a334bSdrh static int auth_callback(
466e22a334bSdrh   void *pArg,
467e22a334bSdrh   int code,
468e22a334bSdrh   const char *zArg1,
469e22a334bSdrh   const char *zArg2,
470e22a334bSdrh   const char *zArg3,
471e22a334bSdrh   const char *zArg4
472e22a334bSdrh ){
473e22a334bSdrh   char *zCode;
474e22a334bSdrh   Tcl_DString str;
475e22a334bSdrh   int rc;
476e22a334bSdrh   const char *zReply;
477e22a334bSdrh   SqliteDb *pDb = (SqliteDb*)pArg;
478e22a334bSdrh 
479e22a334bSdrh   switch( code ){
480e22a334bSdrh     case SQLITE_COPY              : zCode="SQLITE_COPY"; break;
481e22a334bSdrh     case SQLITE_CREATE_INDEX      : zCode="SQLITE_CREATE_INDEX"; break;
482e22a334bSdrh     case SQLITE_CREATE_TABLE      : zCode="SQLITE_CREATE_TABLE"; break;
483e22a334bSdrh     case SQLITE_CREATE_TEMP_INDEX : zCode="SQLITE_CREATE_TEMP_INDEX"; break;
484e22a334bSdrh     case SQLITE_CREATE_TEMP_TABLE : zCode="SQLITE_CREATE_TEMP_TABLE"; break;
485e22a334bSdrh     case SQLITE_CREATE_TEMP_TRIGGER: zCode="SQLITE_CREATE_TEMP_TRIGGER"; break;
486e22a334bSdrh     case SQLITE_CREATE_TEMP_VIEW  : zCode="SQLITE_CREATE_TEMP_VIEW"; break;
487e22a334bSdrh     case SQLITE_CREATE_TRIGGER    : zCode="SQLITE_CREATE_TRIGGER"; break;
488e22a334bSdrh     case SQLITE_CREATE_VIEW       : zCode="SQLITE_CREATE_VIEW"; break;
489e22a334bSdrh     case SQLITE_DELETE            : zCode="SQLITE_DELETE"; break;
490e22a334bSdrh     case SQLITE_DROP_INDEX        : zCode="SQLITE_DROP_INDEX"; break;
491e22a334bSdrh     case SQLITE_DROP_TABLE        : zCode="SQLITE_DROP_TABLE"; break;
492e22a334bSdrh     case SQLITE_DROP_TEMP_INDEX   : zCode="SQLITE_DROP_TEMP_INDEX"; break;
493e22a334bSdrh     case SQLITE_DROP_TEMP_TABLE   : zCode="SQLITE_DROP_TEMP_TABLE"; break;
494e22a334bSdrh     case SQLITE_DROP_TEMP_TRIGGER : zCode="SQLITE_DROP_TEMP_TRIGGER"; break;
495e22a334bSdrh     case SQLITE_DROP_TEMP_VIEW    : zCode="SQLITE_DROP_TEMP_VIEW"; break;
496e22a334bSdrh     case SQLITE_DROP_TRIGGER      : zCode="SQLITE_DROP_TRIGGER"; break;
497e22a334bSdrh     case SQLITE_DROP_VIEW         : zCode="SQLITE_DROP_VIEW"; break;
498e22a334bSdrh     case SQLITE_INSERT            : zCode="SQLITE_INSERT"; break;
499e22a334bSdrh     case SQLITE_PRAGMA            : zCode="SQLITE_PRAGMA"; break;
500e22a334bSdrh     case SQLITE_READ              : zCode="SQLITE_READ"; break;
501e22a334bSdrh     case SQLITE_SELECT            : zCode="SQLITE_SELECT"; break;
502e22a334bSdrh     case SQLITE_TRANSACTION       : zCode="SQLITE_TRANSACTION"; break;
503e22a334bSdrh     case SQLITE_UPDATE            : zCode="SQLITE_UPDATE"; break;
50481e293b4Sdrh     case SQLITE_ATTACH            : zCode="SQLITE_ATTACH"; break;
50581e293b4Sdrh     case SQLITE_DETACH            : zCode="SQLITE_DETACH"; break;
5061c8c23ccSdanielk1977     case SQLITE_ALTER_TABLE       : zCode="SQLITE_ALTER_TABLE"; break;
5071d54df88Sdanielk1977     case SQLITE_REINDEX           : zCode="SQLITE_REINDEX"; break;
508e6e04969Sdrh     case SQLITE_ANALYZE           : zCode="SQLITE_ANALYZE"; break;
509e22a334bSdrh     default                       : zCode="????"; break;
510e22a334bSdrh   }
511e22a334bSdrh   Tcl_DStringInit(&str);
512e22a334bSdrh   Tcl_DStringAppend(&str, pDb->zAuth, -1);
513e22a334bSdrh   Tcl_DStringAppendElement(&str, zCode);
514e22a334bSdrh   Tcl_DStringAppendElement(&str, zArg1 ? zArg1 : "");
515e22a334bSdrh   Tcl_DStringAppendElement(&str, zArg2 ? zArg2 : "");
516e22a334bSdrh   Tcl_DStringAppendElement(&str, zArg3 ? zArg3 : "");
517e22a334bSdrh   Tcl_DStringAppendElement(&str, zArg4 ? zArg4 : "");
518e22a334bSdrh   rc = Tcl_GlobalEval(pDb->interp, Tcl_DStringValue(&str));
519e22a334bSdrh   Tcl_DStringFree(&str);
520e22a334bSdrh   zReply = Tcl_GetStringResult(pDb->interp);
521e22a334bSdrh   if( strcmp(zReply,"SQLITE_OK")==0 ){
522e22a334bSdrh     rc = SQLITE_OK;
523e22a334bSdrh   }else if( strcmp(zReply,"SQLITE_DENY")==0 ){
524e22a334bSdrh     rc = SQLITE_DENY;
525e22a334bSdrh   }else if( strcmp(zReply,"SQLITE_IGNORE")==0 ){
526e22a334bSdrh     rc = SQLITE_IGNORE;
527e22a334bSdrh   }else{
528e22a334bSdrh     rc = 999;
529e22a334bSdrh   }
530e22a334bSdrh   return rc;
531e22a334bSdrh }
532e22a334bSdrh #endif /* SQLITE_OMIT_AUTHORIZATION */
533cabb0819Sdrh 
534cabb0819Sdrh /*
535ef2cb63eSdanielk1977 ** zText is a pointer to text obtained via an sqlite3_result_text()
536ef2cb63eSdanielk1977 ** or similar interface. This routine returns a Tcl string object,
537ef2cb63eSdanielk1977 ** reference count set to 0, containing the text. If a translation
538ef2cb63eSdanielk1977 ** between iso8859 and UTF-8 is required, it is preformed.
539ef2cb63eSdanielk1977 */
540ef2cb63eSdanielk1977 static Tcl_Obj *dbTextToObj(char const *zText){
541ef2cb63eSdanielk1977   Tcl_Obj *pVal;
542ef2cb63eSdanielk1977 #ifdef UTF_TRANSLATION_NEEDED
543ef2cb63eSdanielk1977   Tcl_DString dCol;
544ef2cb63eSdanielk1977   Tcl_DStringInit(&dCol);
545ef2cb63eSdanielk1977   Tcl_ExternalToUtfDString(NULL, zText, -1, &dCol);
546ef2cb63eSdanielk1977   pVal = Tcl_NewStringObj(Tcl_DStringValue(&dCol), -1);
547ef2cb63eSdanielk1977   Tcl_DStringFree(&dCol);
548ef2cb63eSdanielk1977 #else
549ef2cb63eSdanielk1977   pVal = Tcl_NewStringObj(zText, -1);
550ef2cb63eSdanielk1977 #endif
551ef2cb63eSdanielk1977   return pVal;
552ef2cb63eSdanielk1977 }
553ef2cb63eSdanielk1977 
554ef2cb63eSdanielk1977 /*
5551067fe11Stpoindex ** This routine reads a line of text from FILE in, stores
5561067fe11Stpoindex ** the text in memory obtained from malloc() and returns a pointer
5571067fe11Stpoindex ** to the text.  NULL is returned at end of file, or if malloc()
5581067fe11Stpoindex ** fails.
5591067fe11Stpoindex **
5601067fe11Stpoindex ** The interface is like "readline" but no command-line editing
5611067fe11Stpoindex ** is done.
5621067fe11Stpoindex **
5631067fe11Stpoindex ** copied from shell.c from '.import' command
5641067fe11Stpoindex */
5651067fe11Stpoindex static char *local_getline(char *zPrompt, FILE *in){
5661067fe11Stpoindex   char *zLine;
5671067fe11Stpoindex   int nLine;
5681067fe11Stpoindex   int n;
5691067fe11Stpoindex   int eol;
5701067fe11Stpoindex 
5711067fe11Stpoindex   nLine = 100;
5721067fe11Stpoindex   zLine = malloc( nLine );
5731067fe11Stpoindex   if( zLine==0 ) return 0;
5741067fe11Stpoindex   n = 0;
5751067fe11Stpoindex   eol = 0;
5761067fe11Stpoindex   while( !eol ){
5771067fe11Stpoindex     if( n+100>nLine ){
5781067fe11Stpoindex       nLine = nLine*2 + 100;
5791067fe11Stpoindex       zLine = realloc(zLine, nLine);
5801067fe11Stpoindex       if( zLine==0 ) return 0;
5811067fe11Stpoindex     }
5821067fe11Stpoindex     if( fgets(&zLine[n], nLine - n, in)==0 ){
5831067fe11Stpoindex       if( n==0 ){
5841067fe11Stpoindex         free(zLine);
5851067fe11Stpoindex         return 0;
5861067fe11Stpoindex       }
5871067fe11Stpoindex       zLine[n] = 0;
5881067fe11Stpoindex       eol = 1;
5891067fe11Stpoindex       break;
5901067fe11Stpoindex     }
5911067fe11Stpoindex     while( zLine[n] ){ n++; }
5921067fe11Stpoindex     if( n>0 && zLine[n-1]=='\n' ){
5931067fe11Stpoindex       n--;
5941067fe11Stpoindex       zLine[n] = 0;
5951067fe11Stpoindex       eol = 1;
5961067fe11Stpoindex     }
5971067fe11Stpoindex   }
5981067fe11Stpoindex   zLine = realloc( zLine, n+1 );
5991067fe11Stpoindex   return zLine;
6001067fe11Stpoindex }
6011067fe11Stpoindex 
6021067fe11Stpoindex /*
60375897234Sdrh ** The "sqlite" command below creates a new Tcl command for each
60475897234Sdrh ** connection it opens to an SQLite database.  This routine is invoked
60575897234Sdrh ** whenever one of those connection-specific commands is executed
60675897234Sdrh ** in Tcl.  For example, if you run Tcl code like this:
60775897234Sdrh **
6089bb575fdSdrh **       sqlite3 db1  "my_database"
60975897234Sdrh **       db1 close
61075897234Sdrh **
61175897234Sdrh ** The first command opens a connection to the "my_database" database
61275897234Sdrh ** and calls that connection "db1".  The second command causes this
61375897234Sdrh ** subroutine to be invoked.
61475897234Sdrh */
6156d31316cSdrh static int DbObjCmd(void *cd, Tcl_Interp *interp, int objc,Tcl_Obj *const*objv){
616bec3f402Sdrh   SqliteDb *pDb = (SqliteDb*)cd;
6176d31316cSdrh   int choice;
61822fbcb8dSdrh   int rc = TCL_OK;
6190de8c112Sdrh   static const char *DB_strs[] = {
620fb7e7651Sdrh     "authorizer",         "busy",              "cache",
621fb7e7651Sdrh     "changes",            "close",             "collate",
622fb7e7651Sdrh     "collation_needed",   "commit_hook",       "complete",
623fb7e7651Sdrh     "copy",               "errorcode",         "eval",
624*97f2ebc1Sdrh     "exists",             "function",          "last_insert_rowid",
625*97f2ebc1Sdrh     "nullvalue",          "onecolumn",         "profile",
626*97f2ebc1Sdrh     "progress",           "rekey",             "timeout",
627*97f2ebc1Sdrh     "total_changes",      "trace",             "transaction",
628*97f2ebc1Sdrh     "version",            0
6296d31316cSdrh   };
630411995dcSdrh   enum DB_enum {
631fb7e7651Sdrh     DB_AUTHORIZER,        DB_BUSY,             DB_CACHE,
632fb7e7651Sdrh     DB_CHANGES,           DB_CLOSE,            DB_COLLATE,
633fb7e7651Sdrh     DB_COLLATION_NEEDED,  DB_COMMIT_HOOK,      DB_COMPLETE,
634fb7e7651Sdrh     DB_COPY,              DB_ERRORCODE,        DB_EVAL,
635*97f2ebc1Sdrh     DB_EXISTS,            DB_FUNCTION,         DB_LAST_INSERT_ROWID,
636*97f2ebc1Sdrh     DB_NULLVALUE,         DB_ONECOLUMN,        DB_PROFILE,
637*97f2ebc1Sdrh     DB_PROGRESS,          DB_REKEY,            DB_TIMEOUT,
638*97f2ebc1Sdrh     DB_TOTAL_CHANGES,     DB_TRACE,            DB_TRANSACTION,
639*97f2ebc1Sdrh     DB_VERSION
6406d31316cSdrh   };
6411067fe11Stpoindex   /* don't leave trailing commas on DB_enum, it confuses the AIX xlc compiler */
6426d31316cSdrh 
6436d31316cSdrh   if( objc<2 ){
6446d31316cSdrh     Tcl_WrongNumArgs(interp, 1, objv, "SUBCOMMAND ...");
64575897234Sdrh     return TCL_ERROR;
64675897234Sdrh   }
647411995dcSdrh   if( Tcl_GetIndexFromObj(interp, objv[1], DB_strs, "option", 0, &choice) ){
6486d31316cSdrh     return TCL_ERROR;
6496d31316cSdrh   }
6506d31316cSdrh 
651411995dcSdrh   switch( (enum DB_enum)choice ){
65275897234Sdrh 
653e22a334bSdrh   /*    $db authorizer ?CALLBACK?
654e22a334bSdrh   **
655e22a334bSdrh   ** Invoke the given callback to authorize each SQL operation as it is
656e22a334bSdrh   ** compiled.  5 arguments are appended to the callback before it is
657e22a334bSdrh   ** invoked:
658e22a334bSdrh   **
659e22a334bSdrh   **   (1) The authorization type (ex: SQLITE_CREATE_TABLE, SQLITE_INSERT, ...)
660e22a334bSdrh   **   (2) First descriptive name (depends on authorization type)
661e22a334bSdrh   **   (3) Second descriptive name
662e22a334bSdrh   **   (4) Name of the database (ex: "main", "temp")
663e22a334bSdrh   **   (5) Name of trigger that is doing the access
664e22a334bSdrh   **
665e22a334bSdrh   ** The callback should return on of the following strings: SQLITE_OK,
666e22a334bSdrh   ** SQLITE_IGNORE, or SQLITE_DENY.  Any other return value is an error.
667e22a334bSdrh   **
668e22a334bSdrh   ** If this method is invoked with no arguments, the current authorization
669e22a334bSdrh   ** callback string is returned.
670e22a334bSdrh   */
671e22a334bSdrh   case DB_AUTHORIZER: {
6721211de37Sdrh #ifdef SQLITE_OMIT_AUTHORIZATION
6731211de37Sdrh     Tcl_AppendResult(interp, "authorization not available in this build", 0);
6741211de37Sdrh     return TCL_ERROR;
6751211de37Sdrh #else
676e22a334bSdrh     if( objc>3 ){
677e22a334bSdrh       Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?");
6780f14e2ebSdrh       return TCL_ERROR;
679e22a334bSdrh     }else if( objc==2 ){
680b5a20d3cSdrh       if( pDb->zAuth ){
681e22a334bSdrh         Tcl_AppendResult(interp, pDb->zAuth, 0);
682e22a334bSdrh       }
683e22a334bSdrh     }else{
684e22a334bSdrh       char *zAuth;
685e22a334bSdrh       int len;
686e22a334bSdrh       if( pDb->zAuth ){
687e22a334bSdrh         Tcl_Free(pDb->zAuth);
688e22a334bSdrh       }
689e22a334bSdrh       zAuth = Tcl_GetStringFromObj(objv[2], &len);
690e22a334bSdrh       if( zAuth && len>0 ){
691e22a334bSdrh         pDb->zAuth = Tcl_Alloc( len + 1 );
692e22a334bSdrh         strcpy(pDb->zAuth, zAuth);
693e22a334bSdrh       }else{
694e22a334bSdrh         pDb->zAuth = 0;
695e22a334bSdrh       }
696e22a334bSdrh       if( pDb->zAuth ){
697e22a334bSdrh         pDb->interp = interp;
6986f8a503dSdanielk1977         sqlite3_set_authorizer(pDb->db, auth_callback, pDb);
699e22a334bSdrh       }else{
7006f8a503dSdanielk1977         sqlite3_set_authorizer(pDb->db, 0, 0);
701e22a334bSdrh       }
702e22a334bSdrh     }
7031211de37Sdrh #endif
704e22a334bSdrh     break;
705e22a334bSdrh   }
706e22a334bSdrh 
707bec3f402Sdrh   /*    $db busy ?CALLBACK?
708bec3f402Sdrh   **
709bec3f402Sdrh   ** Invoke the given callback if an SQL statement attempts to open
710bec3f402Sdrh   ** a locked database file.
711bec3f402Sdrh   */
7126d31316cSdrh   case DB_BUSY: {
7136d31316cSdrh     if( objc>3 ){
7146d31316cSdrh       Tcl_WrongNumArgs(interp, 2, objv, "CALLBACK");
715bec3f402Sdrh       return TCL_ERROR;
7166d31316cSdrh     }else if( objc==2 ){
717bec3f402Sdrh       if( pDb->zBusy ){
718bec3f402Sdrh         Tcl_AppendResult(interp, pDb->zBusy, 0);
719bec3f402Sdrh       }
720bec3f402Sdrh     }else{
7216d31316cSdrh       char *zBusy;
7226d31316cSdrh       int len;
723bec3f402Sdrh       if( pDb->zBusy ){
724bec3f402Sdrh         Tcl_Free(pDb->zBusy);
7256d31316cSdrh       }
7266d31316cSdrh       zBusy = Tcl_GetStringFromObj(objv[2], &len);
7276d31316cSdrh       if( zBusy && len>0 ){
7286d31316cSdrh         pDb->zBusy = Tcl_Alloc( len + 1 );
7296d31316cSdrh         strcpy(pDb->zBusy, zBusy);
7306d31316cSdrh       }else{
731bec3f402Sdrh         pDb->zBusy = 0;
732bec3f402Sdrh       }
733bec3f402Sdrh       if( pDb->zBusy ){
734bec3f402Sdrh         pDb->interp = interp;
7356f8a503dSdanielk1977         sqlite3_busy_handler(pDb->db, DbBusyHandler, pDb);
7366d31316cSdrh       }else{
7376f8a503dSdanielk1977         sqlite3_busy_handler(pDb->db, 0, 0);
738bec3f402Sdrh       }
739bec3f402Sdrh     }
7406d31316cSdrh     break;
7416d31316cSdrh   }
742bec3f402Sdrh 
743fb7e7651Sdrh   /*     $db cache flush
744fb7e7651Sdrh   **     $db cache size n
745fb7e7651Sdrh   **
746fb7e7651Sdrh   ** Flush the prepared statement cache, or set the maximum number of
747fb7e7651Sdrh   ** cached statements.
748fb7e7651Sdrh   */
749fb7e7651Sdrh   case DB_CACHE: {
750fb7e7651Sdrh     char *subCmd;
751fb7e7651Sdrh     int n;
752fb7e7651Sdrh 
753fb7e7651Sdrh     if( objc<=2 ){
754fb7e7651Sdrh       Tcl_WrongNumArgs(interp, 1, objv, "cache option ?arg?");
755fb7e7651Sdrh       return TCL_ERROR;
756fb7e7651Sdrh     }
757fb7e7651Sdrh     subCmd = Tcl_GetStringFromObj( objv[2], 0 );
758fb7e7651Sdrh     if( *subCmd=='f' && strcmp(subCmd,"flush")==0 ){
759fb7e7651Sdrh       if( objc!=3 ){
760fb7e7651Sdrh         Tcl_WrongNumArgs(interp, 2, objv, "flush");
761fb7e7651Sdrh         return TCL_ERROR;
762fb7e7651Sdrh       }else{
763fb7e7651Sdrh         flushStmtCache( pDb );
764fb7e7651Sdrh       }
765fb7e7651Sdrh     }else if( *subCmd=='s' && strcmp(subCmd,"size")==0 ){
766fb7e7651Sdrh       if( objc!=4 ){
767fb7e7651Sdrh         Tcl_WrongNumArgs(interp, 2, objv, "size n");
768fb7e7651Sdrh         return TCL_ERROR;
769fb7e7651Sdrh       }else{
770fb7e7651Sdrh         if( TCL_ERROR==Tcl_GetIntFromObj(interp, objv[3], &n) ){
771fb7e7651Sdrh           Tcl_AppendResult( interp, "cannot convert \"",
772fb7e7651Sdrh                Tcl_GetStringFromObj(objv[3],0), "\" to integer", 0);
773fb7e7651Sdrh           return TCL_ERROR;
774fb7e7651Sdrh         }else{
775fb7e7651Sdrh           if( n<0 ){
776fb7e7651Sdrh             flushStmtCache( pDb );
777fb7e7651Sdrh             n = 0;
778fb7e7651Sdrh           }else if( n>MAX_PREPARED_STMTS ){
779fb7e7651Sdrh             n = MAX_PREPARED_STMTS;
780fb7e7651Sdrh           }
781fb7e7651Sdrh           pDb->maxStmt = n;
782fb7e7651Sdrh         }
783fb7e7651Sdrh       }
784fb7e7651Sdrh     }else{
785fb7e7651Sdrh       Tcl_AppendResult( interp, "bad option \"",
786fb7e7651Sdrh           Tcl_GetStringFromObj(objv[0],0), "\": must be flush or size", 0);
787fb7e7651Sdrh       return TCL_ERROR;
788fb7e7651Sdrh     }
789fb7e7651Sdrh     break;
790fb7e7651Sdrh   }
791fb7e7651Sdrh 
792b28af71aSdanielk1977   /*     $db changes
793c8d30ac1Sdrh   **
794c8d30ac1Sdrh   ** Return the number of rows that were modified, inserted, or deleted by
795b28af71aSdanielk1977   ** the most recent INSERT, UPDATE or DELETE statement, not including
796b28af71aSdanielk1977   ** any changes made by trigger programs.
797c8d30ac1Sdrh   */
798c8d30ac1Sdrh   case DB_CHANGES: {
799c8d30ac1Sdrh     Tcl_Obj *pResult;
800c8d30ac1Sdrh     if( objc!=2 ){
801c8d30ac1Sdrh       Tcl_WrongNumArgs(interp, 2, objv, "");
802c8d30ac1Sdrh       return TCL_ERROR;
803c8d30ac1Sdrh     }
804c8d30ac1Sdrh     pResult = Tcl_GetObjResult(interp);
805b28af71aSdanielk1977     Tcl_SetIntObj(pResult, sqlite3_changes(pDb->db));
806f146a776Srdc     break;
807f146a776Srdc   }
808f146a776Srdc 
80975897234Sdrh   /*    $db close
81075897234Sdrh   **
81175897234Sdrh   ** Shutdown the database
81275897234Sdrh   */
8136d31316cSdrh   case DB_CLOSE: {
8146d31316cSdrh     Tcl_DeleteCommand(interp, Tcl_GetStringFromObj(objv[0], 0));
8156d31316cSdrh     break;
8166d31316cSdrh   }
81775897234Sdrh 
8180f14e2ebSdrh   /*
8190f14e2ebSdrh   **     $db collate NAME SCRIPT
8200f14e2ebSdrh   **
8210f14e2ebSdrh   ** Create a new SQL collation function called NAME.  Whenever
8220f14e2ebSdrh   ** that function is called, invoke SCRIPT to evaluate the function.
8230f14e2ebSdrh   */
8240f14e2ebSdrh   case DB_COLLATE: {
8250f14e2ebSdrh     SqlCollate *pCollate;
8260f14e2ebSdrh     char *zName;
8270f14e2ebSdrh     char *zScript;
8280f14e2ebSdrh     int nScript;
8290f14e2ebSdrh     if( objc!=4 ){
8300f14e2ebSdrh       Tcl_WrongNumArgs(interp, 2, objv, "NAME SCRIPT");
8310f14e2ebSdrh       return TCL_ERROR;
8320f14e2ebSdrh     }
8330f14e2ebSdrh     zName = Tcl_GetStringFromObj(objv[2], 0);
8340f14e2ebSdrh     zScript = Tcl_GetStringFromObj(objv[3], &nScript);
8350f14e2ebSdrh     pCollate = (SqlCollate*)Tcl_Alloc( sizeof(*pCollate) + nScript + 1 );
8360f14e2ebSdrh     if( pCollate==0 ) return TCL_ERROR;
8370f14e2ebSdrh     pCollate->interp = interp;
8380f14e2ebSdrh     pCollate->pNext = pDb->pCollate;
8390f14e2ebSdrh     pCollate->zScript = (char*)&pCollate[1];
8400f14e2ebSdrh     pDb->pCollate = pCollate;
8410f14e2ebSdrh     strcpy(pCollate->zScript, zScript);
8420f14e2ebSdrh     if( sqlite3_create_collation(pDb->db, zName, SQLITE_UTF8,
8430f14e2ebSdrh         pCollate, tclSqlCollate) ){
8449636c4e1Sdanielk1977       Tcl_SetResult(interp, (char *)sqlite3_errmsg(pDb->db), TCL_VOLATILE);
8450f14e2ebSdrh       return TCL_ERROR;
8460f14e2ebSdrh     }
8470f14e2ebSdrh     break;
8480f14e2ebSdrh   }
8490f14e2ebSdrh 
8500f14e2ebSdrh   /*
8510f14e2ebSdrh   **     $db collation_needed SCRIPT
8520f14e2ebSdrh   **
8530f14e2ebSdrh   ** Create a new SQL collation function called NAME.  Whenever
8540f14e2ebSdrh   ** that function is called, invoke SCRIPT to evaluate the function.
8550f14e2ebSdrh   */
8560f14e2ebSdrh   case DB_COLLATION_NEEDED: {
8570f14e2ebSdrh     if( objc!=3 ){
8580f14e2ebSdrh       Tcl_WrongNumArgs(interp, 2, objv, "SCRIPT");
8590f14e2ebSdrh       return TCL_ERROR;
8600f14e2ebSdrh     }
8610f14e2ebSdrh     if( pDb->pCollateNeeded ){
8620f14e2ebSdrh       Tcl_DecrRefCount(pDb->pCollateNeeded);
8630f14e2ebSdrh     }
8640f14e2ebSdrh     pDb->pCollateNeeded = Tcl_DuplicateObj(objv[2]);
8650f14e2ebSdrh     Tcl_IncrRefCount(pDb->pCollateNeeded);
8660f14e2ebSdrh     sqlite3_collation_needed(pDb->db, pDb, tclCollateNeeded);
8670f14e2ebSdrh     break;
8680f14e2ebSdrh   }
8690f14e2ebSdrh 
87019e2d37fSdrh   /*    $db commit_hook ?CALLBACK?
87119e2d37fSdrh   **
87219e2d37fSdrh   ** Invoke the given callback just before committing every SQL transaction.
87319e2d37fSdrh   ** If the callback throws an exception or returns non-zero, then the
87419e2d37fSdrh   ** transaction is aborted.  If CALLBACK is an empty string, the callback
87519e2d37fSdrh   ** is disabled.
87619e2d37fSdrh   */
87719e2d37fSdrh   case DB_COMMIT_HOOK: {
87819e2d37fSdrh     if( objc>3 ){
87919e2d37fSdrh       Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?");
88019e2d37fSdrh       return TCL_ERROR;
88119e2d37fSdrh     }else if( objc==2 ){
88219e2d37fSdrh       if( pDb->zCommit ){
88319e2d37fSdrh         Tcl_AppendResult(interp, pDb->zCommit, 0);
88419e2d37fSdrh       }
88519e2d37fSdrh     }else{
88619e2d37fSdrh       char *zCommit;
88719e2d37fSdrh       int len;
88819e2d37fSdrh       if( pDb->zCommit ){
88919e2d37fSdrh         Tcl_Free(pDb->zCommit);
89019e2d37fSdrh       }
89119e2d37fSdrh       zCommit = Tcl_GetStringFromObj(objv[2], &len);
89219e2d37fSdrh       if( zCommit && len>0 ){
89319e2d37fSdrh         pDb->zCommit = Tcl_Alloc( len + 1 );
89419e2d37fSdrh         strcpy(pDb->zCommit, zCommit);
89519e2d37fSdrh       }else{
89619e2d37fSdrh         pDb->zCommit = 0;
89719e2d37fSdrh       }
89819e2d37fSdrh       if( pDb->zCommit ){
89919e2d37fSdrh         pDb->interp = interp;
90019e2d37fSdrh         sqlite3_commit_hook(pDb->db, DbCommitHandler, pDb);
90119e2d37fSdrh       }else{
90219e2d37fSdrh         sqlite3_commit_hook(pDb->db, 0, 0);
90319e2d37fSdrh       }
90419e2d37fSdrh     }
90519e2d37fSdrh     break;
90619e2d37fSdrh   }
90719e2d37fSdrh 
90875897234Sdrh   /*    $db complete SQL
90975897234Sdrh   **
91075897234Sdrh   ** Return TRUE if SQL is a complete SQL statement.  Return FALSE if
91175897234Sdrh   ** additional lines of input are needed.  This is similar to the
91275897234Sdrh   ** built-in "info complete" command of Tcl.
91375897234Sdrh   */
9146d31316cSdrh   case DB_COMPLETE: {
915ccae6026Sdrh #ifndef SQLITE_OMIT_COMPLETE
9166d31316cSdrh     Tcl_Obj *pResult;
9176d31316cSdrh     int isComplete;
9186d31316cSdrh     if( objc!=3 ){
9196d31316cSdrh       Tcl_WrongNumArgs(interp, 2, objv, "SQL");
92075897234Sdrh       return TCL_ERROR;
92175897234Sdrh     }
9226f8a503dSdanielk1977     isComplete = sqlite3_complete( Tcl_GetStringFromObj(objv[2], 0) );
9236d31316cSdrh     pResult = Tcl_GetObjResult(interp);
9246d31316cSdrh     Tcl_SetBooleanObj(pResult, isComplete);
925ccae6026Sdrh #endif
9266d31316cSdrh     break;
9276d31316cSdrh   }
92875897234Sdrh 
92919e2d37fSdrh   /*    $db copy conflict-algorithm table filename ?SEPARATOR? ?NULLINDICATOR?
93019e2d37fSdrh   **
93119e2d37fSdrh   ** Copy data into table from filename, optionally using SEPARATOR
93219e2d37fSdrh   ** as column separators.  If a column contains a null string, or the
93319e2d37fSdrh   ** value of NULLINDICATOR, a NULL is inserted for the column.
93419e2d37fSdrh   ** conflict-algorithm is one of the sqlite conflict algorithms:
93519e2d37fSdrh   **    rollback, abort, fail, ignore, replace
93619e2d37fSdrh   ** On success, return the number of lines processed, not necessarily same
93719e2d37fSdrh   ** as 'db changes' due to conflict-algorithm selected.
93819e2d37fSdrh   **
93919e2d37fSdrh   ** This code is basically an implementation/enhancement of
94019e2d37fSdrh   ** the sqlite3 shell.c ".import" command.
94119e2d37fSdrh   **
94219e2d37fSdrh   ** This command usage is equivalent to the sqlite2.x COPY statement,
94319e2d37fSdrh   ** which imports file data into a table using the PostgreSQL COPY file format:
94419e2d37fSdrh   **   $db copy $conflit_algo $table_name $filename \t \\N
94519e2d37fSdrh   */
94619e2d37fSdrh   case DB_COPY: {
94719e2d37fSdrh     char *zTable;               /* Insert data into this table */
94819e2d37fSdrh     char *zFile;                /* The file from which to extract data */
94919e2d37fSdrh     char *zConflict;            /* The conflict algorithm to use */
95019e2d37fSdrh     sqlite3_stmt *pStmt;        /* A statement */
95119e2d37fSdrh     int rc;                     /* Result code */
95219e2d37fSdrh     int nCol;                   /* Number of columns in the table */
95319e2d37fSdrh     int nByte;                  /* Number of bytes in an SQL string */
95419e2d37fSdrh     int i, j;                   /* Loop counters */
95519e2d37fSdrh     int nSep;                   /* Number of bytes in zSep[] */
95619e2d37fSdrh     int nNull;                  /* Number of bytes in zNull[] */
95719e2d37fSdrh     char *zSql;                 /* An SQL statement */
95819e2d37fSdrh     char *zLine;                /* A single line of input from the file */
95919e2d37fSdrh     char **azCol;               /* zLine[] broken up into columns */
96019e2d37fSdrh     char *zCommit;              /* How to commit changes */
96119e2d37fSdrh     FILE *in;                   /* The input file */
96219e2d37fSdrh     int lineno = 0;             /* Line number of input file */
96319e2d37fSdrh     char zLineNum[80];          /* Line number print buffer */
96419e2d37fSdrh     Tcl_Obj *pResult;           /* interp result */
96519e2d37fSdrh 
96619e2d37fSdrh     char *zSep;
96719e2d37fSdrh     char *zNull;
96819e2d37fSdrh     if( objc<5 || objc>7 ){
96919e2d37fSdrh       Tcl_WrongNumArgs(interp, 2, objv,
97019e2d37fSdrh          "CONFLICT-ALGORITHM TABLE FILENAME ?SEPARATOR? ?NULLINDICATOR?");
97119e2d37fSdrh       return TCL_ERROR;
97219e2d37fSdrh     }
97319e2d37fSdrh     if( objc>=6 ){
97419e2d37fSdrh       zSep = Tcl_GetStringFromObj(objv[5], 0);
97519e2d37fSdrh     }else{
97619e2d37fSdrh       zSep = "\t";
97719e2d37fSdrh     }
97819e2d37fSdrh     if( objc>=7 ){
97919e2d37fSdrh       zNull = Tcl_GetStringFromObj(objv[6], 0);
98019e2d37fSdrh     }else{
98119e2d37fSdrh       zNull = "";
98219e2d37fSdrh     }
98319e2d37fSdrh     zConflict = Tcl_GetStringFromObj(objv[2], 0);
98419e2d37fSdrh     zTable = Tcl_GetStringFromObj(objv[3], 0);
98519e2d37fSdrh     zFile = Tcl_GetStringFromObj(objv[4], 0);
98619e2d37fSdrh     nSep = strlen(zSep);
98719e2d37fSdrh     nNull = strlen(zNull);
98819e2d37fSdrh     if( nSep==0 ){
98919e2d37fSdrh       Tcl_AppendResult(interp, "Error: non-null separator required for copy", 0);
99019e2d37fSdrh       return TCL_ERROR;
99119e2d37fSdrh     }
99219e2d37fSdrh     if(sqlite3StrICmp(zConflict, "rollback") != 0 &&
99319e2d37fSdrh        sqlite3StrICmp(zConflict, "abort"   ) != 0 &&
99419e2d37fSdrh        sqlite3StrICmp(zConflict, "fail"    ) != 0 &&
99519e2d37fSdrh        sqlite3StrICmp(zConflict, "ignore"  ) != 0 &&
99619e2d37fSdrh        sqlite3StrICmp(zConflict, "replace" ) != 0 ) {
99719e2d37fSdrh       Tcl_AppendResult(interp, "Error: \"", zConflict,
99819e2d37fSdrh             "\", conflict-algorithm must be one of: rollback, "
99919e2d37fSdrh             "abort, fail, ignore, or replace", 0);
100019e2d37fSdrh       return TCL_ERROR;
100119e2d37fSdrh     }
100219e2d37fSdrh     zSql = sqlite3_mprintf("SELECT * FROM '%q'", zTable);
100319e2d37fSdrh     if( zSql==0 ){
100419e2d37fSdrh       Tcl_AppendResult(interp, "Error: no such table: ", zTable, 0);
100519e2d37fSdrh       return TCL_ERROR;
100619e2d37fSdrh     }
100719e2d37fSdrh     nByte = strlen(zSql);
100819e2d37fSdrh     rc = sqlite3_prepare(pDb->db, zSql, 0, &pStmt, 0);
100919e2d37fSdrh     sqlite3_free(zSql);
101019e2d37fSdrh     if( rc ){
101119e2d37fSdrh       Tcl_AppendResult(interp, "Error: ", sqlite3_errmsg(pDb->db), 0);
101219e2d37fSdrh       nCol = 0;
101319e2d37fSdrh     }else{
101419e2d37fSdrh       nCol = sqlite3_column_count(pStmt);
101519e2d37fSdrh     }
101619e2d37fSdrh     sqlite3_finalize(pStmt);
101719e2d37fSdrh     if( nCol==0 ) {
101819e2d37fSdrh       return TCL_ERROR;
101919e2d37fSdrh     }
102019e2d37fSdrh     zSql = malloc( nByte + 50 + nCol*2 );
102119e2d37fSdrh     if( zSql==0 ) {
102219e2d37fSdrh       Tcl_AppendResult(interp, "Error: can't malloc()", 0);
102319e2d37fSdrh       return TCL_ERROR;
102419e2d37fSdrh     }
102519e2d37fSdrh     sqlite3_snprintf(nByte+50, zSql, "INSERT OR %q INTO '%q' VALUES(?",
102619e2d37fSdrh          zConflict, zTable);
102719e2d37fSdrh     j = strlen(zSql);
102819e2d37fSdrh     for(i=1; i<nCol; i++){
102919e2d37fSdrh       zSql[j++] = ',';
103019e2d37fSdrh       zSql[j++] = '?';
103119e2d37fSdrh     }
103219e2d37fSdrh     zSql[j++] = ')';
103319e2d37fSdrh     zSql[j] = 0;
103419e2d37fSdrh     rc = sqlite3_prepare(pDb->db, zSql, 0, &pStmt, 0);
103519e2d37fSdrh     free(zSql);
103619e2d37fSdrh     if( rc ){
103719e2d37fSdrh       Tcl_AppendResult(interp, "Error: ", sqlite3_errmsg(pDb->db), 0);
103819e2d37fSdrh       sqlite3_finalize(pStmt);
103919e2d37fSdrh       return TCL_ERROR;
104019e2d37fSdrh     }
104119e2d37fSdrh     in = fopen(zFile, "rb");
104219e2d37fSdrh     if( in==0 ){
104319e2d37fSdrh       Tcl_AppendResult(interp, "Error: cannot open file: ", zFile, NULL);
104419e2d37fSdrh       sqlite3_finalize(pStmt);
104519e2d37fSdrh       return TCL_ERROR;
104619e2d37fSdrh     }
104719e2d37fSdrh     azCol = malloc( sizeof(azCol[0])*(nCol+1) );
104819e2d37fSdrh     if( azCol==0 ) {
104919e2d37fSdrh       Tcl_AppendResult(interp, "Error: can't malloc()", 0);
105019e2d37fSdrh       return TCL_ERROR;
105119e2d37fSdrh     }
105219e2d37fSdrh     sqlite3_exec(pDb->db, "BEGIN", 0, 0, 0);
105319e2d37fSdrh     zCommit = "COMMIT";
105419e2d37fSdrh     while( (zLine = local_getline(0, in))!=0 ){
105519e2d37fSdrh       char *z;
105619e2d37fSdrh       i = 0;
105719e2d37fSdrh       lineno++;
105819e2d37fSdrh       azCol[0] = zLine;
105919e2d37fSdrh       for(i=0, z=zLine; *z; z++){
106019e2d37fSdrh         if( *z==zSep[0] && strncmp(z, zSep, nSep)==0 ){
106119e2d37fSdrh           *z = 0;
106219e2d37fSdrh           i++;
106319e2d37fSdrh           if( i<nCol ){
106419e2d37fSdrh             azCol[i] = &z[nSep];
106519e2d37fSdrh             z += nSep-1;
106619e2d37fSdrh           }
106719e2d37fSdrh         }
106819e2d37fSdrh       }
106919e2d37fSdrh       if( i+1!=nCol ){
107019e2d37fSdrh         char *zErr;
107119e2d37fSdrh         zErr = malloc(200 + strlen(zFile));
107219e2d37fSdrh         sprintf(zErr,"Error: %s line %d: expected %d columns of data but found %d",
107319e2d37fSdrh            zFile, lineno, nCol, i+1);
107419e2d37fSdrh         Tcl_AppendResult(interp, zErr, 0);
107519e2d37fSdrh         free(zErr);
107619e2d37fSdrh         zCommit = "ROLLBACK";
107719e2d37fSdrh         break;
107819e2d37fSdrh       }
107919e2d37fSdrh       for(i=0; i<nCol; i++){
108019e2d37fSdrh         /* check for null data, if so, bind as null */
108119e2d37fSdrh         if ((nNull>0 && strcmp(azCol[i], zNull)==0) || strlen(azCol[i])==0) {
108219e2d37fSdrh           sqlite3_bind_null(pStmt, i+1);
108319e2d37fSdrh         }else{
108419e2d37fSdrh           sqlite3_bind_text(pStmt, i+1, azCol[i], -1, SQLITE_STATIC);
108519e2d37fSdrh         }
108619e2d37fSdrh       }
108719e2d37fSdrh       sqlite3_step(pStmt);
108819e2d37fSdrh       rc = sqlite3_reset(pStmt);
108919e2d37fSdrh       free(zLine);
109019e2d37fSdrh       if( rc!=SQLITE_OK ){
109119e2d37fSdrh         Tcl_AppendResult(interp,"Error: ", sqlite3_errmsg(pDb->db), 0);
109219e2d37fSdrh         zCommit = "ROLLBACK";
109319e2d37fSdrh         break;
109419e2d37fSdrh       }
109519e2d37fSdrh     }
109619e2d37fSdrh     free(azCol);
109719e2d37fSdrh     fclose(in);
109819e2d37fSdrh     sqlite3_finalize(pStmt);
109919e2d37fSdrh     sqlite3_exec(pDb->db, zCommit, 0, 0, 0);
110019e2d37fSdrh 
110119e2d37fSdrh     if( zCommit[0] == 'C' ){
110219e2d37fSdrh       /* success, set result as number of lines processed */
110319e2d37fSdrh       pResult = Tcl_GetObjResult(interp);
110419e2d37fSdrh       Tcl_SetIntObj(pResult, lineno);
110519e2d37fSdrh       rc = TCL_OK;
110619e2d37fSdrh     }else{
110719e2d37fSdrh       /* failure, append lineno where failed */
110819e2d37fSdrh       sprintf(zLineNum,"%d",lineno);
110919e2d37fSdrh       Tcl_AppendResult(interp,", failed while processing line: ",zLineNum,0);
111019e2d37fSdrh       rc = TCL_ERROR;
111119e2d37fSdrh     }
111219e2d37fSdrh     break;
111319e2d37fSdrh   }
111419e2d37fSdrh 
111575897234Sdrh   /*
1116dcd997eaSdrh   **    $db errorcode
1117dcd997eaSdrh   **
1118dcd997eaSdrh   ** Return the numeric error code that was returned by the most recent
11196f8a503dSdanielk1977   ** call to sqlite3_exec().
1120dcd997eaSdrh   */
1121dcd997eaSdrh   case DB_ERRORCODE: {
1122f3ce83f5Sdanielk1977     Tcl_SetObjResult(interp, Tcl_NewIntObj(sqlite3_errcode(pDb->db)));
1123dcd997eaSdrh     break;
1124dcd997eaSdrh   }
1125dcd997eaSdrh 
1126dcd997eaSdrh   /*
1127895d7472Sdrh   **    $db eval $sql ?array? ?{  ...code... }?
11281807ce37Sdrh   **    $db onecolumn $sql
112975897234Sdrh   **
113075897234Sdrh   ** The SQL statement in $sql is evaluated.  For each row, the values are
1131bec3f402Sdrh   ** placed in elements of the array named "array" and ...code... is executed.
113275897234Sdrh   ** If "array" and "code" are omitted, then no callback is every invoked.
113375897234Sdrh   ** If "array" is an empty string, then the values are placed in variables
113475897234Sdrh   ** that have the same name as the fields extracted by the query.
11351807ce37Sdrh   **
11361807ce37Sdrh   ** The onecolumn method is the equivalent of:
11371807ce37Sdrh   **     lindex [$db eval $sql] 0
113875897234Sdrh   */
11391807ce37Sdrh   case DB_ONECOLUMN:
1140*97f2ebc1Sdrh   case DB_EVAL:
1141*97f2ebc1Sdrh   case DB_EXISTS: {
11429d74b4c5Sdrh     char const *zSql;      /* Next SQL statement to execute */
11439d74b4c5Sdrh     char const *zLeft;     /* What is left after first stmt in zSql */
11449d74b4c5Sdrh     sqlite3_stmt *pStmt;   /* Compiled SQL statment */
114592febd92Sdrh     Tcl_Obj *pArray;       /* Name of array into which results are written */
114692febd92Sdrh     Tcl_Obj *pScript;      /* Script to run for each result set */
11471d895039Sdrh     Tcl_Obj **apParm;      /* Parameters that need a Tcl_DecrRefCount() */
11481d895039Sdrh     int nParm;             /* Number of entries used in apParm[] */
11491d895039Sdrh     Tcl_Obj *aParm[10];    /* Static space for apParm[] in the common case */
11501807ce37Sdrh     Tcl_Obj *pRet;         /* Value to be returned */
1151fb7e7651Sdrh     SqlPreparedStmt *pPreStmt;  /* Pointer to a prepared statement */
1152fb7e7651Sdrh     int rc2;
1153ef2cb63eSdanielk1977 
1154*97f2ebc1Sdrh     if( choice==DB_EVAL ){
115592febd92Sdrh       if( objc<3 || objc>5 ){
1156895d7472Sdrh         Tcl_WrongNumArgs(interp, 2, objv, "SQL ?ARRAY-NAME? ?SCRIPT?");
115730ccda10Sdanielk1977         return TCL_ERROR;
115830ccda10Sdanielk1977       }
11591807ce37Sdrh       pRet = Tcl_NewObj();
11601807ce37Sdrh       Tcl_IncrRefCount(pRet);
1161*97f2ebc1Sdrh     }else{
1162*97f2ebc1Sdrh       if( objc!=3 ){
1163*97f2ebc1Sdrh         Tcl_WrongNumArgs(interp, 2, objv, "SQL");
1164*97f2ebc1Sdrh         return TCL_ERROR;
1165*97f2ebc1Sdrh       }
1166*97f2ebc1Sdrh       pRet = 0;
1167*97f2ebc1Sdrh       if( choice==DB_EXISTS ){
1168*97f2ebc1Sdrh         Tcl_SetObjResult(interp, Tcl_NewBooleanObj(0));
1169*97f2ebc1Sdrh       }
11701807ce37Sdrh     }
117192febd92Sdrh     if( objc==3 ){
117292febd92Sdrh       pArray = pScript = 0;
117392febd92Sdrh     }else if( objc==4 ){
117492febd92Sdrh       pArray = 0;
117592febd92Sdrh       pScript = objv[3];
117692febd92Sdrh     }else{
117792febd92Sdrh       pArray = objv[3];
117892febd92Sdrh       if( Tcl_GetString(pArray)[0]==0 ) pArray = 0;
117992febd92Sdrh       pScript = objv[4];
118092febd92Sdrh     }
118130ccda10Sdanielk1977 
11821d895039Sdrh     Tcl_IncrRefCount(objv[2]);
118330ccda10Sdanielk1977     zSql = Tcl_GetStringFromObj(objv[2], 0);
118490b6bb19Sdrh     while( rc==TCL_OK && zSql[0] ){
118592febd92Sdrh       int i;                     /* Loop counter */
1186fb7e7651Sdrh       int nVar;                  /* Number of bind parameters in the pStmt */
118792febd92Sdrh       int nCol;                  /* Number of columns in the result set */
118892febd92Sdrh       Tcl_Obj **apColName = 0;   /* Array of column names */
1189fb7e7651Sdrh       int len;                   /* String length of zSql */
119030ccda10Sdanielk1977 
1191fb7e7651Sdrh       /* Try to find a SQL statement that has already been compiled and
1192fb7e7651Sdrh       ** which matches the next sequence of SQL.
1193fb7e7651Sdrh       */
1194fb7e7651Sdrh       pStmt = 0;
1195fb7e7651Sdrh       pPreStmt = pDb->stmtList;
1196fb7e7651Sdrh       len = strlen(zSql);
1197fb7e7651Sdrh       if( pPreStmt && sqlite3_expired(pPreStmt->pStmt) ){
1198fb7e7651Sdrh         flushStmtCache(pDb);
1199fb7e7651Sdrh         pPreStmt = 0;
1200fb7e7651Sdrh       }
1201fb7e7651Sdrh       for(; pPreStmt; pPreStmt=pPreStmt->pNext){
1202fb7e7651Sdrh         int n = pPreStmt->nSql;
1203fb7e7651Sdrh         if( len>=n
1204fb7e7651Sdrh             && memcmp(pPreStmt->zSql, zSql, n)==0
1205fb7e7651Sdrh             && (zSql[n]==0 || zSql[n-1]==';')
1206fb7e7651Sdrh         ){
1207fb7e7651Sdrh           pStmt = pPreStmt->pStmt;
1208fb7e7651Sdrh           zLeft = &zSql[pPreStmt->nSql];
1209fb7e7651Sdrh 
1210fb7e7651Sdrh           /* When a prepared statement is found, unlink it from the
1211fb7e7651Sdrh           ** cache list.  It will later be added back to the beginning
1212fb7e7651Sdrh           ** of the cache list in order to implement LRU replacement.
1213fb7e7651Sdrh           */
1214fb7e7651Sdrh           if( pPreStmt->pPrev ){
1215fb7e7651Sdrh             pPreStmt->pPrev->pNext = pPreStmt->pNext;
1216fb7e7651Sdrh           }else{
1217fb7e7651Sdrh             pDb->stmtList = pPreStmt->pNext;
1218fb7e7651Sdrh           }
1219fb7e7651Sdrh           if( pPreStmt->pNext ){
1220fb7e7651Sdrh             pPreStmt->pNext->pPrev = pPreStmt->pPrev;
1221fb7e7651Sdrh           }else{
1222fb7e7651Sdrh             pDb->stmtLast = pPreStmt->pPrev;
1223fb7e7651Sdrh           }
1224fb7e7651Sdrh           pDb->nStmt--;
1225fb7e7651Sdrh           break;
1226fb7e7651Sdrh         }
1227fb7e7651Sdrh       }
1228fb7e7651Sdrh 
1229fb7e7651Sdrh       /* If no prepared statement was found.  Compile the SQL text
1230fb7e7651Sdrh       */
1231fb7e7651Sdrh       if( pStmt==0 ){
123230ccda10Sdanielk1977         if( SQLITE_OK!=sqlite3_prepare(pDb->db, zSql, -1, &pStmt, &zLeft) ){
1233ef2cb63eSdanielk1977           Tcl_SetObjResult(interp, dbTextToObj(sqlite3_errmsg(pDb->db)));
123430ccda10Sdanielk1977           rc = TCL_ERROR;
123530ccda10Sdanielk1977           break;
123630ccda10Sdanielk1977         }
123792febd92Sdrh         if( pStmt==0 ){
123892febd92Sdrh           if( SQLITE_OK!=sqlite3_errcode(pDb->db) ){
1239fb7e7651Sdrh             /* A compile-time error in the statement
1240fb7e7651Sdrh             */
124192febd92Sdrh             Tcl_SetObjResult(interp, dbTextToObj(sqlite3_errmsg(pDb->db)));
124292febd92Sdrh             rc = TCL_ERROR;
124392febd92Sdrh             break;
124492febd92Sdrh           }else{
1245fb7e7651Sdrh             /* The statement was a no-op.  Continue to the next statement
1246fb7e7651Sdrh             ** in the SQL string.
1247fb7e7651Sdrh             */
124892febd92Sdrh             zSql = zLeft;
124992febd92Sdrh             continue;
125092febd92Sdrh           }
125192febd92Sdrh         }
1252fb7e7651Sdrh         assert( pPreStmt==0 );
1253fb7e7651Sdrh       }
125430ccda10Sdanielk1977 
1255fb7e7651Sdrh       /* Bind values to parameters that begin with $ or :
1256fb7e7651Sdrh       */
125792febd92Sdrh       nVar = sqlite3_bind_parameter_count(pStmt);
12581d895039Sdrh       nParm = 0;
12591d895039Sdrh       if( nVar>sizeof(aParm)/sizeof(aParm[0]) ){
12601d895039Sdrh         apParm = (Tcl_Obj**)Tcl_Alloc(nVar*sizeof(apParm[0]));
12611d895039Sdrh       }else{
12621d895039Sdrh         apParm = aParm;
12631d895039Sdrh       }
126492febd92Sdrh       for(i=1; i<=nVar; i++){
126592febd92Sdrh         const char *zVar = sqlite3_bind_parameter_name(pStmt, i);
1266c8f9079cSdrh         if( zVar!=0 && (zVar[0]=='$' || zVar[0]==':') ){
126792febd92Sdrh           Tcl_Obj *pVar = Tcl_GetVar2Ex(interp, &zVar[1], 0, 0);
126892febd92Sdrh           if( pVar ){
126992febd92Sdrh             int n;
127092febd92Sdrh             u8 *data;
127192febd92Sdrh             char *zType = pVar->typePtr ? pVar->typePtr->name : "";
127292febd92Sdrh             char c = zType[0];
1273df0bddaeSdrh             if( c=='b' && strcmp(zType,"bytearray")==0 && pVar->bytes==0 ){
1274df0bddaeSdrh               /* Only load a BLOB type if the Tcl variable is a bytearray and
1275df0bddaeSdrh               ** has no string representation. */
127692febd92Sdrh               data = Tcl_GetByteArrayFromObj(pVar, &n);
127792febd92Sdrh               sqlite3_bind_blob(pStmt, i, data, n, SQLITE_STATIC);
12781d895039Sdrh               Tcl_IncrRefCount(pVar);
12791d895039Sdrh               apParm[nParm++] = pVar;
128092febd92Sdrh             }else if( (c=='b' && strcmp(zType,"boolean")==0) ||
128192febd92Sdrh                   (c=='i' && strcmp(zType,"int")==0) ){
128292febd92Sdrh               Tcl_GetIntFromObj(interp, pVar, &n);
128392febd92Sdrh               sqlite3_bind_int(pStmt, i, n);
128492febd92Sdrh             }else if( c=='d' && strcmp(zType,"double")==0 ){
128592febd92Sdrh               double r;
128692febd92Sdrh               Tcl_GetDoubleFromObj(interp, pVar, &r);
128792febd92Sdrh               sqlite3_bind_double(pStmt, i, r);
1288df0bddaeSdrh             }else if( c=='w' && strcmp(zType,"wideInt")==0 ){
1289df0bddaeSdrh               Tcl_WideInt v;
1290df0bddaeSdrh               Tcl_GetWideIntFromObj(interp, pVar, &v);
1291df0bddaeSdrh               sqlite3_bind_int64(pStmt, i, v);
129292febd92Sdrh             }else{
129300fd957bSdanielk1977               data = (unsigned char *)Tcl_GetStringFromObj(pVar, &n);
129400fd957bSdanielk1977               sqlite3_bind_text(pStmt, i, (char *)data, n, SQLITE_STATIC);
12951d895039Sdrh               Tcl_IncrRefCount(pVar);
12961d895039Sdrh               apParm[nParm++] = pVar;
129792febd92Sdrh             }
1298fb7e7651Sdrh           }else{
1299fb7e7651Sdrh             sqlite3_bind_null( pStmt, i );
130092febd92Sdrh           }
130192febd92Sdrh         }
130292febd92Sdrh       }
130392febd92Sdrh 
130492febd92Sdrh       /* Compute column names */
130592febd92Sdrh       nCol = sqlite3_column_count(pStmt);
130692febd92Sdrh       if( pScript ){
130792febd92Sdrh         apColName = (Tcl_Obj**)Tcl_Alloc( sizeof(Tcl_Obj*)*nCol );
130892febd92Sdrh         if( apColName==0 ) break;
130992febd92Sdrh         for(i=0; i<nCol; i++){
131092febd92Sdrh           apColName[i] = dbTextToObj(sqlite3_column_name(pStmt,i));
131192febd92Sdrh           Tcl_IncrRefCount(apColName[i]);
131292febd92Sdrh         }
131392febd92Sdrh       }
131492febd92Sdrh 
131592febd92Sdrh       /* If results are being stored in an array variable, then create
131692febd92Sdrh       ** the array(*) entry for that array
131792febd92Sdrh       */
131892febd92Sdrh       if( pArray ){
131930ccda10Sdanielk1977         Tcl_Obj *pColList = Tcl_NewObj();
13203ced14a6Sdrh         Tcl_Obj *pStar = Tcl_NewStringObj("*", -1);
132130ccda10Sdanielk1977         Tcl_IncrRefCount(pColList);
132292febd92Sdrh         for(i=0; i<nCol; i++){
132392febd92Sdrh           Tcl_ListObjAppendElement(interp, pColList, apColName[i]);
132430ccda10Sdanielk1977         }
13253ced14a6Sdrh         Tcl_ObjSetVar2(interp, pArray, pStar, pColList,0);
13263ced14a6Sdrh         Tcl_DecrRefCount(pColList);
13273ced14a6Sdrh         Tcl_DecrRefCount(pStar);
132830ccda10Sdanielk1977       }
132930ccda10Sdanielk1977 
133092febd92Sdrh       /* Execute the SQL
133192febd92Sdrh       */
133290b6bb19Sdrh       while( rc==TCL_OK && pStmt && SQLITE_ROW==sqlite3_step(pStmt) ){
133392febd92Sdrh         for(i=0; i<nCol; i++){
133430ccda10Sdanielk1977           Tcl_Obj *pVal;
133530ccda10Sdanielk1977 
133630ccda10Sdanielk1977           /* Set pVal to contain the i'th column of this row. */
133792febd92Sdrh           switch( sqlite3_column_type(pStmt, i) ){
133892febd92Sdrh             case SQLITE_BLOB: {
13393fd0a736Sdanielk1977               int bytes = sqlite3_column_bytes(pStmt, i);
13403fd0a736Sdanielk1977               pVal = Tcl_NewByteArrayObj(sqlite3_column_blob(pStmt, i), bytes);
134192febd92Sdrh               break;
134292febd92Sdrh             }
134392febd92Sdrh             case SQLITE_INTEGER: {
134492febd92Sdrh               sqlite_int64 v = sqlite3_column_int64(pStmt, i);
134592febd92Sdrh               if( v>=-2147483647 && v<=2147483647 ){
134692febd92Sdrh                 pVal = Tcl_NewIntObj(v);
134792febd92Sdrh               }else{
134892febd92Sdrh                 pVal = Tcl_NewWideIntObj(v);
134992febd92Sdrh               }
135092febd92Sdrh               break;
135192febd92Sdrh             }
135292febd92Sdrh             case SQLITE_FLOAT: {
135392febd92Sdrh               double r = sqlite3_column_double(pStmt, i);
135492febd92Sdrh               pVal = Tcl_NewDoubleObj(r);
135592febd92Sdrh               break;
135692febd92Sdrh             }
135755c45f2eSdanielk1977             case SQLITE_NULL: {
135855c45f2eSdanielk1977               pVal = dbTextToObj(pDb->zNull);
135955c45f2eSdanielk1977               break;
136055c45f2eSdanielk1977             }
136192febd92Sdrh             default: {
136200fd957bSdanielk1977               pVal = dbTextToObj((char *)sqlite3_column_text(pStmt, i));
136392febd92Sdrh               break;
136492febd92Sdrh             }
136530ccda10Sdanielk1977           }
136630ccda10Sdanielk1977 
136792febd92Sdrh           if( pScript ){
136892febd92Sdrh             if( pArray==0 ){
136992febd92Sdrh               Tcl_ObjSetVar2(interp, apColName[i], 0, pVal, 0);
137030ccda10Sdanielk1977             }else{
137192febd92Sdrh               Tcl_ObjSetVar2(interp, pArray, apColName[i], pVal, 0);
137230ccda10Sdanielk1977             }
13731807ce37Sdrh           }else if( choice==DB_ONECOLUMN ){
1374*97f2ebc1Sdrh             assert( pRet==0 );
13751807ce37Sdrh             if( pRet==0 ){
13761807ce37Sdrh               pRet = pVal;
13771807ce37Sdrh               Tcl_IncrRefCount(pRet);
13781807ce37Sdrh             }
137990b6bb19Sdrh             rc = TCL_BREAK;
1380*97f2ebc1Sdrh             i = nCol;
1381*97f2ebc1Sdrh           }else if( choice==DB_EXISTS ){
1382*97f2ebc1Sdrh             assert( pRet==0 );
1383*97f2ebc1Sdrh             Tcl_SetObjResult(interp, Tcl_NewBooleanObj(1));
1384*97f2ebc1Sdrh             rc = TCL_BREAK;
1385*97f2ebc1Sdrh             i = nCol;
138630ccda10Sdanielk1977           }else{
138730ccda10Sdanielk1977             Tcl_ListObjAppendElement(interp, pRet, pVal);
138830ccda10Sdanielk1977           }
138930ccda10Sdanielk1977         }
139030ccda10Sdanielk1977 
139192febd92Sdrh         if( pScript ){
139292febd92Sdrh           rc = Tcl_EvalObjEx(interp, pScript, 0);
139390b6bb19Sdrh           if( rc==TCL_CONTINUE ){
139490b6bb19Sdrh             rc = TCL_OK;
139530ccda10Sdanielk1977           }
139630ccda10Sdanielk1977         }
139790b6bb19Sdrh       }
139890b6bb19Sdrh       if( rc==TCL_BREAK ){
139990b6bb19Sdrh         rc = TCL_OK;
140090b6bb19Sdrh       }
140130ccda10Sdanielk1977 
140292febd92Sdrh       /* Free the column name objects */
140392febd92Sdrh       if( pScript ){
140492febd92Sdrh         for(i=0; i<nCol; i++){
140592febd92Sdrh           Tcl_DecrRefCount(apColName[i]);
140692febd92Sdrh         }
140792febd92Sdrh         Tcl_Free((char*)apColName);
140892febd92Sdrh       }
140992febd92Sdrh 
14101d895039Sdrh       /* Free the bound string and blob parameters */
14111d895039Sdrh       for(i=0; i<nParm; i++){
14121d895039Sdrh         Tcl_DecrRefCount(apParm[i]);
14131d895039Sdrh       }
14141d895039Sdrh       if( apParm!=aParm ){
14151d895039Sdrh         Tcl_Free((char*)apParm);
14161d895039Sdrh       }
14171d895039Sdrh 
1418fb7e7651Sdrh       /* Reset the statement.  If the result code is SQLITE_SCHEMA, then
1419fb7e7651Sdrh       ** flush the statement cache and try the statement again.
142092febd92Sdrh       */
1421fb7e7651Sdrh       rc2 = sqlite3_reset(pStmt);
1422fb7e7651Sdrh       if( SQLITE_SCHEMA==rc2 ){
1423fb7e7651Sdrh         /* After a schema change, flush the cache and try to run the
1424fb7e7651Sdrh         ** statement again
1425fb7e7651Sdrh         */
1426fb7e7651Sdrh         flushStmtCache( pDb );
1427fb7e7651Sdrh         sqlite3_finalize(pStmt);
1428fb7e7651Sdrh         if( pPreStmt ) Tcl_Free((char*)pPreStmt);
142930ccda10Sdanielk1977         continue;
1430fb7e7651Sdrh       }else if( SQLITE_OK!=rc2 ){
1431fb7e7651Sdrh         /* If a run-time error occurs, report the error and stop reading
1432fb7e7651Sdrh         ** the SQL
1433fb7e7651Sdrh         */
1434ef2cb63eSdanielk1977         Tcl_SetObjResult(interp, dbTextToObj(sqlite3_errmsg(pDb->db)));
1435fb7e7651Sdrh         sqlite3_finalize(pStmt);
143630ccda10Sdanielk1977         rc = TCL_ERROR;
1437fb7e7651Sdrh         if( pPreStmt ) Tcl_Free((char*)pPreStmt);
143830ccda10Sdanielk1977         break;
1439fb7e7651Sdrh       }else if( pDb->maxStmt<=0 ){
1440fb7e7651Sdrh         /* If the cache is turned off, deallocated the statement */
1441fb7e7651Sdrh         if( pPreStmt ) Tcl_Free((char*)pPreStmt);
1442fb7e7651Sdrh         sqlite3_finalize(pStmt);
1443fb7e7651Sdrh       }else{
1444fb7e7651Sdrh         /* Everything worked and the cache is operational.
1445fb7e7651Sdrh         ** Create a new SqlPreparedStmt structure if we need one.
1446fb7e7651Sdrh         ** (If we already have one we can just reuse it.)
1447fb7e7651Sdrh         */
1448fb7e7651Sdrh         if( pPreStmt==0 ){
1449fb7e7651Sdrh           len = zLeft - zSql;
1450fb7e7651Sdrh           pPreStmt = (SqlPreparedStmt*)Tcl_Alloc( sizeof(*pPreStmt) + len );
1451fb7e7651Sdrh           if( pPreStmt==0 ) return TCL_ERROR;
1452fb7e7651Sdrh           pPreStmt->pStmt = pStmt;
1453fb7e7651Sdrh           pPreStmt->nSql = len;
1454fb7e7651Sdrh           memcpy(pPreStmt->zSql, zSql, len);
1455fb7e7651Sdrh           pPreStmt->zSql[len] = 0;
145630ccda10Sdanielk1977         }
145730ccda10Sdanielk1977 
1458fb7e7651Sdrh         /* Add the prepared statement to the beginning of the cache list
1459fb7e7651Sdrh         */
1460fb7e7651Sdrh         pPreStmt->pNext = pDb->stmtList;
1461fb7e7651Sdrh         pPreStmt->pPrev = 0;
1462fb7e7651Sdrh         if( pDb->stmtList ){
1463fb7e7651Sdrh          pDb->stmtList->pPrev = pPreStmt;
1464fb7e7651Sdrh         }
1465fb7e7651Sdrh         pDb->stmtList = pPreStmt;
1466fb7e7651Sdrh         if( pDb->stmtLast==0 ){
1467fb7e7651Sdrh           assert( pDb->nStmt==0 );
1468fb7e7651Sdrh           pDb->stmtLast = pPreStmt;
1469fb7e7651Sdrh         }else{
1470fb7e7651Sdrh           assert( pDb->nStmt>0 );
1471fb7e7651Sdrh         }
1472fb7e7651Sdrh         pDb->nStmt++;
1473fb7e7651Sdrh 
1474fb7e7651Sdrh         /* If we have too many statement in cache, remove the surplus from the
1475fb7e7651Sdrh         ** end of the cache list.
1476fb7e7651Sdrh         */
1477fb7e7651Sdrh         while( pDb->nStmt>pDb->maxStmt ){
1478fb7e7651Sdrh           sqlite3_finalize(pDb->stmtLast->pStmt);
1479fb7e7651Sdrh           pDb->stmtLast = pDb->stmtLast->pPrev;
1480fb7e7651Sdrh           Tcl_Free((char*)pDb->stmtLast->pNext);
1481fb7e7651Sdrh           pDb->stmtLast->pNext = 0;
1482fb7e7651Sdrh           pDb->nStmt--;
1483fb7e7651Sdrh         }
1484fb7e7651Sdrh       }
1485fb7e7651Sdrh 
1486fb7e7651Sdrh       /* Proceed to the next statement */
148730ccda10Sdanielk1977       zSql = zLeft;
148830ccda10Sdanielk1977     }
14891d895039Sdrh     Tcl_DecrRefCount(objv[2]);
149030ccda10Sdanielk1977 
14911807ce37Sdrh     if( pRet ){
1492ef2cb63eSdanielk1977       if( rc==TCL_OK ){
149330ccda10Sdanielk1977         Tcl_SetObjResult(interp, pRet);
149430ccda10Sdanielk1977       }
1495ef2cb63eSdanielk1977       Tcl_DecrRefCount(pRet);
14961807ce37Sdrh     }
149730ccda10Sdanielk1977     break;
149830ccda10Sdanielk1977   }
1499bec3f402Sdrh 
1500bec3f402Sdrh   /*
1501cabb0819Sdrh   **     $db function NAME SCRIPT
1502cabb0819Sdrh   **
1503cabb0819Sdrh   ** Create a new SQL function called NAME.  Whenever that function is
1504cabb0819Sdrh   ** called, invoke SCRIPT to evaluate the function.
1505cabb0819Sdrh   */
1506cabb0819Sdrh   case DB_FUNCTION: {
1507cabb0819Sdrh     SqlFunc *pFunc;
1508d1e4733dSdrh     Tcl_Obj *pScript;
1509cabb0819Sdrh     char *zName;
1510cabb0819Sdrh     if( objc!=4 ){
1511cabb0819Sdrh       Tcl_WrongNumArgs(interp, 2, objv, "NAME SCRIPT");
1512cabb0819Sdrh       return TCL_ERROR;
1513cabb0819Sdrh     }
1514cabb0819Sdrh     zName = Tcl_GetStringFromObj(objv[2], 0);
1515d1e4733dSdrh     pScript = objv[3];
1516d1e4733dSdrh     pFunc = findSqlFunc(pDb, zName);
1517cabb0819Sdrh     if( pFunc==0 ) return TCL_ERROR;
1518d1e4733dSdrh     if( pFunc->pScript ){
1519d1e4733dSdrh       Tcl_DecrRefCount(pFunc->pScript);
1520d1e4733dSdrh     }
1521d1e4733dSdrh     pFunc->pScript = pScript;
1522d1e4733dSdrh     Tcl_IncrRefCount(pScript);
1523d1e4733dSdrh     pFunc->useEvalObjv = safeToUseEvalObjv(interp, pScript);
1524c8c1158bSdanielk1977     rc = sqlite3_create_function(pDb->db, zName, -1, SQLITE_UTF8,
1525d8123366Sdanielk1977         pFunc, tclSqlFunc, 0, 0);
1526fb7e7651Sdrh     if( rc!=SQLITE_OK ){
1527fb7e7651Sdrh       rc = TCL_ERROR;
15289636c4e1Sdanielk1977       Tcl_SetResult(interp, (char *)sqlite3_errmsg(pDb->db), TCL_VOLATILE);
1529fb7e7651Sdrh     }else{
1530fb7e7651Sdrh       /* Must flush any cached statements */
1531fb7e7651Sdrh       flushStmtCache( pDb );
1532fb7e7651Sdrh     }
1533cabb0819Sdrh     break;
1534cabb0819Sdrh   }
1535cabb0819Sdrh 
1536cabb0819Sdrh   /*
153719e2d37fSdrh   **     $db nullvalue ?STRING?
153819e2d37fSdrh   **
153919e2d37fSdrh   ** Change text used when a NULL comes back from the database. If ?STRING?
154019e2d37fSdrh   ** is not present, then the current string used for NULL is returned.
154119e2d37fSdrh   ** If STRING is present, then STRING is returned.
154219e2d37fSdrh   **
154319e2d37fSdrh   */
154419e2d37fSdrh   case DB_NULLVALUE: {
154519e2d37fSdrh     if( objc!=2 && objc!=3 ){
154619e2d37fSdrh       Tcl_WrongNumArgs(interp, 2, objv, "NULLVALUE");
154719e2d37fSdrh       return TCL_ERROR;
154819e2d37fSdrh     }
154919e2d37fSdrh     if( objc==3 ){
155019e2d37fSdrh       int len;
155119e2d37fSdrh       char *zNull = Tcl_GetStringFromObj(objv[2], &len);
155219e2d37fSdrh       if( pDb->zNull ){
155319e2d37fSdrh         Tcl_Free(pDb->zNull);
155419e2d37fSdrh       }
155519e2d37fSdrh       if( zNull && len>0 ){
155619e2d37fSdrh         pDb->zNull = Tcl_Alloc( len + 1 );
155719e2d37fSdrh         strncpy(pDb->zNull, zNull, len);
155819e2d37fSdrh         pDb->zNull[len] = '\0';
155919e2d37fSdrh       }else{
156019e2d37fSdrh         pDb->zNull = 0;
156119e2d37fSdrh       }
156219e2d37fSdrh     }
156319e2d37fSdrh     Tcl_SetObjResult(interp, dbTextToObj(pDb->zNull));
156419e2d37fSdrh     break;
156519e2d37fSdrh   }
156619e2d37fSdrh 
156719e2d37fSdrh   /*
1568af9ff33aSdrh   **     $db last_insert_rowid
1569af9ff33aSdrh   **
1570af9ff33aSdrh   ** Return an integer which is the ROWID for the most recent insert.
1571af9ff33aSdrh   */
1572af9ff33aSdrh   case DB_LAST_INSERT_ROWID: {
1573af9ff33aSdrh     Tcl_Obj *pResult;
1574af9ff33aSdrh     int rowid;
1575af9ff33aSdrh     if( objc!=2 ){
1576af9ff33aSdrh       Tcl_WrongNumArgs(interp, 2, objv, "");
1577af9ff33aSdrh       return TCL_ERROR;
1578af9ff33aSdrh     }
15796f8a503dSdanielk1977     rowid = sqlite3_last_insert_rowid(pDb->db);
1580af9ff33aSdrh     pResult = Tcl_GetObjResult(interp);
1581af9ff33aSdrh     Tcl_SetIntObj(pResult, rowid);
1582af9ff33aSdrh     break;
1583af9ff33aSdrh   }
1584af9ff33aSdrh 
1585af9ff33aSdrh   /*
15861807ce37Sdrh   ** The DB_ONECOLUMN method is implemented together with DB_EVAL.
15875d9d7576Sdrh   */
15881807ce37Sdrh 
15891807ce37Sdrh   /*    $db progress ?N CALLBACK?
15901807ce37Sdrh   **
15911807ce37Sdrh   ** Invoke the given callback every N virtual machine opcodes while executing
15921807ce37Sdrh   ** queries.
15931807ce37Sdrh   */
15941807ce37Sdrh   case DB_PROGRESS: {
15951807ce37Sdrh     if( objc==2 ){
15961807ce37Sdrh       if( pDb->zProgress ){
15971807ce37Sdrh         Tcl_AppendResult(interp, pDb->zProgress, 0);
15985d9d7576Sdrh       }
15991807ce37Sdrh     }else if( objc==4 ){
16001807ce37Sdrh       char *zProgress;
16011807ce37Sdrh       int len;
16021807ce37Sdrh       int N;
16031807ce37Sdrh       if( TCL_OK!=Tcl_GetIntFromObj(interp, objv[2], &N) ){
16041807ce37Sdrh 	return TCL_ERROR;
16051807ce37Sdrh       };
16061807ce37Sdrh       if( pDb->zProgress ){
16071807ce37Sdrh         Tcl_Free(pDb->zProgress);
16081807ce37Sdrh       }
16091807ce37Sdrh       zProgress = Tcl_GetStringFromObj(objv[3], &len);
16101807ce37Sdrh       if( zProgress && len>0 ){
16111807ce37Sdrh         pDb->zProgress = Tcl_Alloc( len + 1 );
16121807ce37Sdrh         strcpy(pDb->zProgress, zProgress);
16131807ce37Sdrh       }else{
16141807ce37Sdrh         pDb->zProgress = 0;
16151807ce37Sdrh       }
16161807ce37Sdrh #ifndef SQLITE_OMIT_PROGRESS_CALLBACK
16171807ce37Sdrh       if( pDb->zProgress ){
16181807ce37Sdrh         pDb->interp = interp;
16191807ce37Sdrh         sqlite3_progress_handler(pDb->db, N, DbProgressHandler, pDb);
16201807ce37Sdrh       }else{
16211807ce37Sdrh         sqlite3_progress_handler(pDb->db, 0, 0, 0);
16221807ce37Sdrh       }
16231807ce37Sdrh #endif
16241807ce37Sdrh     }else{
16251807ce37Sdrh       Tcl_WrongNumArgs(interp, 2, objv, "N CALLBACK");
16261807ce37Sdrh       return TCL_ERROR;
16275d9d7576Sdrh     }
16285d9d7576Sdrh     break;
16295d9d7576Sdrh   }
16305d9d7576Sdrh 
163119e2d37fSdrh   /*    $db profile ?CALLBACK?
163219e2d37fSdrh   **
163319e2d37fSdrh   ** Make arrangements to invoke the CALLBACK routine after each SQL statement
163419e2d37fSdrh   ** that has run.  The text of the SQL and the amount of elapse time are
163519e2d37fSdrh   ** appended to CALLBACK before the script is run.
163619e2d37fSdrh   */
163719e2d37fSdrh   case DB_PROFILE: {
163819e2d37fSdrh     if( objc>3 ){
163919e2d37fSdrh       Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?");
164019e2d37fSdrh       return TCL_ERROR;
164119e2d37fSdrh     }else if( objc==2 ){
164219e2d37fSdrh       if( pDb->zProfile ){
164319e2d37fSdrh         Tcl_AppendResult(interp, pDb->zProfile, 0);
164419e2d37fSdrh       }
164519e2d37fSdrh     }else{
164619e2d37fSdrh       char *zProfile;
164719e2d37fSdrh       int len;
164819e2d37fSdrh       if( pDb->zProfile ){
164919e2d37fSdrh         Tcl_Free(pDb->zProfile);
165019e2d37fSdrh       }
165119e2d37fSdrh       zProfile = Tcl_GetStringFromObj(objv[2], &len);
165219e2d37fSdrh       if( zProfile && len>0 ){
165319e2d37fSdrh         pDb->zProfile = Tcl_Alloc( len + 1 );
165419e2d37fSdrh         strcpy(pDb->zProfile, zProfile);
165519e2d37fSdrh       }else{
165619e2d37fSdrh         pDb->zProfile = 0;
165719e2d37fSdrh       }
165819e2d37fSdrh #ifndef SQLITE_OMIT_TRACE
165919e2d37fSdrh       if( pDb->zProfile ){
166019e2d37fSdrh         pDb->interp = interp;
166119e2d37fSdrh         sqlite3_profile(pDb->db, DbProfileHandler, pDb);
166219e2d37fSdrh       }else{
166319e2d37fSdrh         sqlite3_profile(pDb->db, 0, 0);
166419e2d37fSdrh       }
166519e2d37fSdrh #endif
166619e2d37fSdrh     }
166719e2d37fSdrh     break;
166819e2d37fSdrh   }
166919e2d37fSdrh 
16705d9d7576Sdrh   /*
167122fbcb8dSdrh   **     $db rekey KEY
167222fbcb8dSdrh   **
167322fbcb8dSdrh   ** Change the encryption key on the currently open database.
167422fbcb8dSdrh   */
167522fbcb8dSdrh   case DB_REKEY: {
167622fbcb8dSdrh     int nKey;
167722fbcb8dSdrh     void *pKey;
167822fbcb8dSdrh     if( objc!=3 ){
167922fbcb8dSdrh       Tcl_WrongNumArgs(interp, 2, objv, "KEY");
168022fbcb8dSdrh       return TCL_ERROR;
168122fbcb8dSdrh     }
168222fbcb8dSdrh     pKey = Tcl_GetByteArrayFromObj(objv[2], &nKey);
16839eb9e26bSdrh #ifdef SQLITE_HAS_CODEC
16842011d5f5Sdrh     rc = sqlite3_rekey(pDb->db, pKey, nKey);
168522fbcb8dSdrh     if( rc ){
1686f20b21c8Sdanielk1977       Tcl_AppendResult(interp, sqlite3ErrStr(rc), 0);
168722fbcb8dSdrh       rc = TCL_ERROR;
168822fbcb8dSdrh     }
168922fbcb8dSdrh #endif
169022fbcb8dSdrh     break;
169122fbcb8dSdrh   }
169222fbcb8dSdrh 
169322fbcb8dSdrh   /*
1694bec3f402Sdrh   **     $db timeout MILLESECONDS
1695bec3f402Sdrh   **
1696bec3f402Sdrh   ** Delay for the number of milliseconds specified when a file is locked.
1697bec3f402Sdrh   */
16986d31316cSdrh   case DB_TIMEOUT: {
1699bec3f402Sdrh     int ms;
17006d31316cSdrh     if( objc!=3 ){
17016d31316cSdrh       Tcl_WrongNumArgs(interp, 2, objv, "MILLISECONDS");
1702bec3f402Sdrh       return TCL_ERROR;
170375897234Sdrh     }
17046d31316cSdrh     if( Tcl_GetIntFromObj(interp, objv[2], &ms) ) return TCL_ERROR;
17056f8a503dSdanielk1977     sqlite3_busy_timeout(pDb->db, ms);
17066d31316cSdrh     break;
170775897234Sdrh   }
1708b5a20d3cSdrh 
17090f14e2ebSdrh   /*
17100f14e2ebSdrh   **     $db total_changes
17110f14e2ebSdrh   **
17120f14e2ebSdrh   ** Return the number of rows that were modified, inserted, or deleted
17130f14e2ebSdrh   ** since the database handle was created.
17140f14e2ebSdrh   */
17150f14e2ebSdrh   case DB_TOTAL_CHANGES: {
17160f14e2ebSdrh     Tcl_Obj *pResult;
17170f14e2ebSdrh     if( objc!=2 ){
17180f14e2ebSdrh       Tcl_WrongNumArgs(interp, 2, objv, "");
17190f14e2ebSdrh       return TCL_ERROR;
17200f14e2ebSdrh     }
17210f14e2ebSdrh     pResult = Tcl_GetObjResult(interp);
17220f14e2ebSdrh     Tcl_SetIntObj(pResult, sqlite3_total_changes(pDb->db));
17230f14e2ebSdrh     break;
17240f14e2ebSdrh   }
17250f14e2ebSdrh 
1726b5a20d3cSdrh   /*    $db trace ?CALLBACK?
1727b5a20d3cSdrh   **
1728b5a20d3cSdrh   ** Make arrangements to invoke the CALLBACK routine for each SQL statement
1729b5a20d3cSdrh   ** that is executed.  The text of the SQL is appended to CALLBACK before
1730b5a20d3cSdrh   ** it is executed.
1731b5a20d3cSdrh   */
1732b5a20d3cSdrh   case DB_TRACE: {
1733b5a20d3cSdrh     if( objc>3 ){
1734b5a20d3cSdrh       Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?");
1735b97759edSdrh       return TCL_ERROR;
1736b5a20d3cSdrh     }else if( objc==2 ){
1737b5a20d3cSdrh       if( pDb->zTrace ){
1738b5a20d3cSdrh         Tcl_AppendResult(interp, pDb->zTrace, 0);
1739b5a20d3cSdrh       }
1740b5a20d3cSdrh     }else{
1741b5a20d3cSdrh       char *zTrace;
1742b5a20d3cSdrh       int len;
1743b5a20d3cSdrh       if( pDb->zTrace ){
1744b5a20d3cSdrh         Tcl_Free(pDb->zTrace);
1745b5a20d3cSdrh       }
1746b5a20d3cSdrh       zTrace = Tcl_GetStringFromObj(objv[2], &len);
1747b5a20d3cSdrh       if( zTrace && len>0 ){
1748b5a20d3cSdrh         pDb->zTrace = Tcl_Alloc( len + 1 );
1749b5a20d3cSdrh         strcpy(pDb->zTrace, zTrace);
1750b5a20d3cSdrh       }else{
1751b5a20d3cSdrh         pDb->zTrace = 0;
1752b5a20d3cSdrh       }
175319e2d37fSdrh #ifndef SQLITE_OMIT_TRACE
1754b5a20d3cSdrh       if( pDb->zTrace ){
1755b5a20d3cSdrh         pDb->interp = interp;
17566f8a503dSdanielk1977         sqlite3_trace(pDb->db, DbTraceHandler, pDb);
1757b5a20d3cSdrh       }else{
17586f8a503dSdanielk1977         sqlite3_trace(pDb->db, 0, 0);
1759b5a20d3cSdrh       }
176019e2d37fSdrh #endif
1761b5a20d3cSdrh     }
1762b5a20d3cSdrh     break;
1763b5a20d3cSdrh   }
1764b5a20d3cSdrh 
17653d21423cSdrh   /*    $db transaction [-deferred|-immediate|-exclusive] SCRIPT
17663d21423cSdrh   **
17673d21423cSdrh   ** Start a new transaction (if we are not already in the midst of a
17683d21423cSdrh   ** transaction) and execute the TCL script SCRIPT.  After SCRIPT
17693d21423cSdrh   ** completes, either commit the transaction or roll it back if SCRIPT
17703d21423cSdrh   ** throws an exception.  Or if no new transation was started, do nothing.
17713d21423cSdrh   ** pass the exception on up the stack.
17723d21423cSdrh   **
17733d21423cSdrh   ** This command was inspired by Dave Thomas's talk on Ruby at the
17743d21423cSdrh   ** 2005 O'Reilly Open Source Convention (OSCON).
17753d21423cSdrh   */
17763d21423cSdrh   case DB_TRANSACTION: {
17773d21423cSdrh     int inTrans;
17783d21423cSdrh     Tcl_Obj *pScript;
17793d21423cSdrh     const char *zBegin = "BEGIN";
17803d21423cSdrh     if( objc!=3 && objc!=4 ){
17813d21423cSdrh       Tcl_WrongNumArgs(interp, 2, objv, "[TYPE] SCRIPT");
17823d21423cSdrh       return TCL_ERROR;
17833d21423cSdrh     }
17843d21423cSdrh     if( objc==3 ){
17853d21423cSdrh       pScript = objv[2];
17863d21423cSdrh     } else {
17873d21423cSdrh       static const char *TTYPE_strs[] = {
1788ce604012Sdrh         "deferred",   "exclusive",  "immediate", 0
17893d21423cSdrh       };
17903d21423cSdrh       enum TTYPE_enum {
17913d21423cSdrh         TTYPE_DEFERRED, TTYPE_EXCLUSIVE, TTYPE_IMMEDIATE
17923d21423cSdrh       };
17933d21423cSdrh       int ttype;
1794b5555e7eSdrh       if( Tcl_GetIndexFromObj(interp, objv[2], TTYPE_strs, "transaction type",
17953d21423cSdrh                               0, &ttype) ){
17963d21423cSdrh         return TCL_ERROR;
17973d21423cSdrh       }
17983d21423cSdrh       switch( (enum TTYPE_enum)ttype ){
17993d21423cSdrh         case TTYPE_DEFERRED:    /* no-op */;                 break;
18003d21423cSdrh         case TTYPE_EXCLUSIVE:   zBegin = "BEGIN EXCLUSIVE";  break;
18013d21423cSdrh         case TTYPE_IMMEDIATE:   zBegin = "BEGIN IMMEDIATE";  break;
18023d21423cSdrh       }
18033d21423cSdrh       pScript = objv[3];
18043d21423cSdrh     }
18053d21423cSdrh     inTrans = !sqlite3_get_autocommit(pDb->db);
18063d21423cSdrh     if( !inTrans ){
18073d21423cSdrh       sqlite3_exec(pDb->db, zBegin, 0, 0, 0);
18083d21423cSdrh     }
18093d21423cSdrh     rc = Tcl_EvalObjEx(interp, pScript, 0);
18103d21423cSdrh     if( !inTrans ){
18113d21423cSdrh       const char *zEnd;
18123d21423cSdrh       if( rc==TCL_ERROR ){
18133d21423cSdrh         zEnd = "ROLLBACK";
18143d21423cSdrh       } else {
18153d21423cSdrh         zEnd = "COMMIT";
18163d21423cSdrh       }
18173d21423cSdrh       sqlite3_exec(pDb->db, zEnd, 0, 0, 0);
18183d21423cSdrh     }
18193d21423cSdrh     break;
18203d21423cSdrh   }
18213d21423cSdrh 
18224397de57Sdanielk1977   /*    $db version
18234397de57Sdanielk1977   **
18244397de57Sdanielk1977   ** Return the version string for this database.
18254397de57Sdanielk1977   */
18264397de57Sdanielk1977   case DB_VERSION: {
18274397de57Sdanielk1977     Tcl_SetResult(interp, (char *)sqlite3_libversion(), TCL_STATIC);
18284397de57Sdanielk1977     break;
18294397de57Sdanielk1977   }
18304397de57Sdanielk1977 
18311067fe11Stpoindex 
18326d31316cSdrh   } /* End of the SWITCH statement */
183322fbcb8dSdrh   return rc;
183475897234Sdrh }
183575897234Sdrh 
183675897234Sdrh /*
18379bb575fdSdrh **   sqlite3 DBNAME FILENAME ?MODE? ?-key KEY?
183875897234Sdrh **
183975897234Sdrh ** This is the main Tcl command.  When the "sqlite" Tcl command is
184075897234Sdrh ** invoked, this routine runs to process that command.
184175897234Sdrh **
184275897234Sdrh ** The first argument, DBNAME, is an arbitrary name for a new
184375897234Sdrh ** database connection.  This command creates a new command named
184475897234Sdrh ** DBNAME that is used to control that connection.  The database
184575897234Sdrh ** connection is deleted when the DBNAME command is deleted.
184675897234Sdrh **
184775897234Sdrh ** The second argument is the name of the directory that contains
184875897234Sdrh ** the sqlite database that is to be accessed.
1849fbc3eab8Sdrh **
1850fbc3eab8Sdrh ** For testing purposes, we also support the following:
1851fbc3eab8Sdrh **
18529bb575fdSdrh **  sqlite3 -encoding
1853fbc3eab8Sdrh **
1854fbc3eab8Sdrh **       Return the encoding used by LIKE and GLOB operators.  Choices
1855fbc3eab8Sdrh **       are UTF-8 and iso8859.
1856fbc3eab8Sdrh **
18579bb575fdSdrh **  sqlite3 -version
1858647cb0e1Sdrh **
1859647cb0e1Sdrh **       Return the version number of the SQLite library.
1860647cb0e1Sdrh **
18619bb575fdSdrh **  sqlite3 -tcl-uses-utf
1862fbc3eab8Sdrh **
1863fbc3eab8Sdrh **       Return "1" if compiled with a Tcl uses UTF-8.  Return "0" if
1864fbc3eab8Sdrh **       not.  Used by tests to make sure the library was compiled
1865fbc3eab8Sdrh **       correctly.
186675897234Sdrh */
186722fbcb8dSdrh static int DbMain(void *cd, Tcl_Interp *interp, int objc,Tcl_Obj *const*objv){
1868bec3f402Sdrh   SqliteDb *p;
186922fbcb8dSdrh   void *pKey = 0;
187022fbcb8dSdrh   int nKey = 0;
187122fbcb8dSdrh   const char *zArg;
187275897234Sdrh   char *zErrMsg;
187322fbcb8dSdrh   const char *zFile;
187406b2718aSdrh   char zBuf[80];
187522fbcb8dSdrh   if( objc==2 ){
187622fbcb8dSdrh     zArg = Tcl_GetStringFromObj(objv[1], 0);
187722fbcb8dSdrh     if( strcmp(zArg,"-version")==0 ){
18786f8a503dSdanielk1977       Tcl_AppendResult(interp,sqlite3_version,0);
1879647cb0e1Sdrh       return TCL_OK;
1880647cb0e1Sdrh     }
18819eb9e26bSdrh     if( strcmp(zArg,"-has-codec")==0 ){
18829eb9e26bSdrh #ifdef SQLITE_HAS_CODEC
188322fbcb8dSdrh       Tcl_AppendResult(interp,"1",0);
188422fbcb8dSdrh #else
188522fbcb8dSdrh       Tcl_AppendResult(interp,"0",0);
188622fbcb8dSdrh #endif
188722fbcb8dSdrh       return TCL_OK;
188822fbcb8dSdrh     }
188922fbcb8dSdrh     if( strcmp(zArg,"-tcl-uses-utf")==0 ){
1890fbc3eab8Sdrh #ifdef TCL_UTF_MAX
1891fbc3eab8Sdrh       Tcl_AppendResult(interp,"1",0);
1892fbc3eab8Sdrh #else
1893fbc3eab8Sdrh       Tcl_AppendResult(interp,"0",0);
1894fbc3eab8Sdrh #endif
1895fbc3eab8Sdrh       return TCL_OK;
1896fbc3eab8Sdrh     }
1897fbc3eab8Sdrh   }
189822fbcb8dSdrh   if( objc==5 || objc==6 ){
189922fbcb8dSdrh     zArg = Tcl_GetStringFromObj(objv[objc-2], 0);
190022fbcb8dSdrh     if( strcmp(zArg,"-key")==0 ){
190122fbcb8dSdrh       pKey = Tcl_GetByteArrayFromObj(objv[objc-1], &nKey);
190222fbcb8dSdrh       objc -= 2;
190322fbcb8dSdrh     }
190422fbcb8dSdrh   }
190522fbcb8dSdrh   if( objc!=3 && objc!=4 ){
190622fbcb8dSdrh     Tcl_WrongNumArgs(interp, 1, objv,
19079eb9e26bSdrh #ifdef SQLITE_HAS_CODEC
19089eb9e26bSdrh       "HANDLE FILENAME ?-key CODEC-KEY?"
190922fbcb8dSdrh #else
191022fbcb8dSdrh       "HANDLE FILENAME ?MODE?"
191122fbcb8dSdrh #endif
191222fbcb8dSdrh     );
191375897234Sdrh     return TCL_ERROR;
191475897234Sdrh   }
191575897234Sdrh   zErrMsg = 0;
19164cdc9e84Sdrh   p = (SqliteDb*)Tcl_Alloc( sizeof(*p) );
191775897234Sdrh   if( p==0 ){
1918bec3f402Sdrh     Tcl_SetResult(interp, "malloc failed", TCL_STATIC);
1919bec3f402Sdrh     return TCL_ERROR;
1920bec3f402Sdrh   }
1921bec3f402Sdrh   memset(p, 0, sizeof(*p));
192222fbcb8dSdrh   zFile = Tcl_GetStringFromObj(objv[2], 0);
19234f057f90Sdanielk1977   sqlite3_open(zFile, &p->db);
192480290863Sdanielk1977   if( SQLITE_OK!=sqlite3_errcode(p->db) ){
192580290863Sdanielk1977     zErrMsg = strdup(sqlite3_errmsg(p->db));
192680290863Sdanielk1977     sqlite3_close(p->db);
192780290863Sdanielk1977     p->db = 0;
192880290863Sdanielk1977   }
19292011d5f5Sdrh #ifdef SQLITE_HAS_CODEC
19302011d5f5Sdrh   sqlite3_key(p->db, pKey, nKey);
1931eb8ed70dSdrh #endif
1932bec3f402Sdrh   if( p->db==0 ){
193375897234Sdrh     Tcl_SetResult(interp, zErrMsg, TCL_VOLATILE);
1934bec3f402Sdrh     Tcl_Free((char*)p);
193575897234Sdrh     free(zErrMsg);
193675897234Sdrh     return TCL_ERROR;
193775897234Sdrh   }
1938fb7e7651Sdrh   p->maxStmt = NUM_PREPARED_STMTS;
193922fbcb8dSdrh   zArg = Tcl_GetStringFromObj(objv[1], 0);
194022fbcb8dSdrh   Tcl_CreateObjCommand(interp, zArg, DbObjCmd, (char*)p, DbDeleteCmd);
1941c22bd47dSdrh 
194206b2718aSdrh   /* The return value is the value of the sqlite* pointer
194306b2718aSdrh   */
194406b2718aSdrh   sprintf(zBuf, "%p", p->db);
19455e5377fbSdrh   if( strncmp(zBuf,"0x",2) ){
19465e5377fbSdrh     sprintf(zBuf, "0x%p", p->db);
19475e5377fbSdrh   }
194806b2718aSdrh   Tcl_AppendResult(interp, zBuf, 0);
194906b2718aSdrh 
1950c22bd47dSdrh   /* If compiled with SQLITE_TEST turned on, then register the "md5sum"
195106b2718aSdrh   ** SQL function.
1952c22bd47dSdrh   */
195328b4e489Sdrh #ifdef SQLITE_TEST
195428b4e489Sdrh   {
19559bb575fdSdrh     extern void Md5_Register(sqlite3*);
19563b93bc8cSdrh #ifdef SQLITE_MEMDEBUG
195713073931Sdanielk1977     int mallocfail = sqlite3_iMallocFail;
195813073931Sdanielk1977     sqlite3_iMallocFail = 0;
19595b59af85Sdanielk1977 #endif
196028b4e489Sdrh     Md5_Register(p->db);
19613b93bc8cSdrh #ifdef SQLITE_MEMDEBUG
196213073931Sdanielk1977     sqlite3_iMallocFail = mallocfail;
19635b59af85Sdanielk1977 #endif
196428b4e489Sdrh    }
196528b4e489Sdrh #endif
19667cedc8d4Sdanielk1977   p->interp = interp;
196775897234Sdrh   return TCL_OK;
196875897234Sdrh }
196975897234Sdrh 
197075897234Sdrh /*
197190ca9753Sdrh ** Provide a dummy Tcl_InitStubs if we are using this as a static
197290ca9753Sdrh ** library.
197390ca9753Sdrh */
197490ca9753Sdrh #ifndef USE_TCL_STUBS
197590ca9753Sdrh # undef  Tcl_InitStubs
197690ca9753Sdrh # define Tcl_InitStubs(a,b,c)
197790ca9753Sdrh #endif
197890ca9753Sdrh 
197990ca9753Sdrh /*
198029bc4615Sdrh ** Make sure we have a PACKAGE_VERSION macro defined.  This will be
198129bc4615Sdrh ** defined automatically by the TEA makefile.  But other makefiles
198229bc4615Sdrh ** do not define it.
198329bc4615Sdrh */
198429bc4615Sdrh #ifndef PACKAGE_VERSION
198529bc4615Sdrh # define PACKAGE_VERSION SQLITE_VERSION
198629bc4615Sdrh #endif
198729bc4615Sdrh 
198829bc4615Sdrh /*
198975897234Sdrh ** Initialize this module.
199075897234Sdrh **
199175897234Sdrh ** This Tcl module contains only a single new Tcl command named "sqlite".
199275897234Sdrh ** (Hence there is no namespace.  There is no point in using a namespace
199375897234Sdrh ** if the extension only supplies one new name!)  The "sqlite" command is
199475897234Sdrh ** used to open a new SQLite database.  See the DbMain() routine above
199575897234Sdrh ** for additional information.
199675897234Sdrh */
199729bc4615Sdrh EXTERN int Sqlite3_Init(Tcl_Interp *interp){
199892febd92Sdrh   Tcl_InitStubs(interp, "8.4", 0);
1999ef4ac8f9Sdrh   Tcl_CreateObjCommand(interp, "sqlite3", (Tcl_ObjCmdProc*)DbMain, 0, 0);
200029bc4615Sdrh   Tcl_PkgProvide(interp, "sqlite3", PACKAGE_VERSION);
200149766d6cSdrh   Tcl_CreateObjCommand(interp, "sqlite", (Tcl_ObjCmdProc*)DbMain, 0, 0);
200229bc4615Sdrh   Tcl_PkgProvide(interp, "sqlite", PACKAGE_VERSION);
200390ca9753Sdrh   return TCL_OK;
200490ca9753Sdrh }
200529bc4615Sdrh EXTERN int Tclsqlite3_Init(Tcl_Interp *interp){ return Sqlite3_Init(interp); }
200629bc4615Sdrh EXTERN int Sqlite3_SafeInit(Tcl_Interp *interp){ return TCL_OK; }
200729bc4615Sdrh EXTERN int Tclsqlite3_SafeInit(Tcl_Interp *interp){ return TCL_OK; }
200849766d6cSdrh 
200949766d6cSdrh #ifndef SQLITE_3_SUFFIX_ONLY
201029bc4615Sdrh EXTERN int Sqlite_Init(Tcl_Interp *interp){ return Sqlite3_Init(interp); }
201129bc4615Sdrh EXTERN int Tclsqlite_Init(Tcl_Interp *interp){ return Sqlite3_Init(interp); }
201229bc4615Sdrh EXTERN int Sqlite_SafeInit(Tcl_Interp *interp){ return TCL_OK; }
201329bc4615Sdrh EXTERN int Tclsqlite_SafeInit(Tcl_Interp *interp){ return TCL_OK; }
201449766d6cSdrh #endif
201575897234Sdrh 
20163e27c026Sdrh #ifdef TCLSH
20173e27c026Sdrh /*****************************************************************************
20183e27c026Sdrh ** The code that follows is used to build standalone TCL interpreters
201975897234Sdrh */
2020348784efSdrh 
2021348784efSdrh /*
20223e27c026Sdrh ** If the macro TCLSH is one, then put in code this for the
20233e27c026Sdrh ** "main" routine that will initialize Tcl and take input from
20243e27c026Sdrh ** standard input.
2025348784efSdrh */
20263e27c026Sdrh #if TCLSH==1
2027348784efSdrh static char zMainloop[] =
2028348784efSdrh   "set line {}\n"
2029348784efSdrh   "while {![eof stdin]} {\n"
2030348784efSdrh     "if {$line!=\"\"} {\n"
2031348784efSdrh       "puts -nonewline \"> \"\n"
2032348784efSdrh     "} else {\n"
2033348784efSdrh       "puts -nonewline \"% \"\n"
2034348784efSdrh     "}\n"
2035348784efSdrh     "flush stdout\n"
2036348784efSdrh     "append line [gets stdin]\n"
2037348784efSdrh     "if {[info complete $line]} {\n"
2038348784efSdrh       "if {[catch {uplevel #0 $line} result]} {\n"
2039348784efSdrh         "puts stderr \"Error: $result\"\n"
2040348784efSdrh       "} elseif {$result!=\"\"} {\n"
2041348784efSdrh         "puts $result\n"
2042348784efSdrh       "}\n"
2043348784efSdrh       "set line {}\n"
2044348784efSdrh     "} else {\n"
2045348784efSdrh       "append line \\n\n"
2046348784efSdrh     "}\n"
2047348784efSdrh   "}\n"
2048348784efSdrh ;
20493e27c026Sdrh #endif
20503e27c026Sdrh 
20513e27c026Sdrh /*
20523e27c026Sdrh ** If the macro TCLSH is two, then get the main loop code out of
20533e27c026Sdrh ** the separate file "spaceanal_tcl.h".
20543e27c026Sdrh */
20553e27c026Sdrh #if TCLSH==2
20563e27c026Sdrh static char zMainloop[] =
20573e27c026Sdrh #include "spaceanal_tcl.h"
20583e27c026Sdrh ;
20593e27c026Sdrh #endif
2060348784efSdrh 
2061348784efSdrh #define TCLSH_MAIN main   /* Needed to fake out mktclapp */
2062348784efSdrh int TCLSH_MAIN(int argc, char **argv){
2063348784efSdrh   Tcl_Interp *interp;
2064297ecf14Sdrh   Tcl_FindExecutable(argv[0]);
2065348784efSdrh   interp = Tcl_CreateInterp();
206638f8271fSdrh   Sqlite3_Init(interp);
2067d9b0257aSdrh #ifdef SQLITE_TEST
2068d1bf3512Sdrh   {
2069d1bf3512Sdrh     extern int Sqlitetest1_Init(Tcl_Interp*);
20705c4d9703Sdrh     extern int Sqlitetest2_Init(Tcl_Interp*);
20715c4d9703Sdrh     extern int Sqlitetest3_Init(Tcl_Interp*);
2072a6064dcfSdrh     extern int Sqlitetest4_Init(Tcl_Interp*);
2073998b56c3Sdanielk1977     extern int Sqlitetest5_Init(Tcl_Interp*);
20749c06c953Sdrh     extern int Sqlitetest6_Init(Tcl_Interp*);
2075efc251daSdrh     extern int Md5_Init(Tcl_Interp*);
20762e66f0b9Sdrh     extern int Sqlitetestsse_Init(Tcl_Interp*);
20772e66f0b9Sdrh 
20786490bebdSdanielk1977     Sqlitetest1_Init(interp);
20795c4d9703Sdrh     Sqlitetest2_Init(interp);
2080de647130Sdrh     Sqlitetest3_Init(interp);
2081fc57d7bfSdanielk1977     Sqlitetest4_Init(interp);
2082998b56c3Sdanielk1977     Sqlitetest5_Init(interp);
20839c06c953Sdrh     Sqlitetest6_Init(interp);
2084efc251daSdrh     Md5_Init(interp);
208589dec819Sdrh #ifdef SQLITE_SSE
20862e66f0b9Sdrh     Sqlitetestsse_Init(interp);
20872e66f0b9Sdrh #endif
2088d1bf3512Sdrh   }
2089d1bf3512Sdrh #endif
20903e27c026Sdrh   if( argc>=2 || TCLSH==2 ){
2091348784efSdrh     int i;
2092348784efSdrh     Tcl_SetVar(interp,"argv0",argv[1],TCL_GLOBAL_ONLY);
2093348784efSdrh     Tcl_SetVar(interp,"argv", "", TCL_GLOBAL_ONLY);
209461212b69Sdrh     for(i=3-TCLSH; i<argc; i++){
2095348784efSdrh       Tcl_SetVar(interp, "argv", argv[i],
2096348784efSdrh           TCL_GLOBAL_ONLY | TCL_LIST_ELEMENT | TCL_APPEND_VALUE);
2097348784efSdrh     }
20983e27c026Sdrh     if( TCLSH==1 && Tcl_EvalFile(interp, argv[1])!=TCL_OK ){
20990de8c112Sdrh       const char *zInfo = Tcl_GetVar(interp, "errorInfo", TCL_GLOBAL_ONLY);
2100c61053b7Sdrh       if( zInfo==0 ) zInfo = interp->result;
2101c61053b7Sdrh       fprintf(stderr,"%s: %s\n", *argv, zInfo);
2102348784efSdrh       return 1;
2103348784efSdrh     }
21043e27c026Sdrh   }
21053e27c026Sdrh   if( argc<=1 || TCLSH==2 ){
2106348784efSdrh     Tcl_GlobalEval(interp, zMainloop);
2107348784efSdrh   }
2108348784efSdrh   return 0;
2109348784efSdrh }
2110348784efSdrh #endif /* TCLSH */
21116d31316cSdrh 
21126d31316cSdrh #endif /* !defined(NO_TCL) */
2113