third_party/sqlite/src/test/threadtest2.c

   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 }