# 2004 August 30

#

# 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.

#

# This file implements tests to make sure SQLite does not crash or

# segfault if it sees a corrupt database file.

#

# $Id: corrupt.test,v 1.10 2008/08/25 12:14:09 drh Exp $

catch {file delete -force test.db test.db-journal test.bu}

set testdir [file dirname $argv0]

source $testdir/tester.tcl

# Construct a large database for testing.

#

do_test corrupt-1.1 {

  execsql {

    BEGIN;

    CREATE TABLE t1(x);

    INSERT INTO t1 VALUES(randstr(100,100));

    INSERT INTO t1 VALUES(randstr(90,90));

    INSERT INTO t1 VALUES(randstr(80,80));

    INSERT INTO t1 SELECT x || randstr(5,5) FROM t1;

    INSERT INTO t1 SELECT x || randstr(6,6) FROM t1;

    INSERT INTO t1 SELECT x || randstr(7,7) FROM t1;

    INSERT INTO t1 SELECT x || randstr(8,8) FROM t1;

    INSERT INTO t1 VALUES(randstr(3000,3000));

    INSERT INTO t1 SELECT x || randstr(9,9) FROM t1;

    INSERT INTO t1 SELECT x || randstr(10,10) FROM t1;

    INSERT INTO t1 SELECT x || randstr(11,11) FROM t1;

    INSERT INTO t1 SELECT x || randstr(12,12) FROM t1;

    CREATE INDEX t1i1 ON t1(x);

    CREATE TABLE t2 AS SELECT * FROM t1;

    DELETE FROM t2 WHERE rowid%5!=0;

    COMMIT;

  }

} {}

integrity_check corrupt-1.2

# Copy file $from into $to

#

proc copy_file {from to} {

  set f [open $from]

  fconfigure $f -translation binary

  set t [open $to w]

  fconfigure $t -translation binary

  puts -nonewline $t [read $f [file size $from]]

  close $t

  close $f

}

# Setup for the tests.  Make a backup copy of the good database in test.bu.

# Create a string of garbage data that is 256 bytes long.

#

copy_file test.db test.bu

set fsize [file size test.db]

set junk "abcdefghijklmnopqrstuvwxyz0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"

while {[string length $junk]<256} {append junk $junk}

set junk [string range $junk 0 255]

# Go through the database and write garbage data into each 256 segment

# of the file.  Then do various operations on the file to make sure that

# the database engine can recover gracefully from the corruption.

#

for {set i [expr {1*256}]} {$i<$fsize-256} {incr i 256} {

  set tn [expr {$i/256}]

  db close

  copy_file test.bu test.db

  set fd [open test.db r+]

  fconfigure $fd -translation binary

  seek $fd $i

  puts -nonewline $fd $junk

  close $fd

  do_test corrupt-2.$tn.1 {

    sqlite3 db test.db

    catchsql {SELECT count(*) FROM sqlite_master}

    set x {}

  } {}

  do_test corrupt-2.$tn.2 {

    catchsql {SELECT count(*) FROM t1}

    set x {}

  } {}

  do_test corrupt-2.$tn.3 {

    catchsql {SELECT count(*) FROM t1 WHERE x>'abcdef'}

    set x {}

  } {}

  do_test corrupt-2.$tn.4 {

    catchsql {SELECT count(*) FROM t2}

    set x {}

  } {}

  do_test corrupt-2.$tn.5 {

    catchsql {CREATE TABLE t3 AS SELECT * FROM t1}

    set x {}

  } {}

  do_test corrupt-2.$tn.6 {

    catchsql {DROP TABLE t1}

    set x {}

  } {}

  do_test corrupt-2.$tn.7 {

    catchsql {PRAGMA integrity_check}

    set x {}

  } {}

  # Check that no page references were leaked.

  do_test corrupt-2.$tn.8 {

    set bt [btree_from_db db]

    db_enter db

    array set stats [btree_pager_stats $bt]

    db_leave db

    set stats(ref)

  } {0}

}  

#------------------------------------------------------------------------

# For these tests, swap the rootpage entries of t1 (a table) and t1i1 (an

# index on t1) in sqlite_master. Then perform a few different queries

# and make sure this is detected as corruption.

#

do_test corrupt-3.1 {

  db close

  copy_file test.bu test.db

  sqlite3 db test.db

  list

} {}

do_test corrupt-3.2 {

  set t1_r [execsql {SELECT rootpage FROM sqlite_master WHERE name = 't1i1'}]

  set t1i1_r [execsql {SELECT rootpage FROM sqlite_master WHERE name = 't1'}]

  set cookie [expr [execsql {PRAGMA schema_version}] + 1]

  execsql "

    PRAGMA writable_schema = 1;

    UPDATE sqlite_master SET rootpage = $t1_r WHERE name = 't1';

    UPDATE sqlite_master SET rootpage = $t1i1_r WHERE name = 't1i1';

    PRAGMA writable_schema = 0;

    PRAGMA schema_version = $cookie;

  "

} {}

# This one tests the case caught by code in checkin [2313].

do_test corrupt-3.3 {

  db close

  sqlite3 db test.db

  catchsql {

    INSERT INTO t1 VALUES('abc');

  }

} {1 {database disk image is malformed}}

do_test corrupt-3.4 {

  db close

  sqlite3 db test.db

  catchsql {

    SELECT * FROM t1;

  }

} {1 {database disk image is malformed}}

do_test corrupt-3.5 {

  db close

  sqlite3 db test.db

  catchsql {

    SELECT * FROM t1 WHERE oid = 10;

  }

} {1 {database disk image is malformed}}

do_test corrupt-3.6 {

  db close

  sqlite3 db test.db

  catchsql {

    SELECT * FROM t1 WHERE x = 'abcde';

  }

} {1 {database disk image is malformed}}

finish_test