15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/* 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** This program tests the ability of SQLite database to recover from a crash. 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** This program runs under Unix only, but the results are applicable to all 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** systems. 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** The main process first constructs a test database, then starts creating 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** subprocesses that write to that database. Each subprocess is killed off, 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** without a chance to clean up its database connection, after a random 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** delay. This killing of the subprocesses simulates a crash or power 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** failure. The next subprocess to open the database should rollback 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** whatever operation was in process at the time of the simulated crash. 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** If any problems are encountered, an error is reported and the test stops. 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** If no problems are seen after a large number of tests, we assume that 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)** the rollback mechanism is working. 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)*/ 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <stdio.h> 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <unistd.h> 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/types.h> 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/wait.h> 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <signal.h> 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <stdlib.h> 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string.h> 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sched.h> 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sqlite.h" 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static void do_some_sql(int parent){ 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char *zErr; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rc = SQLITE_OK; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sqlite *db; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int cnt = 0; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static char zBig[] = 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "-abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ" 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "-abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if( access("./test.db-journal",0)==0 ){ 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) /*printf("pid %d: journal exists. rollback will be required\n",getpid());*/ unlink("test.db-saved"); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) system("cp test.db test.db-saved"); 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unlink("test.db-journal-saved"); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) system("cp test.db-journal test.db-journal-saved"); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) db = sqlite_open("./test.db", 0, &zErr); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if( db==0 ){ 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) printf("ERROR: %s\n", zErr); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if( strcmp(zErr,"database disk image is malformed")==0 ){ 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kill(parent, SIGKILL); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) exit(1); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) srand(getpid()); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while( rc==SQLITE_OK ){ 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cnt++; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) rc = sqlite_exec_printf(db, 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "INSERT INTO t1 VALUES(%d,'%d%s')", 0, 0, &zErr, 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) rand(), rand(), zBig); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if( rc!=SQLITE_OK ){ 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) printf("ERROR #%d: %s\n", rc, zErr); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if( rc==SQLITE_CORRUPT ){ 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kill(parent, SIGKILL); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) printf("pid %d: cnt=%d\n", getpid(), cnt); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int main(int argc, char **argv){ 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int i; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sqlite *db; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char *zErr; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int status; 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int parent = getpid(); 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unlink("test.db"); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unlink("test.db-journal"); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) db = sqlite_open("test.db", 0, &zErr); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if( db==0 ){ 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) printf("Cannot initialize: %s\n", zErr); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1; 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sqlite_exec(db, "CREATE TABLE t1(a,b)", 0, 0, 0); 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sqlite_close(db); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for(i=0; i<10000; i++){ 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pid = fork(); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if( pid==0 ){ 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sched_yield(); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) do_some_sql(parent); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) printf("test %d, pid=%d\n", i, pid); 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) usleep(rand()%10000 + 1000); 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kill(pid, SIGKILL); 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) waitpid(pid, &status, 0); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 97