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 *************************************************************************
12bd08af48Sdrh ** A TCL Interface to SQLite. Append this file to sqlite3.c and
13bd08af48Sdrh ** compile the whole thing to build a TCL-enabled version of SQLite.
1457a0227fSdrh **
1557a0227fSdrh ** Compile-time options:
1657a0227fSdrh **
1796a206faSdrh ** -DTCLSH Add a "main()" routine that works as a tclsh.
1857a0227fSdrh **
1996a206faSdrh ** -DTCLSH_INIT_PROC=name
2057a0227fSdrh **
2196a206faSdrh ** Invoke name(interp) to initialize the Tcl interpreter.
2296a206faSdrh ** If name(interp) returns a non-NULL string, then run
2396a206faSdrh ** that string as a Tcl script to launch the application.
2496a206faSdrh ** If name(interp) returns NULL, then run the regular
2596a206faSdrh ** tclsh-emulator code.
2675897234Sdrh */
2796a206faSdrh #ifdef TCLSH_INIT_PROC
2896a206faSdrh # define TCLSH 1
2996a206faSdrh #endif
3027b2f053Smistachkin
3127b2f053Smistachkin /*
3227b2f053Smistachkin ** If requested, include the SQLite compiler options file for MSVC.
3327b2f053Smistachkin */
3427b2f053Smistachkin #if defined(INCLUDE_MSVC_H)
3527b2f053Smistachkin # include "msvc.h"
3627b2f053Smistachkin #endif
3727b2f053Smistachkin
3852b1dbb5Smistachkin #if defined(INCLUDE_SQLITE_TCL_H)
3952b1dbb5Smistachkin # include "sqlite_tcl.h"
4052b1dbb5Smistachkin #else
41bd08af48Sdrh # include "tcl.h"
427617e4a8Smistachkin # ifndef SQLITE_TCLAPI
437617e4a8Smistachkin # define SQLITE_TCLAPI
447617e4a8Smistachkin # endif
4552b1dbb5Smistachkin #endif
46b4e9af9fSdanielk1977 #include <errno.h>
476d31316cSdrh
48bd08af48Sdrh /*
49bd08af48Sdrh ** Some additional include files are needed if this file is not
50bd08af48Sdrh ** appended to the amalgamation.
51bd08af48Sdrh */
52bd08af48Sdrh #ifndef SQLITE_AMALGAMATION
5365e8c82eSdrh # include "sqlite3.h"
5475897234Sdrh # include <stdlib.h>
5575897234Sdrh # include <string.h>
56ce927065Sdrh # include <assert.h>
5765e8c82eSdrh typedef unsigned char u8;
58bd08af48Sdrh #endif
59eb206381Sdrh #include <ctype.h>
6075897234Sdrh
611f28e070Smistachkin /* Used to get the current process ID */
621f28e070Smistachkin #if !defined(_WIN32)
631e8487dbSmistachkin # include <signal.h>
641f28e070Smistachkin # include <unistd.h>
651f28e070Smistachkin # define GETPID getpid
661f28e070Smistachkin #elif !defined(_WIN32_WCE)
671f28e070Smistachkin # ifndef SQLITE_AMALGAMATION
68176b3a09Smistachkin # ifndef WIN32_LEAN_AND_MEAN
691f28e070Smistachkin # define WIN32_LEAN_AND_MEAN
70176b3a09Smistachkin # endif
711f28e070Smistachkin # include <windows.h>
721f28e070Smistachkin # endif
731e8487dbSmistachkin # include <io.h>
741e8487dbSmistachkin # define isatty(h) _isatty(h)
751f28e070Smistachkin # define GETPID (int)GetCurrentProcessId
761f28e070Smistachkin #endif
771f28e070Smistachkin
78ad6e1370Sdrh /*
79ad6e1370Sdrh * Windows needs to know which symbols to export. Unix does not.
80ad6e1370Sdrh * BUILD_sqlite should be undefined for Unix.
81ad6e1370Sdrh */
82ad6e1370Sdrh #ifdef BUILD_sqlite
83ad6e1370Sdrh #undef TCL_STORAGE_CLASS
84ad6e1370Sdrh #define TCL_STORAGE_CLASS DLLEXPORT
85ad6e1370Sdrh #endif /* BUILD_sqlite */
8629bc4615Sdrh
87a21c6b6fSdanielk1977 #define NUM_PREPARED_STMTS 10
88fb7e7651Sdrh #define MAX_PREPARED_STMTS 100
89fb7e7651Sdrh
90c45e6716Sdrh /* Forward declaration */
91c45e6716Sdrh typedef struct SqliteDb SqliteDb;
9298808babSdrh
9398808babSdrh /*
94cabb0819Sdrh ** New SQL functions can be created as TCL scripts. Each such function
95cabb0819Sdrh ** is described by an instance of the following structure.
9689d24936Sdan **
9789d24936Sdan ** Variable eType may be set to SQLITE_INTEGER, SQLITE_FLOAT, SQLITE_TEXT,
9889d24936Sdan ** SQLITE_BLOB or SQLITE_NULL. If it is SQLITE_NULL, then the implementation
9989d24936Sdan ** attempts to determine the type of the result based on the Tcl object.
10089d24936Sdan ** If it is SQLITE_TEXT or SQLITE_BLOB, then a text (sqlite3_result_text())
10189d24936Sdan ** or blob (sqlite3_result_blob()) is returned. If it is SQLITE_INTEGER
10289d24936Sdan ** or SQLITE_FLOAT, then an attempt is made to return an integer or float
10389d24936Sdan ** value, falling back to float and then text if this is not possible.
104cabb0819Sdrh */
105cabb0819Sdrh typedef struct SqlFunc SqlFunc;
106cabb0819Sdrh struct SqlFunc {
107cabb0819Sdrh Tcl_Interp *interp; /* The TCL interpret to execute the function */
108d1e4733dSdrh Tcl_Obj *pScript; /* The Tcl_Obj representation of the script */
109c45e6716Sdrh SqliteDb *pDb; /* Database connection that owns this function */
110d1e4733dSdrh int useEvalObjv; /* True if it is safe to use Tcl_EvalObjv */
11189d24936Sdan int eType; /* Type of value to return */
112d1e4733dSdrh char *zName; /* Name of this function */
113cabb0819Sdrh SqlFunc *pNext; /* Next function on the list of them all */
114cabb0819Sdrh };
115cabb0819Sdrh
116cabb0819Sdrh /*
1170202b29eSdanielk1977 ** New collation sequences function can be created as TCL scripts. Each such
1180202b29eSdanielk1977 ** function is described by an instance of the following structure.
1190202b29eSdanielk1977 */
1200202b29eSdanielk1977 typedef struct SqlCollate SqlCollate;
1210202b29eSdanielk1977 struct SqlCollate {
1220202b29eSdanielk1977 Tcl_Interp *interp; /* The TCL interpret to execute the function */
1230202b29eSdanielk1977 char *zScript; /* The script to be run */
1240202b29eSdanielk1977 SqlCollate *pNext; /* Next function on the list of them all */
1250202b29eSdanielk1977 };
1260202b29eSdanielk1977
1270202b29eSdanielk1977 /*
128fb7e7651Sdrh ** Prepared statements are cached for faster execution. Each prepared
129fb7e7651Sdrh ** statement is described by an instance of the following structure.
130fb7e7651Sdrh */
131fb7e7651Sdrh typedef struct SqlPreparedStmt SqlPreparedStmt;
132fb7e7651Sdrh struct SqlPreparedStmt {
133fb7e7651Sdrh SqlPreparedStmt *pNext; /* Next in linked list */
134fb7e7651Sdrh SqlPreparedStmt *pPrev; /* Previous on the list */
135fb7e7651Sdrh sqlite3_stmt *pStmt; /* The prepared statement */
136fb7e7651Sdrh int nSql; /* chars in zSql[] */
137d0e2a854Sdanielk1977 const char *zSql; /* Text of the SQL statement */
1384a4c11aaSdan int nParm; /* Size of apParm array */
1394a4c11aaSdan Tcl_Obj **apParm; /* Array of referenced object pointers */
140fb7e7651Sdrh };
141fb7e7651Sdrh
142d0441796Sdanielk1977 typedef struct IncrblobChannel IncrblobChannel;
143d0441796Sdanielk1977
144fb7e7651Sdrh /*
145bec3f402Sdrh ** There is one instance of this structure for each SQLite database
146bec3f402Sdrh ** that has been opened by the SQLite TCL interface.
147c431fd55Sdan **
148c431fd55Sdan ** If this module is built with SQLITE_TEST defined (to create the SQLite
149c431fd55Sdan ** testfixture executable), then it may be configured to use either
150c431fd55Sdan ** sqlite3_prepare_v2() or sqlite3_prepare() to prepare SQL statements.
151c431fd55Sdan ** If SqliteDb.bLegacyPrepare is true, sqlite3_prepare() is used.
152bec3f402Sdrh */
153bec3f402Sdrh struct SqliteDb {
154dddca286Sdrh sqlite3 *db; /* The "real" database structure. MUST BE FIRST */
155bec3f402Sdrh Tcl_Interp *interp; /* The interpreter used for this database */
1566d31316cSdrh char *zBusy; /* The busy callback routine */
157aa940eacSdrh char *zCommit; /* The commit hook callback routine */
158b5a20d3cSdrh char *zTrace; /* The trace callback routine */
159b56660f5Smistachkin char *zTraceV2; /* The trace_v2 callback routine */
16019e2d37fSdrh char *zProfile; /* The profile callback routine */
161348bb5d6Sdanielk1977 char *zProgress; /* The progress callback routine */
162c06ede10Sdrh char *zBindFallback; /* Callback to invoke on a binding miss */
163e22a334bSdrh char *zAuth; /* The authorization callback routine */
1641f1549f8Sdrh int disableAuth; /* Disable the authorizer if it exists */
16555c45f2eSdanielk1977 char *zNull; /* Text to substitute for an SQL NULL value */
166cabb0819Sdrh SqlFunc *pFunc; /* List of SQL functions */
16794eb6a14Sdanielk1977 Tcl_Obj *pUpdateHook; /* Update hook script (if any) */
16846c47d46Sdan Tcl_Obj *pPreUpdateHook; /* Pre-update hook script (if any) */
16971fd80bfSdanielk1977 Tcl_Obj *pRollbackHook; /* Rollback hook script (if any) */
1705def0843Sdrh Tcl_Obj *pWalHook; /* WAL hook script (if any) */
171404ca075Sdanielk1977 Tcl_Obj *pUnlockNotify; /* Unlock notify script (if any) */
1720202b29eSdanielk1977 SqlCollate *pCollate; /* List of SQL collation functions */
1736f8a503dSdanielk1977 int rc; /* Return code of most recent sqlite3_exec() */
1747cedc8d4Sdanielk1977 Tcl_Obj *pCollateNeeded; /* Collation needed script */
175fb7e7651Sdrh SqlPreparedStmt *stmtList; /* List of prepared statements*/
176fb7e7651Sdrh SqlPreparedStmt *stmtLast; /* Last statement in the list */
177fb7e7651Sdrh int maxStmt; /* The next maximum number of stmtList */
178fb7e7651Sdrh int nStmt; /* Number of statements in stmtList */
179d0441796Sdanielk1977 IncrblobChannel *pIncrblob;/* Linked list of open incrblob channels */
1803c379b01Sdrh int nStep, nSort, nIndex; /* Statistics for most recent operation */
181c456a76fSdan int nVMStep; /* Another statistic for most recent operation */
182cd38d520Sdanielk1977 int nTransaction; /* Number of nested [transaction] methods */
183147ef394Sdrh int openFlags; /* Flags used to open. (SQLITE_OPEN_URI) */
184bea28c73Sdan int nRef; /* Delete object when this reaches 0 */
185c431fd55Sdan #ifdef SQLITE_TEST
186c431fd55Sdan int bLegacyPrepare; /* True to use sqlite3_prepare() */
187c431fd55Sdan #endif
18898808babSdrh };
189297ecf14Sdrh
190b4e9af9fSdanielk1977 struct IncrblobChannel {
191d0441796Sdanielk1977 sqlite3_blob *pBlob; /* sqlite3 blob handle */
192dcbb5d3fSdanielk1977 SqliteDb *pDb; /* Associated database connection */
193b4e9af9fSdanielk1977 int iSeek; /* Current seek offset */
194d0441796Sdanielk1977 Tcl_Channel channel; /* Channel identifier */
195d0441796Sdanielk1977 IncrblobChannel *pNext; /* Linked list of all open incrblob channels */
196d0441796Sdanielk1977 IncrblobChannel *pPrev; /* Linked list of all open incrblob channels */
197b4e9af9fSdanielk1977 };
198b4e9af9fSdanielk1977
199ea678832Sdrh /*
200ea678832Sdrh ** Compute a string length that is limited to what can be stored in
201ea678832Sdrh ** lower 30 bits of a 32-bit signed integer.
202ea678832Sdrh */
strlen30(const char * z)2034f21c4afSdrh static int strlen30(const char *z){
204ea678832Sdrh const char *z2 = z;
205ea678832Sdrh while( *z2 ){ z2++; }
206ea678832Sdrh return 0x3fffffff & (int)(z2 - z);
207ea678832Sdrh }
208ea678832Sdrh
209ea678832Sdrh
21032a0d8bbSdanielk1977 #ifndef SQLITE_OMIT_INCRBLOB
211b4e9af9fSdanielk1977 /*
212d0441796Sdanielk1977 ** Close all incrblob channels opened using database connection pDb.
213d0441796Sdanielk1977 ** This is called when shutting down the database connection.
214d0441796Sdanielk1977 */
closeIncrblobChannels(SqliteDb * pDb)215d0441796Sdanielk1977 static void closeIncrblobChannels(SqliteDb *pDb){
216d0441796Sdanielk1977 IncrblobChannel *p;
217d0441796Sdanielk1977 IncrblobChannel *pNext;
218d0441796Sdanielk1977
219d0441796Sdanielk1977 for(p=pDb->pIncrblob; p; p=pNext){
220d0441796Sdanielk1977 pNext = p->pNext;
221d0441796Sdanielk1977
222d0441796Sdanielk1977 /* Note: Calling unregister here call Tcl_Close on the incrblob channel,
223d0441796Sdanielk1977 ** which deletes the IncrblobChannel structure at *p. So do not
224d0441796Sdanielk1977 ** call Tcl_Free() here.
225d0441796Sdanielk1977 */
226d0441796Sdanielk1977 Tcl_UnregisterChannel(pDb->interp, p->channel);
227d0441796Sdanielk1977 }
228d0441796Sdanielk1977 }
229d0441796Sdanielk1977
230d0441796Sdanielk1977 /*
231b4e9af9fSdanielk1977 ** Close an incremental blob channel.
232b4e9af9fSdanielk1977 */
incrblobClose(ClientData instanceData,Tcl_Interp * interp)2337617e4a8Smistachkin static int SQLITE_TCLAPI incrblobClose(
2347617e4a8Smistachkin ClientData instanceData,
2357617e4a8Smistachkin Tcl_Interp *interp
2367617e4a8Smistachkin ){
237b4e9af9fSdanielk1977 IncrblobChannel *p = (IncrblobChannel *)instanceData;
23892d4d7a9Sdanielk1977 int rc = sqlite3_blob_close(p->pBlob);
23992d4d7a9Sdanielk1977 sqlite3 *db = p->pDb->db;
240d0441796Sdanielk1977
241d0441796Sdanielk1977 /* Remove the channel from the SqliteDb.pIncrblob list. */
242d0441796Sdanielk1977 if( p->pNext ){
243d0441796Sdanielk1977 p->pNext->pPrev = p->pPrev;
244d0441796Sdanielk1977 }
245d0441796Sdanielk1977 if( p->pPrev ){
246d0441796Sdanielk1977 p->pPrev->pNext = p->pNext;
247d0441796Sdanielk1977 }
248d0441796Sdanielk1977 if( p->pDb->pIncrblob==p ){
249d0441796Sdanielk1977 p->pDb->pIncrblob = p->pNext;
250d0441796Sdanielk1977 }
251d0441796Sdanielk1977
25292d4d7a9Sdanielk1977 /* Free the IncrblobChannel structure */
253b4e9af9fSdanielk1977 Tcl_Free((char *)p);
25492d4d7a9Sdanielk1977
25592d4d7a9Sdanielk1977 if( rc!=SQLITE_OK ){
25692d4d7a9Sdanielk1977 Tcl_SetResult(interp, (char *)sqlite3_errmsg(db), TCL_VOLATILE);
25792d4d7a9Sdanielk1977 return TCL_ERROR;
25892d4d7a9Sdanielk1977 }
259b4e9af9fSdanielk1977 return TCL_OK;
260b4e9af9fSdanielk1977 }
261b4e9af9fSdanielk1977
262b4e9af9fSdanielk1977 /*
263b4e9af9fSdanielk1977 ** Read data from an incremental blob channel.
264b4e9af9fSdanielk1977 */
incrblobInput(ClientData instanceData,char * buf,int bufSize,int * errorCodePtr)2657617e4a8Smistachkin static int SQLITE_TCLAPI incrblobInput(
266b4e9af9fSdanielk1977 ClientData instanceData,
267b4e9af9fSdanielk1977 char *buf,
268b4e9af9fSdanielk1977 int bufSize,
269b4e9af9fSdanielk1977 int *errorCodePtr
270b4e9af9fSdanielk1977 ){
271b4e9af9fSdanielk1977 IncrblobChannel *p = (IncrblobChannel *)instanceData;
272b4e9af9fSdanielk1977 int nRead = bufSize; /* Number of bytes to read */
273b4e9af9fSdanielk1977 int nBlob; /* Total size of the blob */
274b4e9af9fSdanielk1977 int rc; /* sqlite error code */
275b4e9af9fSdanielk1977
276b4e9af9fSdanielk1977 nBlob = sqlite3_blob_bytes(p->pBlob);
277b4e9af9fSdanielk1977 if( (p->iSeek+nRead)>nBlob ){
278b4e9af9fSdanielk1977 nRead = nBlob-p->iSeek;
279b4e9af9fSdanielk1977 }
280b4e9af9fSdanielk1977 if( nRead<=0 ){
281b4e9af9fSdanielk1977 return 0;
282b4e9af9fSdanielk1977 }
283b4e9af9fSdanielk1977
284b4e9af9fSdanielk1977 rc = sqlite3_blob_read(p->pBlob, (void *)buf, nRead, p->iSeek);
285b4e9af9fSdanielk1977 if( rc!=SQLITE_OK ){
286b4e9af9fSdanielk1977 *errorCodePtr = rc;
287b4e9af9fSdanielk1977 return -1;
288b4e9af9fSdanielk1977 }
289b4e9af9fSdanielk1977
290b4e9af9fSdanielk1977 p->iSeek += nRead;
291b4e9af9fSdanielk1977 return nRead;
292b4e9af9fSdanielk1977 }
293b4e9af9fSdanielk1977
294d0441796Sdanielk1977 /*
295d0441796Sdanielk1977 ** Write data to an incremental blob channel.
296d0441796Sdanielk1977 */
incrblobOutput(ClientData instanceData,CONST char * buf,int toWrite,int * errorCodePtr)2977617e4a8Smistachkin static int SQLITE_TCLAPI incrblobOutput(
298b4e9af9fSdanielk1977 ClientData instanceData,
299b4e9af9fSdanielk1977 CONST char *buf,
300b4e9af9fSdanielk1977 int toWrite,
301b4e9af9fSdanielk1977 int *errorCodePtr
302b4e9af9fSdanielk1977 ){
303b4e9af9fSdanielk1977 IncrblobChannel *p = (IncrblobChannel *)instanceData;
304b4e9af9fSdanielk1977 int nWrite = toWrite; /* Number of bytes to write */
305b4e9af9fSdanielk1977 int nBlob; /* Total size of the blob */
306b4e9af9fSdanielk1977 int rc; /* sqlite error code */
307b4e9af9fSdanielk1977
308b4e9af9fSdanielk1977 nBlob = sqlite3_blob_bytes(p->pBlob);
309b4e9af9fSdanielk1977 if( (p->iSeek+nWrite)>nBlob ){
310b4e9af9fSdanielk1977 *errorCodePtr = EINVAL;
311b4e9af9fSdanielk1977 return -1;
312b4e9af9fSdanielk1977 }
313b4e9af9fSdanielk1977 if( nWrite<=0 ){
314b4e9af9fSdanielk1977 return 0;
315b4e9af9fSdanielk1977 }
316b4e9af9fSdanielk1977
317b4e9af9fSdanielk1977 rc = sqlite3_blob_write(p->pBlob, (void *)buf, nWrite, p->iSeek);
318b4e9af9fSdanielk1977 if( rc!=SQLITE_OK ){
319b4e9af9fSdanielk1977 *errorCodePtr = EIO;
320b4e9af9fSdanielk1977 return -1;
321b4e9af9fSdanielk1977 }
322b4e9af9fSdanielk1977
323b4e9af9fSdanielk1977 p->iSeek += nWrite;
324b4e9af9fSdanielk1977 return nWrite;
325b4e9af9fSdanielk1977 }
326b4e9af9fSdanielk1977
327b4e9af9fSdanielk1977 /*
328b4e9af9fSdanielk1977 ** Seek an incremental blob channel.
329b4e9af9fSdanielk1977 */
incrblobSeek(ClientData instanceData,long offset,int seekMode,int * errorCodePtr)3307617e4a8Smistachkin static int SQLITE_TCLAPI incrblobSeek(
331b4e9af9fSdanielk1977 ClientData instanceData,
332b4e9af9fSdanielk1977 long offset,
333b4e9af9fSdanielk1977 int seekMode,
334b4e9af9fSdanielk1977 int *errorCodePtr
335b4e9af9fSdanielk1977 ){
336b4e9af9fSdanielk1977 IncrblobChannel *p = (IncrblobChannel *)instanceData;
337b4e9af9fSdanielk1977
338b4e9af9fSdanielk1977 switch( seekMode ){
339b4e9af9fSdanielk1977 case SEEK_SET:
340b4e9af9fSdanielk1977 p->iSeek = offset;
341b4e9af9fSdanielk1977 break;
342b4e9af9fSdanielk1977 case SEEK_CUR:
343b4e9af9fSdanielk1977 p->iSeek += offset;
344b4e9af9fSdanielk1977 break;
345b4e9af9fSdanielk1977 case SEEK_END:
346b4e9af9fSdanielk1977 p->iSeek = sqlite3_blob_bytes(p->pBlob) + offset;
347b4e9af9fSdanielk1977 break;
348b4e9af9fSdanielk1977
349b4e9af9fSdanielk1977 default: assert(!"Bad seekMode");
350b4e9af9fSdanielk1977 }
351b4e9af9fSdanielk1977
352b4e9af9fSdanielk1977 return p->iSeek;
353b4e9af9fSdanielk1977 }
354b4e9af9fSdanielk1977
355b4e9af9fSdanielk1977
incrblobWatch(ClientData instanceData,int mode)3567617e4a8Smistachkin static void SQLITE_TCLAPI incrblobWatch(
3577617e4a8Smistachkin ClientData instanceData,
3587617e4a8Smistachkin int mode
3597617e4a8Smistachkin ){
360b4e9af9fSdanielk1977 /* NO-OP */
361b4e9af9fSdanielk1977 }
incrblobHandle(ClientData instanceData,int dir,ClientData * hPtr)3627617e4a8Smistachkin static int SQLITE_TCLAPI incrblobHandle(
3637617e4a8Smistachkin ClientData instanceData,
3647617e4a8Smistachkin int dir,
3657617e4a8Smistachkin ClientData *hPtr
3667617e4a8Smistachkin ){
367b4e9af9fSdanielk1977 return TCL_ERROR;
368b4e9af9fSdanielk1977 }
369b4e9af9fSdanielk1977
370b4e9af9fSdanielk1977 static Tcl_ChannelType IncrblobChannelType = {
371b4e9af9fSdanielk1977 "incrblob", /* typeName */
372b4e9af9fSdanielk1977 TCL_CHANNEL_VERSION_2, /* version */
373b4e9af9fSdanielk1977 incrblobClose, /* closeProc */
374b4e9af9fSdanielk1977 incrblobInput, /* inputProc */
375b4e9af9fSdanielk1977 incrblobOutput, /* outputProc */
376b4e9af9fSdanielk1977 incrblobSeek, /* seekProc */
377b4e9af9fSdanielk1977 0, /* setOptionProc */
378b4e9af9fSdanielk1977 0, /* getOptionProc */
379b4e9af9fSdanielk1977 incrblobWatch, /* watchProc (this is a no-op) */
380b4e9af9fSdanielk1977 incrblobHandle, /* getHandleProc (always returns error) */
381b4e9af9fSdanielk1977 0, /* close2Proc */
382b4e9af9fSdanielk1977 0, /* blockModeProc */
383b4e9af9fSdanielk1977 0, /* flushProc */
384b4e9af9fSdanielk1977 0, /* handlerProc */
385b4e9af9fSdanielk1977 0, /* wideSeekProc */
386b4e9af9fSdanielk1977 };
387b4e9af9fSdanielk1977
388b4e9af9fSdanielk1977 /*
389b4e9af9fSdanielk1977 ** Create a new incrblob channel.
390b4e9af9fSdanielk1977 */
createIncrblobChannel(Tcl_Interp * interp,SqliteDb * pDb,const char * zDb,const char * zTable,const char * zColumn,sqlite_int64 iRow,int isReadonly)391b4e9af9fSdanielk1977 static int createIncrblobChannel(
392b4e9af9fSdanielk1977 Tcl_Interp *interp,
393b4e9af9fSdanielk1977 SqliteDb *pDb,
394b4e9af9fSdanielk1977 const char *zDb,
395b4e9af9fSdanielk1977 const char *zTable,
396b4e9af9fSdanielk1977 const char *zColumn,
3978cbadb02Sdanielk1977 sqlite_int64 iRow,
3988cbadb02Sdanielk1977 int isReadonly
399b4e9af9fSdanielk1977 ){
400b4e9af9fSdanielk1977 IncrblobChannel *p;
4018cbadb02Sdanielk1977 sqlite3 *db = pDb->db;
402b4e9af9fSdanielk1977 sqlite3_blob *pBlob;
403b4e9af9fSdanielk1977 int rc;
4048cbadb02Sdanielk1977 int flags = TCL_READABLE|(isReadonly ? 0 : TCL_WRITABLE);
405b4e9af9fSdanielk1977
406b4e9af9fSdanielk1977 /* This variable is used to name the channels: "incrblob_[incr count]" */
407b4e9af9fSdanielk1977 static int count = 0;
408b4e9af9fSdanielk1977 char zChannel[64];
409b4e9af9fSdanielk1977
4108cbadb02Sdanielk1977 rc = sqlite3_blob_open(db, zDb, zTable, zColumn, iRow, !isReadonly, &pBlob);
411b4e9af9fSdanielk1977 if( rc!=SQLITE_OK ){
412b4e9af9fSdanielk1977 Tcl_SetResult(interp, (char *)sqlite3_errmsg(pDb->db), TCL_VOLATILE);
413b4e9af9fSdanielk1977 return TCL_ERROR;
414b4e9af9fSdanielk1977 }
415b4e9af9fSdanielk1977
416b4e9af9fSdanielk1977 p = (IncrblobChannel *)Tcl_Alloc(sizeof(IncrblobChannel));
417b4e9af9fSdanielk1977 p->iSeek = 0;
418b4e9af9fSdanielk1977 p->pBlob = pBlob;
419b4e9af9fSdanielk1977
4205bb3eb9bSdrh sqlite3_snprintf(sizeof(zChannel), zChannel, "incrblob_%d", ++count);
421d0441796Sdanielk1977 p->channel = Tcl_CreateChannel(&IncrblobChannelType, zChannel, p, flags);
422d0441796Sdanielk1977 Tcl_RegisterChannel(interp, p->channel);
423b4e9af9fSdanielk1977
424d0441796Sdanielk1977 /* Link the new channel into the SqliteDb.pIncrblob list. */
425d0441796Sdanielk1977 p->pNext = pDb->pIncrblob;
426d0441796Sdanielk1977 p->pPrev = 0;
427d0441796Sdanielk1977 if( p->pNext ){
428d0441796Sdanielk1977 p->pNext->pPrev = p;
429d0441796Sdanielk1977 }
430d0441796Sdanielk1977 pDb->pIncrblob = p;
431d0441796Sdanielk1977 p->pDb = pDb;
432d0441796Sdanielk1977
433d0441796Sdanielk1977 Tcl_SetResult(interp, (char *)Tcl_GetChannelName(p->channel), TCL_VOLATILE);
434b4e9af9fSdanielk1977 return TCL_OK;
435b4e9af9fSdanielk1977 }
43632a0d8bbSdanielk1977 #else /* else clause for "#ifndef SQLITE_OMIT_INCRBLOB" */
43732a0d8bbSdanielk1977 #define closeIncrblobChannels(pDb)
43832a0d8bbSdanielk1977 #endif
439b4e9af9fSdanielk1977
4406d31316cSdrh /*
441d1e4733dSdrh ** Look at the script prefix in pCmd. We will be executing this script
442d1e4733dSdrh ** after first appending one or more arguments. This routine analyzes
443d1e4733dSdrh ** the script to see if it is safe to use Tcl_EvalObjv() on the script
444d1e4733dSdrh ** rather than the more general Tcl_EvalEx(). Tcl_EvalObjv() is much
445d1e4733dSdrh ** faster.
446d1e4733dSdrh **
447d1e4733dSdrh ** Scripts that are safe to use with Tcl_EvalObjv() consists of a
448d1e4733dSdrh ** command name followed by zero or more arguments with no [...] or $
449d1e4733dSdrh ** or {...} or ; to be seen anywhere. Most callback scripts consist
450d1e4733dSdrh ** of just a single procedure name and they meet this requirement.
451d1e4733dSdrh */
safeToUseEvalObjv(Tcl_Interp * interp,Tcl_Obj * pCmd)452d1e4733dSdrh static int safeToUseEvalObjv(Tcl_Interp *interp, Tcl_Obj *pCmd){
453d1e4733dSdrh /* We could try to do something with Tcl_Parse(). But we will instead
454d1e4733dSdrh ** just do a search for forbidden characters. If any of the forbidden
455d1e4733dSdrh ** characters appear in pCmd, we will report the string as unsafe.
456d1e4733dSdrh */
457d1e4733dSdrh const char *z;
458d1e4733dSdrh int n;
459d1e4733dSdrh z = Tcl_GetStringFromObj(pCmd, &n);
460d1e4733dSdrh while( n-- > 0 ){
461d1e4733dSdrh int c = *(z++);
462d1e4733dSdrh if( c=='$' || c=='[' || c==';' ) return 0;
463d1e4733dSdrh }
464d1e4733dSdrh return 1;
465d1e4733dSdrh }
466d1e4733dSdrh
467d1e4733dSdrh /*
468d1e4733dSdrh ** Find an SqlFunc structure with the given name. Or create a new
469d1e4733dSdrh ** one if an existing one cannot be found. Return a pointer to the
470d1e4733dSdrh ** structure.
471d1e4733dSdrh */
findSqlFunc(SqliteDb * pDb,const char * zName)472d1e4733dSdrh static SqlFunc *findSqlFunc(SqliteDb *pDb, const char *zName){
473d1e4733dSdrh SqlFunc *p, *pNew;
4740425f189Sdrh int nName = strlen30(zName);
4750425f189Sdrh pNew = (SqlFunc*)Tcl_Alloc( sizeof(*pNew) + nName + 1 );
476d1e4733dSdrh pNew->zName = (char*)&pNew[1];
4770425f189Sdrh memcpy(pNew->zName, zName, nName+1);
478d1e4733dSdrh for(p=pDb->pFunc; p; p=p->pNext){
4790425f189Sdrh if( sqlite3_stricmp(p->zName, pNew->zName)==0 ){
480d1e4733dSdrh Tcl_Free((char*)pNew);
481d1e4733dSdrh return p;
482d1e4733dSdrh }
483d1e4733dSdrh }
484d1e4733dSdrh pNew->interp = pDb->interp;
485c45e6716Sdrh pNew->pDb = pDb;
486d1e4733dSdrh pNew->pScript = 0;
487d1e4733dSdrh pNew->pNext = pDb->pFunc;
488d1e4733dSdrh pDb->pFunc = pNew;
489d1e4733dSdrh return pNew;
490d1e4733dSdrh }
491d1e4733dSdrh
492d1e4733dSdrh /*
493c431fd55Sdan ** Free a single SqlPreparedStmt object.
494c431fd55Sdan */
dbFreeStmt(SqlPreparedStmt * pStmt)495c431fd55Sdan static void dbFreeStmt(SqlPreparedStmt *pStmt){
496c431fd55Sdan #ifdef SQLITE_TEST
497c431fd55Sdan if( sqlite3_sql(pStmt->pStmt)==0 ){
498c431fd55Sdan Tcl_Free((char *)pStmt->zSql);
499c431fd55Sdan }
500c431fd55Sdan #endif
501c431fd55Sdan sqlite3_finalize(pStmt->pStmt);
502c431fd55Sdan Tcl_Free((char *)pStmt);
503c431fd55Sdan }
504c431fd55Sdan
505c431fd55Sdan /*
506fb7e7651Sdrh ** Finalize and free a list of prepared statements
507fb7e7651Sdrh */
flushStmtCache(SqliteDb * pDb)508fb7e7651Sdrh static void flushStmtCache(SqliteDb *pDb){
509fb7e7651Sdrh SqlPreparedStmt *pPreStmt;
510c431fd55Sdan SqlPreparedStmt *pNext;
511fb7e7651Sdrh
512c431fd55Sdan for(pPreStmt = pDb->stmtList; pPreStmt; pPreStmt=pNext){
513c431fd55Sdan pNext = pPreStmt->pNext;
514c431fd55Sdan dbFreeStmt(pPreStmt);
515fb7e7651Sdrh }
516fb7e7651Sdrh pDb->nStmt = 0;
517fb7e7651Sdrh pDb->stmtLast = 0;
518c431fd55Sdan pDb->stmtList = 0;
519fb7e7651Sdrh }
520fb7e7651Sdrh
521fb7e7651Sdrh /*
522bea28c73Sdan ** Increment the reference counter on the SqliteDb object. The reference
523bea28c73Sdan ** should be released by calling delDatabaseRef().
52475897234Sdrh */
addDatabaseRef(SqliteDb * pDb)525bea28c73Sdan static void addDatabaseRef(SqliteDb *pDb){
526bea28c73Sdan pDb->nRef++;
527bea28c73Sdan }
528bea28c73Sdan
529bea28c73Sdan /*
530bea28c73Sdan ** Decrement the reference counter associated with the SqliteDb object.
531bea28c73Sdan ** If it reaches zero, delete the object.
532bea28c73Sdan */
delDatabaseRef(SqliteDb * pDb)533bea28c73Sdan static void delDatabaseRef(SqliteDb *pDb){
534bea28c73Sdan assert( pDb->nRef>0 );
535bea28c73Sdan pDb->nRef--;
536bea28c73Sdan if( pDb->nRef==0 ){
537fb7e7651Sdrh flushStmtCache(pDb);
538d0441796Sdanielk1977 closeIncrblobChannels(pDb);
5396f8a503dSdanielk1977 sqlite3_close(pDb->db);
540cabb0819Sdrh while( pDb->pFunc ){
541cabb0819Sdrh SqlFunc *pFunc = pDb->pFunc;
542cabb0819Sdrh pDb->pFunc = pFunc->pNext;
543c45e6716Sdrh assert( pFunc->pDb==pDb );
544d1e4733dSdrh Tcl_DecrRefCount(pFunc->pScript);
545cabb0819Sdrh Tcl_Free((char*)pFunc);
546cabb0819Sdrh }
5470202b29eSdanielk1977 while( pDb->pCollate ){
5480202b29eSdanielk1977 SqlCollate *pCollate = pDb->pCollate;
5490202b29eSdanielk1977 pDb->pCollate = pCollate->pNext;
5500202b29eSdanielk1977 Tcl_Free((char*)pCollate);
5510202b29eSdanielk1977 }
552bec3f402Sdrh if( pDb->zBusy ){
553bec3f402Sdrh Tcl_Free(pDb->zBusy);
554bec3f402Sdrh }
555b5a20d3cSdrh if( pDb->zTrace ){
556b5a20d3cSdrh Tcl_Free(pDb->zTrace);
5570d1a643aSdrh }
558b56660f5Smistachkin if( pDb->zTraceV2 ){
559b56660f5Smistachkin Tcl_Free(pDb->zTraceV2);
560b56660f5Smistachkin }
56119e2d37fSdrh if( pDb->zProfile ){
56219e2d37fSdrh Tcl_Free(pDb->zProfile);
56319e2d37fSdrh }
564c06ede10Sdrh if( pDb->zBindFallback ){
565c06ede10Sdrh Tcl_Free(pDb->zBindFallback);
566c06ede10Sdrh }
567e22a334bSdrh if( pDb->zAuth ){
568e22a334bSdrh Tcl_Free(pDb->zAuth);
569e22a334bSdrh }
57055c45f2eSdanielk1977 if( pDb->zNull ){
57155c45f2eSdanielk1977 Tcl_Free(pDb->zNull);
57255c45f2eSdanielk1977 }
57394eb6a14Sdanielk1977 if( pDb->pUpdateHook ){
57494eb6a14Sdanielk1977 Tcl_DecrRefCount(pDb->pUpdateHook);
57594eb6a14Sdanielk1977 }
57646c47d46Sdan if( pDb->pPreUpdateHook ){
57746c47d46Sdan Tcl_DecrRefCount(pDb->pPreUpdateHook);
57846c47d46Sdan }
57971fd80bfSdanielk1977 if( pDb->pRollbackHook ){
58071fd80bfSdanielk1977 Tcl_DecrRefCount(pDb->pRollbackHook);
58171fd80bfSdanielk1977 }
5825def0843Sdrh if( pDb->pWalHook ){
5835def0843Sdrh Tcl_DecrRefCount(pDb->pWalHook);
5848d22a174Sdan }
58594eb6a14Sdanielk1977 if( pDb->pCollateNeeded ){
58694eb6a14Sdanielk1977 Tcl_DecrRefCount(pDb->pCollateNeeded);
58794eb6a14Sdanielk1977 }
588bec3f402Sdrh Tcl_Free((char*)pDb);
589bec3f402Sdrh }
590bea28c73Sdan }
591bea28c73Sdan
592bea28c73Sdan /*
593bea28c73Sdan ** TCL calls this procedure when an sqlite3 database command is
594bea28c73Sdan ** deleted.
595bea28c73Sdan */
DbDeleteCmd(void * db)596bea28c73Sdan static void SQLITE_TCLAPI DbDeleteCmd(void *db){
597bea28c73Sdan SqliteDb *pDb = (SqliteDb*)db;
598bea28c73Sdan delDatabaseRef(pDb);
599bea28c73Sdan }
600bec3f402Sdrh
601bec3f402Sdrh /*
602bec3f402Sdrh ** This routine is called when a database file is locked while trying
603bec3f402Sdrh ** to execute SQL.
604bec3f402Sdrh */
DbBusyHandler(void * cd,int nTries)6052a764eb0Sdanielk1977 static int DbBusyHandler(void *cd, int nTries){
606bec3f402Sdrh SqliteDb *pDb = (SqliteDb*)cd;
607bec3f402Sdrh int rc;
608bec3f402Sdrh char zVal[30];
609bec3f402Sdrh
6105bb3eb9bSdrh sqlite3_snprintf(sizeof(zVal), zVal, "%d", nTries);
611d1e4733dSdrh rc = Tcl_VarEval(pDb->interp, pDb->zBusy, " ", zVal, (char*)0);
612bec3f402Sdrh if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){
613bec3f402Sdrh return 0;
614bec3f402Sdrh }
615bec3f402Sdrh return 1;
61675897234Sdrh }
61775897234Sdrh
61826e4a8b1Sdrh #ifndef SQLITE_OMIT_PROGRESS_CALLBACK
61975897234Sdrh /*
620348bb5d6Sdanielk1977 ** This routine is invoked as the 'progress callback' for the database.
621348bb5d6Sdanielk1977 */
DbProgressHandler(void * cd)622348bb5d6Sdanielk1977 static int DbProgressHandler(void *cd){
623348bb5d6Sdanielk1977 SqliteDb *pDb = (SqliteDb*)cd;
624348bb5d6Sdanielk1977 int rc;
625348bb5d6Sdanielk1977
626348bb5d6Sdanielk1977 assert( pDb->zProgress );
627348bb5d6Sdanielk1977 rc = Tcl_Eval(pDb->interp, pDb->zProgress);
628348bb5d6Sdanielk1977 if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){
629348bb5d6Sdanielk1977 return 1;
630348bb5d6Sdanielk1977 }
631348bb5d6Sdanielk1977 return 0;
632348bb5d6Sdanielk1977 }
63326e4a8b1Sdrh #endif
634348bb5d6Sdanielk1977
6352eb22af0Sdrh #if !defined(SQLITE_OMIT_TRACE) && !defined(SQLITE_OMIT_FLOATING_POINT) && \
6362eb22af0Sdrh !defined(SQLITE_OMIT_DEPRECATED)
637348bb5d6Sdanielk1977 /*
638b5a20d3cSdrh ** This routine is called by the SQLite trace handler whenever a new
639b5a20d3cSdrh ** block of SQL is executed. The TCL script in pDb->zTrace is executed.
6400d1a643aSdrh */
DbTraceHandler(void * cd,const char * zSql)641b5a20d3cSdrh static void DbTraceHandler(void *cd, const char *zSql){
6420d1a643aSdrh SqliteDb *pDb = (SqliteDb*)cd;
643b5a20d3cSdrh Tcl_DString str;
6440d1a643aSdrh
645b5a20d3cSdrh Tcl_DStringInit(&str);
646b5a20d3cSdrh Tcl_DStringAppend(&str, pDb->zTrace, -1);
647b5a20d3cSdrh Tcl_DStringAppendElement(&str, zSql);
648b5a20d3cSdrh Tcl_Eval(pDb->interp, Tcl_DStringValue(&str));
649b5a20d3cSdrh Tcl_DStringFree(&str);
650b5a20d3cSdrh Tcl_ResetResult(pDb->interp);
6510d1a643aSdrh }
652d1167393Sdrh #endif
6530d1a643aSdrh
654d1167393Sdrh #ifndef SQLITE_OMIT_TRACE
6550d1a643aSdrh /*
656b56660f5Smistachkin ** This routine is called by the SQLite trace_v2 handler whenever a new
657b56660f5Smistachkin ** supported event is generated. Unsupported event types are ignored.
658b56660f5Smistachkin ** The TCL script in pDb->zTraceV2 is executed, with the arguments for
659b56660f5Smistachkin ** the event appended to it (as list elements).
660b56660f5Smistachkin */
DbTraceV2Handler(unsigned type,void * cd,void * pd,void * xd)661b56660f5Smistachkin static int DbTraceV2Handler(
662b56660f5Smistachkin unsigned type, /* One of the SQLITE_TRACE_* event types. */
663b56660f5Smistachkin void *cd, /* The original context data pointer. */
664b56660f5Smistachkin void *pd, /* Primary event data, depends on event type. */
665b56660f5Smistachkin void *xd /* Extra event data, depends on event type. */
666b56660f5Smistachkin ){
667b56660f5Smistachkin SqliteDb *pDb = (SqliteDb*)cd;
668b56660f5Smistachkin Tcl_Obj *pCmd;
669b56660f5Smistachkin
670b56660f5Smistachkin switch( type ){
671b56660f5Smistachkin case SQLITE_TRACE_STMT: {
672b56660f5Smistachkin sqlite3_stmt *pStmt = (sqlite3_stmt *)pd;
673b56660f5Smistachkin char *zSql = (char *)xd;
674b56660f5Smistachkin
675b56660f5Smistachkin pCmd = Tcl_NewStringObj(pDb->zTraceV2, -1);
676b56660f5Smistachkin Tcl_IncrRefCount(pCmd);
677b56660f5Smistachkin Tcl_ListObjAppendElement(pDb->interp, pCmd,
678b56660f5Smistachkin Tcl_NewWideIntObj((Tcl_WideInt)pStmt));
679b56660f5Smistachkin Tcl_ListObjAppendElement(pDb->interp, pCmd,
680b56660f5Smistachkin Tcl_NewStringObj(zSql, -1));
681b56660f5Smistachkin Tcl_EvalObjEx(pDb->interp, pCmd, TCL_EVAL_DIRECT);
682b56660f5Smistachkin Tcl_DecrRefCount(pCmd);
683b56660f5Smistachkin Tcl_ResetResult(pDb->interp);
684b56660f5Smistachkin break;
685b56660f5Smistachkin }
686b56660f5Smistachkin case SQLITE_TRACE_PROFILE: {
687b56660f5Smistachkin sqlite3_stmt *pStmt = (sqlite3_stmt *)pd;
688ffdab721Sdrh sqlite3_int64 ns = *(sqlite3_int64*)xd;
689b56660f5Smistachkin
690b56660f5Smistachkin pCmd = Tcl_NewStringObj(pDb->zTraceV2, -1);
691b56660f5Smistachkin Tcl_IncrRefCount(pCmd);
692b56660f5Smistachkin Tcl_ListObjAppendElement(pDb->interp, pCmd,
693b56660f5Smistachkin Tcl_NewWideIntObj((Tcl_WideInt)pStmt));
694b56660f5Smistachkin Tcl_ListObjAppendElement(pDb->interp, pCmd,
695b56660f5Smistachkin Tcl_NewWideIntObj((Tcl_WideInt)ns));
696b56660f5Smistachkin Tcl_EvalObjEx(pDb->interp, pCmd, TCL_EVAL_DIRECT);
697b56660f5Smistachkin Tcl_DecrRefCount(pCmd);
698b56660f5Smistachkin Tcl_ResetResult(pDb->interp);
699b56660f5Smistachkin break;
700b56660f5Smistachkin }
701b56660f5Smistachkin case SQLITE_TRACE_ROW: {
702b56660f5Smistachkin sqlite3_stmt *pStmt = (sqlite3_stmt *)pd;
703b56660f5Smistachkin
704b56660f5Smistachkin pCmd = Tcl_NewStringObj(pDb->zTraceV2, -1);
705b56660f5Smistachkin Tcl_IncrRefCount(pCmd);
706b56660f5Smistachkin Tcl_ListObjAppendElement(pDb->interp, pCmd,
707b56660f5Smistachkin Tcl_NewWideIntObj((Tcl_WideInt)pStmt));
708b56660f5Smistachkin Tcl_EvalObjEx(pDb->interp, pCmd, TCL_EVAL_DIRECT);
709b56660f5Smistachkin Tcl_DecrRefCount(pCmd);
710b56660f5Smistachkin Tcl_ResetResult(pDb->interp);
711b56660f5Smistachkin break;
712b56660f5Smistachkin }
713b56660f5Smistachkin case SQLITE_TRACE_CLOSE: {
714b56660f5Smistachkin sqlite3 *db = (sqlite3 *)pd;
715b56660f5Smistachkin
716b56660f5Smistachkin pCmd = Tcl_NewStringObj(pDb->zTraceV2, -1);
717b56660f5Smistachkin Tcl_IncrRefCount(pCmd);
718b56660f5Smistachkin Tcl_ListObjAppendElement(pDb->interp, pCmd,
719b56660f5Smistachkin Tcl_NewWideIntObj((Tcl_WideInt)db));
720b56660f5Smistachkin Tcl_EvalObjEx(pDb->interp, pCmd, TCL_EVAL_DIRECT);
721b56660f5Smistachkin Tcl_DecrRefCount(pCmd);
722b56660f5Smistachkin Tcl_ResetResult(pDb->interp);
723b56660f5Smistachkin break;
724b56660f5Smistachkin }
725b56660f5Smistachkin }
726b56660f5Smistachkin return SQLITE_OK;
727b56660f5Smistachkin }
728b56660f5Smistachkin #endif
729b56660f5Smistachkin
7302eb22af0Sdrh #if !defined(SQLITE_OMIT_TRACE) && !defined(SQLITE_OMIT_FLOATING_POINT) && \
7312eb22af0Sdrh !defined(SQLITE_OMIT_DEPRECATED)
732b56660f5Smistachkin /*
73319e2d37fSdrh ** This routine is called by the SQLite profile handler after a statement
73419e2d37fSdrh ** SQL has executed. The TCL script in pDb->zProfile is evaluated.
73519e2d37fSdrh */
DbProfileHandler(void * cd,const char * zSql,sqlite_uint64 tm)73619e2d37fSdrh static void DbProfileHandler(void *cd, const char *zSql, sqlite_uint64 tm){
73719e2d37fSdrh SqliteDb *pDb = (SqliteDb*)cd;
73819e2d37fSdrh Tcl_DString str;
73919e2d37fSdrh char zTm[100];
74019e2d37fSdrh
74119e2d37fSdrh sqlite3_snprintf(sizeof(zTm)-1, zTm, "%lld", tm);
74219e2d37fSdrh Tcl_DStringInit(&str);
74319e2d37fSdrh Tcl_DStringAppend(&str, pDb->zProfile, -1);
74419e2d37fSdrh Tcl_DStringAppendElement(&str, zSql);
74519e2d37fSdrh Tcl_DStringAppendElement(&str, zTm);
74619e2d37fSdrh Tcl_Eval(pDb->interp, Tcl_DStringValue(&str));
74719e2d37fSdrh Tcl_DStringFree(&str);
74819e2d37fSdrh Tcl_ResetResult(pDb->interp);
74919e2d37fSdrh }
750d1167393Sdrh #endif
75119e2d37fSdrh
75219e2d37fSdrh /*
753aa940eacSdrh ** This routine is called when a transaction is committed. The
754aa940eacSdrh ** TCL script in pDb->zCommit is executed. If it returns non-zero or
755aa940eacSdrh ** if it throws an exception, the transaction is rolled back instead
756aa940eacSdrh ** of being committed.
757aa940eacSdrh */
DbCommitHandler(void * cd)758aa940eacSdrh static int DbCommitHandler(void *cd){
759aa940eacSdrh SqliteDb *pDb = (SqliteDb*)cd;
760aa940eacSdrh int rc;
761aa940eacSdrh
762aa940eacSdrh rc = Tcl_Eval(pDb->interp, pDb->zCommit);
763aa940eacSdrh if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){
764aa940eacSdrh return 1;
765aa940eacSdrh }
766aa940eacSdrh return 0;
767aa940eacSdrh }
768aa940eacSdrh
DbRollbackHandler(void * clientData)76971fd80bfSdanielk1977 static void DbRollbackHandler(void *clientData){
77071fd80bfSdanielk1977 SqliteDb *pDb = (SqliteDb*)clientData;
77171fd80bfSdanielk1977 assert(pDb->pRollbackHook);
77271fd80bfSdanielk1977 if( TCL_OK!=Tcl_EvalObjEx(pDb->interp, pDb->pRollbackHook, 0) ){
77371fd80bfSdanielk1977 Tcl_BackgroundError(pDb->interp);
77471fd80bfSdanielk1977 }
77571fd80bfSdanielk1977 }
77671fd80bfSdanielk1977
7775def0843Sdrh /*
7785def0843Sdrh ** This procedure handles wal_hook callbacks.
7795def0843Sdrh */
DbWalHandler(void * clientData,sqlite3 * db,const char * zDb,int nEntry)7805def0843Sdrh static int DbWalHandler(
7818d22a174Sdan void *clientData,
7828d22a174Sdan sqlite3 *db,
7838d22a174Sdan const char *zDb,
7848d22a174Sdan int nEntry
7858d22a174Sdan ){
7865def0843Sdrh int ret = SQLITE_OK;
7878d22a174Sdan Tcl_Obj *p;
7888d22a174Sdan SqliteDb *pDb = (SqliteDb*)clientData;
7898d22a174Sdan Tcl_Interp *interp = pDb->interp;
7905def0843Sdrh assert(pDb->pWalHook);
7918d22a174Sdan
7926e45e0c8Sdan assert( db==pDb->db );
7935def0843Sdrh p = Tcl_DuplicateObj(pDb->pWalHook);
7948d22a174Sdan Tcl_IncrRefCount(p);
7958d22a174Sdan Tcl_ListObjAppendElement(interp, p, Tcl_NewStringObj(zDb, -1));
7968d22a174Sdan Tcl_ListObjAppendElement(interp, p, Tcl_NewIntObj(nEntry));
7978d22a174Sdan if( TCL_OK!=Tcl_EvalObjEx(interp, p, 0)
7988d22a174Sdan || TCL_OK!=Tcl_GetIntFromObj(interp, Tcl_GetObjResult(interp), &ret)
7998d22a174Sdan ){
8008d22a174Sdan Tcl_BackgroundError(interp);
8018d22a174Sdan }
8028d22a174Sdan Tcl_DecrRefCount(p);
8038d22a174Sdan
8048d22a174Sdan return ret;
8058d22a174Sdan }
8068d22a174Sdan
807bcf4f484Sdrh #if defined(SQLITE_TEST) && defined(SQLITE_ENABLE_UNLOCK_NOTIFY)
setTestUnlockNotifyVars(Tcl_Interp * interp,int iArg,int nArg)808404ca075Sdanielk1977 static void setTestUnlockNotifyVars(Tcl_Interp *interp, int iArg, int nArg){
809404ca075Sdanielk1977 char zBuf[64];
81065545b59Sdrh sqlite3_snprintf(sizeof(zBuf), zBuf, "%d", iArg);
811404ca075Sdanielk1977 Tcl_SetVar(interp, "sqlite_unlock_notify_arg", zBuf, TCL_GLOBAL_ONLY);
81265545b59Sdrh sqlite3_snprintf(sizeof(zBuf), zBuf, "%d", nArg);
813404ca075Sdanielk1977 Tcl_SetVar(interp, "sqlite_unlock_notify_argcount", zBuf, TCL_GLOBAL_ONLY);
814404ca075Sdanielk1977 }
815404ca075Sdanielk1977 #else
816404ca075Sdanielk1977 # define setTestUnlockNotifyVars(x,y,z)
817404ca075Sdanielk1977 #endif
818404ca075Sdanielk1977
81969910da9Sdrh #ifdef SQLITE_ENABLE_UNLOCK_NOTIFY
DbUnlockNotify(void ** apArg,int nArg)820404ca075Sdanielk1977 static void DbUnlockNotify(void **apArg, int nArg){
821404ca075Sdanielk1977 int i;
822404ca075Sdanielk1977 for(i=0; i<nArg; i++){
823404ca075Sdanielk1977 const int flags = (TCL_EVAL_GLOBAL|TCL_EVAL_DIRECT);
824404ca075Sdanielk1977 SqliteDb *pDb = (SqliteDb *)apArg[i];
825404ca075Sdanielk1977 setTestUnlockNotifyVars(pDb->interp, i, nArg);
826404ca075Sdanielk1977 assert( pDb->pUnlockNotify);
827404ca075Sdanielk1977 Tcl_EvalObjEx(pDb->interp, pDb->pUnlockNotify, flags);
828404ca075Sdanielk1977 Tcl_DecrRefCount(pDb->pUnlockNotify);
829404ca075Sdanielk1977 pDb->pUnlockNotify = 0;
830404ca075Sdanielk1977 }
831404ca075Sdanielk1977 }
83269910da9Sdrh #endif
833404ca075Sdanielk1977
8349b1c62d4Sdrh #ifdef SQLITE_ENABLE_PREUPDATE_HOOK
83546c47d46Sdan /*
83646c47d46Sdan ** Pre-update hook callback.
83746c47d46Sdan */
DbPreUpdateHandler(void * p,sqlite3 * db,int op,const char * zDb,const char * zTbl,sqlite_int64 iKey1,sqlite_int64 iKey2)83846c47d46Sdan static void DbPreUpdateHandler(
83946c47d46Sdan void *p,
84046c47d46Sdan sqlite3 *db,
84146c47d46Sdan int op,
84246c47d46Sdan const char *zDb,
84346c47d46Sdan const char *zTbl,
84446c47d46Sdan sqlite_int64 iKey1,
84546c47d46Sdan sqlite_int64 iKey2
84646c47d46Sdan ){
84746c47d46Sdan SqliteDb *pDb = (SqliteDb *)p;
84846c47d46Sdan Tcl_Obj *pCmd;
84946c47d46Sdan static const char *azStr[] = {"DELETE", "INSERT", "UPDATE"};
85046c47d46Sdan
85146c47d46Sdan assert( (SQLITE_DELETE-1)/9 == 0 );
85246c47d46Sdan assert( (SQLITE_INSERT-1)/9 == 1 );
85346c47d46Sdan assert( (SQLITE_UPDATE-1)/9 == 2 );
85446c47d46Sdan assert( pDb->pPreUpdateHook );
85546c47d46Sdan assert( db==pDb->db );
85646c47d46Sdan assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE );
85746c47d46Sdan
85846c47d46Sdan pCmd = Tcl_DuplicateObj(pDb->pPreUpdateHook);
85946c47d46Sdan Tcl_IncrRefCount(pCmd);
86046c47d46Sdan Tcl_ListObjAppendElement(0, pCmd, Tcl_NewStringObj(azStr[(op-1)/9], -1));
86146c47d46Sdan Tcl_ListObjAppendElement(0, pCmd, Tcl_NewStringObj(zDb, -1));
86246c47d46Sdan Tcl_ListObjAppendElement(0, pCmd, Tcl_NewStringObj(zTbl, -1));
86346c47d46Sdan Tcl_ListObjAppendElement(0, pCmd, Tcl_NewWideIntObj(iKey1));
86446c47d46Sdan Tcl_ListObjAppendElement(0, pCmd, Tcl_NewWideIntObj(iKey2));
86546c47d46Sdan Tcl_EvalObjEx(pDb->interp, pCmd, TCL_EVAL_DIRECT);
86646c47d46Sdan Tcl_DecrRefCount(pCmd);
86746c47d46Sdan }
8689b1c62d4Sdrh #endif /* SQLITE_ENABLE_PREUPDATE_HOOK */
86946c47d46Sdan
DbUpdateHandler(void * p,int op,const char * zDb,const char * zTbl,sqlite_int64 rowid)87094eb6a14Sdanielk1977 static void DbUpdateHandler(
87194eb6a14Sdanielk1977 void *p,
87294eb6a14Sdanielk1977 int op,
87394eb6a14Sdanielk1977 const char *zDb,
87494eb6a14Sdanielk1977 const char *zTbl,
87594eb6a14Sdanielk1977 sqlite_int64 rowid
87694eb6a14Sdanielk1977 ){
87794eb6a14Sdanielk1977 SqliteDb *pDb = (SqliteDb *)p;
87894eb6a14Sdanielk1977 Tcl_Obj *pCmd;
87946c47d46Sdan static const char *azStr[] = {"DELETE", "INSERT", "UPDATE"};
88046c47d46Sdan
88146c47d46Sdan assert( (SQLITE_DELETE-1)/9 == 0 );
88246c47d46Sdan assert( (SQLITE_INSERT-1)/9 == 1 );
88346c47d46Sdan assert( (SQLITE_UPDATE-1)/9 == 2 );
88494eb6a14Sdanielk1977
88594eb6a14Sdanielk1977 assert( pDb->pUpdateHook );
88694eb6a14Sdanielk1977 assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE );
88794eb6a14Sdanielk1977
88894eb6a14Sdanielk1977 pCmd = Tcl_DuplicateObj(pDb->pUpdateHook);
88994eb6a14Sdanielk1977 Tcl_IncrRefCount(pCmd);
89046c47d46Sdan Tcl_ListObjAppendElement(0, pCmd, Tcl_NewStringObj(azStr[(op-1)/9], -1));
89194eb6a14Sdanielk1977 Tcl_ListObjAppendElement(0, pCmd, Tcl_NewStringObj(zDb, -1));
89294eb6a14Sdanielk1977 Tcl_ListObjAppendElement(0, pCmd, Tcl_NewStringObj(zTbl, -1));
89394eb6a14Sdanielk1977 Tcl_ListObjAppendElement(0, pCmd, Tcl_NewWideIntObj(rowid));
89494eb6a14Sdanielk1977 Tcl_EvalObjEx(pDb->interp, pCmd, TCL_EVAL_DIRECT);
895efdde169Sdrh Tcl_DecrRefCount(pCmd);
89694eb6a14Sdanielk1977 }
89794eb6a14Sdanielk1977
tclCollateNeeded(void * pCtx,sqlite3 * db,int enc,const char * zName)8987cedc8d4Sdanielk1977 static void tclCollateNeeded(
8997cedc8d4Sdanielk1977 void *pCtx,
9009bb575fdSdrh sqlite3 *db,
9017cedc8d4Sdanielk1977 int enc,
9027cedc8d4Sdanielk1977 const char *zName
9037cedc8d4Sdanielk1977 ){
9047cedc8d4Sdanielk1977 SqliteDb *pDb = (SqliteDb *)pCtx;
9057cedc8d4Sdanielk1977 Tcl_Obj *pScript = Tcl_DuplicateObj(pDb->pCollateNeeded);
9067cedc8d4Sdanielk1977 Tcl_IncrRefCount(pScript);
9077cedc8d4Sdanielk1977 Tcl_ListObjAppendElement(0, pScript, Tcl_NewStringObj(zName, -1));
9087cedc8d4Sdanielk1977 Tcl_EvalObjEx(pDb->interp, pScript, 0);
9097cedc8d4Sdanielk1977 Tcl_DecrRefCount(pScript);
9107cedc8d4Sdanielk1977 }
9117cedc8d4Sdanielk1977
912aa940eacSdrh /*
9130202b29eSdanielk1977 ** This routine is called to evaluate an SQL collation function implemented
9140202b29eSdanielk1977 ** using TCL script.
9150202b29eSdanielk1977 */
tclSqlCollate(void * pCtx,int nA,const void * zA,int nB,const void * zB)9160202b29eSdanielk1977 static int tclSqlCollate(
9170202b29eSdanielk1977 void *pCtx,
9180202b29eSdanielk1977 int nA,
9190202b29eSdanielk1977 const void *zA,
9200202b29eSdanielk1977 int nB,
9210202b29eSdanielk1977 const void *zB
9220202b29eSdanielk1977 ){
9230202b29eSdanielk1977 SqlCollate *p = (SqlCollate *)pCtx;
9240202b29eSdanielk1977 Tcl_Obj *pCmd;
9250202b29eSdanielk1977
9260202b29eSdanielk1977 pCmd = Tcl_NewStringObj(p->zScript, -1);
9270202b29eSdanielk1977 Tcl_IncrRefCount(pCmd);
9280202b29eSdanielk1977 Tcl_ListObjAppendElement(p->interp, pCmd, Tcl_NewStringObj(zA, nA));
9290202b29eSdanielk1977 Tcl_ListObjAppendElement(p->interp, pCmd, Tcl_NewStringObj(zB, nB));
930d1e4733dSdrh Tcl_EvalObjEx(p->interp, pCmd, TCL_EVAL_DIRECT);
9310202b29eSdanielk1977 Tcl_DecrRefCount(pCmd);
9320202b29eSdanielk1977 return (atoi(Tcl_GetStringResult(p->interp)));
9330202b29eSdanielk1977 }
9340202b29eSdanielk1977
9350202b29eSdanielk1977 /*
936cabb0819Sdrh ** This routine is called to evaluate an SQL function implemented
937cabb0819Sdrh ** using TCL script.
938cabb0819Sdrh */
tclSqlFunc(sqlite3_context * context,int argc,sqlite3_value ** argv)9390ae8b831Sdanielk1977 static void tclSqlFunc(sqlite3_context *context, int argc, sqlite3_value**argv){
9406f8a503dSdanielk1977 SqlFunc *p = sqlite3_user_data(context);
941d1e4733dSdrh Tcl_Obj *pCmd;
942cabb0819Sdrh int i;
943cabb0819Sdrh int rc;
944cabb0819Sdrh
945d1e4733dSdrh if( argc==0 ){
946d1e4733dSdrh /* If there are no arguments to the function, call Tcl_EvalObjEx on the
947d1e4733dSdrh ** script object directly. This allows the TCL compiler to generate
948d1e4733dSdrh ** bytecode for the command on the first invocation and thus make
949d1e4733dSdrh ** subsequent invocations much faster. */
950d1e4733dSdrh pCmd = p->pScript;
951d1e4733dSdrh Tcl_IncrRefCount(pCmd);
952d1e4733dSdrh rc = Tcl_EvalObjEx(p->interp, pCmd, 0);
953d1e4733dSdrh Tcl_DecrRefCount(pCmd);
95451ad0ecdSdanielk1977 }else{
955d1e4733dSdrh /* If there are arguments to the function, make a shallow copy of the
956d1e4733dSdrh ** script object, lappend the arguments, then evaluate the copy.
957d1e4733dSdrh **
95860ec914cSpeter.d.reid ** By "shallow" copy, we mean only the outer list Tcl_Obj is duplicated.
959d1e4733dSdrh ** The new Tcl_Obj contains pointers to the original list elements.
960d1e4733dSdrh ** That way, when Tcl_EvalObjv() is run and shimmers the first element
961d1e4733dSdrh ** of the list to tclCmdNameType, that alternate representation will
962d1e4733dSdrh ** be preserved and reused on the next invocation.
963d1e4733dSdrh */
964d1e4733dSdrh Tcl_Obj **aArg;
965d1e4733dSdrh int nArg;
966d1e4733dSdrh if( Tcl_ListObjGetElements(p->interp, p->pScript, &nArg, &aArg) ){
967d1e4733dSdrh sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1);
968d1e4733dSdrh return;
969d1e4733dSdrh }
970d1e4733dSdrh pCmd = Tcl_NewListObj(nArg, aArg);
971d1e4733dSdrh Tcl_IncrRefCount(pCmd);
972d1e4733dSdrh for(i=0; i<argc; i++){
973d1e4733dSdrh sqlite3_value *pIn = argv[i];
974d1e4733dSdrh Tcl_Obj *pVal;
975d1e4733dSdrh
976d1e4733dSdrh /* Set pVal to contain the i'th column of this row. */
977d1e4733dSdrh switch( sqlite3_value_type(pIn) ){
978d1e4733dSdrh case SQLITE_BLOB: {
979d1e4733dSdrh int bytes = sqlite3_value_bytes(pIn);
980d1e4733dSdrh pVal = Tcl_NewByteArrayObj(sqlite3_value_blob(pIn), bytes);
981d1e4733dSdrh break;
982d1e4733dSdrh }
983d1e4733dSdrh case SQLITE_INTEGER: {
984d1e4733dSdrh sqlite_int64 v = sqlite3_value_int64(pIn);
985d1e4733dSdrh if( v>=-2147483647 && v<=2147483647 ){
9867fd33929Sdrh pVal = Tcl_NewIntObj((int)v);
987d1e4733dSdrh }else{
988d1e4733dSdrh pVal = Tcl_NewWideIntObj(v);
989d1e4733dSdrh }
990d1e4733dSdrh break;
991d1e4733dSdrh }
992d1e4733dSdrh case SQLITE_FLOAT: {
993d1e4733dSdrh double r = sqlite3_value_double(pIn);
994d1e4733dSdrh pVal = Tcl_NewDoubleObj(r);
995d1e4733dSdrh break;
996d1e4733dSdrh }
997d1e4733dSdrh case SQLITE_NULL: {
998c45e6716Sdrh pVal = Tcl_NewStringObj(p->pDb->zNull, -1);
999d1e4733dSdrh break;
1000d1e4733dSdrh }
1001d1e4733dSdrh default: {
1002d1e4733dSdrh int bytes = sqlite3_value_bytes(pIn);
100300fd957bSdanielk1977 pVal = Tcl_NewStringObj((char *)sqlite3_value_text(pIn), bytes);
1004d1e4733dSdrh break;
100551ad0ecdSdanielk1977 }
1006cabb0819Sdrh }
1007d1e4733dSdrh rc = Tcl_ListObjAppendElement(p->interp, pCmd, pVal);
1008d1e4733dSdrh if( rc ){
1009d1e4733dSdrh Tcl_DecrRefCount(pCmd);
1010d1e4733dSdrh sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1);
1011d1e4733dSdrh return;
1012d1e4733dSdrh }
1013d1e4733dSdrh }
1014d1e4733dSdrh if( !p->useEvalObjv ){
1015d1e4733dSdrh /* Tcl_EvalObjEx() will automatically call Tcl_EvalObjv() if pCmd
1016d1e4733dSdrh ** is a list without a string representation. To prevent this from
1017d1e4733dSdrh ** happening, make sure pCmd has a valid string representation */
1018d1e4733dSdrh Tcl_GetString(pCmd);
1019d1e4733dSdrh }
1020d1e4733dSdrh rc = Tcl_EvalObjEx(p->interp, pCmd, TCL_EVAL_DIRECT);
1021d1e4733dSdrh Tcl_DecrRefCount(pCmd);
1022d1e4733dSdrh }
1023562e8d3cSdanielk1977
1024c7f269d5Sdrh if( rc && rc!=TCL_RETURN ){
10257e18c259Sdanielk1977 sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1);
1026cabb0819Sdrh }else{
1027c7f269d5Sdrh Tcl_Obj *pVar = Tcl_GetObjResult(p->interp);
1028c7f269d5Sdrh int n;
1029c7f269d5Sdrh u8 *data;
10304a4c11aaSdan const char *zType = (pVar->typePtr ? pVar->typePtr->name : "");
1031c7f269d5Sdrh char c = zType[0];
103289d24936Sdan int eType = p->eType;
103389d24936Sdan
103489d24936Sdan if( eType==SQLITE_NULL ){
1035df0bddaeSdrh if( c=='b' && strcmp(zType,"bytearray")==0 && pVar->bytes==0 ){
1036d1e4733dSdrh /* Only return a BLOB type if the Tcl variable is a bytearray and
1037df0bddaeSdrh ** has no string representation. */
103889d24936Sdan eType = SQLITE_BLOB;
103989d24936Sdan }else if( (c=='b' && strcmp(zType,"boolean")==0)
104089d24936Sdan || (c=='w' && strcmp(zType,"wideInt")==0)
104189d24936Sdan || (c=='i' && strcmp(zType,"int")==0)
104289d24936Sdan ){
104389d24936Sdan eType = SQLITE_INTEGER;
104489d24936Sdan }else if( c=='d' && strcmp(zType,"double")==0 ){
104589d24936Sdan eType = SQLITE_FLOAT;
104689d24936Sdan }else{
104789d24936Sdan eType = SQLITE_TEXT;
104889d24936Sdan }
104989d24936Sdan }
105089d24936Sdan
105189d24936Sdan switch( eType ){
105289d24936Sdan case SQLITE_BLOB: {
1053c7f269d5Sdrh data = Tcl_GetByteArrayFromObj(pVar, &n);
1054c7f269d5Sdrh sqlite3_result_blob(context, data, n, SQLITE_TRANSIENT);
105589d24936Sdan break;
105689d24936Sdan }
105789d24936Sdan case SQLITE_INTEGER: {
1058df0bddaeSdrh Tcl_WideInt v;
105989d24936Sdan if( TCL_OK==Tcl_GetWideIntFromObj(0, pVar, &v) ){
1060df0bddaeSdrh sqlite3_result_int64(context, v);
106189d24936Sdan break;
106289d24936Sdan }
106389d24936Sdan /* fall-through */
106489d24936Sdan }
106589d24936Sdan case SQLITE_FLOAT: {
106689d24936Sdan double r;
106789d24936Sdan if( TCL_OK==Tcl_GetDoubleFromObj(0, pVar, &r) ){
106889d24936Sdan sqlite3_result_double(context, r);
106989d24936Sdan break;
107089d24936Sdan }
107189d24936Sdan /* fall-through */
107289d24936Sdan }
107389d24936Sdan default: {
107400fd957bSdanielk1977 data = (unsigned char *)Tcl_GetStringFromObj(pVar, &n);
107500fd957bSdanielk1977 sqlite3_result_text(context, (char *)data, n, SQLITE_TRANSIENT);
107689d24936Sdan break;
1077c7f269d5Sdrh }
1078cabb0819Sdrh }
107989d24936Sdan
108089d24936Sdan }
1081cabb0819Sdrh }
1082895d7472Sdrh
1083e22a334bSdrh #ifndef SQLITE_OMIT_AUTHORIZATION
1084e22a334bSdrh /*
1085e22a334bSdrh ** This is the authentication function. It appends the authentication
1086e22a334bSdrh ** type code and the two arguments to zCmd[] then invokes the result
1087e22a334bSdrh ** on the interpreter. The reply is examined to determine if the
1088e22a334bSdrh ** authentication fails or succeeds.
1089e22a334bSdrh */
auth_callback(void * pArg,int code,const char * zArg1,const char * zArg2,const char * zArg3,const char * zArg4,const char * zArg5)1090e22a334bSdrh static int auth_callback(
1091e22a334bSdrh void *pArg,
1092e22a334bSdrh int code,
1093e22a334bSdrh const char *zArg1,
1094e22a334bSdrh const char *zArg2,
1095e22a334bSdrh const char *zArg3,
1096e22a334bSdrh const char *zArg4
109732c6a48bSdrh #ifdef SQLITE_USER_AUTHENTICATION
109832c6a48bSdrh ,const char *zArg5
109932c6a48bSdrh #endif
1100e22a334bSdrh ){
11016ef5e12eSmistachkin const char *zCode;
1102e22a334bSdrh Tcl_DString str;
1103e22a334bSdrh int rc;
1104e22a334bSdrh const char *zReply;
11059418921cSdrh /* EVIDENCE-OF: R-38590-62769 The first parameter to the authorizer
11069418921cSdrh ** callback is a copy of the third parameter to the
11079418921cSdrh ** sqlite3_set_authorizer() interface.
11089418921cSdrh */
1109e22a334bSdrh SqliteDb *pDb = (SqliteDb*)pArg;
11101f1549f8Sdrh if( pDb->disableAuth ) return SQLITE_OK;
1111e22a334bSdrh
11129418921cSdrh /* EVIDENCE-OF: R-56518-44310 The second parameter to the callback is an
11139418921cSdrh ** integer action code that specifies the particular action to be
11149418921cSdrh ** authorized. */
1115e22a334bSdrh switch( code ){
1116e22a334bSdrh case SQLITE_COPY : zCode="SQLITE_COPY"; break;
1117e22a334bSdrh case SQLITE_CREATE_INDEX : zCode="SQLITE_CREATE_INDEX"; break;
1118e22a334bSdrh case SQLITE_CREATE_TABLE : zCode="SQLITE_CREATE_TABLE"; break;
1119e22a334bSdrh case SQLITE_CREATE_TEMP_INDEX : zCode="SQLITE_CREATE_TEMP_INDEX"; break;
1120e22a334bSdrh case SQLITE_CREATE_TEMP_TABLE : zCode="SQLITE_CREATE_TEMP_TABLE"; break;
1121e22a334bSdrh case SQLITE_CREATE_TEMP_TRIGGER: zCode="SQLITE_CREATE_TEMP_TRIGGER"; break;
1122e22a334bSdrh case SQLITE_CREATE_TEMP_VIEW : zCode="SQLITE_CREATE_TEMP_VIEW"; break;
1123e22a334bSdrh case SQLITE_CREATE_TRIGGER : zCode="SQLITE_CREATE_TRIGGER"; break;
1124e22a334bSdrh case SQLITE_CREATE_VIEW : zCode="SQLITE_CREATE_VIEW"; break;
1125e22a334bSdrh case SQLITE_DELETE : zCode="SQLITE_DELETE"; break;
1126e22a334bSdrh case SQLITE_DROP_INDEX : zCode="SQLITE_DROP_INDEX"; break;
1127e22a334bSdrh case SQLITE_DROP_TABLE : zCode="SQLITE_DROP_TABLE"; break;
1128e22a334bSdrh case SQLITE_DROP_TEMP_INDEX : zCode="SQLITE_DROP_TEMP_INDEX"; break;
1129e22a334bSdrh case SQLITE_DROP_TEMP_TABLE : zCode="SQLITE_DROP_TEMP_TABLE"; break;
1130e22a334bSdrh case SQLITE_DROP_TEMP_TRIGGER : zCode="SQLITE_DROP_TEMP_TRIGGER"; break;
1131e22a334bSdrh case SQLITE_DROP_TEMP_VIEW : zCode="SQLITE_DROP_TEMP_VIEW"; break;
1132e22a334bSdrh case SQLITE_DROP_TRIGGER : zCode="SQLITE_DROP_TRIGGER"; break;
1133e22a334bSdrh case SQLITE_DROP_VIEW : zCode="SQLITE_DROP_VIEW"; break;
1134e22a334bSdrh case SQLITE_INSERT : zCode="SQLITE_INSERT"; break;
1135e22a334bSdrh case SQLITE_PRAGMA : zCode="SQLITE_PRAGMA"; break;
1136e22a334bSdrh case SQLITE_READ : zCode="SQLITE_READ"; break;
1137e22a334bSdrh case SQLITE_SELECT : zCode="SQLITE_SELECT"; break;
1138e22a334bSdrh case SQLITE_TRANSACTION : zCode="SQLITE_TRANSACTION"; break;
1139e22a334bSdrh case SQLITE_UPDATE : zCode="SQLITE_UPDATE"; break;
114081e293b4Sdrh case SQLITE_ATTACH : zCode="SQLITE_ATTACH"; break;
114181e293b4Sdrh case SQLITE_DETACH : zCode="SQLITE_DETACH"; break;
11421c8c23ccSdanielk1977 case SQLITE_ALTER_TABLE : zCode="SQLITE_ALTER_TABLE"; break;
11431d54df88Sdanielk1977 case SQLITE_REINDEX : zCode="SQLITE_REINDEX"; break;
1144e6e04969Sdrh case SQLITE_ANALYZE : zCode="SQLITE_ANALYZE"; break;
1145f1a381e7Sdanielk1977 case SQLITE_CREATE_VTABLE : zCode="SQLITE_CREATE_VTABLE"; break;
1146f1a381e7Sdanielk1977 case SQLITE_DROP_VTABLE : zCode="SQLITE_DROP_VTABLE"; break;
11475169bbc6Sdrh case SQLITE_FUNCTION : zCode="SQLITE_FUNCTION"; break;
1148ab9b703fSdanielk1977 case SQLITE_SAVEPOINT : zCode="SQLITE_SAVEPOINT"; break;
114965a2aaa6Sdrh case SQLITE_RECURSIVE : zCode="SQLITE_RECURSIVE"; break;
1150e22a334bSdrh default : zCode="????"; break;
1151e22a334bSdrh }
1152e22a334bSdrh Tcl_DStringInit(&str);
1153e22a334bSdrh Tcl_DStringAppend(&str, pDb->zAuth, -1);
1154e22a334bSdrh Tcl_DStringAppendElement(&str, zCode);
1155e22a334bSdrh Tcl_DStringAppendElement(&str, zArg1 ? zArg1 : "");
1156e22a334bSdrh Tcl_DStringAppendElement(&str, zArg2 ? zArg2 : "");
1157e22a334bSdrh Tcl_DStringAppendElement(&str, zArg3 ? zArg3 : "");
1158e22a334bSdrh Tcl_DStringAppendElement(&str, zArg4 ? zArg4 : "");
115932c6a48bSdrh #ifdef SQLITE_USER_AUTHENTICATION
116032c6a48bSdrh Tcl_DStringAppendElement(&str, zArg5 ? zArg5 : "");
116132c6a48bSdrh #endif
1162e22a334bSdrh rc = Tcl_GlobalEval(pDb->interp, Tcl_DStringValue(&str));
1163e22a334bSdrh Tcl_DStringFree(&str);
1164b07028f7Sdrh zReply = rc==TCL_OK ? Tcl_GetStringResult(pDb->interp) : "SQLITE_DENY";
1165e22a334bSdrh if( strcmp(zReply,"SQLITE_OK")==0 ){
1166e22a334bSdrh rc = SQLITE_OK;
1167e22a334bSdrh }else if( strcmp(zReply,"SQLITE_DENY")==0 ){
1168e22a334bSdrh rc = SQLITE_DENY;
1169e22a334bSdrh }else if( strcmp(zReply,"SQLITE_IGNORE")==0 ){
1170e22a334bSdrh rc = SQLITE_IGNORE;
1171e22a334bSdrh }else{
1172e22a334bSdrh rc = 999;
1173e22a334bSdrh }
1174e22a334bSdrh return rc;
1175e22a334bSdrh }
1176e22a334bSdrh #endif /* SQLITE_OMIT_AUTHORIZATION */
1177cabb0819Sdrh
1178cabb0819Sdrh /*
11791067fe11Stpoindex ** This routine reads a line of text from FILE in, stores
11801067fe11Stpoindex ** the text in memory obtained from malloc() and returns a pointer
11811067fe11Stpoindex ** to the text. NULL is returned at end of file, or if malloc()
11821067fe11Stpoindex ** fails.
11831067fe11Stpoindex **
11841067fe11Stpoindex ** The interface is like "readline" but no command-line editing
11851067fe11Stpoindex ** is done.
11861067fe11Stpoindex **
11871067fe11Stpoindex ** copied from shell.c from '.import' command
11881067fe11Stpoindex */
local_getline(char * zPrompt,FILE * in)11891067fe11Stpoindex static char *local_getline(char *zPrompt, FILE *in){
11901067fe11Stpoindex char *zLine;
11911067fe11Stpoindex int nLine;
11921067fe11Stpoindex int n;
11931067fe11Stpoindex
11941067fe11Stpoindex nLine = 100;
11951067fe11Stpoindex zLine = malloc( nLine );
11961067fe11Stpoindex if( zLine==0 ) return 0;
11971067fe11Stpoindex n = 0;
1198b07028f7Sdrh while( 1 ){
11991067fe11Stpoindex if( n+100>nLine ){
12001067fe11Stpoindex nLine = nLine*2 + 100;
12011067fe11Stpoindex zLine = realloc(zLine, nLine);
12021067fe11Stpoindex if( zLine==0 ) return 0;
12031067fe11Stpoindex }
12041067fe11Stpoindex if( fgets(&zLine[n], nLine - n, in)==0 ){
12051067fe11Stpoindex if( n==0 ){
12061067fe11Stpoindex free(zLine);
12071067fe11Stpoindex return 0;
12081067fe11Stpoindex }
12091067fe11Stpoindex zLine[n] = 0;
12101067fe11Stpoindex break;
12111067fe11Stpoindex }
12121067fe11Stpoindex while( zLine[n] ){ n++; }
12131067fe11Stpoindex if( n>0 && zLine[n-1]=='\n' ){
12141067fe11Stpoindex n--;
12151067fe11Stpoindex zLine[n] = 0;
1216b07028f7Sdrh break;
12171067fe11Stpoindex }
12181067fe11Stpoindex }
12191067fe11Stpoindex zLine = realloc( zLine, n+1 );
12201067fe11Stpoindex return zLine;
12211067fe11Stpoindex }
12221067fe11Stpoindex
12238e556520Sdanielk1977
12248e556520Sdanielk1977 /*
12254a4c11aaSdan ** This function is part of the implementation of the command:
12268e556520Sdanielk1977 **
12274a4c11aaSdan ** $db transaction [-deferred|-immediate|-exclusive] SCRIPT
12288e556520Sdanielk1977 **
12294a4c11aaSdan ** It is invoked after evaluating the script SCRIPT to commit or rollback
12304a4c11aaSdan ** the transaction or savepoint opened by the [transaction] command.
12314a4c11aaSdan */
DbTransPostCmd(ClientData data[],Tcl_Interp * interp,int result)1232a121cc7cSmistachkin static int SQLITE_TCLAPI DbTransPostCmd(
12334a4c11aaSdan ClientData data[], /* data[0] is the Sqlite3Db* for $db */
12344a4c11aaSdan Tcl_Interp *interp, /* Tcl interpreter */
12354a4c11aaSdan int result /* Result of evaluating SCRIPT */
12364a4c11aaSdan ){
12376ef5e12eSmistachkin static const char *const azEnd[] = {
12384a4c11aaSdan "RELEASE _tcl_transaction", /* rc==TCL_ERROR, nTransaction!=0 */
12394a4c11aaSdan "COMMIT", /* rc!=TCL_ERROR, nTransaction==0 */
12404a4c11aaSdan "ROLLBACK TO _tcl_transaction ; RELEASE _tcl_transaction",
12414a4c11aaSdan "ROLLBACK" /* rc==TCL_ERROR, nTransaction==0 */
12424a4c11aaSdan };
12434a4c11aaSdan SqliteDb *pDb = (SqliteDb*)data[0];
12444a4c11aaSdan int rc = result;
12454a4c11aaSdan const char *zEnd;
12464a4c11aaSdan
12474a4c11aaSdan pDb->nTransaction--;
12484a4c11aaSdan zEnd = azEnd[(rc==TCL_ERROR)*2 + (pDb->nTransaction==0)];
12494a4c11aaSdan
12504a4c11aaSdan pDb->disableAuth++;
12514a4c11aaSdan if( sqlite3_exec(pDb->db, zEnd, 0, 0, 0) ){
12524a4c11aaSdan /* This is a tricky scenario to handle. The most likely cause of an
12534a4c11aaSdan ** error is that the exec() above was an attempt to commit the
12544a4c11aaSdan ** top-level transaction that returned SQLITE_BUSY. Or, less likely,
125548864df9Smistachkin ** that an IO-error has occurred. In either case, throw a Tcl exception
12564a4c11aaSdan ** and try to rollback the transaction.
12574a4c11aaSdan **
12584a4c11aaSdan ** But it could also be that the user executed one or more BEGIN,
12594a4c11aaSdan ** COMMIT, SAVEPOINT, RELEASE or ROLLBACK commands that are confusing
12604a4c11aaSdan ** this method's logic. Not clear how this would be best handled.
12614a4c11aaSdan */
12624a4c11aaSdan if( rc!=TCL_ERROR ){
1263a198f2b5Sdrh Tcl_AppendResult(interp, sqlite3_errmsg(pDb->db), (char*)0);
12644a4c11aaSdan rc = TCL_ERROR;
12654a4c11aaSdan }
12664a4c11aaSdan sqlite3_exec(pDb->db, "ROLLBACK", 0, 0, 0);
12674a4c11aaSdan }
12684a4c11aaSdan pDb->disableAuth--;
12694a4c11aaSdan
1270bea28c73Sdan delDatabaseRef(pDb);
12714a4c11aaSdan return rc;
12724a4c11aaSdan }
12734a4c11aaSdan
12744a4c11aaSdan /*
1275c431fd55Sdan ** Unless SQLITE_TEST is defined, this function is a simple wrapper around
1276c431fd55Sdan ** sqlite3_prepare_v2(). If SQLITE_TEST is defined, then it uses either
1277c431fd55Sdan ** sqlite3_prepare_v2() or legacy interface sqlite3_prepare(), depending
1278c431fd55Sdan ** on whether or not the [db_use_legacy_prepare] command has been used to
1279c431fd55Sdan ** configure the connection.
1280c431fd55Sdan */
dbPrepare(SqliteDb * pDb,const char * zSql,sqlite3_stmt ** ppStmt,const char ** pzOut)1281c431fd55Sdan static int dbPrepare(
1282c431fd55Sdan SqliteDb *pDb, /* Database object */
1283c431fd55Sdan const char *zSql, /* SQL to compile */
1284c431fd55Sdan sqlite3_stmt **ppStmt, /* OUT: Prepared statement */
1285c431fd55Sdan const char **pzOut /* OUT: Pointer to next SQL statement */
1286c431fd55Sdan ){
12872c2f392dSdrh unsigned int prepFlags = 0;
1288c431fd55Sdan #ifdef SQLITE_TEST
1289c431fd55Sdan if( pDb->bLegacyPrepare ){
1290c431fd55Sdan return sqlite3_prepare(pDb->db, zSql, -1, ppStmt, pzOut);
1291c431fd55Sdan }
1292c431fd55Sdan #endif
12932c2f392dSdrh /* If the statement cache is large, use the SQLITE_PREPARE_PERSISTENT
12942c2f392dSdrh ** flags, which uses less lookaside memory. But if the cache is small,
12952c2f392dSdrh ** omit that flag to make full use of lookaside */
12962c2f392dSdrh if( pDb->maxStmt>5 ) prepFlags = SQLITE_PREPARE_PERSISTENT;
12972c2f392dSdrh
12982c2f392dSdrh return sqlite3_prepare_v3(pDb->db, zSql, -1, prepFlags, ppStmt, pzOut);
1299c431fd55Sdan }
1300c431fd55Sdan
1301c431fd55Sdan /*
13024a4c11aaSdan ** Search the cache for a prepared-statement object that implements the
13034a4c11aaSdan ** first SQL statement in the buffer pointed to by parameter zIn. If
13044a4c11aaSdan ** no such prepared-statement can be found, allocate and prepare a new
13054a4c11aaSdan ** one. In either case, bind the current values of the relevant Tcl
13064a4c11aaSdan ** variables to any $var, :var or @var variables in the statement. Before
13074a4c11aaSdan ** returning, set *ppPreStmt to point to the prepared-statement object.
13084a4c11aaSdan **
13094a4c11aaSdan ** Output parameter *pzOut is set to point to the next SQL statement in
13104a4c11aaSdan ** buffer zIn, or to the '\0' byte at the end of zIn if there is no
13114a4c11aaSdan ** next statement.
13124a4c11aaSdan **
13134a4c11aaSdan ** If successful, TCL_OK is returned. Otherwise, TCL_ERROR is returned
13144a4c11aaSdan ** and an error message loaded into interpreter pDb->interp.
13154a4c11aaSdan */
dbPrepareAndBind(SqliteDb * pDb,char const * zIn,char const ** pzOut,SqlPreparedStmt ** ppPreStmt)13164a4c11aaSdan static int dbPrepareAndBind(
13174a4c11aaSdan SqliteDb *pDb, /* Database object */
13184a4c11aaSdan char const *zIn, /* SQL to compile */
13194a4c11aaSdan char const **pzOut, /* OUT: Pointer to next SQL statement */
13204a4c11aaSdan SqlPreparedStmt **ppPreStmt /* OUT: Object used to cache statement */
13214a4c11aaSdan ){
13224a4c11aaSdan const char *zSql = zIn; /* Pointer to first SQL statement in zIn */
13237bb22ac7Smistachkin sqlite3_stmt *pStmt = 0; /* Prepared statement object */
13244a4c11aaSdan SqlPreparedStmt *pPreStmt; /* Pointer to cached statement */
13254a4c11aaSdan int nSql; /* Length of zSql in bytes */
13267bb22ac7Smistachkin int nVar = 0; /* Number of variables in statement */
13274a4c11aaSdan int iParm = 0; /* Next free entry in apParm */
13280425f189Sdrh char c;
13294a4c11aaSdan int i;
1330c06ede10Sdrh int needResultReset = 0; /* Need to invoke Tcl_ResetResult() */
1331c06ede10Sdrh int rc = SQLITE_OK; /* Value to return */
13324a4c11aaSdan Tcl_Interp *interp = pDb->interp;
13334a4c11aaSdan
13344a4c11aaSdan *ppPreStmt = 0;
13354a4c11aaSdan
13364a4c11aaSdan /* Trim spaces from the start of zSql and calculate the remaining length. */
13370425f189Sdrh while( (c = zSql[0])==' ' || c=='\t' || c=='\r' || c=='\n' ){ zSql++; }
13384a4c11aaSdan nSql = strlen30(zSql);
13394a4c11aaSdan
13404a4c11aaSdan for(pPreStmt = pDb->stmtList; pPreStmt; pPreStmt=pPreStmt->pNext){
13414a4c11aaSdan int n = pPreStmt->nSql;
13424a4c11aaSdan if( nSql>=n
13434a4c11aaSdan && memcmp(pPreStmt->zSql, zSql, n)==0
13444a4c11aaSdan && (zSql[n]==0 || zSql[n-1]==';')
13454a4c11aaSdan ){
13464a4c11aaSdan pStmt = pPreStmt->pStmt;
13474a4c11aaSdan *pzOut = &zSql[pPreStmt->nSql];
13484a4c11aaSdan
13494a4c11aaSdan /* When a prepared statement is found, unlink it from the
13504a4c11aaSdan ** cache list. It will later be added back to the beginning
13514a4c11aaSdan ** of the cache list in order to implement LRU replacement.
13524a4c11aaSdan */
13534a4c11aaSdan if( pPreStmt->pPrev ){
13544a4c11aaSdan pPreStmt->pPrev->pNext = pPreStmt->pNext;
13554a4c11aaSdan }else{
13564a4c11aaSdan pDb->stmtList = pPreStmt->pNext;
13574a4c11aaSdan }
13584a4c11aaSdan if( pPreStmt->pNext ){
13594a4c11aaSdan pPreStmt->pNext->pPrev = pPreStmt->pPrev;
13604a4c11aaSdan }else{
13614a4c11aaSdan pDb->stmtLast = pPreStmt->pPrev;
13624a4c11aaSdan }
13634a4c11aaSdan pDb->nStmt--;
13644a4c11aaSdan nVar = sqlite3_bind_parameter_count(pStmt);
13654a4c11aaSdan break;
13664a4c11aaSdan }
13674a4c11aaSdan }
13684a4c11aaSdan
13694a4c11aaSdan /* If no prepared statement was found. Compile the SQL text. Also allocate
13704a4c11aaSdan ** a new SqlPreparedStmt structure. */
13714a4c11aaSdan if( pPreStmt==0 ){
13724a4c11aaSdan int nByte;
13734a4c11aaSdan
1374c431fd55Sdan if( SQLITE_OK!=dbPrepare(pDb, zSql, &pStmt, pzOut) ){
1375c45e6716Sdrh Tcl_SetObjResult(interp, Tcl_NewStringObj(sqlite3_errmsg(pDb->db), -1));
13764a4c11aaSdan return TCL_ERROR;
13774a4c11aaSdan }
13784a4c11aaSdan if( pStmt==0 ){
13794a4c11aaSdan if( SQLITE_OK!=sqlite3_errcode(pDb->db) ){
13804a4c11aaSdan /* A compile-time error in the statement. */
1381c45e6716Sdrh Tcl_SetObjResult(interp, Tcl_NewStringObj(sqlite3_errmsg(pDb->db), -1));
13824a4c11aaSdan return TCL_ERROR;
13834a4c11aaSdan }else{
13844a4c11aaSdan /* The statement was a no-op. Continue to the next statement
13854a4c11aaSdan ** in the SQL string.
13864a4c11aaSdan */
13874a4c11aaSdan return TCL_OK;
13884a4c11aaSdan }
13894a4c11aaSdan }
13904a4c11aaSdan
13914a4c11aaSdan assert( pPreStmt==0 );
13924a4c11aaSdan nVar = sqlite3_bind_parameter_count(pStmt);
13934a4c11aaSdan nByte = sizeof(SqlPreparedStmt) + nVar*sizeof(Tcl_Obj *);
13944a4c11aaSdan pPreStmt = (SqlPreparedStmt*)Tcl_Alloc(nByte);
13954a4c11aaSdan memset(pPreStmt, 0, nByte);
13964a4c11aaSdan
13974a4c11aaSdan pPreStmt->pStmt = pStmt;
13987ed243b7Sdrh pPreStmt->nSql = (int)(*pzOut - zSql);
13994a4c11aaSdan pPreStmt->zSql = sqlite3_sql(pStmt);
14004a4c11aaSdan pPreStmt->apParm = (Tcl_Obj **)&pPreStmt[1];
1401c431fd55Sdan #ifdef SQLITE_TEST
1402c431fd55Sdan if( pPreStmt->zSql==0 ){
1403c431fd55Sdan char *zCopy = Tcl_Alloc(pPreStmt->nSql + 1);
1404c431fd55Sdan memcpy(zCopy, zSql, pPreStmt->nSql);
1405c431fd55Sdan zCopy[pPreStmt->nSql] = '\0';
1406c431fd55Sdan pPreStmt->zSql = zCopy;
1407c431fd55Sdan }
1408c431fd55Sdan #endif
14094a4c11aaSdan }
14104a4c11aaSdan assert( pPreStmt );
14114a4c11aaSdan assert( strlen30(pPreStmt->zSql)==pPreStmt->nSql );
14124a4c11aaSdan assert( 0==memcmp(pPreStmt->zSql, zSql, pPreStmt->nSql) );
14134a4c11aaSdan
14144a4c11aaSdan /* Bind values to parameters that begin with $ or : */
14154a4c11aaSdan for(i=1; i<=nVar; i++){
14164a4c11aaSdan const char *zVar = sqlite3_bind_parameter_name(pStmt, i);
14174a4c11aaSdan if( zVar!=0 && (zVar[0]=='$' || zVar[0]==':' || zVar[0]=='@') ){
14184a4c11aaSdan Tcl_Obj *pVar = Tcl_GetVar2Ex(interp, &zVar[1], 0, 0);
1419c06ede10Sdrh if( pVar==0 && pDb->zBindFallback!=0 ){
1420c06ede10Sdrh Tcl_Obj *pCmd;
1421c06ede10Sdrh int rx;
1422c06ede10Sdrh pCmd = Tcl_NewStringObj(pDb->zBindFallback, -1);
1423c06ede10Sdrh Tcl_IncrRefCount(pCmd);
1424c06ede10Sdrh Tcl_ListObjAppendElement(interp, pCmd, Tcl_NewStringObj(zVar,-1));
1425c06ede10Sdrh if( needResultReset ) Tcl_ResetResult(interp);
1426c06ede10Sdrh needResultReset = 1;
1427c06ede10Sdrh rx = Tcl_EvalObjEx(interp, pCmd, TCL_EVAL_DIRECT);
1428c06ede10Sdrh Tcl_DecrRefCount(pCmd);
1429c06ede10Sdrh if( rx==TCL_OK ){
1430c06ede10Sdrh pVar = Tcl_GetObjResult(interp);
1431c06ede10Sdrh }else if( rx==TCL_ERROR ){
1432c06ede10Sdrh rc = TCL_ERROR;
1433c06ede10Sdrh break;
1434c06ede10Sdrh }else{
1435c06ede10Sdrh pVar = 0;
1436c06ede10Sdrh }
1437c06ede10Sdrh }
14384a4c11aaSdan if( pVar ){
14394a4c11aaSdan int n;
14404a4c11aaSdan u8 *data;
14414a4c11aaSdan const char *zType = (pVar->typePtr ? pVar->typePtr->name : "");
14428e18922fSmistachkin c = zType[0];
14434a4c11aaSdan if( zVar[0]=='@' ||
14444a4c11aaSdan (c=='b' && strcmp(zType,"bytearray")==0 && pVar->bytes==0) ){
14454a4c11aaSdan /* Load a BLOB type if the Tcl variable is a bytearray and
14464a4c11aaSdan ** it has no string representation or the host
14474a4c11aaSdan ** parameter name begins with "@". */
14484a4c11aaSdan data = Tcl_GetByteArrayFromObj(pVar, &n);
14494a4c11aaSdan sqlite3_bind_blob(pStmt, i, data, n, SQLITE_STATIC);
14504a4c11aaSdan Tcl_IncrRefCount(pVar);
14514a4c11aaSdan pPreStmt->apParm[iParm++] = pVar;
14524a4c11aaSdan }else if( c=='b' && strcmp(zType,"boolean")==0 ){
14534a4c11aaSdan Tcl_GetIntFromObj(interp, pVar, &n);
14544a4c11aaSdan sqlite3_bind_int(pStmt, i, n);
14554a4c11aaSdan }else if( c=='d' && strcmp(zType,"double")==0 ){
14564a4c11aaSdan double r;
14574a4c11aaSdan Tcl_GetDoubleFromObj(interp, pVar, &r);
14584a4c11aaSdan sqlite3_bind_double(pStmt, i, r);
14594a4c11aaSdan }else if( (c=='w' && strcmp(zType,"wideInt")==0) ||
14604a4c11aaSdan (c=='i' && strcmp(zType,"int")==0) ){
14614a4c11aaSdan Tcl_WideInt v;
14624a4c11aaSdan Tcl_GetWideIntFromObj(interp, pVar, &v);
14634a4c11aaSdan sqlite3_bind_int64(pStmt, i, v);
14644a4c11aaSdan }else{
14654a4c11aaSdan data = (unsigned char *)Tcl_GetStringFromObj(pVar, &n);
14664a4c11aaSdan sqlite3_bind_text(pStmt, i, (char *)data, n, SQLITE_STATIC);
14674a4c11aaSdan Tcl_IncrRefCount(pVar);
14684a4c11aaSdan pPreStmt->apParm[iParm++] = pVar;
14694a4c11aaSdan }
14704a4c11aaSdan }else{
14714a4c11aaSdan sqlite3_bind_null(pStmt, i);
14724a4c11aaSdan }
1473c06ede10Sdrh if( needResultReset ) Tcl_ResetResult(pDb->interp);
14744a4c11aaSdan }
14754a4c11aaSdan }
14764a4c11aaSdan pPreStmt->nParm = iParm;
14774a4c11aaSdan *ppPreStmt = pPreStmt;
1478c06ede10Sdrh if( needResultReset && rc==TCL_OK ) Tcl_ResetResult(pDb->interp);
1479937d0deaSdan
1480c06ede10Sdrh return rc;
14814a4c11aaSdan }
14824a4c11aaSdan
14834a4c11aaSdan /*
14844a4c11aaSdan ** Release a statement reference obtained by calling dbPrepareAndBind().
14854a4c11aaSdan ** There should be exactly one call to this function for each call to
14864a4c11aaSdan ** dbPrepareAndBind().
14874a4c11aaSdan **
14884a4c11aaSdan ** If the discard parameter is non-zero, then the statement is deleted
14894a4c11aaSdan ** immediately. Otherwise it is added to the LRU list and may be returned
14904a4c11aaSdan ** by a subsequent call to dbPrepareAndBind().
14914a4c11aaSdan */
dbReleaseStmt(SqliteDb * pDb,SqlPreparedStmt * pPreStmt,int discard)14924a4c11aaSdan static void dbReleaseStmt(
14934a4c11aaSdan SqliteDb *pDb, /* Database handle */
14944a4c11aaSdan SqlPreparedStmt *pPreStmt, /* Prepared statement handle to release */
14954a4c11aaSdan int discard /* True to delete (not cache) the pPreStmt */
14964a4c11aaSdan ){
14974a4c11aaSdan int i;
14984a4c11aaSdan
14994a4c11aaSdan /* Free the bound string and blob parameters */
15004a4c11aaSdan for(i=0; i<pPreStmt->nParm; i++){
15014a4c11aaSdan Tcl_DecrRefCount(pPreStmt->apParm[i]);
15024a4c11aaSdan }
15034a4c11aaSdan pPreStmt->nParm = 0;
15044a4c11aaSdan
15054a4c11aaSdan if( pDb->maxStmt<=0 || discard ){
15064a4c11aaSdan /* If the cache is turned off, deallocated the statement */
1507c431fd55Sdan dbFreeStmt(pPreStmt);
15084a4c11aaSdan }else{
15094a4c11aaSdan /* Add the prepared statement to the beginning of the cache list. */
15104a4c11aaSdan pPreStmt->pNext = pDb->stmtList;
15114a4c11aaSdan pPreStmt->pPrev = 0;
15124a4c11aaSdan if( pDb->stmtList ){
15134a4c11aaSdan pDb->stmtList->pPrev = pPreStmt;
15144a4c11aaSdan }
15154a4c11aaSdan pDb->stmtList = pPreStmt;
15164a4c11aaSdan if( pDb->stmtLast==0 ){
15174a4c11aaSdan assert( pDb->nStmt==0 );
15184a4c11aaSdan pDb->stmtLast = pPreStmt;
15194a4c11aaSdan }else{
15204a4c11aaSdan assert( pDb->nStmt>0 );
15214a4c11aaSdan }
15224a4c11aaSdan pDb->nStmt++;
15234a4c11aaSdan
15244a4c11aaSdan /* If we have too many statement in cache, remove the surplus from
15254a4c11aaSdan ** the end of the cache list. */
15264a4c11aaSdan while( pDb->nStmt>pDb->maxStmt ){
1527c431fd55Sdan SqlPreparedStmt *pLast = pDb->stmtLast;
1528c431fd55Sdan pDb->stmtLast = pLast->pPrev;
15294a4c11aaSdan pDb->stmtLast->pNext = 0;
15304a4c11aaSdan pDb->nStmt--;
1531c431fd55Sdan dbFreeStmt(pLast);
15324a4c11aaSdan }
15334a4c11aaSdan }
15344a4c11aaSdan }
15354a4c11aaSdan
15364a4c11aaSdan /*
15374a4c11aaSdan ** Structure used with dbEvalXXX() functions:
15384a4c11aaSdan **
15394a4c11aaSdan ** dbEvalInit()
15404a4c11aaSdan ** dbEvalStep()
15414a4c11aaSdan ** dbEvalFinalize()
15424a4c11aaSdan ** dbEvalRowInfo()
15434a4c11aaSdan ** dbEvalColumnValue()
15444a4c11aaSdan */
15454a4c11aaSdan typedef struct DbEvalContext DbEvalContext;
15464a4c11aaSdan struct DbEvalContext {
15474a4c11aaSdan SqliteDb *pDb; /* Database handle */
15484a4c11aaSdan Tcl_Obj *pSql; /* Object holding string zSql */
15494a4c11aaSdan const char *zSql; /* Remaining SQL to execute */
15504a4c11aaSdan SqlPreparedStmt *pPreStmt; /* Current statement */
15514a4c11aaSdan int nCol; /* Number of columns returned by pStmt */
1552af38cdbcSdrh int evalFlags; /* Flags used */
15534a4c11aaSdan Tcl_Obj *pArray; /* Name of array variable */
15544a4c11aaSdan Tcl_Obj **apColName; /* Array of column names */
15554a4c11aaSdan };
15564a4c11aaSdan
1557af38cdbcSdrh #define SQLITE_EVAL_WITHOUTNULLS 0x00001 /* Unset array(*) for NULL */
1558af38cdbcSdrh
15594a4c11aaSdan /*
15604a4c11aaSdan ** Release any cache of column names currently held as part of
15614a4c11aaSdan ** the DbEvalContext structure passed as the first argument.
15624a4c11aaSdan */
dbReleaseColumnNames(DbEvalContext * p)15634a4c11aaSdan static void dbReleaseColumnNames(DbEvalContext *p){
15644a4c11aaSdan if( p->apColName ){
15654a4c11aaSdan int i;
15664a4c11aaSdan for(i=0; i<p->nCol; i++){
15674a4c11aaSdan Tcl_DecrRefCount(p->apColName[i]);
15684a4c11aaSdan }
15694a4c11aaSdan Tcl_Free((char *)p->apColName);
15704a4c11aaSdan p->apColName = 0;
15714a4c11aaSdan }
15724a4c11aaSdan p->nCol = 0;
15734a4c11aaSdan }
15744a4c11aaSdan
15754a4c11aaSdan /*
15764a4c11aaSdan ** Initialize a DbEvalContext structure.
15778e556520Sdanielk1977 **
15788e556520Sdanielk1977 ** If pArray is not NULL, then it contains the name of a Tcl array
15798e556520Sdanielk1977 ** variable. The "*" member of this array is set to a list containing
15804a4c11aaSdan ** the names of the columns returned by the statement as part of each
15814a4c11aaSdan ** call to dbEvalStep(), in order from left to right. e.g. if the names
15824a4c11aaSdan ** of the returned columns are a, b and c, it does the equivalent of the
15834a4c11aaSdan ** tcl command:
15848e556520Sdanielk1977 **
15858e556520Sdanielk1977 ** set ${pArray}(*) {a b c}
15868e556520Sdanielk1977 */
dbEvalInit(DbEvalContext * p,SqliteDb * pDb,Tcl_Obj * pSql,Tcl_Obj * pArray,int evalFlags)15874a4c11aaSdan static void dbEvalInit(
15884a4c11aaSdan DbEvalContext *p, /* Pointer to structure to initialize */
15894a4c11aaSdan SqliteDb *pDb, /* Database handle */
15904a4c11aaSdan Tcl_Obj *pSql, /* Object containing SQL script */
1591af38cdbcSdrh Tcl_Obj *pArray, /* Name of Tcl array to set (*) element of */
1592af38cdbcSdrh int evalFlags /* Flags controlling evaluation */
15938e556520Sdanielk1977 ){
15944a4c11aaSdan memset(p, 0, sizeof(DbEvalContext));
15954a4c11aaSdan p->pDb = pDb;
15964a4c11aaSdan p->zSql = Tcl_GetString(pSql);
15974a4c11aaSdan p->pSql = pSql;
15984a4c11aaSdan Tcl_IncrRefCount(pSql);
15994a4c11aaSdan if( pArray ){
16004a4c11aaSdan p->pArray = pArray;
16014a4c11aaSdan Tcl_IncrRefCount(pArray);
16024a4c11aaSdan }
1603af38cdbcSdrh p->evalFlags = evalFlags;
1604bea28c73Sdan addDatabaseRef(p->pDb);
16054a4c11aaSdan }
16068e556520Sdanielk1977
16074a4c11aaSdan /*
16084a4c11aaSdan ** Obtain information about the row that the DbEvalContext passed as the
16094a4c11aaSdan ** first argument currently points to.
16104a4c11aaSdan */
dbEvalRowInfo(DbEvalContext * p,int * pnCol,Tcl_Obj *** papColName)16114a4c11aaSdan static void dbEvalRowInfo(
16124a4c11aaSdan DbEvalContext *p, /* Evaluation context */
16134a4c11aaSdan int *pnCol, /* OUT: Number of column names */
16144a4c11aaSdan Tcl_Obj ***papColName /* OUT: Array of column names */
16154a4c11aaSdan ){
16168e556520Sdanielk1977 /* Compute column names */
16174a4c11aaSdan if( 0==p->apColName ){
16184a4c11aaSdan sqlite3_stmt *pStmt = p->pPreStmt->pStmt;
16194a4c11aaSdan int i; /* Iterator variable */
16204a4c11aaSdan int nCol; /* Number of columns returned by pStmt */
16214a4c11aaSdan Tcl_Obj **apColName = 0; /* Array of column names */
16224a4c11aaSdan
16234a4c11aaSdan p->nCol = nCol = sqlite3_column_count(pStmt);
16244a4c11aaSdan if( nCol>0 && (papColName || p->pArray) ){
16254a4c11aaSdan apColName = (Tcl_Obj**)Tcl_Alloc( sizeof(Tcl_Obj*)*nCol );
16268e556520Sdanielk1977 for(i=0; i<nCol; i++){
1627c45e6716Sdrh apColName[i] = Tcl_NewStringObj(sqlite3_column_name(pStmt,i), -1);
16288e556520Sdanielk1977 Tcl_IncrRefCount(apColName[i]);
16298e556520Sdanielk1977 }
16304a4c11aaSdan p->apColName = apColName;
16314a4c11aaSdan }
16328e556520Sdanielk1977
16338e556520Sdanielk1977 /* If results are being stored in an array variable, then create
16348e556520Sdanielk1977 ** the array(*) entry for that array
16358e556520Sdanielk1977 */
16364a4c11aaSdan if( p->pArray ){
16374a4c11aaSdan Tcl_Interp *interp = p->pDb->interp;
16388e556520Sdanielk1977 Tcl_Obj *pColList = Tcl_NewObj();
16398e556520Sdanielk1977 Tcl_Obj *pStar = Tcl_NewStringObj("*", -1);
16404a4c11aaSdan
16418e556520Sdanielk1977 for(i=0; i<nCol; i++){
16428e556520Sdanielk1977 Tcl_ListObjAppendElement(interp, pColList, apColName[i]);
16438e556520Sdanielk1977 }
16448e556520Sdanielk1977 Tcl_IncrRefCount(pStar);
16454a4c11aaSdan Tcl_ObjSetVar2(interp, p->pArray, pStar, pColList, 0);
16468e556520Sdanielk1977 Tcl_DecrRefCount(pStar);
16478e556520Sdanielk1977 }
16488e556520Sdanielk1977 }
16498e556520Sdanielk1977
16504a4c11aaSdan if( papColName ){
16514a4c11aaSdan *papColName = p->apColName;
16524a4c11aaSdan }
16534a4c11aaSdan if( pnCol ){
16544a4c11aaSdan *pnCol = p->nCol;
16554a4c11aaSdan }
16564a4c11aaSdan }
16574a4c11aaSdan
16584a4c11aaSdan /*
16594a4c11aaSdan ** Return one of TCL_OK, TCL_BREAK or TCL_ERROR. If TCL_ERROR is
16604a4c11aaSdan ** returned, then an error message is stored in the interpreter before
16614a4c11aaSdan ** returning.
16624a4c11aaSdan **
16634a4c11aaSdan ** A return value of TCL_OK means there is a row of data available. The
16644a4c11aaSdan ** data may be accessed using dbEvalRowInfo() and dbEvalColumnValue(). This
16654a4c11aaSdan ** is analogous to a return of SQLITE_ROW from sqlite3_step(). If TCL_BREAK
16664a4c11aaSdan ** is returned, then the SQL script has finished executing and there are
16674a4c11aaSdan ** no further rows available. This is similar to SQLITE_DONE.
16684a4c11aaSdan */
dbEvalStep(DbEvalContext * p)16694a4c11aaSdan static int dbEvalStep(DbEvalContext *p){
1670c431fd55Sdan const char *zPrevSql = 0; /* Previous value of p->zSql */
1671c431fd55Sdan
16724a4c11aaSdan while( p->zSql[0] || p->pPreStmt ){
16734a4c11aaSdan int rc;
16744a4c11aaSdan if( p->pPreStmt==0 ){
1675c431fd55Sdan zPrevSql = (p->zSql==zPrevSql ? 0 : p->zSql);
16764a4c11aaSdan rc = dbPrepareAndBind(p->pDb, p->zSql, &p->zSql, &p->pPreStmt);
16774a4c11aaSdan if( rc!=TCL_OK ) return rc;
16784a4c11aaSdan }else{
16794a4c11aaSdan int rcs;
16804a4c11aaSdan SqliteDb *pDb = p->pDb;
16814a4c11aaSdan SqlPreparedStmt *pPreStmt = p->pPreStmt;
16824a4c11aaSdan sqlite3_stmt *pStmt = pPreStmt->pStmt;
16834a4c11aaSdan
16844a4c11aaSdan rcs = sqlite3_step(pStmt);
16854a4c11aaSdan if( rcs==SQLITE_ROW ){
16864a4c11aaSdan return TCL_OK;
16874a4c11aaSdan }
16884a4c11aaSdan if( p->pArray ){
16894a4c11aaSdan dbEvalRowInfo(p, 0, 0);
16904a4c11aaSdan }
16914a4c11aaSdan rcs = sqlite3_reset(pStmt);
16924a4c11aaSdan
16934a4c11aaSdan pDb->nStep = sqlite3_stmt_status(pStmt,SQLITE_STMTSTATUS_FULLSCAN_STEP,1);
16944a4c11aaSdan pDb->nSort = sqlite3_stmt_status(pStmt,SQLITE_STMTSTATUS_SORT,1);
16953c379b01Sdrh pDb->nIndex = sqlite3_stmt_status(pStmt,SQLITE_STMTSTATUS_AUTOINDEX,1);
1696c456a76fSdan pDb->nVMStep = sqlite3_stmt_status(pStmt,SQLITE_STMTSTATUS_VM_STEP,1);
16974a4c11aaSdan dbReleaseColumnNames(p);
16984a4c11aaSdan p->pPreStmt = 0;
16994a4c11aaSdan
17004a4c11aaSdan if( rcs!=SQLITE_OK ){
17014a4c11aaSdan /* If a run-time error occurs, report the error and stop reading
17024a4c11aaSdan ** the SQL. */
17034a4c11aaSdan dbReleaseStmt(pDb, pPreStmt, 1);
1704c431fd55Sdan #if SQLITE_TEST
1705c431fd55Sdan if( p->pDb->bLegacyPrepare && rcs==SQLITE_SCHEMA && zPrevSql ){
1706c431fd55Sdan /* If the runtime error was an SQLITE_SCHEMA, and the database
1707c431fd55Sdan ** handle is configured to use the legacy sqlite3_prepare()
1708c431fd55Sdan ** interface, retry prepare()/step() on the same SQL statement.
1709c431fd55Sdan ** This only happens once. If there is a second SQLITE_SCHEMA
1710c431fd55Sdan ** error, the error will be returned to the caller. */
1711c431fd55Sdan p->zSql = zPrevSql;
1712c431fd55Sdan continue;
1713c431fd55Sdan }
1714c431fd55Sdan #endif
1715c45e6716Sdrh Tcl_SetObjResult(pDb->interp,
1716c45e6716Sdrh Tcl_NewStringObj(sqlite3_errmsg(pDb->db), -1));
17174a4c11aaSdan return TCL_ERROR;
17184a4c11aaSdan }else{
17194a4c11aaSdan dbReleaseStmt(pDb, pPreStmt, 0);
17204a4c11aaSdan }
17214a4c11aaSdan }
17224a4c11aaSdan }
17234a4c11aaSdan
17244a4c11aaSdan /* Finished */
17254a4c11aaSdan return TCL_BREAK;
17264a4c11aaSdan }
17274a4c11aaSdan
17284a4c11aaSdan /*
17294a4c11aaSdan ** Free all resources currently held by the DbEvalContext structure passed
17304a4c11aaSdan ** as the first argument. There should be exactly one call to this function
17314a4c11aaSdan ** for each call to dbEvalInit().
17324a4c11aaSdan */
dbEvalFinalize(DbEvalContext * p)17334a4c11aaSdan static void dbEvalFinalize(DbEvalContext *p){
17344a4c11aaSdan if( p->pPreStmt ){
17354a4c11aaSdan sqlite3_reset(p->pPreStmt->pStmt);
17364a4c11aaSdan dbReleaseStmt(p->pDb, p->pPreStmt, 0);
17374a4c11aaSdan p->pPreStmt = 0;
17384a4c11aaSdan }
17394a4c11aaSdan if( p->pArray ){
17404a4c11aaSdan Tcl_DecrRefCount(p->pArray);
17414a4c11aaSdan p->pArray = 0;
17424a4c11aaSdan }
17434a4c11aaSdan Tcl_DecrRefCount(p->pSql);
17444a4c11aaSdan dbReleaseColumnNames(p);
1745bea28c73Sdan delDatabaseRef(p->pDb);
17464a4c11aaSdan }
17474a4c11aaSdan
17484a4c11aaSdan /*
17494a4c11aaSdan ** Return a pointer to a Tcl_Obj structure with ref-count 0 that contains
17504a4c11aaSdan ** the value for the iCol'th column of the row currently pointed to by
17514a4c11aaSdan ** the DbEvalContext structure passed as the first argument.
17524a4c11aaSdan */
dbEvalColumnValue(DbEvalContext * p,int iCol)17534a4c11aaSdan static Tcl_Obj *dbEvalColumnValue(DbEvalContext *p, int iCol){
17544a4c11aaSdan sqlite3_stmt *pStmt = p->pPreStmt->pStmt;
17554a4c11aaSdan switch( sqlite3_column_type(pStmt, iCol) ){
17564a4c11aaSdan case SQLITE_BLOB: {
17574a4c11aaSdan int bytes = sqlite3_column_bytes(pStmt, iCol);
17584a4c11aaSdan const char *zBlob = sqlite3_column_blob(pStmt, iCol);
17594a4c11aaSdan if( !zBlob ) bytes = 0;
17604a4c11aaSdan return Tcl_NewByteArrayObj((u8*)zBlob, bytes);
17614a4c11aaSdan }
17624a4c11aaSdan case SQLITE_INTEGER: {
17634a4c11aaSdan sqlite_int64 v = sqlite3_column_int64(pStmt, iCol);
17644a4c11aaSdan if( v>=-2147483647 && v<=2147483647 ){
17657fd33929Sdrh return Tcl_NewIntObj((int)v);
17664a4c11aaSdan }else{
17674a4c11aaSdan return Tcl_NewWideIntObj(v);
17684a4c11aaSdan }
17694a4c11aaSdan }
17704a4c11aaSdan case SQLITE_FLOAT: {
17714a4c11aaSdan return Tcl_NewDoubleObj(sqlite3_column_double(pStmt, iCol));
17724a4c11aaSdan }
17734a4c11aaSdan case SQLITE_NULL: {
1774c45e6716Sdrh return Tcl_NewStringObj(p->pDb->zNull, -1);
17754a4c11aaSdan }
17764a4c11aaSdan }
17774a4c11aaSdan
1778325eff58Sdrh return Tcl_NewStringObj((char*)sqlite3_column_text(pStmt, iCol), -1);
17794a4c11aaSdan }
17804a4c11aaSdan
17814a4c11aaSdan /*
17824a4c11aaSdan ** If using Tcl version 8.6 or greater, use the NR functions to avoid
17834a4c11aaSdan ** recursive evalution of scripts by the [db eval] and [db trans]
17844a4c11aaSdan ** commands. Even if the headers used while compiling the extension
17854a4c11aaSdan ** are 8.6 or newer, the code still tests the Tcl version at runtime.
17864a4c11aaSdan ** This allows stubs-enabled builds to be used with older Tcl libraries.
17874a4c11aaSdan */
17884a4c11aaSdan #if TCL_MAJOR_VERSION>8 || (TCL_MAJOR_VERSION==8 && TCL_MINOR_VERSION>=6)
1789a2c8a95bSdrh # define SQLITE_TCL_NRE 1
DbUseNre(void)17904a4c11aaSdan static int DbUseNre(void){
17914a4c11aaSdan int major, minor;
17924a4c11aaSdan Tcl_GetVersion(&major, &minor, 0, 0);
17934a4c11aaSdan return( (major==8 && minor>=6) || major>8 );
17944a4c11aaSdan }
17954a4c11aaSdan #else
17964a4c11aaSdan /*
17974a4c11aaSdan ** Compiling using headers earlier than 8.6. In this case NR cannot be
17984a4c11aaSdan ** used, so DbUseNre() to always return zero. Add #defines for the other
17994a4c11aaSdan ** Tcl_NRxxx() functions to prevent them from causing compilation errors,
18004a4c11aaSdan ** even though the only invocations of them are within conditional blocks
18014a4c11aaSdan ** of the form:
18024a4c11aaSdan **
18034a4c11aaSdan ** if( DbUseNre() ) { ... }
18044a4c11aaSdan */
1805a2c8a95bSdrh # define SQLITE_TCL_NRE 0
18064a4c11aaSdan # define DbUseNre() 0
1807a47941feSdrh # define Tcl_NRAddCallback(a,b,c,d,e,f) (void)0
18084a4c11aaSdan # define Tcl_NREvalObj(a,b,c) 0
1809a47941feSdrh # define Tcl_NRCreateCommand(a,b,c,d,e,f) (void)0
18104a4c11aaSdan #endif
18114a4c11aaSdan
18124a4c11aaSdan /*
18134a4c11aaSdan ** This function is part of the implementation of the command:
18144a4c11aaSdan **
18154a4c11aaSdan ** $db eval SQL ?ARRAYNAME? SCRIPT
18164a4c11aaSdan */
DbEvalNextCmd(ClientData data[],Tcl_Interp * interp,int result)1817a121cc7cSmistachkin static int SQLITE_TCLAPI DbEvalNextCmd(
18184a4c11aaSdan ClientData data[], /* data[0] is the (DbEvalContext*) */
18194a4c11aaSdan Tcl_Interp *interp, /* Tcl interpreter */
18204a4c11aaSdan int result /* Result so far */
18214a4c11aaSdan ){
18224a4c11aaSdan int rc = result; /* Return code */
18234a4c11aaSdan
18244a4c11aaSdan /* The first element of the data[] array is a pointer to a DbEvalContext
18254a4c11aaSdan ** structure allocated using Tcl_Alloc(). The second element of data[]
18264a4c11aaSdan ** is a pointer to a Tcl_Obj containing the script to run for each row
18274a4c11aaSdan ** returned by the queries encapsulated in data[0]. */
18284a4c11aaSdan DbEvalContext *p = (DbEvalContext *)data[0];
18294a4c11aaSdan Tcl_Obj *pScript = (Tcl_Obj *)data[1];
18304a4c11aaSdan Tcl_Obj *pArray = p->pArray;
18314a4c11aaSdan
18324a4c11aaSdan while( (rc==TCL_OK || rc==TCL_CONTINUE) && TCL_OK==(rc = dbEvalStep(p)) ){
18334a4c11aaSdan int i;
18344a4c11aaSdan int nCol;
18354a4c11aaSdan Tcl_Obj **apColName;
18364a4c11aaSdan dbEvalRowInfo(p, &nCol, &apColName);
18374a4c11aaSdan for(i=0; i<nCol; i++){
18384a4c11aaSdan if( pArray==0 ){
1839af38cdbcSdrh Tcl_ObjSetVar2(interp, apColName[i], 0, dbEvalColumnValue(p,i), 0);
1840af38cdbcSdrh }else if( (p->evalFlags & SQLITE_EVAL_WITHOUTNULLS)!=0
1841af38cdbcSdrh && sqlite3_column_type(p->pPreStmt->pStmt, i)==SQLITE_NULL
1842af38cdbcSdrh ){
1843af38cdbcSdrh Tcl_UnsetVar2(interp, Tcl_GetString(pArray),
1844af38cdbcSdrh Tcl_GetString(apColName[i]), 0);
18454a4c11aaSdan }else{
1846af38cdbcSdrh Tcl_ObjSetVar2(interp, pArray, apColName[i], dbEvalColumnValue(p,i), 0);
18474a4c11aaSdan }
18484a4c11aaSdan }
18494a4c11aaSdan
18504a4c11aaSdan /* The required interpreter variables are now populated with the data
18514a4c11aaSdan ** from the current row. If using NRE, schedule callbacks to evaluate
18524a4c11aaSdan ** script pScript, then to invoke this function again to fetch the next
18534a4c11aaSdan ** row (or clean up if there is no next row or the script throws an
18544a4c11aaSdan ** exception). After scheduling the callbacks, return control to the
18554a4c11aaSdan ** caller.
18564a4c11aaSdan **
18574a4c11aaSdan ** If not using NRE, evaluate pScript directly and continue with the
18584a4c11aaSdan ** next iteration of this while(...) loop. */
18594a4c11aaSdan if( DbUseNre() ){
18604a4c11aaSdan Tcl_NRAddCallback(interp, DbEvalNextCmd, (void*)p, (void*)pScript, 0, 0);
18614a4c11aaSdan return Tcl_NREvalObj(interp, pScript, 0);
18624a4c11aaSdan }else{
18634a4c11aaSdan rc = Tcl_EvalObjEx(interp, pScript, 0);
18644a4c11aaSdan }
18654a4c11aaSdan }
18664a4c11aaSdan
18674a4c11aaSdan Tcl_DecrRefCount(pScript);
18684a4c11aaSdan dbEvalFinalize(p);
18694a4c11aaSdan Tcl_Free((char *)p);
18704a4c11aaSdan
18714a4c11aaSdan if( rc==TCL_OK || rc==TCL_BREAK ){
18724a4c11aaSdan Tcl_ResetResult(interp);
18734a4c11aaSdan rc = TCL_OK;
18744a4c11aaSdan }
18754a4c11aaSdan return rc;
18768e556520Sdanielk1977 }
18778e556520Sdanielk1977
18781067fe11Stpoindex /*
187946c47d46Sdan ** This function is used by the implementations of the following database
188046c47d46Sdan ** handle sub-commands:
188146c47d46Sdan **
188246c47d46Sdan ** $db update_hook ?SCRIPT?
188346c47d46Sdan ** $db wal_hook ?SCRIPT?
188446c47d46Sdan ** $db commit_hook ?SCRIPT?
188546c47d46Sdan ** $db preupdate hook ?SCRIPT?
188646c47d46Sdan */
DbHookCmd(Tcl_Interp * interp,SqliteDb * pDb,Tcl_Obj * pArg,Tcl_Obj ** ppHook)188746c47d46Sdan static void DbHookCmd(
188846c47d46Sdan Tcl_Interp *interp, /* Tcl interpreter */
188946c47d46Sdan SqliteDb *pDb, /* Database handle */
189046c47d46Sdan Tcl_Obj *pArg, /* SCRIPT argument (or NULL) */
189146c47d46Sdan Tcl_Obj **ppHook /* Pointer to member of SqliteDb */
189246c47d46Sdan ){
189346c47d46Sdan sqlite3 *db = pDb->db;
189446c47d46Sdan
189546c47d46Sdan if( *ppHook ){
189646c47d46Sdan Tcl_SetObjResult(interp, *ppHook);
189746c47d46Sdan if( pArg ){
189846c47d46Sdan Tcl_DecrRefCount(*ppHook);
189946c47d46Sdan *ppHook = 0;
190046c47d46Sdan }
190146c47d46Sdan }
190246c47d46Sdan if( pArg ){
190346c47d46Sdan assert( !(*ppHook) );
190446c47d46Sdan if( Tcl_GetCharLength(pArg)>0 ){
190546c47d46Sdan *ppHook = pArg;
190646c47d46Sdan Tcl_IncrRefCount(*ppHook);
190746c47d46Sdan }
190846c47d46Sdan }
190946c47d46Sdan
19109b1c62d4Sdrh #ifdef SQLITE_ENABLE_PREUPDATE_HOOK
191146c47d46Sdan sqlite3_preupdate_hook(db, (pDb->pPreUpdateHook?DbPreUpdateHandler:0), pDb);
19129b1c62d4Sdrh #endif
191346c47d46Sdan sqlite3_update_hook(db, (pDb->pUpdateHook?DbUpdateHandler:0), pDb);
191446c47d46Sdan sqlite3_rollback_hook(db, (pDb->pRollbackHook?DbRollbackHandler:0), pDb);
191546c47d46Sdan sqlite3_wal_hook(db, (pDb->pWalHook?DbWalHandler:0), pDb);
191646c47d46Sdan }
191746c47d46Sdan
191846c47d46Sdan /*
191975897234Sdrh ** The "sqlite" command below creates a new Tcl command for each
192075897234Sdrh ** connection it opens to an SQLite database. This routine is invoked
192175897234Sdrh ** whenever one of those connection-specific commands is executed
192275897234Sdrh ** in Tcl. For example, if you run Tcl code like this:
192375897234Sdrh **
19249bb575fdSdrh ** sqlite3 db1 "my_database"
192575897234Sdrh ** db1 close
192675897234Sdrh **
192775897234Sdrh ** The first command opens a connection to the "my_database" database
192875897234Sdrh ** and calls that connection "db1". The second command causes this
192975897234Sdrh ** subroutine to be invoked.
193075897234Sdrh */
DbObjCmd(void * cd,Tcl_Interp * interp,int objc,Tcl_Obj * const * objv)19317617e4a8Smistachkin static int SQLITE_TCLAPI DbObjCmd(
19327617e4a8Smistachkin void *cd,
19337617e4a8Smistachkin Tcl_Interp *interp,
19347617e4a8Smistachkin int objc,
19357617e4a8Smistachkin Tcl_Obj *const*objv
19367617e4a8Smistachkin ){
1937bec3f402Sdrh SqliteDb *pDb = (SqliteDb*)cd;
19386d31316cSdrh int choice;
193922fbcb8dSdrh int rc = TCL_OK;
19400de8c112Sdrh static const char *DB_strs[] = {
1941c06ede10Sdrh "authorizer", "backup", "bind_fallback",
1942c06ede10Sdrh "busy", "cache", "changes",
1943c06ede10Sdrh "close", "collate", "collation_needed",
194411d88e68Sdrh "commit_hook", "complete", "config",
194511d88e68Sdrh "copy", "deserialize", "enable_load_extension",
1946be4e3c88Sdrh "errorcode", "erroroffset", "eval",
1947be4e3c88Sdrh "exists", "function", "incrblob",
1948be4e3c88Sdrh "interrupt", "last_insert_rowid", "nullvalue",
1949be4e3c88Sdrh "onecolumn", "preupdate", "profile",
1950be4e3c88Sdrh "progress", "rekey", "restore",
1951be4e3c88Sdrh "rollback_hook", "serialize", "status",
1952be4e3c88Sdrh "timeout", "total_changes", "trace",
1953be4e3c88Sdrh "trace_v2", "transaction", "unlock_notify",
1954be4e3c88Sdrh "update_hook", "version", "wal_hook",
1955be4e3c88Sdrh 0
19566d31316cSdrh };
1957411995dcSdrh enum DB_enum {
1958c06ede10Sdrh DB_AUTHORIZER, DB_BACKUP, DB_BIND_FALLBACK,
1959c06ede10Sdrh DB_BUSY, DB_CACHE, DB_CHANGES,
1960c06ede10Sdrh DB_CLOSE, DB_COLLATE, DB_COLLATION_NEEDED,
196111d88e68Sdrh DB_COMMIT_HOOK, DB_COMPLETE, DB_CONFIG,
196211d88e68Sdrh DB_COPY, DB_DESERIALIZE, DB_ENABLE_LOAD_EXTENSION,
1963be4e3c88Sdrh DB_ERRORCODE, DB_ERROROFFSET, DB_EVAL,
1964be4e3c88Sdrh DB_EXISTS, DB_FUNCTION, DB_INCRBLOB,
1965be4e3c88Sdrh DB_INTERRUPT, DB_LAST_INSERT_ROWID, DB_NULLVALUE,
1966be4e3c88Sdrh DB_ONECOLUMN, DB_PREUPDATE, DB_PROFILE,
1967be4e3c88Sdrh DB_PROGRESS, DB_REKEY, DB_RESTORE,
1968be4e3c88Sdrh DB_ROLLBACK_HOOK, DB_SERIALIZE, DB_STATUS,
1969be4e3c88Sdrh DB_TIMEOUT, DB_TOTAL_CHANGES, DB_TRACE,
1970be4e3c88Sdrh DB_TRACE_V2, DB_TRANSACTION, DB_UNLOCK_NOTIFY,
1971be4e3c88Sdrh DB_UPDATE_HOOK, DB_VERSION, DB_WAL_HOOK,
19726d31316cSdrh };
19731067fe11Stpoindex /* don't leave trailing commas on DB_enum, it confuses the AIX xlc compiler */
19746d31316cSdrh
19756d31316cSdrh if( objc<2 ){
19766d31316cSdrh Tcl_WrongNumArgs(interp, 1, objv, "SUBCOMMAND ...");
197775897234Sdrh return TCL_ERROR;
197875897234Sdrh }
1979411995dcSdrh if( Tcl_GetIndexFromObj(interp, objv[1], DB_strs, "option", 0, &choice) ){
19806d31316cSdrh return TCL_ERROR;
19816d31316cSdrh }
19826d31316cSdrh
1983411995dcSdrh switch( (enum DB_enum)choice ){
198475897234Sdrh
1985e22a334bSdrh /* $db authorizer ?CALLBACK?
1986e22a334bSdrh **
1987e22a334bSdrh ** Invoke the given callback to authorize each SQL operation as it is
1988e22a334bSdrh ** compiled. 5 arguments are appended to the callback before it is
1989e22a334bSdrh ** invoked:
1990e22a334bSdrh **
1991e22a334bSdrh ** (1) The authorization type (ex: SQLITE_CREATE_TABLE, SQLITE_INSERT, ...)
1992e22a334bSdrh ** (2) First descriptive name (depends on authorization type)
1993e22a334bSdrh ** (3) Second descriptive name
1994e22a334bSdrh ** (4) Name of the database (ex: "main", "temp")
1995e22a334bSdrh ** (5) Name of trigger that is doing the access
1996e22a334bSdrh **
1997e22a334bSdrh ** The callback should return on of the following strings: SQLITE_OK,
1998e22a334bSdrh ** SQLITE_IGNORE, or SQLITE_DENY. Any other return value is an error.
1999e22a334bSdrh **
2000e22a334bSdrh ** If this method is invoked with no arguments, the current authorization
2001e22a334bSdrh ** callback string is returned.
2002e22a334bSdrh */
2003e22a334bSdrh case DB_AUTHORIZER: {
20041211de37Sdrh #ifdef SQLITE_OMIT_AUTHORIZATION
2005a198f2b5Sdrh Tcl_AppendResult(interp, "authorization not available in this build",
2006a198f2b5Sdrh (char*)0);
20071211de37Sdrh return TCL_ERROR;
20081211de37Sdrh #else
2009e22a334bSdrh if( objc>3 ){
2010e22a334bSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?");
20110f14e2ebSdrh return TCL_ERROR;
2012e22a334bSdrh }else if( objc==2 ){
2013b5a20d3cSdrh if( pDb->zAuth ){
2014a198f2b5Sdrh Tcl_AppendResult(interp, pDb->zAuth, (char*)0);
2015e22a334bSdrh }
2016e22a334bSdrh }else{
2017e22a334bSdrh char *zAuth;
2018e22a334bSdrh int len;
2019e22a334bSdrh if( pDb->zAuth ){
2020e22a334bSdrh Tcl_Free(pDb->zAuth);
2021e22a334bSdrh }
2022e22a334bSdrh zAuth = Tcl_GetStringFromObj(objv[2], &len);
2023e22a334bSdrh if( zAuth && len>0 ){
2024e22a334bSdrh pDb->zAuth = Tcl_Alloc( len + 1 );
20255bb3eb9bSdrh memcpy(pDb->zAuth, zAuth, len+1);
2026e22a334bSdrh }else{
2027e22a334bSdrh pDb->zAuth = 0;
2028e22a334bSdrh }
2029e22a334bSdrh if( pDb->zAuth ){
203032c6a48bSdrh typedef int (*sqlite3_auth_cb)(
203132c6a48bSdrh void*,int,const char*,const char*,
203232c6a48bSdrh const char*,const char*);
2033e22a334bSdrh pDb->interp = interp;
203432c6a48bSdrh sqlite3_set_authorizer(pDb->db,(sqlite3_auth_cb)auth_callback,pDb);
2035e22a334bSdrh }else{
20366f8a503dSdanielk1977 sqlite3_set_authorizer(pDb->db, 0, 0);
2037e22a334bSdrh }
2038e22a334bSdrh }
20391211de37Sdrh #endif
2040e22a334bSdrh break;
2041e22a334bSdrh }
2042e22a334bSdrh
2043dc2c4915Sdrh /* $db backup ?DATABASE? FILENAME
2044dc2c4915Sdrh **
2045dc2c4915Sdrh ** Open or create a database file named FILENAME. Transfer the
2046dc2c4915Sdrh ** content of local database DATABASE (default: "main") into the
2047dc2c4915Sdrh ** FILENAME database.
2048dc2c4915Sdrh */
2049dc2c4915Sdrh case DB_BACKUP: {
2050dc2c4915Sdrh const char *zDestFile;
2051dc2c4915Sdrh const char *zSrcDb;
2052dc2c4915Sdrh sqlite3 *pDest;
2053dc2c4915Sdrh sqlite3_backup *pBackup;
2054dc2c4915Sdrh
2055dc2c4915Sdrh if( objc==3 ){
2056dc2c4915Sdrh zSrcDb = "main";
2057dc2c4915Sdrh zDestFile = Tcl_GetString(objv[2]);
2058dc2c4915Sdrh }else if( objc==4 ){
2059dc2c4915Sdrh zSrcDb = Tcl_GetString(objv[2]);
2060dc2c4915Sdrh zDestFile = Tcl_GetString(objv[3]);
2061dc2c4915Sdrh }else{
2062dc2c4915Sdrh Tcl_WrongNumArgs(interp, 2, objv, "?DATABASE? FILENAME");
2063dc2c4915Sdrh return TCL_ERROR;
2064dc2c4915Sdrh }
2065147ef394Sdrh rc = sqlite3_open_v2(zDestFile, &pDest,
2066147ef394Sdrh SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE| pDb->openFlags, 0);
2067dc2c4915Sdrh if( rc!=SQLITE_OK ){
2068dc2c4915Sdrh Tcl_AppendResult(interp, "cannot open target database: ",
2069dc2c4915Sdrh sqlite3_errmsg(pDest), (char*)0);
2070dc2c4915Sdrh sqlite3_close(pDest);
2071dc2c4915Sdrh return TCL_ERROR;
2072dc2c4915Sdrh }
2073dc2c4915Sdrh pBackup = sqlite3_backup_init(pDest, "main", pDb->db, zSrcDb);
2074dc2c4915Sdrh if( pBackup==0 ){
2075dc2c4915Sdrh Tcl_AppendResult(interp, "backup failed: ",
2076dc2c4915Sdrh sqlite3_errmsg(pDest), (char*)0);
2077dc2c4915Sdrh sqlite3_close(pDest);
2078dc2c4915Sdrh return TCL_ERROR;
2079dc2c4915Sdrh }
2080dc2c4915Sdrh while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK ){}
2081dc2c4915Sdrh sqlite3_backup_finish(pBackup);
2082dc2c4915Sdrh if( rc==SQLITE_DONE ){
2083dc2c4915Sdrh rc = TCL_OK;
2084dc2c4915Sdrh }else{
2085dc2c4915Sdrh Tcl_AppendResult(interp, "backup failed: ",
2086dc2c4915Sdrh sqlite3_errmsg(pDest), (char*)0);
2087dc2c4915Sdrh rc = TCL_ERROR;
2088dc2c4915Sdrh }
2089dc2c4915Sdrh sqlite3_close(pDest);
2090dc2c4915Sdrh break;
2091dc2c4915Sdrh }
2092dc2c4915Sdrh
2093c06ede10Sdrh /* $db bind_fallback ?CALLBACK?
2094c06ede10Sdrh **
2095c06ede10Sdrh ** When resolving bind parameters in an SQL statement, if the parameter
2096c06ede10Sdrh ** cannot be associated with a TCL variable then invoke CALLBACK with a
2097c06ede10Sdrh ** single argument that is the name of the parameter and use the return
2098c06ede10Sdrh ** value of the CALLBACK as the binding. If CALLBACK returns something
2099c06ede10Sdrh ** other than TCL_OK or TCL_ERROR then bind a NULL.
2100c06ede10Sdrh **
2101c06ede10Sdrh ** If CALLBACK is an empty string, then revert to the default behavior
2102c06ede10Sdrh ** which is to set the binding to NULL.
2103c06ede10Sdrh **
2104c06ede10Sdrh ** If CALLBACK returns an error, that causes the statement execution to
2105c06ede10Sdrh ** abort. Hence, to configure a connection so that it throws an error
2106c06ede10Sdrh ** on an attempt to bind an unknown variable, do something like this:
2107c06ede10Sdrh **
2108c06ede10Sdrh ** proc bind_error {name} {error "no such variable: $name"}
2109c06ede10Sdrh ** db bind_fallback bind_error
2110c06ede10Sdrh */
2111c06ede10Sdrh case DB_BIND_FALLBACK: {
2112c06ede10Sdrh if( objc>3 ){
2113c06ede10Sdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?");
2114c06ede10Sdrh return TCL_ERROR;
2115c06ede10Sdrh }else if( objc==2 ){
2116c06ede10Sdrh if( pDb->zBindFallback ){
2117c06ede10Sdrh Tcl_AppendResult(interp, pDb->zBindFallback, (char*)0);
2118c06ede10Sdrh }
2119c06ede10Sdrh }else{
2120c06ede10Sdrh char *zCallback;
2121c06ede10Sdrh int len;
2122c06ede10Sdrh if( pDb->zBindFallback ){
2123c06ede10Sdrh Tcl_Free(pDb->zBindFallback);
2124c06ede10Sdrh }
2125c06ede10Sdrh zCallback = Tcl_GetStringFromObj(objv[2], &len);
2126c06ede10Sdrh if( zCallback && len>0 ){
2127c06ede10Sdrh pDb->zBindFallback = Tcl_Alloc( len + 1 );
2128c06ede10Sdrh memcpy(pDb->zBindFallback, zCallback, len+1);
2129c06ede10Sdrh }else{
2130c06ede10Sdrh pDb->zBindFallback = 0;
2131c06ede10Sdrh }
2132c06ede10Sdrh }
2133c06ede10Sdrh break;
2134c06ede10Sdrh }
2135c06ede10Sdrh
2136bec3f402Sdrh /* $db busy ?CALLBACK?
2137bec3f402Sdrh **
2138bec3f402Sdrh ** Invoke the given callback if an SQL statement attempts to open
2139bec3f402Sdrh ** a locked database file.
2140bec3f402Sdrh */
21416d31316cSdrh case DB_BUSY: {
21426d31316cSdrh if( objc>3 ){
21436d31316cSdrh Tcl_WrongNumArgs(interp, 2, objv, "CALLBACK");
2144bec3f402Sdrh return TCL_ERROR;
21456d31316cSdrh }else if( objc==2 ){
2146bec3f402Sdrh if( pDb->zBusy ){
2147a198f2b5Sdrh Tcl_AppendResult(interp, pDb->zBusy, (char*)0);
2148bec3f402Sdrh }
2149bec3f402Sdrh }else{
21506d31316cSdrh char *zBusy;
21516d31316cSdrh int len;
2152bec3f402Sdrh if( pDb->zBusy ){
2153bec3f402Sdrh Tcl_Free(pDb->zBusy);
21546d31316cSdrh }
21556d31316cSdrh zBusy = Tcl_GetStringFromObj(objv[2], &len);
21566d31316cSdrh if( zBusy && len>0 ){
21576d31316cSdrh pDb->zBusy = Tcl_Alloc( len + 1 );
21585bb3eb9bSdrh memcpy(pDb->zBusy, zBusy, len+1);
21596d31316cSdrh }else{
2160bec3f402Sdrh pDb->zBusy = 0;
2161bec3f402Sdrh }
2162bec3f402Sdrh if( pDb->zBusy ){
2163bec3f402Sdrh pDb->interp = interp;
21646f8a503dSdanielk1977 sqlite3_busy_handler(pDb->db, DbBusyHandler, pDb);
21656d31316cSdrh }else{
21666f8a503dSdanielk1977 sqlite3_busy_handler(pDb->db, 0, 0);
2167bec3f402Sdrh }
2168bec3f402Sdrh }
21696d31316cSdrh break;
21706d31316cSdrh }
2171bec3f402Sdrh
2172fb7e7651Sdrh /* $db cache flush
2173fb7e7651Sdrh ** $db cache size n
2174fb7e7651Sdrh **
2175fb7e7651Sdrh ** Flush the prepared statement cache, or set the maximum number of
2176fb7e7651Sdrh ** cached statements.
2177fb7e7651Sdrh */
2178fb7e7651Sdrh case DB_CACHE: {
2179fb7e7651Sdrh char *subCmd;
2180fb7e7651Sdrh int n;
2181fb7e7651Sdrh
2182fb7e7651Sdrh if( objc<=2 ){
2183fb7e7651Sdrh Tcl_WrongNumArgs(interp, 1, objv, "cache option ?arg?");
2184fb7e7651Sdrh return TCL_ERROR;
2185fb7e7651Sdrh }
2186fb7e7651Sdrh subCmd = Tcl_GetStringFromObj( objv[2], 0 );
2187fb7e7651Sdrh if( *subCmd=='f' && strcmp(subCmd,"flush")==0 ){
2188fb7e7651Sdrh if( objc!=3 ){
2189fb7e7651Sdrh Tcl_WrongNumArgs(interp, 2, objv, "flush");
2190fb7e7651Sdrh return TCL_ERROR;
2191fb7e7651Sdrh }else{
2192fb7e7651Sdrh flushStmtCache( pDb );
2193fb7e7651Sdrh }
2194fb7e7651Sdrh }else if( *subCmd=='s' && strcmp(subCmd,"size")==0 ){
2195fb7e7651Sdrh if( objc!=4 ){
2196fb7e7651Sdrh Tcl_WrongNumArgs(interp, 2, objv, "size n");
2197fb7e7651Sdrh return TCL_ERROR;
2198fb7e7651Sdrh }else{
2199fb7e7651Sdrh if( TCL_ERROR==Tcl_GetIntFromObj(interp, objv[3], &n) ){
2200fb7e7651Sdrh Tcl_AppendResult( interp, "cannot convert \"",
2201a198f2b5Sdrh Tcl_GetStringFromObj(objv[3],0), "\" to integer", (char*)0);
2202fb7e7651Sdrh return TCL_ERROR;
2203fb7e7651Sdrh }else{
2204fb7e7651Sdrh if( n<0 ){
2205fb7e7651Sdrh flushStmtCache( pDb );
2206fb7e7651Sdrh n = 0;
2207fb7e7651Sdrh }else if( n>MAX_PREPARED_STMTS ){
2208fb7e7651Sdrh n = MAX_PREPARED_STMTS;
2209fb7e7651Sdrh }
2210fb7e7651Sdrh pDb->maxStmt = n;
2211fb7e7651Sdrh }
2212fb7e7651Sdrh }
2213fb7e7651Sdrh }else{
2214fb7e7651Sdrh Tcl_AppendResult( interp, "bad option \"",
2215a198f2b5Sdrh Tcl_GetStringFromObj(objv[2],0), "\": must be flush or size",
2216a198f2b5Sdrh (char*)0);
2217fb7e7651Sdrh return TCL_ERROR;
2218fb7e7651Sdrh }
2219fb7e7651Sdrh break;
2220fb7e7651Sdrh }
2221fb7e7651Sdrh
2222b28af71aSdanielk1977 /* $db changes
2223c8d30ac1Sdrh **
2224c8d30ac1Sdrh ** Return the number of rows that were modified, inserted, or deleted by
2225b28af71aSdanielk1977 ** the most recent INSERT, UPDATE or DELETE statement, not including
2226b28af71aSdanielk1977 ** any changes made by trigger programs.
2227c8d30ac1Sdrh */
2228c8d30ac1Sdrh case DB_CHANGES: {
2229c8d30ac1Sdrh Tcl_Obj *pResult;
2230c8d30ac1Sdrh if( objc!=2 ){
2231c8d30ac1Sdrh Tcl_WrongNumArgs(interp, 2, objv, "");
2232c8d30ac1Sdrh return TCL_ERROR;
2233c8d30ac1Sdrh }
2234c8d30ac1Sdrh pResult = Tcl_GetObjResult(interp);
22352c718873Sdan Tcl_SetWideIntObj(pResult, sqlite3_changes64(pDb->db));
2236f146a776Srdc break;
2237f146a776Srdc }
2238f146a776Srdc
223975897234Sdrh /* $db close
224075897234Sdrh **
224175897234Sdrh ** Shutdown the database
224275897234Sdrh */
22436d31316cSdrh case DB_CLOSE: {
22446d31316cSdrh Tcl_DeleteCommand(interp, Tcl_GetStringFromObj(objv[0], 0));
22456d31316cSdrh break;
22466d31316cSdrh }
224775897234Sdrh
22480f14e2ebSdrh /*
22490f14e2ebSdrh ** $db collate NAME SCRIPT
22500f14e2ebSdrh **
22510f14e2ebSdrh ** Create a new SQL collation function called NAME. Whenever
22520f14e2ebSdrh ** that function is called, invoke SCRIPT to evaluate the function.
22530f14e2ebSdrh */
22540f14e2ebSdrh case DB_COLLATE: {
22550f14e2ebSdrh SqlCollate *pCollate;
22560f14e2ebSdrh char *zName;
22570f14e2ebSdrh char *zScript;
22580f14e2ebSdrh int nScript;
22590f14e2ebSdrh if( objc!=4 ){
22600f14e2ebSdrh Tcl_WrongNumArgs(interp, 2, objv, "NAME SCRIPT");
22610f14e2ebSdrh return TCL_ERROR;
22620f14e2ebSdrh }
22630f14e2ebSdrh zName = Tcl_GetStringFromObj(objv[2], 0);
22640f14e2ebSdrh zScript = Tcl_GetStringFromObj(objv[3], &nScript);
22650f14e2ebSdrh pCollate = (SqlCollate*)Tcl_Alloc( sizeof(*pCollate) + nScript + 1 );
22660f14e2ebSdrh if( pCollate==0 ) return TCL_ERROR;
22670f14e2ebSdrh pCollate->interp = interp;
22680f14e2ebSdrh pCollate->pNext = pDb->pCollate;
22690f14e2ebSdrh pCollate->zScript = (char*)&pCollate[1];
22700f14e2ebSdrh pDb->pCollate = pCollate;
22715bb3eb9bSdrh memcpy(pCollate->zScript, zScript, nScript+1);
22720f14e2ebSdrh if( sqlite3_create_collation(pDb->db, zName, SQLITE_UTF8,
22730f14e2ebSdrh pCollate, tclSqlCollate) ){
22749636c4e1Sdanielk1977 Tcl_SetResult(interp, (char *)sqlite3_errmsg(pDb->db), TCL_VOLATILE);
22750f14e2ebSdrh return TCL_ERROR;
22760f14e2ebSdrh }
22770f14e2ebSdrh break;
22780f14e2ebSdrh }
22790f14e2ebSdrh
22800f14e2ebSdrh /*
22810f14e2ebSdrh ** $db collation_needed SCRIPT
22820f14e2ebSdrh **
22830f14e2ebSdrh ** Create a new SQL collation function called NAME. Whenever
22840f14e2ebSdrh ** that function is called, invoke SCRIPT to evaluate the function.
22850f14e2ebSdrh */
22860f14e2ebSdrh case DB_COLLATION_NEEDED: {
22870f14e2ebSdrh if( objc!=3 ){
22880f14e2ebSdrh Tcl_WrongNumArgs(interp, 2, objv, "SCRIPT");
22890f14e2ebSdrh return TCL_ERROR;
22900f14e2ebSdrh }
22910f14e2ebSdrh if( pDb->pCollateNeeded ){
22920f14e2ebSdrh Tcl_DecrRefCount(pDb->pCollateNeeded);
22930f14e2ebSdrh }
22940f14e2ebSdrh pDb->pCollateNeeded = Tcl_DuplicateObj(objv[2]);
22950f14e2ebSdrh Tcl_IncrRefCount(pDb->pCollateNeeded);
22960f14e2ebSdrh sqlite3_collation_needed(pDb->db, pDb, tclCollateNeeded);
22970f14e2ebSdrh break;
22980f14e2ebSdrh }
22990f14e2ebSdrh
230019e2d37fSdrh /* $db commit_hook ?CALLBACK?
230119e2d37fSdrh **
230219e2d37fSdrh ** Invoke the given callback just before committing every SQL transaction.
230319e2d37fSdrh ** If the callback throws an exception or returns non-zero, then the
230419e2d37fSdrh ** transaction is aborted. If CALLBACK is an empty string, the callback
230519e2d37fSdrh ** is disabled.
230619e2d37fSdrh */
230719e2d37fSdrh case DB_COMMIT_HOOK: {
230819e2d37fSdrh if( objc>3 ){
230919e2d37fSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?");
231019e2d37fSdrh return TCL_ERROR;
231119e2d37fSdrh }else if( objc==2 ){
231219e2d37fSdrh if( pDb->zCommit ){
2313a198f2b5Sdrh Tcl_AppendResult(interp, pDb->zCommit, (char*)0);
231419e2d37fSdrh }
231519e2d37fSdrh }else{
23166ef5e12eSmistachkin const char *zCommit;
231719e2d37fSdrh int len;
231819e2d37fSdrh if( pDb->zCommit ){
231919e2d37fSdrh Tcl_Free(pDb->zCommit);
232019e2d37fSdrh }
232119e2d37fSdrh zCommit = Tcl_GetStringFromObj(objv[2], &len);
232219e2d37fSdrh if( zCommit && len>0 ){
232319e2d37fSdrh pDb->zCommit = Tcl_Alloc( len + 1 );
23245bb3eb9bSdrh memcpy(pDb->zCommit, zCommit, len+1);
232519e2d37fSdrh }else{
232619e2d37fSdrh pDb->zCommit = 0;
232719e2d37fSdrh }
232819e2d37fSdrh if( pDb->zCommit ){
232919e2d37fSdrh pDb->interp = interp;
233019e2d37fSdrh sqlite3_commit_hook(pDb->db, DbCommitHandler, pDb);
233119e2d37fSdrh }else{
233219e2d37fSdrh sqlite3_commit_hook(pDb->db, 0, 0);
233319e2d37fSdrh }
233419e2d37fSdrh }
233519e2d37fSdrh break;
233619e2d37fSdrh }
233719e2d37fSdrh
233875897234Sdrh /* $db complete SQL
233975897234Sdrh **
234075897234Sdrh ** Return TRUE if SQL is a complete SQL statement. Return FALSE if
234175897234Sdrh ** additional lines of input are needed. This is similar to the
234275897234Sdrh ** built-in "info complete" command of Tcl.
234375897234Sdrh */
23446d31316cSdrh case DB_COMPLETE: {
2345ccae6026Sdrh #ifndef SQLITE_OMIT_COMPLETE
23466d31316cSdrh Tcl_Obj *pResult;
23476d31316cSdrh int isComplete;
23486d31316cSdrh if( objc!=3 ){
23496d31316cSdrh Tcl_WrongNumArgs(interp, 2, objv, "SQL");
235075897234Sdrh return TCL_ERROR;
235175897234Sdrh }
23526f8a503dSdanielk1977 isComplete = sqlite3_complete( Tcl_GetStringFromObj(objv[2], 0) );
23536d31316cSdrh pResult = Tcl_GetObjResult(interp);
23546d31316cSdrh Tcl_SetBooleanObj(pResult, isComplete);
2355ccae6026Sdrh #endif
23566d31316cSdrh break;
23576d31316cSdrh }
235875897234Sdrh
235911d88e68Sdrh /* $db config ?OPTION? ?BOOLEAN?
236011d88e68Sdrh **
236111d88e68Sdrh ** Configure the database connection using the sqlite3_db_config()
236211d88e68Sdrh ** interface.
236311d88e68Sdrh */
236411d88e68Sdrh case DB_CONFIG: {
236511d88e68Sdrh static const struct DbConfigChoices {
236611d88e68Sdrh const char *zName;
236711d88e68Sdrh int op;
236811d88e68Sdrh } aDbConfig[] = {
236976a1fce5Sdrh { "defensive", SQLITE_DBCONFIG_DEFENSIVE },
237076a1fce5Sdrh { "dqs_ddl", SQLITE_DBCONFIG_DQS_DDL },
237176a1fce5Sdrh { "dqs_dml", SQLITE_DBCONFIG_DQS_DML },
237211d88e68Sdrh { "enable_fkey", SQLITE_DBCONFIG_ENABLE_FKEY },
237376a1fce5Sdrh { "enable_qpsg", SQLITE_DBCONFIG_ENABLE_QPSG },
237411d88e68Sdrh { "enable_trigger", SQLITE_DBCONFIG_ENABLE_TRIGGER },
237511d88e68Sdrh { "enable_view", SQLITE_DBCONFIG_ENABLE_VIEW },
237611d88e68Sdrh { "fts3_tokenizer", SQLITE_DBCONFIG_ENABLE_FTS3_TOKENIZER },
237776a1fce5Sdrh { "legacy_alter_table", SQLITE_DBCONFIG_LEGACY_ALTER_TABLE },
237876a1fce5Sdrh { "legacy_file_format", SQLITE_DBCONFIG_LEGACY_FILE_FORMAT },
237911d88e68Sdrh { "load_extension", SQLITE_DBCONFIG_ENABLE_LOAD_EXTENSION },
238011d88e68Sdrh { "no_ckpt_on_close", SQLITE_DBCONFIG_NO_CKPT_ON_CLOSE },
238111d88e68Sdrh { "reset_database", SQLITE_DBCONFIG_RESET_DATABASE },
238276a1fce5Sdrh { "trigger_eqp", SQLITE_DBCONFIG_TRIGGER_EQP },
238376a1fce5Sdrh { "trusted_schema", SQLITE_DBCONFIG_TRUSTED_SCHEMA },
238411d88e68Sdrh { "writable_schema", SQLITE_DBCONFIG_WRITABLE_SCHEMA },
238511d88e68Sdrh };
238611d88e68Sdrh Tcl_Obj *pResult;
238711d88e68Sdrh int ii;
238811d88e68Sdrh if( objc>4 ){
238911d88e68Sdrh Tcl_WrongNumArgs(interp, 2, objv, "?OPTION? ?BOOLEAN?");
239011d88e68Sdrh return TCL_ERROR;
239111d88e68Sdrh }
239211d88e68Sdrh if( objc==2 ){
239311d88e68Sdrh /* With no arguments, list all configuration options and with the
239411d88e68Sdrh ** current value */
239511d88e68Sdrh pResult = Tcl_NewListObj(0,0);
239611d88e68Sdrh for(ii=0; ii<sizeof(aDbConfig)/sizeof(aDbConfig[0]); ii++){
239711d88e68Sdrh int v = 0;
23984043cfefSdrh sqlite3_db_config(pDb->db, aDbConfig[ii].op, -1, &v);
239911d88e68Sdrh Tcl_ListObjAppendElement(interp, pResult,
240011d88e68Sdrh Tcl_NewStringObj(aDbConfig[ii].zName,-1));
240111d88e68Sdrh Tcl_ListObjAppendElement(interp, pResult,
240211d88e68Sdrh Tcl_NewIntObj(v));
240311d88e68Sdrh }
240411d88e68Sdrh }else{
240511d88e68Sdrh const char *zOpt = Tcl_GetString(objv[2]);
240611d88e68Sdrh int onoff = -1;
240711d88e68Sdrh int v = 0;
240811d88e68Sdrh if( zOpt[0]=='-' ) zOpt++;
240911d88e68Sdrh for(ii=0; ii<sizeof(aDbConfig)/sizeof(aDbConfig[0]); ii++){
241011d88e68Sdrh if( strcmp(aDbConfig[ii].zName, zOpt)==0 ) break;
241111d88e68Sdrh }
241211d88e68Sdrh if( ii>=sizeof(aDbConfig)/sizeof(aDbConfig[0]) ){
241311d88e68Sdrh Tcl_AppendResult(interp, "unknown config option: \"", zOpt,
241411d88e68Sdrh "\"", (void*)0);
241511d88e68Sdrh return TCL_ERROR;
241611d88e68Sdrh }
241711d88e68Sdrh if( objc==4 ){
241811d88e68Sdrh if( Tcl_GetBooleanFromObj(interp, objv[3], &onoff) ){
241911d88e68Sdrh return TCL_ERROR;
242011d88e68Sdrh }
242111d88e68Sdrh }
24224043cfefSdrh sqlite3_db_config(pDb->db, aDbConfig[ii].op, onoff, &v);
242311d88e68Sdrh pResult = Tcl_NewIntObj(v);
242411d88e68Sdrh }
242511d88e68Sdrh Tcl_SetObjResult(interp, pResult);
242611d88e68Sdrh break;
242711d88e68Sdrh }
242811d88e68Sdrh
242919e2d37fSdrh /* $db copy conflict-algorithm table filename ?SEPARATOR? ?NULLINDICATOR?
243019e2d37fSdrh **
243119e2d37fSdrh ** Copy data into table from filename, optionally using SEPARATOR
243219e2d37fSdrh ** as column separators. If a column contains a null string, or the
243319e2d37fSdrh ** value of NULLINDICATOR, a NULL is inserted for the column.
243419e2d37fSdrh ** conflict-algorithm is one of the sqlite conflict algorithms:
243519e2d37fSdrh ** rollback, abort, fail, ignore, replace
243619e2d37fSdrh ** On success, return the number of lines processed, not necessarily same
243719e2d37fSdrh ** as 'db changes' due to conflict-algorithm selected.
243819e2d37fSdrh **
243919e2d37fSdrh ** This code is basically an implementation/enhancement of
244019e2d37fSdrh ** the sqlite3 shell.c ".import" command.
244119e2d37fSdrh **
244219e2d37fSdrh ** This command usage is equivalent to the sqlite2.x COPY statement,
244319e2d37fSdrh ** which imports file data into a table using the PostgreSQL COPY file format:
244419e2d37fSdrh ** $db copy $conflit_algo $table_name $filename \t \\N
244519e2d37fSdrh */
244619e2d37fSdrh case DB_COPY: {
244719e2d37fSdrh char *zTable; /* Insert data into this table */
244819e2d37fSdrh char *zFile; /* The file from which to extract data */
244919e2d37fSdrh char *zConflict; /* The conflict algorithm to use */
245019e2d37fSdrh sqlite3_stmt *pStmt; /* A statement */
245119e2d37fSdrh int nCol; /* Number of columns in the table */
245219e2d37fSdrh int nByte; /* Number of bytes in an SQL string */
245319e2d37fSdrh int i, j; /* Loop counters */
245419e2d37fSdrh int nSep; /* Number of bytes in zSep[] */
245519e2d37fSdrh int nNull; /* Number of bytes in zNull[] */
245619e2d37fSdrh char *zSql; /* An SQL statement */
245719e2d37fSdrh char *zLine; /* A single line of input from the file */
245819e2d37fSdrh char **azCol; /* zLine[] broken up into columns */
24596ef5e12eSmistachkin const char *zCommit; /* How to commit changes */
246019e2d37fSdrh FILE *in; /* The input file */
246119e2d37fSdrh int lineno = 0; /* Line number of input file */
246219e2d37fSdrh char zLineNum[80]; /* Line number print buffer */
246319e2d37fSdrh Tcl_Obj *pResult; /* interp result */
246419e2d37fSdrh
24656ef5e12eSmistachkin const char *zSep;
24666ef5e12eSmistachkin const char *zNull;
246719e2d37fSdrh if( objc<5 || objc>7 ){
246819e2d37fSdrh Tcl_WrongNumArgs(interp, 2, objv,
246919e2d37fSdrh "CONFLICT-ALGORITHM TABLE FILENAME ?SEPARATOR? ?NULLINDICATOR?");
247019e2d37fSdrh return TCL_ERROR;
247119e2d37fSdrh }
247219e2d37fSdrh if( objc>=6 ){
247319e2d37fSdrh zSep = Tcl_GetStringFromObj(objv[5], 0);
247419e2d37fSdrh }else{
247519e2d37fSdrh zSep = "\t";
247619e2d37fSdrh }
247719e2d37fSdrh if( objc>=7 ){
247819e2d37fSdrh zNull = Tcl_GetStringFromObj(objv[6], 0);
247919e2d37fSdrh }else{
248019e2d37fSdrh zNull = "";
248119e2d37fSdrh }
248219e2d37fSdrh zConflict = Tcl_GetStringFromObj(objv[2], 0);
248319e2d37fSdrh zTable = Tcl_GetStringFromObj(objv[3], 0);
248419e2d37fSdrh zFile = Tcl_GetStringFromObj(objv[4], 0);
24854f21c4afSdrh nSep = strlen30(zSep);
24864f21c4afSdrh nNull = strlen30(zNull);
248719e2d37fSdrh if( nSep==0 ){
2488a198f2b5Sdrh Tcl_AppendResult(interp,"Error: non-null separator required for copy",
2489a198f2b5Sdrh (char*)0);
249019e2d37fSdrh return TCL_ERROR;
249119e2d37fSdrh }
24923e59c012Sdrh if(strcmp(zConflict, "rollback") != 0 &&
24933e59c012Sdrh strcmp(zConflict, "abort" ) != 0 &&
24943e59c012Sdrh strcmp(zConflict, "fail" ) != 0 &&
24953e59c012Sdrh strcmp(zConflict, "ignore" ) != 0 &&
24963e59c012Sdrh strcmp(zConflict, "replace" ) != 0 ) {
249719e2d37fSdrh Tcl_AppendResult(interp, "Error: \"", zConflict,
249819e2d37fSdrh "\", conflict-algorithm must be one of: rollback, "
2499a198f2b5Sdrh "abort, fail, ignore, or replace", (char*)0);
250019e2d37fSdrh return TCL_ERROR;
250119e2d37fSdrh }
250219e2d37fSdrh zSql = sqlite3_mprintf("SELECT * FROM '%q'", zTable);
250319e2d37fSdrh if( zSql==0 ){
2504a198f2b5Sdrh Tcl_AppendResult(interp, "Error: no such table: ", zTable, (char*)0);
250519e2d37fSdrh return TCL_ERROR;
250619e2d37fSdrh }
25074f21c4afSdrh nByte = strlen30(zSql);
25083e701a18Sdrh rc = sqlite3_prepare(pDb->db, zSql, -1, &pStmt, 0);
250919e2d37fSdrh sqlite3_free(zSql);
251019e2d37fSdrh if( rc ){
2511a198f2b5Sdrh Tcl_AppendResult(interp, "Error: ", sqlite3_errmsg(pDb->db), (char*)0);
251219e2d37fSdrh nCol = 0;
251319e2d37fSdrh }else{
251419e2d37fSdrh nCol = sqlite3_column_count(pStmt);
251519e2d37fSdrh }
251619e2d37fSdrh sqlite3_finalize(pStmt);
251719e2d37fSdrh if( nCol==0 ) {
251819e2d37fSdrh return TCL_ERROR;
251919e2d37fSdrh }
252019e2d37fSdrh zSql = malloc( nByte + 50 + nCol*2 );
252119e2d37fSdrh if( zSql==0 ) {
2522a198f2b5Sdrh Tcl_AppendResult(interp, "Error: can't malloc()", (char*)0);
252319e2d37fSdrh return TCL_ERROR;
252419e2d37fSdrh }
252519e2d37fSdrh sqlite3_snprintf(nByte+50, zSql, "INSERT OR %q INTO '%q' VALUES(?",
252619e2d37fSdrh zConflict, zTable);
25274f21c4afSdrh j = strlen30(zSql);
252819e2d37fSdrh for(i=1; i<nCol; i++){
252919e2d37fSdrh zSql[j++] = ',';
253019e2d37fSdrh zSql[j++] = '?';
253119e2d37fSdrh }
253219e2d37fSdrh zSql[j++] = ')';
253319e2d37fSdrh zSql[j] = 0;
25343e701a18Sdrh rc = sqlite3_prepare(pDb->db, zSql, -1, &pStmt, 0);
253519e2d37fSdrh free(zSql);
253619e2d37fSdrh if( rc ){
2537a198f2b5Sdrh Tcl_AppendResult(interp, "Error: ", sqlite3_errmsg(pDb->db), (char*)0);
253819e2d37fSdrh sqlite3_finalize(pStmt);
253919e2d37fSdrh return TCL_ERROR;
254019e2d37fSdrh }
254119e2d37fSdrh in = fopen(zFile, "rb");
254219e2d37fSdrh if( in==0 ){
2543ea8f0a15Sdrh Tcl_AppendResult(interp, "Error: cannot open file: ", zFile, (char*)0);
254419e2d37fSdrh sqlite3_finalize(pStmt);
254519e2d37fSdrh return TCL_ERROR;
254619e2d37fSdrh }
254719e2d37fSdrh azCol = malloc( sizeof(azCol[0])*(nCol+1) );
254819e2d37fSdrh if( azCol==0 ) {
2549a198f2b5Sdrh Tcl_AppendResult(interp, "Error: can't malloc()", (char*)0);
255043617e9aSdrh fclose(in);
255119e2d37fSdrh return TCL_ERROR;
255219e2d37fSdrh }
25533752785fSdrh (void)sqlite3_exec(pDb->db, "BEGIN", 0, 0, 0);
255419e2d37fSdrh zCommit = "COMMIT";
255519e2d37fSdrh while( (zLine = local_getline(0, in))!=0 ){
255619e2d37fSdrh char *z;
255719e2d37fSdrh lineno++;
255819e2d37fSdrh azCol[0] = zLine;
255919e2d37fSdrh for(i=0, z=zLine; *z; z++){
256019e2d37fSdrh if( *z==zSep[0] && strncmp(z, zSep, nSep)==0 ){
256119e2d37fSdrh *z = 0;
256219e2d37fSdrh i++;
256319e2d37fSdrh if( i<nCol ){
256419e2d37fSdrh azCol[i] = &z[nSep];
256519e2d37fSdrh z += nSep-1;
256619e2d37fSdrh }
256719e2d37fSdrh }
256819e2d37fSdrh }
256919e2d37fSdrh if( i+1!=nCol ){
257019e2d37fSdrh char *zErr;
25714f21c4afSdrh int nErr = strlen30(zFile) + 200;
25725bb3eb9bSdrh zErr = malloc(nErr);
2573c1f4494eSdrh if( zErr ){
25745bb3eb9bSdrh sqlite3_snprintf(nErr, zErr,
2575955de52cSdanielk1977 "Error: %s line %d: expected %d columns of data but found %d",
257619e2d37fSdrh zFile, lineno, nCol, i+1);
2577a198f2b5Sdrh Tcl_AppendResult(interp, zErr, (char*)0);
257819e2d37fSdrh free(zErr);
2579c1f4494eSdrh }
258019e2d37fSdrh zCommit = "ROLLBACK";
258119e2d37fSdrh break;
258219e2d37fSdrh }
258319e2d37fSdrh for(i=0; i<nCol; i++){
258419e2d37fSdrh /* check for null data, if so, bind as null */
2585ea678832Sdrh if( (nNull>0 && strcmp(azCol[i], zNull)==0)
25864f21c4afSdrh || strlen30(azCol[i])==0
2587ea678832Sdrh ){
258819e2d37fSdrh sqlite3_bind_null(pStmt, i+1);
258919e2d37fSdrh }else{
259019e2d37fSdrh sqlite3_bind_text(pStmt, i+1, azCol[i], -1, SQLITE_STATIC);
259119e2d37fSdrh }
259219e2d37fSdrh }
259319e2d37fSdrh sqlite3_step(pStmt);
259419e2d37fSdrh rc = sqlite3_reset(pStmt);
259519e2d37fSdrh free(zLine);
259619e2d37fSdrh if( rc!=SQLITE_OK ){
2597a198f2b5Sdrh Tcl_AppendResult(interp,"Error: ", sqlite3_errmsg(pDb->db), (char*)0);
259819e2d37fSdrh zCommit = "ROLLBACK";
259919e2d37fSdrh break;
260019e2d37fSdrh }
260119e2d37fSdrh }
260219e2d37fSdrh free(azCol);
260319e2d37fSdrh fclose(in);
260419e2d37fSdrh sqlite3_finalize(pStmt);
26053752785fSdrh (void)sqlite3_exec(pDb->db, zCommit, 0, 0, 0);
260619e2d37fSdrh
260719e2d37fSdrh if( zCommit[0] == 'C' ){
260819e2d37fSdrh /* success, set result as number of lines processed */
260919e2d37fSdrh pResult = Tcl_GetObjResult(interp);
261019e2d37fSdrh Tcl_SetIntObj(pResult, lineno);
261119e2d37fSdrh rc = TCL_OK;
261219e2d37fSdrh }else{
261319e2d37fSdrh /* failure, append lineno where failed */
26145bb3eb9bSdrh sqlite3_snprintf(sizeof(zLineNum), zLineNum,"%d",lineno);
2615a198f2b5Sdrh Tcl_AppendResult(interp,", failed while processing line: ",zLineNum,
2616a198f2b5Sdrh (char*)0);
261719e2d37fSdrh rc = TCL_ERROR;
261819e2d37fSdrh }
261919e2d37fSdrh break;
262019e2d37fSdrh }
262119e2d37fSdrh
262275897234Sdrh /*
26236ca64481Sdrh ** $db deserialize ?-maxsize N? ?-readonly BOOL? ?DATABASE? VALUE
2624cb7d541dSdrh **
2625cb7d541dSdrh ** Reopen DATABASE (default "main") using the content in $VALUE
2626cb7d541dSdrh */
2627cb7d541dSdrh case DB_DESERIALIZE: {
26288d889afcSdrh #ifdef SQLITE_OMIT_DESERIALIZE
26293ec8665eSdrh Tcl_AppendResult(interp, "MEMDB not available in this build",
26303ec8665eSdrh (char*)0);
26313ec8665eSdrh rc = TCL_ERROR;
26323ec8665eSdrh #else
26336ca64481Sdrh const char *zSchema = 0;
26346ca64481Sdrh Tcl_Obj *pValue = 0;
26353ec8665eSdrh unsigned char *pBA;
26363ec8665eSdrh unsigned char *pData;
26373ec8665eSdrh int len, xrc;
26386ca64481Sdrh sqlite3_int64 mxSize = 0;
26396ca64481Sdrh int i;
26406ca64481Sdrh int isReadonly = 0;
26413ec8665eSdrh
26426ca64481Sdrh
26436ca64481Sdrh if( objc<3 ){
26443ec8665eSdrh Tcl_WrongNumArgs(interp, 2, objv, "?DATABASE? VALUE");
26453ec8665eSdrh rc = TCL_ERROR;
26463ec8665eSdrh break;
26473ec8665eSdrh }
26486ca64481Sdrh for(i=2; i<objc-1; i++){
26496ca64481Sdrh const char *z = Tcl_GetString(objv[i]);
26506ca64481Sdrh if( strcmp(z,"-maxsize")==0 && i<objc-2 ){
26517907d372Sdrh Tcl_WideInt x;
26527907d372Sdrh rc = Tcl_GetWideIntFromObj(interp, objv[++i], &x);
26536ca64481Sdrh if( rc ) goto deserialize_error;
26547907d372Sdrh mxSize = x;
26556ca64481Sdrh continue;
26566ca64481Sdrh }
26576ca64481Sdrh if( strcmp(z,"-readonly")==0 && i<objc-2 ){
26586ca64481Sdrh rc = Tcl_GetBooleanFromObj(interp, objv[++i], &isReadonly);
26596ca64481Sdrh if( rc ) goto deserialize_error;
26606ca64481Sdrh continue;
26616ca64481Sdrh }
26626ca64481Sdrh if( zSchema==0 && i==objc-2 && z[0]!='-' ){
26636ca64481Sdrh zSchema = z;
26646ca64481Sdrh continue;
26656ca64481Sdrh }
26666ca64481Sdrh Tcl_AppendResult(interp, "unknown option: ", z, (char*)0);
26676ca64481Sdrh rc = TCL_ERROR;
26686ca64481Sdrh goto deserialize_error;
26696ca64481Sdrh }
26706ca64481Sdrh pValue = objv[objc-1];
26713ec8665eSdrh pBA = Tcl_GetByteArrayFromObj(pValue, &len);
26723ec8665eSdrh pData = sqlite3_malloc64( len );
2673a5bb4351Sdrh if( pData==0 && len>0 ){
26743ec8665eSdrh Tcl_AppendResult(interp, "out of memory", (char*)0);
26753ec8665eSdrh rc = TCL_ERROR;
26763ec8665eSdrh }else{
26776ca64481Sdrh int flags;
2678a5bb4351Sdrh if( len>0 ) memcpy(pData, pBA, len);
26796ca64481Sdrh if( isReadonly ){
26806ca64481Sdrh flags = SQLITE_DESERIALIZE_FREEONCLOSE | SQLITE_DESERIALIZE_READONLY;
26816ca64481Sdrh }else{
26826ca64481Sdrh flags = SQLITE_DESERIALIZE_FREEONCLOSE | SQLITE_DESERIALIZE_RESIZEABLE;
26836ca64481Sdrh }
26846ca64481Sdrh xrc = sqlite3_deserialize(pDb->db, zSchema, pData, len, len, flags);
26853ec8665eSdrh if( xrc ){
26863ec8665eSdrh Tcl_AppendResult(interp, "unable to set MEMDB content", (char*)0);
26873ec8665eSdrh rc = TCL_ERROR;
26883ec8665eSdrh }
26896ca64481Sdrh if( mxSize>0 ){
26906ca64481Sdrh sqlite3_file_control(pDb->db, zSchema,SQLITE_FCNTL_SIZE_LIMIT,&mxSize);
26913ec8665eSdrh }
26926ca64481Sdrh }
26936ca64481Sdrh deserialize_error:
26943ec8665eSdrh #endif
2695cb7d541dSdrh break;
2696cb7d541dSdrh }
2697cb7d541dSdrh
2698cb7d541dSdrh /*
26994144905bSdrh ** $db enable_load_extension BOOLEAN
27004144905bSdrh **
27014144905bSdrh ** Turn the extension loading feature on or off. It if off by
27024144905bSdrh ** default.
27034144905bSdrh */
27044144905bSdrh case DB_ENABLE_LOAD_EXTENSION: {
2705f533acc0Sdrh #ifndef SQLITE_OMIT_LOAD_EXTENSION
27064144905bSdrh int onoff;
27074144905bSdrh if( objc!=3 ){
27084144905bSdrh Tcl_WrongNumArgs(interp, 2, objv, "BOOLEAN");
27094144905bSdrh return TCL_ERROR;
27104144905bSdrh }
27114144905bSdrh if( Tcl_GetBooleanFromObj(interp, objv[2], &onoff) ){
27124144905bSdrh return TCL_ERROR;
27134144905bSdrh }
27144144905bSdrh sqlite3_enable_load_extension(pDb->db, onoff);
27154144905bSdrh break;
2716f533acc0Sdrh #else
2717f533acc0Sdrh Tcl_AppendResult(interp, "extension loading is turned off at compile-time",
2718a198f2b5Sdrh (char*)0);
2719f533acc0Sdrh return TCL_ERROR;
2720f533acc0Sdrh #endif
27214144905bSdrh }
27224144905bSdrh
27234144905bSdrh /*
2724dcd997eaSdrh ** $db errorcode
2725dcd997eaSdrh **
2726dcd997eaSdrh ** Return the numeric error code that was returned by the most recent
27276f8a503dSdanielk1977 ** call to sqlite3_exec().
2728dcd997eaSdrh */
2729dcd997eaSdrh case DB_ERRORCODE: {
2730f3ce83f5Sdanielk1977 Tcl_SetObjResult(interp, Tcl_NewIntObj(sqlite3_errcode(pDb->db)));
2731dcd997eaSdrh break;
2732dcd997eaSdrh }
2733dcd997eaSdrh
2734dcd997eaSdrh /*
2735be4e3c88Sdrh ** $db erroroffset
2736be4e3c88Sdrh **
2737be4e3c88Sdrh ** Return the numeric error code that was returned by the most recent
2738be4e3c88Sdrh ** call to sqlite3_exec().
2739be4e3c88Sdrh */
2740be4e3c88Sdrh case DB_ERROROFFSET: {
2741be4e3c88Sdrh Tcl_SetObjResult(interp, Tcl_NewIntObj(sqlite3_error_offset(pDb->db)));
2742be4e3c88Sdrh break;
2743be4e3c88Sdrh }
2744be4e3c88Sdrh
2745be4e3c88Sdrh /*
27464a4c11aaSdan ** $db exists $sql
27471807ce37Sdrh ** $db onecolumn $sql
274875897234Sdrh **
27494a4c11aaSdan ** The onecolumn method is the equivalent of:
27504a4c11aaSdan ** lindex [$db eval $sql] 0
27514a4c11aaSdan */
27524a4c11aaSdan case DB_EXISTS:
27534a4c11aaSdan case DB_ONECOLUMN: {
2754edc4024bSdrh Tcl_Obj *pResult = 0;
27554a4c11aaSdan DbEvalContext sEval;
27564a4c11aaSdan if( objc!=3 ){
27574a4c11aaSdan Tcl_WrongNumArgs(interp, 2, objv, "SQL");
27584a4c11aaSdan return TCL_ERROR;
27594a4c11aaSdan }
27604a4c11aaSdan
2761af38cdbcSdrh dbEvalInit(&sEval, pDb, objv[2], 0, 0);
27624a4c11aaSdan rc = dbEvalStep(&sEval);
27634a4c11aaSdan if( choice==DB_ONECOLUMN ){
27644a4c11aaSdan if( rc==TCL_OK ){
2765edc4024bSdrh pResult = dbEvalColumnValue(&sEval, 0);
2766d5f12cd5Sdan }else if( rc==TCL_BREAK ){
2767d5f12cd5Sdan Tcl_ResetResult(interp);
27684a4c11aaSdan }
27694a4c11aaSdan }else if( rc==TCL_BREAK || rc==TCL_OK ){
2770edc4024bSdrh pResult = Tcl_NewBooleanObj(rc==TCL_OK);
27714a4c11aaSdan }
27724a4c11aaSdan dbEvalFinalize(&sEval);
2773edc4024bSdrh if( pResult ) Tcl_SetObjResult(interp, pResult);
27744a4c11aaSdan
27754a4c11aaSdan if( rc==TCL_BREAK ){
27764a4c11aaSdan rc = TCL_OK;
27774a4c11aaSdan }
27784a4c11aaSdan break;
27794a4c11aaSdan }
27804a4c11aaSdan
27814a4c11aaSdan /*
2782af38cdbcSdrh ** $db eval ?options? $sql ?array? ?{ ...code... }?
27834a4c11aaSdan **
278475897234Sdrh ** The SQL statement in $sql is evaluated. For each row, the values are
2785bec3f402Sdrh ** placed in elements of the array named "array" and ...code... is executed.
278675897234Sdrh ** If "array" and "code" are omitted, then no callback is every invoked.
278775897234Sdrh ** If "array" is an empty string, then the values are placed in variables
278875897234Sdrh ** that have the same name as the fields extracted by the query.
278975897234Sdrh */
27904a4c11aaSdan case DB_EVAL: {
2791af38cdbcSdrh int evalFlags = 0;
2792af38cdbcSdrh const char *zOpt;
2793af38cdbcSdrh while( objc>3 && (zOpt = Tcl_GetString(objv[2]))!=0 && zOpt[0]=='-' ){
2794af38cdbcSdrh if( strcmp(zOpt, "-withoutnulls")==0 ){
2795af38cdbcSdrh evalFlags |= SQLITE_EVAL_WITHOUTNULLS;
2796af38cdbcSdrh }
2797af38cdbcSdrh else{
2798af38cdbcSdrh Tcl_AppendResult(interp, "unknown option: \"", zOpt, "\"", (void*)0);
2799af38cdbcSdrh return TCL_ERROR;
2800af38cdbcSdrh }
2801af38cdbcSdrh objc--;
2802af38cdbcSdrh objv++;
2803af38cdbcSdrh }
280492febd92Sdrh if( objc<3 || objc>5 ){
2805af38cdbcSdrh Tcl_WrongNumArgs(interp, 2, objv,
2806af38cdbcSdrh "?OPTIONS? SQL ?ARRAY-NAME? ?SCRIPT?");
280730ccda10Sdanielk1977 return TCL_ERROR;
280830ccda10Sdanielk1977 }
28094a4c11aaSdan
281092febd92Sdrh if( objc==3 ){
28114a4c11aaSdan DbEvalContext sEval;
28124a4c11aaSdan Tcl_Obj *pRet = Tcl_NewObj();
28134a4c11aaSdan Tcl_IncrRefCount(pRet);
2814af38cdbcSdrh dbEvalInit(&sEval, pDb, objv[2], 0, 0);
28154a4c11aaSdan while( TCL_OK==(rc = dbEvalStep(&sEval)) ){
28164a4c11aaSdan int i;
28174a4c11aaSdan int nCol;
28184a4c11aaSdan dbEvalRowInfo(&sEval, &nCol, 0);
281992febd92Sdrh for(i=0; i<nCol; i++){
28204a4c11aaSdan Tcl_ListObjAppendElement(interp, pRet, dbEvalColumnValue(&sEval, i));
282192febd92Sdrh }
282230ccda10Sdanielk1977 }
28234a4c11aaSdan dbEvalFinalize(&sEval);
282490b6bb19Sdrh if( rc==TCL_BREAK ){
28254a4c11aaSdan Tcl_SetObjResult(interp, pRet);
282690b6bb19Sdrh rc = TCL_OK;
282790b6bb19Sdrh }
2828ef2cb63eSdanielk1977 Tcl_DecrRefCount(pRet);
28294a4c11aaSdan }else{
28308e18922fSmistachkin ClientData cd2[2];
28314a4c11aaSdan DbEvalContext *p;
28324a4c11aaSdan Tcl_Obj *pArray = 0;
28334a4c11aaSdan Tcl_Obj *pScript;
28344a4c11aaSdan
2835af38cdbcSdrh if( objc>=5 && *(char *)Tcl_GetString(objv[3]) ){
28364a4c11aaSdan pArray = objv[3];
28374a4c11aaSdan }
28384a4c11aaSdan pScript = objv[objc-1];
28394a4c11aaSdan Tcl_IncrRefCount(pScript);
28404a4c11aaSdan
28414a4c11aaSdan p = (DbEvalContext *)Tcl_Alloc(sizeof(DbEvalContext));
2842af38cdbcSdrh dbEvalInit(p, pDb, objv[2], pArray, evalFlags);
28434a4c11aaSdan
28448e18922fSmistachkin cd2[0] = (void *)p;
28458e18922fSmistachkin cd2[1] = (void *)pScript;
28468e18922fSmistachkin rc = DbEvalNextCmd(cd2, interp, TCL_OK);
28471807ce37Sdrh }
284830ccda10Sdanielk1977 break;
284930ccda10Sdanielk1977 }
2850bec3f402Sdrh
2851bec3f402Sdrh /*
285242d2fce7Sdrh ** $db function NAME [OPTIONS] SCRIPT
2853cabb0819Sdrh **
2854cabb0819Sdrh ** Create a new SQL function called NAME. Whenever that function is
2855cabb0819Sdrh ** called, invoke SCRIPT to evaluate the function.
285642d2fce7Sdrh **
285742d2fce7Sdrh ** Options:
285842d2fce7Sdrh ** --argcount N Function has exactly N arguments
285942d2fce7Sdrh ** --deterministic The function is pure
286042d2fce7Sdrh ** --directonly Prohibit use inside triggers and views
286115f3eacfSdrh ** --innocuous Has no side effects or information leaks
286242d2fce7Sdrh ** --returntype TYPE Specify the return type of the function
2863cabb0819Sdrh */
2864cabb0819Sdrh case DB_FUNCTION: {
28653df30592Sdan int flags = SQLITE_UTF8;
2866cabb0819Sdrh SqlFunc *pFunc;
2867d1e4733dSdrh Tcl_Obj *pScript;
2868cabb0819Sdrh char *zName;
2869e3602be8Sdrh int nArg = -1;
28703df30592Sdan int i;
287189d24936Sdan int eType = SQLITE_NULL;
28723df30592Sdan if( objc<4 ){
28733df30592Sdan Tcl_WrongNumArgs(interp, 2, objv, "NAME ?SWITCHES? SCRIPT");
28743df30592Sdan return TCL_ERROR;
28753df30592Sdan }
28763df30592Sdan for(i=3; i<(objc-1); i++){
28773df30592Sdan const char *z = Tcl_GetString(objv[i]);
28784f21c4afSdrh int n = strlen30(z);
287989d24936Sdan if( n>1 && strncmp(z, "-argcount",n)==0 ){
28803df30592Sdan if( i==(objc-2) ){
2881ea8f0a15Sdrh Tcl_AppendResult(interp, "option requires an argument: ", z,(char*)0);
28823df30592Sdan return TCL_ERROR;
28833df30592Sdan }
28843df30592Sdan if( Tcl_GetIntFromObj(interp, objv[i+1], &nArg) ) return TCL_ERROR;
2885e3602be8Sdrh if( nArg<0 ){
2886e3602be8Sdrh Tcl_AppendResult(interp, "number of arguments must be non-negative",
2887e3602be8Sdrh (char*)0);
2888cabb0819Sdrh return TCL_ERROR;
2889cabb0819Sdrh }
28903df30592Sdan i++;
28913df30592Sdan }else
289289d24936Sdan if( n>1 && strncmp(z, "-deterministic",n)==0 ){
28933df30592Sdan flags |= SQLITE_DETERMINISTIC;
289489d24936Sdan }else
289542d2fce7Sdrh if( n>1 && strncmp(z, "-directonly",n)==0 ){
289642d2fce7Sdrh flags |= SQLITE_DIRECTONLY;
289742d2fce7Sdrh }else
289815f3eacfSdrh if( n>1 && strncmp(z, "-innocuous",n)==0 ){
289915f3eacfSdrh flags |= SQLITE_INNOCUOUS;
290015f3eacfSdrh }else
290189d24936Sdan if( n>1 && strncmp(z, "-returntype", n)==0 ){
290289d24936Sdan const char *azType[] = {"integer", "real", "text", "blob", "any", 0};
290389d24936Sdan assert( SQLITE_INTEGER==1 && SQLITE_FLOAT==2 && SQLITE_TEXT==3 );
290489d24936Sdan assert( SQLITE_BLOB==4 && SQLITE_NULL==5 );
290589d24936Sdan if( i==(objc-2) ){
290689d24936Sdan Tcl_AppendResult(interp, "option requires an argument: ", z,(char*)0);
290789d24936Sdan return TCL_ERROR;
290889d24936Sdan }
290989d24936Sdan i++;
291089d24936Sdan if( Tcl_GetIndexFromObj(interp, objv[i], azType, "type", 0, &eType) ){
291189d24936Sdan return TCL_ERROR;
291289d24936Sdan }
291389d24936Sdan eType++;
2914e3602be8Sdrh }else{
29153df30592Sdan Tcl_AppendResult(interp, "bad option \"", z,
291642d2fce7Sdrh "\": must be -argcount, -deterministic, -directonly,"
291715f3eacfSdrh " -innocuous, or -returntype", (char*)0
29183df30592Sdan );
29193df30592Sdan return TCL_ERROR;
2920e3602be8Sdrh }
29213df30592Sdan }
29223df30592Sdan
29233df30592Sdan pScript = objv[objc-1];
2924e3602be8Sdrh zName = Tcl_GetStringFromObj(objv[2], 0);
2925d1e4733dSdrh pFunc = findSqlFunc(pDb, zName);
2926cabb0819Sdrh if( pFunc==0 ) return TCL_ERROR;
2927d1e4733dSdrh if( pFunc->pScript ){
2928d1e4733dSdrh Tcl_DecrRefCount(pFunc->pScript);
2929d1e4733dSdrh }
2930d1e4733dSdrh pFunc->pScript = pScript;
2931d1e4733dSdrh Tcl_IncrRefCount(pScript);
2932d1e4733dSdrh pFunc->useEvalObjv = safeToUseEvalObjv(interp, pScript);
293389d24936Sdan pFunc->eType = eType;
29343df30592Sdan rc = sqlite3_create_function(pDb->db, zName, nArg, flags,
2935d8123366Sdanielk1977 pFunc, tclSqlFunc, 0, 0);
2936fb7e7651Sdrh if( rc!=SQLITE_OK ){
2937fb7e7651Sdrh rc = TCL_ERROR;
29389636c4e1Sdanielk1977 Tcl_SetResult(interp, (char *)sqlite3_errmsg(pDb->db), TCL_VOLATILE);
2939fb7e7651Sdrh }
2940cabb0819Sdrh break;
2941cabb0819Sdrh }
2942cabb0819Sdrh
2943cabb0819Sdrh /*
29448cbadb02Sdanielk1977 ** $db incrblob ?-readonly? ?DB? TABLE COLUMN ROWID
2945b4e9af9fSdanielk1977 */
2946b4e9af9fSdanielk1977 case DB_INCRBLOB: {
294732a0d8bbSdanielk1977 #ifdef SQLITE_OMIT_INCRBLOB
2948a198f2b5Sdrh Tcl_AppendResult(interp, "incrblob not available in this build", (char*)0);
294932a0d8bbSdanielk1977 return TCL_ERROR;
295032a0d8bbSdanielk1977 #else
29518cbadb02Sdanielk1977 int isReadonly = 0;
2952b4e9af9fSdanielk1977 const char *zDb = "main";
2953b4e9af9fSdanielk1977 const char *zTable;
2954b4e9af9fSdanielk1977 const char *zColumn;
2955b3f787f4Sdrh Tcl_WideInt iRow;
2956b4e9af9fSdanielk1977
29578cbadb02Sdanielk1977 /* Check for the -readonly option */
29588cbadb02Sdanielk1977 if( objc>3 && strcmp(Tcl_GetString(objv[2]), "-readonly")==0 ){
29598cbadb02Sdanielk1977 isReadonly = 1;
29608cbadb02Sdanielk1977 }
29618cbadb02Sdanielk1977
29628cbadb02Sdanielk1977 if( objc!=(5+isReadonly) && objc!=(6+isReadonly) ){
29638cbadb02Sdanielk1977 Tcl_WrongNumArgs(interp, 2, objv, "?-readonly? ?DB? TABLE COLUMN ROWID");
2964b4e9af9fSdanielk1977 return TCL_ERROR;
2965b4e9af9fSdanielk1977 }
2966b4e9af9fSdanielk1977
29678cbadb02Sdanielk1977 if( objc==(6+isReadonly) ){
2968*f74f4ed5Sdan zDb = Tcl_GetString(objv[2+isReadonly]);
2969b4e9af9fSdanielk1977 }
2970b4e9af9fSdanielk1977 zTable = Tcl_GetString(objv[objc-3]);
2971b4e9af9fSdanielk1977 zColumn = Tcl_GetString(objv[objc-2]);
2972b4e9af9fSdanielk1977 rc = Tcl_GetWideIntFromObj(interp, objv[objc-1], &iRow);
2973b4e9af9fSdanielk1977
2974b4e9af9fSdanielk1977 if( rc==TCL_OK ){
29758cbadb02Sdanielk1977 rc = createIncrblobChannel(
2976edf5b165Sdan interp, pDb, zDb, zTable, zColumn, (sqlite3_int64)iRow, isReadonly
29778cbadb02Sdanielk1977 );
2978b4e9af9fSdanielk1977 }
297932a0d8bbSdanielk1977 #endif
2980b4e9af9fSdanielk1977 break;
2981b4e9af9fSdanielk1977 }
2982b4e9af9fSdanielk1977
2983b4e9af9fSdanielk1977 /*
2984f11bded5Sdrh ** $db interrupt
2985f11bded5Sdrh **
2986f11bded5Sdrh ** Interrupt the execution of the inner-most SQL interpreter. This
2987f11bded5Sdrh ** causes the SQL statement to return an error of SQLITE_INTERRUPT.
2988f11bded5Sdrh */
2989f11bded5Sdrh case DB_INTERRUPT: {
2990f11bded5Sdrh sqlite3_interrupt(pDb->db);
2991f11bded5Sdrh break;
2992f11bded5Sdrh }
2993f11bded5Sdrh
2994f11bded5Sdrh /*
299519e2d37fSdrh ** $db nullvalue ?STRING?
299619e2d37fSdrh **
299719e2d37fSdrh ** Change text used when a NULL comes back from the database. If ?STRING?
299819e2d37fSdrh ** is not present, then the current string used for NULL is returned.
299919e2d37fSdrh ** If STRING is present, then STRING is returned.
300019e2d37fSdrh **
300119e2d37fSdrh */
300219e2d37fSdrh case DB_NULLVALUE: {
300319e2d37fSdrh if( objc!=2 && objc!=3 ){
300419e2d37fSdrh Tcl_WrongNumArgs(interp, 2, objv, "NULLVALUE");
300519e2d37fSdrh return TCL_ERROR;
300619e2d37fSdrh }
300719e2d37fSdrh if( objc==3 ){
300819e2d37fSdrh int len;
300919e2d37fSdrh char *zNull = Tcl_GetStringFromObj(objv[2], &len);
301019e2d37fSdrh if( pDb->zNull ){
301119e2d37fSdrh Tcl_Free(pDb->zNull);
301219e2d37fSdrh }
301319e2d37fSdrh if( zNull && len>0 ){
301419e2d37fSdrh pDb->zNull = Tcl_Alloc( len + 1 );
30157fd33929Sdrh memcpy(pDb->zNull, zNull, len);
301619e2d37fSdrh pDb->zNull[len] = '\0';
301719e2d37fSdrh }else{
301819e2d37fSdrh pDb->zNull = 0;
301919e2d37fSdrh }
302019e2d37fSdrh }
3021c45e6716Sdrh Tcl_SetObjResult(interp, Tcl_NewStringObj(pDb->zNull, -1));
302219e2d37fSdrh break;
302319e2d37fSdrh }
302419e2d37fSdrh
302519e2d37fSdrh /*
3026af9ff33aSdrh ** $db last_insert_rowid
3027af9ff33aSdrh **
3028af9ff33aSdrh ** Return an integer which is the ROWID for the most recent insert.
3029af9ff33aSdrh */
3030af9ff33aSdrh case DB_LAST_INSERT_ROWID: {
3031af9ff33aSdrh Tcl_Obj *pResult;
3032f7e678d6Sdrh Tcl_WideInt rowid;
3033af9ff33aSdrh if( objc!=2 ){
3034af9ff33aSdrh Tcl_WrongNumArgs(interp, 2, objv, "");
3035af9ff33aSdrh return TCL_ERROR;
3036af9ff33aSdrh }
30376f8a503dSdanielk1977 rowid = sqlite3_last_insert_rowid(pDb->db);
3038af9ff33aSdrh pResult = Tcl_GetObjResult(interp);
3039f7e678d6Sdrh Tcl_SetWideIntObj(pResult, rowid);
3040af9ff33aSdrh break;
3041af9ff33aSdrh }
3042af9ff33aSdrh
3043af9ff33aSdrh /*
30444a4c11aaSdan ** The DB_ONECOLUMN method is implemented together with DB_EXISTS.
30455d9d7576Sdrh */
30461807ce37Sdrh
30471807ce37Sdrh /* $db progress ?N CALLBACK?
30481807ce37Sdrh **
30491807ce37Sdrh ** Invoke the given callback every N virtual machine opcodes while executing
30501807ce37Sdrh ** queries.
30511807ce37Sdrh */
30521807ce37Sdrh case DB_PROGRESS: {
30531807ce37Sdrh if( objc==2 ){
30541807ce37Sdrh if( pDb->zProgress ){
3055a198f2b5Sdrh Tcl_AppendResult(interp, pDb->zProgress, (char*)0);
30565d9d7576Sdrh }
30571807ce37Sdrh }else if( objc==4 ){
30581807ce37Sdrh char *zProgress;
30591807ce37Sdrh int len;
30601807ce37Sdrh int N;
30611807ce37Sdrh if( TCL_OK!=Tcl_GetIntFromObj(interp, objv[2], &N) ){
30621807ce37Sdrh return TCL_ERROR;
30631807ce37Sdrh };
30641807ce37Sdrh if( pDb->zProgress ){
30651807ce37Sdrh Tcl_Free(pDb->zProgress);
30661807ce37Sdrh }
30671807ce37Sdrh zProgress = Tcl_GetStringFromObj(objv[3], &len);
30681807ce37Sdrh if( zProgress && len>0 ){
30691807ce37Sdrh pDb->zProgress = Tcl_Alloc( len + 1 );
30705bb3eb9bSdrh memcpy(pDb->zProgress, zProgress, len+1);
30711807ce37Sdrh }else{
30721807ce37Sdrh pDb->zProgress = 0;
30731807ce37Sdrh }
30741807ce37Sdrh #ifndef SQLITE_OMIT_PROGRESS_CALLBACK
30751807ce37Sdrh if( pDb->zProgress ){
30761807ce37Sdrh pDb->interp = interp;
30771807ce37Sdrh sqlite3_progress_handler(pDb->db, N, DbProgressHandler, pDb);
30781807ce37Sdrh }else{
30791807ce37Sdrh sqlite3_progress_handler(pDb->db, 0, 0, 0);
30801807ce37Sdrh }
30811807ce37Sdrh #endif
30821807ce37Sdrh }else{
30831807ce37Sdrh Tcl_WrongNumArgs(interp, 2, objv, "N CALLBACK");
30841807ce37Sdrh return TCL_ERROR;
30855d9d7576Sdrh }
30865d9d7576Sdrh break;
30875d9d7576Sdrh }
30885d9d7576Sdrh
308919e2d37fSdrh /* $db profile ?CALLBACK?
309019e2d37fSdrh **
309119e2d37fSdrh ** Make arrangements to invoke the CALLBACK routine after each SQL statement
309219e2d37fSdrh ** that has run. The text of the SQL and the amount of elapse time are
309319e2d37fSdrh ** appended to CALLBACK before the script is run.
309419e2d37fSdrh */
309519e2d37fSdrh case DB_PROFILE: {
309619e2d37fSdrh if( objc>3 ){
309719e2d37fSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?");
309819e2d37fSdrh return TCL_ERROR;
309919e2d37fSdrh }else if( objc==2 ){
310019e2d37fSdrh if( pDb->zProfile ){
3101a198f2b5Sdrh Tcl_AppendResult(interp, pDb->zProfile, (char*)0);
310219e2d37fSdrh }
310319e2d37fSdrh }else{
310419e2d37fSdrh char *zProfile;
310519e2d37fSdrh int len;
310619e2d37fSdrh if( pDb->zProfile ){
310719e2d37fSdrh Tcl_Free(pDb->zProfile);
310819e2d37fSdrh }
310919e2d37fSdrh zProfile = Tcl_GetStringFromObj(objv[2], &len);
311019e2d37fSdrh if( zProfile && len>0 ){
311119e2d37fSdrh pDb->zProfile = Tcl_Alloc( len + 1 );
31125bb3eb9bSdrh memcpy(pDb->zProfile, zProfile, len+1);
311319e2d37fSdrh }else{
311419e2d37fSdrh pDb->zProfile = 0;
311519e2d37fSdrh }
31162eb22af0Sdrh #if !defined(SQLITE_OMIT_TRACE) && !defined(SQLITE_OMIT_FLOATING_POINT) && \
31172eb22af0Sdrh !defined(SQLITE_OMIT_DEPRECATED)
311819e2d37fSdrh if( pDb->zProfile ){
311919e2d37fSdrh pDb->interp = interp;
312019e2d37fSdrh sqlite3_profile(pDb->db, DbProfileHandler, pDb);
312119e2d37fSdrh }else{
312219e2d37fSdrh sqlite3_profile(pDb->db, 0, 0);
312319e2d37fSdrh }
312419e2d37fSdrh #endif
312519e2d37fSdrh }
312619e2d37fSdrh break;
312719e2d37fSdrh }
312819e2d37fSdrh
31295d9d7576Sdrh /*
313022fbcb8dSdrh ** $db rekey KEY
313122fbcb8dSdrh **
313222fbcb8dSdrh ** Change the encryption key on the currently open database.
313322fbcb8dSdrh */
313422fbcb8dSdrh case DB_REKEY: {
313522fbcb8dSdrh if( objc!=3 ){
313622fbcb8dSdrh Tcl_WrongNumArgs(interp, 2, objv, "KEY");
313722fbcb8dSdrh return TCL_ERROR;
313822fbcb8dSdrh }
313922fbcb8dSdrh break;
314022fbcb8dSdrh }
314122fbcb8dSdrh
3142dc2c4915Sdrh /* $db restore ?DATABASE? FILENAME
3143dc2c4915Sdrh **
3144dc2c4915Sdrh ** Open a database file named FILENAME. Transfer the content
3145dc2c4915Sdrh ** of FILENAME into the local database DATABASE (default: "main").
3146dc2c4915Sdrh */
3147dc2c4915Sdrh case DB_RESTORE: {
3148dc2c4915Sdrh const char *zSrcFile;
3149dc2c4915Sdrh const char *zDestDb;
3150dc2c4915Sdrh sqlite3 *pSrc;
3151dc2c4915Sdrh sqlite3_backup *pBackup;
3152dc2c4915Sdrh int nTimeout = 0;
3153dc2c4915Sdrh
3154dc2c4915Sdrh if( objc==3 ){
3155dc2c4915Sdrh zDestDb = "main";
3156dc2c4915Sdrh zSrcFile = Tcl_GetString(objv[2]);
3157dc2c4915Sdrh }else if( objc==4 ){
3158dc2c4915Sdrh zDestDb = Tcl_GetString(objv[2]);
3159dc2c4915Sdrh zSrcFile = Tcl_GetString(objv[3]);
3160dc2c4915Sdrh }else{
3161dc2c4915Sdrh Tcl_WrongNumArgs(interp, 2, objv, "?DATABASE? FILENAME");
3162dc2c4915Sdrh return TCL_ERROR;
3163dc2c4915Sdrh }
3164147ef394Sdrh rc = sqlite3_open_v2(zSrcFile, &pSrc,
3165147ef394Sdrh SQLITE_OPEN_READONLY | pDb->openFlags, 0);
3166dc2c4915Sdrh if( rc!=SQLITE_OK ){
3167dc2c4915Sdrh Tcl_AppendResult(interp, "cannot open source database: ",
3168dc2c4915Sdrh sqlite3_errmsg(pSrc), (char*)0);
3169dc2c4915Sdrh sqlite3_close(pSrc);
3170dc2c4915Sdrh return TCL_ERROR;
3171dc2c4915Sdrh }
3172dc2c4915Sdrh pBackup = sqlite3_backup_init(pDb->db, zDestDb, pSrc, "main");
3173dc2c4915Sdrh if( pBackup==0 ){
3174dc2c4915Sdrh Tcl_AppendResult(interp, "restore failed: ",
3175dc2c4915Sdrh sqlite3_errmsg(pDb->db), (char*)0);
3176dc2c4915Sdrh sqlite3_close(pSrc);
3177dc2c4915Sdrh return TCL_ERROR;
3178dc2c4915Sdrh }
3179dc2c4915Sdrh while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK
3180dc2c4915Sdrh || rc==SQLITE_BUSY ){
3181dc2c4915Sdrh if( rc==SQLITE_BUSY ){
3182dc2c4915Sdrh if( nTimeout++ >= 3 ) break;
3183dc2c4915Sdrh sqlite3_sleep(100);
3184dc2c4915Sdrh }
3185dc2c4915Sdrh }
3186dc2c4915Sdrh sqlite3_backup_finish(pBackup);
3187dc2c4915Sdrh if( rc==SQLITE_DONE ){
3188dc2c4915Sdrh rc = TCL_OK;
3189dc2c4915Sdrh }else if( rc==SQLITE_BUSY || rc==SQLITE_LOCKED ){
3190dc2c4915Sdrh Tcl_AppendResult(interp, "restore failed: source database busy",
3191dc2c4915Sdrh (char*)0);
3192dc2c4915Sdrh rc = TCL_ERROR;
3193dc2c4915Sdrh }else{
3194dc2c4915Sdrh Tcl_AppendResult(interp, "restore failed: ",
3195dc2c4915Sdrh sqlite3_errmsg(pDb->db), (char*)0);
3196dc2c4915Sdrh rc = TCL_ERROR;
3197dc2c4915Sdrh }
3198dc2c4915Sdrh sqlite3_close(pSrc);
3199dc2c4915Sdrh break;
3200dc2c4915Sdrh }
3201dc2c4915Sdrh
320222fbcb8dSdrh /*
3203cb7d541dSdrh ** $db serialize ?DATABASE?
3204cb7d541dSdrh **
3205cb7d541dSdrh ** Return a serialization of a database.
3206cb7d541dSdrh */
3207cb7d541dSdrh case DB_SERIALIZE: {
32088d889afcSdrh #ifdef SQLITE_OMIT_DESERIALIZE
3209cb7d541dSdrh Tcl_AppendResult(interp, "MEMDB not available in this build",
3210cb7d541dSdrh (char*)0);
3211cb7d541dSdrh rc = TCL_ERROR;
3212cb7d541dSdrh #else
3213cb7d541dSdrh const char *zSchema = objc>=3 ? Tcl_GetString(objv[2]) : "main";
3214cb7d541dSdrh sqlite3_int64 sz = 0;
3215cb7d541dSdrh unsigned char *pData;
3216cb7d541dSdrh if( objc!=2 && objc!=3 ){
3217cb7d541dSdrh Tcl_WrongNumArgs(interp, 2, objv, "?DATABASE?");
3218cb7d541dSdrh rc = TCL_ERROR;
3219cb7d541dSdrh }else{
3220cb7d541dSdrh int needFree;
3221cb7d541dSdrh pData = sqlite3_serialize(pDb->db, zSchema, &sz, SQLITE_SERIALIZE_NOCOPY);
3222cb7d541dSdrh if( pData ){
3223cb7d541dSdrh needFree = 0;
3224cb7d541dSdrh }else{
3225cb7d541dSdrh pData = sqlite3_serialize(pDb->db, zSchema, &sz, 0);
3226cb7d541dSdrh needFree = 1;
3227cb7d541dSdrh }
3228cb7d541dSdrh Tcl_SetObjResult(interp, Tcl_NewByteArrayObj(pData,sz));
3229cb7d541dSdrh if( needFree ) sqlite3_free(pData);
3230cb7d541dSdrh }
3231cb7d541dSdrh #endif
3232cb7d541dSdrh break;
3233cb7d541dSdrh }
3234cb7d541dSdrh
3235cb7d541dSdrh /*
3236c456a76fSdan ** $db status (step|sort|autoindex|vmstep)
3237d1d38488Sdrh **
3238d1d38488Sdrh ** Display SQLITE_STMTSTATUS_FULLSCAN_STEP or
3239d1d38488Sdrh ** SQLITE_STMTSTATUS_SORT for the most recent eval.
3240d1d38488Sdrh */
3241d1d38488Sdrh case DB_STATUS: {
3242d1d38488Sdrh int v;
3243d1d38488Sdrh const char *zOp;
3244d1d38488Sdrh if( objc!=3 ){
32451c320a43Sdrh Tcl_WrongNumArgs(interp, 2, objv, "(step|sort|autoindex)");
3246d1d38488Sdrh return TCL_ERROR;
3247d1d38488Sdrh }
3248d1d38488Sdrh zOp = Tcl_GetString(objv[2]);
3249d1d38488Sdrh if( strcmp(zOp, "step")==0 ){
3250d1d38488Sdrh v = pDb->nStep;
3251d1d38488Sdrh }else if( strcmp(zOp, "sort")==0 ){
3252d1d38488Sdrh v = pDb->nSort;
32533c379b01Sdrh }else if( strcmp(zOp, "autoindex")==0 ){
32543c379b01Sdrh v = pDb->nIndex;
3255c456a76fSdan }else if( strcmp(zOp, "vmstep")==0 ){
3256c456a76fSdan v = pDb->nVMStep;
3257d1d38488Sdrh }else{
32583c379b01Sdrh Tcl_AppendResult(interp,
3259c456a76fSdan "bad argument: should be autoindex, step, sort or vmstep",
3260d1d38488Sdrh (char*)0);
3261d1d38488Sdrh return TCL_ERROR;
3262d1d38488Sdrh }
3263d1d38488Sdrh Tcl_SetObjResult(interp, Tcl_NewIntObj(v));
3264d1d38488Sdrh break;
3265d1d38488Sdrh }
3266d1d38488Sdrh
3267d1d38488Sdrh /*
3268bec3f402Sdrh ** $db timeout MILLESECONDS
3269bec3f402Sdrh **
3270bec3f402Sdrh ** Delay for the number of milliseconds specified when a file is locked.
3271bec3f402Sdrh */
32726d31316cSdrh case DB_TIMEOUT: {
3273bec3f402Sdrh int ms;
32746d31316cSdrh if( objc!=3 ){
32756d31316cSdrh Tcl_WrongNumArgs(interp, 2, objv, "MILLISECONDS");
3276bec3f402Sdrh return TCL_ERROR;
327775897234Sdrh }
32786d31316cSdrh if( Tcl_GetIntFromObj(interp, objv[2], &ms) ) return TCL_ERROR;
32796f8a503dSdanielk1977 sqlite3_busy_timeout(pDb->db, ms);
32806d31316cSdrh break;
328175897234Sdrh }
3282b5a20d3cSdrh
32830f14e2ebSdrh /*
32840f14e2ebSdrh ** $db total_changes
32850f14e2ebSdrh **
32860f14e2ebSdrh ** Return the number of rows that were modified, inserted, or deleted
32870f14e2ebSdrh ** since the database handle was created.
32880f14e2ebSdrh */
32890f14e2ebSdrh case DB_TOTAL_CHANGES: {
32900f14e2ebSdrh Tcl_Obj *pResult;
32910f14e2ebSdrh if( objc!=2 ){
32920f14e2ebSdrh Tcl_WrongNumArgs(interp, 2, objv, "");
32930f14e2ebSdrh return TCL_ERROR;
32940f14e2ebSdrh }
32950f14e2ebSdrh pResult = Tcl_GetObjResult(interp);
32962c718873Sdan Tcl_SetWideIntObj(pResult, sqlite3_total_changes64(pDb->db));
32970f14e2ebSdrh break;
32980f14e2ebSdrh }
32990f14e2ebSdrh
3300b5a20d3cSdrh /* $db trace ?CALLBACK?
3301b5a20d3cSdrh **
3302b5a20d3cSdrh ** Make arrangements to invoke the CALLBACK routine for each SQL statement
3303b5a20d3cSdrh ** that is executed. The text of the SQL is appended to CALLBACK before
3304b5a20d3cSdrh ** it is executed.
3305b5a20d3cSdrh */
3306b5a20d3cSdrh case DB_TRACE: {
3307b5a20d3cSdrh if( objc>3 ){
3308b5a20d3cSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?");
3309b97759edSdrh return TCL_ERROR;
3310b5a20d3cSdrh }else if( objc==2 ){
3311b5a20d3cSdrh if( pDb->zTrace ){
3312a198f2b5Sdrh Tcl_AppendResult(interp, pDb->zTrace, (char*)0);
3313b5a20d3cSdrh }
3314b5a20d3cSdrh }else{
3315b5a20d3cSdrh char *zTrace;
3316b5a20d3cSdrh int len;
3317b5a20d3cSdrh if( pDb->zTrace ){
3318b5a20d3cSdrh Tcl_Free(pDb->zTrace);
3319b5a20d3cSdrh }
3320b5a20d3cSdrh zTrace = Tcl_GetStringFromObj(objv[2], &len);
3321b5a20d3cSdrh if( zTrace && len>0 ){
3322b5a20d3cSdrh pDb->zTrace = Tcl_Alloc( len + 1 );
33235bb3eb9bSdrh memcpy(pDb->zTrace, zTrace, len+1);
3324b5a20d3cSdrh }else{
3325b5a20d3cSdrh pDb->zTrace = 0;
3326b5a20d3cSdrh }
33272eb22af0Sdrh #if !defined(SQLITE_OMIT_TRACE) && !defined(SQLITE_OMIT_FLOATING_POINT) && \
33282eb22af0Sdrh !defined(SQLITE_OMIT_DEPRECATED)
3329b5a20d3cSdrh if( pDb->zTrace ){
3330b5a20d3cSdrh pDb->interp = interp;
33316f8a503dSdanielk1977 sqlite3_trace(pDb->db, DbTraceHandler, pDb);
3332b5a20d3cSdrh }else{
33336f8a503dSdanielk1977 sqlite3_trace(pDb->db, 0, 0);
3334b5a20d3cSdrh }
333519e2d37fSdrh #endif
3336b5a20d3cSdrh }
3337b5a20d3cSdrh break;
3338b5a20d3cSdrh }
3339b5a20d3cSdrh
3340b56660f5Smistachkin /* $db trace_v2 ?CALLBACK? ?MASK?
3341b56660f5Smistachkin **
3342b56660f5Smistachkin ** Make arrangements to invoke the CALLBACK routine for each trace event
3343b56660f5Smistachkin ** matching the mask that is generated. The parameters are appended to
3344b56660f5Smistachkin ** CALLBACK before it is executed.
3345b56660f5Smistachkin */
3346b56660f5Smistachkin case DB_TRACE_V2: {
3347b56660f5Smistachkin if( objc>4 ){
3348b56660f5Smistachkin Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK? ?MASK?");
3349b56660f5Smistachkin return TCL_ERROR;
3350b56660f5Smistachkin }else if( objc==2 ){
3351b56660f5Smistachkin if( pDb->zTraceV2 ){
3352b56660f5Smistachkin Tcl_AppendResult(interp, pDb->zTraceV2, (char*)0);
3353b56660f5Smistachkin }
3354b56660f5Smistachkin }else{
3355b56660f5Smistachkin char *zTraceV2;
3356b56660f5Smistachkin int len;
3357b52dcd89Smistachkin Tcl_WideInt wMask = 0;
3358b56660f5Smistachkin if( objc==4 ){
3359b52dcd89Smistachkin static const char *TTYPE_strs[] = {
3360b52dcd89Smistachkin "statement", "profile", "row", "close", 0
3361b52dcd89Smistachkin };
3362b52dcd89Smistachkin enum TTYPE_enum {
3363b52dcd89Smistachkin TTYPE_STMT, TTYPE_PROFILE, TTYPE_ROW, TTYPE_CLOSE
3364b52dcd89Smistachkin };
3365b52dcd89Smistachkin int i;
3366b52dcd89Smistachkin if( TCL_OK!=Tcl_ListObjLength(interp, objv[3], &len) ){
3367b52dcd89Smistachkin return TCL_ERROR;
3368b52dcd89Smistachkin }
3369b52dcd89Smistachkin for(i=0; i<len; i++){
3370b52dcd89Smistachkin Tcl_Obj *pObj;
3371b52dcd89Smistachkin int ttype;
3372b52dcd89Smistachkin if( TCL_OK!=Tcl_ListObjIndex(interp, objv[3], i, &pObj) ){
3373b52dcd89Smistachkin return TCL_ERROR;
3374b52dcd89Smistachkin }
3375b52dcd89Smistachkin if( Tcl_GetIndexFromObj(interp, pObj, TTYPE_strs, "trace type",
3376b52dcd89Smistachkin 0, &ttype)!=TCL_OK ){
3377b52dcd89Smistachkin Tcl_WideInt wType;
3378b52dcd89Smistachkin Tcl_Obj *pError = Tcl_DuplicateObj(Tcl_GetObjResult(interp));
3379b52dcd89Smistachkin Tcl_IncrRefCount(pError);
3380b52dcd89Smistachkin if( TCL_OK==Tcl_GetWideIntFromObj(interp, pObj, &wType) ){
3381b52dcd89Smistachkin Tcl_DecrRefCount(pError);
3382b52dcd89Smistachkin wMask |= wType;
3383b52dcd89Smistachkin }else{
3384b52dcd89Smistachkin Tcl_SetObjResult(interp, pError);
3385b52dcd89Smistachkin Tcl_DecrRefCount(pError);
3386b56660f5Smistachkin return TCL_ERROR;
3387b56660f5Smistachkin }
3388b56660f5Smistachkin }else{
3389b52dcd89Smistachkin switch( (enum TTYPE_enum)ttype ){
3390b52dcd89Smistachkin case TTYPE_STMT: wMask |= SQLITE_TRACE_STMT; break;
3391b52dcd89Smistachkin case TTYPE_PROFILE: wMask |= SQLITE_TRACE_PROFILE; break;
3392b52dcd89Smistachkin case TTYPE_ROW: wMask |= SQLITE_TRACE_ROW; break;
3393b52dcd89Smistachkin case TTYPE_CLOSE: wMask |= SQLITE_TRACE_CLOSE; break;
3394b52dcd89Smistachkin }
3395b52dcd89Smistachkin }
3396b52dcd89Smistachkin }
3397b52dcd89Smistachkin }else{
3398b52dcd89Smistachkin wMask = SQLITE_TRACE_STMT; /* use the "legacy" default */
3399b56660f5Smistachkin }
3400b56660f5Smistachkin if( pDb->zTraceV2 ){
3401b56660f5Smistachkin Tcl_Free(pDb->zTraceV2);
3402b56660f5Smistachkin }
3403b56660f5Smistachkin zTraceV2 = Tcl_GetStringFromObj(objv[2], &len);
3404b56660f5Smistachkin if( zTraceV2 && len>0 ){
3405b56660f5Smistachkin pDb->zTraceV2 = Tcl_Alloc( len + 1 );
3406b56660f5Smistachkin memcpy(pDb->zTraceV2, zTraceV2, len+1);
3407b56660f5Smistachkin }else{
3408b56660f5Smistachkin pDb->zTraceV2 = 0;
3409b56660f5Smistachkin }
3410b56660f5Smistachkin #if !defined(SQLITE_OMIT_TRACE) && !defined(SQLITE_OMIT_FLOATING_POINT)
3411b56660f5Smistachkin if( pDb->zTraceV2 ){
3412b56660f5Smistachkin pDb->interp = interp;
3413b56660f5Smistachkin sqlite3_trace_v2(pDb->db, (unsigned)wMask, DbTraceV2Handler, pDb);
3414b56660f5Smistachkin }else{
3415b56660f5Smistachkin sqlite3_trace_v2(pDb->db, 0, 0, 0);
3416b56660f5Smistachkin }
3417b56660f5Smistachkin #endif
3418b56660f5Smistachkin }
3419b56660f5Smistachkin break;
3420b56660f5Smistachkin }
3421b56660f5Smistachkin
34223d21423cSdrh /* $db transaction [-deferred|-immediate|-exclusive] SCRIPT
34233d21423cSdrh **
34243d21423cSdrh ** Start a new transaction (if we are not already in the midst of a
34253d21423cSdrh ** transaction) and execute the TCL script SCRIPT. After SCRIPT
34263d21423cSdrh ** completes, either commit the transaction or roll it back if SCRIPT
34273d21423cSdrh ** throws an exception. Or if no new transation was started, do nothing.
34283d21423cSdrh ** pass the exception on up the stack.
34293d21423cSdrh **
34303d21423cSdrh ** This command was inspired by Dave Thomas's talk on Ruby at the
34313d21423cSdrh ** 2005 O'Reilly Open Source Convention (OSCON).
34323d21423cSdrh */
34333d21423cSdrh case DB_TRANSACTION: {
34343d21423cSdrh Tcl_Obj *pScript;
3435cd38d520Sdanielk1977 const char *zBegin = "SAVEPOINT _tcl_transaction";
34363d21423cSdrh if( objc!=3 && objc!=4 ){
34373d21423cSdrh Tcl_WrongNumArgs(interp, 2, objv, "[TYPE] SCRIPT");
34383d21423cSdrh return TCL_ERROR;
34393d21423cSdrh }
3440cd38d520Sdanielk1977
34414a4c11aaSdan if( pDb->nTransaction==0 && objc==4 ){
34423d21423cSdrh static const char *TTYPE_strs[] = {
3443ce604012Sdrh "deferred", "exclusive", "immediate", 0
34443d21423cSdrh };
34453d21423cSdrh enum TTYPE_enum {
34463d21423cSdrh TTYPE_DEFERRED, TTYPE_EXCLUSIVE, TTYPE_IMMEDIATE
34473d21423cSdrh };
34483d21423cSdrh int ttype;
3449b5555e7eSdrh if( Tcl_GetIndexFromObj(interp, objv[2], TTYPE_strs, "transaction type",
34503d21423cSdrh 0, &ttype) ){
34513d21423cSdrh return TCL_ERROR;
34523d21423cSdrh }
34533d21423cSdrh switch( (enum TTYPE_enum)ttype ){
34543d21423cSdrh case TTYPE_DEFERRED: /* no-op */; break;
34553d21423cSdrh case TTYPE_EXCLUSIVE: zBegin = "BEGIN EXCLUSIVE"; break;
34563d21423cSdrh case TTYPE_IMMEDIATE: zBegin = "BEGIN IMMEDIATE"; break;
34573d21423cSdrh }
34583d21423cSdrh }
3459cd38d520Sdanielk1977 pScript = objv[objc-1];
3460cd38d520Sdanielk1977
34614a4c11aaSdan /* Run the SQLite BEGIN command to open a transaction or savepoint. */
34621f1549f8Sdrh pDb->disableAuth++;
3463cd38d520Sdanielk1977 rc = sqlite3_exec(pDb->db, zBegin, 0, 0, 0);
34641f1549f8Sdrh pDb->disableAuth--;
3465cd38d520Sdanielk1977 if( rc!=SQLITE_OK ){
3466a198f2b5Sdrh Tcl_AppendResult(interp, sqlite3_errmsg(pDb->db), (char*)0);
3467cd38d520Sdanielk1977 return TCL_ERROR;
34683d21423cSdrh }
3469cd38d520Sdanielk1977 pDb->nTransaction++;
3470cd38d520Sdanielk1977
34714a4c11aaSdan /* If using NRE, schedule a callback to invoke the script pScript, then
34724a4c11aaSdan ** a second callback to commit (or rollback) the transaction or savepoint
34734a4c11aaSdan ** opened above. If not using NRE, evaluate the script directly, then
34744a4c11aaSdan ** call function DbTransPostCmd() to commit (or rollback) the transaction
34754a4c11aaSdan ** or savepoint. */
3476bea28c73Sdan addDatabaseRef(pDb); /* DbTransPostCmd() calls delDatabaseRef() */
34774a4c11aaSdan if( DbUseNre() ){
34784a4c11aaSdan Tcl_NRAddCallback(interp, DbTransPostCmd, cd, 0, 0, 0);
3479a47941feSdrh (void)Tcl_NREvalObj(interp, pScript, 0);
34803d21423cSdrh }else{
34814a4c11aaSdan rc = DbTransPostCmd(&cd, interp, Tcl_EvalObjEx(interp, pScript, 0));
34823d21423cSdrh }
34833d21423cSdrh break;
34843d21423cSdrh }
34853d21423cSdrh
348694eb6a14Sdanielk1977 /*
3487404ca075Sdanielk1977 ** $db unlock_notify ?script?
3488404ca075Sdanielk1977 */
3489404ca075Sdanielk1977 case DB_UNLOCK_NOTIFY: {
3490404ca075Sdanielk1977 #ifndef SQLITE_ENABLE_UNLOCK_NOTIFY
3491a198f2b5Sdrh Tcl_AppendResult(interp, "unlock_notify not available in this build",
3492a198f2b5Sdrh (char*)0);
3493404ca075Sdanielk1977 rc = TCL_ERROR;
3494404ca075Sdanielk1977 #else
3495404ca075Sdanielk1977 if( objc!=2 && objc!=3 ){
3496404ca075Sdanielk1977 Tcl_WrongNumArgs(interp, 2, objv, "?SCRIPT?");
3497404ca075Sdanielk1977 rc = TCL_ERROR;
3498404ca075Sdanielk1977 }else{
3499404ca075Sdanielk1977 void (*xNotify)(void **, int) = 0;
3500404ca075Sdanielk1977 void *pNotifyArg = 0;
3501404ca075Sdanielk1977
3502404ca075Sdanielk1977 if( pDb->pUnlockNotify ){
3503404ca075Sdanielk1977 Tcl_DecrRefCount(pDb->pUnlockNotify);
3504404ca075Sdanielk1977 pDb->pUnlockNotify = 0;
3505404ca075Sdanielk1977 }
3506404ca075Sdanielk1977
3507404ca075Sdanielk1977 if( objc==3 ){
3508404ca075Sdanielk1977 xNotify = DbUnlockNotify;
3509404ca075Sdanielk1977 pNotifyArg = (void *)pDb;
3510404ca075Sdanielk1977 pDb->pUnlockNotify = objv[2];
3511404ca075Sdanielk1977 Tcl_IncrRefCount(pDb->pUnlockNotify);
3512404ca075Sdanielk1977 }
3513404ca075Sdanielk1977
3514404ca075Sdanielk1977 if( sqlite3_unlock_notify(pDb->db, xNotify, pNotifyArg) ){
3515a198f2b5Sdrh Tcl_AppendResult(interp, sqlite3_errmsg(pDb->db), (char*)0);
3516404ca075Sdanielk1977 rc = TCL_ERROR;
3517404ca075Sdanielk1977 }
3518404ca075Sdanielk1977 }
3519404ca075Sdanielk1977 #endif
3520404ca075Sdanielk1977 break;
3521404ca075Sdanielk1977 }
3522404ca075Sdanielk1977
3523304637c0Sdrh /*
3524304637c0Sdrh ** $db preupdate_hook count
3525304637c0Sdrh ** $db preupdate_hook hook ?SCRIPT?
3526304637c0Sdrh ** $db preupdate_hook new INDEX
3527304637c0Sdrh ** $db preupdate_hook old INDEX
3528304637c0Sdrh */
352946c47d46Sdan case DB_PREUPDATE: {
35309b1c62d4Sdrh #ifndef SQLITE_ENABLE_PREUPDATE_HOOK
35312b519ab0Sdan Tcl_AppendResult(interp, "preupdate_hook was omitted at compile-time",
35322b519ab0Sdan (char*)0);
35339b1c62d4Sdrh rc = TCL_ERROR;
35349b1c62d4Sdrh #else
35351e7a2d43Sdan static const char *azSub[] = {"count", "depth", "hook", "new", "old", 0};
353646c47d46Sdan enum DbPreupdateSubCmd {
35371e7a2d43Sdan PRE_COUNT, PRE_DEPTH, PRE_HOOK, PRE_NEW, PRE_OLD
353846c47d46Sdan };
353946c47d46Sdan int iSub;
354046c47d46Sdan
354146c47d46Sdan if( objc<3 ){
354246c47d46Sdan Tcl_WrongNumArgs(interp, 2, objv, "SUB-COMMAND ?ARGS?");
354346c47d46Sdan }
354446c47d46Sdan if( Tcl_GetIndexFromObj(interp, objv[2], azSub, "sub-command", 0, &iSub) ){
354546c47d46Sdan return TCL_ERROR;
354646c47d46Sdan }
354746c47d46Sdan
354846c47d46Sdan switch( (enum DbPreupdateSubCmd)iSub ){
354946c47d46Sdan case PRE_COUNT: {
355046c47d46Sdan int nCol = sqlite3_preupdate_count(pDb->db);
355146c47d46Sdan Tcl_SetObjResult(interp, Tcl_NewIntObj(nCol));
355246c47d46Sdan break;
355346c47d46Sdan }
355446c47d46Sdan
355546c47d46Sdan case PRE_HOOK: {
355646c47d46Sdan if( objc>4 ){
355746c47d46Sdan Tcl_WrongNumArgs(interp, 2, objv, "hook ?SCRIPT?");
355846c47d46Sdan return TCL_ERROR;
355946c47d46Sdan }
356046c47d46Sdan DbHookCmd(interp, pDb, (objc==4 ? objv[3] : 0), &pDb->pPreUpdateHook);
356146c47d46Sdan break;
356246c47d46Sdan }
356346c47d46Sdan
35641e7a2d43Sdan case PRE_DEPTH: {
35651e7a2d43Sdan Tcl_Obj *pRet;
35661e7a2d43Sdan if( objc!=3 ){
35671e7a2d43Sdan Tcl_WrongNumArgs(interp, 3, objv, "");
35681e7a2d43Sdan return TCL_ERROR;
35691e7a2d43Sdan }
35701e7a2d43Sdan pRet = Tcl_NewIntObj(sqlite3_preupdate_depth(pDb->db));
35711e7a2d43Sdan Tcl_SetObjResult(interp, pRet);
35721e7a2d43Sdan break;
35731e7a2d43Sdan }
35741e7a2d43Sdan
357537db03bfSdan case PRE_NEW:
357646c47d46Sdan case PRE_OLD: {
357746c47d46Sdan int iIdx;
357837db03bfSdan sqlite3_value *pValue;
357946c47d46Sdan if( objc!=4 ){
358046c47d46Sdan Tcl_WrongNumArgs(interp, 3, objv, "INDEX");
358146c47d46Sdan return TCL_ERROR;
358246c47d46Sdan }
358346c47d46Sdan if( Tcl_GetIntFromObj(interp, objv[3], &iIdx) ){
358446c47d46Sdan return TCL_ERROR;
358546c47d46Sdan }
358646c47d46Sdan
358737db03bfSdan if( iSub==PRE_OLD ){
358846c47d46Sdan rc = sqlite3_preupdate_old(pDb->db, iIdx, &pValue);
358937db03bfSdan }else{
359037db03bfSdan assert( iSub==PRE_NEW );
359137db03bfSdan rc = sqlite3_preupdate_new(pDb->db, iIdx, &pValue);
359237db03bfSdan }
359337db03bfSdan
359446c47d46Sdan if( rc==SQLITE_OK ){
3595304637c0Sdrh Tcl_Obj *pObj;
3596304637c0Sdrh pObj = Tcl_NewStringObj((char*)sqlite3_value_text(pValue), -1);
359746c47d46Sdan Tcl_SetObjResult(interp, pObj);
359837db03bfSdan }else{
3599ea8f0a15Sdrh Tcl_AppendResult(interp, sqlite3_errmsg(pDb->db), (char*)0);
360046c47d46Sdan return TCL_ERROR;
360146c47d46Sdan }
360246c47d46Sdan }
360346c47d46Sdan }
36049b1c62d4Sdrh #endif /* SQLITE_ENABLE_PREUPDATE_HOOK */
360546c47d46Sdan break;
360646c47d46Sdan }
360746c47d46Sdan
3608404ca075Sdanielk1977 /*
3609833bf968Sdrh ** $db wal_hook ?script?
361094eb6a14Sdanielk1977 ** $db update_hook ?script?
361171fd80bfSdanielk1977 ** $db rollback_hook ?script?
361294eb6a14Sdanielk1977 */
3613833bf968Sdrh case DB_WAL_HOOK:
361471fd80bfSdanielk1977 case DB_UPDATE_HOOK:
36156566ebe1Sdan case DB_ROLLBACK_HOOK: {
361671fd80bfSdanielk1977 /* set ppHook to point at pUpdateHook or pRollbackHook, depending on
361771fd80bfSdanielk1977 ** whether [$db update_hook] or [$db rollback_hook] was invoked.
361871fd80bfSdanielk1977 */
3619c5896b5cSmistachkin Tcl_Obj **ppHook = 0;
362046c47d46Sdan if( choice==DB_WAL_HOOK ) ppHook = &pDb->pWalHook;
362146c47d46Sdan if( choice==DB_UPDATE_HOOK ) ppHook = &pDb->pUpdateHook;
362246c47d46Sdan if( choice==DB_ROLLBACK_HOOK ) ppHook = &pDb->pRollbackHook;
362346c47d46Sdan if( objc>3 ){
362494eb6a14Sdanielk1977 Tcl_WrongNumArgs(interp, 2, objv, "?SCRIPT?");
362594eb6a14Sdanielk1977 return TCL_ERROR;
362694eb6a14Sdanielk1977 }
362771fd80bfSdanielk1977
362846c47d46Sdan DbHookCmd(interp, pDb, (objc==3 ? objv[2] : 0), ppHook);
362994eb6a14Sdanielk1977 break;
363094eb6a14Sdanielk1977 }
363194eb6a14Sdanielk1977
36324397de57Sdanielk1977 /* $db version
36334397de57Sdanielk1977 **
36344397de57Sdanielk1977 ** Return the version string for this database.
36354397de57Sdanielk1977 */
36364397de57Sdanielk1977 case DB_VERSION: {
36371df6470dSdrh int i;
36381df6470dSdrh for(i=2; i<objc; i++){
36391df6470dSdrh const char *zArg = Tcl_GetString(objv[i]);
36401df6470dSdrh /* Optional arguments to $db version are used for testing purpose */
36411df6470dSdrh #ifdef SQLITE_TEST
36421df6470dSdrh /* $db version -use-legacy-prepare BOOLEAN
36431df6470dSdrh **
36441df6470dSdrh ** Turn the use of legacy sqlite3_prepare() on or off.
36451df6470dSdrh */
36461df6470dSdrh if( strcmp(zArg, "-use-legacy-prepare")==0 && i+1<objc ){
36471df6470dSdrh i++;
36481df6470dSdrh if( Tcl_GetBooleanFromObj(interp, objv[i], &pDb->bLegacyPrepare) ){
36491df6470dSdrh return TCL_ERROR;
36501df6470dSdrh }
36511df6470dSdrh }else
36521df6470dSdrh
36531df6470dSdrh /* $db version -last-stmt-ptr
36541df6470dSdrh **
36551df6470dSdrh ** Return a string which is a hex encoding of the pointer to the
36561df6470dSdrh ** most recent sqlite3_stmt in the statement cache.
36571df6470dSdrh */
36581df6470dSdrh if( strcmp(zArg, "-last-stmt-ptr")==0 ){
36591df6470dSdrh char zBuf[100];
36601df6470dSdrh sqlite3_snprintf(sizeof(zBuf), zBuf, "%p",
36611df6470dSdrh pDb->stmtList ? pDb->stmtList->pStmt: 0);
36621df6470dSdrh Tcl_SetResult(interp, zBuf, TCL_VOLATILE);
36631df6470dSdrh }else
36641df6470dSdrh #endif /* SQLITE_TEST */
36651df6470dSdrh {
36661df6470dSdrh Tcl_AppendResult(interp, "unknown argument: ", zArg, (char*)0);
36671df6470dSdrh return TCL_ERROR;
36681df6470dSdrh }
36691df6470dSdrh }
36701df6470dSdrh if( i==2 ){
36714397de57Sdanielk1977 Tcl_SetResult(interp, (char *)sqlite3_libversion(), TCL_STATIC);
36721df6470dSdrh }
36734397de57Sdanielk1977 break;
36744397de57Sdanielk1977 }
36754397de57Sdanielk1977
36761067fe11Stpoindex
36776d31316cSdrh } /* End of the SWITCH statement */
367822fbcb8dSdrh return rc;
367975897234Sdrh }
368075897234Sdrh
3681a2c8a95bSdrh #if SQLITE_TCL_NRE
3682a2c8a95bSdrh /*
3683a2c8a95bSdrh ** Adaptor that provides an objCmd interface to the NRE-enabled
3684a2c8a95bSdrh ** interface implementation.
3685a2c8a95bSdrh */
DbObjCmdAdaptor(void * cd,Tcl_Interp * interp,int objc,Tcl_Obj * const * objv)3686a121cc7cSmistachkin static int SQLITE_TCLAPI DbObjCmdAdaptor(
3687a2c8a95bSdrh void *cd,
3688a2c8a95bSdrh Tcl_Interp *interp,
3689a2c8a95bSdrh int objc,
3690a2c8a95bSdrh Tcl_Obj *const*objv
3691a2c8a95bSdrh ){
3692a2c8a95bSdrh return Tcl_NRCallObjProc(interp, DbObjCmd, cd, objc, objv);
3693a2c8a95bSdrh }
3694a2c8a95bSdrh #endif /* SQLITE_TCL_NRE */
3695a2c8a95bSdrh
369675897234Sdrh /*
36974dcac40eSdrh ** Issue the usage message when the "sqlite3" command arguments are
36984dcac40eSdrh ** incorrect.
36994dcac40eSdrh */
sqliteCmdUsage(Tcl_Interp * interp,Tcl_Obj * const * objv)37004dcac40eSdrh static int sqliteCmdUsage(
37014dcac40eSdrh Tcl_Interp *interp,
37024dcac40eSdrh Tcl_Obj *const*objv
37034dcac40eSdrh ){
37044dcac40eSdrh Tcl_WrongNumArgs(interp, 1, objv,
37054dcac40eSdrh "HANDLE ?FILENAME? ?-vfs VFSNAME? ?-readonly BOOLEAN? ?-create BOOLEAN?"
37060933aad7Sdrh " ?-nofollow BOOLEAN?"
37074dcac40eSdrh " ?-nomutex BOOLEAN? ?-fullmutex BOOLEAN? ?-uri BOOLEAN?"
37084dcac40eSdrh );
37094dcac40eSdrh return TCL_ERROR;
37104dcac40eSdrh }
37114dcac40eSdrh
37124dcac40eSdrh /*
37133570ad93Sdrh ** sqlite3 DBNAME FILENAME ?-vfs VFSNAME? ?-key KEY? ?-readonly BOOLEAN?
37149a6284c1Sdanielk1977 ** ?-create BOOLEAN? ?-nomutex BOOLEAN?
37150933aad7Sdrh ** ?-nofollow BOOLEAN?
371675897234Sdrh **
371775897234Sdrh ** This is the main Tcl command. When the "sqlite" Tcl command is
371875897234Sdrh ** invoked, this routine runs to process that command.
371975897234Sdrh **
372075897234Sdrh ** The first argument, DBNAME, is an arbitrary name for a new
372175897234Sdrh ** database connection. This command creates a new command named
372275897234Sdrh ** DBNAME that is used to control that connection. The database
372375897234Sdrh ** connection is deleted when the DBNAME command is deleted.
372475897234Sdrh **
37253570ad93Sdrh ** The second argument is the name of the database file.
3726fbc3eab8Sdrh **
372775897234Sdrh */
DbMain(void * cd,Tcl_Interp * interp,int objc,Tcl_Obj * const * objv)37287617e4a8Smistachkin static int SQLITE_TCLAPI DbMain(
37297617e4a8Smistachkin void *cd,
37307617e4a8Smistachkin Tcl_Interp *interp,
37317617e4a8Smistachkin int objc,
37327617e4a8Smistachkin Tcl_Obj *const*objv
37337617e4a8Smistachkin ){
3734bec3f402Sdrh SqliteDb *p;
373522fbcb8dSdrh const char *zArg;
373675897234Sdrh char *zErrMsg;
37373570ad93Sdrh int i;
37384dcac40eSdrh const char *zFile = 0;
37393570ad93Sdrh const char *zVfs = 0;
3740d9da78a2Sdrh int flags;
3741a8e41ecaSmistachkin int bTranslateFileName = 1;
3742882e8e4dSdrh Tcl_DString translatedFilename;
3743540ebf82Smistachkin int rc;
3744d9da78a2Sdrh
3745d9da78a2Sdrh /* In normal use, each TCL interpreter runs in a single thread. So
37464dcac40eSdrh ** by default, we can turn off mutexing on SQLite database connections.
3747d9da78a2Sdrh ** However, for testing purposes it is useful to have mutexes turned
3748d9da78a2Sdrh ** on. So, by default, mutexes default off. But if compiled with
3749d9da78a2Sdrh ** SQLITE_TCL_DEFAULT_FULLMUTEX then mutexes default on.
3750d9da78a2Sdrh */
3751d9da78a2Sdrh #ifdef SQLITE_TCL_DEFAULT_FULLMUTEX
3752d9da78a2Sdrh flags = SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE | SQLITE_OPEN_FULLMUTEX;
3753d9da78a2Sdrh #else
3754d9da78a2Sdrh flags = SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE | SQLITE_OPEN_NOMUTEX;
3755d9da78a2Sdrh #endif
3756d9da78a2Sdrh
3757c6727c83Sdrh if( objc==1 ) return sqliteCmdUsage(interp, objv);
375822fbcb8dSdrh if( objc==2 ){
375922fbcb8dSdrh zArg = Tcl_GetStringFromObj(objv[1], 0);
376022fbcb8dSdrh if( strcmp(zArg,"-version")==0 ){
3761a198f2b5Sdrh Tcl_AppendResult(interp,sqlite3_libversion(), (char*)0);
3762647cb0e1Sdrh return TCL_OK;
3763647cb0e1Sdrh }
376472bf6a3eSdrh if( strcmp(zArg,"-sourceid")==0 ){
376572bf6a3eSdrh Tcl_AppendResult(interp,sqlite3_sourceid(), (char*)0);
376672bf6a3eSdrh return TCL_OK;
376772bf6a3eSdrh }
37689eb9e26bSdrh if( strcmp(zArg,"-has-codec")==0 ){
3769a198f2b5Sdrh Tcl_AppendResult(interp,"0",(char*)0);
377022fbcb8dSdrh return TCL_OK;
377122fbcb8dSdrh }
37724dcac40eSdrh if( zArg[0]=='-' ) return sqliteCmdUsage(interp, objv);
3773fbc3eab8Sdrh }
37744dcac40eSdrh for(i=2; i<objc; i++){
37753570ad93Sdrh zArg = Tcl_GetString(objv[i]);
37764dcac40eSdrh if( zArg[0]!='-' ){
37774dcac40eSdrh if( zFile!=0 ) return sqliteCmdUsage(interp, objv);
37784dcac40eSdrh zFile = zArg;
37794dcac40eSdrh continue;
37804dcac40eSdrh }
37814dcac40eSdrh if( i==objc-1 ) return sqliteCmdUsage(interp, objv);
37824dcac40eSdrh i++;
378322fbcb8dSdrh if( strcmp(zArg,"-key")==0 ){
3784b48c0d59Sdrh /* no-op */
37853570ad93Sdrh }else if( strcmp(zArg, "-vfs")==0 ){
37864dcac40eSdrh zVfs = Tcl_GetString(objv[i]);
37873570ad93Sdrh }else if( strcmp(zArg, "-readonly")==0 ){
37883570ad93Sdrh int b;
37894dcac40eSdrh if( Tcl_GetBooleanFromObj(interp, objv[i], &b) ) return TCL_ERROR;
37903570ad93Sdrh if( b ){
379133f4e02aSdrh flags &= ~(SQLITE_OPEN_READWRITE|SQLITE_OPEN_CREATE);
37923570ad93Sdrh flags |= SQLITE_OPEN_READONLY;
37933570ad93Sdrh }else{
37943570ad93Sdrh flags &= ~SQLITE_OPEN_READONLY;
37953570ad93Sdrh flags |= SQLITE_OPEN_READWRITE;
37963570ad93Sdrh }
37973570ad93Sdrh }else if( strcmp(zArg, "-create")==0 ){
37983570ad93Sdrh int b;
37994dcac40eSdrh if( Tcl_GetBooleanFromObj(interp, objv[i], &b) ) return TCL_ERROR;
380033f4e02aSdrh if( b && (flags & SQLITE_OPEN_READONLY)==0 ){
38013570ad93Sdrh flags |= SQLITE_OPEN_CREATE;
38023570ad93Sdrh }else{
38033570ad93Sdrh flags &= ~SQLITE_OPEN_CREATE;
38043570ad93Sdrh }
38050933aad7Sdrh }else if( strcmp(zArg, "-nofollow")==0 ){
38060933aad7Sdrh int b;
38070933aad7Sdrh if( Tcl_GetBooleanFromObj(interp, objv[i], &b) ) return TCL_ERROR;
38080933aad7Sdrh if( b ){
38090933aad7Sdrh flags |= SQLITE_OPEN_NOFOLLOW;
38100933aad7Sdrh }else{
38110933aad7Sdrh flags &= ~SQLITE_OPEN_NOFOLLOW;
38120933aad7Sdrh }
38139a6284c1Sdanielk1977 }else if( strcmp(zArg, "-nomutex")==0 ){
38149a6284c1Sdanielk1977 int b;
38154dcac40eSdrh if( Tcl_GetBooleanFromObj(interp, objv[i], &b) ) return TCL_ERROR;
38169a6284c1Sdanielk1977 if( b ){
38179a6284c1Sdanielk1977 flags |= SQLITE_OPEN_NOMUTEX;
3818039963adSdrh flags &= ~SQLITE_OPEN_FULLMUTEX;
38199a6284c1Sdanielk1977 }else{
38209a6284c1Sdanielk1977 flags &= ~SQLITE_OPEN_NOMUTEX;
38219a6284c1Sdanielk1977 }
3822039963adSdrh }else if( strcmp(zArg, "-fullmutex")==0 ){
3823039963adSdrh int b;
38244dcac40eSdrh if( Tcl_GetBooleanFromObj(interp, objv[i], &b) ) return TCL_ERROR;
3825039963adSdrh if( b ){
3826039963adSdrh flags |= SQLITE_OPEN_FULLMUTEX;
3827039963adSdrh flags &= ~SQLITE_OPEN_NOMUTEX;
3828039963adSdrh }else{
3829039963adSdrh flags &= ~SQLITE_OPEN_FULLMUTEX;
3830039963adSdrh }
3831f12b3f60Sdrh }else if( strcmp(zArg, "-uri")==0 ){
3832f12b3f60Sdrh int b;
38334dcac40eSdrh if( Tcl_GetBooleanFromObj(interp, objv[i], &b) ) return TCL_ERROR;
3834f12b3f60Sdrh if( b ){
3835f12b3f60Sdrh flags |= SQLITE_OPEN_URI;
3836f12b3f60Sdrh }else{
3837f12b3f60Sdrh flags &= ~SQLITE_OPEN_URI;
3838f12b3f60Sdrh }
3839a8e41ecaSmistachkin }else if( strcmp(zArg, "-translatefilename")==0 ){
3840a8e41ecaSmistachkin if( Tcl_GetBooleanFromObj(interp, objv[i], &bTranslateFileName) ){
3841a8e41ecaSmistachkin return TCL_ERROR;
3842a8e41ecaSmistachkin }
38433570ad93Sdrh }else{
38443570ad93Sdrh Tcl_AppendResult(interp, "unknown option: ", zArg, (char*)0);
38453570ad93Sdrh return TCL_ERROR;
384622fbcb8dSdrh }
384722fbcb8dSdrh }
384875897234Sdrh zErrMsg = 0;
38494cdc9e84Sdrh p = (SqliteDb*)Tcl_Alloc( sizeof(*p) );
3850bec3f402Sdrh memset(p, 0, sizeof(*p));
38513ec8665eSdrh if( zFile==0 ) zFile = "";
3852a8e41ecaSmistachkin if( bTranslateFileName ){
3853882e8e4dSdrh zFile = Tcl_TranslateFileName(interp, zFile, &translatedFilename);
3854a8e41ecaSmistachkin }
3855540ebf82Smistachkin rc = sqlite3_open_v2(zFile, &p->db, flags, zVfs);
3856a8e41ecaSmistachkin if( bTranslateFileName ){
3857882e8e4dSdrh Tcl_DStringFree(&translatedFilename);
3858a8e41ecaSmistachkin }
3859540ebf82Smistachkin if( p->db ){
386080290863Sdanielk1977 if( SQLITE_OK!=sqlite3_errcode(p->db) ){
38619404d50eSdrh zErrMsg = sqlite3_mprintf("%s", sqlite3_errmsg(p->db));
386280290863Sdanielk1977 sqlite3_close(p->db);
386380290863Sdanielk1977 p->db = 0;
386480290863Sdanielk1977 }
3865540ebf82Smistachkin }else{
38665dac8432Smistachkin zErrMsg = sqlite3_mprintf("%s", sqlite3_errstr(rc));
3867540ebf82Smistachkin }
3868bec3f402Sdrh if( p->db==0 ){
386975897234Sdrh Tcl_SetResult(interp, zErrMsg, TCL_VOLATILE);
3870bec3f402Sdrh Tcl_Free((char*)p);
38719404d50eSdrh sqlite3_free(zErrMsg);
387275897234Sdrh return TCL_ERROR;
387375897234Sdrh }
3874fb7e7651Sdrh p->maxStmt = NUM_PREPARED_STMTS;
3875147ef394Sdrh p->openFlags = flags & SQLITE_OPEN_URI;
38765169bbc6Sdrh p->interp = interp;
387722fbcb8dSdrh zArg = Tcl_GetStringFromObj(objv[1], 0);
38784a4c11aaSdan if( DbUseNre() ){
3879a2c8a95bSdrh Tcl_NRCreateCommand(interp, zArg, DbObjCmdAdaptor, DbObjCmd,
3880a2c8a95bSdrh (char*)p, DbDeleteCmd);
38814a4c11aaSdan }else{
388222fbcb8dSdrh Tcl_CreateObjCommand(interp, zArg, DbObjCmd, (char*)p, DbDeleteCmd);
38834a4c11aaSdan }
3884bea28c73Sdan p->nRef = 1;
388575897234Sdrh return TCL_OK;
388675897234Sdrh }
388775897234Sdrh
388875897234Sdrh /*
388990ca9753Sdrh ** Provide a dummy Tcl_InitStubs if we are using this as a static
389090ca9753Sdrh ** library.
389190ca9753Sdrh */
389290ca9753Sdrh #ifndef USE_TCL_STUBS
389390ca9753Sdrh # undef Tcl_InitStubs
38940e85ccfcSdrh # define Tcl_InitStubs(a,b,c) TCL_VERSION
389590ca9753Sdrh #endif
389690ca9753Sdrh
389790ca9753Sdrh /*
389829bc4615Sdrh ** Make sure we have a PACKAGE_VERSION macro defined. This will be
389929bc4615Sdrh ** defined automatically by the TEA makefile. But other makefiles
390029bc4615Sdrh ** do not define it.
390129bc4615Sdrh */
390229bc4615Sdrh #ifndef PACKAGE_VERSION
390329bc4615Sdrh # define PACKAGE_VERSION SQLITE_VERSION
390429bc4615Sdrh #endif
390529bc4615Sdrh
390629bc4615Sdrh /*
390775897234Sdrh ** Initialize this module.
390875897234Sdrh **
390975897234Sdrh ** This Tcl module contains only a single new Tcl command named "sqlite".
391075897234Sdrh ** (Hence there is no namespace. There is no point in using a namespace
391175897234Sdrh ** if the extension only supplies one new name!) The "sqlite" command is
391275897234Sdrh ** used to open a new SQLite database. See the DbMain() routine above
391375897234Sdrh ** for additional information.
3914b652f432Sdrh **
3915b652f432Sdrh ** The EXTERN macros are required by TCL in order to work on windows.
391675897234Sdrh */
Sqlite3_Init(Tcl_Interp * interp)3917b652f432Sdrh EXTERN int Sqlite3_Init(Tcl_Interp *interp){
391827b2f053Smistachkin int rc = Tcl_InitStubs(interp, "8.4", 0) ? TCL_OK : TCL_ERROR;
39196dc8cbe0Sdrh if( rc==TCL_OK ){
3920ef4ac8f9Sdrh Tcl_CreateObjCommand(interp, "sqlite3", (Tcl_ObjCmdProc*)DbMain, 0, 0);
39211cca0d22Sdrh #ifndef SQLITE_3_SUFFIX_ONLY
39221cca0d22Sdrh /* The "sqlite" alias is undocumented. It is here only to support
39231cca0d22Sdrh ** legacy scripts. All new scripts should use only the "sqlite3"
39246dc8cbe0Sdrh ** command. */
392549766d6cSdrh Tcl_CreateObjCommand(interp, "sqlite", (Tcl_ObjCmdProc*)DbMain, 0, 0);
39264c0f1649Sdrh #endif
39276dc8cbe0Sdrh rc = Tcl_PkgProvide(interp, "sqlite3", PACKAGE_VERSION);
39286dc8cbe0Sdrh }
39296dc8cbe0Sdrh return rc;
393090ca9753Sdrh }
Tclsqlite3_Init(Tcl_Interp * interp)3931b652f432Sdrh EXTERN int Tclsqlite3_Init(Tcl_Interp *interp){ return Sqlite3_Init(interp); }
Sqlite3_Unload(Tcl_Interp * interp,int flags)3932b652f432Sdrh EXTERN int Sqlite3_Unload(Tcl_Interp *interp, int flags){ return TCL_OK; }
Tclsqlite3_Unload(Tcl_Interp * interp,int flags)3933b652f432Sdrh EXTERN int Tclsqlite3_Unload(Tcl_Interp *interp, int flags){ return TCL_OK; }
3934e2c3a659Sdrh
3935d878cab5Sdrh /* Because it accesses the file-system and uses persistent state, SQLite
3936e75a9eb9Sdrh ** is not considered appropriate for safe interpreters. Hence, we cause
3937e75a9eb9Sdrh ** the _SafeInit() interfaces return TCL_ERROR.
3938d878cab5Sdrh */
Sqlite3_SafeInit(Tcl_Interp * interp)3939e75a9eb9Sdrh EXTERN int Sqlite3_SafeInit(Tcl_Interp *interp){ return TCL_ERROR; }
Sqlite3_SafeUnload(Tcl_Interp * interp,int flags)3940e75a9eb9Sdrh EXTERN int Sqlite3_SafeUnload(Tcl_Interp *interp, int flags){return TCL_ERROR;}
3941e75a9eb9Sdrh
3942e75a9eb9Sdrh
394349766d6cSdrh
394449766d6cSdrh #ifndef SQLITE_3_SUFFIX_ONLY
Sqlite_Init(Tcl_Interp * interp)3945a3e63c4aSdan int Sqlite_Init(Tcl_Interp *interp){ return Sqlite3_Init(interp); }
Tclsqlite_Init(Tcl_Interp * interp)3946a3e63c4aSdan int Tclsqlite_Init(Tcl_Interp *interp){ return Sqlite3_Init(interp); }
Sqlite_Unload(Tcl_Interp * interp,int flags)3947a3e63c4aSdan int Sqlite_Unload(Tcl_Interp *interp, int flags){ return TCL_OK; }
Tclsqlite_Unload(Tcl_Interp * interp,int flags)3948a3e63c4aSdan int Tclsqlite_Unload(Tcl_Interp *interp, int flags){ return TCL_OK; }
394949766d6cSdrh #endif
395075897234Sdrh
3951c318f730Sdrh /*
395296a206faSdrh ** If the TCLSH macro is defined, add code to make a stand-alone program.
3953c318f730Sdrh */
395496a206faSdrh #if defined(TCLSH)
395557a0227fSdrh
395696a206faSdrh /* This is the main routine for an ordinary TCL shell. If there are
395796a206faSdrh ** are arguments, run the first argument as a script. Otherwise,
395896a206faSdrh ** read TCL commands from standard input
3959348784efSdrh */
tclsh_main_loop(void)39600ae479dfSdan static const char *tclsh_main_loop(void){
39610ae479dfSdan static const char zMainloop[] =
396296a206faSdrh "if {[llength $argv]>=1} {\n"
396396a206faSdrh "set argv0 [lindex $argv 0]\n"
396496a206faSdrh "set argv [lrange $argv 1 end]\n"
396596a206faSdrh "source $argv0\n"
396696a206faSdrh "} else {\n"
3967348784efSdrh "set line {}\n"
3968348784efSdrh "while {![eof stdin]} {\n"
3969348784efSdrh "if {$line!=\"\"} {\n"
3970348784efSdrh "puts -nonewline \"> \"\n"
3971348784efSdrh "} else {\n"
3972348784efSdrh "puts -nonewline \"% \"\n"
3973348784efSdrh "}\n"
3974348784efSdrh "flush stdout\n"
3975348784efSdrh "append line [gets stdin]\n"
3976348784efSdrh "if {[info complete $line]} {\n"
3977348784efSdrh "if {[catch {uplevel #0 $line} result]} {\n"
3978348784efSdrh "puts stderr \"Error: $result\"\n"
3979348784efSdrh "} elseif {$result!=\"\"} {\n"
3980348784efSdrh "puts $result\n"
3981348784efSdrh "}\n"
3982348784efSdrh "set line {}\n"
3983348784efSdrh "} else {\n"
3984348784efSdrh "append line \\n\n"
3985348784efSdrh "}\n"
3986348784efSdrh "}\n"
398796a206faSdrh "}\n"
3988348784efSdrh ;
39890ae479dfSdan return zMainloop;
39900ae479dfSdan }
39913e27c026Sdrh
39925e90794bSdrh #ifndef TCLSH_MAIN
39935e90794bSdrh # define TCLSH_MAIN main
39945e90794bSdrh #endif
TCLSH_MAIN(int argc,char ** argv)399569def7ffSmistachkin int SQLITE_CDECL TCLSH_MAIN(int argc, char **argv){
3996c1a60c51Sdan Tcl_Interp *interp;
399796a206faSdrh int i;
399896a206faSdrh const char *zScript = 0;
399996a206faSdrh char zArgc[32];
400096a206faSdrh #if defined(TCLSH_INIT_PROC)
400196a206faSdrh extern const char *TCLSH_INIT_PROC(Tcl_Interp*);
400296a206faSdrh #endif
4003c1a60c51Sdan
40041f28e070Smistachkin #if !defined(_WIN32_WCE)
40051e8487dbSmistachkin if( getenv("SQLITE_DEBUG_BREAK") ){
40061e8487dbSmistachkin if( isatty(0) && isatty(2) ){
40071f28e070Smistachkin fprintf(stderr,
40081f28e070Smistachkin "attach debugger to process %d and press any key to continue.\n",
40091f28e070Smistachkin GETPID());
40101f28e070Smistachkin fgetc(stdin);
40111e8487dbSmistachkin }else{
40121e8487dbSmistachkin #if defined(_WIN32) || defined(WIN32)
40131e8487dbSmistachkin DebugBreak();
40141e8487dbSmistachkin #elif defined(SIGTRAP)
40151e8487dbSmistachkin raise(SIGTRAP);
40161e8487dbSmistachkin #endif
40171e8487dbSmistachkin }
40181f28e070Smistachkin }
40191f28e070Smistachkin #endif
40201f28e070Smistachkin
4021c1a60c51Sdan /* Call sqlite3_shutdown() once before doing anything else. This is to
4022c1a60c51Sdan ** test that sqlite3_shutdown() can be safely called by a process before
4023c1a60c51Sdan ** sqlite3_initialize() is. */
4024c1a60c51Sdan sqlite3_shutdown();
4025c1a60c51Sdan
40260ae479dfSdan Tcl_FindExecutable(argv[0]);
40272953ba9eSmistachkin Tcl_SetSystemEncoding(NULL, "utf-8");
40280ae479dfSdan interp = Tcl_CreateInterp();
4029c318f730Sdrh Sqlite3_Init(interp);
403096a206faSdrh
403196a206faSdrh sqlite3_snprintf(sizeof(zArgc), zArgc, "%d", argc-1);
4032ad42c3a3Sshess Tcl_SetVar(interp,"argc", zArgc, TCL_GLOBAL_ONLY);
403396a206faSdrh Tcl_SetVar(interp,"argv0",argv[0],TCL_GLOBAL_ONLY);
4034348784efSdrh Tcl_SetVar(interp,"argv", "", TCL_GLOBAL_ONLY);
403596a206faSdrh for(i=1; i<argc; i++){
4036348784efSdrh Tcl_SetVar(interp, "argv", argv[i],
4037348784efSdrh TCL_GLOBAL_ONLY | TCL_LIST_ELEMENT | TCL_APPEND_VALUE);
4038348784efSdrh }
403996a206faSdrh #if defined(TCLSH_INIT_PROC)
404096a206faSdrh zScript = TCLSH_INIT_PROC(interp);
404196a206faSdrh #endif
404296a206faSdrh if( zScript==0 ){
404396a206faSdrh zScript = tclsh_main_loop();
404496a206faSdrh }
404596a206faSdrh if( Tcl_GlobalEval(interp, zScript)!=TCL_OK ){
40460de8c112Sdrh const char *zInfo = Tcl_GetVar(interp, "errorInfo", TCL_GLOBAL_ONLY);
4047a81c64a2Sdrh if( zInfo==0 ) zInfo = Tcl_GetStringResult(interp);
4048c61053b7Sdrh fprintf(stderr,"%s: %s\n", *argv, zInfo);
4049348784efSdrh return 1;
4050348784efSdrh }
4051348784efSdrh return 0;
4052348784efSdrh }
4053348784efSdrh #endif /* TCLSH */
4054