1 /*
2 ** 2004 January 13
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 ** This file implements a simple standalone program used to test whether
13 ** or not the SQLite library is threadsafe.
14 **
15 ** This file is NOT part of the standard SQLite library. It is used for
16 ** testing only.
17 */
18 #include <stdio.h>
19 #include <unistd.h>
20 #include <pthread.h>
21 #include <string.h>
22 #include <stdlib.h>
23 #include "sqlite.h"
24
25 /*
26 ** Name of the database
27 */
28 #define DB_FILE "test.db"
29
30 /*
31 ** When this variable becomes non-zero, all threads stop
32 ** what they are doing.
33 */
34 volatile int all_stop = 0;
35
36 /*
37 ** Callback from the integrity check. If the result is anything other
38 ** than "ok" it means the integrity check has failed. Set the "all_stop"
39 ** global variable to stop all other activity. Print the error message
40 ** or print OK if the string "ok" is seen.
41 */
check_callback(void * pid,int argc,char ** argv,char ** notUsed2)42 int check_callback(void *pid, int argc, char **argv, char **notUsed2){
43 int id = (int)pid;
44 if( strcmp(argv[0],"ok") ){
45 all_stop = 1;
46 fprintf(stderr,"id: %s\n", id, argv[0]);
47 }else{
48 /* fprintf(stderr,"%d: OK\n", id); */
49 }
50 return 0;
51 }
52
53 /*
54 ** Do an integrity check on the database. If the first integrity check
55 ** fails, try it a second time.
56 */
integrity_check(sqlite * db,int id)57 int integrity_check(sqlite *db, int id){
58 int rc;
59 if( all_stop ) return 0;
60 /* fprintf(stderr,"%d: CHECK\n", id); */
61 rc = sqlite3_exec(db, "pragma integrity_check", check_callback, 0, 0);
62 if( rc!=SQLITE_OK && rc!=SQLITE_BUSY ){
63 fprintf(stderr,"%d, Integrity check returns %d\n", id, rc);
64 }
65 if( all_stop ){
66 sqlite3_exec(db, "pragma integrity_check", check_callback, 0, 0);
67 }
68 return 0;
69 }
70
71 /*
72 ** This is the worker thread
73 */
worker(void * workerArg)74 void *worker(void *workerArg){
75 sqlite *db;
76 int id = (int)workerArg;
77 int rc;
78 int cnt = 0;
79 fprintf(stderr, "Starting worker %d\n", id);
80 while( !all_stop && cnt++<10000 ){
81 if( cnt%100==0 ) printf("%d: %d\n", id, cnt);
82 while( (sqlite3_open(DB_FILE, &db))!=SQLITE_OK ) sched_yield();
83 sqlite3_exec(db, "PRAGMA synchronous=OFF", 0, 0, 0);
84 /* integrity_check(db, id); */
85 if( all_stop ){ sqlite3_close(db); break; }
86 /* fprintf(stderr, "%d: BEGIN\n", id); */
87 rc = sqlite3_exec(db, "INSERT INTO t1 VALUES('bogus data')", 0, 0, 0);
88 /* fprintf(stderr, "%d: END rc=%d\n", id, rc); */
89 sqlite3_close(db);
90 }
91 fprintf(stderr, "Worker %d finished\n", id);
92 return 0;
93 }
94
95 /*
96 ** Initialize the database and start the threads
97 */
main(int argc,char ** argv)98 int main(int argc, char **argv){
99 sqlite *db;
100 int i, rc;
101 pthread_t aThread[5];
102
103 if( strcmp(DB_FILE,":memory:") ){
104 char *zJournal = sqlite3_mprintf("%s-journal", DB_FILE);
105 unlink(DB_FILE);
106 unlink(zJournal);
107 sqlite3_free(zJournal);
108 }
109 sqlite3_open(DB_FILE, &db);
110 if( db==0 ){
111 fprintf(stderr,"unable to initialize database\n");
112 exit(1);
113 }
114 rc = sqlite3_exec(db, "CREATE TABLE t1(x);", 0,0,0);
115 if( rc ){
116 fprintf(stderr,"cannot create table t1: %d\n", rc);
117 exit(1);
118 }
119 sqlite3_close(db);
120 for(i=0; i<sizeof(aThread)/sizeof(aThread[0]); i++){
121 pthread_create(&aThread[i], 0, worker, (void*)i);
122 }
123 for(i=0; i<sizeof(aThread)/sizeof(aThread[i]); i++){
124 pthread_join(aThread[i], 0);
125 }
126 if( !all_stop ){
127 printf("Everything seems ok.\n");
128 return 0;
129 }else{
130 printf("We hit an error.\n");
131 return 1;
132 }
133 }
134