1 files changed, 94 insertions, 0 deletions
diff --git a/test/fuzz_malloc.test b/test/fuzz_malloc.test
new file mode 100644
index 0000000..1e31bab
--- /dev/null
+++ b/test/fuzz_malloc.test
@@ -0,0 +1,94 @@
+#
+# 2007 May 10
+#
+# 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 tests malloc failures in concert with fuzzy SQL generation.
+#
+# $Id: fuzz_malloc.test,v 1.10 2008/08/20 16:35:10 drh Exp $
+
+set testdir [file dirname $argv0]
+source $testdir/tester.tcl
+
+ifcapable !memdebug {
+  finish_test
+  return
+}
+
+source $testdir/malloc_common.tcl
+source $testdir/fuzz_common.tcl
+
+if {[info exists G(isquick)]} {
+  set ::REPEATS 20
+} elseif {[info exists G(issoak)]} {
+  set ::REPEATS 100
+} else {
+  set ::REPEATS 40
+}
+
+#
+# Usage: do_fuzzy_malloc_test <testname> ?<options>?
+# 
+#     -template
+#     -sqlprep
+#     -repeats
+#     
+proc do_fuzzy_malloc_test {testname args} {
+  set ::fuzzyopts(-repeats) $::REPEATS
+  set ::fuzzyopts(-sqlprep) {}
+  array set ::fuzzyopts $args
+
+  sqlite3_memdebug_fail -1
+  db close
+  delete_file test.db test.db-journal
+  sqlite3 db test.db
+  set ::prep $::fuzzyopts(-sqlprep)
+  execsql $::prep
+  set jj 0
+  for {set ii 0} {$ii < $::fuzzyopts(-repeats)} {incr ii} {
+    expr srand($jj)
+    incr jj
+    set ::sql [subst $::fuzzyopts(-template)]
+    # puts fuzyy-sql=\[$::sql\]; flush stdout
+    foreach {rc res} [catchsql "$::sql"] {}
+    if {$rc==0} {
+      do_malloc_test $testname-$ii -sqlbody $::sql -sqlprep $::prep
+    } else {
+      incr ii -1
+    }
+  }
+}
+
+#----------------------------------------------------------------
+# Test malloc failure during parsing (and execution) of a fuzzily 
+# generated expressions.
+#
+do_fuzzy_malloc_test fuzzy_malloc-1 -template {Select [Expr]}
+do_fuzzy_malloc_test fuzzy_malloc-2 -template {[Select]}
+
+set ::SQLPREP {
+  BEGIN;
+    CREATE TABLE abc(a, b, c);
+    CREATE TABLE def(a, b, c);
+    CREATE TABLE ghi(a, b, c);
+    INSERT INTO abc VALUES(1.5, 3, 'a short string');
+    INSERT INTO def VALUES(NULL, X'ABCDEF', 
+        'a longer string. Long enough that it doesn''t fit in Mem.zShort');
+    INSERT INTO ghi VALUES(zeroblob(1000), 'hello world', -1257900987654321);
+  COMMIT;
+}
+set ::TableList  [list abc def ghi]
+set ::ColumnList [list a b c]
+
+do_fuzzy_malloc_test fuzzy_malloc-3 \
+  -template {[Select]}              \
+  -sqlprep $::SQLPREP
+
+finish_test