303 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
			
		
		
	
	
			303 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
| # 2011 December 1
 | |
| #
 | |
| # 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.
 | |
| #
 | |
| #***********************************************************************
 | |
| #
 | |
| 
 | |
| set testdir [file dirname $argv0]
 | |
| source $testdir/tester.tcl
 | |
| 
 | |
| # If SQLITE_CURDIR is not defined, omit this file.
 | |
| ifcapable !curdir {
 | |
|   finish_test
 | |
|   return
 | |
| }
 | |
| 
 | |
| source $testdir/malloc_common.tcl
 | |
| 
 | |
| db close
 | |
| sqlite3_quota_initialize "" 1
 | |
| 
 | |
| foreach dir {quota2a/x1 quota2a/x2 quota2a quota2b quota2c} {
 | |
|   forcedelete $dir
 | |
| }
 | |
| foreach dir {quota2a quota2a/x1 quota2a/x2 quota2b quota2c} {
 | |
|   file mkdir $dir
 | |
| }
 | |
| 
 | |
| # The standard_path procedure converts a pathname into a standard format
 | |
| # that is the same across platforms.
 | |
| #
 | |
| unset -nocomplain ::quota_pwd ::quota_mapping
 | |
| set ::quota_pwd [string map {\\ /} [get_pwd]]
 | |
| set ::quota_mapping [list $::quota_pwd PWD]
 | |
| proc standard_path {x} {
 | |
|   set x [string map {\\ /} $x]
 | |
|   return [string map $::quota_mapping $x]
 | |
| }
 | |
| 
 | |
| # The quota_check procedure is a callback from the quota handler.
 | |
| # It has three arguments which are (1) the full pathname of the file
 | |
| # that has gone over quota, (2) the quota limit, (3) the requested
 | |
| # new quota size to cover the last write.  These three values are
 | |
| # appended to the global variable $::quota.  The filename is processed
 | |
| # to convert every \ character into / and to change the name of the
 | |
| # working directory to PWD.  
 | |
| #
 | |
| # The quota is increased to the request if the ::quota_request_ok 
 | |
| # global variable is true.
 | |
| #
 | |
| set ::quota {}
 | |
| set ::quota_request_ok 0
 | |
| 
 | |
| proc quota_check {filename limitvar size} {
 | |
|   upvar $limitvar limit
 | |
|   lappend ::quota [standard_path $filename] [set limit] $size
 | |
|   if {$::quota_request_ok} {set limit $size}
 | |
| }
 | |
| 
 | |
| sqlite3_quota_set */quota2a/* 4000 quota_check
 | |
| sqlite3_quota_set */quota2b/* 5000 quota_check
 | |
| 
 | |
| unset -nocomplain bigtext
 | |
| for {set i 1} {$i<=1000} {incr i} {
 | |
|   if {$i%10==0} {
 | |
|     append bigtext [format "%06d\n" $i]
 | |
|   } else {
 | |
|     append bigtext [format "%06d " $i]
 | |
|   }
 | |
| }
 | |
| 
 | |
| catch { unset h1 }
 | |
| catch { unset x }
 | |
| do_test quota2-1.1 {
 | |
|   set ::h1 [sqlite3_quota_fopen quota2a/xyz.txt w+b]
 | |
|   sqlite3_quota_fwrite $::h1 1 7000 $bigtext
 | |
| } {4000}
 | |
| do_test quota2-1.2 {
 | |
|   set ::quota
 | |
| } {PWD/quota2a/xyz.txt 4000 7000}
 | |
| do_test quota2-1.2.1 {
 | |
|   sqlite3_quota_file_size $::h1
 | |
| } {4000}
 | |
| do_test quota2-1.2.2 {
 | |
|   sqlite3_quota_fflush $::h1 1
 | |
|   sqlite3_quota_file_truesize $::h1
 | |
| } {4000}
 | |
| do_test quota2-1.3 {
 | |
|   sqlite3_quota_rewind $::h1
 | |
|   set ::x [sqlite3_quota_fread $::h1 1001 7]
 | |
|   string length $::x
 | |
| } {3003}
 | |
| do_test quota2-1.4 {
 | |
|   string match $::x [string range $::bigtext 0 3002]
 | |
| } {1}
 | |
| do_test quota2-1.5 {
 | |
|   sqlite3_quota_fseek $::h1 0 SEEK_END
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {4000}
 | |
| do_test quota2-1.6 {
 | |
|   sqlite3_quota_fseek $::h1 -100 SEEK_END
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {3900}
 | |
| do_test quota2-1.7 {
 | |
|   sqlite3_quota_fseek $::h1 -100 SEEK_CUR
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {3800}
 | |
| do_test quota2-1.8 {
 | |
|   sqlite3_quota_fseek $::h1 50 SEEK_CUR
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {3850}
 | |
| do_test quota2-1.9 {
 | |
|   sqlite3_quota_fseek $::h1 50 SEEK_SET
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {50}
 | |
| do_test quota2-1.10 {
 | |
|   sqlite3_quota_rewind $::h1
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {0}
 | |
| do_test quota2-1.11 {
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 4000 {PWD/quota2a/xyz.txt 4000 1 0}}}
 | |
| do_test quota2-1.12 {
 | |
|   sqlite3_quota_ftruncate $::h1 3500
 | |
|   sqlite3_quota_file_size $::h1
 | |
| } {3500}
 | |
| do_test quota2-1.13 {
 | |
|   sqlite3_quota_file_truesize $::h1
 | |
| } {3500}
 | |
| do_test quota2-1.14 {
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 3500 {PWD/quota2a/xyz.txt 3500 1 0}}}
 | |
| do_test quota2-1.15 {
 | |
|   sqlite3_quota_fseek $::h1 0 SEEK_END
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {3500}
 | |
| do_test quota2-1.16 {
 | |
|   sqlite3_quota_fwrite $::h1 1 7000 $bigtext
 | |
| } {500}
 | |
| do_test quota2-1.17 {
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {4000}
 | |
| do_test quota2-1.18 {
 | |
|   sqlite3_quota_file_size $::h1
 | |
| } {4000}
 | |
| do_test quota2-1.19 {
 | |
|   sqlite3_quota_fflush $::h1 1
 | |
|   sqlite3_quota_file_truesize $::h1
 | |
| } {4000}
 | |
| do_test quota2-1.20 {
 | |
|   sqlite3_quota_fclose $::h1
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 4000 {PWD/quota2a/xyz.txt 4000 0 0}}}
 | |
| do_test quota2-1.21 {
 | |
|   sqlite3_quota_remove quota2a/xyz.txt
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 0}}
 | |
| 
 | |
| 
 | |
| 
 | |
| set quota {}
 | |
| do_test quota2-2.1 {
 | |
|   set ::h1 [sqlite3_quota_fopen quota2c/xyz.txt w+b]
 | |
|   sqlite3_quota_fwrite $::h1 1 7000 $bigtext
 | |
| } {7000}
 | |
| do_test quota2-2.2 {
 | |
|   set ::quota
 | |
| } {}
 | |
| do_test quota2-2.3.1 {
 | |
|   sqlite3_quota_rewind $::h1
 | |
|   sqlite3_quota_file_available $::h1
 | |
| } {7000}
 | |
| do_test quota2-2.3.2 {
 | |
|   set ::x [sqlite3_quota_fread $::h1 1001 7]
 | |
|   string length $::x
 | |
| } {6006}
 | |
| do_test quota2-2.3.3 {
 | |
|   sqlite3_quota_file_available $::h1
 | |
| } {0}
 | |
| do_test quota2-2.4 {
 | |
|   string match $::x [string range $::bigtext 0 6005]
 | |
| } {1}
 | |
| do_test quota2-2.5 {
 | |
|   sqlite3_quota_fseek $::h1 0 SEEK_END
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {7000}
 | |
| do_test quota2-2.6 {
 | |
|   sqlite3_quota_fseek $::h1 -100 SEEK_END
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {6900}
 | |
| do_test quota2-2.6.1 {
 | |
|   sqlite3_quota_file_available $::h1
 | |
| } {100}
 | |
| do_test quota2-2.7 {
 | |
|   sqlite3_quota_fseek $::h1 -100 SEEK_CUR
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {6800}
 | |
| do_test quota2-2.7.1 {
 | |
|   sqlite3_quota_file_available $::h1
 | |
| } {200}
 | |
| do_test quota2-2.8 {
 | |
|   sqlite3_quota_fseek $::h1 50 SEEK_CUR
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {6850}
 | |
| do_test quota2-2.8.1 {
 | |
|   sqlite3_quota_file_available $::h1
 | |
| } {150}
 | |
| do_test quota2-2.9 {
 | |
|   sqlite3_quota_fseek $::h1 50 SEEK_SET
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {50}
 | |
| do_test quota2-2.9.1 {
 | |
|   sqlite3_quota_file_available $::h1
 | |
| } {6950}
 | |
| do_test quota2-2.10 {
 | |
|   sqlite3_quota_rewind $::h1
 | |
|   sqlite3_quota_ftell $::h1
 | |
| } {0}
 | |
| do_test quota2-2.10.1 {
 | |
|   sqlite3_quota_file_available $::h1
 | |
| } {7000}
 | |
| do_test quota2-2.10.2 {
 | |
|   sqlite3_quota_ferror $::h1
 | |
| } {0}
 | |
| do_test quota2-2.11 {
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 0}}
 | |
| do_test quota2-2.12 {
 | |
|   sqlite3_quota_fclose $::h1
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 0}}
 | |
| 
 | |
| do_test quota2-3.1 {
 | |
|   sqlite3_quota_set */quota2b/* 0 quota_check
 | |
|   set ::h1 [sqlite3_quota_fopen quota2a/x1/a.txt a]
 | |
|   sqlite3_quota_fwrite $::h1 10 10 $bigtext
 | |
| } {10}
 | |
| do_test quota2-3.2 {
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 1 0}}}
 | |
| do_test quota2-3.3a {
 | |
|   sqlite3_quota_fflush $::h1 0
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 1 0}}}
 | |
| do_test quota2-3.3b {
 | |
|   sqlite3_quota_fflush $::h1 1
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 1 0}}}
 | |
| do_test quota2-3.3c {
 | |
|   sqlite3_quota_fflush $::h1
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 1 0}}}
 | |
| do_test quota2-3.4 {
 | |
|   sqlite3_quota_fclose $::h1
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 0 0}}}
 | |
| do_test quota2-3.5 {
 | |
|   set ::h2 [sqlite3_quota_fopen quota2a/x2/b.txt a]
 | |
|   sqlite3_quota_fwrite $::h2 10 20 $bigtext
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2a/* 4000 300 {PWD/quota2a/x2/b.txt 200 1 0} {PWD/quota2a/x1/a.txt 100 0 0}}}
 | |
| do_test quota2-3.6 {
 | |
|   set ::h3 [sqlite3_quota_fopen quota2a/x1/c.txt a]
 | |
|   sqlite3_quota_fwrite $::h3 10 50 $bigtext
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2a/* 4000 800 {PWD/quota2a/x1/c.txt 500 1 0} {PWD/quota2a/x2/b.txt 200 1 0} {PWD/quota2a/x1/a.txt 100 0 0}}}
 | |
| do_test quota2-3.7 {
 | |
|   file exists quota2a/x1/a.txt
 | |
| } {1}
 | |
| do_test quota2-3.8 {
 | |
|   file exists quota2a/x2/b.txt
 | |
| } {1}
 | |
| do_test quota2-3.9 {
 | |
|   file exists quota2a/x1/c.txt
 | |
| } {1}
 | |
| do_test quota2-3.10 {
 | |
|   sqlite3_quota_remove quota2a/x1
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2a/* 4000 700 {PWD/quota2a/x1/c.txt 500 1 1} {PWD/quota2a/x2/b.txt 200 1 0}}}
 | |
| do_test quota2-3.11 {
 | |
|   sqlite3_quota_fclose $::h2
 | |
|   sqlite3_quota_fclose $::h3
 | |
|   standard_path [sqlite3_quota_dump]
 | |
| } {{*/quota2a/* 4000 200 {PWD/quota2a/x2/b.txt 200 0 0}}}
 | |
| do_test quota2-3.12 {
 | |
|   file exists quota2a/x1/a.txt
 | |
| } {0}
 | |
| do_test quota2-3.13 {
 | |
|   file exists quota2a/x2/b.txt
 | |
| } {1}
 | |
| do_test quota2-3.14 {
 | |
|   file exists quota2a/x1/c.txt
 | |
| } {0}
 | |
| 
 | |
| catch { sqlite3_quota_shutdown }
 | |
| catch { unset quota_request_ok }
 | |
| finish_test
 | 
