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: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>