summaryrefslogtreecommitdiffstats
path: root/test/snapshot_fault.test
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-13 14:07:11 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-13 14:07:11 +0000
commit63847496f14c813a5d80efd5b7de0f1294ffe1e3 (patch)
tree01c7571c7c762ceee70638549a99834fdd7c411b /test/snapshot_fault.test
parentInitial commit. (diff)
downloadsqlite3-63847496f14c813a5d80efd5b7de0f1294ffe1e3.tar.xz
sqlite3-63847496f14c813a5d80efd5b7de0f1294ffe1e3.zip
Adding upstream version 3.45.1.upstream/3.45.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'test/snapshot_fault.test')
-rw-r--r--test/snapshot_fault.test252
1 files changed, 252 insertions, 0 deletions
diff --git a/test/snapshot_fault.test b/test/snapshot_fault.test
new file mode 100644
index 0000000..10c5094
--- /dev/null
+++ b/test/snapshot_fault.test
@@ -0,0 +1,252 @@
+# 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 {
+ catch { db2 close }
+ 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