src/test_thread.c

   1 /*
   2 ** 2007 September 9
   3 **
   4 ** The author disclaims copyright to this source code.  In place of
   5 ** a legal notice, here is a blessing:
   6 **
   7 **    May you do good and not evil.
   8 **    May you find forgiveness for yourself and forgive others.
   9 **    May you share freely, never taking more than you give.
  10 **
  11 *************************************************************************
  12 **
  13 ** This file contains the implementation of some Tcl commands used to
  14 ** test that sqlite3 database handles may be concurrently accessed by
  15 ** multiple threads. Right now this only works on unix.
  16 */
  17
  18 #include "sqliteInt.h"
  19 #if defined(INCLUDE_SQLITE_TCL_H)
  20 #  include "sqlite_tcl.h"
  21 #else
  22 #  include "tcl.h"
  23 #endif
  24
  25 #if SQLITE_THREADSAFE
  26
  27 #include <errno.h>
  28
  29 #if !defined(_MSC_VER)
  30 #include <unistd.h>
  31 #endif
  32
  33 /*
  34 ** One of these is allocated for each thread created by [sqlthread spawn].
  35 */
  36 typedef struct SqlThread SqlThread;
  37 struct SqlThread {
  38   Tcl_ThreadId parent;     /* Thread id of parent thread */
  39   Tcl_Interp *interp;      /* Parent interpreter */
  40   char *zScript;           /* The script to execute. */
  41   char *zVarname;          /* Varname in parent script */
  42 };
  43
  44 /*
  45 ** A custom Tcl_Event type used by this module. When the event is
  46 ** handled, script zScript is evaluated in interpreter interp. If
  47 ** the evaluation throws an exception (returns TCL_ERROR), then the
  48 ** error is handled by Tcl_BackgroundError(). If no error occurs,
  49 ** the result is simply discarded.
  50 */
  51 typedef struct EvalEvent EvalEvent;
  52 struct EvalEvent {
  53   Tcl_Event base;          /* Base class of type Tcl_Event */
  54   char *zScript;           /* The script to execute. */
  55   Tcl_Interp *interp;      /* The interpreter to execute it in. */
  56 };
  57
  58 static Tcl_ObjCmdProc sqlthread_proc;
  59 static Tcl_ObjCmdProc clock_seconds_proc;
  60 #if SQLITE_OS_UNIX && defined(SQLITE_ENABLE_UNLOCK_NOTIFY)
  61 static Tcl_ObjCmdProc blocking_step_proc;
  62 static Tcl_ObjCmdProc blocking_prepare_v2_proc;
  63 #endif
  64 int Sqlitetest1_Init(Tcl_Interp *);
  65 int Sqlite3_Init(Tcl_Interp *);
  66
  67 /* Functions from main.c */
  68 extern const char *sqlite3ErrName(int);
  69
  70 /* Functions from test1.c */
  71 extern void *sqlite3TestTextToPtr(const char *);
  72 extern int getDbPointer(Tcl_Interp *, const char *, sqlite3 **);
  73 extern int sqlite3TestMakePointerStr(Tcl_Interp *, char *, void *);
  74 extern int sqlite3TestErrCode(Tcl_Interp *, sqlite3 *, int);
  75
  76 /*
  77 ** Handler for events of type EvalEvent.
  78 */
  79 static int SQLITE_TCLAPI tclScriptEvent(Tcl_Event *evPtr, int flags){
  80   int rc;
  81   EvalEvent *p = (EvalEvent *)evPtr;
  82   rc = Tcl_Eval(p->interp, p->zScript);
  83   if( rc!=TCL_OK ){
  84     Tcl_BackgroundError(p->interp);
  85   }
  86   UNUSED_PARAMETER(flags);
  87   return 1;
  88 }
  89
  90 /*
  91 ** Register an EvalEvent to evaluate the script pScript in the
  92 ** parent interpreter/thread of SqlThread p.
  93 */
  94 static void postToParent(SqlThread *p, Tcl_Obj *pScript){
  95   EvalEvent *pEvent;
  96   char *zMsg;
  97   int nMsg;
  98
  99   zMsg = Tcl_GetStringFromObj(pScript, &nMsg);
 100   pEvent = (EvalEvent *)ckalloc(sizeof(EvalEvent)+nMsg+1);
 101   pEvent->base.nextPtr = 0;
 102   pEvent->base.proc = tclScriptEvent;
 103   pEvent->zScript = (char *)&pEvent[1];
 104   memcpy(pEvent->zScript, zMsg, nMsg+1);
 105   pEvent->interp = p->interp;
 106
 107   Tcl_ThreadQueueEvent(p->parent, (Tcl_Event *)pEvent, TCL_QUEUE_TAIL);
 108   Tcl_ThreadAlert(p->parent);
 109 }
 110
 111 /*
 112 ** The main function for threads created with [sqlthread spawn].
 113 */
 114 static Tcl_ThreadCreateType tclScriptThread(ClientData pSqlThread){
 115   Tcl_Interp *interp;
 116   Tcl_Obj *pRes;
 117   Tcl_Obj *pList;
 118   int rc;
 119   SqlThread *p = (SqlThread *)pSqlThread;
 120   extern int Sqlitetest_mutex_Init(Tcl_Interp*);
 121
 122   interp = Tcl_CreateInterp();
 123   Tcl_CreateObjCommand(interp, "clock_seconds", clock_seconds_proc, 0, 0);
 124   Tcl_CreateObjCommand(interp, "sqlthread", sqlthread_proc, pSqlThread, 0);
 125 #if SQLITE_OS_UNIX && defined(SQLITE_ENABLE_UNLOCK_NOTIFY)
 126   Tcl_CreateObjCommand(interp, "sqlite3_blocking_step", blocking_step_proc,0,0);
 127   Tcl_CreateObjCommand(interp,
 128       "sqlite3_blocking_prepare_v2", blocking_prepare_v2_proc, (void *)1, 0);
 129   Tcl_CreateObjCommand(interp,
 130       "sqlite3_nonblocking_prepare_v2", blocking_prepare_v2_proc, 0, 0);
 131 #endif
 132   Sqlitetest1_Init(interp);
 133   Sqlitetest_mutex_Init(interp);
 134   Sqlite3_Init(interp);
 135
 136   rc = Tcl_Eval(interp, p->zScript);
 137   pRes = Tcl_GetObjResult(interp);
 138   pList = Tcl_NewObj();
 139   Tcl_IncrRefCount(pList);
 140   Tcl_IncrRefCount(pRes);
 141
 142   if( rc!=TCL_OK ){
 143     Tcl_ListObjAppendElement(interp, pList, Tcl_NewStringObj("error", -1));
 144     Tcl_ListObjAppendElement(interp, pList, pRes);
 145     postToParent(p, pList);
 146     Tcl_DecrRefCount(pList);
 147     pList = Tcl_NewObj();
 148   }
 149
 150   Tcl_ListObjAppendElement(interp, pList, Tcl_NewStringObj("set", -1));
 151   Tcl_ListObjAppendElement(interp, pList, Tcl_NewStringObj(p->zVarname, -1));
 152   Tcl_ListObjAppendElement(interp, pList, pRes);
 153   postToParent(p, pList);
 154
 155   ckfree((void *)p);
 156   Tcl_DecrRefCount(pList);
 157   Tcl_DecrRefCount(pRes);
 158   Tcl_DeleteInterp(interp);
 159   while( Tcl_DoOneEvent(TCL_ALL_EVENTS|TCL_DONT_WAIT) );
 160   Tcl_ExitThread(0);
 161   TCL_THREAD_CREATE_RETURN;
 162 }
 163
 164 /*
 165 ** sqlthread spawn VARNAME SCRIPT
 166 **
 167 **     Spawn a new thread with its own Tcl interpreter and run the
 168 **     specified SCRIPT(s) in it. The thread terminates after running
 169 **     the script. The result of the script is stored in the variable
 170 **     VARNAME.
 171 **
 172 **     The caller can wait for the script to terminate using [vwait VARNAME].
 173 */
 174 static int SQLITE_TCLAPI sqlthread_spawn(
 175   ClientData clientData,
 176   Tcl_Interp *interp,
 177   int objc,
 178   Tcl_Obj *CONST objv[]
 179 ){
 180   Tcl_ThreadId x;
 181   SqlThread *pNew;
 182   int rc;
 183
 184   int nVarname; char *zVarname;
 185   int nScript; char *zScript;
 186
 187   /* Parameters for thread creation */
 188   const int nStack = TCL_THREAD_STACK_DEFAULT;
 189   const int flags = TCL_THREAD_NOFLAGS;
 190
 191   assert(objc==4);
 192   UNUSED_PARAMETER(clientData);
 193   UNUSED_PARAMETER(objc);
 194
 195   zVarname = Tcl_GetStringFromObj(objv[2], &nVarname);
 196   zScript = Tcl_GetStringFromObj(objv[3], &nScript);
 197
 198   pNew = (SqlThread *)ckalloc(sizeof(SqlThread)+nVarname+nScript+2);
 199   pNew->zVarname = (char *)&pNew[1];
 200   pNew->zScript = (char *)&pNew->zVarname[nVarname+1];
 201   memcpy(pNew->zVarname, zVarname, nVarname+1);
 202   memcpy(pNew->zScript, zScript, nScript+1);
 203   pNew->parent = Tcl_GetCurrentThread();
 204   pNew->interp = interp;
 205
 206   rc = Tcl_CreateThread(&x, tclScriptThread, (void *)pNew, nStack, flags);
 207   if( rc!=TCL_OK ){
 208     Tcl_AppendResult(interp, "Error in Tcl_CreateThread()", 0);
 209     ckfree((char *)pNew);
 210     return TCL_ERROR;
 211   }
 212
 213   return TCL_OK;
 214 }
 215
 216 /*
 217 ** sqlthread parent SCRIPT
 218 **
 219 **     This can be called by spawned threads only. It sends the specified
 220 **     script back to the parent thread for execution. The result of
 221 **     evaluating the SCRIPT is returned. The parent thread must enter
 222 **     the event loop for this to work - otherwise the caller will
 223 **     block indefinitely.
 224 **
 225 **     NOTE: At the moment, this doesn't work. FIXME.
 226 */
 227 static int SQLITE_TCLAPI sqlthread_parent(
 228   ClientData clientData,
 229   Tcl_Interp *interp,
 230   int objc,
 231   Tcl_Obj *CONST objv[]
 232 ){
 233   EvalEvent *pEvent;
 234   char *zMsg;
 235   int nMsg;
 236   SqlThread *p = (SqlThread *)clientData;
 237
 238   assert(objc==3);
 239   UNUSED_PARAMETER(objc);
 240
 241   if( p==0 ){
 242     Tcl_AppendResult(interp, "no parent thread", 0);
 243     return TCL_ERROR;
 244   }
 245
 246   zMsg = Tcl_GetStringFromObj(objv[2], &nMsg);
 247   pEvent = (EvalEvent *)ckalloc(sizeof(EvalEvent)+nMsg+1);
 248   pEvent->base.nextPtr = 0;
 249   pEvent->base.proc = tclScriptEvent;
 250   pEvent->zScript = (char *)&pEvent[1];
 251   memcpy(pEvent->zScript, zMsg, nMsg+1);
 252   pEvent->interp = p->interp;
 253   Tcl_ThreadQueueEvent(p->parent, (Tcl_Event *)pEvent, TCL_QUEUE_TAIL);
 254   Tcl_ThreadAlert(p->parent);
 255
 256   return TCL_OK;
 257 }
 258
 259 static int xBusy(void *pArg, int nBusy){
 260   UNUSED_PARAMETER(pArg);
 261   UNUSED_PARAMETER(nBusy);
 262   sqlite3_sleep(50);
 263   return 1;             /* Try again... */
 264 }
 265
 266 /*
 267 ** sqlthread open
 268 **
 269 **     Open a database handle and return the string representation of
 270 **     the pointer value.
 271 */
 272 static int SQLITE_TCLAPI sqlthread_open(
 273   ClientData clientData,
 274   Tcl_Interp *interp,
 275   int objc,
 276   Tcl_Obj *CONST objv[]
 277 ){
 278   int sqlite3TestMakePointerStr(Tcl_Interp *interp, char *zPtr, void *p);
 279
 280   const char *zFilename;
 281   sqlite3 *db;
 282   char zBuf[100];
 283   extern int Md5_Register(sqlite3*,char**,const sqlite3_api_routines*);
 284
 285   UNUSED_PARAMETER(clientData);
 286   UNUSED_PARAMETER(objc);
 287
 288   zFilename = Tcl_GetString(objv[2]);
 289   sqlite3_open(zFilename, &db);
 290 /* BEGIN SQLCIPHER */
 291 #ifdef SQLITE_HAS_CODEC
 292   if( db && objc>=4 ){
 293     const char *zKey;
 294     int nKey;
 295     int rc;
 296     zKey = Tcl_GetStringFromObj(objv[3], &nKey);
 297     rc = sqlite3_key(db, zKey, nKey);
 298     if( rc!=SQLITE_OK ){
 299       char *zErrMsg = sqlite3_mprintf("error %d: %s", rc, sqlite3_errmsg(db));
 300       sqlite3_close(db);
 301       Tcl_AppendResult(interp, zErrMsg, (char*)0);
 302       sqlite3_free(zErrMsg);
 303       return TCL_ERROR;
 304     }
 305   }
 306 #endif
 307 /* END SQLCIPHER */
 308   Md5_Register(db, 0, 0);
 309   sqlite3_busy_handler(db, xBusy, 0);
 310
 311   if( sqlite3TestMakePointerStr(interp, zBuf, db) ) return TCL_ERROR;
 312   Tcl_AppendResult(interp, zBuf, 0);
 313
 314   return TCL_OK;
 315 }
 316
 317
 318 /*
 319 ** sqlthread open
 320 **
 321 **     Return the current thread-id (Tcl_GetCurrentThread()) cast to
 322 **     an integer.
 323 */
 324 static int SQLITE_TCLAPI sqlthread_id(
 325   ClientData clientData,
 326   Tcl_Interp *interp,
 327   int objc,
 328   Tcl_Obj *CONST objv[]
 329 ){
 330   Tcl_ThreadId id = Tcl_GetCurrentThread();
 331   Tcl_SetObjResult(interp, Tcl_NewIntObj(SQLITE_PTR_TO_INT(id)));
 332   UNUSED_PARAMETER(clientData);
 333   UNUSED_PARAMETER(objc);
 334   UNUSED_PARAMETER(objv);
 335   return TCL_OK;
 336 }
 337
 338
 339 /*
 340 ** Dispatch routine for the sub-commands of [sqlthread].
 341 */
 342 static int SQLITE_TCLAPI sqlthread_proc(
 343   ClientData clientData,
 344   Tcl_Interp *interp,
 345   int objc,
 346   Tcl_Obj *CONST objv[]
 347 ){
 348   struct SubCommand {
 349     char *zName;
 350     Tcl_ObjCmdProc *xProc;
 351     int nArg;
 352     char *zUsage;
 353   } aSub[] = {
 354     {"parent", sqlthread_parent, 1, "SCRIPT"},
 355     {"spawn",  sqlthread_spawn,  2, "VARNAME SCRIPT"},
 356     {"open",   sqlthread_open,   1, "DBNAME"},
 357     {"id",     sqlthread_id,     0, ""},
 358     {0, 0, 0}
 359   };
 360   struct SubCommand *pSub;
 361   int rc;
 362   int iIndex;
 363
 364   if( objc<2 ){
 365     Tcl_WrongNumArgs(interp, 1, objv, "SUB-COMMAND");
 366     return TCL_ERROR;
 367   }
 368
 369   rc = Tcl_GetIndexFromObjStruct(
 370       interp, objv[1], aSub, sizeof(aSub[0]), "sub-command", 0, &iIndex
 371   );
 372   if( rc!=TCL_OK ) return rc;
 373   pSub = &aSub[iIndex];
 374
 375   if( objc<(pSub->nArg+2) ){
 376     Tcl_WrongNumArgs(interp, 2, objv, pSub->zUsage);
 377     return TCL_ERROR;
 378   }
 379
 380   return pSub->xProc(clientData, interp, objc, objv);
 381 }
 382
 383 /*
 384 ** The [clock_seconds] command. This is more or less the same as the
 385 ** regular tcl [clock seconds], except that it is available in testfixture
 386 ** when linked against both Tcl 8.4 and 8.5. Because [clock seconds] is
 387 ** implemented as a script in Tcl 8.5, it is not usually available to
 388 ** testfixture.
 389 */
 390 static int SQLITE_TCLAPI clock_seconds_proc(
 391   ClientData clientData,
 392   Tcl_Interp *interp,
 393   int objc,
 394   Tcl_Obj *CONST objv[]
 395 ){
 396   Tcl_Time now;
 397   Tcl_GetTime(&now);
 398   Tcl_SetObjResult(interp, Tcl_NewIntObj(now.sec));
 399   UNUSED_PARAMETER(clientData);
 400   UNUSED_PARAMETER(objc);
 401   UNUSED_PARAMETER(objv);
 402   return TCL_OK;
 403 }
 404
 405 /*************************************************************************
 406 ** This block contains the implementation of the [sqlite3_blocking_step]
 407 ** command available to threads created by [sqlthread spawn] commands. It
 408 ** is only available on UNIX for now. This is because pthread condition
 409 ** variables are used.
 410 **
 411 ** The source code for the C functions sqlite3_blocking_step(),
 412 ** blocking_step_notify() and the structure UnlockNotification is
 413 ** automatically extracted from this file and used as part of the
 414 ** documentation for the sqlite3_unlock_notify() API function. This
 415 ** should be considered if these functions are to be extended (i.e. to
 416 ** support windows) in the future.
 417 */
 418 #if SQLITE_OS_UNIX && defined(SQLITE_ENABLE_UNLOCK_NOTIFY)
 419
 420 /* BEGIN_SQLITE_BLOCKING_STEP */
 421 /* This example uses the pthreads API */
 422 #include <pthread.h>
 423
 424 /*
 425 ** A pointer to an instance of this structure is passed as the user-context
 426 ** pointer when registering for an unlock-notify callback.
 427 */
 428 typedef struct UnlockNotification UnlockNotification;
 429 struct UnlockNotification {
 430   int fired;                         /* True after unlock event has occurred */
 431   pthread_cond_t cond;               /* Condition variable to wait on */
 432   pthread_mutex_t mutex;             /* Mutex to protect structure */
 433 };
 434
 435 /*
 436 ** This function is an unlock-notify callback registered with SQLite.
 437 */
 438 static void unlock_notify_cb(void **apArg, int nArg){
 439   int i;
 440   for(i=0; i<nArg; i++){
 441     UnlockNotification *p = (UnlockNotification *)apArg[i];
 442     pthread_mutex_lock(&p->mutex);
 443     p->fired = 1;
 444     pthread_cond_signal(&p->cond);
 445     pthread_mutex_unlock(&p->mutex);
 446   }
 447 }
 448
 449 /*
 450 ** This function assumes that an SQLite API call (either sqlite3_prepare_v2()
 451 ** or sqlite3_step()) has just returned SQLITE_LOCKED. The argument is the
 452 ** associated database connection.
 453 **
 454 ** This function calls sqlite3_unlock_notify() to register for an
 455 ** unlock-notify callback, then blocks until that callback is delivered
 456 ** and returns SQLITE_OK. The caller should then retry the failed operation.
 457 **
 458 ** Or, if sqlite3_unlock_notify() indicates that to block would deadlock
 459 ** the system, then this function returns SQLITE_LOCKED immediately. In
 460 ** this case the caller should not retry the operation and should roll
 461 ** back the current transaction (if any).
 462 */
 463 static int wait_for_unlock_notify(sqlite3 *db){
 464   int rc;
 465   UnlockNotification un;
 466
 467   /* Initialize the UnlockNotification structure. */
 468   un.fired = 0;
 469   pthread_mutex_init(&un.mutex, 0);
 470   pthread_cond_init(&un.cond, 0);
 471
 472   /* Register for an unlock-notify callback. */
 473   rc = sqlite3_unlock_notify(db, unlock_notify_cb, (void *)&un);
 474   assert( rc==SQLITE_LOCKED || rc==SQLITE_OK );
 475
 476   /* The call to sqlite3_unlock_notify() always returns either SQLITE_LOCKED
 477   ** or SQLITE_OK.
 478   **
 479   ** If SQLITE_LOCKED was returned, then the system is deadlocked. In this
 480   ** case this function needs to return SQLITE_LOCKED to the caller so
 481   ** that the current transaction can be rolled back. Otherwise, block
 482   ** until the unlock-notify callback is invoked, then return SQLITE_OK.
 483   */
 484   if( rc==SQLITE_OK ){
 485     pthread_mutex_lock(&un.mutex);
 486     if( !un.fired ){
 487       pthread_cond_wait(&un.cond, &un.mutex);
 488     }
 489     pthread_mutex_unlock(&un.mutex);
 490   }
 491
 492   /* Destroy the mutex and condition variables. */
 493   pthread_cond_destroy(&un.cond);
 494   pthread_mutex_destroy(&un.mutex);
 495
 496   return rc;
 497 }
 498
 499 /*
 500 ** This function is a wrapper around the SQLite function sqlite3_step().
 501 ** It functions in the same way as step(), except that if a required
 502 ** shared-cache lock cannot be obtained, this function may block waiting for
 503 ** the lock to become available. In this scenario the normal API step()
 504 ** function always returns SQLITE_LOCKED.
 505 **
 506 ** If this function returns SQLITE_LOCKED, the caller should rollback
 507 ** the current transaction (if any) and try again later. Otherwise, the
 508 ** system may become deadlocked.
 509 */
 510 int sqlite3_blocking_step(sqlite3_stmt *pStmt){
 511   int rc;
 512   while( SQLITE_LOCKED==(rc = sqlite3_step(pStmt)) ){
 513     rc = wait_for_unlock_notify(sqlite3_db_handle(pStmt));
 514     if( rc!=SQLITE_OK ) break;
 515     sqlite3_reset(pStmt);
 516   }
 517   return rc;
 518 }
 519
 520 /*
 521 ** This function is a wrapper around the SQLite function sqlite3_prepare_v2().
 522 ** It functions in the same way as prepare_v2(), except that if a required
 523 ** shared-cache lock cannot be obtained, this function may block waiting for
 524 ** the lock to become available. In this scenario the normal API prepare_v2()
 525 ** function always returns SQLITE_LOCKED.
 526 **
 527 ** If this function returns SQLITE_LOCKED, the caller should rollback
 528 ** the current transaction (if any) and try again later. Otherwise, the
 529 ** system may become deadlocked.
 530 */
 531 int sqlite3_blocking_prepare_v2(
 532   sqlite3 *db,              /* Database handle. */
 533   const char *zSql,         /* UTF-8 encoded SQL statement. */
 534   int nSql,                 /* Length of zSql in bytes. */
 535   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
 536   const char **pz           /* OUT: End of parsed string */
 537 ){
 538   int rc;
 539   while( SQLITE_LOCKED==(rc = sqlite3_prepare_v2(db, zSql, nSql, ppStmt, pz)) ){
 540     rc = wait_for_unlock_notify(db);
 541     if( rc!=SQLITE_OK ) break;
 542   }
 543   return rc;
 544 }
 545 /* END_SQLITE_BLOCKING_STEP */
 546
 547 /*
 548 ** Usage: sqlite3_blocking_step STMT
 549 **
 550 ** Advance the statement to the next row.
 551 */
 552 static int SQLITE_TCLAPI blocking_step_proc(
 553   void * clientData,
 554   Tcl_Interp *interp,
 555   int objc,
 556   Tcl_Obj *CONST objv[]
 557 ){
 558
 559   sqlite3_stmt *pStmt;
 560   int rc;
 561
 562   if( objc!=2 ){
 563     Tcl_WrongNumArgs(interp, 1, objv, "STMT");
 564     return TCL_ERROR;
 565   }
 566
 567   pStmt = (sqlite3_stmt*)sqlite3TestTextToPtr(Tcl_GetString(objv[1]));
 568   rc = sqlite3_blocking_step(pStmt);
 569
 570   Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), 0);
 571   return TCL_OK;
 572 }
 573
 574 /*
 575 ** Usage: sqlite3_blocking_prepare_v2 DB sql bytes ?tailvar?
 576 ** Usage: sqlite3_nonblocking_prepare_v2 DB sql bytes ?tailvar?
 577 */
 578 static int SQLITE_TCLAPI blocking_prepare_v2_proc(
 579   void * clientData,
 580   Tcl_Interp *interp,
 581   int objc,
 582   Tcl_Obj *CONST objv[]
 583 ){
 584   sqlite3 *db;
 585   const char *zSql;
 586   int bytes;
 587   const char *zTail = 0;
 588   sqlite3_stmt *pStmt = 0;
 589   char zBuf[50];
 590   int rc;
 591   int isBlocking = !(clientData==0);
 592
 593   if( objc!=5 && objc!=4 ){
 594     Tcl_AppendResult(interp, "wrong # args: should be \"",
 595        Tcl_GetString(objv[0]), " DB sql bytes tailvar", 0);
 596     return TCL_ERROR;
 597   }
 598   if( getDbPointer(interp, Tcl_GetString(objv[1]), &db) ) return TCL_ERROR;
 599   zSql = Tcl_GetString(objv[2]);
 600   if( Tcl_GetIntFromObj(interp, objv[3], &bytes) ) return TCL_ERROR;
 601
 602   if( isBlocking ){
 603     rc = sqlite3_blocking_prepare_v2(db, zSql, bytes, &pStmt, &zTail);
 604   }else{
 605     rc = sqlite3_prepare_v2(db, zSql, bytes, &pStmt, &zTail);
 606   }
 607
 608   assert(rc==SQLITE_OK || pStmt==0);
 609   if( zTail && objc>=5 ){
 610     if( bytes>=0 ){
 611       bytes = bytes - (zTail-zSql);
 612     }
 613     Tcl_ObjSetVar2(interp, objv[4], 0, Tcl_NewStringObj(zTail, bytes), 0);
 614   }
 615   if( rc!=SQLITE_OK ){
 616     assert( pStmt==0 );
 617     sqlite3_snprintf(sizeof(zBuf), zBuf, "%s ", (char *)sqlite3ErrName(rc));
 618     Tcl_AppendResult(interp, zBuf, sqlite3_errmsg(db), 0);
 619     return TCL_ERROR;
 620   }
 621
 622   if( pStmt ){
 623     if( sqlite3TestMakePointerStr(interp, zBuf, pStmt) ) return TCL_ERROR;
 624     Tcl_AppendResult(interp, zBuf, 0);
 625   }
 626   return TCL_OK;
 627 }
 628
 629 #endif /* SQLITE_OS_UNIX && SQLITE_ENABLE_UNLOCK_NOTIFY */
 630 /*
 631 ** End of implementation of [sqlite3_blocking_step].
 632 ************************************************************************/
 633
 634 /*
 635 ** Register commands with the TCL interpreter.
 636 */
 637 int SqlitetestThread_Init(Tcl_Interp *interp){
 638   Tcl_CreateObjCommand(interp, "sqlthread", sqlthread_proc, 0, 0);
 639   Tcl_CreateObjCommand(interp, "clock_seconds", clock_seconds_proc, 0, 0);
 640 #if SQLITE_OS_UNIX && defined(SQLITE_ENABLE_UNLOCK_NOTIFY)
 641   Tcl_CreateObjCommand(interp, "sqlite3_blocking_step", blocking_step_proc,0,0);
 642   Tcl_CreateObjCommand(interp,
 643       "sqlite3_blocking_prepare_v2", blocking_prepare_v2_proc, (void *)1, 0);
 644   Tcl_CreateObjCommand(interp,
 645       "sqlite3_nonblocking_prepare_v2", blocking_prepare_v2_proc, 0, 0);
 646 #endif
 647   return TCL_OK;
 648 }
 649 #else
 650 int SqlitetestThread_Init(Tcl_Interp *interp){
 651   return TCL_OK;
 652 }
 653 #endif