AryaWu/sqlite

1 2# 2007 Aug 133#4# The author disclaims copyright to this source code.  In place of5# a legal notice, here is a blessing:6#7#    May you do good and not evil.8#    May you find forgiveness for yourself and forgive others.9#    May you share freely, never taking more than you give.10#11#***********************************************************************12# 13# This file tests aspects of recovery from a malloc() failure14# in a CREATE INDEX statement.15#16# $Id: crash5.test,v 1.3 2008/07/12 14:52:20 drh Exp $17 18set testdir [file dirname $argv0]19source $testdir/tester.tcl20 21# Only run these tests if memory debugging is turned on.22#23ifcapable !crashtest||!memorymanage {24   puts "Skipping crash5 tests: not compiled with -DSQLITE_ENABLE_MEMORY_MANAGEMENT..."25   finish_test26   return27}28 29db close30 31for {set ii 0} {$ii < 10} {incr ii} {32  for {set jj 1} {$jj < 100} {incr jj} {33 34    # Set up the database so that it is an auto-vacuum database 35    # containing a single table (root page 3) with a single row. 36    # The row has an overflow page (page 4).37    forcedelete test.db test.db-journal38    sqlite3 db test.db39    set c [string repeat 3 1500]40    db eval {41      pragma auto_vacuum = 1;42      CREATE TABLE t1(a, b, c);43      INSERT INTO t1 VALUES('1111111111', '2222222222', $c);44    }45    db close46 47    do_test crash5-$ii.$jj.1 {48      crashsql -delay 1 -file test.db-journal -seed $ii -tclbody [join [list \49        [list set iFail $jj] {50        proc get_pwd {} {51          if {$::tcl_platform(platform) eq "windows"} {52            if {[info exists ::env(ComSpec)]} {53              set comSpec $::env(ComSpec)54            } else {55              # NOTE: Hard-code the typical default value.56              set comSpec {C:\Windows\system32\cmd.exe}57            }58            return [string map [list \\ /] \59              [string trim [exec -- $comSpec /c echo %CD%]]]60          } else {61            return [pwd]62          }63        }64        sqlite3_crashparams 0 [file join [get_pwd] test.db-journal]65      66        # Begin a transaction and evaluate a "CREATE INDEX" statement67        # with the iFail'th malloc() set to fail. This operation will68        # have to move the current contents of page 4 (the overflow69        # page) to make room for the new root page. The bug is that70        # if malloc() fails at a particular point in sqlite3PagerMovepage(),71        # sqlite mistakenly thinks that the page being moved (page 4) has 72        # been safely synced into the journal. If the page is written73        # to later in the transaction, it may be written out to the database74        # before the relevant part of the journal has been synced.75        #76        db eval BEGIN77        sqlite3_memdebug_fail $iFail -repeat 078        set rc [catch {db eval { CREATE UNIQUE INDEX i1 ON t1(a); }} msg]79#       puts "$msg ac=[sqlite3_get_autocommit db] iFail=$iFail"80#       puts "fail=[sqlite3_memdebug_fail -1]"81      82        if {$rc} {83          # If the transaction is still active (it may not be if the malloc()84          # failure occurred in the OS layer), write to the database. Make sure85          # page 4 is among those written.86          #87          if {![sqlite3_get_autocommit db]} {88            db eval {89              DELETE FROM t1;  -- This will put page 4 on the free list.90              INSERT INTO t1 VALUES('111111111', '2222222222', '33333333');91              INSERT INTO t1 SELECT * FROM t1;                     -- 292              INSERT INTO t1 SELECT * FROM t1;                     -- 493              INSERT INTO t1 SELECT * FROM t1;                     -- 894              INSERT INTO t1 SELECT * FROM t1;                     -- 1695              INSERT INTO t1 SELECT * FROM t1;                     -- 3296              INSERT INTO t1 SELECT * FROM t1 WHERE rowid%2;       -- 4897            }98          }99          100          # If the right malloc() failed during the 'CREATE INDEX' above and101          # the transaction was not rolled back, then the sqlite cache now 102          # has a dirty page 4 that it incorrectly believes is already safely103          # in the synced part of the journal file. When 104          # sqlite3_release_memory() is called sqlite tries to free memory105          # by writing page 4 out to the db file. If it crashes later on,106          # before syncing the journal... Corruption!107          #108          sqlite3_crashparams 1 [file join [get_pwd] test.db-journal]109          sqlite3_release_memory 8092110        }111      }]] {}112      expr 1113    } {1}114  115    sqlite3 db test.db116    do_test crash5-$ii.$jj.2 {117      db eval {pragma integrity_check}118    } {ok}119    do_test crash5-$ii.$jj.3 {120      db eval {SELECT * FROM t1}121    } [list 1111111111 2222222222 $::c]122    db close123  }124}125 126 127finish_test128