mirror of
				https://github.com/sqlite/sqlite.git
				synced 2025-11-03 16:53:36 +03:00 
			
		
		
		
	
		
			
				
	
	
		
			105 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			Tcl
		
	
	
	
	
	
			
		
		
	
	
			105 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			Tcl
		
	
	
	
	
	
 | 
						|
# Usage: do_malloc_test <test number> <options...>
 | 
						|
#
 | 
						|
# The first argument, <test number>, is an integer used to name the
 | 
						|
# tests executed by this proc. Options are as follows:
 | 
						|
#
 | 
						|
#     -tclprep          TCL script to run to prepare test.
 | 
						|
#     -sqlprep          SQL script to run to prepare test.
 | 
						|
#     -tclbody          TCL script to run with malloc failure simulation.
 | 
						|
#     -sqlbody          TCL script to run with malloc failure simulation.
 | 
						|
#     -cleanup          TCL script to run after the test.
 | 
						|
#
 | 
						|
# This command runs a series of tests to verify SQLite's ability
 | 
						|
# to handle an out-of-memory condition gracefully. It is assumed
 | 
						|
# that if this condition occurs a malloc() call will return a
 | 
						|
# NULL pointer. Linux, for example, doesn't do that by default. See
 | 
						|
# the "BUGS" section of malloc(3).
 | 
						|
#
 | 
						|
# Each iteration of a loop, the TCL commands in any argument passed
 | 
						|
# to the -tclbody switch, followed by the SQL commands in any argument
 | 
						|
# passed to the -sqlbody switch are executed. Each iteration the
 | 
						|
# Nth call to sqliteMalloc() is made to fail, where N is increased
 | 
						|
# each time the loop runs starting from 1. When all commands execute
 | 
						|
# successfully, the loop ends.
 | 
						|
#
 | 
						|
proc do_malloc_test {tn args} {
 | 
						|
  array unset ::mallocopts 
 | 
						|
  array set ::mallocopts $args
 | 
						|
 | 
						|
  if {[string is integer $tn]} {
 | 
						|
    set tn malloc-$tn
 | 
						|
  }
 | 
						|
  if {[info exists ::mallocopts(-start)]} {
 | 
						|
    set start $::mallocopts(-start)
 | 
						|
  } else {
 | 
						|
    set start 1
 | 
						|
  }
 | 
						|
 | 
						|
  set ::go 1
 | 
						|
  for {set ::n $start} {$::go && $::n < 50000} {incr ::n} {
 | 
						|
    do_test $tn.$::n {
 | 
						|
 | 
						|
      # Remove all traces of database files test.db and test2.db from the files
 | 
						|
      # system. Then open (empty database) "test.db" with the handle [db].
 | 
						|
      # 
 | 
						|
      catch {db close} 
 | 
						|
      catch {file delete -force test.db}
 | 
						|
      catch {file delete -force test.db-journal}
 | 
						|
      catch {file delete -force test2.db}
 | 
						|
      catch {file delete -force test2.db-journal}
 | 
						|
      if {[info exists ::mallocopts(-testdb)]} {
 | 
						|
        file copy $::mallocopts(-testdb) test.db
 | 
						|
      }
 | 
						|
      catch {sqlite3 db test.db} 
 | 
						|
 | 
						|
      # Execute any -tclprep and -sqlprep scripts.
 | 
						|
      #
 | 
						|
      if {[info exists ::mallocopts(-tclprep)]} {
 | 
						|
        eval $::mallocopts(-tclprep)
 | 
						|
      }
 | 
						|
      if {[info exists ::mallocopts(-sqlprep)]} {
 | 
						|
        execsql $::mallocopts(-sqlprep)
 | 
						|
      }
 | 
						|
 | 
						|
      # Now set the ${::n}th malloc() to fail and execute the -tclbody and
 | 
						|
      # -sqlbody scripts.
 | 
						|
      #
 | 
						|
      sqlite3_memdebug_fail $::n 1
 | 
						|
      set ::mallocbody {}
 | 
						|
      if {[info exists ::mallocopts(-tclbody)]} {
 | 
						|
        append ::mallocbody "$::mallocopts(-tclbody)\n"
 | 
						|
      }
 | 
						|
      if {[info exists ::mallocopts(-sqlbody)]} {
 | 
						|
        append ::mallocbody "db eval {$::mallocopts(-sqlbody)}"
 | 
						|
      }
 | 
						|
      set v [catch $::mallocbody msg]
 | 
						|
      set failFlag [sqlite3_memdebug_fail -1 0]
 | 
						|
      set go [expr {$failFlag>0}]
 | 
						|
 | 
						|
 | 
						|
      if {$failFlag==0} {
 | 
						|
        if {$v} {
 | 
						|
          set v2 $msg
 | 
						|
        } else {
 | 
						|
          set v 1
 | 
						|
          set v2 1
 | 
						|
        }
 | 
						|
      } elseif {!$v} {
 | 
						|
        set v2 $msg
 | 
						|
      } elseif {[info command db]=="" || [db errorcode]==7
 | 
						|
                   || $msg=="out of memory"} {
 | 
						|
        set v2 1
 | 
						|
      } else {
 | 
						|
        set v2 $msg
 | 
						|
      }
 | 
						|
      lappend v $v2
 | 
						|
    } {1 1}
 | 
						|
 | 
						|
    if {[info exists ::mallocopts(-cleanup)]} {
 | 
						|
      catch [list uplevel #0 $::mallocopts(-cleanup)] msg
 | 
						|
    }
 | 
						|
  }
 | 
						|
  unset ::mallocopts
 | 
						|
}
 |