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*/
34volatile 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*/
42int 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*/
57int 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*/
74void *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*/
98int 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