xref: /sqlite-3.40.0/test/malloc.test (revision 0aac5613)
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