175897234Sdrh /* 2b19a2bc6Sdrh ** 2001 September 15 375897234Sdrh ** 4b19a2bc6Sdrh ** The author disclaims copyright to this source code. In place of 5b19a2bc6Sdrh ** a legal notice, here is a blessing: 675897234Sdrh ** 7b19a2bc6Sdrh ** May you do good and not evil. 8b19a2bc6Sdrh ** May you find forgiveness for yourself and forgive others. 9b19a2bc6Sdrh ** May you share freely, never taking more than you give. 1075897234Sdrh ** 1175897234Sdrh ************************************************************************* 1275897234Sdrh ** A TCL Interface to SQLite 1375897234Sdrh ** 14*aef0bf64Sdanielk1977 ** $Id: tclsqlite.c,v 1.142 2005/12/30 16:28:02 danielk1977 Exp $ 1575897234Sdrh */ 166d31316cSdrh #ifndef NO_TCL /* Omit this whole file if TCL is unavailable */ 176d31316cSdrh 1806b2718aSdrh #include "sqliteInt.h" 19895d7472Sdrh #include "hash.h" 2017a68934Sdrh #include "tcl.h" 2175897234Sdrh #include <stdlib.h> 2275897234Sdrh #include <string.h> 23ce927065Sdrh #include <assert.h> 24d1e4733dSdrh #include <ctype.h> 2575897234Sdrh 2629bc4615Sdrh /* 2729bc4615Sdrh * Windows needs to know which symbols to export. Unix does not. 2829bc4615Sdrh * BUILD_sqlite should be undefined for Unix. 2929bc4615Sdrh */ 3029bc4615Sdrh 3129bc4615Sdrh #ifdef BUILD_sqlite 3229bc4615Sdrh #undef TCL_STORAGE_CLASS 3329bc4615Sdrh #define TCL_STORAGE_CLASS DLLEXPORT 3429bc4615Sdrh #endif /* BUILD_sqlite */ 3529bc4615Sdrh 36a21c6b6fSdanielk1977 #define NUM_PREPARED_STMTS 10 37fb7e7651Sdrh #define MAX_PREPARED_STMTS 100 38fb7e7651Sdrh 3975897234Sdrh /* 4098808babSdrh ** If TCL uses UTF-8 and SQLite is configured to use iso8859, then we 4198808babSdrh ** have to do a translation when going between the two. Set the 4298808babSdrh ** UTF_TRANSLATION_NEEDED macro to indicate that we need to do 4398808babSdrh ** this translation. 4498808babSdrh */ 4598808babSdrh #if defined(TCL_UTF_MAX) && !defined(SQLITE_UTF8) 4698808babSdrh # define UTF_TRANSLATION_NEEDED 1 4798808babSdrh #endif 4898808babSdrh 4998808babSdrh /* 50cabb0819Sdrh ** New SQL functions can be created as TCL scripts. Each such function 51cabb0819Sdrh ** is described by an instance of the following structure. 52cabb0819Sdrh */ 53cabb0819Sdrh typedef struct SqlFunc SqlFunc; 54cabb0819Sdrh struct SqlFunc { 55cabb0819Sdrh Tcl_Interp *interp; /* The TCL interpret to execute the function */ 56d1e4733dSdrh Tcl_Obj *pScript; /* The Tcl_Obj representation of the script */ 57d1e4733dSdrh int useEvalObjv; /* True if it is safe to use Tcl_EvalObjv */ 58d1e4733dSdrh char *zName; /* Name of this function */ 59cabb0819Sdrh SqlFunc *pNext; /* Next function on the list of them all */ 60cabb0819Sdrh }; 61cabb0819Sdrh 62cabb0819Sdrh /* 630202b29eSdanielk1977 ** New collation sequences function can be created as TCL scripts. Each such 640202b29eSdanielk1977 ** function is described by an instance of the following structure. 650202b29eSdanielk1977 */ 660202b29eSdanielk1977 typedef struct SqlCollate SqlCollate; 670202b29eSdanielk1977 struct SqlCollate { 680202b29eSdanielk1977 Tcl_Interp *interp; /* The TCL interpret to execute the function */ 690202b29eSdanielk1977 char *zScript; /* The script to be run */ 700202b29eSdanielk1977 SqlCollate *pNext; /* Next function on the list of them all */ 710202b29eSdanielk1977 }; 720202b29eSdanielk1977 730202b29eSdanielk1977 /* 74fb7e7651Sdrh ** Prepared statements are cached for faster execution. Each prepared 75fb7e7651Sdrh ** statement is described by an instance of the following structure. 76fb7e7651Sdrh */ 77fb7e7651Sdrh typedef struct SqlPreparedStmt SqlPreparedStmt; 78fb7e7651Sdrh struct SqlPreparedStmt { 79fb7e7651Sdrh SqlPreparedStmt *pNext; /* Next in linked list */ 80fb7e7651Sdrh SqlPreparedStmt *pPrev; /* Previous on the list */ 81fb7e7651Sdrh sqlite3_stmt *pStmt; /* The prepared statement */ 82fb7e7651Sdrh int nSql; /* chars in zSql[] */ 83fb7e7651Sdrh char zSql[1]; /* Text of the SQL statement */ 84fb7e7651Sdrh }; 85fb7e7651Sdrh 86fb7e7651Sdrh /* 87bec3f402Sdrh ** There is one instance of this structure for each SQLite database 88bec3f402Sdrh ** that has been opened by the SQLite TCL interface. 89bec3f402Sdrh */ 90bec3f402Sdrh typedef struct SqliteDb SqliteDb; 91bec3f402Sdrh struct SqliteDb { 92895d7472Sdrh sqlite3 *db; /* The "real" database structure */ 93bec3f402Sdrh Tcl_Interp *interp; /* The interpreter used for this database */ 946d31316cSdrh char *zBusy; /* The busy callback routine */ 95aa940eacSdrh char *zCommit; /* The commit hook callback routine */ 96b5a20d3cSdrh char *zTrace; /* The trace callback routine */ 9719e2d37fSdrh char *zProfile; /* The profile callback routine */ 98348bb5d6Sdanielk1977 char *zProgress; /* The progress callback routine */ 99e22a334bSdrh char *zAuth; /* The authorization callback routine */ 10055c45f2eSdanielk1977 char *zNull; /* Text to substitute for an SQL NULL value */ 101cabb0819Sdrh SqlFunc *pFunc; /* List of SQL functions */ 10294eb6a14Sdanielk1977 Tcl_Obj *pUpdateHook; /* Update hook script (if any) */ 10371fd80bfSdanielk1977 Tcl_Obj *pRollbackHook; /* Rollback hook script (if any) */ 1040202b29eSdanielk1977 SqlCollate *pCollate; /* List of SQL collation functions */ 1056f8a503dSdanielk1977 int rc; /* Return code of most recent sqlite3_exec() */ 1067cedc8d4Sdanielk1977 Tcl_Obj *pCollateNeeded; /* Collation needed script */ 107fb7e7651Sdrh SqlPreparedStmt *stmtList; /* List of prepared statements*/ 108fb7e7651Sdrh SqlPreparedStmt *stmtLast; /* Last statement in the list */ 109fb7e7651Sdrh int maxStmt; /* The next maximum number of stmtList */ 110fb7e7651Sdrh int nStmt; /* Number of statements in stmtList */ 11198808babSdrh }; 112297ecf14Sdrh 1136d31316cSdrh /* 114d1e4733dSdrh ** Look at the script prefix in pCmd. We will be executing this script 115d1e4733dSdrh ** after first appending one or more arguments. This routine analyzes 116d1e4733dSdrh ** the script to see if it is safe to use Tcl_EvalObjv() on the script 117d1e4733dSdrh ** rather than the more general Tcl_EvalEx(). Tcl_EvalObjv() is much 118d1e4733dSdrh ** faster. 119d1e4733dSdrh ** 120d1e4733dSdrh ** Scripts that are safe to use with Tcl_EvalObjv() consists of a 121d1e4733dSdrh ** command name followed by zero or more arguments with no [...] or $ 122d1e4733dSdrh ** or {...} or ; to be seen anywhere. Most callback scripts consist 123d1e4733dSdrh ** of just a single procedure name and they meet this requirement. 124d1e4733dSdrh */ 125d1e4733dSdrh static int safeToUseEvalObjv(Tcl_Interp *interp, Tcl_Obj *pCmd){ 126d1e4733dSdrh /* We could try to do something with Tcl_Parse(). But we will instead 127d1e4733dSdrh ** just do a search for forbidden characters. If any of the forbidden 128d1e4733dSdrh ** characters appear in pCmd, we will report the string as unsafe. 129d1e4733dSdrh */ 130d1e4733dSdrh const char *z; 131d1e4733dSdrh int n; 132d1e4733dSdrh z = Tcl_GetStringFromObj(pCmd, &n); 133d1e4733dSdrh while( n-- > 0 ){ 134d1e4733dSdrh int c = *(z++); 135d1e4733dSdrh if( c=='$' || c=='[' || c==';' ) return 0; 136d1e4733dSdrh } 137d1e4733dSdrh return 1; 138d1e4733dSdrh } 139d1e4733dSdrh 140d1e4733dSdrh /* 141d1e4733dSdrh ** Find an SqlFunc structure with the given name. Or create a new 142d1e4733dSdrh ** one if an existing one cannot be found. Return a pointer to the 143d1e4733dSdrh ** structure. 144d1e4733dSdrh */ 145d1e4733dSdrh static SqlFunc *findSqlFunc(SqliteDb *pDb, const char *zName){ 146d1e4733dSdrh SqlFunc *p, *pNew; 147d1e4733dSdrh int i; 148d1e4733dSdrh pNew = (SqlFunc*)Tcl_Alloc( sizeof(*pNew) + strlen(zName) + 1 ); 149d1e4733dSdrh pNew->zName = (char*)&pNew[1]; 150d1e4733dSdrh for(i=0; zName[i]; i++){ pNew->zName[i] = tolower(zName[i]); } 151d1e4733dSdrh pNew->zName[i] = 0; 152d1e4733dSdrh for(p=pDb->pFunc; p; p=p->pNext){ 153d1e4733dSdrh if( strcmp(p->zName, pNew->zName)==0 ){ 154d1e4733dSdrh Tcl_Free((char*)pNew); 155d1e4733dSdrh return p; 156d1e4733dSdrh } 157d1e4733dSdrh } 158d1e4733dSdrh pNew->interp = pDb->interp; 159d1e4733dSdrh pNew->pScript = 0; 160d1e4733dSdrh pNew->pNext = pDb->pFunc; 161d1e4733dSdrh pDb->pFunc = pNew; 162d1e4733dSdrh return pNew; 163d1e4733dSdrh } 164d1e4733dSdrh 165d1e4733dSdrh /* 166fb7e7651Sdrh ** Finalize and free a list of prepared statements 167fb7e7651Sdrh */ 168fb7e7651Sdrh static void flushStmtCache( SqliteDb *pDb ){ 169fb7e7651Sdrh SqlPreparedStmt *pPreStmt; 170fb7e7651Sdrh 171fb7e7651Sdrh while( pDb->stmtList ){ 172fb7e7651Sdrh sqlite3_finalize( pDb->stmtList->pStmt ); 173fb7e7651Sdrh pPreStmt = pDb->stmtList; 174fb7e7651Sdrh pDb->stmtList = pDb->stmtList->pNext; 175fb7e7651Sdrh Tcl_Free( (char*)pPreStmt ); 176fb7e7651Sdrh } 177fb7e7651Sdrh pDb->nStmt = 0; 178fb7e7651Sdrh pDb->stmtLast = 0; 179fb7e7651Sdrh } 180fb7e7651Sdrh 181fb7e7651Sdrh /* 182895d7472Sdrh ** TCL calls this procedure when an sqlite3 database command is 183895d7472Sdrh ** deleted. 18475897234Sdrh */ 18575897234Sdrh static void DbDeleteCmd(void *db){ 186bec3f402Sdrh SqliteDb *pDb = (SqliteDb*)db; 187fb7e7651Sdrh flushStmtCache(pDb); 1886f8a503dSdanielk1977 sqlite3_close(pDb->db); 189cabb0819Sdrh while( pDb->pFunc ){ 190cabb0819Sdrh SqlFunc *pFunc = pDb->pFunc; 191cabb0819Sdrh pDb->pFunc = pFunc->pNext; 192d1e4733dSdrh Tcl_DecrRefCount(pFunc->pScript); 193cabb0819Sdrh Tcl_Free((char*)pFunc); 194cabb0819Sdrh } 1950202b29eSdanielk1977 while( pDb->pCollate ){ 1960202b29eSdanielk1977 SqlCollate *pCollate = pDb->pCollate; 1970202b29eSdanielk1977 pDb->pCollate = pCollate->pNext; 1980202b29eSdanielk1977 Tcl_Free((char*)pCollate); 1990202b29eSdanielk1977 } 200bec3f402Sdrh if( pDb->zBusy ){ 201bec3f402Sdrh Tcl_Free(pDb->zBusy); 202bec3f402Sdrh } 203b5a20d3cSdrh if( pDb->zTrace ){ 204b5a20d3cSdrh Tcl_Free(pDb->zTrace); 2050d1a643aSdrh } 20619e2d37fSdrh if( pDb->zProfile ){ 20719e2d37fSdrh Tcl_Free(pDb->zProfile); 20819e2d37fSdrh } 209e22a334bSdrh if( pDb->zAuth ){ 210e22a334bSdrh Tcl_Free(pDb->zAuth); 211e22a334bSdrh } 21255c45f2eSdanielk1977 if( pDb->zNull ){ 21355c45f2eSdanielk1977 Tcl_Free(pDb->zNull); 21455c45f2eSdanielk1977 } 21594eb6a14Sdanielk1977 if( pDb->pUpdateHook ){ 21694eb6a14Sdanielk1977 Tcl_DecrRefCount(pDb->pUpdateHook); 21794eb6a14Sdanielk1977 } 21871fd80bfSdanielk1977 if( pDb->pRollbackHook ){ 21971fd80bfSdanielk1977 Tcl_DecrRefCount(pDb->pRollbackHook); 22071fd80bfSdanielk1977 } 22194eb6a14Sdanielk1977 if( pDb->pCollateNeeded ){ 22294eb6a14Sdanielk1977 Tcl_DecrRefCount(pDb->pCollateNeeded); 22394eb6a14Sdanielk1977 } 224bec3f402Sdrh Tcl_Free((char*)pDb); 225bec3f402Sdrh } 226bec3f402Sdrh 227bec3f402Sdrh /* 228bec3f402Sdrh ** This routine is called when a database file is locked while trying 229bec3f402Sdrh ** to execute SQL. 230bec3f402Sdrh */ 2312a764eb0Sdanielk1977 static int DbBusyHandler(void *cd, int nTries){ 232bec3f402Sdrh SqliteDb *pDb = (SqliteDb*)cd; 233bec3f402Sdrh int rc; 234bec3f402Sdrh char zVal[30]; 235bec3f402Sdrh 236bec3f402Sdrh sprintf(zVal, "%d", nTries); 237d1e4733dSdrh rc = Tcl_VarEval(pDb->interp, pDb->zBusy, " ", zVal, (char*)0); 238bec3f402Sdrh if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){ 239bec3f402Sdrh return 0; 240bec3f402Sdrh } 241bec3f402Sdrh return 1; 24275897234Sdrh } 24375897234Sdrh 24475897234Sdrh /* 245348bb5d6Sdanielk1977 ** This routine is invoked as the 'progress callback' for the database. 246348bb5d6Sdanielk1977 */ 247348bb5d6Sdanielk1977 static int DbProgressHandler(void *cd){ 248348bb5d6Sdanielk1977 SqliteDb *pDb = (SqliteDb*)cd; 249348bb5d6Sdanielk1977 int rc; 250348bb5d6Sdanielk1977 251348bb5d6Sdanielk1977 assert( pDb->zProgress ); 252348bb5d6Sdanielk1977 rc = Tcl_Eval(pDb->interp, pDb->zProgress); 253348bb5d6Sdanielk1977 if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){ 254348bb5d6Sdanielk1977 return 1; 255348bb5d6Sdanielk1977 } 256348bb5d6Sdanielk1977 return 0; 257348bb5d6Sdanielk1977 } 258348bb5d6Sdanielk1977 259348bb5d6Sdanielk1977 /* 260b5a20d3cSdrh ** This routine is called by the SQLite trace handler whenever a new 261b5a20d3cSdrh ** block of SQL is executed. The TCL script in pDb->zTrace is executed. 2620d1a643aSdrh */ 263b5a20d3cSdrh static void DbTraceHandler(void *cd, const char *zSql){ 2640d1a643aSdrh SqliteDb *pDb = (SqliteDb*)cd; 265b5a20d3cSdrh Tcl_DString str; 2660d1a643aSdrh 267b5a20d3cSdrh Tcl_DStringInit(&str); 268b5a20d3cSdrh Tcl_DStringAppend(&str, pDb->zTrace, -1); 269b5a20d3cSdrh Tcl_DStringAppendElement(&str, zSql); 270b5a20d3cSdrh Tcl_Eval(pDb->interp, Tcl_DStringValue(&str)); 271b5a20d3cSdrh Tcl_DStringFree(&str); 272b5a20d3cSdrh Tcl_ResetResult(pDb->interp); 2730d1a643aSdrh } 2740d1a643aSdrh 2750d1a643aSdrh /* 27619e2d37fSdrh ** This routine is called by the SQLite profile handler after a statement 27719e2d37fSdrh ** SQL has executed. The TCL script in pDb->zProfile is evaluated. 27819e2d37fSdrh */ 27919e2d37fSdrh static void DbProfileHandler(void *cd, const char *zSql, sqlite_uint64 tm){ 28019e2d37fSdrh SqliteDb *pDb = (SqliteDb*)cd; 28119e2d37fSdrh Tcl_DString str; 28219e2d37fSdrh char zTm[100]; 28319e2d37fSdrh 28419e2d37fSdrh sqlite3_snprintf(sizeof(zTm)-1, zTm, "%lld", tm); 28519e2d37fSdrh Tcl_DStringInit(&str); 28619e2d37fSdrh Tcl_DStringAppend(&str, pDb->zProfile, -1); 28719e2d37fSdrh Tcl_DStringAppendElement(&str, zSql); 28819e2d37fSdrh Tcl_DStringAppendElement(&str, zTm); 28919e2d37fSdrh Tcl_Eval(pDb->interp, Tcl_DStringValue(&str)); 29019e2d37fSdrh Tcl_DStringFree(&str); 29119e2d37fSdrh Tcl_ResetResult(pDb->interp); 29219e2d37fSdrh } 29319e2d37fSdrh 29419e2d37fSdrh /* 295aa940eacSdrh ** This routine is called when a transaction is committed. The 296aa940eacSdrh ** TCL script in pDb->zCommit is executed. If it returns non-zero or 297aa940eacSdrh ** if it throws an exception, the transaction is rolled back instead 298aa940eacSdrh ** of being committed. 299aa940eacSdrh */ 300aa940eacSdrh static int DbCommitHandler(void *cd){ 301aa940eacSdrh SqliteDb *pDb = (SqliteDb*)cd; 302aa940eacSdrh int rc; 303aa940eacSdrh 304aa940eacSdrh rc = Tcl_Eval(pDb->interp, pDb->zCommit); 305aa940eacSdrh if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){ 306aa940eacSdrh return 1; 307aa940eacSdrh } 308aa940eacSdrh return 0; 309aa940eacSdrh } 310aa940eacSdrh 31171fd80bfSdanielk1977 static void DbRollbackHandler(void *clientData){ 31271fd80bfSdanielk1977 SqliteDb *pDb = (SqliteDb*)clientData; 31371fd80bfSdanielk1977 assert(pDb->pRollbackHook); 31471fd80bfSdanielk1977 if( TCL_OK!=Tcl_EvalObjEx(pDb->interp, pDb->pRollbackHook, 0) ){ 31571fd80bfSdanielk1977 Tcl_BackgroundError(pDb->interp); 31671fd80bfSdanielk1977 } 31771fd80bfSdanielk1977 } 31871fd80bfSdanielk1977 31994eb6a14Sdanielk1977 static void DbUpdateHandler( 32094eb6a14Sdanielk1977 void *p, 32194eb6a14Sdanielk1977 int op, 32294eb6a14Sdanielk1977 const char *zDb, 32394eb6a14Sdanielk1977 const char *zTbl, 32494eb6a14Sdanielk1977 sqlite_int64 rowid 32594eb6a14Sdanielk1977 ){ 32694eb6a14Sdanielk1977 SqliteDb *pDb = (SqliteDb *)p; 32794eb6a14Sdanielk1977 Tcl_Obj *pCmd; 32894eb6a14Sdanielk1977 32994eb6a14Sdanielk1977 assert( pDb->pUpdateHook ); 33094eb6a14Sdanielk1977 assert( op==SQLITE_INSERT || op==SQLITE_UPDATE || op==SQLITE_DELETE ); 33194eb6a14Sdanielk1977 33294eb6a14Sdanielk1977 pCmd = Tcl_DuplicateObj(pDb->pUpdateHook); 33394eb6a14Sdanielk1977 Tcl_IncrRefCount(pCmd); 33494eb6a14Sdanielk1977 Tcl_ListObjAppendElement(0, pCmd, Tcl_NewStringObj( 33594eb6a14Sdanielk1977 ( (op==SQLITE_INSERT)?"INSERT":(op==SQLITE_UPDATE)?"UPDATE":"DELETE"), -1)); 33694eb6a14Sdanielk1977 Tcl_ListObjAppendElement(0, pCmd, Tcl_NewStringObj(zDb, -1)); 33794eb6a14Sdanielk1977 Tcl_ListObjAppendElement(0, pCmd, Tcl_NewStringObj(zTbl, -1)); 33894eb6a14Sdanielk1977 Tcl_ListObjAppendElement(0, pCmd, Tcl_NewWideIntObj(rowid)); 33994eb6a14Sdanielk1977 Tcl_EvalObjEx(pDb->interp, pCmd, TCL_EVAL_DIRECT); 34094eb6a14Sdanielk1977 } 34194eb6a14Sdanielk1977 3427cedc8d4Sdanielk1977 static void tclCollateNeeded( 3437cedc8d4Sdanielk1977 void *pCtx, 3449bb575fdSdrh sqlite3 *db, 3457cedc8d4Sdanielk1977 int enc, 3467cedc8d4Sdanielk1977 const char *zName 3477cedc8d4Sdanielk1977 ){ 3487cedc8d4Sdanielk1977 SqliteDb *pDb = (SqliteDb *)pCtx; 3497cedc8d4Sdanielk1977 Tcl_Obj *pScript = Tcl_DuplicateObj(pDb->pCollateNeeded); 3507cedc8d4Sdanielk1977 Tcl_IncrRefCount(pScript); 3517cedc8d4Sdanielk1977 Tcl_ListObjAppendElement(0, pScript, Tcl_NewStringObj(zName, -1)); 3527cedc8d4Sdanielk1977 Tcl_EvalObjEx(pDb->interp, pScript, 0); 3537cedc8d4Sdanielk1977 Tcl_DecrRefCount(pScript); 3547cedc8d4Sdanielk1977 } 3557cedc8d4Sdanielk1977 356aa940eacSdrh /* 3570202b29eSdanielk1977 ** This routine is called to evaluate an SQL collation function implemented 3580202b29eSdanielk1977 ** using TCL script. 3590202b29eSdanielk1977 */ 3600202b29eSdanielk1977 static int tclSqlCollate( 3610202b29eSdanielk1977 void *pCtx, 3620202b29eSdanielk1977 int nA, 3630202b29eSdanielk1977 const void *zA, 3640202b29eSdanielk1977 int nB, 3650202b29eSdanielk1977 const void *zB 3660202b29eSdanielk1977 ){ 3670202b29eSdanielk1977 SqlCollate *p = (SqlCollate *)pCtx; 3680202b29eSdanielk1977 Tcl_Obj *pCmd; 3690202b29eSdanielk1977 3700202b29eSdanielk1977 pCmd = Tcl_NewStringObj(p->zScript, -1); 3710202b29eSdanielk1977 Tcl_IncrRefCount(pCmd); 3720202b29eSdanielk1977 Tcl_ListObjAppendElement(p->interp, pCmd, Tcl_NewStringObj(zA, nA)); 3730202b29eSdanielk1977 Tcl_ListObjAppendElement(p->interp, pCmd, Tcl_NewStringObj(zB, nB)); 374d1e4733dSdrh Tcl_EvalObjEx(p->interp, pCmd, TCL_EVAL_DIRECT); 3750202b29eSdanielk1977 Tcl_DecrRefCount(pCmd); 3760202b29eSdanielk1977 return (atoi(Tcl_GetStringResult(p->interp))); 3770202b29eSdanielk1977 } 3780202b29eSdanielk1977 3790202b29eSdanielk1977 /* 380cabb0819Sdrh ** This routine is called to evaluate an SQL function implemented 381cabb0819Sdrh ** using TCL script. 382cabb0819Sdrh */ 3830ae8b831Sdanielk1977 static void tclSqlFunc(sqlite3_context *context, int argc, sqlite3_value**argv){ 3846f8a503dSdanielk1977 SqlFunc *p = sqlite3_user_data(context); 385d1e4733dSdrh Tcl_Obj *pCmd; 386cabb0819Sdrh int i; 387cabb0819Sdrh int rc; 388cabb0819Sdrh 389d1e4733dSdrh if( argc==0 ){ 390d1e4733dSdrh /* If there are no arguments to the function, call Tcl_EvalObjEx on the 391d1e4733dSdrh ** script object directly. This allows the TCL compiler to generate 392d1e4733dSdrh ** bytecode for the command on the first invocation and thus make 393d1e4733dSdrh ** subsequent invocations much faster. */ 394d1e4733dSdrh pCmd = p->pScript; 395d1e4733dSdrh Tcl_IncrRefCount(pCmd); 396d1e4733dSdrh rc = Tcl_EvalObjEx(p->interp, pCmd, 0); 397d1e4733dSdrh Tcl_DecrRefCount(pCmd); 39851ad0ecdSdanielk1977 }else{ 399d1e4733dSdrh /* If there are arguments to the function, make a shallow copy of the 400d1e4733dSdrh ** script object, lappend the arguments, then evaluate the copy. 401d1e4733dSdrh ** 402d1e4733dSdrh ** By "shallow" copy, we mean a only the outer list Tcl_Obj is duplicated. 403d1e4733dSdrh ** The new Tcl_Obj contains pointers to the original list elements. 404d1e4733dSdrh ** That way, when Tcl_EvalObjv() is run and shimmers the first element 405d1e4733dSdrh ** of the list to tclCmdNameType, that alternate representation will 406d1e4733dSdrh ** be preserved and reused on the next invocation. 407d1e4733dSdrh */ 408d1e4733dSdrh Tcl_Obj **aArg; 409d1e4733dSdrh int nArg; 410d1e4733dSdrh if( Tcl_ListObjGetElements(p->interp, p->pScript, &nArg, &aArg) ){ 411d1e4733dSdrh sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1); 412d1e4733dSdrh return; 413d1e4733dSdrh } 414d1e4733dSdrh pCmd = Tcl_NewListObj(nArg, aArg); 415d1e4733dSdrh Tcl_IncrRefCount(pCmd); 416d1e4733dSdrh for(i=0; i<argc; i++){ 417d1e4733dSdrh sqlite3_value *pIn = argv[i]; 418d1e4733dSdrh Tcl_Obj *pVal; 419d1e4733dSdrh 420d1e4733dSdrh /* Set pVal to contain the i'th column of this row. */ 421d1e4733dSdrh switch( sqlite3_value_type(pIn) ){ 422d1e4733dSdrh case SQLITE_BLOB: { 423d1e4733dSdrh int bytes = sqlite3_value_bytes(pIn); 424d1e4733dSdrh pVal = Tcl_NewByteArrayObj(sqlite3_value_blob(pIn), bytes); 425d1e4733dSdrh break; 426d1e4733dSdrh } 427d1e4733dSdrh case SQLITE_INTEGER: { 428d1e4733dSdrh sqlite_int64 v = sqlite3_value_int64(pIn); 429d1e4733dSdrh if( v>=-2147483647 && v<=2147483647 ){ 430d1e4733dSdrh pVal = Tcl_NewIntObj(v); 431d1e4733dSdrh }else{ 432d1e4733dSdrh pVal = Tcl_NewWideIntObj(v); 433d1e4733dSdrh } 434d1e4733dSdrh break; 435d1e4733dSdrh } 436d1e4733dSdrh case SQLITE_FLOAT: { 437d1e4733dSdrh double r = sqlite3_value_double(pIn); 438d1e4733dSdrh pVal = Tcl_NewDoubleObj(r); 439d1e4733dSdrh break; 440d1e4733dSdrh } 441d1e4733dSdrh case SQLITE_NULL: { 442d1e4733dSdrh pVal = Tcl_NewStringObj("", 0); 443d1e4733dSdrh break; 444d1e4733dSdrh } 445d1e4733dSdrh default: { 446d1e4733dSdrh int bytes = sqlite3_value_bytes(pIn); 44700fd957bSdanielk1977 pVal = Tcl_NewStringObj((char *)sqlite3_value_text(pIn), bytes); 448d1e4733dSdrh break; 44951ad0ecdSdanielk1977 } 450cabb0819Sdrh } 451d1e4733dSdrh rc = Tcl_ListObjAppendElement(p->interp, pCmd, pVal); 452d1e4733dSdrh if( rc ){ 453d1e4733dSdrh Tcl_DecrRefCount(pCmd); 454d1e4733dSdrh sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1); 455d1e4733dSdrh return; 456d1e4733dSdrh } 457d1e4733dSdrh } 458d1e4733dSdrh if( !p->useEvalObjv ){ 459d1e4733dSdrh /* Tcl_EvalObjEx() will automatically call Tcl_EvalObjv() if pCmd 460d1e4733dSdrh ** is a list without a string representation. To prevent this from 461d1e4733dSdrh ** happening, make sure pCmd has a valid string representation */ 462d1e4733dSdrh Tcl_GetString(pCmd); 463d1e4733dSdrh } 464d1e4733dSdrh rc = Tcl_EvalObjEx(p->interp, pCmd, TCL_EVAL_DIRECT); 465d1e4733dSdrh Tcl_DecrRefCount(pCmd); 466d1e4733dSdrh } 467562e8d3cSdanielk1977 468c7f269d5Sdrh if( rc && rc!=TCL_RETURN ){ 4697e18c259Sdanielk1977 sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1); 470cabb0819Sdrh }else{ 471c7f269d5Sdrh Tcl_Obj *pVar = Tcl_GetObjResult(p->interp); 472c7f269d5Sdrh int n; 473c7f269d5Sdrh u8 *data; 474c7f269d5Sdrh char *zType = pVar->typePtr ? pVar->typePtr->name : ""; 475c7f269d5Sdrh char c = zType[0]; 476df0bddaeSdrh if( c=='b' && strcmp(zType,"bytearray")==0 && pVar->bytes==0 ){ 477d1e4733dSdrh /* Only return a BLOB type if the Tcl variable is a bytearray and 478df0bddaeSdrh ** has no string representation. */ 479c7f269d5Sdrh data = Tcl_GetByteArrayFromObj(pVar, &n); 480c7f269d5Sdrh sqlite3_result_blob(context, data, n, SQLITE_TRANSIENT); 481c7f269d5Sdrh }else if( (c=='b' && strcmp(zType,"boolean")==0) || 482c7f269d5Sdrh (c=='i' && strcmp(zType,"int")==0) ){ 483c7f269d5Sdrh Tcl_GetIntFromObj(0, pVar, &n); 484c7f269d5Sdrh sqlite3_result_int(context, n); 485c7f269d5Sdrh }else if( c=='d' && strcmp(zType,"double")==0 ){ 486c7f269d5Sdrh double r; 487c7f269d5Sdrh Tcl_GetDoubleFromObj(0, pVar, &r); 488c7f269d5Sdrh sqlite3_result_double(context, r); 489df0bddaeSdrh }else if( c=='w' && strcmp(zType,"wideInt")==0 ){ 490df0bddaeSdrh Tcl_WideInt v; 491df0bddaeSdrh Tcl_GetWideIntFromObj(0, pVar, &v); 492df0bddaeSdrh sqlite3_result_int64(context, v); 493c7f269d5Sdrh }else{ 49400fd957bSdanielk1977 data = (unsigned char *)Tcl_GetStringFromObj(pVar, &n); 49500fd957bSdanielk1977 sqlite3_result_text(context, (char *)data, n, SQLITE_TRANSIENT); 496c7f269d5Sdrh } 497cabb0819Sdrh } 498cabb0819Sdrh } 499895d7472Sdrh 500e22a334bSdrh #ifndef SQLITE_OMIT_AUTHORIZATION 501e22a334bSdrh /* 502e22a334bSdrh ** This is the authentication function. It appends the authentication 503e22a334bSdrh ** type code and the two arguments to zCmd[] then invokes the result 504e22a334bSdrh ** on the interpreter. The reply is examined to determine if the 505e22a334bSdrh ** authentication fails or succeeds. 506e22a334bSdrh */ 507e22a334bSdrh static int auth_callback( 508e22a334bSdrh void *pArg, 509e22a334bSdrh int code, 510e22a334bSdrh const char *zArg1, 511e22a334bSdrh const char *zArg2, 512e22a334bSdrh const char *zArg3, 513e22a334bSdrh const char *zArg4 514e22a334bSdrh ){ 515e22a334bSdrh char *zCode; 516e22a334bSdrh Tcl_DString str; 517e22a334bSdrh int rc; 518e22a334bSdrh const char *zReply; 519e22a334bSdrh SqliteDb *pDb = (SqliteDb*)pArg; 520e22a334bSdrh 521e22a334bSdrh switch( code ){ 522e22a334bSdrh case SQLITE_COPY : zCode="SQLITE_COPY"; break; 523e22a334bSdrh case SQLITE_CREATE_INDEX : zCode="SQLITE_CREATE_INDEX"; break; 524e22a334bSdrh case SQLITE_CREATE_TABLE : zCode="SQLITE_CREATE_TABLE"; break; 525e22a334bSdrh case SQLITE_CREATE_TEMP_INDEX : zCode="SQLITE_CREATE_TEMP_INDEX"; break; 526e22a334bSdrh case SQLITE_CREATE_TEMP_TABLE : zCode="SQLITE_CREATE_TEMP_TABLE"; break; 527e22a334bSdrh case SQLITE_CREATE_TEMP_TRIGGER: zCode="SQLITE_CREATE_TEMP_TRIGGER"; break; 528e22a334bSdrh case SQLITE_CREATE_TEMP_VIEW : zCode="SQLITE_CREATE_TEMP_VIEW"; break; 529e22a334bSdrh case SQLITE_CREATE_TRIGGER : zCode="SQLITE_CREATE_TRIGGER"; break; 530e22a334bSdrh case SQLITE_CREATE_VIEW : zCode="SQLITE_CREATE_VIEW"; break; 531e22a334bSdrh case SQLITE_DELETE : zCode="SQLITE_DELETE"; break; 532e22a334bSdrh case SQLITE_DROP_INDEX : zCode="SQLITE_DROP_INDEX"; break; 533e22a334bSdrh case SQLITE_DROP_TABLE : zCode="SQLITE_DROP_TABLE"; break; 534e22a334bSdrh case SQLITE_DROP_TEMP_INDEX : zCode="SQLITE_DROP_TEMP_INDEX"; break; 535e22a334bSdrh case SQLITE_DROP_TEMP_TABLE : zCode="SQLITE_DROP_TEMP_TABLE"; break; 536e22a334bSdrh case SQLITE_DROP_TEMP_TRIGGER : zCode="SQLITE_DROP_TEMP_TRIGGER"; break; 537e22a334bSdrh case SQLITE_DROP_TEMP_VIEW : zCode="SQLITE_DROP_TEMP_VIEW"; break; 538e22a334bSdrh case SQLITE_DROP_TRIGGER : zCode="SQLITE_DROP_TRIGGER"; break; 539e22a334bSdrh case SQLITE_DROP_VIEW : zCode="SQLITE_DROP_VIEW"; break; 540e22a334bSdrh case SQLITE_INSERT : zCode="SQLITE_INSERT"; break; 541e22a334bSdrh case SQLITE_PRAGMA : zCode="SQLITE_PRAGMA"; break; 542e22a334bSdrh case SQLITE_READ : zCode="SQLITE_READ"; break; 543e22a334bSdrh case SQLITE_SELECT : zCode="SQLITE_SELECT"; break; 544e22a334bSdrh case SQLITE_TRANSACTION : zCode="SQLITE_TRANSACTION"; break; 545e22a334bSdrh case SQLITE_UPDATE : zCode="SQLITE_UPDATE"; break; 54681e293b4Sdrh case SQLITE_ATTACH : zCode="SQLITE_ATTACH"; break; 54781e293b4Sdrh case SQLITE_DETACH : zCode="SQLITE_DETACH"; break; 5481c8c23ccSdanielk1977 case SQLITE_ALTER_TABLE : zCode="SQLITE_ALTER_TABLE"; break; 5491d54df88Sdanielk1977 case SQLITE_REINDEX : zCode="SQLITE_REINDEX"; break; 550e6e04969Sdrh case SQLITE_ANALYZE : zCode="SQLITE_ANALYZE"; break; 551e22a334bSdrh default : zCode="????"; break; 552e22a334bSdrh } 553e22a334bSdrh Tcl_DStringInit(&str); 554e22a334bSdrh Tcl_DStringAppend(&str, pDb->zAuth, -1); 555e22a334bSdrh Tcl_DStringAppendElement(&str, zCode); 556e22a334bSdrh Tcl_DStringAppendElement(&str, zArg1 ? zArg1 : ""); 557e22a334bSdrh Tcl_DStringAppendElement(&str, zArg2 ? zArg2 : ""); 558e22a334bSdrh Tcl_DStringAppendElement(&str, zArg3 ? zArg3 : ""); 559e22a334bSdrh Tcl_DStringAppendElement(&str, zArg4 ? zArg4 : ""); 560e22a334bSdrh rc = Tcl_GlobalEval(pDb->interp, Tcl_DStringValue(&str)); 561e22a334bSdrh Tcl_DStringFree(&str); 562e22a334bSdrh zReply = Tcl_GetStringResult(pDb->interp); 563e22a334bSdrh if( strcmp(zReply,"SQLITE_OK")==0 ){ 564e22a334bSdrh rc = SQLITE_OK; 565e22a334bSdrh }else if( strcmp(zReply,"SQLITE_DENY")==0 ){ 566e22a334bSdrh rc = SQLITE_DENY; 567e22a334bSdrh }else if( strcmp(zReply,"SQLITE_IGNORE")==0 ){ 568e22a334bSdrh rc = SQLITE_IGNORE; 569e22a334bSdrh }else{ 570e22a334bSdrh rc = 999; 571e22a334bSdrh } 572e22a334bSdrh return rc; 573e22a334bSdrh } 574e22a334bSdrh #endif /* SQLITE_OMIT_AUTHORIZATION */ 575cabb0819Sdrh 576cabb0819Sdrh /* 577ef2cb63eSdanielk1977 ** zText is a pointer to text obtained via an sqlite3_result_text() 578ef2cb63eSdanielk1977 ** or similar interface. This routine returns a Tcl string object, 579ef2cb63eSdanielk1977 ** reference count set to 0, containing the text. If a translation 580ef2cb63eSdanielk1977 ** between iso8859 and UTF-8 is required, it is preformed. 581ef2cb63eSdanielk1977 */ 582ef2cb63eSdanielk1977 static Tcl_Obj *dbTextToObj(char const *zText){ 583ef2cb63eSdanielk1977 Tcl_Obj *pVal; 584ef2cb63eSdanielk1977 #ifdef UTF_TRANSLATION_NEEDED 585ef2cb63eSdanielk1977 Tcl_DString dCol; 586ef2cb63eSdanielk1977 Tcl_DStringInit(&dCol); 587ef2cb63eSdanielk1977 Tcl_ExternalToUtfDString(NULL, zText, -1, &dCol); 588ef2cb63eSdanielk1977 pVal = Tcl_NewStringObj(Tcl_DStringValue(&dCol), -1); 589ef2cb63eSdanielk1977 Tcl_DStringFree(&dCol); 590ef2cb63eSdanielk1977 #else 591ef2cb63eSdanielk1977 pVal = Tcl_NewStringObj(zText, -1); 592ef2cb63eSdanielk1977 #endif 593ef2cb63eSdanielk1977 return pVal; 594ef2cb63eSdanielk1977 } 595ef2cb63eSdanielk1977 596ef2cb63eSdanielk1977 /* 5971067fe11Stpoindex ** This routine reads a line of text from FILE in, stores 5981067fe11Stpoindex ** the text in memory obtained from malloc() and returns a pointer 5991067fe11Stpoindex ** to the text. NULL is returned at end of file, or if malloc() 6001067fe11Stpoindex ** fails. 6011067fe11Stpoindex ** 6021067fe11Stpoindex ** The interface is like "readline" but no command-line editing 6031067fe11Stpoindex ** is done. 6041067fe11Stpoindex ** 6051067fe11Stpoindex ** copied from shell.c from '.import' command 6061067fe11Stpoindex */ 6071067fe11Stpoindex static char *local_getline(char *zPrompt, FILE *in){ 6081067fe11Stpoindex char *zLine; 6091067fe11Stpoindex int nLine; 6101067fe11Stpoindex int n; 6111067fe11Stpoindex int eol; 6121067fe11Stpoindex 6131067fe11Stpoindex nLine = 100; 6141067fe11Stpoindex zLine = malloc( nLine ); 6151067fe11Stpoindex if( zLine==0 ) return 0; 6161067fe11Stpoindex n = 0; 6171067fe11Stpoindex eol = 0; 6181067fe11Stpoindex while( !eol ){ 6191067fe11Stpoindex if( n+100>nLine ){ 6201067fe11Stpoindex nLine = nLine*2 + 100; 6211067fe11Stpoindex zLine = realloc(zLine, nLine); 6221067fe11Stpoindex if( zLine==0 ) return 0; 6231067fe11Stpoindex } 6241067fe11Stpoindex if( fgets(&zLine[n], nLine - n, in)==0 ){ 6251067fe11Stpoindex if( n==0 ){ 6261067fe11Stpoindex free(zLine); 6271067fe11Stpoindex return 0; 6281067fe11Stpoindex } 6291067fe11Stpoindex zLine[n] = 0; 6301067fe11Stpoindex eol = 1; 6311067fe11Stpoindex break; 6321067fe11Stpoindex } 6331067fe11Stpoindex while( zLine[n] ){ n++; } 6341067fe11Stpoindex if( n>0 && zLine[n-1]=='\n' ){ 6351067fe11Stpoindex n--; 6361067fe11Stpoindex zLine[n] = 0; 6371067fe11Stpoindex eol = 1; 6381067fe11Stpoindex } 6391067fe11Stpoindex } 6401067fe11Stpoindex zLine = realloc( zLine, n+1 ); 6411067fe11Stpoindex return zLine; 6421067fe11Stpoindex } 6431067fe11Stpoindex 6441067fe11Stpoindex /* 64575897234Sdrh ** The "sqlite" command below creates a new Tcl command for each 64675897234Sdrh ** connection it opens to an SQLite database. This routine is invoked 64775897234Sdrh ** whenever one of those connection-specific commands is executed 64875897234Sdrh ** in Tcl. For example, if you run Tcl code like this: 64975897234Sdrh ** 6509bb575fdSdrh ** sqlite3 db1 "my_database" 65175897234Sdrh ** db1 close 65275897234Sdrh ** 65375897234Sdrh ** The first command opens a connection to the "my_database" database 65475897234Sdrh ** and calls that connection "db1". The second command causes this 65575897234Sdrh ** subroutine to be invoked. 65675897234Sdrh */ 6576d31316cSdrh static int DbObjCmd(void *cd, Tcl_Interp *interp, int objc,Tcl_Obj *const*objv){ 658bec3f402Sdrh SqliteDb *pDb = (SqliteDb*)cd; 6596d31316cSdrh int choice; 66022fbcb8dSdrh int rc = TCL_OK; 6610de8c112Sdrh static const char *DB_strs[] = { 662fb7e7651Sdrh "authorizer", "busy", "cache", 663fb7e7651Sdrh "changes", "close", "collate", 664fb7e7651Sdrh "collation_needed", "commit_hook", "complete", 665fb7e7651Sdrh "copy", "errorcode", "eval", 66697f2ebc1Sdrh "exists", "function", "last_insert_rowid", 66797f2ebc1Sdrh "nullvalue", "onecolumn", "profile", 66871fd80bfSdanielk1977 "progress", "rekey", "rollback_hook", 6690190d1daSdanielk1977 "release_memory", "soft_heap_limit", "timeout", 6700190d1daSdanielk1977 "total_changes", "trace", "transaction", 6710190d1daSdanielk1977 "update_hook", "version", 0 6726d31316cSdrh }; 673411995dcSdrh enum DB_enum { 674fb7e7651Sdrh DB_AUTHORIZER, DB_BUSY, DB_CACHE, 675fb7e7651Sdrh DB_CHANGES, DB_CLOSE, DB_COLLATE, 676fb7e7651Sdrh DB_COLLATION_NEEDED, DB_COMMIT_HOOK, DB_COMPLETE, 677fb7e7651Sdrh DB_COPY, DB_ERRORCODE, DB_EVAL, 67897f2ebc1Sdrh DB_EXISTS, DB_FUNCTION, DB_LAST_INSERT_ROWID, 67997f2ebc1Sdrh DB_NULLVALUE, DB_ONECOLUMN, DB_PROFILE, 68071fd80bfSdanielk1977 DB_PROGRESS, DB_REKEY, DB_ROLLBACK_HOOK, 6810190d1daSdanielk1977 DB_RELEASE_MEMORY, DB_SOFT_HEAP_LIMIT, DB_TIMEOUT, 6820190d1daSdanielk1977 DB_TOTAL_CHANGES, DB_TRACE, DB_TRANSACTION, 6830190d1daSdanielk1977 DB_UPDATE_HOOK, DB_VERSION 6846d31316cSdrh }; 6851067fe11Stpoindex /* don't leave trailing commas on DB_enum, it confuses the AIX xlc compiler */ 6866d31316cSdrh 6876d31316cSdrh if( objc<2 ){ 6886d31316cSdrh Tcl_WrongNumArgs(interp, 1, objv, "SUBCOMMAND ..."); 68975897234Sdrh return TCL_ERROR; 69075897234Sdrh } 691411995dcSdrh if( Tcl_GetIndexFromObj(interp, objv[1], DB_strs, "option", 0, &choice) ){ 6926d31316cSdrh return TCL_ERROR; 6936d31316cSdrh } 6946d31316cSdrh 695411995dcSdrh switch( (enum DB_enum)choice ){ 69675897234Sdrh 697e22a334bSdrh /* $db authorizer ?CALLBACK? 698e22a334bSdrh ** 699e22a334bSdrh ** Invoke the given callback to authorize each SQL operation as it is 700e22a334bSdrh ** compiled. 5 arguments are appended to the callback before it is 701e22a334bSdrh ** invoked: 702e22a334bSdrh ** 703e22a334bSdrh ** (1) The authorization type (ex: SQLITE_CREATE_TABLE, SQLITE_INSERT, ...) 704e22a334bSdrh ** (2) First descriptive name (depends on authorization type) 705e22a334bSdrh ** (3) Second descriptive name 706e22a334bSdrh ** (4) Name of the database (ex: "main", "temp") 707e22a334bSdrh ** (5) Name of trigger that is doing the access 708e22a334bSdrh ** 709e22a334bSdrh ** The callback should return on of the following strings: SQLITE_OK, 710e22a334bSdrh ** SQLITE_IGNORE, or SQLITE_DENY. Any other return value is an error. 711e22a334bSdrh ** 712e22a334bSdrh ** If this method is invoked with no arguments, the current authorization 713e22a334bSdrh ** callback string is returned. 714e22a334bSdrh */ 715e22a334bSdrh case DB_AUTHORIZER: { 7161211de37Sdrh #ifdef SQLITE_OMIT_AUTHORIZATION 7171211de37Sdrh Tcl_AppendResult(interp, "authorization not available in this build", 0); 7181211de37Sdrh return TCL_ERROR; 7191211de37Sdrh #else 720e22a334bSdrh if( objc>3 ){ 721e22a334bSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?"); 7220f14e2ebSdrh return TCL_ERROR; 723e22a334bSdrh }else if( objc==2 ){ 724b5a20d3cSdrh if( pDb->zAuth ){ 725e22a334bSdrh Tcl_AppendResult(interp, pDb->zAuth, 0); 726e22a334bSdrh } 727e22a334bSdrh }else{ 728e22a334bSdrh char *zAuth; 729e22a334bSdrh int len; 730e22a334bSdrh if( pDb->zAuth ){ 731e22a334bSdrh Tcl_Free(pDb->zAuth); 732e22a334bSdrh } 733e22a334bSdrh zAuth = Tcl_GetStringFromObj(objv[2], &len); 734e22a334bSdrh if( zAuth && len>0 ){ 735e22a334bSdrh pDb->zAuth = Tcl_Alloc( len + 1 ); 736e22a334bSdrh strcpy(pDb->zAuth, zAuth); 737e22a334bSdrh }else{ 738e22a334bSdrh pDb->zAuth = 0; 739e22a334bSdrh } 740e22a334bSdrh if( pDb->zAuth ){ 741e22a334bSdrh pDb->interp = interp; 7426f8a503dSdanielk1977 sqlite3_set_authorizer(pDb->db, auth_callback, pDb); 743e22a334bSdrh }else{ 7446f8a503dSdanielk1977 sqlite3_set_authorizer(pDb->db, 0, 0); 745e22a334bSdrh } 746e22a334bSdrh } 7471211de37Sdrh #endif 748e22a334bSdrh break; 749e22a334bSdrh } 750e22a334bSdrh 751bec3f402Sdrh /* $db busy ?CALLBACK? 752bec3f402Sdrh ** 753bec3f402Sdrh ** Invoke the given callback if an SQL statement attempts to open 754bec3f402Sdrh ** a locked database file. 755bec3f402Sdrh */ 7566d31316cSdrh case DB_BUSY: { 7576d31316cSdrh if( objc>3 ){ 7586d31316cSdrh Tcl_WrongNumArgs(interp, 2, objv, "CALLBACK"); 759bec3f402Sdrh return TCL_ERROR; 7606d31316cSdrh }else if( objc==2 ){ 761bec3f402Sdrh if( pDb->zBusy ){ 762bec3f402Sdrh Tcl_AppendResult(interp, pDb->zBusy, 0); 763bec3f402Sdrh } 764bec3f402Sdrh }else{ 7656d31316cSdrh char *zBusy; 7666d31316cSdrh int len; 767bec3f402Sdrh if( pDb->zBusy ){ 768bec3f402Sdrh Tcl_Free(pDb->zBusy); 7696d31316cSdrh } 7706d31316cSdrh zBusy = Tcl_GetStringFromObj(objv[2], &len); 7716d31316cSdrh if( zBusy && len>0 ){ 7726d31316cSdrh pDb->zBusy = Tcl_Alloc( len + 1 ); 7736d31316cSdrh strcpy(pDb->zBusy, zBusy); 7746d31316cSdrh }else{ 775bec3f402Sdrh pDb->zBusy = 0; 776bec3f402Sdrh } 777bec3f402Sdrh if( pDb->zBusy ){ 778bec3f402Sdrh pDb->interp = interp; 7796f8a503dSdanielk1977 sqlite3_busy_handler(pDb->db, DbBusyHandler, pDb); 7806d31316cSdrh }else{ 7816f8a503dSdanielk1977 sqlite3_busy_handler(pDb->db, 0, 0); 782bec3f402Sdrh } 783bec3f402Sdrh } 7846d31316cSdrh break; 7856d31316cSdrh } 786bec3f402Sdrh 787fb7e7651Sdrh /* $db cache flush 788fb7e7651Sdrh ** $db cache size n 789fb7e7651Sdrh ** 790fb7e7651Sdrh ** Flush the prepared statement cache, or set the maximum number of 791fb7e7651Sdrh ** cached statements. 792fb7e7651Sdrh */ 793fb7e7651Sdrh case DB_CACHE: { 794fb7e7651Sdrh char *subCmd; 795fb7e7651Sdrh int n; 796fb7e7651Sdrh 797fb7e7651Sdrh if( objc<=2 ){ 798fb7e7651Sdrh Tcl_WrongNumArgs(interp, 1, objv, "cache option ?arg?"); 799fb7e7651Sdrh return TCL_ERROR; 800fb7e7651Sdrh } 801fb7e7651Sdrh subCmd = Tcl_GetStringFromObj( objv[2], 0 ); 802fb7e7651Sdrh if( *subCmd=='f' && strcmp(subCmd,"flush")==0 ){ 803fb7e7651Sdrh if( objc!=3 ){ 804fb7e7651Sdrh Tcl_WrongNumArgs(interp, 2, objv, "flush"); 805fb7e7651Sdrh return TCL_ERROR; 806fb7e7651Sdrh }else{ 807fb7e7651Sdrh flushStmtCache( pDb ); 808fb7e7651Sdrh } 809fb7e7651Sdrh }else if( *subCmd=='s' && strcmp(subCmd,"size")==0 ){ 810fb7e7651Sdrh if( objc!=4 ){ 811fb7e7651Sdrh Tcl_WrongNumArgs(interp, 2, objv, "size n"); 812fb7e7651Sdrh return TCL_ERROR; 813fb7e7651Sdrh }else{ 814fb7e7651Sdrh if( TCL_ERROR==Tcl_GetIntFromObj(interp, objv[3], &n) ){ 815fb7e7651Sdrh Tcl_AppendResult( interp, "cannot convert \"", 816fb7e7651Sdrh Tcl_GetStringFromObj(objv[3],0), "\" to integer", 0); 817fb7e7651Sdrh return TCL_ERROR; 818fb7e7651Sdrh }else{ 819fb7e7651Sdrh if( n<0 ){ 820fb7e7651Sdrh flushStmtCache( pDb ); 821fb7e7651Sdrh n = 0; 822fb7e7651Sdrh }else if( n>MAX_PREPARED_STMTS ){ 823fb7e7651Sdrh n = MAX_PREPARED_STMTS; 824fb7e7651Sdrh } 825fb7e7651Sdrh pDb->maxStmt = n; 826fb7e7651Sdrh } 827fb7e7651Sdrh } 828fb7e7651Sdrh }else{ 829fb7e7651Sdrh Tcl_AppendResult( interp, "bad option \"", 830fb7e7651Sdrh Tcl_GetStringFromObj(objv[0],0), "\": must be flush or size", 0); 831fb7e7651Sdrh return TCL_ERROR; 832fb7e7651Sdrh } 833fb7e7651Sdrh break; 834fb7e7651Sdrh } 835fb7e7651Sdrh 836b28af71aSdanielk1977 /* $db changes 837c8d30ac1Sdrh ** 838c8d30ac1Sdrh ** Return the number of rows that were modified, inserted, or deleted by 839b28af71aSdanielk1977 ** the most recent INSERT, UPDATE or DELETE statement, not including 840b28af71aSdanielk1977 ** any changes made by trigger programs. 841c8d30ac1Sdrh */ 842c8d30ac1Sdrh case DB_CHANGES: { 843c8d30ac1Sdrh Tcl_Obj *pResult; 844c8d30ac1Sdrh if( objc!=2 ){ 845c8d30ac1Sdrh Tcl_WrongNumArgs(interp, 2, objv, ""); 846c8d30ac1Sdrh return TCL_ERROR; 847c8d30ac1Sdrh } 848c8d30ac1Sdrh pResult = Tcl_GetObjResult(interp); 849b28af71aSdanielk1977 Tcl_SetIntObj(pResult, sqlite3_changes(pDb->db)); 850f146a776Srdc break; 851f146a776Srdc } 852f146a776Srdc 85375897234Sdrh /* $db close 85475897234Sdrh ** 85575897234Sdrh ** Shutdown the database 85675897234Sdrh */ 8576d31316cSdrh case DB_CLOSE: { 8586d31316cSdrh Tcl_DeleteCommand(interp, Tcl_GetStringFromObj(objv[0], 0)); 8596d31316cSdrh break; 8606d31316cSdrh } 86175897234Sdrh 8620f14e2ebSdrh /* 8630f14e2ebSdrh ** $db collate NAME SCRIPT 8640f14e2ebSdrh ** 8650f14e2ebSdrh ** Create a new SQL collation function called NAME. Whenever 8660f14e2ebSdrh ** that function is called, invoke SCRIPT to evaluate the function. 8670f14e2ebSdrh */ 8680f14e2ebSdrh case DB_COLLATE: { 8690f14e2ebSdrh SqlCollate *pCollate; 8700f14e2ebSdrh char *zName; 8710f14e2ebSdrh char *zScript; 8720f14e2ebSdrh int nScript; 8730f14e2ebSdrh if( objc!=4 ){ 8740f14e2ebSdrh Tcl_WrongNumArgs(interp, 2, objv, "NAME SCRIPT"); 8750f14e2ebSdrh return TCL_ERROR; 8760f14e2ebSdrh } 8770f14e2ebSdrh zName = Tcl_GetStringFromObj(objv[2], 0); 8780f14e2ebSdrh zScript = Tcl_GetStringFromObj(objv[3], &nScript); 8790f14e2ebSdrh pCollate = (SqlCollate*)Tcl_Alloc( sizeof(*pCollate) + nScript + 1 ); 8800f14e2ebSdrh if( pCollate==0 ) return TCL_ERROR; 8810f14e2ebSdrh pCollate->interp = interp; 8820f14e2ebSdrh pCollate->pNext = pDb->pCollate; 8830f14e2ebSdrh pCollate->zScript = (char*)&pCollate[1]; 8840f14e2ebSdrh pDb->pCollate = pCollate; 8850f14e2ebSdrh strcpy(pCollate->zScript, zScript); 8860f14e2ebSdrh if( sqlite3_create_collation(pDb->db, zName, SQLITE_UTF8, 8870f14e2ebSdrh pCollate, tclSqlCollate) ){ 8889636c4e1Sdanielk1977 Tcl_SetResult(interp, (char *)sqlite3_errmsg(pDb->db), TCL_VOLATILE); 8890f14e2ebSdrh return TCL_ERROR; 8900f14e2ebSdrh } 8910f14e2ebSdrh break; 8920f14e2ebSdrh } 8930f14e2ebSdrh 8940f14e2ebSdrh /* 8950f14e2ebSdrh ** $db collation_needed SCRIPT 8960f14e2ebSdrh ** 8970f14e2ebSdrh ** Create a new SQL collation function called NAME. Whenever 8980f14e2ebSdrh ** that function is called, invoke SCRIPT to evaluate the function. 8990f14e2ebSdrh */ 9000f14e2ebSdrh case DB_COLLATION_NEEDED: { 9010f14e2ebSdrh if( objc!=3 ){ 9020f14e2ebSdrh Tcl_WrongNumArgs(interp, 2, objv, "SCRIPT"); 9030f14e2ebSdrh return TCL_ERROR; 9040f14e2ebSdrh } 9050f14e2ebSdrh if( pDb->pCollateNeeded ){ 9060f14e2ebSdrh Tcl_DecrRefCount(pDb->pCollateNeeded); 9070f14e2ebSdrh } 9080f14e2ebSdrh pDb->pCollateNeeded = Tcl_DuplicateObj(objv[2]); 9090f14e2ebSdrh Tcl_IncrRefCount(pDb->pCollateNeeded); 9100f14e2ebSdrh sqlite3_collation_needed(pDb->db, pDb, tclCollateNeeded); 9110f14e2ebSdrh break; 9120f14e2ebSdrh } 9130f14e2ebSdrh 91419e2d37fSdrh /* $db commit_hook ?CALLBACK? 91519e2d37fSdrh ** 91619e2d37fSdrh ** Invoke the given callback just before committing every SQL transaction. 91719e2d37fSdrh ** If the callback throws an exception or returns non-zero, then the 91819e2d37fSdrh ** transaction is aborted. If CALLBACK is an empty string, the callback 91919e2d37fSdrh ** is disabled. 92019e2d37fSdrh */ 92119e2d37fSdrh case DB_COMMIT_HOOK: { 92219e2d37fSdrh if( objc>3 ){ 92319e2d37fSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?"); 92419e2d37fSdrh return TCL_ERROR; 92519e2d37fSdrh }else if( objc==2 ){ 92619e2d37fSdrh if( pDb->zCommit ){ 92719e2d37fSdrh Tcl_AppendResult(interp, pDb->zCommit, 0); 92819e2d37fSdrh } 92919e2d37fSdrh }else{ 93019e2d37fSdrh char *zCommit; 93119e2d37fSdrh int len; 93219e2d37fSdrh if( pDb->zCommit ){ 93319e2d37fSdrh Tcl_Free(pDb->zCommit); 93419e2d37fSdrh } 93519e2d37fSdrh zCommit = Tcl_GetStringFromObj(objv[2], &len); 93619e2d37fSdrh if( zCommit && len>0 ){ 93719e2d37fSdrh pDb->zCommit = Tcl_Alloc( len + 1 ); 93819e2d37fSdrh strcpy(pDb->zCommit, zCommit); 93919e2d37fSdrh }else{ 94019e2d37fSdrh pDb->zCommit = 0; 94119e2d37fSdrh } 94219e2d37fSdrh if( pDb->zCommit ){ 94319e2d37fSdrh pDb->interp = interp; 94419e2d37fSdrh sqlite3_commit_hook(pDb->db, DbCommitHandler, pDb); 94519e2d37fSdrh }else{ 94619e2d37fSdrh sqlite3_commit_hook(pDb->db, 0, 0); 94719e2d37fSdrh } 94819e2d37fSdrh } 94919e2d37fSdrh break; 95019e2d37fSdrh } 95119e2d37fSdrh 95275897234Sdrh /* $db complete SQL 95375897234Sdrh ** 95475897234Sdrh ** Return TRUE if SQL is a complete SQL statement. Return FALSE if 95575897234Sdrh ** additional lines of input are needed. This is similar to the 95675897234Sdrh ** built-in "info complete" command of Tcl. 95775897234Sdrh */ 9586d31316cSdrh case DB_COMPLETE: { 959ccae6026Sdrh #ifndef SQLITE_OMIT_COMPLETE 9606d31316cSdrh Tcl_Obj *pResult; 9616d31316cSdrh int isComplete; 9626d31316cSdrh if( objc!=3 ){ 9636d31316cSdrh Tcl_WrongNumArgs(interp, 2, objv, "SQL"); 96475897234Sdrh return TCL_ERROR; 96575897234Sdrh } 9666f8a503dSdanielk1977 isComplete = sqlite3_complete( Tcl_GetStringFromObj(objv[2], 0) ); 9676d31316cSdrh pResult = Tcl_GetObjResult(interp); 9686d31316cSdrh Tcl_SetBooleanObj(pResult, isComplete); 969ccae6026Sdrh #endif 9706d31316cSdrh break; 9716d31316cSdrh } 97275897234Sdrh 97319e2d37fSdrh /* $db copy conflict-algorithm table filename ?SEPARATOR? ?NULLINDICATOR? 97419e2d37fSdrh ** 97519e2d37fSdrh ** Copy data into table from filename, optionally using SEPARATOR 97619e2d37fSdrh ** as column separators. If a column contains a null string, or the 97719e2d37fSdrh ** value of NULLINDICATOR, a NULL is inserted for the column. 97819e2d37fSdrh ** conflict-algorithm is one of the sqlite conflict algorithms: 97919e2d37fSdrh ** rollback, abort, fail, ignore, replace 98019e2d37fSdrh ** On success, return the number of lines processed, not necessarily same 98119e2d37fSdrh ** as 'db changes' due to conflict-algorithm selected. 98219e2d37fSdrh ** 98319e2d37fSdrh ** This code is basically an implementation/enhancement of 98419e2d37fSdrh ** the sqlite3 shell.c ".import" command. 98519e2d37fSdrh ** 98619e2d37fSdrh ** This command usage is equivalent to the sqlite2.x COPY statement, 98719e2d37fSdrh ** which imports file data into a table using the PostgreSQL COPY file format: 98819e2d37fSdrh ** $db copy $conflit_algo $table_name $filename \t \\N 98919e2d37fSdrh */ 99019e2d37fSdrh case DB_COPY: { 99119e2d37fSdrh char *zTable; /* Insert data into this table */ 99219e2d37fSdrh char *zFile; /* The file from which to extract data */ 99319e2d37fSdrh char *zConflict; /* The conflict algorithm to use */ 99419e2d37fSdrh sqlite3_stmt *pStmt; /* A statement */ 99519e2d37fSdrh int rc; /* Result code */ 99619e2d37fSdrh int nCol; /* Number of columns in the table */ 99719e2d37fSdrh int nByte; /* Number of bytes in an SQL string */ 99819e2d37fSdrh int i, j; /* Loop counters */ 99919e2d37fSdrh int nSep; /* Number of bytes in zSep[] */ 100019e2d37fSdrh int nNull; /* Number of bytes in zNull[] */ 100119e2d37fSdrh char *zSql; /* An SQL statement */ 100219e2d37fSdrh char *zLine; /* A single line of input from the file */ 100319e2d37fSdrh char **azCol; /* zLine[] broken up into columns */ 100419e2d37fSdrh char *zCommit; /* How to commit changes */ 100519e2d37fSdrh FILE *in; /* The input file */ 100619e2d37fSdrh int lineno = 0; /* Line number of input file */ 100719e2d37fSdrh char zLineNum[80]; /* Line number print buffer */ 100819e2d37fSdrh Tcl_Obj *pResult; /* interp result */ 100919e2d37fSdrh 101019e2d37fSdrh char *zSep; 101119e2d37fSdrh char *zNull; 101219e2d37fSdrh if( objc<5 || objc>7 ){ 101319e2d37fSdrh Tcl_WrongNumArgs(interp, 2, objv, 101419e2d37fSdrh "CONFLICT-ALGORITHM TABLE FILENAME ?SEPARATOR? ?NULLINDICATOR?"); 101519e2d37fSdrh return TCL_ERROR; 101619e2d37fSdrh } 101719e2d37fSdrh if( objc>=6 ){ 101819e2d37fSdrh zSep = Tcl_GetStringFromObj(objv[5], 0); 101919e2d37fSdrh }else{ 102019e2d37fSdrh zSep = "\t"; 102119e2d37fSdrh } 102219e2d37fSdrh if( objc>=7 ){ 102319e2d37fSdrh zNull = Tcl_GetStringFromObj(objv[6], 0); 102419e2d37fSdrh }else{ 102519e2d37fSdrh zNull = ""; 102619e2d37fSdrh } 102719e2d37fSdrh zConflict = Tcl_GetStringFromObj(objv[2], 0); 102819e2d37fSdrh zTable = Tcl_GetStringFromObj(objv[3], 0); 102919e2d37fSdrh zFile = Tcl_GetStringFromObj(objv[4], 0); 103019e2d37fSdrh nSep = strlen(zSep); 103119e2d37fSdrh nNull = strlen(zNull); 103219e2d37fSdrh if( nSep==0 ){ 103319e2d37fSdrh Tcl_AppendResult(interp, "Error: non-null separator required for copy", 0); 103419e2d37fSdrh return TCL_ERROR; 103519e2d37fSdrh } 103619e2d37fSdrh if(sqlite3StrICmp(zConflict, "rollback") != 0 && 103719e2d37fSdrh sqlite3StrICmp(zConflict, "abort" ) != 0 && 103819e2d37fSdrh sqlite3StrICmp(zConflict, "fail" ) != 0 && 103919e2d37fSdrh sqlite3StrICmp(zConflict, "ignore" ) != 0 && 104019e2d37fSdrh sqlite3StrICmp(zConflict, "replace" ) != 0 ) { 104119e2d37fSdrh Tcl_AppendResult(interp, "Error: \"", zConflict, 104219e2d37fSdrh "\", conflict-algorithm must be one of: rollback, " 104319e2d37fSdrh "abort, fail, ignore, or replace", 0); 104419e2d37fSdrh return TCL_ERROR; 104519e2d37fSdrh } 104619e2d37fSdrh zSql = sqlite3_mprintf("SELECT * FROM '%q'", zTable); 104719e2d37fSdrh if( zSql==0 ){ 104819e2d37fSdrh Tcl_AppendResult(interp, "Error: no such table: ", zTable, 0); 104919e2d37fSdrh return TCL_ERROR; 105019e2d37fSdrh } 105119e2d37fSdrh nByte = strlen(zSql); 105219e2d37fSdrh rc = sqlite3_prepare(pDb->db, zSql, 0, &pStmt, 0); 105319e2d37fSdrh sqlite3_free(zSql); 105419e2d37fSdrh if( rc ){ 105519e2d37fSdrh Tcl_AppendResult(interp, "Error: ", sqlite3_errmsg(pDb->db), 0); 105619e2d37fSdrh nCol = 0; 105719e2d37fSdrh }else{ 105819e2d37fSdrh nCol = sqlite3_column_count(pStmt); 105919e2d37fSdrh } 106019e2d37fSdrh sqlite3_finalize(pStmt); 106119e2d37fSdrh if( nCol==0 ) { 106219e2d37fSdrh return TCL_ERROR; 106319e2d37fSdrh } 106419e2d37fSdrh zSql = malloc( nByte + 50 + nCol*2 ); 106519e2d37fSdrh if( zSql==0 ) { 106619e2d37fSdrh Tcl_AppendResult(interp, "Error: can't malloc()", 0); 106719e2d37fSdrh return TCL_ERROR; 106819e2d37fSdrh } 106919e2d37fSdrh sqlite3_snprintf(nByte+50, zSql, "INSERT OR %q INTO '%q' VALUES(?", 107019e2d37fSdrh zConflict, zTable); 107119e2d37fSdrh j = strlen(zSql); 107219e2d37fSdrh for(i=1; i<nCol; i++){ 107319e2d37fSdrh zSql[j++] = ','; 107419e2d37fSdrh zSql[j++] = '?'; 107519e2d37fSdrh } 107619e2d37fSdrh zSql[j++] = ')'; 107719e2d37fSdrh zSql[j] = 0; 107819e2d37fSdrh rc = sqlite3_prepare(pDb->db, zSql, 0, &pStmt, 0); 107919e2d37fSdrh free(zSql); 108019e2d37fSdrh if( rc ){ 108119e2d37fSdrh Tcl_AppendResult(interp, "Error: ", sqlite3_errmsg(pDb->db), 0); 108219e2d37fSdrh sqlite3_finalize(pStmt); 108319e2d37fSdrh return TCL_ERROR; 108419e2d37fSdrh } 108519e2d37fSdrh in = fopen(zFile, "rb"); 108619e2d37fSdrh if( in==0 ){ 108719e2d37fSdrh Tcl_AppendResult(interp, "Error: cannot open file: ", zFile, NULL); 108819e2d37fSdrh sqlite3_finalize(pStmt); 108919e2d37fSdrh return TCL_ERROR; 109019e2d37fSdrh } 109119e2d37fSdrh azCol = malloc( sizeof(azCol[0])*(nCol+1) ); 109219e2d37fSdrh if( azCol==0 ) { 109319e2d37fSdrh Tcl_AppendResult(interp, "Error: can't malloc()", 0); 109419e2d37fSdrh return TCL_ERROR; 109519e2d37fSdrh } 109619e2d37fSdrh sqlite3_exec(pDb->db, "BEGIN", 0, 0, 0); 109719e2d37fSdrh zCommit = "COMMIT"; 109819e2d37fSdrh while( (zLine = local_getline(0, in))!=0 ){ 109919e2d37fSdrh char *z; 110019e2d37fSdrh i = 0; 110119e2d37fSdrh lineno++; 110219e2d37fSdrh azCol[0] = zLine; 110319e2d37fSdrh for(i=0, z=zLine; *z; z++){ 110419e2d37fSdrh if( *z==zSep[0] && strncmp(z, zSep, nSep)==0 ){ 110519e2d37fSdrh *z = 0; 110619e2d37fSdrh i++; 110719e2d37fSdrh if( i<nCol ){ 110819e2d37fSdrh azCol[i] = &z[nSep]; 110919e2d37fSdrh z += nSep-1; 111019e2d37fSdrh } 111119e2d37fSdrh } 111219e2d37fSdrh } 111319e2d37fSdrh if( i+1!=nCol ){ 111419e2d37fSdrh char *zErr; 111519e2d37fSdrh zErr = malloc(200 + strlen(zFile)); 111619e2d37fSdrh sprintf(zErr,"Error: %s line %d: expected %d columns of data but found %d", 111719e2d37fSdrh zFile, lineno, nCol, i+1); 111819e2d37fSdrh Tcl_AppendResult(interp, zErr, 0); 111919e2d37fSdrh free(zErr); 112019e2d37fSdrh zCommit = "ROLLBACK"; 112119e2d37fSdrh break; 112219e2d37fSdrh } 112319e2d37fSdrh for(i=0; i<nCol; i++){ 112419e2d37fSdrh /* check for null data, if so, bind as null */ 112519e2d37fSdrh if ((nNull>0 && strcmp(azCol[i], zNull)==0) || strlen(azCol[i])==0) { 112619e2d37fSdrh sqlite3_bind_null(pStmt, i+1); 112719e2d37fSdrh }else{ 112819e2d37fSdrh sqlite3_bind_text(pStmt, i+1, azCol[i], -1, SQLITE_STATIC); 112919e2d37fSdrh } 113019e2d37fSdrh } 113119e2d37fSdrh sqlite3_step(pStmt); 113219e2d37fSdrh rc = sqlite3_reset(pStmt); 113319e2d37fSdrh free(zLine); 113419e2d37fSdrh if( rc!=SQLITE_OK ){ 113519e2d37fSdrh Tcl_AppendResult(interp,"Error: ", sqlite3_errmsg(pDb->db), 0); 113619e2d37fSdrh zCommit = "ROLLBACK"; 113719e2d37fSdrh break; 113819e2d37fSdrh } 113919e2d37fSdrh } 114019e2d37fSdrh free(azCol); 114119e2d37fSdrh fclose(in); 114219e2d37fSdrh sqlite3_finalize(pStmt); 114319e2d37fSdrh sqlite3_exec(pDb->db, zCommit, 0, 0, 0); 114419e2d37fSdrh 114519e2d37fSdrh if( zCommit[0] == 'C' ){ 114619e2d37fSdrh /* success, set result as number of lines processed */ 114719e2d37fSdrh pResult = Tcl_GetObjResult(interp); 114819e2d37fSdrh Tcl_SetIntObj(pResult, lineno); 114919e2d37fSdrh rc = TCL_OK; 115019e2d37fSdrh }else{ 115119e2d37fSdrh /* failure, append lineno where failed */ 115219e2d37fSdrh sprintf(zLineNum,"%d",lineno); 115319e2d37fSdrh Tcl_AppendResult(interp,", failed while processing line: ",zLineNum,0); 115419e2d37fSdrh rc = TCL_ERROR; 115519e2d37fSdrh } 115619e2d37fSdrh break; 115719e2d37fSdrh } 115819e2d37fSdrh 115975897234Sdrh /* 1160dcd997eaSdrh ** $db errorcode 1161dcd997eaSdrh ** 1162dcd997eaSdrh ** Return the numeric error code that was returned by the most recent 11636f8a503dSdanielk1977 ** call to sqlite3_exec(). 1164dcd997eaSdrh */ 1165dcd997eaSdrh case DB_ERRORCODE: { 1166f3ce83f5Sdanielk1977 Tcl_SetObjResult(interp, Tcl_NewIntObj(sqlite3_errcode(pDb->db))); 1167dcd997eaSdrh break; 1168dcd997eaSdrh } 1169dcd997eaSdrh 1170dcd997eaSdrh /* 1171895d7472Sdrh ** $db eval $sql ?array? ?{ ...code... }? 11721807ce37Sdrh ** $db onecolumn $sql 117375897234Sdrh ** 117475897234Sdrh ** The SQL statement in $sql is evaluated. For each row, the values are 1175bec3f402Sdrh ** placed in elements of the array named "array" and ...code... is executed. 117675897234Sdrh ** If "array" and "code" are omitted, then no callback is every invoked. 117775897234Sdrh ** If "array" is an empty string, then the values are placed in variables 117875897234Sdrh ** that have the same name as the fields extracted by the query. 11791807ce37Sdrh ** 11801807ce37Sdrh ** The onecolumn method is the equivalent of: 11811807ce37Sdrh ** lindex [$db eval $sql] 0 118275897234Sdrh */ 11831807ce37Sdrh case DB_ONECOLUMN: 118497f2ebc1Sdrh case DB_EVAL: 118597f2ebc1Sdrh case DB_EXISTS: { 11869d74b4c5Sdrh char const *zSql; /* Next SQL statement to execute */ 11879d74b4c5Sdrh char const *zLeft; /* What is left after first stmt in zSql */ 11889d74b4c5Sdrh sqlite3_stmt *pStmt; /* Compiled SQL statment */ 118992febd92Sdrh Tcl_Obj *pArray; /* Name of array into which results are written */ 119092febd92Sdrh Tcl_Obj *pScript; /* Script to run for each result set */ 11911d895039Sdrh Tcl_Obj **apParm; /* Parameters that need a Tcl_DecrRefCount() */ 11921d895039Sdrh int nParm; /* Number of entries used in apParm[] */ 11931d895039Sdrh Tcl_Obj *aParm[10]; /* Static space for apParm[] in the common case */ 11941807ce37Sdrh Tcl_Obj *pRet; /* Value to be returned */ 1195fb7e7651Sdrh SqlPreparedStmt *pPreStmt; /* Pointer to a prepared statement */ 1196fb7e7651Sdrh int rc2; 1197ef2cb63eSdanielk1977 119897f2ebc1Sdrh if( choice==DB_EVAL ){ 119992febd92Sdrh if( objc<3 || objc>5 ){ 1200895d7472Sdrh Tcl_WrongNumArgs(interp, 2, objv, "SQL ?ARRAY-NAME? ?SCRIPT?"); 120130ccda10Sdanielk1977 return TCL_ERROR; 120230ccda10Sdanielk1977 } 12031807ce37Sdrh pRet = Tcl_NewObj(); 12041807ce37Sdrh Tcl_IncrRefCount(pRet); 120597f2ebc1Sdrh }else{ 120697f2ebc1Sdrh if( objc!=3 ){ 120797f2ebc1Sdrh Tcl_WrongNumArgs(interp, 2, objv, "SQL"); 120897f2ebc1Sdrh return TCL_ERROR; 120997f2ebc1Sdrh } 121097f2ebc1Sdrh pRet = 0; 121197f2ebc1Sdrh if( choice==DB_EXISTS ){ 121297f2ebc1Sdrh Tcl_SetObjResult(interp, Tcl_NewBooleanObj(0)); 121397f2ebc1Sdrh } 12141807ce37Sdrh } 121592febd92Sdrh if( objc==3 ){ 121692febd92Sdrh pArray = pScript = 0; 121792febd92Sdrh }else if( objc==4 ){ 121892febd92Sdrh pArray = 0; 121992febd92Sdrh pScript = objv[3]; 122092febd92Sdrh }else{ 122192febd92Sdrh pArray = objv[3]; 122292febd92Sdrh if( Tcl_GetString(pArray)[0]==0 ) pArray = 0; 122392febd92Sdrh pScript = objv[4]; 122492febd92Sdrh } 122530ccda10Sdanielk1977 12261d895039Sdrh Tcl_IncrRefCount(objv[2]); 122730ccda10Sdanielk1977 zSql = Tcl_GetStringFromObj(objv[2], 0); 122890b6bb19Sdrh while( rc==TCL_OK && zSql[0] ){ 122992febd92Sdrh int i; /* Loop counter */ 1230fb7e7651Sdrh int nVar; /* Number of bind parameters in the pStmt */ 123192febd92Sdrh int nCol; /* Number of columns in the result set */ 123292febd92Sdrh Tcl_Obj **apColName = 0; /* Array of column names */ 1233fb7e7651Sdrh int len; /* String length of zSql */ 123430ccda10Sdanielk1977 1235fb7e7651Sdrh /* Try to find a SQL statement that has already been compiled and 1236fb7e7651Sdrh ** which matches the next sequence of SQL. 1237fb7e7651Sdrh */ 1238fb7e7651Sdrh pStmt = 0; 1239fb7e7651Sdrh pPreStmt = pDb->stmtList; 1240fb7e7651Sdrh len = strlen(zSql); 1241fb7e7651Sdrh if( pPreStmt && sqlite3_expired(pPreStmt->pStmt) ){ 1242fb7e7651Sdrh flushStmtCache(pDb); 1243fb7e7651Sdrh pPreStmt = 0; 1244fb7e7651Sdrh } 1245fb7e7651Sdrh for(; pPreStmt; pPreStmt=pPreStmt->pNext){ 1246fb7e7651Sdrh int n = pPreStmt->nSql; 1247fb7e7651Sdrh if( len>=n 1248fb7e7651Sdrh && memcmp(pPreStmt->zSql, zSql, n)==0 1249fb7e7651Sdrh && (zSql[n]==0 || zSql[n-1]==';') 1250fb7e7651Sdrh ){ 1251fb7e7651Sdrh pStmt = pPreStmt->pStmt; 1252fb7e7651Sdrh zLeft = &zSql[pPreStmt->nSql]; 1253fb7e7651Sdrh 1254fb7e7651Sdrh /* When a prepared statement is found, unlink it from the 1255fb7e7651Sdrh ** cache list. It will later be added back to the beginning 1256fb7e7651Sdrh ** of the cache list in order to implement LRU replacement. 1257fb7e7651Sdrh */ 1258fb7e7651Sdrh if( pPreStmt->pPrev ){ 1259fb7e7651Sdrh pPreStmt->pPrev->pNext = pPreStmt->pNext; 1260fb7e7651Sdrh }else{ 1261fb7e7651Sdrh pDb->stmtList = pPreStmt->pNext; 1262fb7e7651Sdrh } 1263fb7e7651Sdrh if( pPreStmt->pNext ){ 1264fb7e7651Sdrh pPreStmt->pNext->pPrev = pPreStmt->pPrev; 1265fb7e7651Sdrh }else{ 1266fb7e7651Sdrh pDb->stmtLast = pPreStmt->pPrev; 1267fb7e7651Sdrh } 1268fb7e7651Sdrh pDb->nStmt--; 1269fb7e7651Sdrh break; 1270fb7e7651Sdrh } 1271fb7e7651Sdrh } 1272fb7e7651Sdrh 1273fb7e7651Sdrh /* If no prepared statement was found. Compile the SQL text 1274fb7e7651Sdrh */ 1275fb7e7651Sdrh if( pStmt==0 ){ 127630ccda10Sdanielk1977 if( SQLITE_OK!=sqlite3_prepare(pDb->db, zSql, -1, &pStmt, &zLeft) ){ 1277ef2cb63eSdanielk1977 Tcl_SetObjResult(interp, dbTextToObj(sqlite3_errmsg(pDb->db))); 127830ccda10Sdanielk1977 rc = TCL_ERROR; 127930ccda10Sdanielk1977 break; 128030ccda10Sdanielk1977 } 128192febd92Sdrh if( pStmt==0 ){ 128292febd92Sdrh if( SQLITE_OK!=sqlite3_errcode(pDb->db) ){ 1283fb7e7651Sdrh /* A compile-time error in the statement 1284fb7e7651Sdrh */ 128592febd92Sdrh Tcl_SetObjResult(interp, dbTextToObj(sqlite3_errmsg(pDb->db))); 128692febd92Sdrh rc = TCL_ERROR; 128792febd92Sdrh break; 128892febd92Sdrh }else{ 1289fb7e7651Sdrh /* The statement was a no-op. Continue to the next statement 1290fb7e7651Sdrh ** in the SQL string. 1291fb7e7651Sdrh */ 129292febd92Sdrh zSql = zLeft; 129392febd92Sdrh continue; 129492febd92Sdrh } 129592febd92Sdrh } 1296fb7e7651Sdrh assert( pPreStmt==0 ); 1297fb7e7651Sdrh } 129830ccda10Sdanielk1977 1299fb7e7651Sdrh /* Bind values to parameters that begin with $ or : 1300fb7e7651Sdrh */ 130192febd92Sdrh nVar = sqlite3_bind_parameter_count(pStmt); 13021d895039Sdrh nParm = 0; 13031d895039Sdrh if( nVar>sizeof(aParm)/sizeof(aParm[0]) ){ 13041d895039Sdrh apParm = (Tcl_Obj**)Tcl_Alloc(nVar*sizeof(apParm[0])); 13051d895039Sdrh }else{ 13061d895039Sdrh apParm = aParm; 13071d895039Sdrh } 130892febd92Sdrh for(i=1; i<=nVar; i++){ 130992febd92Sdrh const char *zVar = sqlite3_bind_parameter_name(pStmt, i); 1310c8f9079cSdrh if( zVar!=0 && (zVar[0]=='$' || zVar[0]==':') ){ 131192febd92Sdrh Tcl_Obj *pVar = Tcl_GetVar2Ex(interp, &zVar[1], 0, 0); 131292febd92Sdrh if( pVar ){ 131392febd92Sdrh int n; 131492febd92Sdrh u8 *data; 131592febd92Sdrh char *zType = pVar->typePtr ? pVar->typePtr->name : ""; 131692febd92Sdrh char c = zType[0]; 1317df0bddaeSdrh if( c=='b' && strcmp(zType,"bytearray")==0 && pVar->bytes==0 ){ 1318df0bddaeSdrh /* Only load a BLOB type if the Tcl variable is a bytearray and 1319df0bddaeSdrh ** has no string representation. */ 132092febd92Sdrh data = Tcl_GetByteArrayFromObj(pVar, &n); 132192febd92Sdrh sqlite3_bind_blob(pStmt, i, data, n, SQLITE_STATIC); 13221d895039Sdrh Tcl_IncrRefCount(pVar); 13231d895039Sdrh apParm[nParm++] = pVar; 132492febd92Sdrh }else if( (c=='b' && strcmp(zType,"boolean")==0) || 132592febd92Sdrh (c=='i' && strcmp(zType,"int")==0) ){ 132692febd92Sdrh Tcl_GetIntFromObj(interp, pVar, &n); 132792febd92Sdrh sqlite3_bind_int(pStmt, i, n); 132892febd92Sdrh }else if( c=='d' && strcmp(zType,"double")==0 ){ 132992febd92Sdrh double r; 133092febd92Sdrh Tcl_GetDoubleFromObj(interp, pVar, &r); 133192febd92Sdrh sqlite3_bind_double(pStmt, i, r); 1332df0bddaeSdrh }else if( c=='w' && strcmp(zType,"wideInt")==0 ){ 1333df0bddaeSdrh Tcl_WideInt v; 1334df0bddaeSdrh Tcl_GetWideIntFromObj(interp, pVar, &v); 1335df0bddaeSdrh sqlite3_bind_int64(pStmt, i, v); 133692febd92Sdrh }else{ 133700fd957bSdanielk1977 data = (unsigned char *)Tcl_GetStringFromObj(pVar, &n); 133800fd957bSdanielk1977 sqlite3_bind_text(pStmt, i, (char *)data, n, SQLITE_STATIC); 13391d895039Sdrh Tcl_IncrRefCount(pVar); 13401d895039Sdrh apParm[nParm++] = pVar; 134192febd92Sdrh } 1342fb7e7651Sdrh }else{ 1343fb7e7651Sdrh sqlite3_bind_null( pStmt, i ); 134492febd92Sdrh } 134592febd92Sdrh } 134692febd92Sdrh } 134792febd92Sdrh 134892febd92Sdrh /* Compute column names */ 134992febd92Sdrh nCol = sqlite3_column_count(pStmt); 135092febd92Sdrh if( pScript ){ 135192febd92Sdrh apColName = (Tcl_Obj**)Tcl_Alloc( sizeof(Tcl_Obj*)*nCol ); 135292febd92Sdrh if( apColName==0 ) break; 135392febd92Sdrh for(i=0; i<nCol; i++){ 135492febd92Sdrh apColName[i] = dbTextToObj(sqlite3_column_name(pStmt,i)); 135592febd92Sdrh Tcl_IncrRefCount(apColName[i]); 135692febd92Sdrh } 135792febd92Sdrh } 135892febd92Sdrh 135992febd92Sdrh /* If results are being stored in an array variable, then create 136092febd92Sdrh ** the array(*) entry for that array 136192febd92Sdrh */ 136292febd92Sdrh if( pArray ){ 136330ccda10Sdanielk1977 Tcl_Obj *pColList = Tcl_NewObj(); 13643ced14a6Sdrh Tcl_Obj *pStar = Tcl_NewStringObj("*", -1); 136530ccda10Sdanielk1977 Tcl_IncrRefCount(pColList); 136692febd92Sdrh for(i=0; i<nCol; i++){ 136792febd92Sdrh Tcl_ListObjAppendElement(interp, pColList, apColName[i]); 136830ccda10Sdanielk1977 } 13693ced14a6Sdrh Tcl_ObjSetVar2(interp, pArray, pStar, pColList,0); 13703ced14a6Sdrh Tcl_DecrRefCount(pColList); 13713ced14a6Sdrh Tcl_DecrRefCount(pStar); 137230ccda10Sdanielk1977 } 137330ccda10Sdanielk1977 137492febd92Sdrh /* Execute the SQL 137592febd92Sdrh */ 137690b6bb19Sdrh while( rc==TCL_OK && pStmt && SQLITE_ROW==sqlite3_step(pStmt) ){ 137792febd92Sdrh for(i=0; i<nCol; i++){ 137830ccda10Sdanielk1977 Tcl_Obj *pVal; 137930ccda10Sdanielk1977 138030ccda10Sdanielk1977 /* Set pVal to contain the i'th column of this row. */ 138192febd92Sdrh switch( sqlite3_column_type(pStmt, i) ){ 138292febd92Sdrh case SQLITE_BLOB: { 13833fd0a736Sdanielk1977 int bytes = sqlite3_column_bytes(pStmt, i); 13843fd0a736Sdanielk1977 pVal = Tcl_NewByteArrayObj(sqlite3_column_blob(pStmt, i), bytes); 138592febd92Sdrh break; 138692febd92Sdrh } 138792febd92Sdrh case SQLITE_INTEGER: { 138892febd92Sdrh sqlite_int64 v = sqlite3_column_int64(pStmt, i); 138992febd92Sdrh if( v>=-2147483647 && v<=2147483647 ){ 139092febd92Sdrh pVal = Tcl_NewIntObj(v); 139192febd92Sdrh }else{ 139292febd92Sdrh pVal = Tcl_NewWideIntObj(v); 139392febd92Sdrh } 139492febd92Sdrh break; 139592febd92Sdrh } 139692febd92Sdrh case SQLITE_FLOAT: { 139792febd92Sdrh double r = sqlite3_column_double(pStmt, i); 139892febd92Sdrh pVal = Tcl_NewDoubleObj(r); 139992febd92Sdrh break; 140092febd92Sdrh } 140155c45f2eSdanielk1977 case SQLITE_NULL: { 140255c45f2eSdanielk1977 pVal = dbTextToObj(pDb->zNull); 140355c45f2eSdanielk1977 break; 140455c45f2eSdanielk1977 } 140592febd92Sdrh default: { 140600fd957bSdanielk1977 pVal = dbTextToObj((char *)sqlite3_column_text(pStmt, i)); 140792febd92Sdrh break; 140892febd92Sdrh } 140930ccda10Sdanielk1977 } 141030ccda10Sdanielk1977 141192febd92Sdrh if( pScript ){ 141292febd92Sdrh if( pArray==0 ){ 141392febd92Sdrh Tcl_ObjSetVar2(interp, apColName[i], 0, pVal, 0); 141430ccda10Sdanielk1977 }else{ 141592febd92Sdrh Tcl_ObjSetVar2(interp, pArray, apColName[i], pVal, 0); 141630ccda10Sdanielk1977 } 14171807ce37Sdrh }else if( choice==DB_ONECOLUMN ){ 141897f2ebc1Sdrh assert( pRet==0 ); 14191807ce37Sdrh if( pRet==0 ){ 14201807ce37Sdrh pRet = pVal; 14211807ce37Sdrh Tcl_IncrRefCount(pRet); 14221807ce37Sdrh } 142390b6bb19Sdrh rc = TCL_BREAK; 142497f2ebc1Sdrh i = nCol; 142597f2ebc1Sdrh }else if( choice==DB_EXISTS ){ 142697f2ebc1Sdrh assert( pRet==0 ); 142797f2ebc1Sdrh Tcl_SetObjResult(interp, Tcl_NewBooleanObj(1)); 142897f2ebc1Sdrh rc = TCL_BREAK; 142997f2ebc1Sdrh i = nCol; 143030ccda10Sdanielk1977 }else{ 143130ccda10Sdanielk1977 Tcl_ListObjAppendElement(interp, pRet, pVal); 143230ccda10Sdanielk1977 } 143330ccda10Sdanielk1977 } 143430ccda10Sdanielk1977 143592febd92Sdrh if( pScript ){ 143692febd92Sdrh rc = Tcl_EvalObjEx(interp, pScript, 0); 143790b6bb19Sdrh if( rc==TCL_CONTINUE ){ 143890b6bb19Sdrh rc = TCL_OK; 143930ccda10Sdanielk1977 } 144030ccda10Sdanielk1977 } 144190b6bb19Sdrh } 144290b6bb19Sdrh if( rc==TCL_BREAK ){ 144390b6bb19Sdrh rc = TCL_OK; 144490b6bb19Sdrh } 144530ccda10Sdanielk1977 144692febd92Sdrh /* Free the column name objects */ 144792febd92Sdrh if( pScript ){ 144892febd92Sdrh for(i=0; i<nCol; i++){ 144992febd92Sdrh Tcl_DecrRefCount(apColName[i]); 145092febd92Sdrh } 145192febd92Sdrh Tcl_Free((char*)apColName); 145292febd92Sdrh } 145392febd92Sdrh 14541d895039Sdrh /* Free the bound string and blob parameters */ 14551d895039Sdrh for(i=0; i<nParm; i++){ 14561d895039Sdrh Tcl_DecrRefCount(apParm[i]); 14571d895039Sdrh } 14581d895039Sdrh if( apParm!=aParm ){ 14591d895039Sdrh Tcl_Free((char*)apParm); 14601d895039Sdrh } 14611d895039Sdrh 1462fb7e7651Sdrh /* Reset the statement. If the result code is SQLITE_SCHEMA, then 1463fb7e7651Sdrh ** flush the statement cache and try the statement again. 146492febd92Sdrh */ 1465fb7e7651Sdrh rc2 = sqlite3_reset(pStmt); 1466fb7e7651Sdrh if( SQLITE_SCHEMA==rc2 ){ 1467fb7e7651Sdrh /* After a schema change, flush the cache and try to run the 1468fb7e7651Sdrh ** statement again 1469fb7e7651Sdrh */ 1470fb7e7651Sdrh flushStmtCache( pDb ); 1471fb7e7651Sdrh sqlite3_finalize(pStmt); 1472fb7e7651Sdrh if( pPreStmt ) Tcl_Free((char*)pPreStmt); 147330ccda10Sdanielk1977 continue; 1474fb7e7651Sdrh }else if( SQLITE_OK!=rc2 ){ 1475fb7e7651Sdrh /* If a run-time error occurs, report the error and stop reading 1476fb7e7651Sdrh ** the SQL 1477fb7e7651Sdrh */ 1478ef2cb63eSdanielk1977 Tcl_SetObjResult(interp, dbTextToObj(sqlite3_errmsg(pDb->db))); 1479fb7e7651Sdrh sqlite3_finalize(pStmt); 148030ccda10Sdanielk1977 rc = TCL_ERROR; 1481fb7e7651Sdrh if( pPreStmt ) Tcl_Free((char*)pPreStmt); 148230ccda10Sdanielk1977 break; 1483fb7e7651Sdrh }else if( pDb->maxStmt<=0 ){ 1484fb7e7651Sdrh /* If the cache is turned off, deallocated the statement */ 1485fb7e7651Sdrh if( pPreStmt ) Tcl_Free((char*)pPreStmt); 1486fb7e7651Sdrh sqlite3_finalize(pStmt); 1487fb7e7651Sdrh }else{ 1488fb7e7651Sdrh /* Everything worked and the cache is operational. 1489fb7e7651Sdrh ** Create a new SqlPreparedStmt structure if we need one. 1490fb7e7651Sdrh ** (If we already have one we can just reuse it.) 1491fb7e7651Sdrh */ 1492fb7e7651Sdrh if( pPreStmt==0 ){ 1493fb7e7651Sdrh len = zLeft - zSql; 1494fb7e7651Sdrh pPreStmt = (SqlPreparedStmt*)Tcl_Alloc( sizeof(*pPreStmt) + len ); 1495fb7e7651Sdrh if( pPreStmt==0 ) return TCL_ERROR; 1496fb7e7651Sdrh pPreStmt->pStmt = pStmt; 1497fb7e7651Sdrh pPreStmt->nSql = len; 1498fb7e7651Sdrh memcpy(pPreStmt->zSql, zSql, len); 1499fb7e7651Sdrh pPreStmt->zSql[len] = 0; 150030ccda10Sdanielk1977 } 150130ccda10Sdanielk1977 1502fb7e7651Sdrh /* Add the prepared statement to the beginning of the cache list 1503fb7e7651Sdrh */ 1504fb7e7651Sdrh pPreStmt->pNext = pDb->stmtList; 1505fb7e7651Sdrh pPreStmt->pPrev = 0; 1506fb7e7651Sdrh if( pDb->stmtList ){ 1507fb7e7651Sdrh pDb->stmtList->pPrev = pPreStmt; 1508fb7e7651Sdrh } 1509fb7e7651Sdrh pDb->stmtList = pPreStmt; 1510fb7e7651Sdrh if( pDb->stmtLast==0 ){ 1511fb7e7651Sdrh assert( pDb->nStmt==0 ); 1512fb7e7651Sdrh pDb->stmtLast = pPreStmt; 1513fb7e7651Sdrh }else{ 1514fb7e7651Sdrh assert( pDb->nStmt>0 ); 1515fb7e7651Sdrh } 1516fb7e7651Sdrh pDb->nStmt++; 1517fb7e7651Sdrh 1518fb7e7651Sdrh /* If we have too many statement in cache, remove the surplus from the 1519fb7e7651Sdrh ** end of the cache list. 1520fb7e7651Sdrh */ 1521fb7e7651Sdrh while( pDb->nStmt>pDb->maxStmt ){ 1522fb7e7651Sdrh sqlite3_finalize(pDb->stmtLast->pStmt); 1523fb7e7651Sdrh pDb->stmtLast = pDb->stmtLast->pPrev; 1524fb7e7651Sdrh Tcl_Free((char*)pDb->stmtLast->pNext); 1525fb7e7651Sdrh pDb->stmtLast->pNext = 0; 1526fb7e7651Sdrh pDb->nStmt--; 1527fb7e7651Sdrh } 1528fb7e7651Sdrh } 1529fb7e7651Sdrh 1530fb7e7651Sdrh /* Proceed to the next statement */ 153130ccda10Sdanielk1977 zSql = zLeft; 153230ccda10Sdanielk1977 } 15331d895039Sdrh Tcl_DecrRefCount(objv[2]); 153430ccda10Sdanielk1977 15351807ce37Sdrh if( pRet ){ 1536ef2cb63eSdanielk1977 if( rc==TCL_OK ){ 153730ccda10Sdanielk1977 Tcl_SetObjResult(interp, pRet); 153830ccda10Sdanielk1977 } 1539ef2cb63eSdanielk1977 Tcl_DecrRefCount(pRet); 15401807ce37Sdrh } 154130ccda10Sdanielk1977 break; 154230ccda10Sdanielk1977 } 1543bec3f402Sdrh 1544bec3f402Sdrh /* 1545cabb0819Sdrh ** $db function NAME SCRIPT 1546cabb0819Sdrh ** 1547cabb0819Sdrh ** Create a new SQL function called NAME. Whenever that function is 1548cabb0819Sdrh ** called, invoke SCRIPT to evaluate the function. 1549cabb0819Sdrh */ 1550cabb0819Sdrh case DB_FUNCTION: { 1551cabb0819Sdrh SqlFunc *pFunc; 1552d1e4733dSdrh Tcl_Obj *pScript; 1553cabb0819Sdrh char *zName; 1554cabb0819Sdrh if( objc!=4 ){ 1555cabb0819Sdrh Tcl_WrongNumArgs(interp, 2, objv, "NAME SCRIPT"); 1556cabb0819Sdrh return TCL_ERROR; 1557cabb0819Sdrh } 1558cabb0819Sdrh zName = Tcl_GetStringFromObj(objv[2], 0); 1559d1e4733dSdrh pScript = objv[3]; 1560d1e4733dSdrh pFunc = findSqlFunc(pDb, zName); 1561cabb0819Sdrh if( pFunc==0 ) return TCL_ERROR; 1562d1e4733dSdrh if( pFunc->pScript ){ 1563d1e4733dSdrh Tcl_DecrRefCount(pFunc->pScript); 1564d1e4733dSdrh } 1565d1e4733dSdrh pFunc->pScript = pScript; 1566d1e4733dSdrh Tcl_IncrRefCount(pScript); 1567d1e4733dSdrh pFunc->useEvalObjv = safeToUseEvalObjv(interp, pScript); 1568c8c1158bSdanielk1977 rc = sqlite3_create_function(pDb->db, zName, -1, SQLITE_UTF8, 1569d8123366Sdanielk1977 pFunc, tclSqlFunc, 0, 0); 1570fb7e7651Sdrh if( rc!=SQLITE_OK ){ 1571fb7e7651Sdrh rc = TCL_ERROR; 15729636c4e1Sdanielk1977 Tcl_SetResult(interp, (char *)sqlite3_errmsg(pDb->db), TCL_VOLATILE); 1573fb7e7651Sdrh }else{ 1574fb7e7651Sdrh /* Must flush any cached statements */ 1575fb7e7651Sdrh flushStmtCache( pDb ); 1576fb7e7651Sdrh } 1577cabb0819Sdrh break; 1578cabb0819Sdrh } 1579cabb0819Sdrh 1580cabb0819Sdrh /* 158119e2d37fSdrh ** $db nullvalue ?STRING? 158219e2d37fSdrh ** 158319e2d37fSdrh ** Change text used when a NULL comes back from the database. If ?STRING? 158419e2d37fSdrh ** is not present, then the current string used for NULL is returned. 158519e2d37fSdrh ** If STRING is present, then STRING is returned. 158619e2d37fSdrh ** 158719e2d37fSdrh */ 158819e2d37fSdrh case DB_NULLVALUE: { 158919e2d37fSdrh if( objc!=2 && objc!=3 ){ 159019e2d37fSdrh Tcl_WrongNumArgs(interp, 2, objv, "NULLVALUE"); 159119e2d37fSdrh return TCL_ERROR; 159219e2d37fSdrh } 159319e2d37fSdrh if( objc==3 ){ 159419e2d37fSdrh int len; 159519e2d37fSdrh char *zNull = Tcl_GetStringFromObj(objv[2], &len); 159619e2d37fSdrh if( pDb->zNull ){ 159719e2d37fSdrh Tcl_Free(pDb->zNull); 159819e2d37fSdrh } 159919e2d37fSdrh if( zNull && len>0 ){ 160019e2d37fSdrh pDb->zNull = Tcl_Alloc( len + 1 ); 160119e2d37fSdrh strncpy(pDb->zNull, zNull, len); 160219e2d37fSdrh pDb->zNull[len] = '\0'; 160319e2d37fSdrh }else{ 160419e2d37fSdrh pDb->zNull = 0; 160519e2d37fSdrh } 160619e2d37fSdrh } 160719e2d37fSdrh Tcl_SetObjResult(interp, dbTextToObj(pDb->zNull)); 160819e2d37fSdrh break; 160919e2d37fSdrh } 161019e2d37fSdrh 161119e2d37fSdrh /* 1612af9ff33aSdrh ** $db last_insert_rowid 1613af9ff33aSdrh ** 1614af9ff33aSdrh ** Return an integer which is the ROWID for the most recent insert. 1615af9ff33aSdrh */ 1616af9ff33aSdrh case DB_LAST_INSERT_ROWID: { 1617af9ff33aSdrh Tcl_Obj *pResult; 1618af9ff33aSdrh int rowid; 1619af9ff33aSdrh if( objc!=2 ){ 1620af9ff33aSdrh Tcl_WrongNumArgs(interp, 2, objv, ""); 1621af9ff33aSdrh return TCL_ERROR; 1622af9ff33aSdrh } 16236f8a503dSdanielk1977 rowid = sqlite3_last_insert_rowid(pDb->db); 1624af9ff33aSdrh pResult = Tcl_GetObjResult(interp); 1625af9ff33aSdrh Tcl_SetIntObj(pResult, rowid); 1626af9ff33aSdrh break; 1627af9ff33aSdrh } 1628af9ff33aSdrh 1629af9ff33aSdrh /* 16301807ce37Sdrh ** The DB_ONECOLUMN method is implemented together with DB_EVAL. 16315d9d7576Sdrh */ 16321807ce37Sdrh 16331807ce37Sdrh /* $db progress ?N CALLBACK? 16341807ce37Sdrh ** 16351807ce37Sdrh ** Invoke the given callback every N virtual machine opcodes while executing 16361807ce37Sdrh ** queries. 16371807ce37Sdrh */ 16381807ce37Sdrh case DB_PROGRESS: { 16391807ce37Sdrh if( objc==2 ){ 16401807ce37Sdrh if( pDb->zProgress ){ 16411807ce37Sdrh Tcl_AppendResult(interp, pDb->zProgress, 0); 16425d9d7576Sdrh } 16431807ce37Sdrh }else if( objc==4 ){ 16441807ce37Sdrh char *zProgress; 16451807ce37Sdrh int len; 16461807ce37Sdrh int N; 16471807ce37Sdrh if( TCL_OK!=Tcl_GetIntFromObj(interp, objv[2], &N) ){ 16481807ce37Sdrh return TCL_ERROR; 16491807ce37Sdrh }; 16501807ce37Sdrh if( pDb->zProgress ){ 16511807ce37Sdrh Tcl_Free(pDb->zProgress); 16521807ce37Sdrh } 16531807ce37Sdrh zProgress = Tcl_GetStringFromObj(objv[3], &len); 16541807ce37Sdrh if( zProgress && len>0 ){ 16551807ce37Sdrh pDb->zProgress = Tcl_Alloc( len + 1 ); 16561807ce37Sdrh strcpy(pDb->zProgress, zProgress); 16571807ce37Sdrh }else{ 16581807ce37Sdrh pDb->zProgress = 0; 16591807ce37Sdrh } 16601807ce37Sdrh #ifndef SQLITE_OMIT_PROGRESS_CALLBACK 16611807ce37Sdrh if( pDb->zProgress ){ 16621807ce37Sdrh pDb->interp = interp; 16631807ce37Sdrh sqlite3_progress_handler(pDb->db, N, DbProgressHandler, pDb); 16641807ce37Sdrh }else{ 16651807ce37Sdrh sqlite3_progress_handler(pDb->db, 0, 0, 0); 16661807ce37Sdrh } 16671807ce37Sdrh #endif 16681807ce37Sdrh }else{ 16691807ce37Sdrh Tcl_WrongNumArgs(interp, 2, objv, "N CALLBACK"); 16701807ce37Sdrh return TCL_ERROR; 16715d9d7576Sdrh } 16725d9d7576Sdrh break; 16735d9d7576Sdrh } 16745d9d7576Sdrh 167519e2d37fSdrh /* $db profile ?CALLBACK? 167619e2d37fSdrh ** 167719e2d37fSdrh ** Make arrangements to invoke the CALLBACK routine after each SQL statement 167819e2d37fSdrh ** that has run. The text of the SQL and the amount of elapse time are 167919e2d37fSdrh ** appended to CALLBACK before the script is run. 168019e2d37fSdrh */ 168119e2d37fSdrh case DB_PROFILE: { 168219e2d37fSdrh if( objc>3 ){ 168319e2d37fSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?"); 168419e2d37fSdrh return TCL_ERROR; 168519e2d37fSdrh }else if( objc==2 ){ 168619e2d37fSdrh if( pDb->zProfile ){ 168719e2d37fSdrh Tcl_AppendResult(interp, pDb->zProfile, 0); 168819e2d37fSdrh } 168919e2d37fSdrh }else{ 169019e2d37fSdrh char *zProfile; 169119e2d37fSdrh int len; 169219e2d37fSdrh if( pDb->zProfile ){ 169319e2d37fSdrh Tcl_Free(pDb->zProfile); 169419e2d37fSdrh } 169519e2d37fSdrh zProfile = Tcl_GetStringFromObj(objv[2], &len); 169619e2d37fSdrh if( zProfile && len>0 ){ 169719e2d37fSdrh pDb->zProfile = Tcl_Alloc( len + 1 ); 169819e2d37fSdrh strcpy(pDb->zProfile, zProfile); 169919e2d37fSdrh }else{ 170019e2d37fSdrh pDb->zProfile = 0; 170119e2d37fSdrh } 170219e2d37fSdrh #ifndef SQLITE_OMIT_TRACE 170319e2d37fSdrh if( pDb->zProfile ){ 170419e2d37fSdrh pDb->interp = interp; 170519e2d37fSdrh sqlite3_profile(pDb->db, DbProfileHandler, pDb); 170619e2d37fSdrh }else{ 170719e2d37fSdrh sqlite3_profile(pDb->db, 0, 0); 170819e2d37fSdrh } 170919e2d37fSdrh #endif 171019e2d37fSdrh } 171119e2d37fSdrh break; 171219e2d37fSdrh } 171319e2d37fSdrh 17145d9d7576Sdrh /* 171522fbcb8dSdrh ** $db rekey KEY 171622fbcb8dSdrh ** 171722fbcb8dSdrh ** Change the encryption key on the currently open database. 171822fbcb8dSdrh */ 171922fbcb8dSdrh case DB_REKEY: { 172022fbcb8dSdrh int nKey; 172122fbcb8dSdrh void *pKey; 172222fbcb8dSdrh if( objc!=3 ){ 172322fbcb8dSdrh Tcl_WrongNumArgs(interp, 2, objv, "KEY"); 172422fbcb8dSdrh return TCL_ERROR; 172522fbcb8dSdrh } 172622fbcb8dSdrh pKey = Tcl_GetByteArrayFromObj(objv[2], &nKey); 17279eb9e26bSdrh #ifdef SQLITE_HAS_CODEC 17282011d5f5Sdrh rc = sqlite3_rekey(pDb->db, pKey, nKey); 172922fbcb8dSdrh if( rc ){ 1730f20b21c8Sdanielk1977 Tcl_AppendResult(interp, sqlite3ErrStr(rc), 0); 173122fbcb8dSdrh rc = TCL_ERROR; 173222fbcb8dSdrh } 173322fbcb8dSdrh #endif 173422fbcb8dSdrh break; 173522fbcb8dSdrh } 173622fbcb8dSdrh 173722fbcb8dSdrh /* 1738bec3f402Sdrh ** $db timeout MILLESECONDS 1739bec3f402Sdrh ** 1740bec3f402Sdrh ** Delay for the number of milliseconds specified when a file is locked. 1741bec3f402Sdrh */ 17426d31316cSdrh case DB_TIMEOUT: { 1743bec3f402Sdrh int ms; 17446d31316cSdrh if( objc!=3 ){ 17456d31316cSdrh Tcl_WrongNumArgs(interp, 2, objv, "MILLISECONDS"); 1746bec3f402Sdrh return TCL_ERROR; 174775897234Sdrh } 17486d31316cSdrh if( Tcl_GetIntFromObj(interp, objv[2], &ms) ) return TCL_ERROR; 17496f8a503dSdanielk1977 sqlite3_busy_timeout(pDb->db, ms); 17506d31316cSdrh break; 175175897234Sdrh } 1752b5a20d3cSdrh 17530f14e2ebSdrh /* 17547ddad969Sdanielk1977 ** $db soft_heap_limit N 17557ddad969Sdanielk1977 ** 17567ddad969Sdanielk1977 ** Set the soft-heap-limit for this thread. Note that the limit is 1757*aef0bf64Sdanielk1977 ** per-thread, not per-database. The previous limit is returned. 17587ddad969Sdanielk1977 */ 17597ddad969Sdanielk1977 case DB_SOFT_HEAP_LIMIT: { 17600190d1daSdanielk1977 #ifndef SQLITE_OMIT_MEMORY_MANAGEMENT 17617ddad969Sdanielk1977 int n; 1762*aef0bf64Sdanielk1977 int ret; 17637ddad969Sdanielk1977 if( objc!=3 ){ 17647ddad969Sdanielk1977 Tcl_WrongNumArgs(interp, 2, objv, "BYTES"); 17657ddad969Sdanielk1977 return TCL_ERROR; 17667ddad969Sdanielk1977 } 17677ddad969Sdanielk1977 if( Tcl_GetIntFromObj(interp, objv[2], &n) ){ 17687ddad969Sdanielk1977 return TCL_ERROR; 17697ddad969Sdanielk1977 } 1770*aef0bf64Sdanielk1977 ret = sqlite3Tsd()->nSoftHeapLimit; 17717ddad969Sdanielk1977 sqlite3_soft_heap_limit(n); 1772*aef0bf64Sdanielk1977 Tcl_SetObjResult(interp, Tcl_NewIntObj(ret)); 17730190d1daSdanielk1977 #endif 17740190d1daSdanielk1977 break; 17750190d1daSdanielk1977 } 17760190d1daSdanielk1977 17770190d1daSdanielk1977 /* 17780190d1daSdanielk1977 ** $db release_memory ?N? 17790190d1daSdanielk1977 ** 17800190d1daSdanielk1977 ** Try to release memory currently held (but not really required) by 17810190d1daSdanielk1977 ** SQLite database connections opened by the current thread. If an 17820190d1daSdanielk1977 ** integer argument is supplied, then SQLite stops trying to free memory 17830190d1daSdanielk1977 ** after N bytes have been freed. 17840190d1daSdanielk1977 ** 17850190d1daSdanielk1977 ** The value returned is the number of bytes actually freed. 17860190d1daSdanielk1977 **/ 17870190d1daSdanielk1977 case DB_RELEASE_MEMORY: { 17880190d1daSdanielk1977 int nRelease = 0; 17890190d1daSdanielk1977 int N = -1; 17900190d1daSdanielk1977 if( objc!=2 && objc!=3 ){ 17910190d1daSdanielk1977 Tcl_WrongNumArgs(interp, 2, objv, "?N?"); 17920190d1daSdanielk1977 return TCL_ERROR; 17930190d1daSdanielk1977 } 17940190d1daSdanielk1977 #ifndef SQLITE_OMIT_MEMORY_MANAGEMENT 17950190d1daSdanielk1977 if( objc==3 && TCL_OK!=Tcl_GetIntFromObj(interp, objv[2], &N) ){ 17960190d1daSdanielk1977 return TCL_ERROR; 17970190d1daSdanielk1977 } 17980190d1daSdanielk1977 nRelease = sqlite3_release_memory(N); 17990190d1daSdanielk1977 #endif 18000190d1daSdanielk1977 Tcl_SetObjResult(interp, Tcl_NewIntObj(nRelease)); 18017ddad969Sdanielk1977 break; 18027ddad969Sdanielk1977 } 18037ddad969Sdanielk1977 18047ddad969Sdanielk1977 /* 18050f14e2ebSdrh ** $db total_changes 18060f14e2ebSdrh ** 18070f14e2ebSdrh ** Return the number of rows that were modified, inserted, or deleted 18080f14e2ebSdrh ** since the database handle was created. 18090f14e2ebSdrh */ 18100f14e2ebSdrh case DB_TOTAL_CHANGES: { 18110f14e2ebSdrh Tcl_Obj *pResult; 18120f14e2ebSdrh if( objc!=2 ){ 18130f14e2ebSdrh Tcl_WrongNumArgs(interp, 2, objv, ""); 18140f14e2ebSdrh return TCL_ERROR; 18150f14e2ebSdrh } 18160f14e2ebSdrh pResult = Tcl_GetObjResult(interp); 18170f14e2ebSdrh Tcl_SetIntObj(pResult, sqlite3_total_changes(pDb->db)); 18180f14e2ebSdrh break; 18190f14e2ebSdrh } 18200f14e2ebSdrh 1821b5a20d3cSdrh /* $db trace ?CALLBACK? 1822b5a20d3cSdrh ** 1823b5a20d3cSdrh ** Make arrangements to invoke the CALLBACK routine for each SQL statement 1824b5a20d3cSdrh ** that is executed. The text of the SQL is appended to CALLBACK before 1825b5a20d3cSdrh ** it is executed. 1826b5a20d3cSdrh */ 1827b5a20d3cSdrh case DB_TRACE: { 1828b5a20d3cSdrh if( objc>3 ){ 1829b5a20d3cSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?"); 1830b97759edSdrh return TCL_ERROR; 1831b5a20d3cSdrh }else if( objc==2 ){ 1832b5a20d3cSdrh if( pDb->zTrace ){ 1833b5a20d3cSdrh Tcl_AppendResult(interp, pDb->zTrace, 0); 1834b5a20d3cSdrh } 1835b5a20d3cSdrh }else{ 1836b5a20d3cSdrh char *zTrace; 1837b5a20d3cSdrh int len; 1838b5a20d3cSdrh if( pDb->zTrace ){ 1839b5a20d3cSdrh Tcl_Free(pDb->zTrace); 1840b5a20d3cSdrh } 1841b5a20d3cSdrh zTrace = Tcl_GetStringFromObj(objv[2], &len); 1842b5a20d3cSdrh if( zTrace && len>0 ){ 1843b5a20d3cSdrh pDb->zTrace = Tcl_Alloc( len + 1 ); 1844b5a20d3cSdrh strcpy(pDb->zTrace, zTrace); 1845b5a20d3cSdrh }else{ 1846b5a20d3cSdrh pDb->zTrace = 0; 1847b5a20d3cSdrh } 184819e2d37fSdrh #ifndef SQLITE_OMIT_TRACE 1849b5a20d3cSdrh if( pDb->zTrace ){ 1850b5a20d3cSdrh pDb->interp = interp; 18516f8a503dSdanielk1977 sqlite3_trace(pDb->db, DbTraceHandler, pDb); 1852b5a20d3cSdrh }else{ 18536f8a503dSdanielk1977 sqlite3_trace(pDb->db, 0, 0); 1854b5a20d3cSdrh } 185519e2d37fSdrh #endif 1856b5a20d3cSdrh } 1857b5a20d3cSdrh break; 1858b5a20d3cSdrh } 1859b5a20d3cSdrh 18603d21423cSdrh /* $db transaction [-deferred|-immediate|-exclusive] SCRIPT 18613d21423cSdrh ** 18623d21423cSdrh ** Start a new transaction (if we are not already in the midst of a 18633d21423cSdrh ** transaction) and execute the TCL script SCRIPT. After SCRIPT 18643d21423cSdrh ** completes, either commit the transaction or roll it back if SCRIPT 18653d21423cSdrh ** throws an exception. Or if no new transation was started, do nothing. 18663d21423cSdrh ** pass the exception on up the stack. 18673d21423cSdrh ** 18683d21423cSdrh ** This command was inspired by Dave Thomas's talk on Ruby at the 18693d21423cSdrh ** 2005 O'Reilly Open Source Convention (OSCON). 18703d21423cSdrh */ 18713d21423cSdrh case DB_TRANSACTION: { 18723d21423cSdrh int inTrans; 18733d21423cSdrh Tcl_Obj *pScript; 18743d21423cSdrh const char *zBegin = "BEGIN"; 18753d21423cSdrh if( objc!=3 && objc!=4 ){ 18763d21423cSdrh Tcl_WrongNumArgs(interp, 2, objv, "[TYPE] SCRIPT"); 18773d21423cSdrh return TCL_ERROR; 18783d21423cSdrh } 18793d21423cSdrh if( objc==3 ){ 18803d21423cSdrh pScript = objv[2]; 18813d21423cSdrh } else { 18823d21423cSdrh static const char *TTYPE_strs[] = { 1883ce604012Sdrh "deferred", "exclusive", "immediate", 0 18843d21423cSdrh }; 18853d21423cSdrh enum TTYPE_enum { 18863d21423cSdrh TTYPE_DEFERRED, TTYPE_EXCLUSIVE, TTYPE_IMMEDIATE 18873d21423cSdrh }; 18883d21423cSdrh int ttype; 1889b5555e7eSdrh if( Tcl_GetIndexFromObj(interp, objv[2], TTYPE_strs, "transaction type", 18903d21423cSdrh 0, &ttype) ){ 18913d21423cSdrh return TCL_ERROR; 18923d21423cSdrh } 18933d21423cSdrh switch( (enum TTYPE_enum)ttype ){ 18943d21423cSdrh case TTYPE_DEFERRED: /* no-op */; break; 18953d21423cSdrh case TTYPE_EXCLUSIVE: zBegin = "BEGIN EXCLUSIVE"; break; 18963d21423cSdrh case TTYPE_IMMEDIATE: zBegin = "BEGIN IMMEDIATE"; break; 18973d21423cSdrh } 18983d21423cSdrh pScript = objv[3]; 18993d21423cSdrh } 19003d21423cSdrh inTrans = !sqlite3_get_autocommit(pDb->db); 19013d21423cSdrh if( !inTrans ){ 19023d21423cSdrh sqlite3_exec(pDb->db, zBegin, 0, 0, 0); 19033d21423cSdrh } 19043d21423cSdrh rc = Tcl_EvalObjEx(interp, pScript, 0); 19053d21423cSdrh if( !inTrans ){ 19063d21423cSdrh const char *zEnd; 19073d21423cSdrh if( rc==TCL_ERROR ){ 19083d21423cSdrh zEnd = "ROLLBACK"; 19093d21423cSdrh } else { 19103d21423cSdrh zEnd = "COMMIT"; 19113d21423cSdrh } 19123d21423cSdrh sqlite3_exec(pDb->db, zEnd, 0, 0, 0); 19133d21423cSdrh } 19143d21423cSdrh break; 19153d21423cSdrh } 19163d21423cSdrh 191794eb6a14Sdanielk1977 /* 191894eb6a14Sdanielk1977 ** $db update_hook ?script? 191971fd80bfSdanielk1977 ** $db rollback_hook ?script? 192094eb6a14Sdanielk1977 */ 192171fd80bfSdanielk1977 case DB_UPDATE_HOOK: 192271fd80bfSdanielk1977 case DB_ROLLBACK_HOOK: { 192371fd80bfSdanielk1977 192471fd80bfSdanielk1977 /* set ppHook to point at pUpdateHook or pRollbackHook, depending on 192571fd80bfSdanielk1977 ** whether [$db update_hook] or [$db rollback_hook] was invoked. 192671fd80bfSdanielk1977 */ 192771fd80bfSdanielk1977 Tcl_Obj **ppHook; 192871fd80bfSdanielk1977 if( choice==DB_UPDATE_HOOK ){ 192971fd80bfSdanielk1977 ppHook = &pDb->pUpdateHook; 193071fd80bfSdanielk1977 }else{ 193171fd80bfSdanielk1977 ppHook = &pDb->pRollbackHook; 193271fd80bfSdanielk1977 } 193371fd80bfSdanielk1977 193494eb6a14Sdanielk1977 if( objc!=2 && objc!=3 ){ 193594eb6a14Sdanielk1977 Tcl_WrongNumArgs(interp, 2, objv, "?SCRIPT?"); 193694eb6a14Sdanielk1977 return TCL_ERROR; 193794eb6a14Sdanielk1977 } 193871fd80bfSdanielk1977 if( *ppHook ){ 193971fd80bfSdanielk1977 Tcl_SetObjResult(interp, *ppHook); 194094eb6a14Sdanielk1977 if( objc==3 ){ 194171fd80bfSdanielk1977 Tcl_DecrRefCount(*ppHook); 194271fd80bfSdanielk1977 *ppHook = 0; 194394eb6a14Sdanielk1977 } 194494eb6a14Sdanielk1977 } 194594eb6a14Sdanielk1977 if( objc==3 ){ 194671fd80bfSdanielk1977 assert( !(*ppHook) ); 194794eb6a14Sdanielk1977 if( Tcl_GetCharLength(objv[2])>0 ){ 194871fd80bfSdanielk1977 *ppHook = objv[2]; 194971fd80bfSdanielk1977 Tcl_IncrRefCount(*ppHook); 195094eb6a14Sdanielk1977 } 195194eb6a14Sdanielk1977 } 195271fd80bfSdanielk1977 195371fd80bfSdanielk1977 sqlite3_update_hook(pDb->db, (pDb->pUpdateHook?DbUpdateHandler:0), pDb); 195471fd80bfSdanielk1977 sqlite3_rollback_hook(pDb->db,(pDb->pRollbackHook?DbRollbackHandler:0),pDb); 195571fd80bfSdanielk1977 195694eb6a14Sdanielk1977 break; 195794eb6a14Sdanielk1977 } 195894eb6a14Sdanielk1977 19594397de57Sdanielk1977 /* $db version 19604397de57Sdanielk1977 ** 19614397de57Sdanielk1977 ** Return the version string for this database. 19624397de57Sdanielk1977 */ 19634397de57Sdanielk1977 case DB_VERSION: { 19644397de57Sdanielk1977 Tcl_SetResult(interp, (char *)sqlite3_libversion(), TCL_STATIC); 19654397de57Sdanielk1977 break; 19664397de57Sdanielk1977 } 19674397de57Sdanielk1977 19681067fe11Stpoindex 19696d31316cSdrh } /* End of the SWITCH statement */ 197022fbcb8dSdrh return rc; 197175897234Sdrh } 197275897234Sdrh 197375897234Sdrh /* 19749bb575fdSdrh ** sqlite3 DBNAME FILENAME ?MODE? ?-key KEY? 197575897234Sdrh ** 197675897234Sdrh ** This is the main Tcl command. When the "sqlite" Tcl command is 197775897234Sdrh ** invoked, this routine runs to process that command. 197875897234Sdrh ** 197975897234Sdrh ** The first argument, DBNAME, is an arbitrary name for a new 198075897234Sdrh ** database connection. This command creates a new command named 198175897234Sdrh ** DBNAME that is used to control that connection. The database 198275897234Sdrh ** connection is deleted when the DBNAME command is deleted. 198375897234Sdrh ** 198475897234Sdrh ** The second argument is the name of the directory that contains 198575897234Sdrh ** the sqlite database that is to be accessed. 1986fbc3eab8Sdrh ** 1987fbc3eab8Sdrh ** For testing purposes, we also support the following: 1988fbc3eab8Sdrh ** 19899bb575fdSdrh ** sqlite3 -encoding 1990fbc3eab8Sdrh ** 1991fbc3eab8Sdrh ** Return the encoding used by LIKE and GLOB operators. Choices 1992fbc3eab8Sdrh ** are UTF-8 and iso8859. 1993fbc3eab8Sdrh ** 19949bb575fdSdrh ** sqlite3 -version 1995647cb0e1Sdrh ** 1996647cb0e1Sdrh ** Return the version number of the SQLite library. 1997647cb0e1Sdrh ** 19989bb575fdSdrh ** sqlite3 -tcl-uses-utf 1999fbc3eab8Sdrh ** 2000fbc3eab8Sdrh ** Return "1" if compiled with a Tcl uses UTF-8. Return "0" if 2001fbc3eab8Sdrh ** not. Used by tests to make sure the library was compiled 2002fbc3eab8Sdrh ** correctly. 200375897234Sdrh */ 200422fbcb8dSdrh static int DbMain(void *cd, Tcl_Interp *interp, int objc,Tcl_Obj *const*objv){ 2005bec3f402Sdrh SqliteDb *p; 200622fbcb8dSdrh void *pKey = 0; 200722fbcb8dSdrh int nKey = 0; 200822fbcb8dSdrh const char *zArg; 200975897234Sdrh char *zErrMsg; 201022fbcb8dSdrh const char *zFile; 201106b2718aSdrh char zBuf[80]; 201222fbcb8dSdrh if( objc==2 ){ 201322fbcb8dSdrh zArg = Tcl_GetStringFromObj(objv[1], 0); 201422fbcb8dSdrh if( strcmp(zArg,"-version")==0 ){ 20156f8a503dSdanielk1977 Tcl_AppendResult(interp,sqlite3_version,0); 2016647cb0e1Sdrh return TCL_OK; 2017647cb0e1Sdrh } 20189eb9e26bSdrh if( strcmp(zArg,"-has-codec")==0 ){ 20199eb9e26bSdrh #ifdef SQLITE_HAS_CODEC 202022fbcb8dSdrh Tcl_AppendResult(interp,"1",0); 202122fbcb8dSdrh #else 202222fbcb8dSdrh Tcl_AppendResult(interp,"0",0); 202322fbcb8dSdrh #endif 202422fbcb8dSdrh return TCL_OK; 202522fbcb8dSdrh } 202622fbcb8dSdrh if( strcmp(zArg,"-tcl-uses-utf")==0 ){ 2027fbc3eab8Sdrh #ifdef TCL_UTF_MAX 2028fbc3eab8Sdrh Tcl_AppendResult(interp,"1",0); 2029fbc3eab8Sdrh #else 2030fbc3eab8Sdrh Tcl_AppendResult(interp,"0",0); 2031fbc3eab8Sdrh #endif 2032fbc3eab8Sdrh return TCL_OK; 2033fbc3eab8Sdrh } 2034fbc3eab8Sdrh } 203522fbcb8dSdrh if( objc==5 || objc==6 ){ 203622fbcb8dSdrh zArg = Tcl_GetStringFromObj(objv[objc-2], 0); 203722fbcb8dSdrh if( strcmp(zArg,"-key")==0 ){ 203822fbcb8dSdrh pKey = Tcl_GetByteArrayFromObj(objv[objc-1], &nKey); 203922fbcb8dSdrh objc -= 2; 204022fbcb8dSdrh } 204122fbcb8dSdrh } 204222fbcb8dSdrh if( objc!=3 && objc!=4 ){ 204322fbcb8dSdrh Tcl_WrongNumArgs(interp, 1, objv, 20449eb9e26bSdrh #ifdef SQLITE_HAS_CODEC 20459eb9e26bSdrh "HANDLE FILENAME ?-key CODEC-KEY?" 204622fbcb8dSdrh #else 204722fbcb8dSdrh "HANDLE FILENAME ?MODE?" 204822fbcb8dSdrh #endif 204922fbcb8dSdrh ); 205075897234Sdrh return TCL_ERROR; 205175897234Sdrh } 205275897234Sdrh zErrMsg = 0; 20534cdc9e84Sdrh p = (SqliteDb*)Tcl_Alloc( sizeof(*p) ); 205475897234Sdrh if( p==0 ){ 2055bec3f402Sdrh Tcl_SetResult(interp, "malloc failed", TCL_STATIC); 2056bec3f402Sdrh return TCL_ERROR; 2057bec3f402Sdrh } 2058bec3f402Sdrh memset(p, 0, sizeof(*p)); 205922fbcb8dSdrh zFile = Tcl_GetStringFromObj(objv[2], 0); 20604f057f90Sdanielk1977 sqlite3_open(zFile, &p->db); 206180290863Sdanielk1977 if( SQLITE_OK!=sqlite3_errcode(p->db) ){ 206280290863Sdanielk1977 zErrMsg = strdup(sqlite3_errmsg(p->db)); 206380290863Sdanielk1977 sqlite3_close(p->db); 206480290863Sdanielk1977 p->db = 0; 206580290863Sdanielk1977 } 20662011d5f5Sdrh #ifdef SQLITE_HAS_CODEC 20672011d5f5Sdrh sqlite3_key(p->db, pKey, nKey); 2068eb8ed70dSdrh #endif 2069bec3f402Sdrh if( p->db==0 ){ 207075897234Sdrh Tcl_SetResult(interp, zErrMsg, TCL_VOLATILE); 2071bec3f402Sdrh Tcl_Free((char*)p); 207275897234Sdrh free(zErrMsg); 207375897234Sdrh return TCL_ERROR; 207475897234Sdrh } 2075fb7e7651Sdrh p->maxStmt = NUM_PREPARED_STMTS; 207622fbcb8dSdrh zArg = Tcl_GetStringFromObj(objv[1], 0); 207722fbcb8dSdrh Tcl_CreateObjCommand(interp, zArg, DbObjCmd, (char*)p, DbDeleteCmd); 2078c22bd47dSdrh 207906b2718aSdrh /* The return value is the value of the sqlite* pointer 208006b2718aSdrh */ 208106b2718aSdrh sprintf(zBuf, "%p", p->db); 20825e5377fbSdrh if( strncmp(zBuf,"0x",2) ){ 20835e5377fbSdrh sprintf(zBuf, "0x%p", p->db); 20845e5377fbSdrh } 208506b2718aSdrh Tcl_AppendResult(interp, zBuf, 0); 208606b2718aSdrh 2087c22bd47dSdrh /* If compiled with SQLITE_TEST turned on, then register the "md5sum" 208806b2718aSdrh ** SQL function. 2089c22bd47dSdrh */ 209028b4e489Sdrh #ifdef SQLITE_TEST 209128b4e489Sdrh { 20929bb575fdSdrh extern void Md5_Register(sqlite3*); 20933b93bc8cSdrh #ifdef SQLITE_MEMDEBUG 209413073931Sdanielk1977 int mallocfail = sqlite3_iMallocFail; 209513073931Sdanielk1977 sqlite3_iMallocFail = 0; 20965b59af85Sdanielk1977 #endif 209728b4e489Sdrh Md5_Register(p->db); 20983b93bc8cSdrh #ifdef SQLITE_MEMDEBUG 209913073931Sdanielk1977 sqlite3_iMallocFail = mallocfail; 21005b59af85Sdanielk1977 #endif 210128b4e489Sdrh } 210228b4e489Sdrh #endif 21037cedc8d4Sdanielk1977 p->interp = interp; 210475897234Sdrh return TCL_OK; 210575897234Sdrh } 210675897234Sdrh 210775897234Sdrh /* 210890ca9753Sdrh ** Provide a dummy Tcl_InitStubs if we are using this as a static 210990ca9753Sdrh ** library. 211090ca9753Sdrh */ 211190ca9753Sdrh #ifndef USE_TCL_STUBS 211290ca9753Sdrh # undef Tcl_InitStubs 211390ca9753Sdrh # define Tcl_InitStubs(a,b,c) 211490ca9753Sdrh #endif 211590ca9753Sdrh 211690ca9753Sdrh /* 211729bc4615Sdrh ** Make sure we have a PACKAGE_VERSION macro defined. This will be 211829bc4615Sdrh ** defined automatically by the TEA makefile. But other makefiles 211929bc4615Sdrh ** do not define it. 212029bc4615Sdrh */ 212129bc4615Sdrh #ifndef PACKAGE_VERSION 212229bc4615Sdrh # define PACKAGE_VERSION SQLITE_VERSION 212329bc4615Sdrh #endif 212429bc4615Sdrh 212529bc4615Sdrh /* 212675897234Sdrh ** Initialize this module. 212775897234Sdrh ** 212875897234Sdrh ** This Tcl module contains only a single new Tcl command named "sqlite". 212975897234Sdrh ** (Hence there is no namespace. There is no point in using a namespace 213075897234Sdrh ** if the extension only supplies one new name!) The "sqlite" command is 213175897234Sdrh ** used to open a new SQLite database. See the DbMain() routine above 213275897234Sdrh ** for additional information. 213375897234Sdrh */ 213429bc4615Sdrh EXTERN int Sqlite3_Init(Tcl_Interp *interp){ 213592febd92Sdrh Tcl_InitStubs(interp, "8.4", 0); 2136ef4ac8f9Sdrh Tcl_CreateObjCommand(interp, "sqlite3", (Tcl_ObjCmdProc*)DbMain, 0, 0); 213729bc4615Sdrh Tcl_PkgProvide(interp, "sqlite3", PACKAGE_VERSION); 213849766d6cSdrh Tcl_CreateObjCommand(interp, "sqlite", (Tcl_ObjCmdProc*)DbMain, 0, 0); 213929bc4615Sdrh Tcl_PkgProvide(interp, "sqlite", PACKAGE_VERSION); 214090ca9753Sdrh return TCL_OK; 214190ca9753Sdrh } 214229bc4615Sdrh EXTERN int Tclsqlite3_Init(Tcl_Interp *interp){ return Sqlite3_Init(interp); } 214329bc4615Sdrh EXTERN int Sqlite3_SafeInit(Tcl_Interp *interp){ return TCL_OK; } 214429bc4615Sdrh EXTERN int Tclsqlite3_SafeInit(Tcl_Interp *interp){ return TCL_OK; } 214549766d6cSdrh 214649766d6cSdrh #ifndef SQLITE_3_SUFFIX_ONLY 214729bc4615Sdrh EXTERN int Sqlite_Init(Tcl_Interp *interp){ return Sqlite3_Init(interp); } 214829bc4615Sdrh EXTERN int Tclsqlite_Init(Tcl_Interp *interp){ return Sqlite3_Init(interp); } 214929bc4615Sdrh EXTERN int Sqlite_SafeInit(Tcl_Interp *interp){ return TCL_OK; } 215029bc4615Sdrh EXTERN int Tclsqlite_SafeInit(Tcl_Interp *interp){ return TCL_OK; } 215149766d6cSdrh #endif 215275897234Sdrh 21533e27c026Sdrh #ifdef TCLSH 21543e27c026Sdrh /***************************************************************************** 21553e27c026Sdrh ** The code that follows is used to build standalone TCL interpreters 215675897234Sdrh */ 2157348784efSdrh 2158348784efSdrh /* 21593e27c026Sdrh ** If the macro TCLSH is one, then put in code this for the 21603e27c026Sdrh ** "main" routine that will initialize Tcl and take input from 21613e27c026Sdrh ** standard input. 2162348784efSdrh */ 21633e27c026Sdrh #if TCLSH==1 2164348784efSdrh static char zMainloop[] = 2165348784efSdrh "set line {}\n" 2166348784efSdrh "while {![eof stdin]} {\n" 2167348784efSdrh "if {$line!=\"\"} {\n" 2168348784efSdrh "puts -nonewline \"> \"\n" 2169348784efSdrh "} else {\n" 2170348784efSdrh "puts -nonewline \"% \"\n" 2171348784efSdrh "}\n" 2172348784efSdrh "flush stdout\n" 2173348784efSdrh "append line [gets stdin]\n" 2174348784efSdrh "if {[info complete $line]} {\n" 2175348784efSdrh "if {[catch {uplevel #0 $line} result]} {\n" 2176348784efSdrh "puts stderr \"Error: $result\"\n" 2177348784efSdrh "} elseif {$result!=\"\"} {\n" 2178348784efSdrh "puts $result\n" 2179348784efSdrh "}\n" 2180348784efSdrh "set line {}\n" 2181348784efSdrh "} else {\n" 2182348784efSdrh "append line \\n\n" 2183348784efSdrh "}\n" 2184348784efSdrh "}\n" 2185348784efSdrh ; 21863e27c026Sdrh #endif 21873e27c026Sdrh 21883e27c026Sdrh /* 21893e27c026Sdrh ** If the macro TCLSH is two, then get the main loop code out of 21903e27c026Sdrh ** the separate file "spaceanal_tcl.h". 21913e27c026Sdrh */ 21923e27c026Sdrh #if TCLSH==2 21933e27c026Sdrh static char zMainloop[] = 21943e27c026Sdrh #include "spaceanal_tcl.h" 21953e27c026Sdrh ; 21963e27c026Sdrh #endif 2197348784efSdrh 2198348784efSdrh #define TCLSH_MAIN main /* Needed to fake out mktclapp */ 2199348784efSdrh int TCLSH_MAIN(int argc, char **argv){ 2200348784efSdrh Tcl_Interp *interp; 2201297ecf14Sdrh Tcl_FindExecutable(argv[0]); 2202348784efSdrh interp = Tcl_CreateInterp(); 220338f8271fSdrh Sqlite3_Init(interp); 2204d9b0257aSdrh #ifdef SQLITE_TEST 2205d1bf3512Sdrh { 2206d1bf3512Sdrh extern int Sqlitetest1_Init(Tcl_Interp*); 22075c4d9703Sdrh extern int Sqlitetest2_Init(Tcl_Interp*); 22085c4d9703Sdrh extern int Sqlitetest3_Init(Tcl_Interp*); 2209a6064dcfSdrh extern int Sqlitetest4_Init(Tcl_Interp*); 2210998b56c3Sdanielk1977 extern int Sqlitetest5_Init(Tcl_Interp*); 22119c06c953Sdrh extern int Sqlitetest6_Init(Tcl_Interp*); 221213a68c3fSdanielk1977 extern int Sqlitetestasync_Init(Tcl_Interp*); 2213efc251daSdrh extern int Md5_Init(Tcl_Interp*); 22142e66f0b9Sdrh extern int Sqlitetestsse_Init(Tcl_Interp*); 22152e66f0b9Sdrh 22166490bebdSdanielk1977 Sqlitetest1_Init(interp); 22175c4d9703Sdrh Sqlitetest2_Init(interp); 2218de647130Sdrh Sqlitetest3_Init(interp); 2219fc57d7bfSdanielk1977 Sqlitetest4_Init(interp); 2220998b56c3Sdanielk1977 Sqlitetest5_Init(interp); 22219c06c953Sdrh Sqlitetest6_Init(interp); 222271fd80bfSdanielk1977 /* Sqlitetestasync_Init(interp); */ 2223efc251daSdrh Md5_Init(interp); 222489dec819Sdrh #ifdef SQLITE_SSE 22252e66f0b9Sdrh Sqlitetestsse_Init(interp); 22262e66f0b9Sdrh #endif 2227d1bf3512Sdrh } 2228d1bf3512Sdrh #endif 22293e27c026Sdrh if( argc>=2 || TCLSH==2 ){ 2230348784efSdrh int i; 2231348784efSdrh Tcl_SetVar(interp,"argv0",argv[1],TCL_GLOBAL_ONLY); 2232348784efSdrh Tcl_SetVar(interp,"argv", "", TCL_GLOBAL_ONLY); 223361212b69Sdrh for(i=3-TCLSH; i<argc; i++){ 2234348784efSdrh Tcl_SetVar(interp, "argv", argv[i], 2235348784efSdrh TCL_GLOBAL_ONLY | TCL_LIST_ELEMENT | TCL_APPEND_VALUE); 2236348784efSdrh } 22373e27c026Sdrh if( TCLSH==1 && Tcl_EvalFile(interp, argv[1])!=TCL_OK ){ 22380de8c112Sdrh const char *zInfo = Tcl_GetVar(interp, "errorInfo", TCL_GLOBAL_ONLY); 2239c61053b7Sdrh if( zInfo==0 ) zInfo = interp->result; 2240c61053b7Sdrh fprintf(stderr,"%s: %s\n", *argv, zInfo); 2241348784efSdrh return 1; 2242348784efSdrh } 22433e27c026Sdrh } 22443e27c026Sdrh if( argc<=1 || TCLSH==2 ){ 2245348784efSdrh Tcl_GlobalEval(interp, zMainloop); 2246348784efSdrh } 2247348784efSdrh return 0; 2248348784efSdrh } 2249348784efSdrh #endif /* TCLSH */ 22506d31316cSdrh 22516d31316cSdrh #endif /* !defined(NO_TCL) */ 2252