Index: third_party/sqlite/sqlite-src-3080704/test/malloc4.test |
diff --git a/third_party/sqlite/sqlite-src-3080704/test/malloc4.test b/third_party/sqlite/sqlite-src-3080704/test/malloc4.test |
new file mode 100644 |
index 0000000000000000000000000000000000000000..e2a616a5896c61b51bb176da7baab2fab4d60cbe |
--- /dev/null |
+++ b/third_party/sqlite/sqlite-src-3080704/test/malloc4.test |
@@ -0,0 +1,194 @@ |
+# 2005 November 30 |
+# |
+# The author disclaims copyright to this source code. In place of |
+# a legal notice, here is a blessing: |
+# |
+# May you do good and not evil. |
+# May you find forgiveness for yourself and forgive others. |
+# May you share freely, never taking more than you give. |
+# |
+#*********************************************************************** |
+# |
+# This file contains tests to ensure that the library handles malloc() failures |
+# correctly. The emphasis in this file is on sqlite3_column_XXX() APIs. |
+# |
+# $Id: malloc4.test,v 1.10 2008/02/18 22:24:58 drh Exp $ |
+ |
+#--------------------------------------------------------------------------- |
+# NOTES ON EXPECTED BEHAVIOUR |
+# |
+# [193] When a memory allocation failure occurs during sqlite3_column_name(), |
+# sqlite3_column_name16(), sqlite3_column_decltype(), or |
+# sqlite3_column_decltype16() the function shall return NULL. |
+# |
+#--------------------------------------------------------------------------- |
+ |
+set testdir [file dirname $argv0] |
+source $testdir/tester.tcl |
+source $testdir/malloc_common.tcl |
+ |
+# Only run these tests if memory debugging is turned on. |
+if {!$MEMDEBUG} { |
+ puts "Skipping malloc4 tests: not compiled with -DSQLITE_MEMDEBUG..." |
+ finish_test |
+ return |
+} |
+ |
+ifcapable !utf16 { |
+ finish_test |
+ return |
+} |
+ |
+proc do_stmt_test {id sql} { |
+ set ::sql $sql |
+ set go 1 |
+ for {set n 0} {$go} {incr n} { |
+ set testid "malloc4-$id.$n" |
+ |
+ # Prepare the statement |
+ do_test ${testid}.1 { |
+ set ::STMT [sqlite3_prepare $::DB $sql -1 TAIL] |
+ expr [string length $::STMT] > 0 |
+ } {1} |
+ |
+ # Set the Nth malloc() to fail. |
+ sqlite3_memdebug_fail $n -repeat 0 |
+ |
+ # Test malloc failure in the _name(), _name16(), decltype() and |
+ # decltype16() APIs. Calls that occur after the malloc() failure should |
+ # return NULL. No error is raised though. |
+ # |
+ # ${testid}.2.1 - Call _name() |
+ # ${testid}.2.2 - Call _name16() |
+ # ${testid}.2.3 - Call _name() |
+ # ${testid}.2.4 - Check that the return values of the above three calls are |
+ # consistent with each other and with the simulated |
+ # malloc() failures. |
+ # |
+ # Because the code that implements the _decltype() and _decltype16() APIs |
+ # is the same as the _name() and _name16() implementations, we don't worry |
+ # about explicitly testing them. |
+ # |
+ do_test ${testid}.2.1 { |
+ set mf1 [expr [sqlite3_memdebug_pending] < 0] |
+ set ::name8 [sqlite3_column_name $::STMT 0] |
+ set mf2 [expr [sqlite3_memdebug_pending] < 0] |
+ expr {$mf1 == $mf2 || $::name8 == ""} |
+ } {1} |
+ do_test ${testid}.2.2 { |
+ set mf1 [expr [sqlite3_memdebug_pending] < 0] |
+ set ::name16 [sqlite3_column_name16 $::STMT 0] |
+ set ::name16 [encoding convertfrom unicode $::name16] |
+ set ::name16 [string range $::name16 0 end-1] |
+ set mf2 [expr [sqlite3_memdebug_pending] < 0] |
+ expr {$mf1 == $mf2 || $::name16 == ""} |
+ } {1} |
+ do_test ${testid}.2.3 { |
+ set mf1 [expr [sqlite3_memdebug_pending] < 0] |
+ set ::name8_2 [sqlite3_column_name $::STMT 0] |
+ set mf2 [expr [sqlite3_memdebug_pending] < 0] |
+ expr {$mf1 == $mf2 || $::name8_2 == ""} |
+ } {1} |
+ set ::mallocFailed [expr [sqlite3_memdebug_pending] < 0] |
+ do_test ${testid}.2.4 { |
+ expr { |
+ $::name8 == $::name8_2 && $::name16 == $::name8 && !$::mallocFailed || |
+ $::name8 == $::name8_2 && $::name16 == "" && $::mallocFailed || |
+ $::name8 == $::name16 && $::name8_2 == "" && $::mallocFailed || |
+ $::name8_2 == $::name16 && $::name8 == "" && $::mallocFailed |
+ } |
+ } {1} |
+ |
+ # Step the statement so that we can call _text() and _text16(). Before |
+ # running sqlite3_step(), make sure that malloc() is not about to fail. |
+ # Memory allocation failures that occur within sqlite3_step() are tested |
+ # elsewhere. |
+ set mf [sqlite3_memdebug_pending] |
+ sqlite3_memdebug_fail -1 |
+ do_test ${testid}.3 { |
+ sqlite3_step $::STMT |
+ } {SQLITE_ROW} |
+ sqlite3_memdebug_fail $mf |
+ |
+ # Test for malloc() failures within _text() and _text16(). |
+ # |
+ do_test ${testid}.4.1 { |
+ set ::text8 [sqlite3_column_text $::STMT 0] |
+ set mf [expr [sqlite3_memdebug_pending] < 0 && !$::mallocFailed] |
+ expr {$mf==0 || $::text8 == ""} |
+ } {1} |
+ do_test ${testid}.4.2 { |
+ set ::text16 [sqlite3_column_text16 $::STMT 0] |
+ set ::text16 [encoding convertfrom unicode $::text16] |
+ set ::text16 [string range $::text16 0 end-1] |
+ set mf [expr [sqlite3_memdebug_pending] < 0 && !$::mallocFailed] |
+ expr {$mf==0 || $::text16 == ""} |
+ } {1} |
+ do_test ${testid}.4.3 { |
+ set ::text8_2 [sqlite3_column_text $::STMT 0] |
+ set mf [expr [sqlite3_memdebug_pending] < 0 && !$::mallocFailed] |
+ expr {$mf==0 || $::text8_2 == "" || ($::text16 == "" && $::text8 != "")} |
+ } {1} |
+ |
+ # Test for malloc() failures within _int(), _int64() and _real(). The only |
+ # way this can occur is if the string has to be translated from UTF-16 to |
+ # UTF-8 before being converted to a numeric value. |
+ do_test ${testid}.4.4.1 { |
+ set mf [sqlite3_memdebug_pending] |
+ sqlite3_memdebug_fail -1 |
+ sqlite3_column_text16 $::STMT 0 |
+ sqlite3_memdebug_fail $mf |
+ sqlite3_column_int $::STMT 0 |
+ } {0} |
+ do_test ${testid}.4.5 { |
+ set mf [sqlite3_memdebug_pending] |
+ sqlite3_memdebug_fail -1 |
+ sqlite3_column_text16 $::STMT 0 |
+ sqlite3_memdebug_fail $mf |
+ sqlite3_column_int64 $::STMT 0 |
+ } {0} |
+ |
+ do_test ${testid}.4.6 { |
+ set mf [sqlite3_memdebug_pending] |
+ sqlite3_memdebug_fail -1 |
+ sqlite3_column_text16 $::STMT 0 |
+ sqlite3_memdebug_fail $mf |
+ sqlite3_column_double $::STMT 0 |
+ } {0.0} |
+ |
+ set mallocFailedAfterStep [expr \ |
+ [sqlite3_memdebug_pending] < 0 && !$::mallocFailed |
+ ] |
+ |
+ sqlite3_memdebug_fail -1 |
+ # Test that if a malloc() failed the next call to sqlite3_step() returns |
+ # SQLITE_ERROR. If malloc() did not fail, it should return SQLITE_DONE. |
+ # |
+ do_test ${testid}.5 { |
+ sqlite3_step $::STMT |
+ } [expr {$mallocFailedAfterStep ? "SQLITE_ERROR" : "SQLITE_DONE"}] |
+ |
+ do_test ${testid}.6 { |
+ sqlite3_finalize $::STMT |
+ } [expr {$mallocFailedAfterStep ? "SQLITE_NOMEM" : "SQLITE_OK"}] |
+ |
+ if {$::mallocFailed == 0 && $mallocFailedAfterStep == 0} { |
+ sqlite3_memdebug_fail -1 |
+ set go 0 |
+ } |
+ } |
+} |
+ |
+execsql { |
+ CREATE TABLE tbl( |
+ the_first_reasonably_long_column_name that_also_has_quite_a_lengthy_type |
+ ); |
+ INSERT INTO tbl VALUES( |
+ 'An extra long string. Far too long to be stored in NBFS bytes.' |
+ ); |
+} |
+ |
+do_stmt_test 1 "SELECT * FROM tbl" |
+ |
+sqlite3_memdebug_fail -1 |
+finish_test |