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 */ 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 */ 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 */ 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 */ 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
