AryaWu/sqlite

1# 2012 January 122#3# The author disclaims copyright to this source code.  In place of4# a legal notice, here is a blessing:5#6#    May you do good and not evil.7#    May you find forgiveness for yourself and forgive others.8#    May you share freely, never taking more than you give.9#10#***********************************************************************11#12 13set testdir [file dirname $argv0]14source $testdir/tester.tcl15set testprefix corruptF16 17# Do not use a codec for tests in this file, as the database file is18# manipulated directly using tcl scripts (using the [hexio_write] command).19#20do_not_use_codec21 22# These tests deal with corrupt database files23#24database_may_be_corrupt25 26proc str {i} { format %08d $i }27 28# Create a 6 page database containing a single table - t1. Table t1 29# consists of page 2 (the root page) and pages 5 and 6 (leaf pages). 30# Database pages 3 and 4 are on the free list.31#32proc create_test_db {} {33  catch { db close }34  forcedelete test.db35  sqlite3 db test.db36  db func str str37  execsql {38    PRAGMA auto_vacuum = 0;39    PRAGMA page_size = 1024;40    CREATE TABLE t1(x);         /* root page = 2 */41    CREATE TABLE t2(x);         /* root page = 3 */42    CREATE TABLE t3(x);         /* root page = 4 */43 44    INSERT INTO t1 VALUES(str(1));45    INSERT INTO t1 SELECT str(rowid+1) FROM t1;46    INSERT INTO t1 SELECT str(rowid+2) FROM t1;47    INSERT INTO t1 SELECT str(rowid+4) FROM t1;48    INSERT INTO t1 SELECT str(rowid+8) FROM t1;49    INSERT INTO t1 SELECT str(rowid+16) FROM t1;50    INSERT INTO t1 SELECT str(rowid+32) FROM t1;51    INSERT INTO t1 SELECT str(rowid+64) FROM t1;52    DROP TABLE t2;53    DROP TABLE t3;54  }55  db close56}57 58do_test 1.1 { create_test_db } {}59 60# Check the db is as we expect. 6 pages in total, with 3 and 4 on the free61# list. Page 3 is the free list trunk and page 4 is a leaf.62#63do_test 1.2 { file size test.db } [expr 6*1024]64do_test 1.3 { hexio_read test.db 32 4 } 0000000365do_test 1.4 { hexio_read test.db [expr 2*1024] 12 } 00000000000000010000000466 67# Change the free-list entry to page 6 and reopen the db file.68do_test 1.5 { 69  hexio_write test.db [expr 2*1024 + 8] 00000006 70  sqlite3 db test.db71} {}72 73# Now create a new table in the database file. The root of the new table74# is page 6, which is also the right-most leaf page in table t1.75#76do_execsql_test 1.6 { 77  CREATE TABLE t4(x);78  SELECT * FROM sqlite_master;79} {80  table t1 t1 2 {CREATE TABLE t1(x)} 81  table t4 t4 6 {CREATE TABLE t4(x)}82}83 84# At one point this was causing an assert to fail.85#86# This statement opens a cursor on table t1 and does a full table scan. As87# each row is visited, it is copied into table t4. There is no temporary88# table.89#90# When the t1 cursor reaches page 6 (which is both the right-most leaf of91# t1 and the root of t4), it continues to iterate through the keys within92# it (which at this point are keys that have been inserted into t4). And93# for each row visited, another row is inserted into page 6 - it being the94# root page of t4. Eventually, page 6 becomes full and the height of the95# b-tree for table t4 increased. From the point of view of the t1 cursor,96# this unexpectedly reduces the number of keys on page 6 in the middle of97# its iteration, which causes an assert() to fail.98#99db_save_and_close100if 1 {101for {set i 0} {$i < 128} {incr i} {102  db_restore_and_reopen103  do_test 1.7.$i { 104    set res [105      catchsql { INSERT INTO t4 SELECT x FROM t1 WHERE rowid>$i }106    ]107    if {$res == "0 {}" || $res == "1 {database disk image is malformed}"} {108      set res ""109    }110    set res111  } {}112}113}114 115do_test 2.1 { create_test_db } {}116do_test 2.2 { file size test.db } [expr 6*1024]117do_test 2.3 { hexio_read test.db 32 4 } 00000003118do_test 2.4 { hexio_read test.db [expr 2*1024] 12 } 000000000000000100000004119 120# Change the free-list entry to page 5 and reopen the db file.121do_test 2.5 { 122  hexio_write test.db [expr 2*1024 + 8] 00000005 123  sqlite3 db test.db124} {}125 126# Now create a new table in the database file. The root of the new table127# is page 5, which is also the right-most leaf page in table t1.128#129do_execsql_test 2.6 { 130  CREATE TABLE t4(x);131  SELECT * FROM sqlite_master;132} {133  table t1 t1 2 {CREATE TABLE t1(x)} 134  table t4 t4 5 {CREATE TABLE t4(x)}135}136 137db_save_and_close138for {set i 127} {$i >= 0} {incr i -1} {139  db_restore_and_reopen140  do_test 2.7.$i { 141    set res [142      catchsql { 143        INSERT INTO t4 SELECT x FROM t1 WHERE rowid<$i ORDER BY rowid DESC 144      }145    ]146    if {$res == "0 {}" || $res == "1 {database disk image is malformed}"} {147      set res ""148    }149    set res150  } {}151}152 153finish_test154