OLD | NEW |
(Empty) | |
| 1 # 2001 September 15 |
| 2 # |
| 3 # The author disclaims copyright to this source code. In place of |
| 4 # a legal notice, here is a blessing: |
| 5 # |
| 6 # May you do good and not evil. |
| 7 # May you find forgiveness for yourself and forgive others. |
| 8 # May you share freely, never taking more than you give. |
| 9 # |
| 10 #*********************************************************************** |
| 11 # |
| 12 # This file attempts to check the behavior of the SQLite library in |
| 13 # an out-of-memory situation. When compiled with -DSQLITE_DEBUG=1, |
| 14 # the SQLite library accepts a special command (sqlite3_memdebug_fail N C) |
| 15 # which causes the N-th malloc to fail. This special feature is used |
| 16 # to see what happens in the library if a malloc were to really fail |
| 17 # due to an out-of-memory situation. |
| 18 # |
| 19 # $Id: malloc.test,v 1.81 2009/06/24 13:13:45 drh Exp $ |
| 20 |
| 21 set testdir [file dirname $argv0] |
| 22 source $testdir/tester.tcl |
| 23 |
| 24 |
| 25 # Only run these tests if memory debugging is turned on. |
| 26 # |
| 27 source $testdir/malloc_common.tcl |
| 28 if {!$MEMDEBUG} { |
| 29 puts "Skipping malloc tests: not compiled with -DSQLITE_MEMDEBUG..." |
| 30 finish_test |
| 31 return |
| 32 } |
| 33 |
| 34 # Do a couple of memory dumps just to exercise the memory dump logic |
| 35 # that that we can say that we have. |
| 36 # |
| 37 puts stderr "This is a test. Ignore the error that follows:" |
| 38 sqlite3_memdebug_dump $testdir |
| 39 puts "Memory dump to file memdump.txt..." |
| 40 sqlite3_memdebug_dump memdump.txt |
| 41 |
| 42 ifcapable bloblit&&subquery { |
| 43 do_malloc_test 1 -tclprep { |
| 44 db close |
| 45 } -tclbody { |
| 46 if {[catch {sqlite3 db test.db}]} { |
| 47 error "out of memory" |
| 48 } |
| 49 sqlite3_extended_result_codes db 1 |
| 50 } -sqlbody { |
| 51 DROP TABLE IF EXISTS t1; |
| 52 CREATE TABLE t1( |
| 53 a int, b float, c double, d text, e varchar(20), |
| 54 primary key(a,b,c) |
| 55 ); |
| 56 CREATE INDEX i1 ON t1(a,b); |
| 57 INSERT INTO t1 VALUES(1,2.3,4.5,'hi',x'746865726500'); |
| 58 INSERT INTO t1 VALUES(6,7.0,0.8,'hello','out yonder'); |
| 59 SELECT * FROM t1; |
| 60 SELECT avg(b) FROM t1 GROUP BY a HAVING b>20.0; |
| 61 DELETE FROM t1 WHERE a IN (SELECT min(a) FROM t1); |
| 62 SELECT count(*), group_concat(e) FROM t1; |
| 63 SELECT b FROM t1 ORDER BY 1 COLLATE nocase; |
| 64 } |
| 65 } |
| 66 |
| 67 # Ensure that no file descriptors were leaked. |
| 68 do_test malloc-1.X { |
| 69 catch {db close} |
| 70 set sqlite_open_file_count |
| 71 } {0} |
| 72 |
| 73 ifcapable subquery { |
| 74 do_malloc_test 2 -sqlbody { |
| 75 CREATE TABLE t1(a int, b int default 'abc', c int default 1); |
| 76 CREATE INDEX i1 ON t1(a,b); |
| 77 INSERT INTO t1 VALUES(1,1,'99 abcdefghijklmnopqrstuvwxyz'); |
| 78 INSERT INTO t1 VALUES(2,4,'98 abcdefghijklmnopqrstuvwxyz'); |
| 79 INSERT INTO t1 VALUES(3,9,'97 abcdefghijklmnopqrstuvwxyz'); |
| 80 INSERT INTO t1 VALUES(4,16,'96 abcdefghijklmnopqrstuvwxyz'); |
| 81 INSERT INTO t1 VALUES(5,25,'95 abcdefghijklmnopqrstuvwxyz'); |
| 82 INSERT INTO t1 VALUES(6,36,'94 abcdefghijklmnopqrstuvwxyz'); |
| 83 SELECT 'stuff', count(*) as 'other stuff', max(a+10) FROM t1; |
| 84 UPDATE t1 SET b=b||b||b||b; |
| 85 UPDATE t1 SET b=a WHERE a in (10,12,22); |
| 86 INSERT INTO t1(c,b,a) VALUES(20,10,5); |
| 87 INSERT INTO t1 SELECT * FROM t1 |
| 88 WHERE a IN (SELECT a FROM t1 WHERE a<10); |
| 89 DELETE FROM t1 WHERE a>=10; |
| 90 DROP INDEX i1; |
| 91 DELETE FROM t1; |
| 92 } |
| 93 } |
| 94 |
| 95 # Ensure that no file descriptors were leaked. |
| 96 do_test malloc-2.X { |
| 97 catch {db close} |
| 98 set sqlite_open_file_count |
| 99 } {0} |
| 100 |
| 101 do_malloc_test 3 -sqlbody { |
| 102 BEGIN TRANSACTION; |
| 103 CREATE TABLE t1(a int, b int, c int); |
| 104 CREATE INDEX i1 ON t1(a,b); |
| 105 INSERT INTO t1 VALUES(1,1,99); |
| 106 INSERT INTO t1 VALUES(2,4,98); |
| 107 INSERT INTO t1 VALUES(3,9,97); |
| 108 INSERT INTO t1 VALUES(4,16,96); |
| 109 INSERT INTO t1 VALUES(5,25,95); |
| 110 INSERT INTO t1 VALUES(6,36,94); |
| 111 INSERT INTO t1(c,b,a) VALUES(20,10,5); |
| 112 DELETE FROM t1 WHERE a>=10; |
| 113 DROP INDEX i1; |
| 114 DELETE FROM t1; |
| 115 ROLLBACK; |
| 116 } |
| 117 |
| 118 |
| 119 # Ensure that no file descriptors were leaked. |
| 120 do_test malloc-3.X { |
| 121 catch {db close} |
| 122 set sqlite_open_file_count |
| 123 } {0} |
| 124 |
| 125 ifcapable subquery { |
| 126 do_malloc_test 4 -sqlbody { |
| 127 BEGIN TRANSACTION; |
| 128 CREATE TABLE t1(a int, b int, c int); |
| 129 CREATE INDEX i1 ON t1(a,b); |
| 130 INSERT INTO t1 VALUES(1,1,99); |
| 131 INSERT INTO t1 VALUES(2,4,98); |
| 132 INSERT INTO t1 VALUES(3,9,97); |
| 133 INSERT INTO t1 VALUES(4,16,96); |
| 134 INSERT INTO t1 VALUES(5,25,95); |
| 135 INSERT INTO t1 VALUES(6,36,94); |
| 136 UPDATE t1 SET b=a WHERE a in (10,12,22); |
| 137 INSERT INTO t1 SELECT * FROM t1 |
| 138 WHERE a IN (SELECT a FROM t1 WHERE a<10); |
| 139 DROP INDEX i1; |
| 140 DELETE FROM t1; |
| 141 COMMIT; |
| 142 } |
| 143 } |
| 144 |
| 145 # Ensure that no file descriptors were leaked. |
| 146 do_test malloc-4.X { |
| 147 catch {db close} |
| 148 set sqlite_open_file_count |
| 149 } {0} |
| 150 |
| 151 ifcapable trigger { |
| 152 do_malloc_test 5 -sqlbody { |
| 153 BEGIN TRANSACTION; |
| 154 CREATE TABLE t1(a,b); |
| 155 CREATE TABLE t2(x,y); |
| 156 CREATE TRIGGER r1 AFTER INSERT ON t1 WHEN new.a = 2 BEGIN |
| 157 INSERT INTO t2(x,y) VALUES(new.rowid,1); |
| 158 INSERT INTO t2(x,y) SELECT * FROM t2; |
| 159 INSERT INTO t2 SELECT * FROM t2; |
| 160 UPDATE t2 SET y=y+1 WHERE x=new.rowid; |
| 161 SELECT 123; |
| 162 DELETE FROM t2 WHERE x=new.rowid; |
| 163 END; |
| 164 INSERT INTO t1(a,b) VALUES(2,3); |
| 165 COMMIT; |
| 166 } |
| 167 } |
| 168 |
| 169 # Ensure that no file descriptors were leaked. |
| 170 do_test malloc-5.X { |
| 171 catch {db close} |
| 172 set sqlite_open_file_count |
| 173 } {0} |
| 174 |
| 175 ifcapable vacuum { |
| 176 do_malloc_test 6 -sqlprep { |
| 177 BEGIN TRANSACTION; |
| 178 CREATE TABLE t1(a); |
| 179 INSERT INTO t1 VALUES(1); |
| 180 INSERT INTO t1 SELECT a*2 FROM t1; |
| 181 INSERT INTO t1 SELECT a*2 FROM t1; |
| 182 INSERT INTO t1 SELECT a*2 FROM t1; |
| 183 INSERT INTO t1 SELECT a*2 FROM t1; |
| 184 INSERT INTO t1 SELECT a*2 FROM t1; |
| 185 INSERT INTO t1 SELECT a*2 FROM t1; |
| 186 INSERT INTO t1 SELECT a*2 FROM t1; |
| 187 INSERT INTO t1 SELECT a*2 FROM t1; |
| 188 INSERT INTO t1 SELECT a*2 FROM t1; |
| 189 INSERT INTO t1 SELECT a*2 FROM t1; |
| 190 DELETE FROM t1 where rowid%5 = 0; |
| 191 COMMIT; |
| 192 } -sqlbody { |
| 193 VACUUM; |
| 194 } |
| 195 } |
| 196 |
| 197 autoinstall_test_functions |
| 198 do_malloc_test 7 -sqlprep { |
| 199 CREATE TABLE t1(a, b); |
| 200 INSERT INTO t1 VALUES(1, 2); |
| 201 INSERT INTO t1 VALUES(3, 4); |
| 202 INSERT INTO t1 VALUES(5, 6); |
| 203 INSERT INTO t1 VALUES(7, randstr(1200,1200)); |
| 204 } -sqlbody { |
| 205 SELECT min(a) FROM t1 WHERE a<6 GROUP BY b; |
| 206 SELECT a FROM t1 WHERE a<6 ORDER BY a; |
| 207 SELECT b FROM t1 WHERE a>6; |
| 208 } |
| 209 |
| 210 # This block is designed to test that some malloc failures that may |
| 211 # occur in vdbeapi.c. Specifically, if a malloc failure that occurs |
| 212 # when converting UTF-16 text to integers and real numbers is handled |
| 213 # correctly. |
| 214 # |
| 215 # This is done by retrieving a string from the database engine and |
| 216 # manipulating it using the sqlite3_column_*** APIs. This doesn't |
| 217 # actually return an error to the user when a malloc() fails.. That |
| 218 # could be viewed as a bug. |
| 219 # |
| 220 # These tests only run if UTF-16 support is compiled in. |
| 221 # |
| 222 ifcapable utf16 { |
| 223 set ::STMT {} |
| 224 do_malloc_test 8 -tclprep { |
| 225 set sql "SELECT '[string repeat abc 20]', '[string repeat def 20]', ?" |
| 226 set ::STMT [sqlite3_prepare db $sql -1 X] |
| 227 sqlite3_step $::STMT |
| 228 if { $::tcl_platform(byteOrder)=="littleEndian" } { |
| 229 set ::bomstr "\xFF\xFE" |
| 230 } else { |
| 231 set ::bomstr "\xFE\xFF" |
| 232 } |
| 233 append ::bomstr [encoding convertto unicode "123456789_123456789_123456789"] |
| 234 } -tclbody { |
| 235 sqlite3_column_text16 $::STMT 0 |
| 236 sqlite3_column_int $::STMT 0 |
| 237 sqlite3_column_text16 $::STMT 1 |
| 238 sqlite3_column_double $::STMT 1 |
| 239 set rc [sqlite3_reset $::STMT] |
| 240 if {$rc eq "SQLITE_NOMEM"} {error "out of memory"} |
| 241 sqlite3_bind_text16 $::STMT 1 $::bomstr 60 |
| 242 #catch {sqlite3_finalize $::STMT} |
| 243 #if {[lindex [sqlite_malloc_stat] 2]<=0} { |
| 244 # error "out of memory" |
| 245 #} |
| 246 } -cleanup { |
| 247 if {$::STMT!=""} { |
| 248 sqlite3_finalize $::STMT |
| 249 set ::STMT {} |
| 250 } |
| 251 } |
| 252 } |
| 253 |
| 254 # This block tests that malloc() failures that occur whilst commiting |
| 255 # a multi-file transaction are handled correctly. |
| 256 # |
| 257 do_malloc_test 9 -sqlprep { |
| 258 ATTACH 'test2.db' as test2; |
| 259 CREATE TABLE abc1(a, b, c); |
| 260 CREATE TABLE test2.abc2(a, b, c); |
| 261 } -sqlbody { |
| 262 BEGIN; |
| 263 INSERT INTO abc1 VALUES(1, 2, 3); |
| 264 INSERT INTO abc2 VALUES(1, 2, 3); |
| 265 COMMIT; |
| 266 } |
| 267 |
| 268 # This block tests malloc() failures that occur while opening a |
| 269 # connection to a database. |
| 270 do_malloc_test 10 -tclprep { |
| 271 catch {db2 close} |
| 272 db close |
| 273 file delete -force test.db test.db-journal |
| 274 sqlite3 db test.db |
| 275 sqlite3_extended_result_codes db 1 |
| 276 db eval {CREATE TABLE abc(a, b, c)} |
| 277 } -tclbody { |
| 278 db close |
| 279 sqlite3 db2 test.db |
| 280 sqlite3_extended_result_codes db2 1 |
| 281 db2 eval {SELECT * FROM sqlite_master} |
| 282 db2 close |
| 283 } |
| 284 |
| 285 # This block tests malloc() failures that occur within calls to |
| 286 # sqlite3_create_function(). |
| 287 do_malloc_test 11 -tclbody { |
| 288 set rc [sqlite3_create_function db] |
| 289 if {[string match $rc SQLITE_OK]} { |
| 290 set rc [sqlite3_create_aggregate db] |
| 291 } |
| 292 if {[string match $rc SQLITE_NOMEM]} { |
| 293 error "out of memory" |
| 294 } |
| 295 } |
| 296 |
| 297 do_malloc_test 12 -tclbody { |
| 298 set sql16 [encoding convertto unicode "SELECT * FROM sqlite_master"] |
| 299 append sql16 "\00\00" |
| 300 set ::STMT [sqlite3_prepare16 db $sql16 -1 DUMMY] |
| 301 sqlite3_finalize $::STMT |
| 302 } |
| 303 |
| 304 # Test malloc errors when replaying two hot journals from a 2-file |
| 305 # transaction. |
| 306 ifcapable crashtest&&attach { |
| 307 do_malloc_test 13 -tclprep { |
| 308 set rc [crashsql -delay 1 -file test2.db { |
| 309 ATTACH 'test2.db' as aux; |
| 310 PRAGMA cache_size = 10; |
| 311 BEGIN; |
| 312 CREATE TABLE aux.t2(a, b, c); |
| 313 CREATE TABLE t1(a, b, c); |
| 314 COMMIT; |
| 315 }] |
| 316 if {$rc!="1 {child process exited abnormally}"} { |
| 317 error "Wrong error message: $rc" |
| 318 } |
| 319 } -tclbody { |
| 320 db eval {ATTACH 'test2.db' as aux;} |
| 321 set rc [catch {db eval { |
| 322 SELECT * FROM t1; |
| 323 SELECT * FROM t2; |
| 324 }} err] |
| 325 if {$rc && $err!="no such table: t1"} { |
| 326 error $err |
| 327 } |
| 328 } |
| 329 } |
| 330 |
| 331 if {$tcl_platform(platform)!="windows"} { |
| 332 do_malloc_test 14 -tclprep { |
| 333 catch {db close} |
| 334 sqlite3 db2 test2.db |
| 335 sqlite3_extended_result_codes db2 1 |
| 336 db2 eval { |
| 337 PRAGMA journal_mode = DELETE; /* For inmemory_journal permutation */ |
| 338 PRAGMA synchronous = 0; |
| 339 CREATE TABLE t1(a, b); |
| 340 INSERT INTO t1 VALUES(1, 2); |
| 341 BEGIN; |
| 342 INSERT INTO t1 VALUES(3, 4); |
| 343 } |
| 344 copy_file test2.db test.db |
| 345 copy_file test2.db-journal test.db-journal |
| 346 db2 close |
| 347 } -tclbody { |
| 348 sqlite3 db test.db |
| 349 sqlite3_extended_result_codes db 1 |
| 350 |
| 351 # If an out-of-memory occurs within a call to a VFS layer function during |
| 352 # hot-journal rollback, sqlite will report SQLITE_CORRUPT. See commit |
| 353 # [5668] for details. |
| 354 set rc [catch {db eval { SELECT * FROM t1 }} msg] |
| 355 if {$msg eq "database disk image is malformed"} { set msg "out of memory" } |
| 356 if {$rc} { error $msg } |
| 357 set msg |
| 358 } |
| 359 } |
| 360 |
| 361 proc string_compare {a b} { |
| 362 return [string compare $a $b] |
| 363 } |
| 364 |
| 365 # Test for malloc() failures in sqlite3_create_collation() and |
| 366 # sqlite3_create_collation16(). |
| 367 # |
| 368 ifcapable utf16 { |
| 369 do_malloc_test 15 -start 4 -tclbody { |
| 370 db collate string_compare string_compare |
| 371 if {[catch {add_test_collate db 1 1 1} msg]} { |
| 372 if {$msg=="SQLITE_NOMEM"} {set msg "out of memory"} |
| 373 error $msg |
| 374 } |
| 375 |
| 376 db complete {SELECT "hello """||'world"' [microsoft], * FROM anicetable;} |
| 377 db complete {-- Useful comment} |
| 378 |
| 379 execsql { |
| 380 CREATE TABLE t1(a, b COLLATE string_compare); |
| 381 INSERT INTO t1 VALUES(10, 'string'); |
| 382 INSERT INTO t1 VALUES(10, 'string2'); |
| 383 } |
| 384 } |
| 385 } |
| 386 |
| 387 # Also test sqlite3_complete(). There are (currently) no malloc() |
| 388 # calls in this function, but test anyway against future changes. |
| 389 # |
| 390 do_malloc_test 16 -tclbody { |
| 391 db complete {SELECT "hello """||'world"' [microsoft], * FROM anicetable;} |
| 392 db complete {-- Useful comment} |
| 393 db eval { |
| 394 SELECT * FROM sqlite_master; |
| 395 } |
| 396 } |
| 397 |
| 398 # Test handling of malloc() failures in sqlite3_open16(). |
| 399 # |
| 400 ifcapable utf16 { |
| 401 do_malloc_test 17 -tclbody { |
| 402 set DB2 0 |
| 403 set STMT 0 |
| 404 |
| 405 # open database using sqlite3_open16() |
| 406 set filename [encoding convertto unicode test.db] |
| 407 append filename "\x00\x00" |
| 408 set DB2 [sqlite3_open16 $filename -unused] |
| 409 if {0==$DB2} { |
| 410 error "out of memory" |
| 411 } |
| 412 sqlite3_extended_result_codes $DB2 1 |
| 413 |
| 414 # Prepare statement |
| 415 set rc [catch {sqlite3_prepare $DB2 {SELECT * FROM sqlite_master} -1 X} msg] |
| 416 if {[sqlite3_errcode $DB2] eq "SQLITE_IOERR+12"} { |
| 417 error "out of memory" |
| 418 } |
| 419 if {[regexp ".*automatic extension loading.*" [sqlite3_errmsg $DB2]]} { |
| 420 error "out of memory" |
| 421 } |
| 422 if {$rc} { |
| 423 error [string range $msg 4 end] |
| 424 } |
| 425 set STMT $msg |
| 426 |
| 427 # Finalize statement |
| 428 set rc [sqlite3_finalize $STMT] |
| 429 if {$rc!="SQLITE_OK"} { |
| 430 error [sqlite3_errmsg $DB2] |
| 431 } |
| 432 set STMT 0 |
| 433 |
| 434 # Close database |
| 435 set rc [sqlite3_close $DB2] |
| 436 if {$rc!="SQLITE_OK"} { |
| 437 error [sqlite3_errmsg $DB2] |
| 438 } |
| 439 set DB2 0 |
| 440 } -cleanup { |
| 441 if {$STMT!="0"} { |
| 442 sqlite3_finalize $STMT |
| 443 } |
| 444 if {$DB2!="0"} { |
| 445 set rc [sqlite3_close $DB2] |
| 446 } |
| 447 } |
| 448 } |
| 449 |
| 450 # Test handling of malloc() failures in sqlite3_errmsg16(). |
| 451 # |
| 452 ifcapable utf16 { |
| 453 do_malloc_test 18 -tclprep { |
| 454 catch { |
| 455 db eval "SELECT [string repeat longcolumnname 10] FROM sqlite_master" |
| 456 } |
| 457 } -tclbody { |
| 458 set utf16 [sqlite3_errmsg16 [sqlite3_connection_pointer db]] |
| 459 binary scan $utf16 c* bytes |
| 460 if {[llength $bytes]==0} { |
| 461 error "out of memory" |
| 462 } |
| 463 } |
| 464 } |
| 465 |
| 466 # This test is aimed at coverage testing. Specificly, it is supposed to |
| 467 # cause a malloc() only used when converting between the two utf-16 |
| 468 # encodings to fail (i.e. little-endian->big-endian). It only actually |
| 469 # hits this malloc() on little-endian hosts. |
| 470 # |
| 471 set static_string "\x00h\x00e\x00l\x00l\x00o" |
| 472 for {set l 0} {$l<10} {incr l} { |
| 473 append static_string $static_string |
| 474 } |
| 475 append static_string "\x00\x00" |
| 476 do_malloc_test 19 -tclprep { |
| 477 execsql { |
| 478 PRAGMA encoding = "UTF16be"; |
| 479 CREATE TABLE abc(a, b, c); |
| 480 } |
| 481 } -tclbody { |
| 482 unset -nocomplain ::STMT |
| 483 set r [catch { |
| 484 set ::STMT [sqlite3_prepare db {SELECT ?} -1 DUMMY] |
| 485 sqlite3_bind_text16 -static $::STMT 1 $static_string 112 |
| 486 } msg] |
| 487 if {$r} {error [string range $msg 4 end]} |
| 488 set msg |
| 489 } -cleanup { |
| 490 if {[info exists ::STMT]} { |
| 491 sqlite3_finalize $::STMT |
| 492 } |
| 493 } |
| 494 unset static_string |
| 495 |
| 496 # Make sure SQLITE_NOMEM is reported out on an ATTACH failure even |
| 497 # when the malloc failure occurs within the nested parse. |
| 498 # |
| 499 ifcapable attach { |
| 500 do_malloc_test 20 -tclprep { |
| 501 db close |
| 502 file delete -force test2.db test2.db-journal |
| 503 sqlite3 db test2.db |
| 504 sqlite3_extended_result_codes db 1 |
| 505 db eval {CREATE TABLE t1(x);} |
| 506 db close |
| 507 } -tclbody { |
| 508 if {[catch {sqlite3 db test.db}]} { |
| 509 error "out of memory" |
| 510 } |
| 511 sqlite3_extended_result_codes db 1 |
| 512 } -sqlbody { |
| 513 ATTACH DATABASE 'test2.db' AS t2; |
| 514 SELECT * FROM t1; |
| 515 DETACH DATABASE t2; |
| 516 } |
| 517 } |
| 518 |
| 519 # Test malloc failure whilst installing a foreign key. |
| 520 # |
| 521 ifcapable foreignkey { |
| 522 do_malloc_test 21 -sqlbody { |
| 523 CREATE TABLE abc(a, b, c, FOREIGN KEY(a) REFERENCES abc(b)) |
| 524 } |
| 525 } |
| 526 |
| 527 # Test malloc failure in an sqlite3_prepare_v2() call. |
| 528 # |
| 529 do_malloc_test 22 -tclbody { |
| 530 set ::STMT "" |
| 531 set r [catch { |
| 532 set ::STMT [ |
| 533 sqlite3_prepare_v2 db "SELECT * FROM sqlite_master" -1 DUMMY |
| 534 ] |
| 535 } msg] |
| 536 if {$r} {error [string range $msg 4 end]} |
| 537 } -cleanup { |
| 538 if {$::STMT ne ""} { |
| 539 sqlite3_finalize $::STMT |
| 540 set ::STMT "" |
| 541 } |
| 542 } |
| 543 |
| 544 ifcapable {pager_pragmas} { |
| 545 # This tests a special case - that an error that occurs while the pager |
| 546 # is trying to recover from error-state in exclusive-access mode works. |
| 547 # |
| 548 do_malloc_test 23 -tclprep { |
| 549 db eval { |
| 550 PRAGMA cache_size = 10; |
| 551 PRAGMA locking_mode = exclusive; |
| 552 BEGIN; |
| 553 CREATE TABLE abc(a, b, c); |
| 554 CREATE INDEX abc_i ON abc(a, b, c); |
| 555 INSERT INTO abc |
| 556 VALUES(randstr(100,100), randstr(100,100), randstr(100,100)); |
| 557 INSERT INTO abc |
| 558 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; |
| 559 INSERT INTO abc |
| 560 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; |
| 561 INSERT INTO abc |
| 562 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; |
| 563 INSERT INTO abc |
| 564 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; |
| 565 INSERT INTO abc |
| 566 SELECT randstr(100,100), randstr(100,100), randstr(100,100) FROM abc; |
| 567 COMMIT; |
| 568 } |
| 569 |
| 570 # This puts the pager into error state. |
| 571 # |
| 572 db eval BEGIN |
| 573 db eval {UPDATE abc SET a = 0 WHERE oid%2} |
| 574 set ::sqlite_io_error_pending 10 |
| 575 catch {db eval {ROLLBACK}} msg |
| 576 |
| 577 } -tclbody { |
| 578 # If an out-of-memory occurs within a call to a VFS layer function during |
| 579 # hot-journal rollback, sqlite will report SQLITE_CORRUPT. See commit |
| 580 # [5668] for details. |
| 581 set rc [catch {db eval { SELECT * FROM abc LIMIT 10 }} msg] |
| 582 if {$msg eq "database disk image is malformed"} { set msg "out of memory" } |
| 583 if {$rc} { error $msg } |
| 584 set msg |
| 585 } -cleanup { |
| 586 set e [db eval {PRAGMA integrity_check}] |
| 587 if {$e ne "ok"} {error $e} |
| 588 } |
| 589 } |
| 590 |
| 591 ifcapable compound { |
| 592 do_malloc_test 24 -sqlprep { |
| 593 CREATE TABLE t1(a, b, c) |
| 594 } -sqlbody { |
| 595 SELECT 1 FROM t1 UNION SELECT 2 FROM t1 ORDER BY 1 |
| 596 } |
| 597 } |
| 598 |
| 599 ifcapable view&&trigger { |
| 600 do_malloc_test 25 -sqlprep { |
| 601 CREATE TABLE t1(a, b, c); |
| 602 CREATE VIEW v1 AS SELECT * FROM t1; |
| 603 CREATE TRIGGER v1t1 INSTEAD OF DELETE ON v1 BEGIN SELECT 1; END; |
| 604 CREATE TRIGGER v1t2 INSTEAD OF INSERT ON v1 BEGIN SELECT 1; END; |
| 605 CREATE TRIGGER v1t3 INSTEAD OF UPDATE ON v1 BEGIN SELECT 1; END; |
| 606 } -sqlbody { |
| 607 DELETE FROM v1 WHERE a = 1; |
| 608 INSERT INTO v1 VALUES(1, 2, 3); |
| 609 UPDATE v1 SET a = 1 WHERE b = 2; |
| 610 } |
| 611 } |
| 612 |
| 613 do_malloc_test 25 -sqlprep { |
| 614 CREATE TABLE abc(a, b, c); |
| 615 CREATE INDEX i1 ON abc(a, b); |
| 616 INSERT INTO abc VALUES(1, 2, 3); |
| 617 INSERT INTO abc VALUES(4, 5, 6); |
| 618 } -tclbody { |
| 619 # For each UPDATE executed, the cursor used for the SELECT statement |
| 620 # must be "saved". Because the cursor is open on an index, this requires |
| 621 # a malloc() to allocate space to save the index key. This test case is |
| 622 # aimed at testing the response of the library to a failure in that |
| 623 # particular malloc() call. |
| 624 db eval {SELECT a FROM abc ORDER BY a} { |
| 625 db eval {UPDATE abc SET b = b - 1 WHERE a = $a} |
| 626 } |
| 627 } |
| 628 |
| 629 # This test is designed to test a specific juncture in the sqlite code. |
| 630 # The database set up by -sqlprep script contains a single table B-Tree |
| 631 # of height 2. In the -tclbody script, the existing database connection |
| 632 # is closed and a new one opened and used to insert a new row into the |
| 633 # table B-Tree. By using a new connection, the outcome of a malloc() |
| 634 # failure while seeking to the right-hand side of the B-Tree to insert |
| 635 # a new record can be tested. |
| 636 # |
| 637 do_malloc_test 26 -sqlprep { |
| 638 BEGIN; |
| 639 CREATE TABLE t1(a, b); |
| 640 INSERT INTO t1 VALUES(1, randomblob(210)); |
| 641 INSERT INTO t1 VALUES(1, randomblob(210)); |
| 642 INSERT INTO t1 VALUES(1, randomblob(210)); |
| 643 INSERT INTO t1 VALUES(1, randomblob(210)); |
| 644 INSERT INTO t1 VALUES(1, randomblob(210)); |
| 645 COMMIT; |
| 646 } -tclbody { |
| 647 db close |
| 648 sqlite3 db test.db |
| 649 db eval { INSERT INTO t1 VALUES(1, randomblob(210)) } |
| 650 } |
| 651 |
| 652 # Test that no memory is leaked following a malloc() failure in |
| 653 # sqlite3_initialize(). |
| 654 # |
| 655 do_malloc_test 27 -tclprep { |
| 656 db close |
| 657 sqlite3_shutdown |
| 658 } -tclbody { |
| 659 set rc [sqlite3_initialize] |
| 660 if {$rc == "SQLITE_NOMEM"} { |
| 661 error "out of memory" |
| 662 } |
| 663 } |
| 664 autoinstall_test_functions |
| 665 |
| 666 # Test that malloc failures that occur while processing INDEXED BY |
| 667 # clauses are handled correctly. |
| 668 do_malloc_test 28 -sqlprep { |
| 669 CREATE TABLE t1(a, b); |
| 670 CREATE INDEX i1 ON t1(a); |
| 671 CREATE VIEW v1 AS SELECT * FROM t1 INDEXED BY i1 WHERE a = 10; |
| 672 } -sqlbody { |
| 673 SELECT * FROM t1 INDEXED BY i1 ORDER BY a; |
| 674 SELECT * FROM v1; |
| 675 } |
| 676 |
| 677 do_malloc_test 29 -sqlprep { |
| 678 CREATE TABLE t1(a TEXT, b TEXT); |
| 679 } -sqlbody { |
| 680 INSERT INTO t1 VALUES(1, -234); |
| 681 INSERT INTO t1 SELECT * FROM t1 UNION ALL SELECT * FROM t1; |
| 682 } |
| 683 |
| 684 do_malloc_test 30 -tclprep { |
| 685 db eval { |
| 686 CREATE TABLE t1(x PRIMARY KEY); |
| 687 INSERT INTO t1 VALUES(randstr(500,500)); |
| 688 INSERT INTO t1 VALUES(randstr(500,500)); |
| 689 INSERT INTO t1 VALUES(randstr(500,500)); |
| 690 } |
| 691 db close |
| 692 sqlite3 db test.db |
| 693 |
| 694 # The DELETE command in the following block moves the overflow pages that |
| 695 # are part of the primary key index to the free-list. But it does not |
| 696 # actually load the content of the pages. This leads to the peculiar |
| 697 # situation where cache entries exist, but are not populated with data. |
| 698 # They are populated next time they are requested by the b-tree layer. |
| 699 # |
| 700 db eval { |
| 701 BEGIN; |
| 702 DELETE FROM t1; |
| 703 ROLLBACK; |
| 704 } |
| 705 } -sqlbody { |
| 706 -- This statement requires the 'no-content' pages loaded by the DELETE |
| 707 -- statement above. When requesting the pages, the content is loaded |
| 708 -- from the database file. The point of this test case is to test handling |
| 709 -- of malloc errors (including SQLITE_IOERR_NOMEM errors) when loading |
| 710 -- the content. |
| 711 SELECT * FROM t1 ORDER BY x; |
| 712 } |
| 713 |
| 714 # After committing a transaction in persistent-journal mode, if a journal |
| 715 # size limit is configured SQLite may attempt to truncate the journal file. |
| 716 # This test verifies the libraries response to a malloc() failure during |
| 717 # this operation. |
| 718 # |
| 719 do_malloc_test 31 -sqlprep { |
| 720 PRAGMA journal_mode = persist; |
| 721 PRAGMA journal_size_limit = 1024; |
| 722 CREATE TABLE t1(a PRIMARY KEY, b); |
| 723 } -sqlbody { |
| 724 INSERT INTO t1 VALUES(1, 2); |
| 725 } |
| 726 |
| 727 # When written, this test provoked an obscure change-counter bug. |
| 728 # |
| 729 # If, when running in exclusive mode, a malloc() failure occurs |
| 730 # after the database file change-counter has been written but |
| 731 # before the transaction has been committed, then the transaction |
| 732 # is automatically rolled back. However, internally the |
| 733 # Pager.changeCounterDone flag was being left set. This means |
| 734 # that if the same connection attempts another transaction following |
| 735 # the malloc failure and rollback, the change counter will not |
| 736 # be updated. This could corrupt another processes cache. |
| 737 # |
| 738 do_malloc_test 32 -tclprep { |
| 739 # Build a small database containing an indexed table. |
| 740 # |
| 741 db eval { |
| 742 PRAGMA locking_mode = normal; |
| 743 BEGIN; |
| 744 CREATE TABLE t1(a PRIMARY KEY, b); |
| 745 INSERT INTO t1 VALUES(1, 'one'); |
| 746 INSERT INTO t1 VALUES(2, 'two'); |
| 747 INSERT INTO t1 VALUES(3, 'three'); |
| 748 COMMIT; |
| 749 PRAGMA locking_mode = exclusive; |
| 750 } |
| 751 |
| 752 # Open a second database connection. Load the table (but not index) |
| 753 # into the second connections pager cache. |
| 754 # |
| 755 sqlite3 db2 test.db |
| 756 db2 eval { |
| 757 PRAGMA locking_mode = normal; |
| 758 SELECT b FROM t1; |
| 759 } |
| 760 |
| 761 } -tclbody { |
| 762 # Running in exclusive mode, perform a database transaction that |
| 763 # modifies both the database table and index. For iterations where |
| 764 # the malloc failure occurs after updating the change counter but |
| 765 # before committing the transaction, this should result in the |
| 766 # transaction being rolled back but the changeCounterDone flag |
| 767 # left set. |
| 768 # |
| 769 db eval { UPDATE t1 SET a = a + 3 } |
| 770 } -cleanup { |
| 771 |
| 772 # Perform another transaction using the first connection. Unlock |
| 773 # the database after doing so. If this is one of the right iterations, |
| 774 # then this should result in the database contents being updated but |
| 775 # the change-counter left as it is. |
| 776 # |
| 777 db eval { |
| 778 PRAGMA locking_mode = normal; |
| 779 UPDATE t1 SET a = a + 3; |
| 780 } |
| 781 |
| 782 # Now do an integrity check with the second connection. The second |
| 783 # connection still has the database table in its cache. If this is |
| 784 # one of the magic iterations and the change counter was not modified, |
| 785 # then it won't realize that the cached data is out of date. Since |
| 786 # the cached data won't match the up to date index data read from |
| 787 # the database file, the integrity check should fail. |
| 788 # |
| 789 set zRepeat "transient" |
| 790 if {$::iRepeat} {set zRepeat "persistent"} |
| 791 do_test malloc-32.$zRepeat.${::n}.integrity { |
| 792 execsql {PRAGMA integrity_check} db2 |
| 793 } {ok} |
| 794 db2 close |
| 795 } |
| 796 |
| 797 # The following two OOM tests verify that OOM handling works in the |
| 798 # code used to optimize "SELECT count(*) FROM <tbl>". |
| 799 # |
| 800 do_malloc_test 33 -tclprep { |
| 801 db eval { PRAGMA cache_size = 10 } |
| 802 db transaction { |
| 803 db eval { CREATE TABLE abc(a, b) } |
| 804 for {set i 0} {$i<500} {incr i} { |
| 805 db eval {INSERT INTO abc VALUES(randstr(100,100), randstr(1000,1000))} |
| 806 } |
| 807 } |
| 808 } -sqlbody { |
| 809 SELECT count(*) FROM abc; |
| 810 } |
| 811 do_malloc_test 34 -tclprep { |
| 812 db eval { PRAGMA cache_size = 10 } |
| 813 db transaction { |
| 814 db eval { CREATE TABLE abc(a PRIMARY KEY, b) } |
| 815 for {set i 0} {$i<500} {incr i} { |
| 816 db eval {INSERT INTO abc VALUES(randstr(100,100), randstr(1000,1000))} |
| 817 } |
| 818 } |
| 819 } -sqlbody { |
| 820 SELECT count(*) FROM abc; |
| 821 } |
| 822 |
| 823 proc f {args} { error "Quite a long error!" } |
| 824 do_malloc_test 35 -tclprep { |
| 825 db func f f |
| 826 set ::STMT [sqlite3_prepare db "SELECT f()" -1 DUMMY] |
| 827 sqlite3_step $::STMT |
| 828 } -tclbody { |
| 829 sqlite3_finalize $::STMT |
| 830 } -cleanup { |
| 831 # At one point an assert( !db->mallocFailed ) could fail in the following |
| 832 # call to sqlite3_errmsg(). Because sqlite3_finalize() had failed to clear |
| 833 # the flag before returning. |
| 834 sqlite3_errmsg16 db |
| 835 } |
| 836 |
| 837 do_malloc_test 36 -sqlprep { |
| 838 CREATE TABLE t1(a, b); |
| 839 INSERT INTO t1 VALUES(1, 2); |
| 840 INSERT INTO t1 VALUES(3, 4); |
| 841 } -sqlbody { |
| 842 SELECT test_agg_errmsg16(), group_concat(a) FROM t1 |
| 843 } |
| 844 |
| 845 # At one point, if an OOM occured immediately after obtaining a shared lock |
| 846 # on the database file, the file remained locked. This test case ensures |
| 847 # that bug has been fixed.i |
| 848 if {[db eval {PRAGMA locking_mode}]!="exclusive"} { |
| 849 do_malloc_test 37 -tclprep { |
| 850 sqlite3 db2 test.db |
| 851 execsql { |
| 852 CREATE TABLE t1(a, b); |
| 853 INSERT INTO t1 VALUES(1, 2); |
| 854 } db2 |
| 855 } -sqlbody { |
| 856 SELECT * FROM t1; |
| 857 } -cleanup { |
| 858 # Try to write to the database using connection [db2]. If connection [db] |
| 859 # has correctly released the shared lock, this write attempt should |
| 860 # succeed. If [db] has not released the lock, this should hit an |
| 861 # SQLITE_BUSY error. |
| 862 do_test malloc-36.$zRepeat.${::n}.unlocked { |
| 863 execsql {INSERT INTO t1 VALUES(3, 4)} db2 |
| 864 } {} |
| 865 db2 close |
| 866 } |
| 867 catch { db2 close } |
| 868 } |
| 869 |
| 870 ifcapable stat2&&utf16 { |
| 871 do_malloc_test 38 -tclprep { |
| 872 add_test_collate db 0 0 1 |
| 873 execsql { |
| 874 ANALYZE; |
| 875 CREATE TABLE t4(x COLLATE test_collate); |
| 876 CREATE INDEX t4x ON t4(x); |
| 877 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 0, 'aaa'); |
| 878 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 1, 'aaa'); |
| 879 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 2, 'aaa'); |
| 880 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 3, 'aaa'); |
| 881 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 4, 'aaa'); |
| 882 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 5, 'aaa'); |
| 883 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 6, 'aaa'); |
| 884 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 7, 'aaa'); |
| 885 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 8, 'aaa'); |
| 886 INSERT INTO sqlite_stat2 VALUES('t4', 't4x', 9, 'aaa'); |
| 887 } |
| 888 db close |
| 889 sqlite3 db test.db |
| 890 sqlite3_db_config_lookaside db 0 0 0 |
| 891 add_test_collate db 0 0 1 |
| 892 } -sqlbody { |
| 893 SELECT * FROM t4 AS t41, t4 AS t42 WHERE t41.x>'ddd' AND t42.x>'ccc' |
| 894 } |
| 895 } |
| 896 |
| 897 # Ensure that no file descriptors were leaked. |
| 898 do_test malloc-99.X { |
| 899 catch {db close} |
| 900 set sqlite_open_file_count |
| 901 } {0} |
| 902 |
| 903 puts open-file-count=$sqlite_open_file_count |
| 904 finish_test |
OLD | NEW |