OLD | NEW |
| (Empty) |
1 # 2005 November 30 | |
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 contains tests to ensure that the library handles malloc() failures | |
13 # correctly. The emphasis in this file is on sqlite3_column_XXX() APIs. | |
14 # | |
15 # $Id: malloc4.test,v 1.10 2008/02/18 22:24:58 drh Exp $ | |
16 | |
17 #--------------------------------------------------------------------------- | |
18 # NOTES ON EXPECTED BEHAVIOUR | |
19 # | |
20 # [193] When a memory allocation failure occurs during sqlite3_column_name(), | |
21 # sqlite3_column_name16(), sqlite3_column_decltype(), or | |
22 # sqlite3_column_decltype16() the function shall return NULL. | |
23 # | |
24 #--------------------------------------------------------------------------- | |
25 | |
26 set testdir [file dirname $argv0] | |
27 source $testdir/tester.tcl | |
28 source $testdir/malloc_common.tcl | |
29 | |
30 # Only run these tests if memory debugging is turned on. | |
31 if {!$MEMDEBUG} { | |
32 puts "Skipping malloc4 tests: not compiled with -DSQLITE_MEMDEBUG..." | |
33 finish_test | |
34 return | |
35 } | |
36 | |
37 ifcapable !utf16 { | |
38 finish_test | |
39 return | |
40 } | |
41 | |
42 proc do_stmt_test {id sql} { | |
43 set ::sql $sql | |
44 set go 1 | |
45 for {set n 0} {$go} {incr n} { | |
46 set testid "malloc4-$id.$n" | |
47 | |
48 # Prepare the statement | |
49 do_test ${testid}.1 { | |
50 set ::STMT [sqlite3_prepare $::DB $sql -1 TAIL] | |
51 expr [string length $::STMT] > 0 | |
52 } {1} | |
53 | |
54 # Set the Nth malloc() to fail. | |
55 sqlite3_memdebug_fail $n -repeat 0 | |
56 | |
57 # Test malloc failure in the _name(), _name16(), decltype() and | |
58 # decltype16() APIs. Calls that occur after the malloc() failure should | |
59 # return NULL. No error is raised though. | |
60 # | |
61 # ${testid}.2.1 - Call _name() | |
62 # ${testid}.2.2 - Call _name16() | |
63 # ${testid}.2.3 - Call _name() | |
64 # ${testid}.2.4 - Check that the return values of the above three calls are | |
65 # consistent with each other and with the simulated | |
66 # malloc() failures. | |
67 # | |
68 # Because the code that implements the _decltype() and _decltype16() APIs | |
69 # is the same as the _name() and _name16() implementations, we don't worry | |
70 # about explicitly testing them. | |
71 # | |
72 do_test ${testid}.2.1 { | |
73 set mf1 [expr [sqlite3_memdebug_pending] < 0] | |
74 set ::name8 [sqlite3_column_name $::STMT 0] | |
75 set mf2 [expr [sqlite3_memdebug_pending] < 0] | |
76 expr {$mf1 == $mf2 || $::name8 == ""} | |
77 } {1} | |
78 do_test ${testid}.2.2 { | |
79 set mf1 [expr [sqlite3_memdebug_pending] < 0] | |
80 set ::name16 [sqlite3_column_name16 $::STMT 0] | |
81 set ::name16 [encoding convertfrom unicode $::name16] | |
82 set ::name16 [string range $::name16 0 end-1] | |
83 set mf2 [expr [sqlite3_memdebug_pending] < 0] | |
84 expr {$mf1 == $mf2 || $::name16 == ""} | |
85 } {1} | |
86 do_test ${testid}.2.3 { | |
87 set mf1 [expr [sqlite3_memdebug_pending] < 0] | |
88 set ::name8_2 [sqlite3_column_name $::STMT 0] | |
89 set mf2 [expr [sqlite3_memdebug_pending] < 0] | |
90 expr {$mf1 == $mf2 || $::name8_2 == ""} | |
91 } {1} | |
92 set ::mallocFailed [expr [sqlite3_memdebug_pending] < 0] | |
93 do_test ${testid}.2.4 { | |
94 expr { | |
95 $::name8 == $::name8_2 && $::name16 == $::name8 && !$::mallocFailed || | |
96 $::name8 == $::name8_2 && $::name16 == "" && $::mallocFailed || | |
97 $::name8 == $::name16 && $::name8_2 == "" && $::mallocFailed || | |
98 $::name8_2 == $::name16 && $::name8 == "" && $::mallocFailed | |
99 } | |
100 } {1} | |
101 | |
102 # Step the statement so that we can call _text() and _text16(). Before | |
103 # running sqlite3_step(), make sure that malloc() is not about to fail. | |
104 # Memory allocation failures that occur within sqlite3_step() are tested | |
105 # elsewhere. | |
106 set mf [sqlite3_memdebug_pending] | |
107 sqlite3_memdebug_fail -1 | |
108 do_test ${testid}.3 { | |
109 sqlite3_step $::STMT | |
110 } {SQLITE_ROW} | |
111 sqlite3_memdebug_fail $mf | |
112 | |
113 # Test for malloc() failures within _text() and _text16(). | |
114 # | |
115 do_test ${testid}.4.1 { | |
116 set ::text8 [sqlite3_column_text $::STMT 0] | |
117 set mf [expr [sqlite3_memdebug_pending] < 0 && !$::mallocFailed] | |
118 expr {$mf==0 || $::text8 == ""} | |
119 } {1} | |
120 do_test ${testid}.4.2 { | |
121 set ::text16 [sqlite3_column_text16 $::STMT 0] | |
122 set ::text16 [encoding convertfrom unicode $::text16] | |
123 set ::text16 [string range $::text16 0 end-1] | |
124 set mf [expr [sqlite3_memdebug_pending] < 0 && !$::mallocFailed] | |
125 expr {$mf==0 || $::text16 == ""} | |
126 } {1} | |
127 do_test ${testid}.4.3 { | |
128 set ::text8_2 [sqlite3_column_text $::STMT 0] | |
129 set mf [expr [sqlite3_memdebug_pending] < 0 && !$::mallocFailed] | |
130 expr {$mf==0 || $::text8_2 == "" || ($::text16 == "" && $::text8 != "")} | |
131 } {1} | |
132 | |
133 # Test for malloc() failures within _int(), _int64() and _real(). The only | |
134 # way this can occur is if the string has to be translated from UTF-16 to | |
135 # UTF-8 before being converted to a numeric value. | |
136 do_test ${testid}.4.4.1 { | |
137 set mf [sqlite3_memdebug_pending] | |
138 sqlite3_memdebug_fail -1 | |
139 sqlite3_column_text16 $::STMT 0 | |
140 sqlite3_memdebug_fail $mf | |
141 sqlite3_column_int $::STMT 0 | |
142 } {0} | |
143 do_test ${testid}.4.5 { | |
144 set mf [sqlite3_memdebug_pending] | |
145 sqlite3_memdebug_fail -1 | |
146 sqlite3_column_text16 $::STMT 0 | |
147 sqlite3_memdebug_fail $mf | |
148 sqlite3_column_int64 $::STMT 0 | |
149 } {0} | |
150 | |
151 do_test ${testid}.4.6 { | |
152 set mf [sqlite3_memdebug_pending] | |
153 sqlite3_memdebug_fail -1 | |
154 sqlite3_column_text16 $::STMT 0 | |
155 sqlite3_memdebug_fail $mf | |
156 sqlite3_column_double $::STMT 0 | |
157 } {0.0} | |
158 | |
159 set mallocFailedAfterStep [expr \ | |
160 [sqlite3_memdebug_pending] < 0 && !$::mallocFailed | |
161 ] | |
162 | |
163 sqlite3_memdebug_fail -1 | |
164 # Test that if a malloc() failed the next call to sqlite3_step() returns | |
165 # SQLITE_ERROR. If malloc() did not fail, it should return SQLITE_DONE. | |
166 # | |
167 do_test ${testid}.5 { | |
168 sqlite3_step $::STMT | |
169 } [expr {$mallocFailedAfterStep ? "SQLITE_ERROR" : "SQLITE_DONE"}] | |
170 | |
171 do_test ${testid}.6 { | |
172 sqlite3_finalize $::STMT | |
173 } [expr {$mallocFailedAfterStep ? "SQLITE_NOMEM" : "SQLITE_OK"}] | |
174 | |
175 if {$::mallocFailed == 0 && $mallocFailedAfterStep == 0} { | |
176 sqlite3_memdebug_fail -1 | |
177 set go 0 | |
178 } | |
179 } | |
180 } | |
181 | |
182 execsql { | |
183 CREATE TABLE tbl( | |
184 the_first_reasonably_long_column_name that_also_has_quite_a_lengthy_type | |
185 ); | |
186 INSERT INTO tbl VALUES( | |
187 'An extra long string. Far too long to be stored in NBFS bytes.' | |
188 ); | |
189 } | |
190 | |
191 do_stmt_test 1 "SELECT * FROM tbl" | |
192 | |
193 sqlite3_memdebug_fail -1 | |
194 finish_test | |
OLD | NEW |