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