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*9ee3cdcbSdrh ** $Id: tclsqlite.c,v 1.111 2004/12/17 20:48:06 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> 2475897234Sdrh 2575897234Sdrh /* 2698808babSdrh ** If TCL uses UTF-8 and SQLite is configured to use iso8859, then we 2798808babSdrh ** have to do a translation when going between the two. Set the 2898808babSdrh ** UTF_TRANSLATION_NEEDED macro to indicate that we need to do 2998808babSdrh ** this translation. 3098808babSdrh */ 3198808babSdrh #if defined(TCL_UTF_MAX) && !defined(SQLITE_UTF8) 3298808babSdrh # define UTF_TRANSLATION_NEEDED 1 3398808babSdrh #endif 3498808babSdrh 3598808babSdrh /* 36cabb0819Sdrh ** New SQL functions can be created as TCL scripts. Each such function 37cabb0819Sdrh ** is described by an instance of the following structure. 38cabb0819Sdrh */ 39cabb0819Sdrh typedef struct SqlFunc SqlFunc; 40cabb0819Sdrh struct SqlFunc { 41cabb0819Sdrh Tcl_Interp *interp; /* The TCL interpret to execute the function */ 42cabb0819Sdrh char *zScript; /* The script to be run */ 43cabb0819Sdrh SqlFunc *pNext; /* Next function on the list of them all */ 44cabb0819Sdrh }; 45cabb0819Sdrh 46cabb0819Sdrh /* 470202b29eSdanielk1977 ** New collation sequences function can be created as TCL scripts. Each such 480202b29eSdanielk1977 ** function is described by an instance of the following structure. 490202b29eSdanielk1977 */ 500202b29eSdanielk1977 typedef struct SqlCollate SqlCollate; 510202b29eSdanielk1977 struct SqlCollate { 520202b29eSdanielk1977 Tcl_Interp *interp; /* The TCL interpret to execute the function */ 530202b29eSdanielk1977 char *zScript; /* The script to be run */ 540202b29eSdanielk1977 SqlCollate *pNext; /* Next function on the list of them all */ 550202b29eSdanielk1977 }; 560202b29eSdanielk1977 570202b29eSdanielk1977 /* 58bec3f402Sdrh ** There is one instance of this structure for each SQLite database 59bec3f402Sdrh ** that has been opened by the SQLite TCL interface. 60bec3f402Sdrh */ 61bec3f402Sdrh typedef struct SqliteDb SqliteDb; 62bec3f402Sdrh struct SqliteDb { 63895d7472Sdrh sqlite3 *db; /* The "real" database structure */ 64bec3f402Sdrh Tcl_Interp *interp; /* The interpreter used for this database */ 656d31316cSdrh char *zBusy; /* The busy callback routine */ 66aa940eacSdrh char *zCommit; /* The commit hook callback routine */ 67b5a20d3cSdrh char *zTrace; /* The trace callback routine */ 68348bb5d6Sdanielk1977 char *zProgress; /* The progress callback routine */ 69e22a334bSdrh char *zAuth; /* The authorization callback routine */ 70cabb0819Sdrh SqlFunc *pFunc; /* List of SQL functions */ 710202b29eSdanielk1977 SqlCollate *pCollate; /* List of SQL collation functions */ 726f8a503dSdanielk1977 int rc; /* Return code of most recent sqlite3_exec() */ 737cedc8d4Sdanielk1977 Tcl_Obj *pCollateNeeded; /* Collation needed script */ 7498808babSdrh }; 75297ecf14Sdrh 766d31316cSdrh /* 77895d7472Sdrh ** TCL calls this procedure when an sqlite3 database command is 78895d7472Sdrh ** deleted. 7975897234Sdrh */ 8075897234Sdrh static void DbDeleteCmd(void *db){ 81bec3f402Sdrh SqliteDb *pDb = (SqliteDb*)db; 826f8a503dSdanielk1977 sqlite3_close(pDb->db); 83cabb0819Sdrh while( pDb->pFunc ){ 84cabb0819Sdrh SqlFunc *pFunc = pDb->pFunc; 85cabb0819Sdrh pDb->pFunc = pFunc->pNext; 86cabb0819Sdrh Tcl_Free((char*)pFunc); 87cabb0819Sdrh } 880202b29eSdanielk1977 while( pDb->pCollate ){ 890202b29eSdanielk1977 SqlCollate *pCollate = pDb->pCollate; 900202b29eSdanielk1977 pDb->pCollate = pCollate->pNext; 910202b29eSdanielk1977 Tcl_Free((char*)pCollate); 920202b29eSdanielk1977 } 93bec3f402Sdrh if( pDb->zBusy ){ 94bec3f402Sdrh Tcl_Free(pDb->zBusy); 95bec3f402Sdrh } 96b5a20d3cSdrh if( pDb->zTrace ){ 97b5a20d3cSdrh Tcl_Free(pDb->zTrace); 980d1a643aSdrh } 99e22a334bSdrh if( pDb->zAuth ){ 100e22a334bSdrh Tcl_Free(pDb->zAuth); 101e22a334bSdrh } 102bec3f402Sdrh Tcl_Free((char*)pDb); 103bec3f402Sdrh } 104bec3f402Sdrh 105bec3f402Sdrh /* 106bec3f402Sdrh ** This routine is called when a database file is locked while trying 107bec3f402Sdrh ** to execute SQL. 108bec3f402Sdrh */ 1092a764eb0Sdanielk1977 static int DbBusyHandler(void *cd, int nTries){ 110bec3f402Sdrh SqliteDb *pDb = (SqliteDb*)cd; 111bec3f402Sdrh int rc; 112bec3f402Sdrh char zVal[30]; 113bec3f402Sdrh char *zCmd; 114bec3f402Sdrh Tcl_DString cmd; 115bec3f402Sdrh 116bec3f402Sdrh Tcl_DStringInit(&cmd); 117bec3f402Sdrh Tcl_DStringAppend(&cmd, pDb->zBusy, -1); 118bec3f402Sdrh sprintf(zVal, "%d", nTries); 1192a764eb0Sdanielk1977 Tcl_DStringAppendElement(&cmd, zVal); 120bec3f402Sdrh zCmd = Tcl_DStringValue(&cmd); 121bec3f402Sdrh rc = Tcl_Eval(pDb->interp, zCmd); 122bec3f402Sdrh Tcl_DStringFree(&cmd); 123bec3f402Sdrh if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){ 124bec3f402Sdrh return 0; 125bec3f402Sdrh } 126bec3f402Sdrh return 1; 12775897234Sdrh } 12875897234Sdrh 12975897234Sdrh /* 130348bb5d6Sdanielk1977 ** This routine is invoked as the 'progress callback' for the database. 131348bb5d6Sdanielk1977 */ 132348bb5d6Sdanielk1977 static int DbProgressHandler(void *cd){ 133348bb5d6Sdanielk1977 SqliteDb *pDb = (SqliteDb*)cd; 134348bb5d6Sdanielk1977 int rc; 135348bb5d6Sdanielk1977 136348bb5d6Sdanielk1977 assert( pDb->zProgress ); 137348bb5d6Sdanielk1977 rc = Tcl_Eval(pDb->interp, pDb->zProgress); 138348bb5d6Sdanielk1977 if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){ 139348bb5d6Sdanielk1977 return 1; 140348bb5d6Sdanielk1977 } 141348bb5d6Sdanielk1977 return 0; 142348bb5d6Sdanielk1977 } 143348bb5d6Sdanielk1977 144348bb5d6Sdanielk1977 /* 145b5a20d3cSdrh ** This routine is called by the SQLite trace handler whenever a new 146b5a20d3cSdrh ** block of SQL is executed. The TCL script in pDb->zTrace is executed. 1470d1a643aSdrh */ 148b5a20d3cSdrh static void DbTraceHandler(void *cd, const char *zSql){ 1490d1a643aSdrh SqliteDb *pDb = (SqliteDb*)cd; 150b5a20d3cSdrh Tcl_DString str; 1510d1a643aSdrh 152b5a20d3cSdrh Tcl_DStringInit(&str); 153b5a20d3cSdrh Tcl_DStringAppend(&str, pDb->zTrace, -1); 154b5a20d3cSdrh Tcl_DStringAppendElement(&str, zSql); 155b5a20d3cSdrh Tcl_Eval(pDb->interp, Tcl_DStringValue(&str)); 156b5a20d3cSdrh Tcl_DStringFree(&str); 157b5a20d3cSdrh Tcl_ResetResult(pDb->interp); 1580d1a643aSdrh } 1590d1a643aSdrh 1600d1a643aSdrh /* 161aa940eacSdrh ** This routine is called when a transaction is committed. The 162aa940eacSdrh ** TCL script in pDb->zCommit is executed. If it returns non-zero or 163aa940eacSdrh ** if it throws an exception, the transaction is rolled back instead 164aa940eacSdrh ** of being committed. 165aa940eacSdrh */ 166aa940eacSdrh static int DbCommitHandler(void *cd){ 167aa940eacSdrh SqliteDb *pDb = (SqliteDb*)cd; 168aa940eacSdrh int rc; 169aa940eacSdrh 170aa940eacSdrh rc = Tcl_Eval(pDb->interp, pDb->zCommit); 171aa940eacSdrh if( rc!=TCL_OK || atoi(Tcl_GetStringResult(pDb->interp)) ){ 172aa940eacSdrh return 1; 173aa940eacSdrh } 174aa940eacSdrh return 0; 175aa940eacSdrh } 176aa940eacSdrh 1777cedc8d4Sdanielk1977 static void tclCollateNeeded( 1787cedc8d4Sdanielk1977 void *pCtx, 1799bb575fdSdrh sqlite3 *db, 1807cedc8d4Sdanielk1977 int enc, 1817cedc8d4Sdanielk1977 const char *zName 1827cedc8d4Sdanielk1977 ){ 1837cedc8d4Sdanielk1977 SqliteDb *pDb = (SqliteDb *)pCtx; 1847cedc8d4Sdanielk1977 Tcl_Obj *pScript = Tcl_DuplicateObj(pDb->pCollateNeeded); 1857cedc8d4Sdanielk1977 Tcl_IncrRefCount(pScript); 1867cedc8d4Sdanielk1977 Tcl_ListObjAppendElement(0, pScript, Tcl_NewStringObj(zName, -1)); 1877cedc8d4Sdanielk1977 Tcl_EvalObjEx(pDb->interp, pScript, 0); 1887cedc8d4Sdanielk1977 Tcl_DecrRefCount(pScript); 1897cedc8d4Sdanielk1977 } 1907cedc8d4Sdanielk1977 191aa940eacSdrh /* 1920202b29eSdanielk1977 ** This routine is called to evaluate an SQL collation function implemented 1930202b29eSdanielk1977 ** using TCL script. 1940202b29eSdanielk1977 */ 1950202b29eSdanielk1977 static int tclSqlCollate( 1960202b29eSdanielk1977 void *pCtx, 1970202b29eSdanielk1977 int nA, 1980202b29eSdanielk1977 const void *zA, 1990202b29eSdanielk1977 int nB, 2000202b29eSdanielk1977 const void *zB 2010202b29eSdanielk1977 ){ 2020202b29eSdanielk1977 SqlCollate *p = (SqlCollate *)pCtx; 2030202b29eSdanielk1977 Tcl_Obj *pCmd; 2040202b29eSdanielk1977 2050202b29eSdanielk1977 pCmd = Tcl_NewStringObj(p->zScript, -1); 2060202b29eSdanielk1977 Tcl_IncrRefCount(pCmd); 2070202b29eSdanielk1977 Tcl_ListObjAppendElement(p->interp, pCmd, Tcl_NewStringObj(zA, nA)); 2080202b29eSdanielk1977 Tcl_ListObjAppendElement(p->interp, pCmd, Tcl_NewStringObj(zB, nB)); 2090202b29eSdanielk1977 Tcl_EvalObjEx(p->interp, pCmd, 0); 2100202b29eSdanielk1977 Tcl_DecrRefCount(pCmd); 2110202b29eSdanielk1977 return (atoi(Tcl_GetStringResult(p->interp))); 2120202b29eSdanielk1977 } 2130202b29eSdanielk1977 2140202b29eSdanielk1977 /* 215cabb0819Sdrh ** This routine is called to evaluate an SQL function implemented 216cabb0819Sdrh ** using TCL script. 217cabb0819Sdrh */ 2180ae8b831Sdanielk1977 static void tclSqlFunc(sqlite3_context *context, int argc, sqlite3_value **argv){ 2196f8a503dSdanielk1977 SqlFunc *p = sqlite3_user_data(context); 220cabb0819Sdrh Tcl_DString cmd; 221cabb0819Sdrh int i; 222cabb0819Sdrh int rc; 223cabb0819Sdrh 224cabb0819Sdrh Tcl_DStringInit(&cmd); 225cabb0819Sdrh Tcl_DStringAppend(&cmd, p->zScript, -1); 226cabb0819Sdrh for(i=0; i<argc; i++){ 2279c054830Sdrh if( SQLITE_NULL==sqlite3_value_type(argv[i]) ){ 22851ad0ecdSdanielk1977 Tcl_DStringAppendElement(&cmd, ""); 22951ad0ecdSdanielk1977 }else{ 2304f26d6c4Sdrh Tcl_DStringAppendElement(&cmd, sqlite3_value_text(argv[i])); 23151ad0ecdSdanielk1977 } 232cabb0819Sdrh } 233cabb0819Sdrh rc = Tcl_Eval(p->interp, Tcl_DStringValue(&cmd)); 234cabb0819Sdrh if( rc ){ 2357e18c259Sdanielk1977 sqlite3_result_error(context, Tcl_GetStringResult(p->interp), -1); 236cabb0819Sdrh }else{ 237d8123366Sdanielk1977 sqlite3_result_text(context, Tcl_GetStringResult(p->interp), -1, 238d8123366Sdanielk1977 SQLITE_TRANSIENT); 239cabb0819Sdrh } 240cabb0819Sdrh } 241895d7472Sdrh 242e22a334bSdrh #ifndef SQLITE_OMIT_AUTHORIZATION 243e22a334bSdrh /* 244e22a334bSdrh ** This is the authentication function. It appends the authentication 245e22a334bSdrh ** type code and the two arguments to zCmd[] then invokes the result 246e22a334bSdrh ** on the interpreter. The reply is examined to determine if the 247e22a334bSdrh ** authentication fails or succeeds. 248e22a334bSdrh */ 249e22a334bSdrh static int auth_callback( 250e22a334bSdrh void *pArg, 251e22a334bSdrh int code, 252e22a334bSdrh const char *zArg1, 253e22a334bSdrh const char *zArg2, 254e22a334bSdrh const char *zArg3, 255e22a334bSdrh const char *zArg4 256e22a334bSdrh ){ 257e22a334bSdrh char *zCode; 258e22a334bSdrh Tcl_DString str; 259e22a334bSdrh int rc; 260e22a334bSdrh const char *zReply; 261e22a334bSdrh SqliteDb *pDb = (SqliteDb*)pArg; 262e22a334bSdrh 263e22a334bSdrh switch( code ){ 264e22a334bSdrh case SQLITE_COPY : zCode="SQLITE_COPY"; break; 265e22a334bSdrh case SQLITE_CREATE_INDEX : zCode="SQLITE_CREATE_INDEX"; break; 266e22a334bSdrh case SQLITE_CREATE_TABLE : zCode="SQLITE_CREATE_TABLE"; break; 267e22a334bSdrh case SQLITE_CREATE_TEMP_INDEX : zCode="SQLITE_CREATE_TEMP_INDEX"; break; 268e22a334bSdrh case SQLITE_CREATE_TEMP_TABLE : zCode="SQLITE_CREATE_TEMP_TABLE"; break; 269e22a334bSdrh case SQLITE_CREATE_TEMP_TRIGGER: zCode="SQLITE_CREATE_TEMP_TRIGGER"; break; 270e22a334bSdrh case SQLITE_CREATE_TEMP_VIEW : zCode="SQLITE_CREATE_TEMP_VIEW"; break; 271e22a334bSdrh case SQLITE_CREATE_TRIGGER : zCode="SQLITE_CREATE_TRIGGER"; break; 272e22a334bSdrh case SQLITE_CREATE_VIEW : zCode="SQLITE_CREATE_VIEW"; break; 273e22a334bSdrh case SQLITE_DELETE : zCode="SQLITE_DELETE"; break; 274e22a334bSdrh case SQLITE_DROP_INDEX : zCode="SQLITE_DROP_INDEX"; break; 275e22a334bSdrh case SQLITE_DROP_TABLE : zCode="SQLITE_DROP_TABLE"; break; 276e22a334bSdrh case SQLITE_DROP_TEMP_INDEX : zCode="SQLITE_DROP_TEMP_INDEX"; break; 277e22a334bSdrh case SQLITE_DROP_TEMP_TABLE : zCode="SQLITE_DROP_TEMP_TABLE"; break; 278e22a334bSdrh case SQLITE_DROP_TEMP_TRIGGER : zCode="SQLITE_DROP_TEMP_TRIGGER"; break; 279e22a334bSdrh case SQLITE_DROP_TEMP_VIEW : zCode="SQLITE_DROP_TEMP_VIEW"; break; 280e22a334bSdrh case SQLITE_DROP_TRIGGER : zCode="SQLITE_DROP_TRIGGER"; break; 281e22a334bSdrh case SQLITE_DROP_VIEW : zCode="SQLITE_DROP_VIEW"; break; 282e22a334bSdrh case SQLITE_INSERT : zCode="SQLITE_INSERT"; break; 283e22a334bSdrh case SQLITE_PRAGMA : zCode="SQLITE_PRAGMA"; break; 284e22a334bSdrh case SQLITE_READ : zCode="SQLITE_READ"; break; 285e22a334bSdrh case SQLITE_SELECT : zCode="SQLITE_SELECT"; break; 286e22a334bSdrh case SQLITE_TRANSACTION : zCode="SQLITE_TRANSACTION"; break; 287e22a334bSdrh case SQLITE_UPDATE : zCode="SQLITE_UPDATE"; break; 28881e293b4Sdrh case SQLITE_ATTACH : zCode="SQLITE_ATTACH"; break; 28981e293b4Sdrh case SQLITE_DETACH : zCode="SQLITE_DETACH"; break; 2901c8c23ccSdanielk1977 case SQLITE_ALTER_TABLE : zCode="SQLITE_ALTER_TABLE"; break; 2911d54df88Sdanielk1977 case SQLITE_REINDEX : zCode="SQLITE_REINDEX"; break; 292e22a334bSdrh default : zCode="????"; break; 293e22a334bSdrh } 294e22a334bSdrh Tcl_DStringInit(&str); 295e22a334bSdrh Tcl_DStringAppend(&str, pDb->zAuth, -1); 296e22a334bSdrh Tcl_DStringAppendElement(&str, zCode); 297e22a334bSdrh Tcl_DStringAppendElement(&str, zArg1 ? zArg1 : ""); 298e22a334bSdrh Tcl_DStringAppendElement(&str, zArg2 ? zArg2 : ""); 299e22a334bSdrh Tcl_DStringAppendElement(&str, zArg3 ? zArg3 : ""); 300e22a334bSdrh Tcl_DStringAppendElement(&str, zArg4 ? zArg4 : ""); 301e22a334bSdrh rc = Tcl_GlobalEval(pDb->interp, Tcl_DStringValue(&str)); 302e22a334bSdrh Tcl_DStringFree(&str); 303e22a334bSdrh zReply = Tcl_GetStringResult(pDb->interp); 304e22a334bSdrh if( strcmp(zReply,"SQLITE_OK")==0 ){ 305e22a334bSdrh rc = SQLITE_OK; 306e22a334bSdrh }else if( strcmp(zReply,"SQLITE_DENY")==0 ){ 307e22a334bSdrh rc = SQLITE_DENY; 308e22a334bSdrh }else if( strcmp(zReply,"SQLITE_IGNORE")==0 ){ 309e22a334bSdrh rc = SQLITE_IGNORE; 310e22a334bSdrh }else{ 311e22a334bSdrh rc = 999; 312e22a334bSdrh } 313e22a334bSdrh return rc; 314e22a334bSdrh } 315e22a334bSdrh #endif /* SQLITE_OMIT_AUTHORIZATION */ 316cabb0819Sdrh 317cabb0819Sdrh /* 318ef2cb63eSdanielk1977 ** zText is a pointer to text obtained via an sqlite3_result_text() 319ef2cb63eSdanielk1977 ** or similar interface. This routine returns a Tcl string object, 320ef2cb63eSdanielk1977 ** reference count set to 0, containing the text. If a translation 321ef2cb63eSdanielk1977 ** between iso8859 and UTF-8 is required, it is preformed. 322ef2cb63eSdanielk1977 */ 323ef2cb63eSdanielk1977 static Tcl_Obj *dbTextToObj(char const *zText){ 324ef2cb63eSdanielk1977 Tcl_Obj *pVal; 325ef2cb63eSdanielk1977 #ifdef UTF_TRANSLATION_NEEDED 326ef2cb63eSdanielk1977 Tcl_DString dCol; 327ef2cb63eSdanielk1977 Tcl_DStringInit(&dCol); 328ef2cb63eSdanielk1977 Tcl_ExternalToUtfDString(NULL, zText, -1, &dCol); 329ef2cb63eSdanielk1977 pVal = Tcl_NewStringObj(Tcl_DStringValue(&dCol), -1); 330ef2cb63eSdanielk1977 Tcl_DStringFree(&dCol); 331ef2cb63eSdanielk1977 #else 332ef2cb63eSdanielk1977 pVal = Tcl_NewStringObj(zText, -1); 333ef2cb63eSdanielk1977 #endif 334ef2cb63eSdanielk1977 return pVal; 335ef2cb63eSdanielk1977 } 336ef2cb63eSdanielk1977 337ef2cb63eSdanielk1977 /* 3381067fe11Stpoindex ** This routine reads a line of text from FILE in, stores 3391067fe11Stpoindex ** the text in memory obtained from malloc() and returns a pointer 3401067fe11Stpoindex ** to the text. NULL is returned at end of file, or if malloc() 3411067fe11Stpoindex ** fails. 3421067fe11Stpoindex ** 3431067fe11Stpoindex ** The interface is like "readline" but no command-line editing 3441067fe11Stpoindex ** is done. 3451067fe11Stpoindex ** 3461067fe11Stpoindex ** copied from shell.c from '.import' command 3471067fe11Stpoindex */ 3481067fe11Stpoindex static char *local_getline(char *zPrompt, FILE *in){ 3491067fe11Stpoindex char *zLine; 3501067fe11Stpoindex int nLine; 3511067fe11Stpoindex int n; 3521067fe11Stpoindex int eol; 3531067fe11Stpoindex 3541067fe11Stpoindex nLine = 100; 3551067fe11Stpoindex zLine = malloc( nLine ); 3561067fe11Stpoindex if( zLine==0 ) return 0; 3571067fe11Stpoindex n = 0; 3581067fe11Stpoindex eol = 0; 3591067fe11Stpoindex while( !eol ){ 3601067fe11Stpoindex if( n+100>nLine ){ 3611067fe11Stpoindex nLine = nLine*2 + 100; 3621067fe11Stpoindex zLine = realloc(zLine, nLine); 3631067fe11Stpoindex if( zLine==0 ) return 0; 3641067fe11Stpoindex } 3651067fe11Stpoindex if( fgets(&zLine[n], nLine - n, in)==0 ){ 3661067fe11Stpoindex if( n==0 ){ 3671067fe11Stpoindex free(zLine); 3681067fe11Stpoindex return 0; 3691067fe11Stpoindex } 3701067fe11Stpoindex zLine[n] = 0; 3711067fe11Stpoindex eol = 1; 3721067fe11Stpoindex break; 3731067fe11Stpoindex } 3741067fe11Stpoindex while( zLine[n] ){ n++; } 3751067fe11Stpoindex if( n>0 && zLine[n-1]=='\n' ){ 3761067fe11Stpoindex n--; 3771067fe11Stpoindex zLine[n] = 0; 3781067fe11Stpoindex eol = 1; 3791067fe11Stpoindex } 3801067fe11Stpoindex } 3811067fe11Stpoindex zLine = realloc( zLine, n+1 ); 3821067fe11Stpoindex return zLine; 3831067fe11Stpoindex } 3841067fe11Stpoindex 3851067fe11Stpoindex /* 38675897234Sdrh ** The "sqlite" command below creates a new Tcl command for each 38775897234Sdrh ** connection it opens to an SQLite database. This routine is invoked 38875897234Sdrh ** whenever one of those connection-specific commands is executed 38975897234Sdrh ** in Tcl. For example, if you run Tcl code like this: 39075897234Sdrh ** 3919bb575fdSdrh ** sqlite3 db1 "my_database" 39275897234Sdrh ** db1 close 39375897234Sdrh ** 39475897234Sdrh ** The first command opens a connection to the "my_database" database 39575897234Sdrh ** and calls that connection "db1". The second command causes this 39675897234Sdrh ** subroutine to be invoked. 39775897234Sdrh */ 3986d31316cSdrh static int DbObjCmd(void *cd, Tcl_Interp *interp, int objc,Tcl_Obj *const*objv){ 399bec3f402Sdrh SqliteDb *pDb = (SqliteDb*)cd; 4006d31316cSdrh int choice; 40122fbcb8dSdrh int rc = TCL_OK; 4020de8c112Sdrh static const char *DB_strs[] = { 403b5a20d3cSdrh "authorizer", "busy", "changes", 4040f14e2ebSdrh "close", "collate", "collation_needed", 4051067fe11Stpoindex "commit_hook", "complete", "copy", 4061067fe11Stpoindex "errorcode", "eval", "function", 4071067fe11Stpoindex "last_insert_rowid", "onecolumn", "progress", 4081067fe11Stpoindex "rekey", "timeout", "total_changes", 4091067fe11Stpoindex "trace", 4100f14e2ebSdrh 0 4116d31316cSdrh }; 412411995dcSdrh enum DB_enum { 413b5a20d3cSdrh DB_AUTHORIZER, DB_BUSY, DB_CHANGES, 4140f14e2ebSdrh DB_CLOSE, DB_COLLATE, DB_COLLATION_NEEDED, 4151067fe11Stpoindex DB_COMMIT_HOOK, DB_COMPLETE, DB_COPY, 4161067fe11Stpoindex DB_ERRORCODE, DB_EVAL, DB_FUNCTION, 4171067fe11Stpoindex DB_LAST_INSERT_ROWID, DB_ONECOLUMN, DB_PROGRESS, 4181067fe11Stpoindex DB_REKEY, DB_TIMEOUT, DB_TOTAL_CHANGES, 4191067fe11Stpoindex DB_TRACE 4206d31316cSdrh }; 4211067fe11Stpoindex /* don't leave trailing commas on DB_enum, it confuses the AIX xlc compiler */ 4226d31316cSdrh 4236d31316cSdrh if( objc<2 ){ 4246d31316cSdrh Tcl_WrongNumArgs(interp, 1, objv, "SUBCOMMAND ..."); 42575897234Sdrh return TCL_ERROR; 42675897234Sdrh } 427411995dcSdrh if( Tcl_GetIndexFromObj(interp, objv[1], DB_strs, "option", 0, &choice) ){ 4286d31316cSdrh return TCL_ERROR; 4296d31316cSdrh } 4306d31316cSdrh 431411995dcSdrh switch( (enum DB_enum)choice ){ 43275897234Sdrh 433e22a334bSdrh /* $db authorizer ?CALLBACK? 434e22a334bSdrh ** 435e22a334bSdrh ** Invoke the given callback to authorize each SQL operation as it is 436e22a334bSdrh ** compiled. 5 arguments are appended to the callback before it is 437e22a334bSdrh ** invoked: 438e22a334bSdrh ** 439e22a334bSdrh ** (1) The authorization type (ex: SQLITE_CREATE_TABLE, SQLITE_INSERT, ...) 440e22a334bSdrh ** (2) First descriptive name (depends on authorization type) 441e22a334bSdrh ** (3) Second descriptive name 442e22a334bSdrh ** (4) Name of the database (ex: "main", "temp") 443e22a334bSdrh ** (5) Name of trigger that is doing the access 444e22a334bSdrh ** 445e22a334bSdrh ** The callback should return on of the following strings: SQLITE_OK, 446e22a334bSdrh ** SQLITE_IGNORE, or SQLITE_DENY. Any other return value is an error. 447e22a334bSdrh ** 448e22a334bSdrh ** If this method is invoked with no arguments, the current authorization 449e22a334bSdrh ** callback string is returned. 450e22a334bSdrh */ 451e22a334bSdrh case DB_AUTHORIZER: { 4521211de37Sdrh #ifdef SQLITE_OMIT_AUTHORIZATION 4531211de37Sdrh Tcl_AppendResult(interp, "authorization not available in this build", 0); 4541211de37Sdrh return TCL_ERROR; 4551211de37Sdrh #else 456e22a334bSdrh if( objc>3 ){ 457e22a334bSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?"); 4580f14e2ebSdrh return TCL_ERROR; 459e22a334bSdrh }else if( objc==2 ){ 460b5a20d3cSdrh if( pDb->zAuth ){ 461e22a334bSdrh Tcl_AppendResult(interp, pDb->zAuth, 0); 462e22a334bSdrh } 463e22a334bSdrh }else{ 464e22a334bSdrh char *zAuth; 465e22a334bSdrh int len; 466e22a334bSdrh if( pDb->zAuth ){ 467e22a334bSdrh Tcl_Free(pDb->zAuth); 468e22a334bSdrh } 469e22a334bSdrh zAuth = Tcl_GetStringFromObj(objv[2], &len); 470e22a334bSdrh if( zAuth && len>0 ){ 471e22a334bSdrh pDb->zAuth = Tcl_Alloc( len + 1 ); 472e22a334bSdrh strcpy(pDb->zAuth, zAuth); 473e22a334bSdrh }else{ 474e22a334bSdrh pDb->zAuth = 0; 475e22a334bSdrh } 476e22a334bSdrh if( pDb->zAuth ){ 477e22a334bSdrh pDb->interp = interp; 4786f8a503dSdanielk1977 sqlite3_set_authorizer(pDb->db, auth_callback, pDb); 479e22a334bSdrh }else{ 4806f8a503dSdanielk1977 sqlite3_set_authorizer(pDb->db, 0, 0); 481e22a334bSdrh } 482e22a334bSdrh } 4831211de37Sdrh #endif 484e22a334bSdrh break; 485e22a334bSdrh } 486e22a334bSdrh 487bec3f402Sdrh /* $db busy ?CALLBACK? 488bec3f402Sdrh ** 489bec3f402Sdrh ** Invoke the given callback if an SQL statement attempts to open 490bec3f402Sdrh ** a locked database file. 491bec3f402Sdrh */ 4926d31316cSdrh case DB_BUSY: { 4936d31316cSdrh if( objc>3 ){ 4946d31316cSdrh Tcl_WrongNumArgs(interp, 2, objv, "CALLBACK"); 495bec3f402Sdrh return TCL_ERROR; 4966d31316cSdrh }else if( objc==2 ){ 497bec3f402Sdrh if( pDb->zBusy ){ 498bec3f402Sdrh Tcl_AppendResult(interp, pDb->zBusy, 0); 499bec3f402Sdrh } 500bec3f402Sdrh }else{ 5016d31316cSdrh char *zBusy; 5026d31316cSdrh int len; 503bec3f402Sdrh if( pDb->zBusy ){ 504bec3f402Sdrh Tcl_Free(pDb->zBusy); 5056d31316cSdrh } 5066d31316cSdrh zBusy = Tcl_GetStringFromObj(objv[2], &len); 5076d31316cSdrh if( zBusy && len>0 ){ 5086d31316cSdrh pDb->zBusy = Tcl_Alloc( len + 1 ); 5096d31316cSdrh strcpy(pDb->zBusy, zBusy); 5106d31316cSdrh }else{ 511bec3f402Sdrh pDb->zBusy = 0; 512bec3f402Sdrh } 513bec3f402Sdrh if( pDb->zBusy ){ 514bec3f402Sdrh pDb->interp = interp; 5156f8a503dSdanielk1977 sqlite3_busy_handler(pDb->db, DbBusyHandler, pDb); 5166d31316cSdrh }else{ 5176f8a503dSdanielk1977 sqlite3_busy_handler(pDb->db, 0, 0); 518bec3f402Sdrh } 519bec3f402Sdrh } 5206d31316cSdrh break; 5216d31316cSdrh } 522bec3f402Sdrh 523b28af71aSdanielk1977 /* $db changes 524c8d30ac1Sdrh ** 525c8d30ac1Sdrh ** Return the number of rows that were modified, inserted, or deleted by 526b28af71aSdanielk1977 ** the most recent INSERT, UPDATE or DELETE statement, not including 527b28af71aSdanielk1977 ** any changes made by trigger programs. 528c8d30ac1Sdrh */ 529c8d30ac1Sdrh case DB_CHANGES: { 530c8d30ac1Sdrh Tcl_Obj *pResult; 531c8d30ac1Sdrh if( objc!=2 ){ 532c8d30ac1Sdrh Tcl_WrongNumArgs(interp, 2, objv, ""); 533c8d30ac1Sdrh return TCL_ERROR; 534c8d30ac1Sdrh } 535c8d30ac1Sdrh pResult = Tcl_GetObjResult(interp); 536b28af71aSdanielk1977 Tcl_SetIntObj(pResult, sqlite3_changes(pDb->db)); 537f146a776Srdc break; 538f146a776Srdc } 539f146a776Srdc 54075897234Sdrh /* $db close 54175897234Sdrh ** 54275897234Sdrh ** Shutdown the database 54375897234Sdrh */ 5446d31316cSdrh case DB_CLOSE: { 5456d31316cSdrh Tcl_DeleteCommand(interp, Tcl_GetStringFromObj(objv[0], 0)); 5466d31316cSdrh break; 5476d31316cSdrh } 54875897234Sdrh 549aa940eacSdrh /* $db commit_hook ?CALLBACK? 550aa940eacSdrh ** 551aa940eacSdrh ** Invoke the given callback just before committing every SQL transaction. 552aa940eacSdrh ** If the callback throws an exception or returns non-zero, then the 553aa940eacSdrh ** transaction is aborted. If CALLBACK is an empty string, the callback 554aa940eacSdrh ** is disabled. 555aa940eacSdrh */ 556aa940eacSdrh case DB_COMMIT_HOOK: { 557aa940eacSdrh if( objc>3 ){ 558aa940eacSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?"); 5590f14e2ebSdrh return TCL_ERROR; 560aa940eacSdrh }else if( objc==2 ){ 561aa940eacSdrh if( pDb->zCommit ){ 562aa940eacSdrh Tcl_AppendResult(interp, pDb->zCommit, 0); 563aa940eacSdrh } 564aa940eacSdrh }else{ 565aa940eacSdrh char *zCommit; 566aa940eacSdrh int len; 567aa940eacSdrh if( pDb->zCommit ){ 568aa940eacSdrh Tcl_Free(pDb->zCommit); 569aa940eacSdrh } 570aa940eacSdrh zCommit = Tcl_GetStringFromObj(objv[2], &len); 571aa940eacSdrh if( zCommit && len>0 ){ 572aa940eacSdrh pDb->zCommit = Tcl_Alloc( len + 1 ); 573aa940eacSdrh strcpy(pDb->zCommit, zCommit); 574aa940eacSdrh }else{ 575aa940eacSdrh pDb->zCommit = 0; 576aa940eacSdrh } 577aa940eacSdrh if( pDb->zCommit ){ 578aa940eacSdrh pDb->interp = interp; 5796f8a503dSdanielk1977 sqlite3_commit_hook(pDb->db, DbCommitHandler, pDb); 580aa940eacSdrh }else{ 5816f8a503dSdanielk1977 sqlite3_commit_hook(pDb->db, 0, 0); 582aa940eacSdrh } 583aa940eacSdrh } 584aa940eacSdrh break; 585aa940eacSdrh } 586aa940eacSdrh 5870f14e2ebSdrh /* 5880f14e2ebSdrh ** $db collate NAME SCRIPT 5890f14e2ebSdrh ** 5900f14e2ebSdrh ** Create a new SQL collation function called NAME. Whenever 5910f14e2ebSdrh ** that function is called, invoke SCRIPT to evaluate the function. 5920f14e2ebSdrh */ 5930f14e2ebSdrh case DB_COLLATE: { 5940f14e2ebSdrh SqlCollate *pCollate; 5950f14e2ebSdrh char *zName; 5960f14e2ebSdrh char *zScript; 5970f14e2ebSdrh int nScript; 5980f14e2ebSdrh if( objc!=4 ){ 5990f14e2ebSdrh Tcl_WrongNumArgs(interp, 2, objv, "NAME SCRIPT"); 6000f14e2ebSdrh return TCL_ERROR; 6010f14e2ebSdrh } 6020f14e2ebSdrh zName = Tcl_GetStringFromObj(objv[2], 0); 6030f14e2ebSdrh zScript = Tcl_GetStringFromObj(objv[3], &nScript); 6040f14e2ebSdrh pCollate = (SqlCollate*)Tcl_Alloc( sizeof(*pCollate) + nScript + 1 ); 6050f14e2ebSdrh if( pCollate==0 ) return TCL_ERROR; 6060f14e2ebSdrh pCollate->interp = interp; 6070f14e2ebSdrh pCollate->pNext = pDb->pCollate; 6080f14e2ebSdrh pCollate->zScript = (char*)&pCollate[1]; 6090f14e2ebSdrh pDb->pCollate = pCollate; 6100f14e2ebSdrh strcpy(pCollate->zScript, zScript); 6110f14e2ebSdrh if( sqlite3_create_collation(pDb->db, zName, SQLITE_UTF8, 6120f14e2ebSdrh pCollate, tclSqlCollate) ){ 6130f14e2ebSdrh return TCL_ERROR; 6140f14e2ebSdrh } 6150f14e2ebSdrh break; 6160f14e2ebSdrh } 6170f14e2ebSdrh 6180f14e2ebSdrh /* 6190f14e2ebSdrh ** $db collation_needed SCRIPT 6200f14e2ebSdrh ** 6210f14e2ebSdrh ** Create a new SQL collation function called NAME. Whenever 6220f14e2ebSdrh ** that function is called, invoke SCRIPT to evaluate the function. 6230f14e2ebSdrh */ 6240f14e2ebSdrh case DB_COLLATION_NEEDED: { 6250f14e2ebSdrh if( objc!=3 ){ 6260f14e2ebSdrh Tcl_WrongNumArgs(interp, 2, objv, "SCRIPT"); 6270f14e2ebSdrh return TCL_ERROR; 6280f14e2ebSdrh } 6290f14e2ebSdrh if( pDb->pCollateNeeded ){ 6300f14e2ebSdrh Tcl_DecrRefCount(pDb->pCollateNeeded); 6310f14e2ebSdrh } 6320f14e2ebSdrh pDb->pCollateNeeded = Tcl_DuplicateObj(objv[2]); 6330f14e2ebSdrh Tcl_IncrRefCount(pDb->pCollateNeeded); 6340f14e2ebSdrh sqlite3_collation_needed(pDb->db, pDb, tclCollateNeeded); 6350f14e2ebSdrh break; 6360f14e2ebSdrh } 6370f14e2ebSdrh 63875897234Sdrh /* $db complete SQL 63975897234Sdrh ** 64075897234Sdrh ** Return TRUE if SQL is a complete SQL statement. Return FALSE if 64175897234Sdrh ** additional lines of input are needed. This is similar to the 64275897234Sdrh ** built-in "info complete" command of Tcl. 64375897234Sdrh */ 6446d31316cSdrh case DB_COMPLETE: { 6456d31316cSdrh Tcl_Obj *pResult; 6466d31316cSdrh int isComplete; 6476d31316cSdrh if( objc!=3 ){ 6486d31316cSdrh Tcl_WrongNumArgs(interp, 2, objv, "SQL"); 64975897234Sdrh return TCL_ERROR; 65075897234Sdrh } 6516f8a503dSdanielk1977 isComplete = sqlite3_complete( Tcl_GetStringFromObj(objv[2], 0) ); 6526d31316cSdrh pResult = Tcl_GetObjResult(interp); 6536d31316cSdrh Tcl_SetBooleanObj(pResult, isComplete); 6546d31316cSdrh break; 6556d31316cSdrh } 65675897234Sdrh 65775897234Sdrh /* 658dcd997eaSdrh ** $db errorcode 659dcd997eaSdrh ** 660dcd997eaSdrh ** Return the numeric error code that was returned by the most recent 6616f8a503dSdanielk1977 ** call to sqlite3_exec(). 662dcd997eaSdrh */ 663dcd997eaSdrh case DB_ERRORCODE: { 664f3ce83f5Sdanielk1977 Tcl_SetObjResult(interp, Tcl_NewIntObj(sqlite3_errcode(pDb->db))); 665dcd997eaSdrh break; 666dcd997eaSdrh } 667dcd997eaSdrh 668dcd997eaSdrh /* 669895d7472Sdrh ** $db eval $sql ?array? ?{ ...code... }? 6701807ce37Sdrh ** $db onecolumn $sql 67175897234Sdrh ** 67275897234Sdrh ** The SQL statement in $sql is evaluated. For each row, the values are 673bec3f402Sdrh ** placed in elements of the array named "array" and ...code... is executed. 67475897234Sdrh ** If "array" and "code" are omitted, then no callback is every invoked. 67575897234Sdrh ** If "array" is an empty string, then the values are placed in variables 67675897234Sdrh ** that have the same name as the fields extracted by the query. 6771807ce37Sdrh ** 6781807ce37Sdrh ** The onecolumn method is the equivalent of: 6791807ce37Sdrh ** lindex [$db eval $sql] 0 68075897234Sdrh */ 6811807ce37Sdrh case DB_ONECOLUMN: 6826d31316cSdrh case DB_EVAL: { 6839d74b4c5Sdrh char const *zSql; /* Next SQL statement to execute */ 6849d74b4c5Sdrh char const *zLeft; /* What is left after first stmt in zSql */ 6859d74b4c5Sdrh sqlite3_stmt *pStmt; /* Compiled SQL statment */ 68692febd92Sdrh Tcl_Obj *pArray; /* Name of array into which results are written */ 68792febd92Sdrh Tcl_Obj *pScript; /* Script to run for each result set */ 6881d895039Sdrh Tcl_Obj **apParm; /* Parameters that need a Tcl_DecrRefCount() */ 6891d895039Sdrh int nParm; /* Number of entries used in apParm[] */ 6901d895039Sdrh Tcl_Obj *aParm[10]; /* Static space for apParm[] in the common case */ 6911807ce37Sdrh Tcl_Obj *pRet; /* Value to be returned */ 692ef2cb63eSdanielk1977 6931807ce37Sdrh if( choice==DB_ONECOLUMN ){ 6941807ce37Sdrh if( objc!=3 ){ 6951807ce37Sdrh Tcl_WrongNumArgs(interp, 2, objv, "SQL"); 6961807ce37Sdrh return TCL_ERROR; 6971807ce37Sdrh } 6981807ce37Sdrh pRet = 0; 6991807ce37Sdrh }else{ 70092febd92Sdrh if( objc<3 || objc>5 ){ 701895d7472Sdrh Tcl_WrongNumArgs(interp, 2, objv, "SQL ?ARRAY-NAME? ?SCRIPT?"); 70230ccda10Sdanielk1977 return TCL_ERROR; 70330ccda10Sdanielk1977 } 7041807ce37Sdrh pRet = Tcl_NewObj(); 7051807ce37Sdrh Tcl_IncrRefCount(pRet); 7061807ce37Sdrh } 70792febd92Sdrh if( objc==3 ){ 70892febd92Sdrh pArray = pScript = 0; 70992febd92Sdrh }else if( objc==4 ){ 71092febd92Sdrh pArray = 0; 71192febd92Sdrh pScript = objv[3]; 71292febd92Sdrh }else{ 71392febd92Sdrh pArray = objv[3]; 71492febd92Sdrh if( Tcl_GetString(pArray)[0]==0 ) pArray = 0; 71592febd92Sdrh pScript = objv[4]; 71692febd92Sdrh } 71730ccda10Sdanielk1977 7181d895039Sdrh Tcl_IncrRefCount(objv[2]); 71930ccda10Sdanielk1977 zSql = Tcl_GetStringFromObj(objv[2], 0); 72090b6bb19Sdrh while( rc==TCL_OK && zSql[0] ){ 72192febd92Sdrh int i; /* Loop counter */ 72292febd92Sdrh int nVar; /* Number of wildcards in the SQL */ 72392febd92Sdrh int nCol; /* Number of columns in the result set */ 72492febd92Sdrh Tcl_Obj **apColName = 0; /* Array of column names */ 72530ccda10Sdanielk1977 72692febd92Sdrh /* Compile a single SQL statement */ 72730ccda10Sdanielk1977 if( SQLITE_OK!=sqlite3_prepare(pDb->db, zSql, -1, &pStmt, &zLeft) ){ 728ef2cb63eSdanielk1977 Tcl_SetObjResult(interp, dbTextToObj(sqlite3_errmsg(pDb->db))); 72930ccda10Sdanielk1977 rc = TCL_ERROR; 73030ccda10Sdanielk1977 break; 73130ccda10Sdanielk1977 } 73292febd92Sdrh if( pStmt==0 ){ 73392febd92Sdrh if( SQLITE_OK!=sqlite3_errcode(pDb->db) ){ 73492febd92Sdrh Tcl_SetObjResult(interp, dbTextToObj(sqlite3_errmsg(pDb->db))); 73592febd92Sdrh rc = TCL_ERROR; 73692febd92Sdrh break; 73792febd92Sdrh }else{ 73892febd92Sdrh zSql = zLeft; 73992febd92Sdrh continue; 74092febd92Sdrh } 74192febd92Sdrh } 74230ccda10Sdanielk1977 7431d895039Sdrh /* Bind values to wildcards that begin with $ or : */ 74492febd92Sdrh nVar = sqlite3_bind_parameter_count(pStmt); 7451d895039Sdrh nParm = 0; 7461d895039Sdrh if( nVar>sizeof(aParm)/sizeof(aParm[0]) ){ 7471d895039Sdrh apParm = (Tcl_Obj**)Tcl_Alloc(nVar*sizeof(apParm[0])); 7481d895039Sdrh }else{ 7491d895039Sdrh apParm = aParm; 7501d895039Sdrh } 75192febd92Sdrh for(i=1; i<=nVar; i++){ 75292febd92Sdrh const char *zVar = sqlite3_bind_parameter_name(pStmt, i); 7532c6674cfSdrh if( zVar[0]=='$' || zVar[0]==':' ){ 75492febd92Sdrh Tcl_Obj *pVar = Tcl_GetVar2Ex(interp, &zVar[1], 0, 0); 75592febd92Sdrh if( pVar ){ 75692febd92Sdrh int n; 75792febd92Sdrh u8 *data; 75892febd92Sdrh char *zType = pVar->typePtr ? pVar->typePtr->name : ""; 75992febd92Sdrh char c = zType[0]; 76092febd92Sdrh if( c=='b' && strcmp(zType,"bytearray")==0 ){ 76192febd92Sdrh data = Tcl_GetByteArrayFromObj(pVar, &n); 76292febd92Sdrh sqlite3_bind_blob(pStmt, i, data, n, SQLITE_STATIC); 7631d895039Sdrh Tcl_IncrRefCount(pVar); 7641d895039Sdrh apParm[nParm++] = pVar; 76592febd92Sdrh }else if( (c=='b' && strcmp(zType,"boolean")==0) || 76692febd92Sdrh (c=='i' && strcmp(zType,"int")==0) ){ 76792febd92Sdrh Tcl_GetIntFromObj(interp, pVar, &n); 76892febd92Sdrh sqlite3_bind_int(pStmt, i, n); 76992febd92Sdrh }else if( c=='d' && strcmp(zType,"double")==0 ){ 77092febd92Sdrh double r; 77192febd92Sdrh Tcl_GetDoubleFromObj(interp, pVar, &r); 77292febd92Sdrh sqlite3_bind_double(pStmt, i, r); 77392febd92Sdrh }else{ 77492febd92Sdrh data = Tcl_GetStringFromObj(pVar, &n); 77592febd92Sdrh sqlite3_bind_text(pStmt, i, data, n, SQLITE_STATIC); 7761d895039Sdrh Tcl_IncrRefCount(pVar); 7771d895039Sdrh apParm[nParm++] = pVar; 77892febd92Sdrh } 77992febd92Sdrh } 78092febd92Sdrh } 78192febd92Sdrh } 78292febd92Sdrh 78392febd92Sdrh /* Compute column names */ 78492febd92Sdrh nCol = sqlite3_column_count(pStmt); 78592febd92Sdrh if( pScript ){ 78692febd92Sdrh apColName = (Tcl_Obj**)Tcl_Alloc( sizeof(Tcl_Obj*)*nCol ); 78792febd92Sdrh if( apColName==0 ) break; 78892febd92Sdrh for(i=0; i<nCol; i++){ 78992febd92Sdrh apColName[i] = dbTextToObj(sqlite3_column_name(pStmt,i)); 79092febd92Sdrh Tcl_IncrRefCount(apColName[i]); 79192febd92Sdrh } 79292febd92Sdrh } 79392febd92Sdrh 79492febd92Sdrh /* If results are being stored in an array variable, then create 79592febd92Sdrh ** the array(*) entry for that array 79692febd92Sdrh */ 79792febd92Sdrh if( pArray ){ 79830ccda10Sdanielk1977 Tcl_Obj *pColList = Tcl_NewObj(); 79930ccda10Sdanielk1977 Tcl_IncrRefCount(pColList); 80092febd92Sdrh for(i=0; i<nCol; i++){ 80192febd92Sdrh Tcl_ListObjAppendElement(interp, pColList, apColName[i]); 80230ccda10Sdanielk1977 } 80392febd92Sdrh Tcl_ObjSetVar2(interp, pArray, Tcl_NewStringObj("*",-1), pColList,0); 80430ccda10Sdanielk1977 } 80530ccda10Sdanielk1977 80692febd92Sdrh /* Execute the SQL 80792febd92Sdrh */ 80890b6bb19Sdrh while( rc==TCL_OK && pStmt && SQLITE_ROW==sqlite3_step(pStmt) ){ 80992febd92Sdrh for(i=0; i<nCol; i++){ 81030ccda10Sdanielk1977 Tcl_Obj *pVal; 81130ccda10Sdanielk1977 81230ccda10Sdanielk1977 /* Set pVal to contain the i'th column of this row. */ 81392febd92Sdrh switch( sqlite3_column_type(pStmt, i) ){ 81492febd92Sdrh case SQLITE_BLOB: { 8153fd0a736Sdanielk1977 int bytes = sqlite3_column_bytes(pStmt, i); 8163fd0a736Sdanielk1977 pVal = Tcl_NewByteArrayObj(sqlite3_column_blob(pStmt, i), bytes); 81792febd92Sdrh break; 81892febd92Sdrh } 81992febd92Sdrh case SQLITE_INTEGER: { 82092febd92Sdrh sqlite_int64 v = sqlite3_column_int64(pStmt, i); 82192febd92Sdrh if( v>=-2147483647 && v<=2147483647 ){ 82292febd92Sdrh pVal = Tcl_NewIntObj(v); 82392febd92Sdrh }else{ 82492febd92Sdrh pVal = Tcl_NewWideIntObj(v); 82592febd92Sdrh } 82692febd92Sdrh break; 82792febd92Sdrh } 82892febd92Sdrh case SQLITE_FLOAT: { 82992febd92Sdrh double r = sqlite3_column_double(pStmt, i); 83092febd92Sdrh pVal = Tcl_NewDoubleObj(r); 83192febd92Sdrh break; 83292febd92Sdrh } 83392febd92Sdrh default: { 83492febd92Sdrh pVal = dbTextToObj(sqlite3_column_text(pStmt, i)); 83592febd92Sdrh break; 83692febd92Sdrh } 83730ccda10Sdanielk1977 } 83830ccda10Sdanielk1977 83992febd92Sdrh if( pScript ){ 84092febd92Sdrh if( pArray==0 ){ 84192febd92Sdrh Tcl_ObjSetVar2(interp, apColName[i], 0, pVal, 0); 84230ccda10Sdanielk1977 }else{ 84392febd92Sdrh Tcl_ObjSetVar2(interp, pArray, apColName[i], pVal, 0); 84430ccda10Sdanielk1977 } 8451807ce37Sdrh }else if( choice==DB_ONECOLUMN ){ 8461807ce37Sdrh if( pRet==0 ){ 8471807ce37Sdrh pRet = pVal; 8481807ce37Sdrh Tcl_IncrRefCount(pRet); 8491807ce37Sdrh } 85090b6bb19Sdrh rc = TCL_BREAK; 85130ccda10Sdanielk1977 }else{ 85230ccda10Sdanielk1977 Tcl_ListObjAppendElement(interp, pRet, pVal); 85330ccda10Sdanielk1977 } 85430ccda10Sdanielk1977 } 85530ccda10Sdanielk1977 85692febd92Sdrh if( pScript ){ 85792febd92Sdrh rc = Tcl_EvalObjEx(interp, pScript, 0); 85890b6bb19Sdrh if( rc==TCL_CONTINUE ){ 85990b6bb19Sdrh rc = TCL_OK; 86030ccda10Sdanielk1977 } 86130ccda10Sdanielk1977 } 86290b6bb19Sdrh } 86390b6bb19Sdrh if( rc==TCL_BREAK ){ 86490b6bb19Sdrh rc = TCL_OK; 86590b6bb19Sdrh } 86630ccda10Sdanielk1977 86792febd92Sdrh /* Free the column name objects */ 86892febd92Sdrh if( pScript ){ 86992febd92Sdrh for(i=0; i<nCol; i++){ 87092febd92Sdrh Tcl_DecrRefCount(apColName[i]); 87192febd92Sdrh } 87292febd92Sdrh Tcl_Free((char*)apColName); 87392febd92Sdrh } 87492febd92Sdrh 8751d895039Sdrh /* Free the bound string and blob parameters */ 8761d895039Sdrh for(i=0; i<nParm; i++){ 8771d895039Sdrh Tcl_DecrRefCount(apParm[i]); 8781d895039Sdrh } 8791d895039Sdrh if( apParm!=aParm ){ 8801d895039Sdrh Tcl_Free((char*)apParm); 8811d895039Sdrh } 8821d895039Sdrh 88392febd92Sdrh /* Finalize the statement. If the result code is SQLITE_SCHEMA, then 88492febd92Sdrh ** try again to execute the same statement 88592febd92Sdrh */ 88692febd92Sdrh if( SQLITE_SCHEMA==sqlite3_finalize(pStmt) ){ 88730ccda10Sdanielk1977 continue; 88830ccda10Sdanielk1977 } 88930ccda10Sdanielk1977 89092febd92Sdrh if( SQLITE_OK!=sqlite3_errcode(pDb->db) ){ 891ef2cb63eSdanielk1977 Tcl_SetObjResult(interp, dbTextToObj(sqlite3_errmsg(pDb->db))); 89230ccda10Sdanielk1977 rc = TCL_ERROR; 89330ccda10Sdanielk1977 break; 89430ccda10Sdanielk1977 } 89530ccda10Sdanielk1977 89630ccda10Sdanielk1977 zSql = zLeft; 89730ccda10Sdanielk1977 } 8981d895039Sdrh Tcl_DecrRefCount(objv[2]); 89930ccda10Sdanielk1977 9001807ce37Sdrh if( pRet ){ 901ef2cb63eSdanielk1977 if( rc==TCL_OK ){ 90230ccda10Sdanielk1977 Tcl_SetObjResult(interp, pRet); 90330ccda10Sdanielk1977 } 904ef2cb63eSdanielk1977 Tcl_DecrRefCount(pRet); 9051807ce37Sdrh } 90630ccda10Sdanielk1977 break; 90730ccda10Sdanielk1977 } 908bec3f402Sdrh 909bec3f402Sdrh /* 910cabb0819Sdrh ** $db function NAME SCRIPT 911cabb0819Sdrh ** 912cabb0819Sdrh ** Create a new SQL function called NAME. Whenever that function is 913cabb0819Sdrh ** called, invoke SCRIPT to evaluate the function. 914cabb0819Sdrh */ 915cabb0819Sdrh case DB_FUNCTION: { 916cabb0819Sdrh SqlFunc *pFunc; 917cabb0819Sdrh char *zName; 918cabb0819Sdrh char *zScript; 919cabb0819Sdrh int nScript; 920cabb0819Sdrh if( objc!=4 ){ 921cabb0819Sdrh Tcl_WrongNumArgs(interp, 2, objv, "NAME SCRIPT"); 922cabb0819Sdrh return TCL_ERROR; 923cabb0819Sdrh } 924cabb0819Sdrh zName = Tcl_GetStringFromObj(objv[2], 0); 925cabb0819Sdrh zScript = Tcl_GetStringFromObj(objv[3], &nScript); 926cabb0819Sdrh pFunc = (SqlFunc*)Tcl_Alloc( sizeof(*pFunc) + nScript + 1 ); 927cabb0819Sdrh if( pFunc==0 ) return TCL_ERROR; 928cabb0819Sdrh pFunc->interp = interp; 929cabb0819Sdrh pFunc->pNext = pDb->pFunc; 930cabb0819Sdrh pFunc->zScript = (char*)&pFunc[1]; 9310f14e2ebSdrh pDb->pFunc = pFunc; 932cabb0819Sdrh strcpy(pFunc->zScript, zScript); 933c8c1158bSdanielk1977 rc = sqlite3_create_function(pDb->db, zName, -1, SQLITE_UTF8, 934d8123366Sdanielk1977 pFunc, tclSqlFunc, 0, 0); 935c8c1158bSdanielk1977 if( rc!=SQLITE_OK ) rc = TCL_ERROR; 936cabb0819Sdrh break; 937cabb0819Sdrh } 938cabb0819Sdrh 939cabb0819Sdrh /* 940af9ff33aSdrh ** $db last_insert_rowid 941af9ff33aSdrh ** 942af9ff33aSdrh ** Return an integer which is the ROWID for the most recent insert. 943af9ff33aSdrh */ 944af9ff33aSdrh case DB_LAST_INSERT_ROWID: { 945af9ff33aSdrh Tcl_Obj *pResult; 946af9ff33aSdrh int rowid; 947af9ff33aSdrh if( objc!=2 ){ 948af9ff33aSdrh Tcl_WrongNumArgs(interp, 2, objv, ""); 949af9ff33aSdrh return TCL_ERROR; 950af9ff33aSdrh } 9516f8a503dSdanielk1977 rowid = sqlite3_last_insert_rowid(pDb->db); 952af9ff33aSdrh pResult = Tcl_GetObjResult(interp); 953af9ff33aSdrh Tcl_SetIntObj(pResult, rowid); 954af9ff33aSdrh break; 955af9ff33aSdrh } 956af9ff33aSdrh 957af9ff33aSdrh /* 9581807ce37Sdrh ** The DB_ONECOLUMN method is implemented together with DB_EVAL. 9595d9d7576Sdrh */ 9601807ce37Sdrh 9611807ce37Sdrh /* $db progress ?N CALLBACK? 9621807ce37Sdrh ** 9631807ce37Sdrh ** Invoke the given callback every N virtual machine opcodes while executing 9641807ce37Sdrh ** queries. 9651807ce37Sdrh */ 9661807ce37Sdrh case DB_PROGRESS: { 9671807ce37Sdrh if( objc==2 ){ 9681807ce37Sdrh if( pDb->zProgress ){ 9691807ce37Sdrh Tcl_AppendResult(interp, pDb->zProgress, 0); 9705d9d7576Sdrh } 9711807ce37Sdrh }else if( objc==4 ){ 9721807ce37Sdrh char *zProgress; 9731807ce37Sdrh int len; 9741807ce37Sdrh int N; 9751807ce37Sdrh if( TCL_OK!=Tcl_GetIntFromObj(interp, objv[2], &N) ){ 9761807ce37Sdrh return TCL_ERROR; 9771807ce37Sdrh }; 9781807ce37Sdrh if( pDb->zProgress ){ 9791807ce37Sdrh Tcl_Free(pDb->zProgress); 9801807ce37Sdrh } 9811807ce37Sdrh zProgress = Tcl_GetStringFromObj(objv[3], &len); 9821807ce37Sdrh if( zProgress && len>0 ){ 9831807ce37Sdrh pDb->zProgress = Tcl_Alloc( len + 1 ); 9841807ce37Sdrh strcpy(pDb->zProgress, zProgress); 9851807ce37Sdrh }else{ 9861807ce37Sdrh pDb->zProgress = 0; 9871807ce37Sdrh } 9881807ce37Sdrh #ifndef SQLITE_OMIT_PROGRESS_CALLBACK 9891807ce37Sdrh if( pDb->zProgress ){ 9901807ce37Sdrh pDb->interp = interp; 9911807ce37Sdrh sqlite3_progress_handler(pDb->db, N, DbProgressHandler, pDb); 9921807ce37Sdrh }else{ 9931807ce37Sdrh sqlite3_progress_handler(pDb->db, 0, 0, 0); 9941807ce37Sdrh } 9951807ce37Sdrh #endif 9961807ce37Sdrh }else{ 9971807ce37Sdrh Tcl_WrongNumArgs(interp, 2, objv, "N CALLBACK"); 9981807ce37Sdrh return TCL_ERROR; 9995d9d7576Sdrh } 10005d9d7576Sdrh break; 10015d9d7576Sdrh } 10025d9d7576Sdrh 10035d9d7576Sdrh /* 100422fbcb8dSdrh ** $db rekey KEY 100522fbcb8dSdrh ** 100622fbcb8dSdrh ** Change the encryption key on the currently open database. 100722fbcb8dSdrh */ 100822fbcb8dSdrh case DB_REKEY: { 100922fbcb8dSdrh int nKey; 101022fbcb8dSdrh void *pKey; 101122fbcb8dSdrh if( objc!=3 ){ 101222fbcb8dSdrh Tcl_WrongNumArgs(interp, 2, objv, "KEY"); 101322fbcb8dSdrh return TCL_ERROR; 101422fbcb8dSdrh } 101522fbcb8dSdrh pKey = Tcl_GetByteArrayFromObj(objv[2], &nKey); 10169eb9e26bSdrh #ifdef SQLITE_HAS_CODEC 10172011d5f5Sdrh rc = sqlite3_rekey(pDb->db, pKey, nKey); 101822fbcb8dSdrh if( rc ){ 1019f20b21c8Sdanielk1977 Tcl_AppendResult(interp, sqlite3ErrStr(rc), 0); 102022fbcb8dSdrh rc = TCL_ERROR; 102122fbcb8dSdrh } 102222fbcb8dSdrh #endif 102322fbcb8dSdrh break; 102422fbcb8dSdrh } 102522fbcb8dSdrh 102622fbcb8dSdrh /* 1027bec3f402Sdrh ** $db timeout MILLESECONDS 1028bec3f402Sdrh ** 1029bec3f402Sdrh ** Delay for the number of milliseconds specified when a file is locked. 1030bec3f402Sdrh */ 10316d31316cSdrh case DB_TIMEOUT: { 1032bec3f402Sdrh int ms; 10336d31316cSdrh if( objc!=3 ){ 10346d31316cSdrh Tcl_WrongNumArgs(interp, 2, objv, "MILLISECONDS"); 1035bec3f402Sdrh return TCL_ERROR; 103675897234Sdrh } 10376d31316cSdrh if( Tcl_GetIntFromObj(interp, objv[2], &ms) ) return TCL_ERROR; 10386f8a503dSdanielk1977 sqlite3_busy_timeout(pDb->db, ms); 10396d31316cSdrh break; 104075897234Sdrh } 1041b5a20d3cSdrh 10420f14e2ebSdrh /* 10430f14e2ebSdrh ** $db total_changes 10440f14e2ebSdrh ** 10450f14e2ebSdrh ** Return the number of rows that were modified, inserted, or deleted 10460f14e2ebSdrh ** since the database handle was created. 10470f14e2ebSdrh */ 10480f14e2ebSdrh case DB_TOTAL_CHANGES: { 10490f14e2ebSdrh Tcl_Obj *pResult; 10500f14e2ebSdrh if( objc!=2 ){ 10510f14e2ebSdrh Tcl_WrongNumArgs(interp, 2, objv, ""); 10520f14e2ebSdrh return TCL_ERROR; 10530f14e2ebSdrh } 10540f14e2ebSdrh pResult = Tcl_GetObjResult(interp); 10550f14e2ebSdrh Tcl_SetIntObj(pResult, sqlite3_total_changes(pDb->db)); 10560f14e2ebSdrh break; 10570f14e2ebSdrh } 10580f14e2ebSdrh 1059b5a20d3cSdrh /* $db trace ?CALLBACK? 1060b5a20d3cSdrh ** 1061b5a20d3cSdrh ** Make arrangements to invoke the CALLBACK routine for each SQL statement 1062b5a20d3cSdrh ** that is executed. The text of the SQL is appended to CALLBACK before 1063b5a20d3cSdrh ** it is executed. 1064b5a20d3cSdrh */ 1065b5a20d3cSdrh case DB_TRACE: { 1066b5a20d3cSdrh if( objc>3 ){ 1067b5a20d3cSdrh Tcl_WrongNumArgs(interp, 2, objv, "?CALLBACK?"); 1068b97759edSdrh return TCL_ERROR; 1069b5a20d3cSdrh }else if( objc==2 ){ 1070b5a20d3cSdrh if( pDb->zTrace ){ 1071b5a20d3cSdrh Tcl_AppendResult(interp, pDb->zTrace, 0); 1072b5a20d3cSdrh } 1073b5a20d3cSdrh }else{ 1074b5a20d3cSdrh char *zTrace; 1075b5a20d3cSdrh int len; 1076b5a20d3cSdrh if( pDb->zTrace ){ 1077b5a20d3cSdrh Tcl_Free(pDb->zTrace); 1078b5a20d3cSdrh } 1079b5a20d3cSdrh zTrace = Tcl_GetStringFromObj(objv[2], &len); 1080b5a20d3cSdrh if( zTrace && len>0 ){ 1081b5a20d3cSdrh pDb->zTrace = Tcl_Alloc( len + 1 ); 1082b5a20d3cSdrh strcpy(pDb->zTrace, zTrace); 1083b5a20d3cSdrh }else{ 1084b5a20d3cSdrh pDb->zTrace = 0; 1085b5a20d3cSdrh } 1086b5a20d3cSdrh if( pDb->zTrace ){ 1087b5a20d3cSdrh pDb->interp = interp; 10886f8a503dSdanielk1977 sqlite3_trace(pDb->db, DbTraceHandler, pDb); 1089b5a20d3cSdrh }else{ 10906f8a503dSdanielk1977 sqlite3_trace(pDb->db, 0, 0); 1091b5a20d3cSdrh } 1092b5a20d3cSdrh } 1093b5a20d3cSdrh break; 1094b5a20d3cSdrh } 1095b5a20d3cSdrh 10961067fe11Stpoindex /* $db copy conflict-algorithm table filename ?SEPARATOR? ?NULLINDICATOR? 10971067fe11Stpoindex ** 10981067fe11Stpoindex ** Copy data into table from filename, optionally using SEPARATOR 10991067fe11Stpoindex ** as column separators. If a column contains a null string, or the 11001067fe11Stpoindex ** value of NULLINDICATOR, a NULL is inserted for the column. 11011067fe11Stpoindex ** conflict-algorithm is one of the sqlite conflict algorithms: 11021067fe11Stpoindex ** rollback, abort, fail, ignore, replace 11031067fe11Stpoindex ** On success, return the number of lines processed, not necessarily same 11041067fe11Stpoindex ** as 'db changes' due to conflict-algorithm selected. 11051067fe11Stpoindex ** 11061067fe11Stpoindex ** This code is basically an implementation/enhancement of 11071067fe11Stpoindex ** the sqlite3 shell.c ".import" command. 11081067fe11Stpoindex ** 11091067fe11Stpoindex ** This command usage is equivalent to the sqlite2.x COPY statement, 11101067fe11Stpoindex ** which imports file data into a table using the PostgreSQL COPY file format: 11111067fe11Stpoindex ** $db copy $conflit_algo $table_name $filename \t \\N 11121067fe11Stpoindex */ 11131067fe11Stpoindex case DB_COPY: { 11141067fe11Stpoindex char *zTable; /* Insert data into this table */ 11151067fe11Stpoindex char *zFile; /* The file from which to extract data */ 11161067fe11Stpoindex char *zConflict; /* The conflict algorithm to use */ 11171067fe11Stpoindex sqlite3_stmt *pStmt; /* A statement */ 11181067fe11Stpoindex int rc; /* Result code */ 11191067fe11Stpoindex int nCol; /* Number of columns in the table */ 11201067fe11Stpoindex int nByte; /* Number of bytes in an SQL string */ 11211067fe11Stpoindex int i, j; /* Loop counters */ 11221067fe11Stpoindex int nSep; /* Number of bytes in zSep[] */ 11231067fe11Stpoindex int nNull; /* Number of bytes in zNull[] */ 11241067fe11Stpoindex char *zSql; /* An SQL statement */ 11251067fe11Stpoindex char *zLine; /* A single line of input from the file */ 11261067fe11Stpoindex char **azCol; /* zLine[] broken up into columns */ 11271067fe11Stpoindex char *zCommit; /* How to commit changes */ 11281067fe11Stpoindex FILE *in; /* The input file */ 11291067fe11Stpoindex int lineno = 0; /* Line number of input file */ 11301067fe11Stpoindex char zLineNum[80]; /* Line number print buffer */ 11311067fe11Stpoindex Tcl_Obj *pResult; /* interp result */ 11321067fe11Stpoindex 1133*9ee3cdcbSdrh char *zSep; 1134*9ee3cdcbSdrh char *zNull; 1135*9ee3cdcbSdrh if( objc<5 || objc>7 ){ 1136*9ee3cdcbSdrh Tcl_WrongNumArgs(interp, 2, objv, 1137*9ee3cdcbSdrh "CONFLICT-ALGORITHM TABLE FILENAME ?SEPARATOR? ?NULLINDICATOR?"); 1138*9ee3cdcbSdrh return TCL_ERROR; 1139*9ee3cdcbSdrh } 1140*9ee3cdcbSdrh if( objc>=6 ){ 1141*9ee3cdcbSdrh zSep = Tcl_GetStringFromObj(objv[5], 0); 1142*9ee3cdcbSdrh }else{ 1143*9ee3cdcbSdrh zSep = "\t"; 1144*9ee3cdcbSdrh } 1145*9ee3cdcbSdrh if( objc>=7 ){ 1146*9ee3cdcbSdrh zNull = Tcl_GetStringFromObj(objv[6], 0); 1147*9ee3cdcbSdrh }else{ 1148*9ee3cdcbSdrh zNull = ""; 1149*9ee3cdcbSdrh } 11501067fe11Stpoindex zConflict = Tcl_GetStringFromObj(objv[2], 0); 11511067fe11Stpoindex zTable = Tcl_GetStringFromObj(objv[3], 0); 11521067fe11Stpoindex zFile = Tcl_GetStringFromObj(objv[4], 0); 11531067fe11Stpoindex nSep = strlen(zSep); 11541067fe11Stpoindex nNull = strlen(zNull); 11551067fe11Stpoindex if( nSep==0 ){ 11561067fe11Stpoindex Tcl_AppendResult(interp, "Error: non-null separator required for copy", 0); 11571067fe11Stpoindex return TCL_ERROR; 11581067fe11Stpoindex } 11591067fe11Stpoindex if(sqlite3StrICmp(zConflict, "rollback") != 0 && 11601067fe11Stpoindex sqlite3StrICmp(zConflict, "abort" ) != 0 && 11611067fe11Stpoindex sqlite3StrICmp(zConflict, "fail" ) != 0 && 11621067fe11Stpoindex sqlite3StrICmp(zConflict, "ignore" ) != 0 && 11631067fe11Stpoindex sqlite3StrICmp(zConflict, "replace" ) != 0 ) { 1164*9ee3cdcbSdrh Tcl_AppendResult(interp, "Error: \"", zConflict, 1165*9ee3cdcbSdrh "\", conflict-algorithm must be one of: rollback, " 1166*9ee3cdcbSdrh "abort, fail, ignore, or replace", 0); 11671067fe11Stpoindex return TCL_ERROR; 11681067fe11Stpoindex } 11691067fe11Stpoindex zSql = sqlite3_mprintf("SELECT * FROM '%q'", zTable); 11701067fe11Stpoindex if( zSql==0 ){ 11711067fe11Stpoindex Tcl_AppendResult(interp, "Error: no such table: ", zTable, 0); 11721067fe11Stpoindex return TCL_ERROR; 11731067fe11Stpoindex } 11741067fe11Stpoindex nByte = strlen(zSql); 11751067fe11Stpoindex rc = sqlite3_prepare(pDb->db, zSql, 0, &pStmt, 0); 11761067fe11Stpoindex sqlite3_free(zSql); 11771067fe11Stpoindex if( rc ){ 11781067fe11Stpoindex Tcl_AppendResult(interp, "Error: ", sqlite3_errmsg(pDb->db), 0); 11791067fe11Stpoindex nCol = 0; 11801067fe11Stpoindex }else{ 11811067fe11Stpoindex nCol = sqlite3_column_count(pStmt); 11821067fe11Stpoindex } 11831067fe11Stpoindex sqlite3_finalize(pStmt); 11841067fe11Stpoindex if( nCol==0 ) { 11851067fe11Stpoindex return TCL_ERROR; 11861067fe11Stpoindex } 11871067fe11Stpoindex zSql = malloc( nByte + 50 + nCol*2 ); 11881067fe11Stpoindex if( zSql==0 ) { 11891067fe11Stpoindex Tcl_AppendResult(interp, "Error: can't malloc()", 0); 11901067fe11Stpoindex return TCL_ERROR; 11911067fe11Stpoindex } 1192*9ee3cdcbSdrh sqlite3_snprintf(nByte+50, zSql, "INSERT OR %q INTO '%q' VALUES(?", 1193*9ee3cdcbSdrh zConflict, zTable); 11941067fe11Stpoindex j = strlen(zSql); 11951067fe11Stpoindex for(i=1; i<nCol; i++){ 11961067fe11Stpoindex zSql[j++] = ','; 11971067fe11Stpoindex zSql[j++] = '?'; 11981067fe11Stpoindex } 11991067fe11Stpoindex zSql[j++] = ')'; 12001067fe11Stpoindex zSql[j] = 0; 12011067fe11Stpoindex rc = sqlite3_prepare(pDb->db, zSql, 0, &pStmt, 0); 12021067fe11Stpoindex free(zSql); 12031067fe11Stpoindex if( rc ){ 12041067fe11Stpoindex Tcl_AppendResult(interp, "Error: ", sqlite3_errmsg(pDb->db), 0); 12051067fe11Stpoindex sqlite3_finalize(pStmt); 12061067fe11Stpoindex return TCL_ERROR; 12071067fe11Stpoindex } 12081067fe11Stpoindex in = fopen(zFile, "rb"); 12091067fe11Stpoindex if( in==0 ){ 12101067fe11Stpoindex Tcl_AppendResult(interp, "Error: cannot open file: ", zFile, NULL); 12111067fe11Stpoindex sqlite3_finalize(pStmt); 12121067fe11Stpoindex return TCL_ERROR; 12131067fe11Stpoindex } 12141067fe11Stpoindex azCol = malloc( sizeof(azCol[0])*(nCol+1) ); 12151067fe11Stpoindex if( azCol==0 ) { 12161067fe11Stpoindex Tcl_AppendResult(interp, "Error: can't malloc()", 0); 12171067fe11Stpoindex return TCL_ERROR; 12181067fe11Stpoindex } 12191067fe11Stpoindex sqlite3_exec(pDb->db, "BEGIN", 0, 0, 0); 12201067fe11Stpoindex zCommit = "COMMIT"; 12211067fe11Stpoindex while( (zLine = local_getline(0, in))!=0 ){ 12221067fe11Stpoindex char *z; 12231067fe11Stpoindex i = 0; 12241067fe11Stpoindex lineno++; 12251067fe11Stpoindex azCol[0] = zLine; 12261067fe11Stpoindex for(i=0, z=zLine; *z; z++){ 12271067fe11Stpoindex if( *z==zSep[0] && strncmp(z, zSep, nSep)==0 ){ 12281067fe11Stpoindex *z = 0; 12291067fe11Stpoindex i++; 12301067fe11Stpoindex if( i<nCol ){ 12311067fe11Stpoindex azCol[i] = &z[nSep]; 12321067fe11Stpoindex z += nSep-1; 12331067fe11Stpoindex } 12341067fe11Stpoindex } 12351067fe11Stpoindex } 12361067fe11Stpoindex if( i+1!=nCol ){ 12371067fe11Stpoindex char *zErr; 12381067fe11Stpoindex zErr = malloc(200 + strlen(zFile)); 12391067fe11Stpoindex sprintf(zErr,"Error: %s line %d: expected %d columns of data but found %d", 12401067fe11Stpoindex zFile, lineno, nCol, i+1); 12411067fe11Stpoindex Tcl_AppendResult(interp, zErr, 0); 12421067fe11Stpoindex free(zErr); 12431067fe11Stpoindex zCommit = "ROLLBACK"; 12441067fe11Stpoindex break; 12451067fe11Stpoindex } 12461067fe11Stpoindex for(i=0; i<nCol; i++){ 12471067fe11Stpoindex /* check for null data, if so, bind as null */ 12481067fe11Stpoindex if ((nNull>0 && strcmp(azCol[i], zNull)==0) || strlen(azCol[i])==0) { 12491067fe11Stpoindex sqlite3_bind_null(pStmt, i+1); 12501067fe11Stpoindex }else{ 12511067fe11Stpoindex sqlite3_bind_text(pStmt, i+1, azCol[i], -1, SQLITE_STATIC); 12521067fe11Stpoindex } 12531067fe11Stpoindex } 12541067fe11Stpoindex sqlite3_step(pStmt); 12551067fe11Stpoindex rc = sqlite3_reset(pStmt); 12561067fe11Stpoindex free(zLine); 12571067fe11Stpoindex if( rc!=SQLITE_OK ){ 12581067fe11Stpoindex Tcl_AppendResult(interp,"Error: ", sqlite3_errmsg(pDb->db), 0); 12591067fe11Stpoindex zCommit = "ROLLBACK"; 12601067fe11Stpoindex break; 12611067fe11Stpoindex } 12621067fe11Stpoindex } 12631067fe11Stpoindex free(azCol); 12641067fe11Stpoindex fclose(in); 12651067fe11Stpoindex sqlite3_finalize(pStmt); 12661067fe11Stpoindex sqlite3_exec(pDb->db, zCommit, 0, 0, 0); 12671067fe11Stpoindex 12681067fe11Stpoindex if( zCommit[0] == 'C' ){ 12691067fe11Stpoindex /* success, set result as number of lines processed */ 12701067fe11Stpoindex pResult = Tcl_GetObjResult(interp); 12711067fe11Stpoindex Tcl_SetIntObj(pResult, lineno); 12721067fe11Stpoindex rc = TCL_OK; 12731067fe11Stpoindex }else{ 12741067fe11Stpoindex /* failure, append lineno where failed */ 12751067fe11Stpoindex sprintf(zLineNum,"%d",lineno); 12761067fe11Stpoindex Tcl_AppendResult(interp,", failed while processing line: ",zLineNum,0); 12771067fe11Stpoindex rc = TCL_ERROR; 12781067fe11Stpoindex } 12791067fe11Stpoindex break; 12801067fe11Stpoindex } 12811067fe11Stpoindex 12821067fe11Stpoindex 12836d31316cSdrh } /* End of the SWITCH statement */ 128422fbcb8dSdrh return rc; 128575897234Sdrh } 128675897234Sdrh 128775897234Sdrh /* 12889bb575fdSdrh ** sqlite3 DBNAME FILENAME ?MODE? ?-key KEY? 128975897234Sdrh ** 129075897234Sdrh ** This is the main Tcl command. When the "sqlite" Tcl command is 129175897234Sdrh ** invoked, this routine runs to process that command. 129275897234Sdrh ** 129375897234Sdrh ** The first argument, DBNAME, is an arbitrary name for a new 129475897234Sdrh ** database connection. This command creates a new command named 129575897234Sdrh ** DBNAME that is used to control that connection. The database 129675897234Sdrh ** connection is deleted when the DBNAME command is deleted. 129775897234Sdrh ** 129875897234Sdrh ** The second argument is the name of the directory that contains 129975897234Sdrh ** the sqlite database that is to be accessed. 1300fbc3eab8Sdrh ** 1301fbc3eab8Sdrh ** For testing purposes, we also support the following: 1302fbc3eab8Sdrh ** 13039bb575fdSdrh ** sqlite3 -encoding 1304fbc3eab8Sdrh ** 1305fbc3eab8Sdrh ** Return the encoding used by LIKE and GLOB operators. Choices 1306fbc3eab8Sdrh ** are UTF-8 and iso8859. 1307fbc3eab8Sdrh ** 13089bb575fdSdrh ** sqlite3 -version 1309647cb0e1Sdrh ** 1310647cb0e1Sdrh ** Return the version number of the SQLite library. 1311647cb0e1Sdrh ** 13129bb575fdSdrh ** sqlite3 -tcl-uses-utf 1313fbc3eab8Sdrh ** 1314fbc3eab8Sdrh ** Return "1" if compiled with a Tcl uses UTF-8. Return "0" if 1315fbc3eab8Sdrh ** not. Used by tests to make sure the library was compiled 1316fbc3eab8Sdrh ** correctly. 131775897234Sdrh */ 131822fbcb8dSdrh static int DbMain(void *cd, Tcl_Interp *interp, int objc,Tcl_Obj *const*objv){ 1319bec3f402Sdrh SqliteDb *p; 132022fbcb8dSdrh void *pKey = 0; 132122fbcb8dSdrh int nKey = 0; 132222fbcb8dSdrh const char *zArg; 132375897234Sdrh char *zErrMsg; 132422fbcb8dSdrh const char *zFile; 132506b2718aSdrh char zBuf[80]; 132622fbcb8dSdrh if( objc==2 ){ 132722fbcb8dSdrh zArg = Tcl_GetStringFromObj(objv[1], 0); 132822fbcb8dSdrh if( strcmp(zArg,"-version")==0 ){ 13296f8a503dSdanielk1977 Tcl_AppendResult(interp,sqlite3_version,0); 1330647cb0e1Sdrh return TCL_OK; 1331647cb0e1Sdrh } 13329eb9e26bSdrh if( strcmp(zArg,"-has-codec")==0 ){ 13339eb9e26bSdrh #ifdef SQLITE_HAS_CODEC 133422fbcb8dSdrh Tcl_AppendResult(interp,"1",0); 133522fbcb8dSdrh #else 133622fbcb8dSdrh Tcl_AppendResult(interp,"0",0); 133722fbcb8dSdrh #endif 133822fbcb8dSdrh return TCL_OK; 133922fbcb8dSdrh } 134022fbcb8dSdrh if( strcmp(zArg,"-tcl-uses-utf")==0 ){ 1341fbc3eab8Sdrh #ifdef TCL_UTF_MAX 1342fbc3eab8Sdrh Tcl_AppendResult(interp,"1",0); 1343fbc3eab8Sdrh #else 1344fbc3eab8Sdrh Tcl_AppendResult(interp,"0",0); 1345fbc3eab8Sdrh #endif 1346fbc3eab8Sdrh return TCL_OK; 1347fbc3eab8Sdrh } 1348fbc3eab8Sdrh } 134922fbcb8dSdrh if( objc==5 || objc==6 ){ 135022fbcb8dSdrh zArg = Tcl_GetStringFromObj(objv[objc-2], 0); 135122fbcb8dSdrh if( strcmp(zArg,"-key")==0 ){ 135222fbcb8dSdrh pKey = Tcl_GetByteArrayFromObj(objv[objc-1], &nKey); 135322fbcb8dSdrh objc -= 2; 135422fbcb8dSdrh } 135522fbcb8dSdrh } 135622fbcb8dSdrh if( objc!=3 && objc!=4 ){ 135722fbcb8dSdrh Tcl_WrongNumArgs(interp, 1, objv, 13589eb9e26bSdrh #ifdef SQLITE_HAS_CODEC 13599eb9e26bSdrh "HANDLE FILENAME ?-key CODEC-KEY?" 136022fbcb8dSdrh #else 136122fbcb8dSdrh "HANDLE FILENAME ?MODE?" 136222fbcb8dSdrh #endif 136322fbcb8dSdrh ); 136475897234Sdrh return TCL_ERROR; 136575897234Sdrh } 136675897234Sdrh zErrMsg = 0; 13674cdc9e84Sdrh p = (SqliteDb*)Tcl_Alloc( sizeof(*p) ); 136875897234Sdrh if( p==0 ){ 1369bec3f402Sdrh Tcl_SetResult(interp, "malloc failed", TCL_STATIC); 1370bec3f402Sdrh return TCL_ERROR; 1371bec3f402Sdrh } 1372bec3f402Sdrh memset(p, 0, sizeof(*p)); 137322fbcb8dSdrh zFile = Tcl_GetStringFromObj(objv[2], 0); 13744f057f90Sdanielk1977 sqlite3_open(zFile, &p->db); 137580290863Sdanielk1977 if( SQLITE_OK!=sqlite3_errcode(p->db) ){ 137680290863Sdanielk1977 zErrMsg = strdup(sqlite3_errmsg(p->db)); 137780290863Sdanielk1977 sqlite3_close(p->db); 137880290863Sdanielk1977 p->db = 0; 137980290863Sdanielk1977 } 13802011d5f5Sdrh #ifdef SQLITE_HAS_CODEC 13812011d5f5Sdrh sqlite3_key(p->db, pKey, nKey); 1382eb8ed70dSdrh #endif 1383bec3f402Sdrh if( p->db==0 ){ 138475897234Sdrh Tcl_SetResult(interp, zErrMsg, TCL_VOLATILE); 1385bec3f402Sdrh Tcl_Free((char*)p); 138675897234Sdrh free(zErrMsg); 138775897234Sdrh return TCL_ERROR; 138875897234Sdrh } 138922fbcb8dSdrh zArg = Tcl_GetStringFromObj(objv[1], 0); 139022fbcb8dSdrh Tcl_CreateObjCommand(interp, zArg, DbObjCmd, (char*)p, DbDeleteCmd); 1391c22bd47dSdrh 139206b2718aSdrh /* The return value is the value of the sqlite* pointer 139306b2718aSdrh */ 139406b2718aSdrh sprintf(zBuf, "%p", p->db); 13955e5377fbSdrh if( strncmp(zBuf,"0x",2) ){ 13965e5377fbSdrh sprintf(zBuf, "0x%p", p->db); 13975e5377fbSdrh } 139806b2718aSdrh Tcl_AppendResult(interp, zBuf, 0); 139906b2718aSdrh 1400c22bd47dSdrh /* If compiled with SQLITE_TEST turned on, then register the "md5sum" 140106b2718aSdrh ** SQL function. 1402c22bd47dSdrh */ 140328b4e489Sdrh #ifdef SQLITE_TEST 140428b4e489Sdrh { 14059bb575fdSdrh extern void Md5_Register(sqlite3*); 14065b59af85Sdanielk1977 #ifdef SQLITE_DEBUG 140713073931Sdanielk1977 int mallocfail = sqlite3_iMallocFail; 140813073931Sdanielk1977 sqlite3_iMallocFail = 0; 14095b59af85Sdanielk1977 #endif 141028b4e489Sdrh Md5_Register(p->db); 14115b59af85Sdanielk1977 #ifdef SQLITE_DEBUG 141213073931Sdanielk1977 sqlite3_iMallocFail = mallocfail; 14135b59af85Sdanielk1977 #endif 141428b4e489Sdrh } 141528b4e489Sdrh #endif 14167cedc8d4Sdanielk1977 p->interp = interp; 141775897234Sdrh return TCL_OK; 141875897234Sdrh } 141975897234Sdrh 142075897234Sdrh /* 142190ca9753Sdrh ** Provide a dummy Tcl_InitStubs if we are using this as a static 142290ca9753Sdrh ** library. 142390ca9753Sdrh */ 142490ca9753Sdrh #ifndef USE_TCL_STUBS 142590ca9753Sdrh # undef Tcl_InitStubs 142690ca9753Sdrh # define Tcl_InitStubs(a,b,c) 142790ca9753Sdrh #endif 142890ca9753Sdrh 142990ca9753Sdrh /* 143075897234Sdrh ** Initialize this module. 143175897234Sdrh ** 143275897234Sdrh ** This Tcl module contains only a single new Tcl command named "sqlite". 143375897234Sdrh ** (Hence there is no namespace. There is no point in using a namespace 143475897234Sdrh ** if the extension only supplies one new name!) The "sqlite" command is 143575897234Sdrh ** used to open a new SQLite database. See the DbMain() routine above 143675897234Sdrh ** for additional information. 143775897234Sdrh */ 143838f8271fSdrh int Sqlite3_Init(Tcl_Interp *interp){ 143992febd92Sdrh Tcl_InitStubs(interp, "8.4", 0); 1440ef4ac8f9Sdrh Tcl_CreateObjCommand(interp, "sqlite3", (Tcl_ObjCmdProc*)DbMain, 0, 0); 1441ef4ac8f9Sdrh Tcl_PkgProvide(interp, "sqlite3", "3.0"); 144290ca9753Sdrh return TCL_OK; 144390ca9753Sdrh } 144438f8271fSdrh int Tclsqlite3_Init(Tcl_Interp *interp){ 144592febd92Sdrh Tcl_InitStubs(interp, "8.4", 0); 1446ef4ac8f9Sdrh Tcl_CreateObjCommand(interp, "sqlite3", (Tcl_ObjCmdProc*)DbMain, 0, 0); 1447ef4ac8f9Sdrh Tcl_PkgProvide(interp, "sqlite3", "3.0"); 144875897234Sdrh return TCL_OK; 144975897234Sdrh } 145038f8271fSdrh int Sqlite3_SafeInit(Tcl_Interp *interp){ 145175897234Sdrh return TCL_OK; 145275897234Sdrh } 145338f8271fSdrh int Tclsqlite3_SafeInit(Tcl_Interp *interp){ 145490ca9753Sdrh return TCL_OK; 145590ca9753Sdrh } 145675897234Sdrh 14573e27c026Sdrh #ifdef TCLSH 14583e27c026Sdrh /***************************************************************************** 14593e27c026Sdrh ** The code that follows is used to build standalone TCL interpreters 146075897234Sdrh */ 1461348784efSdrh 1462348784efSdrh /* 14633e27c026Sdrh ** If the macro TCLSH is one, then put in code this for the 14643e27c026Sdrh ** "main" routine that will initialize Tcl and take input from 14653e27c026Sdrh ** standard input. 1466348784efSdrh */ 14673e27c026Sdrh #if TCLSH==1 1468348784efSdrh static char zMainloop[] = 1469348784efSdrh "set line {}\n" 1470348784efSdrh "while {![eof stdin]} {\n" 1471348784efSdrh "if {$line!=\"\"} {\n" 1472348784efSdrh "puts -nonewline \"> \"\n" 1473348784efSdrh "} else {\n" 1474348784efSdrh "puts -nonewline \"% \"\n" 1475348784efSdrh "}\n" 1476348784efSdrh "flush stdout\n" 1477348784efSdrh "append line [gets stdin]\n" 1478348784efSdrh "if {[info complete $line]} {\n" 1479348784efSdrh "if {[catch {uplevel #0 $line} result]} {\n" 1480348784efSdrh "puts stderr \"Error: $result\"\n" 1481348784efSdrh "} elseif {$result!=\"\"} {\n" 1482348784efSdrh "puts $result\n" 1483348784efSdrh "}\n" 1484348784efSdrh "set line {}\n" 1485348784efSdrh "} else {\n" 1486348784efSdrh "append line \\n\n" 1487348784efSdrh "}\n" 1488348784efSdrh "}\n" 1489348784efSdrh ; 14903e27c026Sdrh #endif 14913e27c026Sdrh 14923e27c026Sdrh /* 14933e27c026Sdrh ** If the macro TCLSH is two, then get the main loop code out of 14943e27c026Sdrh ** the separate file "spaceanal_tcl.h". 14953e27c026Sdrh */ 14963e27c026Sdrh #if TCLSH==2 14973e27c026Sdrh static char zMainloop[] = 14983e27c026Sdrh #include "spaceanal_tcl.h" 14993e27c026Sdrh ; 15003e27c026Sdrh #endif 1501348784efSdrh 1502348784efSdrh #define TCLSH_MAIN main /* Needed to fake out mktclapp */ 1503348784efSdrh int TCLSH_MAIN(int argc, char **argv){ 1504348784efSdrh Tcl_Interp *interp; 1505297ecf14Sdrh Tcl_FindExecutable(argv[0]); 1506348784efSdrh interp = Tcl_CreateInterp(); 150738f8271fSdrh Sqlite3_Init(interp); 1508d9b0257aSdrh #ifdef SQLITE_TEST 1509d1bf3512Sdrh { 1510d1bf3512Sdrh extern int Sqlitetest1_Init(Tcl_Interp*); 15115c4d9703Sdrh extern int Sqlitetest2_Init(Tcl_Interp*); 15125c4d9703Sdrh extern int Sqlitetest3_Init(Tcl_Interp*); 1513a6064dcfSdrh extern int Sqlitetest4_Init(Tcl_Interp*); 1514998b56c3Sdanielk1977 extern int Sqlitetest5_Init(Tcl_Interp*); 1515efc251daSdrh extern int Md5_Init(Tcl_Interp*); 15166490bebdSdanielk1977 Sqlitetest1_Init(interp); 15175c4d9703Sdrh Sqlitetest2_Init(interp); 1518de647130Sdrh Sqlitetest3_Init(interp); 1519fc57d7bfSdanielk1977 Sqlitetest4_Init(interp); 1520998b56c3Sdanielk1977 Sqlitetest5_Init(interp); 1521efc251daSdrh Md5_Init(interp); 1522d1bf3512Sdrh } 1523d1bf3512Sdrh #endif 15243e27c026Sdrh if( argc>=2 || TCLSH==2 ){ 1525348784efSdrh int i; 1526348784efSdrh Tcl_SetVar(interp,"argv0",argv[1],TCL_GLOBAL_ONLY); 1527348784efSdrh Tcl_SetVar(interp,"argv", "", TCL_GLOBAL_ONLY); 152861212b69Sdrh for(i=3-TCLSH; i<argc; i++){ 1529348784efSdrh Tcl_SetVar(interp, "argv", argv[i], 1530348784efSdrh TCL_GLOBAL_ONLY | TCL_LIST_ELEMENT | TCL_APPEND_VALUE); 1531348784efSdrh } 15323e27c026Sdrh if( TCLSH==1 && Tcl_EvalFile(interp, argv[1])!=TCL_OK ){ 15330de8c112Sdrh const char *zInfo = Tcl_GetVar(interp, "errorInfo", TCL_GLOBAL_ONLY); 1534c61053b7Sdrh if( zInfo==0 ) zInfo = interp->result; 1535c61053b7Sdrh fprintf(stderr,"%s: %s\n", *argv, zInfo); 1536348784efSdrh return 1; 1537348784efSdrh } 15383e27c026Sdrh } 15393e27c026Sdrh if( argc<=1 || TCLSH==2 ){ 1540348784efSdrh Tcl_GlobalEval(interp, zMainloop); 1541348784efSdrh } 1542348784efSdrh return 0; 1543348784efSdrh } 1544348784efSdrh #endif /* TCLSH */ 15456d31316cSdrh 15466d31316cSdrh #endif /* !defined(NO_TCL) */ 1547