AryaWu/sqlite

1/*2** This program tests the ability of SQLite database to recover from a crash.3** This program runs under Unix only, but the results are applicable to all4** systems.5**6** The main process first constructs a test database, then starts creating7** subprocesses that write to that database.  Each subprocess is killed off,8** without a chance to clean up its database connection, after a random9** delay.  This killing of the subprocesses simulates a crash or power10** failure.  The next subprocess to open the database should rollback11** whatever operation was in process at the time of the simulated crash.12**13** If any problems are encountered, an error is reported and the test stops.14** If no problems are seen after a large number of tests, we assume that15** the rollback mechanism is working.16*/17#include <stdio.h>18#include <unistd.h>19#include <sys/types.h>20#include <sys/wait.h>21#include <signal.h>22#include <stdlib.h>23#include <string.h>24#include <sched.h>25#include "sqlite.h"26 27static void do_some_sql(int parent){28  char *zErr;29  int rc = SQLITE_OK;30  sqlite *db;31  int cnt = 0;32  static char zBig[] = 33    "-abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"34    "-abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";35 36  if( access("./test.db-journal",0)==0 ){37    /*printf("pid %d: journal exists.  rollback will be required\n",getpid());*/    unlink("test.db-saved");38    system("cp test.db test.db-saved");39    unlink("test.db-journal-saved");40    system("cp test.db-journal test.db-journal-saved");41  }42  db = sqlite_open("./test.db", 0, &zErr);43  if( db==0 ){44    printf("ERROR: %s\n", zErr);45    if( strcmp(zErr,"database disk image is malformed")==0 ){46      kill(parent, SIGKILL);47    }48    exit(1);49  }50  srand(getpid());51  while( rc==SQLITE_OK ){52    cnt++;53    rc = sqlite_exec_printf(db, 54       "INSERT INTO t1 VALUES(%d,'%d%s')", 0, 0, &zErr,55       rand(), rand(), zBig);56  }57  if( rc!=SQLITE_OK ){58    printf("ERROR #%d: %s\n", rc, zErr);59    if( rc==SQLITE_CORRUPT ){60      kill(parent, SIGKILL);61    }62  }63  printf("pid %d: cnt=%d\n", getpid(), cnt);64}65 66 67int main(int argc, char **argv){68  int i;69  sqlite *db;70  char *zErr;71  int status;72  int parent = getpid();73 74  unlink("test.db");75  unlink("test.db-journal");76  db = sqlite_open("test.db", 0, &zErr);77  if( db==0 ){78    printf("Cannot initialize: %s\n", zErr);79    return 1;80  }81  sqlite_exec(db, "CREATE TABLE t1(a,b)", 0, 0, 0);82  sqlite_close(db);83  for(i=0; i<10000; i++){84    int pid = fork();85    if( pid==0 ){86      sched_yield();87      do_some_sql(parent);88      return 0;89    }90    printf("test %d, pid=%d\n", i, pid);91    usleep(rand()%10000 + 1000);92    kill(pid, SIGKILL);93    waitpid(pid, &status, 0);94  }95  return 0;96}97