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
|