4 ** The author disclaims copyright to this source code. In place of
5 ** a legal notice, here is a blessing:
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.
11 *************************************************************************
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.
18 #include "sqliteInt.h"
19 #if defined(INCLUDE_SQLITE_TCL_H)
20 # include "sqlite_tcl.h"
29 #if !defined(_MSC_VER)
34 ** One of these is allocated for each thread created by [sqlthread spawn].
36 typedef struct SqlThread 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 */
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.
51 typedef struct EvalEvent 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. */
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
;
64 int Sqlitetest1_Init(Tcl_Interp
*);
65 int Sqlite3_Init(Tcl_Interp
*);
67 /* Functions from main.c */
68 extern const char *sqlite3ErrName(int);
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);
77 ** Handler for events of type EvalEvent.
79 static int SQLITE_TCLAPI
tclScriptEvent(Tcl_Event
*evPtr
, int flags
){
81 EvalEvent
*p
= (EvalEvent
*)evPtr
;
82 rc
= Tcl_Eval(p
->interp
, p
->zScript
);
84 Tcl_BackgroundError(p
->interp
);
86 UNUSED_PARAMETER(flags
);
91 ** Register an EvalEvent to evaluate the script pScript in the
92 ** parent interpreter/thread of SqlThread p.
94 static void postToParent(SqlThread
*p
, Tcl_Obj
*pScript
){
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
;
107 Tcl_ThreadQueueEvent(p
->parent
, (Tcl_Event
*)pEvent
, TCL_QUEUE_TAIL
);
108 Tcl_ThreadAlert(p
->parent
);
112 ** The main function for threads created with [sqlthread spawn].
114 static Tcl_ThreadCreateType
tclScriptThread(ClientData pSqlThread
){
119 SqlThread
*p
= (SqlThread
*)pSqlThread
;
120 extern int Sqlitetest_mutex_Init(Tcl_Interp
*);
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);
132 Sqlitetest1_Init(interp
);
133 Sqlitetest_mutex_Init(interp
);
134 Sqlite3_Init(interp
);
136 rc
= Tcl_Eval(interp
, p
->zScript
);
137 pRes
= Tcl_GetObjResult(interp
);
138 pList
= Tcl_NewObj();
139 Tcl_IncrRefCount(pList
);
140 Tcl_IncrRefCount(pRes
);
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();
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
);
156 Tcl_DecrRefCount(pList
);
157 Tcl_DecrRefCount(pRes
);
158 Tcl_DeleteInterp(interp
);
159 while( Tcl_DoOneEvent(TCL_ALL_EVENTS
|TCL_DONT_WAIT
) );
161 TCL_THREAD_CREATE_RETURN
;
165 ** sqlthread spawn VARNAME SCRIPT
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
172 ** The caller can wait for the script to terminate using [vwait VARNAME].
174 static int SQLITE_TCLAPI
sqlthread_spawn(
175 ClientData clientData
,
178 Tcl_Obj
*CONST objv
[]
184 int nVarname
; char *zVarname
;
185 int nScript
; char *zScript
;
187 /* Parameters for thread creation */
188 const int nStack
= TCL_THREAD_STACK_DEFAULT
;
189 const int flags
= TCL_THREAD_NOFLAGS
;
192 UNUSED_PARAMETER(clientData
);
193 UNUSED_PARAMETER(objc
);
195 zVarname
= Tcl_GetStringFromObj(objv
[2], &nVarname
);
196 zScript
= Tcl_GetStringFromObj(objv
[3], &nScript
);
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
;
206 rc
= Tcl_CreateThread(&x
, tclScriptThread
, (void *)pNew
, nStack
, flags
);
208 Tcl_AppendResult(interp
, "Error in Tcl_CreateThread()", 0);
209 ckfree((char *)pNew
);
217 ** sqlthread parent SCRIPT
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.
225 ** NOTE: At the moment, this doesn't work. FIXME.
227 static int SQLITE_TCLAPI
sqlthread_parent(
228 ClientData clientData
,
231 Tcl_Obj
*CONST objv
[]
236 SqlThread
*p
= (SqlThread
*)clientData
;
239 UNUSED_PARAMETER(objc
);
242 Tcl_AppendResult(interp
, "no parent thread", 0);
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
);
259 static int xBusy(void *pArg
, int nBusy
){
260 UNUSED_PARAMETER(pArg
);
261 UNUSED_PARAMETER(nBusy
);
263 return 1; /* Try again... */
269 ** Open a database handle and return the string representation of
270 ** the pointer value.
272 static int SQLITE_TCLAPI
sqlthread_open(
273 ClientData clientData
,
276 Tcl_Obj
*CONST objv
[]
278 int sqlite3TestMakePointerStr(Tcl_Interp
*interp
, char *zPtr
, void *p
);
280 const char *zFilename
;
283 extern int Md5_Register(sqlite3
*,char**,const sqlite3_api_routines
*);
285 UNUSED_PARAMETER(clientData
);
286 UNUSED_PARAMETER(objc
);
288 zFilename
= Tcl_GetString(objv
[2]);
289 sqlite3_open(zFilename
, &db
);
290 #ifdef SQLITE_HAS_CODEC
295 zKey
= Tcl_GetStringFromObj(objv
[3], &nKey
);
296 rc
= sqlite3_key(db
, zKey
, nKey
);
298 char *zErrMsg
= sqlite3_mprintf("error %d: %s", rc
, sqlite3_errmsg(db
));
300 Tcl_AppendResult(interp
, zErrMsg
, (char*)0);
301 sqlite3_free(zErrMsg
);
306 Md5_Register(db
, 0, 0);
307 sqlite3_busy_handler(db
, xBusy
, 0);
309 if( sqlite3TestMakePointerStr(interp
, zBuf
, db
) ) return TCL_ERROR
;
310 Tcl_AppendResult(interp
, zBuf
, 0);
319 ** Return the current thread-id (Tcl_GetCurrentThread()) cast to
322 static int SQLITE_TCLAPI
sqlthread_id(
323 ClientData clientData
,
326 Tcl_Obj
*CONST objv
[]
328 Tcl_ThreadId id
= Tcl_GetCurrentThread();
329 Tcl_SetObjResult(interp
, Tcl_NewIntObj(SQLITE_PTR_TO_INT(id
)));
330 UNUSED_PARAMETER(clientData
);
331 UNUSED_PARAMETER(objc
);
332 UNUSED_PARAMETER(objv
);
338 ** Dispatch routine for the sub-commands of [sqlthread].
340 static int SQLITE_TCLAPI
sqlthread_proc(
341 ClientData clientData
,
344 Tcl_Obj
*CONST objv
[]
348 Tcl_ObjCmdProc
*xProc
;
352 {"parent", sqlthread_parent
, 1, "SCRIPT"},
353 {"spawn", sqlthread_spawn
, 2, "VARNAME SCRIPT"},
354 {"open", sqlthread_open
, 1, "DBNAME"},
355 {"id", sqlthread_id
, 0, ""},
358 struct SubCommand
*pSub
;
363 Tcl_WrongNumArgs(interp
, 1, objv
, "SUB-COMMAND");
367 rc
= Tcl_GetIndexFromObjStruct(
368 interp
, objv
[1], aSub
, sizeof(aSub
[0]), "sub-command", 0, &iIndex
370 if( rc
!=TCL_OK
) return rc
;
371 pSub
= &aSub
[iIndex
];
373 if( objc
<(pSub
->nArg
+2) ){
374 Tcl_WrongNumArgs(interp
, 2, objv
, pSub
->zUsage
);
378 return pSub
->xProc(clientData
, interp
, objc
, objv
);
382 ** The [clock_seconds] command. This is more or less the same as the
383 ** regular tcl [clock seconds], except that it is available in testfixture
384 ** when linked against both Tcl 8.4 and 8.5. Because [clock seconds] is
385 ** implemented as a script in Tcl 8.5, it is not usually available to
388 static int SQLITE_TCLAPI
clock_seconds_proc(
389 ClientData clientData
,
392 Tcl_Obj
*CONST objv
[]
396 Tcl_SetObjResult(interp
, Tcl_NewIntObj(now
.sec
));
397 UNUSED_PARAMETER(clientData
);
398 UNUSED_PARAMETER(objc
);
399 UNUSED_PARAMETER(objv
);
403 /*************************************************************************
404 ** This block contains the implementation of the [sqlite3_blocking_step]
405 ** command available to threads created by [sqlthread spawn] commands. It
406 ** is only available on UNIX for now. This is because pthread condition
407 ** variables are used.
409 ** The source code for the C functions sqlite3_blocking_step(),
410 ** blocking_step_notify() and the structure UnlockNotification is
411 ** automatically extracted from this file and used as part of the
412 ** documentation for the sqlite3_unlock_notify() API function. This
413 ** should be considered if these functions are to be extended (i.e. to
414 ** support windows) in the future.
416 #if SQLITE_OS_UNIX && defined(SQLITE_ENABLE_UNLOCK_NOTIFY)
418 /* BEGIN_SQLITE_BLOCKING_STEP */
419 /* This example uses the pthreads API */
423 ** A pointer to an instance of this structure is passed as the user-context
424 ** pointer when registering for an unlock-notify callback.
426 typedef struct UnlockNotification UnlockNotification
;
427 struct UnlockNotification
{
428 int fired
; /* True after unlock event has occurred */
429 pthread_cond_t cond
; /* Condition variable to wait on */
430 pthread_mutex_t mutex
; /* Mutex to protect structure */
434 ** This function is an unlock-notify callback registered with SQLite.
436 static void unlock_notify_cb(void **apArg
, int nArg
){
438 for(i
=0; i
<nArg
; i
++){
439 UnlockNotification
*p
= (UnlockNotification
*)apArg
[i
];
440 pthread_mutex_lock(&p
->mutex
);
442 pthread_cond_signal(&p
->cond
);
443 pthread_mutex_unlock(&p
->mutex
);
448 ** This function assumes that an SQLite API call (either sqlite3_prepare_v2()
449 ** or sqlite3_step()) has just returned SQLITE_LOCKED. The argument is the
450 ** associated database connection.
452 ** This function calls sqlite3_unlock_notify() to register for an
453 ** unlock-notify callback, then blocks until that callback is delivered
454 ** and returns SQLITE_OK. The caller should then retry the failed operation.
456 ** Or, if sqlite3_unlock_notify() indicates that to block would deadlock
457 ** the system, then this function returns SQLITE_LOCKED immediately. In
458 ** this case the caller should not retry the operation and should roll
459 ** back the current transaction (if any).
461 static int wait_for_unlock_notify(sqlite3
*db
){
463 UnlockNotification un
;
465 /* Initialize the UnlockNotification structure. */
467 pthread_mutex_init(&un
.mutex
, 0);
468 pthread_cond_init(&un
.cond
, 0);
470 /* Register for an unlock-notify callback. */
471 rc
= sqlite3_unlock_notify(db
, unlock_notify_cb
, (void *)&un
);
472 assert( rc
==SQLITE_LOCKED
|| rc
==SQLITE_OK
);
474 /* The call to sqlite3_unlock_notify() always returns either SQLITE_LOCKED
477 ** If SQLITE_LOCKED was returned, then the system is deadlocked. In this
478 ** case this function needs to return SQLITE_LOCKED to the caller so
479 ** that the current transaction can be rolled back. Otherwise, block
480 ** until the unlock-notify callback is invoked, then return SQLITE_OK.
483 pthread_mutex_lock(&un
.mutex
);
485 pthread_cond_wait(&un
.cond
, &un
.mutex
);
487 pthread_mutex_unlock(&un
.mutex
);
490 /* Destroy the mutex and condition variables. */
491 pthread_cond_destroy(&un
.cond
);
492 pthread_mutex_destroy(&un
.mutex
);
498 ** This function is a wrapper around the SQLite function sqlite3_step().
499 ** It functions in the same way as step(), except that if a required
500 ** shared-cache lock cannot be obtained, this function may block waiting for
501 ** the lock to become available. In this scenario the normal API step()
502 ** function always returns SQLITE_LOCKED.
504 ** If this function returns SQLITE_LOCKED, the caller should rollback
505 ** the current transaction (if any) and try again later. Otherwise, the
506 ** system may become deadlocked.
508 int sqlite3_blocking_step(sqlite3_stmt
*pStmt
){
510 while( SQLITE_LOCKED
==(rc
= sqlite3_step(pStmt
)) ){
511 rc
= wait_for_unlock_notify(sqlite3_db_handle(pStmt
));
512 if( rc
!=SQLITE_OK
) break;
513 sqlite3_reset(pStmt
);
519 ** This function is a wrapper around the SQLite function sqlite3_prepare_v2().
520 ** It functions in the same way as prepare_v2(), except that if a required
521 ** shared-cache lock cannot be obtained, this function may block waiting for
522 ** the lock to become available. In this scenario the normal API prepare_v2()
523 ** function always returns SQLITE_LOCKED.
525 ** If this function returns SQLITE_LOCKED, the caller should rollback
526 ** the current transaction (if any) and try again later. Otherwise, the
527 ** system may become deadlocked.
529 int sqlite3_blocking_prepare_v2(
530 sqlite3
*db
, /* Database handle. */
531 const char *zSql
, /* UTF-8 encoded SQL statement. */
532 int nSql
, /* Length of zSql in bytes. */
533 sqlite3_stmt
**ppStmt
, /* OUT: A pointer to the prepared statement */
534 const char **pz
/* OUT: End of parsed string */
537 while( SQLITE_LOCKED
==(rc
= sqlite3_prepare_v2(db
, zSql
, nSql
, ppStmt
, pz
)) ){
538 rc
= wait_for_unlock_notify(db
);
539 if( rc
!=SQLITE_OK
) break;
543 /* END_SQLITE_BLOCKING_STEP */
546 ** Usage: sqlite3_blocking_step STMT
548 ** Advance the statement to the next row.
550 static int SQLITE_TCLAPI
blocking_step_proc(
554 Tcl_Obj
*CONST objv
[]
561 Tcl_WrongNumArgs(interp
, 1, objv
, "STMT");
565 pStmt
= (sqlite3_stmt
*)sqlite3TestTextToPtr(Tcl_GetString(objv
[1]));
566 rc
= sqlite3_blocking_step(pStmt
);
568 Tcl_SetResult(interp
, (char *)sqlite3ErrName(rc
), 0);
573 ** Usage: sqlite3_blocking_prepare_v2 DB sql bytes ?tailvar?
574 ** Usage: sqlite3_nonblocking_prepare_v2 DB sql bytes ?tailvar?
576 static int SQLITE_TCLAPI
blocking_prepare_v2_proc(
580 Tcl_Obj
*CONST objv
[]
585 const char *zTail
= 0;
586 sqlite3_stmt
*pStmt
= 0;
589 int isBlocking
= !(clientData
==0);
591 if( objc
!=5 && objc
!=4 ){
592 Tcl_AppendResult(interp
, "wrong # args: should be \"",
593 Tcl_GetString(objv
[0]), " DB sql bytes tailvar", 0);
596 if( getDbPointer(interp
, Tcl_GetString(objv
[1]), &db
) ) return TCL_ERROR
;
597 zSql
= Tcl_GetString(objv
[2]);
598 if( Tcl_GetIntFromObj(interp
, objv
[3], &bytes
) ) return TCL_ERROR
;
601 rc
= sqlite3_blocking_prepare_v2(db
, zSql
, bytes
, &pStmt
, &zTail
);
603 rc
= sqlite3_prepare_v2(db
, zSql
, bytes
, &pStmt
, &zTail
);
606 assert(rc
==SQLITE_OK
|| pStmt
==0);
607 if( zTail
&& objc
>=5 ){
609 bytes
= bytes
- (zTail
-zSql
);
611 Tcl_ObjSetVar2(interp
, objv
[4], 0, Tcl_NewStringObj(zTail
, bytes
), 0);
615 sqlite3_snprintf(sizeof(zBuf
), zBuf
, "%s ", (char *)sqlite3ErrName(rc
));
616 Tcl_AppendResult(interp
, zBuf
, sqlite3_errmsg(db
), 0);
621 if( sqlite3TestMakePointerStr(interp
, zBuf
, pStmt
) ) return TCL_ERROR
;
622 Tcl_AppendResult(interp
, zBuf
, 0);
627 #endif /* SQLITE_OS_UNIX && SQLITE_ENABLE_UNLOCK_NOTIFY */
629 ** End of implementation of [sqlite3_blocking_step].
630 ************************************************************************/
633 ** Register commands with the TCL interpreter.
635 int SqlitetestThread_Init(Tcl_Interp
*interp
){
636 Tcl_CreateObjCommand(interp
, "sqlthread", sqlthread_proc
, 0, 0);
637 Tcl_CreateObjCommand(interp
, "clock_seconds", clock_seconds_proc
, 0, 0);
638 #if SQLITE_OS_UNIX && defined(SQLITE_ENABLE_UNLOCK_NOTIFY)
639 Tcl_CreateObjCommand(interp
, "sqlite3_blocking_step", blocking_step_proc
,0,0);
640 Tcl_CreateObjCommand(interp
,
641 "sqlite3_blocking_prepare_v2", blocking_prepare_v2_proc
, (void *)1, 0);
642 Tcl_CreateObjCommand(interp
,
643 "sqlite3_nonblocking_prepare_v2", blocking_prepare_v2_proc
, 0, 0);
648 int SqlitetestThread_Init(Tcl_Interp
*interp
){