1b19a2bc6Sdrh# 2001 September 15 2ed7c855cSdrh# 3b19a2bc6Sdrh# The author disclaims copyright to this source code. In place of 4b19a2bc6Sdrh# a legal notice, here is a blessing: 5ed7c855cSdrh# 6b19a2bc6Sdrh# May you do good and not evil. 7b19a2bc6Sdrh# May you find forgiveness for yourself and forgive others. 8b19a2bc6Sdrh# May you share freely, never taking more than you give. 9ed7c855cSdrh# 10ed7c855cSdrh#*********************************************************************** 11ed7c855cSdrh# 12f3a65f7eSdrh# This file attempts to check the behavior of the SQLite library in 13f3a65f7eSdrh# an out-of-memory situation. When compiled with -DSQLITE_DEBUG=1, 14f3a65f7eSdrh# the SQLite library accepts a special command (sqlite3_memdebug_fail N C) 15f3a65f7eSdrh# which causes the N-th malloc to fail. This special feature is used 16f3a65f7eSdrh# to see what happens in the library if a malloc were to really fail 17f3a65f7eSdrh# due to an out-of-memory situation. 18f3a65f7eSdrh# 19c3927537Sdrh# $Id: malloc.test,v 1.81 2009/06/24 13:13:45 drh Exp $ 20ed7c855cSdrh 21ed7c855cSdrhset testdir [file dirname $argv0] 22ed7c855cSdrhsource $testdir/tester.tcl 2376ccd89dSdanset ::testprefix malloc 24ed7c855cSdrh 25d09414cdSdanielk1977 26ed7c855cSdrh# Only run these tests if memory debugging is turned on. 27ed7c855cSdrh# 28eee4c8caSdrhsource $testdir/malloc_common.tcl 29eee4c8caSdrhif {!$MEMDEBUG} { 30261919ccSdanielk1977 puts "Skipping malloc tests: not compiled with -DSQLITE_MEMDEBUG..." 31ed7c855cSdrh finish_test 32ed7c855cSdrh return 33ed7c855cSdrh} 34ed7c855cSdrh 35de4fcfddSdrh# Do a couple of memory dumps just to exercise the memory dump logic 36de4fcfddSdrh# that that we can say that we have. 37de4fcfddSdrh# 38de4fcfddSdrhputs stderr "This is a test. Ignore the error that follows:" 39de4fcfddSdrhsqlite3_memdebug_dump $testdir 40de4fcfddSdrhputs "Memory dump to file memdump.txt..." 41de4fcfddSdrhsqlite3_memdebug_dump memdump.txt 42de4fcfddSdrh 434152e677Sdanielk1977ifcapable bloblit&&subquery { 44c08d405dSdanielk1977 do_malloc_test 1 -tclprep { 45c08d405dSdanielk1977 db close 46c08d405dSdanielk1977 } -tclbody { 47c08d405dSdanielk1977 if {[catch {sqlite3 db test.db}]} { 48c08d405dSdanielk1977 error "out of memory" 49c08d405dSdanielk1977 } 50ae72d982Sdanielk1977 sqlite3_extended_result_codes db 1 51c08d405dSdanielk1977 } -sqlbody { 52344a6276Sdrh DROP TABLE IF EXISTS t1; 53ed7c855cSdrh CREATE TABLE t1( 54ed7c855cSdrh a int, b float, c double, d text, e varchar(20), 55ed7c855cSdrh primary key(a,b,c) 56ed7c855cSdrh ); 57ed7c855cSdrh CREATE INDEX i1 ON t1(a,b); 5876f80796Sdrh INSERT INTO t1 VALUES(1,2.3,4.5,'hi',x'746865726500'); 59ed7c855cSdrh INSERT INTO t1 VALUES(6,7.0,0.8,'hello','out yonder'); 60ed7c855cSdrh SELECT * FROM t1; 61d400728aSdrh SELECT avg(b) FROM t1 GROUP BY a HAVING b>20.0; 626d4abfbeSdrh DELETE FROM t1 WHERE a IN (SELECT min(a) FROM t1); 63ade86483Sdrh SELECT count(*), group_concat(e) FROM t1; 6415cdbebeSdanielk1977 SELECT b FROM t1 ORDER BY 1 COLLATE nocase; 65ed7c855cSdrh } 664152e677Sdanielk1977} 67d400728aSdrh 68b5548a8bSdanielk1977# Ensure that no file descriptors were leaked. 69b5548a8bSdanielk1977do_test malloc-1.X { 70b5548a8bSdanielk1977 catch {db close} 71b5548a8bSdanielk1977 set sqlite_open_file_count 72b5548a8bSdanielk1977} {0} 73b5548a8bSdanielk1977 744152e677Sdanielk1977ifcapable subquery { 75c08d405dSdanielk1977 do_malloc_test 2 -sqlbody { 76fc23314aSdrh CREATE TABLE t1(a int, b int default 'abc', c int default 1); 77d400728aSdrh CREATE INDEX i1 ON t1(a,b); 785f3b4ab5Sdrh INSERT INTO t1 VALUES(1,1,'99 abcdefghijklmnopqrstuvwxyz'); 795f3b4ab5Sdrh INSERT INTO t1 VALUES(2,4,'98 abcdefghijklmnopqrstuvwxyz'); 805f3b4ab5Sdrh INSERT INTO t1 VALUES(3,9,'97 abcdefghijklmnopqrstuvwxyz'); 815f3b4ab5Sdrh INSERT INTO t1 VALUES(4,16,'96 abcdefghijklmnopqrstuvwxyz'); 825f3b4ab5Sdrh INSERT INTO t1 VALUES(5,25,'95 abcdefghijklmnopqrstuvwxyz'); 835f3b4ab5Sdrh INSERT INTO t1 VALUES(6,36,'94 abcdefghijklmnopqrstuvwxyz'); 846d4abfbeSdrh SELECT 'stuff', count(*) as 'other stuff', max(a+10) FROM t1; 856d4abfbeSdrh UPDATE t1 SET b=b||b||b||b; 86d400728aSdrh UPDATE t1 SET b=a WHERE a in (10,12,22); 876d4abfbeSdrh INSERT INTO t1(c,b,a) VALUES(20,10,5); 886d4abfbeSdrh INSERT INTO t1 SELECT * FROM t1 896d4abfbeSdrh WHERE a IN (SELECT a FROM t1 WHERE a<10); 906d4abfbeSdrh DELETE FROM t1 WHERE a>=10; 91d400728aSdrh DROP INDEX i1; 926d4abfbeSdrh DELETE FROM t1; 936d4abfbeSdrh } 944152e677Sdanielk1977} 956d4abfbeSdrh 96b5548a8bSdanielk1977# Ensure that no file descriptors were leaked. 97b5548a8bSdanielk1977do_test malloc-2.X { 98b5548a8bSdanielk1977 catch {db close} 99b5548a8bSdanielk1977 set sqlite_open_file_count 100b5548a8bSdanielk1977} {0} 101b5548a8bSdanielk1977 102c08d405dSdanielk1977do_malloc_test 3 -sqlbody { 1036d4abfbeSdrh BEGIN TRANSACTION; 1046d4abfbeSdrh CREATE TABLE t1(a int, b int, c int); 1056d4abfbeSdrh CREATE INDEX i1 ON t1(a,b); 1065f3b4ab5Sdrh INSERT INTO t1 VALUES(1,1,99); 1075f3b4ab5Sdrh INSERT INTO t1 VALUES(2,4,98); 1085f3b4ab5Sdrh INSERT INTO t1 VALUES(3,9,97); 1095f3b4ab5Sdrh INSERT INTO t1 VALUES(4,16,96); 1105f3b4ab5Sdrh INSERT INTO t1 VALUES(5,25,95); 1115f3b4ab5Sdrh INSERT INTO t1 VALUES(6,36,94); 1126d4abfbeSdrh INSERT INTO t1(c,b,a) VALUES(20,10,5); 1136d4abfbeSdrh DELETE FROM t1 WHERE a>=10; 1146d4abfbeSdrh DROP INDEX i1; 1156d4abfbeSdrh DELETE FROM t1; 1166d4abfbeSdrh ROLLBACK; 1176d4abfbeSdrh} 118c08d405dSdanielk1977 119b5548a8bSdanielk1977 120b5548a8bSdanielk1977# Ensure that no file descriptors were leaked. 121b5548a8bSdanielk1977do_test malloc-3.X { 122b5548a8bSdanielk1977 catch {db close} 123b5548a8bSdanielk1977 set sqlite_open_file_count 124b5548a8bSdanielk1977} {0} 125b5548a8bSdanielk1977 1264152e677Sdanielk1977ifcapable subquery { 127c08d405dSdanielk1977 do_malloc_test 4 -sqlbody { 1286d4abfbeSdrh BEGIN TRANSACTION; 1296d4abfbeSdrh CREATE TABLE t1(a int, b int, c int); 1306d4abfbeSdrh CREATE INDEX i1 ON t1(a,b); 1315f3b4ab5Sdrh INSERT INTO t1 VALUES(1,1,99); 1325f3b4ab5Sdrh INSERT INTO t1 VALUES(2,4,98); 1335f3b4ab5Sdrh INSERT INTO t1 VALUES(3,9,97); 1345f3b4ab5Sdrh INSERT INTO t1 VALUES(4,16,96); 1355f3b4ab5Sdrh INSERT INTO t1 VALUES(5,25,95); 1365f3b4ab5Sdrh INSERT INTO t1 VALUES(6,36,94); 1376d4abfbeSdrh UPDATE t1 SET b=a WHERE a in (10,12,22); 1386d4abfbeSdrh INSERT INTO t1 SELECT * FROM t1 1396d4abfbeSdrh WHERE a IN (SELECT a FROM t1 WHERE a<10); 1406d4abfbeSdrh DROP INDEX i1; 1416d4abfbeSdrh DELETE FROM t1; 1426d4abfbeSdrh COMMIT; 143d400728aSdrh } 1444152e677Sdanielk1977} 145b5548a8bSdanielk1977 146b5548a8bSdanielk1977# Ensure that no file descriptors were leaked. 147b5548a8bSdanielk1977do_test malloc-4.X { 148b5548a8bSdanielk1977 catch {db close} 149b5548a8bSdanielk1977 set sqlite_open_file_count 150b5548a8bSdanielk1977} {0} 151b5548a8bSdanielk1977 1524152e677Sdanielk1977ifcapable trigger { 153c08d405dSdanielk1977 do_malloc_test 5 -sqlbody { 154e4697f5eSdrh BEGIN TRANSACTION; 155e4697f5eSdrh CREATE TABLE t1(a,b); 156e4697f5eSdrh CREATE TABLE t2(x,y); 157932083c6Sdanielk1977 CREATE TRIGGER r1 AFTER INSERT ON t1 WHEN new.a = 2 BEGIN 158e4697f5eSdrh INSERT INTO t2(x,y) VALUES(new.rowid,1); 1599fb3ecbcSdanielk1977 INSERT INTO t2(x,y) SELECT * FROM t2; 1609fb3ecbcSdanielk1977 INSERT INTO t2 SELECT * FROM t2; 16128f45914Sdrh UPDATE t2 SET y=y+1 WHERE x=new.rowid; 16228f45914Sdrh SELECT 123; 16328f45914Sdrh DELETE FROM t2 WHERE x=new.rowid; 164e4697f5eSdrh END; 165e4697f5eSdrh INSERT INTO t1(a,b) VALUES(2,3); 166e4697f5eSdrh COMMIT; 167e4697f5eSdrh } 1684152e677Sdanielk1977} 169b5548a8bSdanielk1977 170b5548a8bSdanielk1977# Ensure that no file descriptors were leaked. 171b5548a8bSdanielk1977do_test malloc-5.X { 172b5548a8bSdanielk1977 catch {db close} 173b5548a8bSdanielk1977 set sqlite_open_file_count 174b5548a8bSdanielk1977} {0} 175b5548a8bSdanielk1977 1764152e677Sdanielk1977ifcapable vacuum { 177c08d405dSdanielk1977 do_malloc_test 6 -sqlprep { 17896fb0dd5Sdanielk1977 BEGIN TRANSACTION; 17996fb0dd5Sdanielk1977 CREATE TABLE t1(a); 18096fb0dd5Sdanielk1977 INSERT INTO t1 VALUES(1); 18196fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 18296fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 18396fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 18496fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 18596fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 18696fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 18796fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 18896fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 18996fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 19096fb0dd5Sdanielk1977 INSERT INTO t1 SELECT a*2 FROM t1; 19196fb0dd5Sdanielk1977 DELETE FROM t1 where rowid%5 = 0; 19296fb0dd5Sdanielk1977 COMMIT; 193c08d405dSdanielk1977 } -sqlbody { 19496fb0dd5Sdanielk1977 VACUUM; 19596fb0dd5Sdanielk1977 } 1964152e677Sdanielk1977} 19796fb0dd5Sdanielk1977 198bb77b753Sdrhautoinstall_test_functions 199c08d405dSdanielk1977do_malloc_test 7 -sqlprep { 20001427a62Sdanielk1977 CREATE TABLE t1(a, b); 20101427a62Sdanielk1977 INSERT INTO t1 VALUES(1, 2); 20201427a62Sdanielk1977 INSERT INTO t1 VALUES(3, 4); 20301427a62Sdanielk1977 INSERT INTO t1 VALUES(5, 6); 204b5402fbfSdanielk1977 INSERT INTO t1 VALUES(7, randstr(1200,1200)); 205c08d405dSdanielk1977} -sqlbody { 206b5402fbfSdanielk1977 SELECT min(a) FROM t1 WHERE a<6 GROUP BY b; 207b5402fbfSdanielk1977 SELECT a FROM t1 WHERE a<6 ORDER BY a; 208b5402fbfSdanielk1977 SELECT b FROM t1 WHERE a>6; 20901427a62Sdanielk1977} 21001427a62Sdanielk1977 211b5402fbfSdanielk1977# This block is designed to test that some malloc failures that may 212b5402fbfSdanielk1977# occur in vdbeapi.c. Specifically, if a malloc failure that occurs 213b5402fbfSdanielk1977# when converting UTF-16 text to integers and real numbers is handled 214b5402fbfSdanielk1977# correctly. 215b5402fbfSdanielk1977# 2168b60e0f1Sdanielk1977# This is done by retrieving a string from the database engine and 2178b60e0f1Sdanielk1977# manipulating it using the sqlite3_column_*** APIs. This doesn't 2188b60e0f1Sdanielk1977# actually return an error to the user when a malloc() fails.. That 2198b60e0f1Sdanielk1977# could be viewed as a bug. 2208b60e0f1Sdanielk1977# 2218b60e0f1Sdanielk1977# These tests only run if UTF-16 support is compiled in. 222b5402fbfSdanielk1977# 2234152e677Sdanielk1977ifcapable utf16 { 224f3a65f7eSdrh set ::STMT {} 225c08d405dSdanielk1977 do_malloc_test 8 -tclprep { 226b5402fbfSdanielk1977 set sql "SELECT '[string repeat abc 20]', '[string repeat def 20]', ?" 227f3a65f7eSdrh set ::STMT [sqlite3_prepare db $sql -1 X] 228b5402fbfSdanielk1977 sqlite3_step $::STMT 229b5402fbfSdanielk1977 if { $::tcl_platform(byteOrder)=="littleEndian" } { 230b5402fbfSdanielk1977 set ::bomstr "\xFF\xFE" 231b5402fbfSdanielk1977 } else { 232b5402fbfSdanielk1977 set ::bomstr "\xFE\xFF" 233b5402fbfSdanielk1977 } 23408746af4Sdanielk1977 append ::bomstr [encoding convertto unicode "123456789_123456789_123456789"] 235c08d405dSdanielk1977 } -tclbody { 236b5402fbfSdanielk1977 sqlite3_column_text16 $::STMT 0 237b5402fbfSdanielk1977 sqlite3_column_int $::STMT 0 238b5402fbfSdanielk1977 sqlite3_column_text16 $::STMT 1 239b5402fbfSdanielk1977 sqlite3_column_double $::STMT 1 240a1644fd8Sdanielk1977 set rc [sqlite3_reset $::STMT] 241a1644fd8Sdanielk1977 if {$rc eq "SQLITE_NOMEM"} {error "out of memory"} 242b5402fbfSdanielk1977 sqlite3_bind_text16 $::STMT 1 $::bomstr 60 243f3a65f7eSdrh #catch {sqlite3_finalize $::STMT} 244f3a65f7eSdrh #if {[lindex [sqlite_malloc_stat] 2]<=0} { 245f3a65f7eSdrh # error "out of memory" 246f3a65f7eSdrh #} 247c08d405dSdanielk1977 } -cleanup { 248f3a65f7eSdrh if {$::STMT!=""} { 249b5402fbfSdanielk1977 sqlite3_finalize $::STMT 250f3a65f7eSdrh set ::STMT {} 251f3a65f7eSdrh } 252b5402fbfSdanielk1977 } 253b5402fbfSdanielk1977} 254b5402fbfSdanielk1977 2558b60e0f1Sdanielk1977# This block tests that malloc() failures that occur whilst commiting 2568b60e0f1Sdanielk1977# a multi-file transaction are handled correctly. 2578b60e0f1Sdanielk1977# 2584397de57Sdanielk1977do_malloc_test 9 -sqlprep { 2598b60e0f1Sdanielk1977 ATTACH 'test2.db' as test2; 2608b60e0f1Sdanielk1977 CREATE TABLE abc1(a, b, c); 2618b60e0f1Sdanielk1977 CREATE TABLE test2.abc2(a, b, c); 2624397de57Sdanielk1977} -sqlbody { 2638b60e0f1Sdanielk1977 BEGIN; 2648b60e0f1Sdanielk1977 INSERT INTO abc1 VALUES(1, 2, 3); 2658b60e0f1Sdanielk1977 INSERT INTO abc2 VALUES(1, 2, 3); 2668b60e0f1Sdanielk1977 COMMIT; 2678b60e0f1Sdanielk1977} 2688b60e0f1Sdanielk1977 2694397de57Sdanielk1977# This block tests malloc() failures that occur while opening a 2704397de57Sdanielk1977# connection to a database. 271dc05efb4Sdrhdo_malloc_test 10 -tclprep { 272dc05efb4Sdrh catch {db2 close} 273dc05efb4Sdrh db close 274fda06befSmistachkin forcedelete test.db test.db-journal 275dc05efb4Sdrh sqlite3 db test.db 276ae72d982Sdanielk1977 sqlite3_extended_result_codes db 1 277dc05efb4Sdrh db eval {CREATE TABLE abc(a, b, c)} 2784397de57Sdanielk1977} -tclbody { 279c5859718Sdanielk1977 db close 280771151b6Sdanielk1977 sqlite3 db2 test.db 281ae72d982Sdanielk1977 sqlite3_extended_result_codes db2 1 2824397de57Sdanielk1977 db2 eval {SELECT * FROM sqlite_master} 2834397de57Sdanielk1977 db2 close 2844397de57Sdanielk1977} 2854397de57Sdanielk1977 2864397de57Sdanielk1977# This block tests malloc() failures that occur within calls to 2874397de57Sdanielk1977# sqlite3_create_function(). 2884397de57Sdanielk1977do_malloc_test 11 -tclbody { 289f3a65f7eSdrh set rc [sqlite3_create_function db] 290fa18beceSdanielk1977 if {[string match $rc SQLITE_OK]} { 291fa18beceSdanielk1977 set rc [sqlite3_create_aggregate db] 292fa18beceSdanielk1977 } 2932c336549Sdanielk1977 if {[string match $rc SQLITE_NOMEM]} { 2944397de57Sdanielk1977 error "out of memory" 2954397de57Sdanielk1977 } 2964397de57Sdanielk1977} 2974397de57Sdanielk1977 2984397de57Sdanielk1977do_malloc_test 12 -tclbody { 2994397de57Sdanielk1977 set sql16 [encoding convertto unicode "SELECT * FROM sqlite_master"] 3004397de57Sdanielk1977 append sql16 "\00\00" 301f3a65f7eSdrh set ::STMT [sqlite3_prepare16 db $sql16 -1 DUMMY] 3024397de57Sdanielk1977 sqlite3_finalize $::STMT 3034397de57Sdanielk1977} 3048b60e0f1Sdanielk1977 305aca790acSdanielk1977# Test malloc errors when replaying two hot journals from a 2-file 30666560adaSdrh# transaction. 3075a8f9374Sdanielk1977ifcapable crashtest&&attach { 308aca790acSdanielk1977 do_malloc_test 13 -tclprep { 30959a33f98Sdanielk1977 set rc [crashsql -delay 1 -file test2.db { 310aca790acSdanielk1977 ATTACH 'test2.db' as aux; 311aca790acSdanielk1977 PRAGMA cache_size = 10; 312aca790acSdanielk1977 BEGIN; 313aca790acSdanielk1977 CREATE TABLE aux.t2(a, b, c); 314aca790acSdanielk1977 CREATE TABLE t1(a, b, c); 315aca790acSdanielk1977 COMMIT; 316aca790acSdanielk1977 }] 317aca790acSdanielk1977 if {$rc!="1 {child process exited abnormally}"} { 318aca790acSdanielk1977 error "Wrong error message: $rc" 319aca790acSdanielk1977 } 320950f054cSdanielk1977 } -tclbody { 321950f054cSdanielk1977 db eval {ATTACH 'test2.db' as aux;} 322950f054cSdanielk1977 set rc [catch {db eval { 323aca790acSdanielk1977 SELECT * FROM t1; 324aca790acSdanielk1977 SELECT * FROM t2; 325950f054cSdanielk1977 }} err] 326950f054cSdanielk1977 if {$rc && $err!="no such table: t1"} { 327950f054cSdanielk1977 error $err 328950f054cSdanielk1977 } 329aca790acSdanielk1977 } 330aca790acSdanielk1977} 331aca790acSdanielk1977 332*0aac5613Sdanif {$tcl_platform(platform)!="windows" && [atomic_batch_write test.db]==0} { 333aca790acSdanielk1977 do_malloc_test 14 -tclprep { 334aca790acSdanielk1977 catch {db close} 335aca790acSdanielk1977 sqlite3 db2 test2.db 336ae72d982Sdanielk1977 sqlite3_extended_result_codes db2 1 337aca790acSdanielk1977 db2 eval { 338f43d7fceSdan PRAGMA journal_mode = DELETE; /* For inmemory_journal permutation */ 339aca790acSdanielk1977 PRAGMA synchronous = 0; 340aca790acSdanielk1977 CREATE TABLE t1(a, b); 341aca790acSdanielk1977 INSERT INTO t1 VALUES(1, 2); 342aca790acSdanielk1977 BEGIN; 343aca790acSdanielk1977 INSERT INTO t1 VALUES(3, 4); 344aca790acSdanielk1977 } 345fda06befSmistachkin forcecopy test2.db test.db 346fda06befSmistachkin forcecopy test2.db-journal test.db-journal 347aca790acSdanielk1977 db2 close 348aca790acSdanielk1977 } -tclbody { 349aca790acSdanielk1977 sqlite3 db test.db 350ae72d982Sdanielk1977 sqlite3_extended_result_codes db 1 35198c21903Sdanielk1977 35298c21903Sdanielk1977 # If an out-of-memory occurs within a call to a VFS layer function during 35398c21903Sdanielk1977 # hot-journal rollback, sqlite will report SQLITE_CORRUPT. See commit 35498c21903Sdanielk1977 # [5668] for details. 35598c21903Sdanielk1977 set rc [catch {db eval { SELECT * FROM t1 }} msg] 35698c21903Sdanielk1977 if {$msg eq "database disk image is malformed"} { set msg "out of memory" } 35798c21903Sdanielk1977 if {$rc} { error $msg } 35898c21903Sdanielk1977 set msg 359aca790acSdanielk1977 } 36076b047d9Sdanielk1977} 361aca790acSdanielk1977 3629a30cf65Sdanielk1977proc string_compare {a b} { 3639a30cf65Sdanielk1977 return [string compare $a $b] 3649a30cf65Sdanielk1977} 3659a30cf65Sdanielk1977 3669a30cf65Sdanielk1977# Test for malloc() failures in sqlite3_create_collation() and 3679a30cf65Sdanielk1977# sqlite3_create_collation16(). 368950f054cSdanielk1977# 3694152e677Sdanielk1977ifcapable utf16 { 370f3a65f7eSdrh do_malloc_test 15 -start 4 -tclbody { 3719a30cf65Sdanielk1977 db collate string_compare string_compare 372f3a65f7eSdrh if {[catch {add_test_collate db 1 1 1} msg]} { 3739a30cf65Sdanielk1977 if {$msg=="SQLITE_NOMEM"} {set msg "out of memory"} 3749a30cf65Sdanielk1977 error $msg 3759a30cf65Sdanielk1977 } 376950f054cSdanielk1977 377950f054cSdanielk1977 db complete {SELECT "hello """||'world"' [microsoft], * FROM anicetable;} 378950f054cSdanielk1977 db complete {-- Useful comment} 379950f054cSdanielk1977 3809a30cf65Sdanielk1977 execsql { 3819a30cf65Sdanielk1977 CREATE TABLE t1(a, b COLLATE string_compare); 3829a30cf65Sdanielk1977 INSERT INTO t1 VALUES(10, 'string'); 3839a30cf65Sdanielk1977 INSERT INTO t1 VALUES(10, 'string2'); 3849a30cf65Sdanielk1977 } 3859a30cf65Sdanielk1977 } 3864152e677Sdanielk1977} 3879a30cf65Sdanielk1977 388950f054cSdanielk1977# Also test sqlite3_complete(). There are (currently) no malloc() 389950f054cSdanielk1977# calls in this function, but test anyway against future changes. 390950f054cSdanielk1977# 391950f054cSdanielk1977do_malloc_test 16 -tclbody { 392950f054cSdanielk1977 db complete {SELECT "hello """||'world"' [microsoft], * FROM anicetable;} 393950f054cSdanielk1977 db complete {-- Useful comment} 394950f054cSdanielk1977 db eval { 395950f054cSdanielk1977 SELECT * FROM sqlite_master; 396950f054cSdanielk1977 } 397950f054cSdanielk1977} 398950f054cSdanielk1977 399950f054cSdanielk1977# Test handling of malloc() failures in sqlite3_open16(). 400950f054cSdanielk1977# 4014152e677Sdanielk1977ifcapable utf16 { 402950f054cSdanielk1977 do_malloc_test 17 -tclbody { 403950f054cSdanielk1977 set DB2 0 404950f054cSdanielk1977 set STMT 0 405950f054cSdanielk1977 406950f054cSdanielk1977 # open database using sqlite3_open16() 407d9910fe5Sdrh set filename [encoding convertto unicode test.db] 408d9910fe5Sdrh append filename "\x00\x00" 409d9910fe5Sdrh set DB2 [sqlite3_open16 $filename -unused] 410950f054cSdanielk1977 if {0==$DB2} { 411950f054cSdanielk1977 error "out of memory" 412950f054cSdanielk1977 } 413ae72d982Sdanielk1977 sqlite3_extended_result_codes $DB2 1 414950f054cSdanielk1977 415950f054cSdanielk1977 # Prepare statement 416950f054cSdanielk1977 set rc [catch {sqlite3_prepare $DB2 {SELECT * FROM sqlite_master} -1 X} msg] 417ae72d982Sdanielk1977 if {[sqlite3_errcode $DB2] eq "SQLITE_IOERR+12"} { 418ae72d982Sdanielk1977 error "out of memory" 419ae72d982Sdanielk1977 } 42059633aeeSdanielk1977 if {[regexp ".*automatic extension loading.*" [sqlite3_errmsg $DB2]]} { 42159633aeeSdanielk1977 error "out of memory" 42259633aeeSdanielk1977 } 423950f054cSdanielk1977 if {$rc} { 424950f054cSdanielk1977 error [string range $msg 4 end] 425950f054cSdanielk1977 } 426950f054cSdanielk1977 set STMT $msg 427950f054cSdanielk1977 428950f054cSdanielk1977 # Finalize statement 429950f054cSdanielk1977 set rc [sqlite3_finalize $STMT] 430950f054cSdanielk1977 if {$rc!="SQLITE_OK"} { 431950f054cSdanielk1977 error [sqlite3_errmsg $DB2] 432950f054cSdanielk1977 } 433950f054cSdanielk1977 set STMT 0 434950f054cSdanielk1977 435950f054cSdanielk1977 # Close database 436950f054cSdanielk1977 set rc [sqlite3_close $DB2] 437950f054cSdanielk1977 if {$rc!="SQLITE_OK"} { 438950f054cSdanielk1977 error [sqlite3_errmsg $DB2] 439950f054cSdanielk1977 } 440950f054cSdanielk1977 set DB2 0 441950f054cSdanielk1977 } -cleanup { 442950f054cSdanielk1977 if {$STMT!="0"} { 443950f054cSdanielk1977 sqlite3_finalize $STMT 444950f054cSdanielk1977 } 445950f054cSdanielk1977 if {$DB2!="0"} { 446950f054cSdanielk1977 set rc [sqlite3_close $DB2] 447950f054cSdanielk1977 } 448950f054cSdanielk1977 } 4494152e677Sdanielk1977} 450950f054cSdanielk1977 451950f054cSdanielk1977# Test handling of malloc() failures in sqlite3_errmsg16(). 452950f054cSdanielk1977# 4534152e677Sdanielk1977ifcapable utf16 { 454ae72d982Sdanielk1977 do_malloc_test 18 -tclprep { 455950f054cSdanielk1977 catch { 456950f054cSdanielk1977 db eval "SELECT [string repeat longcolumnname 10] FROM sqlite_master" 457ae72d982Sdanielk1977 } 458ae72d982Sdanielk1977 } -tclbody { 459950f054cSdanielk1977 set utf16 [sqlite3_errmsg16 [sqlite3_connection_pointer db]] 460950f054cSdanielk1977 binary scan $utf16 c* bytes 461950f054cSdanielk1977 if {[llength $bytes]==0} { 462950f054cSdanielk1977 error "out of memory" 463950f054cSdanielk1977 } 464950f054cSdanielk1977 } 4654152e677Sdanielk1977} 466950f054cSdanielk1977 467161fb796Sdanielk1977# This test is aimed at coverage testing. Specificly, it is supposed to 468161fb796Sdanielk1977# cause a malloc() only used when converting between the two utf-16 469161fb796Sdanielk1977# encodings to fail (i.e. little-endian->big-endian). It only actually 470161fb796Sdanielk1977# hits this malloc() on little-endian hosts. 471161fb796Sdanielk1977# 472161fb796Sdanielk1977set static_string "\x00h\x00e\x00l\x00l\x00o" 473161fb796Sdanielk1977for {set l 0} {$l<10} {incr l} { 474161fb796Sdanielk1977 append static_string $static_string 475161fb796Sdanielk1977} 476161fb796Sdanielk1977append static_string "\x00\x00" 477161fb796Sdanielk1977do_malloc_test 19 -tclprep { 478161fb796Sdanielk1977 execsql { 479161fb796Sdanielk1977 PRAGMA encoding = "UTF16be"; 480161fb796Sdanielk1977 CREATE TABLE abc(a, b, c); 481161fb796Sdanielk1977 } 482161fb796Sdanielk1977} -tclbody { 483161fb796Sdanielk1977 unset -nocomplain ::STMT 484161fb796Sdanielk1977 set r [catch { 485f3a65f7eSdrh set ::STMT [sqlite3_prepare db {SELECT ?} -1 DUMMY] 486161fb796Sdanielk1977 sqlite3_bind_text16 -static $::STMT 1 $static_string 112 487161fb796Sdanielk1977 } msg] 488161fb796Sdanielk1977 if {$r} {error [string range $msg 4 end]} 489161fb796Sdanielk1977 set msg 490161fb796Sdanielk1977} -cleanup { 491161fb796Sdanielk1977 if {[info exists ::STMT]} { 492161fb796Sdanielk1977 sqlite3_finalize $::STMT 493161fb796Sdanielk1977 } 494161fb796Sdanielk1977} 495161fb796Sdanielk1977unset static_string 496161fb796Sdanielk1977 4976103fe97Sdrh# Make sure SQLITE_NOMEM is reported out on an ATTACH failure even 4986103fe97Sdrh# when the malloc failure occurs within the nested parse. 4996103fe97Sdrh# 5005a8f9374Sdanielk1977ifcapable attach { 5016103fe97Sdrh do_malloc_test 20 -tclprep { 5026103fe97Sdrh db close 503fda06befSmistachkin forcedelete test2.db test2.db-journal 5046103fe97Sdrh sqlite3 db test2.db 505ae72d982Sdanielk1977 sqlite3_extended_result_codes db 1 5066103fe97Sdrh db eval {CREATE TABLE t1(x);} 5076103fe97Sdrh db close 5086103fe97Sdrh } -tclbody { 5096103fe97Sdrh if {[catch {sqlite3 db test.db}]} { 5106103fe97Sdrh error "out of memory" 5116103fe97Sdrh } 512ae72d982Sdanielk1977 sqlite3_extended_result_codes db 1 5136103fe97Sdrh } -sqlbody { 5146103fe97Sdrh ATTACH DATABASE 'test2.db' AS t2; 5156103fe97Sdrh SELECT * FROM t1; 5166103fe97Sdrh DETACH DATABASE t2; 5176103fe97Sdrh } 5185a8f9374Sdanielk1977} 5196103fe97Sdrh 520b5584c0cSdanielk1977# Test malloc failure whilst installing a foreign key. 52169b637b5Sdanielk1977# 5224152e677Sdanielk1977ifcapable foreignkey { 52369b637b5Sdanielk1977 do_malloc_test 21 -sqlbody { 52469b637b5Sdanielk1977 CREATE TABLE abc(a, b, c, FOREIGN KEY(a) REFERENCES abc(b)) 52569b637b5Sdanielk1977 } 5264152e677Sdanielk1977} 52769b637b5Sdanielk1977 5287e29e956Sdanielk1977# Test malloc failure in an sqlite3_prepare_v2() call. 5297e29e956Sdanielk1977# 5307e29e956Sdanielk1977do_malloc_test 22 -tclbody { 5317e29e956Sdanielk1977 set ::STMT "" 5327e29e956Sdanielk1977 set r [catch { 5337e29e956Sdanielk1977 set ::STMT [ 534f3a65f7eSdrh sqlite3_prepare_v2 db "SELECT * FROM sqlite_master" -1 DUMMY 5357e29e956Sdanielk1977 ] 5367e29e956Sdanielk1977 } msg] 5377e29e956Sdanielk1977 if {$r} {error [string range $msg 4 end]} 5387e29e956Sdanielk1977} -cleanup { 5397e29e956Sdanielk1977 if {$::STMT ne ""} { 5407e29e956Sdanielk1977 sqlite3_finalize $::STMT 5417e29e956Sdanielk1977 set ::STMT "" 5427e29e956Sdanielk1977 } 5437e29e956Sdanielk1977} 5447e29e956Sdanielk1977 5452a180ff9Sdanielk1977ifcapable {pager_pragmas} { 5462a180ff9Sdanielk1977 # This tests a special case - that an error that occurs while the pager 5472a180ff9Sdanielk1977 # is trying to recover from error-state in exclusive-access mode works. 5482a180ff9Sdanielk1977 # 5492a180ff9Sdanielk1977 do_malloc_test 23 -tclprep { 5502a180ff9Sdanielk1977 db eval { 5512a180ff9Sdanielk1977 PRAGMA cache_size = 10; 5522a180ff9Sdanielk1977 PRAGMA locking_mode = exclusive; 5532a180ff9Sdanielk1977 BEGIN; 5542a180ff9Sdanielk1977 CREATE TABLE abc(a, b, c); 5552a180ff9Sdanielk1977 CREATE INDEX abc_i ON abc(a, b, c); 5562a180ff9Sdanielk1977 INSERT INTO abc 5572a180ff9Sdanielk1977 VALUES(randstr(100,100), randstr(100,100), randstr(100,100)); 5582a180ff9Sdanielk1977 INSERT INTO abc 5592a180ff9Sdanielk1977 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; 5602a180ff9Sdanielk1977 INSERT INTO abc 5612a180ff9Sdanielk1977 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; 5622a180ff9Sdanielk1977 INSERT INTO abc 5632a180ff9Sdanielk1977 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; 5642a180ff9Sdanielk1977 INSERT INTO abc 5652a180ff9Sdanielk1977 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; 5662a180ff9Sdanielk1977 INSERT INTO abc 5672a180ff9Sdanielk1977 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; 5682a180ff9Sdanielk1977 COMMIT; 5692a180ff9Sdanielk1977 } 5702a180ff9Sdanielk1977 5712a180ff9Sdanielk1977 # This puts the pager into error state. 5722a180ff9Sdanielk1977 # 5732a180ff9Sdanielk1977 db eval BEGIN 5742a180ff9Sdanielk1977 db eval {UPDATE abc SET a = 0 WHERE oid%2} 5752a180ff9Sdanielk1977 set ::sqlite_io_error_pending 10 5762a180ff9Sdanielk1977 catch {db eval {ROLLBACK}} msg 5772a180ff9Sdanielk1977 57898c21903Sdanielk1977 } -tclbody { 57998c21903Sdanielk1977 # If an out-of-memory occurs within a call to a VFS layer function during 58098c21903Sdanielk1977 # hot-journal rollback, sqlite will report SQLITE_CORRUPT. See commit 58198c21903Sdanielk1977 # [5668] for details. 58298c21903Sdanielk1977 set rc [catch {db eval { SELECT * FROM abc LIMIT 10 }} msg] 58398c21903Sdanielk1977 if {$msg eq "database disk image is malformed"} { set msg "out of memory" } 58498c21903Sdanielk1977 if {$rc} { error $msg } 58598c21903Sdanielk1977 set msg 5862a180ff9Sdanielk1977 } -cleanup { 5872a180ff9Sdanielk1977 set e [db eval {PRAGMA integrity_check}] 5882a180ff9Sdanielk1977 if {$e ne "ok"} {error $e} 5892a180ff9Sdanielk1977 } 5902a180ff9Sdanielk1977} 5912a180ff9Sdanielk1977 592ac559264Sdanielk1977ifcapable compound { 593cdf3020cSdanielk1977 do_malloc_test 24 -sqlprep { 594ac559264Sdanielk1977 CREATE TABLE t1(a, b, c) 595ac559264Sdanielk1977 } -sqlbody { 596ac559264Sdanielk1977 SELECT 1 FROM t1 UNION SELECT 2 FROM t1 ORDER BY 1 597ac559264Sdanielk1977 } 598ac559264Sdanielk1977} 599ac559264Sdanielk1977 600cdf3020cSdanielk1977ifcapable view&&trigger { 601cdf3020cSdanielk1977 do_malloc_test 25 -sqlprep { 602cdf3020cSdanielk1977 CREATE TABLE t1(a, b, c); 603cdf3020cSdanielk1977 CREATE VIEW v1 AS SELECT * FROM t1; 604cdf3020cSdanielk1977 CREATE TRIGGER v1t1 INSTEAD OF DELETE ON v1 BEGIN SELECT 1; END; 605cdf3020cSdanielk1977 CREATE TRIGGER v1t2 INSTEAD OF INSERT ON v1 BEGIN SELECT 1; END; 606cdf3020cSdanielk1977 CREATE TRIGGER v1t3 INSTEAD OF UPDATE ON v1 BEGIN SELECT 1; END; 607cdf3020cSdanielk1977 } -sqlbody { 608cdf3020cSdanielk1977 DELETE FROM v1 WHERE a = 1; 609cdf3020cSdanielk1977 INSERT INTO v1 VALUES(1, 2, 3); 610cdf3020cSdanielk1977 UPDATE v1 SET a = 1 WHERE b = 2; 611cdf3020cSdanielk1977 } 612cdf3020cSdanielk1977} 613cdf3020cSdanielk1977 6147eaabcdbSdanielk1977do_malloc_test 25 -sqlprep { 6157eaabcdbSdanielk1977 CREATE TABLE abc(a, b, c); 6167eaabcdbSdanielk1977 CREATE INDEX i1 ON abc(a, b); 6177eaabcdbSdanielk1977 INSERT INTO abc VALUES(1, 2, 3); 6187eaabcdbSdanielk1977 INSERT INTO abc VALUES(4, 5, 6); 6197eaabcdbSdanielk1977} -tclbody { 6207eaabcdbSdanielk1977 # For each UPDATE executed, the cursor used for the SELECT statement 6217eaabcdbSdanielk1977 # must be "saved". Because the cursor is open on an index, this requires 6227eaabcdbSdanielk1977 # a malloc() to allocate space to save the index key. This test case is 6237eaabcdbSdanielk1977 # aimed at testing the response of the library to a failure in that 6247eaabcdbSdanielk1977 # particular malloc() call. 6257eaabcdbSdanielk1977 db eval {SELECT a FROM abc ORDER BY a} { 6267eaabcdbSdanielk1977 db eval {UPDATE abc SET b = b - 1 WHERE a = $a} 6277eaabcdbSdanielk1977 } 6287eaabcdbSdanielk1977} 6297eaabcdbSdanielk1977 630518002e4Sdanielk1977# This test is designed to test a specific juncture in the sqlite code. 631518002e4Sdanielk1977# The database set up by -sqlprep script contains a single table B-Tree 632518002e4Sdanielk1977# of height 2. In the -tclbody script, the existing database connection 633518002e4Sdanielk1977# is closed and a new one opened and used to insert a new row into the 634518002e4Sdanielk1977# table B-Tree. By using a new connection, the outcome of a malloc() 635518002e4Sdanielk1977# failure while seeking to the right-hand side of the B-Tree to insert 636518002e4Sdanielk1977# a new record can be tested. 637518002e4Sdanielk1977# 638518002e4Sdanielk1977do_malloc_test 26 -sqlprep { 639518002e4Sdanielk1977 BEGIN; 640518002e4Sdanielk1977 CREATE TABLE t1(a, b); 641518002e4Sdanielk1977 INSERT INTO t1 VALUES(1, randomblob(210)); 642518002e4Sdanielk1977 INSERT INTO t1 VALUES(1, randomblob(210)); 643518002e4Sdanielk1977 INSERT INTO t1 VALUES(1, randomblob(210)); 644518002e4Sdanielk1977 INSERT INTO t1 VALUES(1, randomblob(210)); 645518002e4Sdanielk1977 INSERT INTO t1 VALUES(1, randomblob(210)); 646518002e4Sdanielk1977 COMMIT; 647518002e4Sdanielk1977} -tclbody { 648518002e4Sdanielk1977 db close 649518002e4Sdanielk1977 sqlite3 db test.db 650518002e4Sdanielk1977 db eval { INSERT INTO t1 VALUES(1, randomblob(210)) } 651518002e4Sdanielk1977} 652518002e4Sdanielk1977 65377eb5bb6Sdanielk1977# Test that no memory is leaked following a malloc() failure in 65477eb5bb6Sdanielk1977# sqlite3_initialize(). 65577eb5bb6Sdanielk1977# 65677eb5bb6Sdanielk1977do_malloc_test 27 -tclprep { 65777eb5bb6Sdanielk1977 db close 65877eb5bb6Sdanielk1977 sqlite3_shutdown 65977eb5bb6Sdanielk1977} -tclbody { 66077eb5bb6Sdanielk1977 set rc [sqlite3_initialize] 66177eb5bb6Sdanielk1977 if {$rc == "SQLITE_NOMEM"} { 66277eb5bb6Sdanielk1977 error "out of memory" 66377eb5bb6Sdanielk1977 } 66477eb5bb6Sdanielk1977} 665bb77b753Sdrhautoinstall_test_functions 66677eb5bb6Sdanielk1977 66785574e31Sdanielk1977# Test that malloc failures that occur while processing INDEXED BY 66885574e31Sdanielk1977# clauses are handled correctly. 66985574e31Sdanielk1977do_malloc_test 28 -sqlprep { 67085574e31Sdanielk1977 CREATE TABLE t1(a, b); 67185574e31Sdanielk1977 CREATE INDEX i1 ON t1(a); 67285574e31Sdanielk1977 CREATE VIEW v1 AS SELECT * FROM t1 INDEXED BY i1 WHERE a = 10; 67385574e31Sdanielk1977} -sqlbody { 67485574e31Sdanielk1977 SELECT * FROM t1 INDEXED BY i1 ORDER BY a; 67585574e31Sdanielk1977 SELECT * FROM v1; 67685574e31Sdanielk1977} 67785574e31Sdanielk1977 678e4359750Sdanielk1977do_malloc_test 29 -sqlprep { 679e4359750Sdanielk1977 CREATE TABLE t1(a TEXT, b TEXT); 680e4359750Sdanielk1977} -sqlbody { 681e4359750Sdanielk1977 INSERT INTO t1 VALUES(1, -234); 682956f4319Sdanielk1977 INSERT INTO t1 SELECT * FROM t1 UNION ALL SELECT * FROM t1; 683e4359750Sdanielk1977} 684e4359750Sdanielk1977 6857cbd589dSdanielk1977do_malloc_test 30 -tclprep { 68695855288Sdanielk1977 db eval { 68795855288Sdanielk1977 CREATE TABLE t1(x PRIMARY KEY); 68895855288Sdanielk1977 INSERT INTO t1 VALUES(randstr(500,500)); 68995855288Sdanielk1977 INSERT INTO t1 VALUES(randstr(500,500)); 69095855288Sdanielk1977 INSERT INTO t1 VALUES(randstr(500,500)); 69195855288Sdanielk1977 } 69295855288Sdanielk1977 db close 69395855288Sdanielk1977 sqlite3 db test.db 69495855288Sdanielk1977 69595855288Sdanielk1977 # The DELETE command in the following block moves the overflow pages that 69695855288Sdanielk1977 # are part of the primary key index to the free-list. But it does not 69795855288Sdanielk1977 # actually load the content of the pages. This leads to the peculiar 69895855288Sdanielk1977 # situation where cache entries exist, but are not populated with data. 69995855288Sdanielk1977 # They are populated next time they are requested by the b-tree layer. 70095855288Sdanielk1977 # 70195855288Sdanielk1977 db eval { 70295855288Sdanielk1977 BEGIN; 70395855288Sdanielk1977 DELETE FROM t1; 70495855288Sdanielk1977 ROLLBACK; 70595855288Sdanielk1977 } 70695855288Sdanielk1977} -sqlbody { 707c87239fbSdanielk1977 -- This statement requires the 'no-content' pages loaded by the DELETE 708c87239fbSdanielk1977 -- statement above. When requesting the pages, the content is loaded 709c87239fbSdanielk1977 -- from the database file. The point of this test case is to test handling 710c87239fbSdanielk1977 -- of malloc errors (including SQLITE_IOERR_NOMEM errors) when loading 711c87239fbSdanielk1977 -- the content. 71295855288Sdanielk1977 SELECT * FROM t1 ORDER BY x; 71395855288Sdanielk1977} 71495855288Sdanielk1977 7157cbd589dSdanielk1977# After committing a transaction in persistent-journal mode, if a journal 7167cbd589dSdanielk1977# size limit is configured SQLite may attempt to truncate the journal file. 7177cbd589dSdanielk1977# This test verifies the libraries response to a malloc() failure during 7187cbd589dSdanielk1977# this operation. 7197cbd589dSdanielk1977# 7207cbd589dSdanielk1977do_malloc_test 31 -sqlprep { 7217cbd589dSdanielk1977 PRAGMA journal_mode = persist; 7227cbd589dSdanielk1977 PRAGMA journal_size_limit = 1024; 7237cbd589dSdanielk1977 CREATE TABLE t1(a PRIMARY KEY, b); 7247cbd589dSdanielk1977} -sqlbody { 7257cbd589dSdanielk1977 INSERT INTO t1 VALUES(1, 2); 7267cbd589dSdanielk1977} 727956f4319Sdanielk1977 728db340397Sdanielk1977# When written, this test provoked an obscure change-counter bug. 729db340397Sdanielk1977# 730db340397Sdanielk1977# If, when running in exclusive mode, a malloc() failure occurs 731db340397Sdanielk1977# after the database file change-counter has been written but 732db340397Sdanielk1977# before the transaction has been committed, then the transaction 733db340397Sdanielk1977# is automatically rolled back. However, internally the 734db340397Sdanielk1977# Pager.changeCounterDone flag was being left set. This means 735db340397Sdanielk1977# that if the same connection attempts another transaction following 736db340397Sdanielk1977# the malloc failure and rollback, the change counter will not 737db340397Sdanielk1977# be updated. This could corrupt another processes cache. 738db340397Sdanielk1977# 739db340397Sdanielk1977do_malloc_test 32 -tclprep { 740db340397Sdanielk1977 # Build a small database containing an indexed table. 741db340397Sdanielk1977 # 742db340397Sdanielk1977 db eval { 743de06e9f2Sdanielk1977 PRAGMA locking_mode = normal; 744db340397Sdanielk1977 BEGIN; 745db340397Sdanielk1977 CREATE TABLE t1(a PRIMARY KEY, b); 746db340397Sdanielk1977 INSERT INTO t1 VALUES(1, 'one'); 747db340397Sdanielk1977 INSERT INTO t1 VALUES(2, 'two'); 748db340397Sdanielk1977 INSERT INTO t1 VALUES(3, 'three'); 749db340397Sdanielk1977 COMMIT; 750db340397Sdanielk1977 PRAGMA locking_mode = exclusive; 751db340397Sdanielk1977 } 752db340397Sdanielk1977 753db340397Sdanielk1977 # Open a second database connection. Load the table (but not index) 754db340397Sdanielk1977 # into the second connections pager cache. 755db340397Sdanielk1977 # 756db340397Sdanielk1977 sqlite3 db2 test.db 757de06e9f2Sdanielk1977 db2 eval { 758de06e9f2Sdanielk1977 PRAGMA locking_mode = normal; 759de06e9f2Sdanielk1977 SELECT b FROM t1; 760de06e9f2Sdanielk1977 } 761db340397Sdanielk1977 762db340397Sdanielk1977} -tclbody { 763db340397Sdanielk1977 # Running in exclusive mode, perform a database transaction that 764db340397Sdanielk1977 # modifies both the database table and index. For iterations where 765db340397Sdanielk1977 # the malloc failure occurs after updating the change counter but 766db340397Sdanielk1977 # before committing the transaction, this should result in the 767db340397Sdanielk1977 # transaction being rolled back but the changeCounterDone flag 768db340397Sdanielk1977 # left set. 769db340397Sdanielk1977 # 770db340397Sdanielk1977 db eval { UPDATE t1 SET a = a + 3 } 771db340397Sdanielk1977} -cleanup { 772db340397Sdanielk1977 773db340397Sdanielk1977 # Perform another transaction using the first connection. Unlock 774db340397Sdanielk1977 # the database after doing so. If this is one of the right iterations, 775db340397Sdanielk1977 # then this should result in the database contents being updated but 776db340397Sdanielk1977 # the change-counter left as it is. 777db340397Sdanielk1977 # 778db340397Sdanielk1977 db eval { 779db340397Sdanielk1977 PRAGMA locking_mode = normal; 780db340397Sdanielk1977 UPDATE t1 SET a = a + 3; 781db340397Sdanielk1977 } 782db340397Sdanielk1977 783db340397Sdanielk1977 # Now do an integrity check with the second connection. The second 784db340397Sdanielk1977 # connection still has the database table in its cache. If this is 785db340397Sdanielk1977 # one of the magic iterations and the change counter was not modified, 786db340397Sdanielk1977 # then it won't realize that the cached data is out of date. Since 787db340397Sdanielk1977 # the cached data won't match the up to date index data read from 788db340397Sdanielk1977 # the database file, the integrity check should fail. 789db340397Sdanielk1977 # 790db340397Sdanielk1977 set zRepeat "transient" 791db340397Sdanielk1977 if {$::iRepeat} {set zRepeat "persistent"} 792db340397Sdanielk1977 do_test malloc-32.$zRepeat.${::n}.integrity { 793db340397Sdanielk1977 execsql {PRAGMA integrity_check} db2 794db340397Sdanielk1977 } {ok} 795db340397Sdanielk1977 db2 close 796db340397Sdanielk1977} 797db340397Sdanielk1977 79802f33725Sdanielk1977# The following two OOM tests verify that OOM handling works in the 79902f33725Sdanielk1977# code used to optimize "SELECT count(*) FROM <tbl>". 80002f33725Sdanielk1977# 80102f33725Sdanielk1977do_malloc_test 33 -tclprep { 80202f33725Sdanielk1977 db eval { PRAGMA cache_size = 10 } 80302f33725Sdanielk1977 db transaction { 80402f33725Sdanielk1977 db eval { CREATE TABLE abc(a, b) } 80502f33725Sdanielk1977 for {set i 0} {$i<500} {incr i} { 80602f33725Sdanielk1977 db eval {INSERT INTO abc VALUES(randstr(100,100), randstr(1000,1000))} 80702f33725Sdanielk1977 } 80802f33725Sdanielk1977 } 80902f33725Sdanielk1977} -sqlbody { 81002f33725Sdanielk1977 SELECT count(*) FROM abc; 81102f33725Sdanielk1977} 81202f33725Sdanielk1977do_malloc_test 34 -tclprep { 81302f33725Sdanielk1977 db eval { PRAGMA cache_size = 10 } 81402f33725Sdanielk1977 db transaction { 81502f33725Sdanielk1977 db eval { CREATE TABLE abc(a PRIMARY KEY, b) } 81602f33725Sdanielk1977 for {set i 0} {$i<500} {incr i} { 81702f33725Sdanielk1977 db eval {INSERT INTO abc VALUES(randstr(100,100), randstr(1000,1000))} 81802f33725Sdanielk1977 } 81902f33725Sdanielk1977 } 82002f33725Sdanielk1977} -sqlbody { 82102f33725Sdanielk1977 SELECT count(*) FROM abc; 82202f33725Sdanielk1977} 82302f33725Sdanielk1977 824238746a6Sdanielk1977proc f {args} { error "Quite a long error!" } 825238746a6Sdanielk1977do_malloc_test 35 -tclprep { 826238746a6Sdanielk1977 db func f f 827238746a6Sdanielk1977 set ::STMT [sqlite3_prepare db "SELECT f()" -1 DUMMY] 828238746a6Sdanielk1977 sqlite3_step $::STMT 829238746a6Sdanielk1977} -tclbody { 830238746a6Sdanielk1977 sqlite3_finalize $::STMT 831238746a6Sdanielk1977} -cleanup { 832238746a6Sdanielk1977 # At one point an assert( !db->mallocFailed ) could fail in the following 833238746a6Sdanielk1977 # call to sqlite3_errmsg(). Because sqlite3_finalize() had failed to clear 834238746a6Sdanielk1977 # the flag before returning. 835238746a6Sdanielk1977 sqlite3_errmsg16 db 836238746a6Sdanielk1977} 837238746a6Sdanielk1977 838238746a6Sdanielk1977do_malloc_test 36 -sqlprep { 839238746a6Sdanielk1977 CREATE TABLE t1(a, b); 840238746a6Sdanielk1977 INSERT INTO t1 VALUES(1, 2); 841238746a6Sdanielk1977 INSERT INTO t1 VALUES(3, 4); 842238746a6Sdanielk1977} -sqlbody { 843238746a6Sdanielk1977 SELECT test_agg_errmsg16(), group_concat(a) FROM t1 844238746a6Sdanielk1977} 845238746a6Sdanielk1977 84648864df9Smistachkin# At one point, if an OOM occurred immediately after obtaining a shared lock 8473582c8f7Sdanielk1977# on the database file, the file remained locked. This test case ensures 848c3927537Sdrh# that bug has been fixed.i 849c3927537Sdrhif {[db eval {PRAGMA locking_mode}]!="exclusive"} { 850c3927537Sdrh do_malloc_test 37 -tclprep { 8513582c8f7Sdanielk1977 sqlite3 db2 test.db 8523582c8f7Sdanielk1977 execsql { 8533582c8f7Sdanielk1977 CREATE TABLE t1(a, b); 8543582c8f7Sdanielk1977 INSERT INTO t1 VALUES(1, 2); 8553582c8f7Sdanielk1977 } db2 8563582c8f7Sdanielk1977 } -sqlbody { 8573582c8f7Sdanielk1977 SELECT * FROM t1; 8583582c8f7Sdanielk1977 } -cleanup { 8593582c8f7Sdanielk1977 # Try to write to the database using connection [db2]. If connection [db] 8603582c8f7Sdanielk1977 # has correctly released the shared lock, this write attempt should 8613582c8f7Sdanielk1977 # succeed. If [db] has not released the lock, this should hit an 8623582c8f7Sdanielk1977 # SQLITE_BUSY error. 8633582c8f7Sdanielk1977 do_test malloc-36.$zRepeat.${::n}.unlocked { 8643582c8f7Sdanielk1977 execsql {INSERT INTO t1 VALUES(3, 4)} db2 8653582c8f7Sdanielk1977 } {} 8663582c8f7Sdanielk1977 db2 close 8673582c8f7Sdanielk1977 } 8683582c8f7Sdanielk1977 catch { db2 close } 869c3927537Sdrh} 8703582c8f7Sdanielk1977 87185c165caSdan 8725f84e14aSdan# Test that if an OOM error occurs, aux-data is still correctly destroyed. 8735f84e14aSdan# This test case was causing either a memory-leak or an assert() failure 8745f84e14aSdan# at one point, depending on the configuration. 8755f84e14aSdan# 8765f84e14aSdando_malloc_test 39 -tclprep { 8775f84e14aSdan sqlite3 db test.db 8785f84e14aSdan} -sqlbody { 8795f84e14aSdan SELECT test_auxdata('abc', 'def'); 8805f84e14aSdan} -cleanup { 8815f84e14aSdan db close 8825f84e14aSdan} 8835f84e14aSdan 88438fdead8Sdanreset_db 88538fdead8Sdanadd_test_utf16bin_collate db 88638fdead8Sdando_execsql_test 40.1 { 88738fdead8Sdan CREATE TABLE t1(a); 88838fdead8Sdan INSERT INTO t1 VALUES('fghij'); 88938fdead8Sdan INSERT INTO t1 VALUES('pqrst'); 89038fdead8Sdan INSERT INTO t1 VALUES('abcde'); 89138fdead8Sdan INSERT INTO t1 VALUES('uvwxy'); 89238fdead8Sdan INSERT INTO t1 VALUES('klmno'); 89338fdead8Sdan} 89438fdead8Sdando_execsql_test 40.2 { 89538fdead8Sdan SELECT * FROM t1 ORDER BY 1 COLLATE utf16bin; 89638fdead8Sdan} {abcde fghij klmno pqrst uvwxy} 89738fdead8Sdando_faultsim_test 40.3 -faults oom-trans* -body { 89838fdead8Sdan execsql { 89938fdead8Sdan SELECT * FROM t1 ORDER BY 1 COLLATE utf16bin; 90038fdead8Sdan } 90138fdead8Sdan} -test { 90238fdead8Sdan faultsim_test_result {0 {abcde fghij klmno pqrst uvwxy}} 90338fdead8Sdan faultsim_integrity_check 90438fdead8Sdan} 90538fdead8Sdan 906a7bf23c6Sdanreset_db 907a7bf23c6Sdanadd_test_utf16bin_collate db 908a7bf23c6Sdanset big [string repeat x 200] 909a7bf23c6Sdando_execsql_test 41.1 { 910a7bf23c6Sdan DROP TABLE IF EXISTS t1; 911a7bf23c6Sdan CREATE TABLE t1(a COLLATE utf16bin); 912a7bf23c6Sdan INSERT INTO t1 VALUES('fghij' || $::big); 913a7bf23c6Sdan INSERT INTO t1 VALUES('pqrst' || $::big); 914a7bf23c6Sdan INSERT INTO t1 VALUES('abcde' || $::big); 915a7bf23c6Sdan INSERT INTO t1 VALUES('uvwxy' || $::big); 916a7bf23c6Sdan INSERT INTO t1 VALUES('klmno' || $::big); 917a7bf23c6Sdan CREATE INDEX i1 ON t1(a); 918a7bf23c6Sdan} 919a7bf23c6Sdando_faultsim_test 41.2 -faults oom* -body { 920a7bf23c6Sdan execsql { SELECT * FROM t1 WHERE a = ('abcde' || $::big)} 921a7bf23c6Sdan} -test { 922a7bf23c6Sdan faultsim_test_result [list 0 "abcde$::big"] 923a7bf23c6Sdan faultsim_integrity_check 924a7bf23c6Sdan} 925a7bf23c6Sdan 926f89aa47fSdanreset_db 927f89aa47fSdando_execsql_test 42.0 { 928f89aa47fSdan CREATE TABLE t1(x INTEGER PRIMARY KEY, y, z); 929f89aa47fSdan CREATE TABLE t2(a, b); 930f89aa47fSdan CREATE VIEW a002 AS SELECT *, sum(b) AS m FROM t2 GROUP BY a; 931f89aa47fSdan} 932f89aa47fSdanfaultsim_save_and_close 933f89aa47fSdando_faultsim_test 42 -faults oom-tran* -prep { 934f89aa47fSdan faultsim_restore_and_reopen 935f89aa47fSdan execsql { SELECT * FROM sqlite_master } 936f89aa47fSdan} -body { 937f89aa47fSdan execsql { 938f89aa47fSdan SELECT t1.z, a002.m 939f89aa47fSdan FROM t1 JOIN a002 ON t1.y=a002.m 940f89aa47fSdan WHERE t1.x IN (1,2,3); 941f89aa47fSdan } 942f89aa47fSdan} -test { 943f89aa47fSdan faultsim_test_result {0 {}} 944f89aa47fSdan} 945f89aa47fSdan 946f89aa47fSdan 94796fb0dd5Sdanielk1977# Ensure that no file descriptors were leaked. 948b5402fbfSdanielk1977do_test malloc-99.X { 94996fb0dd5Sdanielk1977 catch {db close} 95096fb0dd5Sdanielk1977 set sqlite_open_file_count 95196fb0dd5Sdanielk1977} {0} 95296fb0dd5Sdanielk1977 953344a6276Sdrhputs open-file-count=$sqlite_open_file_count 954ed7c855cSdrhfinish_test 955