mirror of
				https://github.com/sqlite/sqlite.git
				synced 2025-10-30 07:05:46 +03:00 
			
		
		
		
	SQLITE_ENABLE_SNAPSHOT. FossilOrigin-Name: 6821c61f1d71be2be7f867e59fd94582a1eaf45a4d1fb88be995807a77cc22ea
		
			
				
	
	
		
			252 lines
		
	
	
		
			7.3 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
			
		
		
	
	
			252 lines
		
	
	
		
			7.3 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
| # 2015 December 10
 | |
| #
 | |
| # The author disclaims copyright to this source code.  In place of
 | |
| # a legal notice, here is a blessing:
 | |
| #
 | |
| #    May you do good and not evil.
 | |
| #    May you find forgiveness for yourself and forgive others.
 | |
| #    May you share freely, never taking more than you give.
 | |
| #
 | |
| #***********************************************************************
 | |
| # This file implements regression tests for SQLite library. The focus
 | |
| # of this file is the sqlite3_snapshot_xxx() APIs.
 | |
| #
 | |
| 
 | |
| set testdir [file dirname $argv0]
 | |
| source $testdir/tester.tcl
 | |
| ifcapable !snapshot {finish_test; return}
 | |
| set testprefix snapshot_fault
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| # Check that an sqlite3_snapshot_open() client cannot be tricked into
 | |
| # reading a corrupt snapshot even if a second client fails while 
 | |
| # checkpointing the db.
 | |
| #
 | |
| do_faultsim_test 1.0 -prep {
 | |
|   faultsim_delete_and_reopen
 | |
|   sqlite3 db2 test.db
 | |
|   db2 eval { 
 | |
|     CREATE TABLE t1(a, b UNIQUE, c UNIQUE);
 | |
|     INSERT INTO t1 VALUES(1, randomblob(500), randomblob(500));
 | |
|     INSERT INTO t1 VALUES(2, randomblob(500), randomblob(500));
 | |
|     PRAGMA journal_mode = wal;
 | |
|     INSERT INTO t1 VALUES(3, randomblob(500), randomblob(500));
 | |
|     BEGIN;
 | |
|       SELECT a FROM t1;
 | |
|   }
 | |
|   set ::snapshot [sqlite3_snapshot_get db2 main] 
 | |
|   db2 eval COMMIT
 | |
|   db2 eval {
 | |
|     UPDATE t1 SET b=randomblob(501), c=randomblob(501) WHERE a=1;
 | |
|     INSERT INTO t1 VALUES(4, randomblob(500), randomblob(500));
 | |
|     INSERT INTO t1 VALUES(5, randomblob(500), randomblob(500));
 | |
|     INSERT INTO t1 VALUES(6, randomblob(500), randomblob(500));
 | |
|   }
 | |
| } -body {
 | |
|   db eval { PRAGMA wal_checkpoint }
 | |
| } -test {
 | |
|   db2 eval BEGIN
 | |
|   if {[catch { sqlite3_snapshot_open db2 main $::snapshot } msg]} {
 | |
|     if {$msg != "SQLITE_ERROR_SNAPSHOT" && $msg != "SQLITE_BUSY"} {
 | |
|       error "error is $msg" 
 | |
|     }
 | |
|   } else {
 | |
|     set res [db2 eval { 
 | |
|       SELECT a FROM t1;
 | |
|       PRAGMA integrity_check;
 | |
|     }]
 | |
|     if {$res != "1 2 3 ok"} { error "res is $res" }
 | |
|   }
 | |
| 
 | |
|   sqlite3_snapshot_free $::snapshot
 | |
| }
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| # This test is similar to the previous one. Except, after the 
 | |
| # "PRAGMA wal_checkpoint" command fails the db is closed and reopened
 | |
| # so as to require wal file recovery. It should not be possible to open
 | |
| # a snapshot that is part of the body of a recovered wal file.
 | |
| #
 | |
| do_faultsim_test 2.0 -prep {
 | |
|   faultsim_delete_and_reopen
 | |
|   db eval { 
 | |
|     CREATE TABLE t1(a, b UNIQUE, c UNIQUE);
 | |
|     INSERT INTO t1 VALUES(1, randomblob(500), randomblob(500));
 | |
|     INSERT INTO t1 VALUES(2, randomblob(500), randomblob(500));
 | |
|     PRAGMA journal_mode = wal;
 | |
|     INSERT INTO t1 VALUES(3, randomblob(500), randomblob(500));
 | |
|     BEGIN;
 | |
|       SELECT a FROM t1;
 | |
|   }
 | |
|   set ::snapshot [sqlite3_snapshot_get db main] 
 | |
|   db eval COMMIT
 | |
| 
 | |
|   db eval {
 | |
|     UPDATE t1 SET b=randomblob(501), c=randomblob(501) WHERE a=1;
 | |
|     INSERT INTO t1 VALUES(4, randomblob(500), randomblob(500));
 | |
|     INSERT INTO t1 VALUES(5, randomblob(500), randomblob(500));
 | |
|     INSERT INTO t1 VALUES(6, randomblob(500), randomblob(500));
 | |
|   }
 | |
| } -body {
 | |
|   db eval { PRAGMA wal_checkpoint }
 | |
| } -test {
 | |
| 
 | |
|   db_save
 | |
|   db close
 | |
|   db_restore_and_reopen
 | |
|   db eval { SELECT * FROM t1 }
 | |
|   
 | |
|   db eval BEGIN
 | |
|   if {[catch { sqlite3_snapshot_open db main $::snapshot } msg]} {
 | |
|     if {$msg != "SQLITE_ERROR_SNAPSHOT" && $msg != "SQLITE_BUSY"} {
 | |
|       error "error is $msg" 
 | |
|     }
 | |
|   } else {
 | |
|     # This branch should actually never be taken. But it was useful in
 | |
|     # determining whether or not this test was actually working (by 
 | |
|     # running a modified version of SQLite that allowed snapshots to be
 | |
|     # opened following a recovery).
 | |
|     error "TEST HAS FAILED"
 | |
| 
 | |
|     set res [db eval { 
 | |
|       SELECT a FROM t1;
 | |
|       PRAGMA integrity_check;
 | |
|     }]
 | |
|     if {$res != "1 2 3 ok"} { error "res is $res" }
 | |
|   }
 | |
| 
 | |
|   sqlite3_snapshot_free $::snapshot
 | |
| }
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| # Test the handling of faults that occur within sqlite3_snapshot_open().
 | |
| #
 | |
| do_faultsim_test 3.0 -prep {
 | |
|   faultsim_delete_and_reopen
 | |
|   db eval { 
 | |
|     CREATE TABLE t1(a, b UNIQUE, c UNIQUE);
 | |
|     INSERT INTO t1 VALUES(1, randomblob(500), randomblob(500));
 | |
|     INSERT INTO t1 VALUES(2, randomblob(500), randomblob(500));
 | |
|     PRAGMA journal_mode = wal;
 | |
|     INSERT INTO t1 VALUES(3, randomblob(500), randomblob(500));
 | |
|     BEGIN;
 | |
|       SELECT a FROM t1;
 | |
|   }
 | |
|   set ::snapshot [sqlite3_snapshot_get db main] 
 | |
|   db eval COMMIT
 | |
|   db eval {
 | |
|     UPDATE t1 SET b=randomblob(501), c=randomblob(501) WHERE a=1;
 | |
|     INSERT INTO t1 VALUES(4, randomblob(500), randomblob(500));
 | |
|     INSERT INTO t1 VALUES(5, randomblob(500), randomblob(500));
 | |
|     INSERT INTO t1 VALUES(6, randomblob(500), randomblob(500));
 | |
|     BEGIN;
 | |
|   }
 | |
| } -body {
 | |
|   if { [catch { sqlite3_snapshot_open db main $::snapshot } msg] } {
 | |
|     error $msg
 | |
|   }
 | |
| } -test {
 | |
|   faultsim_test_result {0 {}} {1 SQLITE_IOERR} {1 SQLITE_NOMEM} \
 | |
|                               {1 SQLITE_IOERR_NOMEM} {1 SQLITE_IOERR_READ}
 | |
|   if {$testrc==0} {
 | |
|     set res [db eval { 
 | |
|       SELECT a FROM t1;
 | |
|       PRAGMA integrity_check;
 | |
|     }]
 | |
|     if {$res != "1 2 3 ok"} { error "res is $res" }
 | |
|   }
 | |
| 
 | |
|   sqlite3_snapshot_free $::snapshot
 | |
| }
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| # Test the handling of faults that occur within sqlite3_snapshot_recover().
 | |
| #
 | |
| reset_db
 | |
| do_execsql_test 4.0 {
 | |
|   PRAGMA journal_mode = wal;
 | |
|   CREATE TABLE t1(zzz);
 | |
|   INSERT INTO t1 VALUES('abc');
 | |
|   INSERT INTO t1 VALUES('def');
 | |
| } {wal}
 | |
| faultsim_save_and_close
 | |
| 
 | |
| do_test 4.0.1 {
 | |
|   faultsim_restore_and_reopen
 | |
|   db eval { SELECT * FROM sqlite_master } 
 | |
|   sqlite3_snapshot_recover db main
 | |
| } {}
 | |
| db close
 | |
| 
 | |
| do_faultsim_test 4.0 -faults oom* -prep {
 | |
|   faultsim_restore_and_reopen
 | |
|   db eval { SELECT * FROM sqlite_master } 
 | |
| } -body {
 | |
|   sqlite3_snapshot_recover db main
 | |
| } -test {
 | |
|   faultsim_test_result {0 {}} {1 SQLITE_NOMEM} {1 SQLITE_IOERR_NOMEM}
 | |
| }
 | |
| 
 | |
| # The following test cases contrive to call sqlite3_snapshot_recover()
 | |
| # before all pages of the *-shm file have been mapped. This tests an
 | |
| # extra branch of error handling logic in snapshot_recover().
 | |
| #
 | |
| reset_db
 | |
| do_execsql_test 4.1.0 {
 | |
|   PRAGMA page_size = 512;
 | |
|   PRAGMA journal_mode = wal;
 | |
|   PRAGMA wal_autocheckpoint = 0;
 | |
|   CREATE TABLE t1(zzz);
 | |
|   INSERT INTO t1 VALUES(randomblob( 500 * 9500 ));
 | |
|   PRAGMA user_version = 211;
 | |
| } {wal 0}
 | |
| 
 | |
| do_test 4.1.1 {
 | |
|   list [file size test.db-shm] [file size test.db]
 | |
| } {98304 512}
 | |
| 
 | |
| faultsim_save_and_close
 | |
| do_faultsim_test 4.1 -faults shm* -prep {
 | |
|   catch { db2 close } 
 | |
|   catch { db close } 
 | |
|   faultsim_restore_and_reopen
 | |
|   sqlite3 db2 test.db
 | |
|   db2 eval { SELECT * FROM sqlite_master } 
 | |
|   db eval BEGIN
 | |
|   sqlite3_snapshot_get_blob db main
 | |
|   db eval COMMIT
 | |
| } -body {
 | |
|   sqlite3_snapshot_recover db main
 | |
| } -test {
 | |
|   faultsim_test_result {0 {}} {1 SQLITE_IOERR}
 | |
| }
 | |
| 
 | |
| #-------------------------------------------------------------------------
 | |
| # Test the handling of faults that occur within sqlite3_snapshot_get().
 | |
| #
 | |
| reset_db
 | |
| do_execsql_test 5.0 {
 | |
|   PRAGMA page_size = 512;
 | |
|   PRAGMA journal_mode = wal;
 | |
|   PRAGMA wal_autocheckpoint = 0;
 | |
|   CREATE TABLE t1(zzz);
 | |
|   INSERT INTO t1 VALUES(randomblob( 5000 ));
 | |
|   PRAGMA user_version = 211;
 | |
| } {wal 0}
 | |
| faultsim_save_and_close
 | |
| 
 | |
| do_faultsim_test 5 -prep {
 | |
|   faultsim_restore_and_reopen
 | |
|   execsql { SELECT count(*) FROM sqlite_master }
 | |
|   execsql BEGIN
 | |
| } -body {
 | |
|   sqlite3_snapshot_get_blob db main
 | |
|   set {} {}
 | |
| } -test {
 | |
|   execsql END
 | |
|   faultsim_test_result {0 {}} {1 SQLITE_IOERR} {1 SQLITE_NOMEM}
 | |
| }
 | |
| 
 | |
| 
 | |
| finish_test
 |