OLD | NEW |
(Empty) | |
| 1 # 2011 December 1 |
| 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 |
| 13 set testdir [file dirname $argv0] |
| 14 source $testdir/tester.tcl |
| 15 |
| 16 # If SQLITE_CURDIR is not defined, omit this file. |
| 17 ifcapable !curdir { |
| 18 finish_test |
| 19 return |
| 20 } |
| 21 |
| 22 source $testdir/malloc_common.tcl |
| 23 |
| 24 db close |
| 25 sqlite3_quota_initialize "" 1 |
| 26 |
| 27 foreach dir {quota2a/x1 quota2a/x2 quota2a quota2b quota2c} { |
| 28 forcedelete $dir |
| 29 } |
| 30 foreach dir {quota2a quota2a/x1 quota2a/x2 quota2b quota2c} { |
| 31 file mkdir $dir |
| 32 } |
| 33 |
| 34 # The standard_path procedure converts a pathname into a standard format |
| 35 # that is the same across platforms. |
| 36 # |
| 37 unset -nocomplain ::quota_pwd ::quota_mapping |
| 38 set ::quota_pwd [string map {\\ /} [get_pwd]] |
| 39 set ::quota_mapping [list $::quota_pwd PWD] |
| 40 proc standard_path {x} { |
| 41 set x [string map {\\ /} $x] |
| 42 return [string map $::quota_mapping $x] |
| 43 } |
| 44 |
| 45 # The quota_check procedure is a callback from the quota handler. |
| 46 # It has three arguments which are (1) the full pathname of the file |
| 47 # that has gone over quota, (2) the quota limit, (3) the requested |
| 48 # new quota size to cover the last write. These three values are |
| 49 # appended to the global variable $::quota. The filename is processed |
| 50 # to convert every \ character into / and to change the name of the |
| 51 # working directory to PWD. |
| 52 # |
| 53 # The quota is increased to the request if the ::quota_request_ok |
| 54 # global variable is true. |
| 55 # |
| 56 set ::quota {} |
| 57 set ::quota_request_ok 0 |
| 58 |
| 59 proc quota_check {filename limitvar size} { |
| 60 upvar $limitvar limit |
| 61 lappend ::quota [standard_path $filename] [set limit] $size |
| 62 if {$::quota_request_ok} {set limit $size} |
| 63 } |
| 64 |
| 65 sqlite3_quota_set */quota2a/* 4000 quota_check |
| 66 sqlite3_quota_set */quota2b/* 5000 quota_check |
| 67 |
| 68 unset -nocomplain bigtext |
| 69 for {set i 1} {$i<=1000} {incr i} { |
| 70 if {$i%10==0} { |
| 71 append bigtext [format "%06d\n" $i] |
| 72 } else { |
| 73 append bigtext [format "%06d " $i] |
| 74 } |
| 75 } |
| 76 |
| 77 catch { unset h1 } |
| 78 catch { unset x } |
| 79 do_test quota2-1.1 { |
| 80 set ::h1 [sqlite3_quota_fopen quota2a/xyz.txt w+b] |
| 81 sqlite3_quota_fwrite $::h1 1 7000 $bigtext |
| 82 } {4000} |
| 83 do_test quota2-1.2 { |
| 84 set ::quota |
| 85 } {PWD/quota2a/xyz.txt 4000 7000} |
| 86 do_test quota2-1.2.1 { |
| 87 sqlite3_quota_file_size $::h1 |
| 88 } {4000} |
| 89 do_test quota2-1.2.2 { |
| 90 sqlite3_quota_fflush $::h1 1 |
| 91 sqlite3_quota_file_truesize $::h1 |
| 92 } {4000} |
| 93 do_test quota2-1.3 { |
| 94 sqlite3_quota_rewind $::h1 |
| 95 set ::x [sqlite3_quota_fread $::h1 1001 7] |
| 96 string length $::x |
| 97 } {3003} |
| 98 do_test quota2-1.4 { |
| 99 string match $::x [string range $::bigtext 0 3002] |
| 100 } {1} |
| 101 do_test quota2-1.5 { |
| 102 sqlite3_quota_fseek $::h1 0 SEEK_END |
| 103 sqlite3_quota_ftell $::h1 |
| 104 } {4000} |
| 105 do_test quota2-1.6 { |
| 106 sqlite3_quota_fseek $::h1 -100 SEEK_END |
| 107 sqlite3_quota_ftell $::h1 |
| 108 } {3900} |
| 109 do_test quota2-1.7 { |
| 110 sqlite3_quota_fseek $::h1 -100 SEEK_CUR |
| 111 sqlite3_quota_ftell $::h1 |
| 112 } {3800} |
| 113 do_test quota2-1.8 { |
| 114 sqlite3_quota_fseek $::h1 50 SEEK_CUR |
| 115 sqlite3_quota_ftell $::h1 |
| 116 } {3850} |
| 117 do_test quota2-1.9 { |
| 118 sqlite3_quota_fseek $::h1 50 SEEK_SET |
| 119 sqlite3_quota_ftell $::h1 |
| 120 } {50} |
| 121 do_test quota2-1.10 { |
| 122 sqlite3_quota_rewind $::h1 |
| 123 sqlite3_quota_ftell $::h1 |
| 124 } {0} |
| 125 do_test quota2-1.11 { |
| 126 standard_path [sqlite3_quota_dump] |
| 127 } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 4000 {PWD/quota2a/xyz.txt 4000 1 0}}} |
| 128 do_test quota2-1.12 { |
| 129 sqlite3_quota_ftruncate $::h1 3500 |
| 130 sqlite3_quota_file_size $::h1 |
| 131 } {3500} |
| 132 do_test quota2-1.13 { |
| 133 sqlite3_quota_file_truesize $::h1 |
| 134 } {3500} |
| 135 do_test quota2-1.14 { |
| 136 standard_path [sqlite3_quota_dump] |
| 137 } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 3500 {PWD/quota2a/xyz.txt 3500 1 0}}} |
| 138 do_test quota2-1.15 { |
| 139 sqlite3_quota_fseek $::h1 0 SEEK_END |
| 140 sqlite3_quota_ftell $::h1 |
| 141 } {3500} |
| 142 do_test quota2-1.16 { |
| 143 sqlite3_quota_fwrite $::h1 1 7000 $bigtext |
| 144 } {500} |
| 145 do_test quota2-1.17 { |
| 146 sqlite3_quota_ftell $::h1 |
| 147 } {4000} |
| 148 do_test quota2-1.18 { |
| 149 sqlite3_quota_file_size $::h1 |
| 150 } {4000} |
| 151 do_test quota2-1.19 { |
| 152 sqlite3_quota_fflush $::h1 1 |
| 153 sqlite3_quota_file_truesize $::h1 |
| 154 } {4000} |
| 155 do_test quota2-1.20 { |
| 156 sqlite3_quota_fclose $::h1 |
| 157 standard_path [sqlite3_quota_dump] |
| 158 } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 4000 {PWD/quota2a/xyz.txt 4000 0 0}}} |
| 159 do_test quota2-1.21 { |
| 160 sqlite3_quota_remove quota2a/xyz.txt |
| 161 standard_path [sqlite3_quota_dump] |
| 162 } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 0}} |
| 163 |
| 164 |
| 165 |
| 166 set quota {} |
| 167 do_test quota2-2.1 { |
| 168 set ::h1 [sqlite3_quota_fopen quota2c/xyz.txt w+b] |
| 169 sqlite3_quota_fwrite $::h1 1 7000 $bigtext |
| 170 } {7000} |
| 171 do_test quota2-2.2 { |
| 172 set ::quota |
| 173 } {} |
| 174 do_test quota2-2.3.1 { |
| 175 sqlite3_quota_rewind $::h1 |
| 176 sqlite3_quota_file_available $::h1 |
| 177 } {7000} |
| 178 do_test quota2-2.3.2 { |
| 179 set ::x [sqlite3_quota_fread $::h1 1001 7] |
| 180 string length $::x |
| 181 } {6006} |
| 182 do_test quota2-2.3.3 { |
| 183 sqlite3_quota_file_available $::h1 |
| 184 } {0} |
| 185 do_test quota2-2.4 { |
| 186 string match $::x [string range $::bigtext 0 6005] |
| 187 } {1} |
| 188 do_test quota2-2.5 { |
| 189 sqlite3_quota_fseek $::h1 0 SEEK_END |
| 190 sqlite3_quota_ftell $::h1 |
| 191 } {7000} |
| 192 do_test quota2-2.6 { |
| 193 sqlite3_quota_fseek $::h1 -100 SEEK_END |
| 194 sqlite3_quota_ftell $::h1 |
| 195 } {6900} |
| 196 do_test quota2-2.6.1 { |
| 197 sqlite3_quota_file_available $::h1 |
| 198 } {100} |
| 199 do_test quota2-2.7 { |
| 200 sqlite3_quota_fseek $::h1 -100 SEEK_CUR |
| 201 sqlite3_quota_ftell $::h1 |
| 202 } {6800} |
| 203 do_test quota2-2.7.1 { |
| 204 sqlite3_quota_file_available $::h1 |
| 205 } {200} |
| 206 do_test quota2-2.8 { |
| 207 sqlite3_quota_fseek $::h1 50 SEEK_CUR |
| 208 sqlite3_quota_ftell $::h1 |
| 209 } {6850} |
| 210 do_test quota2-2.8.1 { |
| 211 sqlite3_quota_file_available $::h1 |
| 212 } {150} |
| 213 do_test quota2-2.9 { |
| 214 sqlite3_quota_fseek $::h1 50 SEEK_SET |
| 215 sqlite3_quota_ftell $::h1 |
| 216 } {50} |
| 217 do_test quota2-2.9.1 { |
| 218 sqlite3_quota_file_available $::h1 |
| 219 } {6950} |
| 220 do_test quota2-2.10 { |
| 221 sqlite3_quota_rewind $::h1 |
| 222 sqlite3_quota_ftell $::h1 |
| 223 } {0} |
| 224 do_test quota2-2.10.1 { |
| 225 sqlite3_quota_file_available $::h1 |
| 226 } {7000} |
| 227 do_test quota2-2.10.2 { |
| 228 sqlite3_quota_ferror $::h1 |
| 229 } {0} |
| 230 do_test quota2-2.11 { |
| 231 standard_path [sqlite3_quota_dump] |
| 232 } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 0}} |
| 233 do_test quota2-2.12 { |
| 234 sqlite3_quota_fclose $::h1 |
| 235 standard_path [sqlite3_quota_dump] |
| 236 } {{*/quota2b/* 5000 0} {*/quota2a/* 4000 0}} |
| 237 |
| 238 do_test quota2-3.1 { |
| 239 sqlite3_quota_set */quota2b/* 0 quota_check |
| 240 set ::h1 [sqlite3_quota_fopen quota2a/x1/a.txt a] |
| 241 sqlite3_quota_fwrite $::h1 10 10 $bigtext |
| 242 } {10} |
| 243 do_test quota2-3.2 { |
| 244 standard_path [sqlite3_quota_dump] |
| 245 } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 1 0}}} |
| 246 do_test quota2-3.3a { |
| 247 sqlite3_quota_fflush $::h1 0 |
| 248 standard_path [sqlite3_quota_dump] |
| 249 } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 1 0}}} |
| 250 do_test quota2-3.3b { |
| 251 sqlite3_quota_fflush $::h1 1 |
| 252 standard_path [sqlite3_quota_dump] |
| 253 } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 1 0}}} |
| 254 do_test quota2-3.3c { |
| 255 sqlite3_quota_fflush $::h1 |
| 256 standard_path [sqlite3_quota_dump] |
| 257 } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 1 0}}} |
| 258 do_test quota2-3.4 { |
| 259 sqlite3_quota_fclose $::h1 |
| 260 standard_path [sqlite3_quota_dump] |
| 261 } {{*/quota2a/* 4000 100 {PWD/quota2a/x1/a.txt 100 0 0}}} |
| 262 do_test quota2-3.5 { |
| 263 set ::h2 [sqlite3_quota_fopen quota2a/x2/b.txt a] |
| 264 sqlite3_quota_fwrite $::h2 10 20 $bigtext |
| 265 standard_path [sqlite3_quota_dump] |
| 266 } {{*/quota2a/* 4000 300 {PWD/quota2a/x2/b.txt 200 1 0} {PWD/quota2a/x1/a.txt 10
0 0 0}}} |
| 267 do_test quota2-3.6 { |
| 268 set ::h3 [sqlite3_quota_fopen quota2a/x1/c.txt a] |
| 269 sqlite3_quota_fwrite $::h3 10 50 $bigtext |
| 270 standard_path [sqlite3_quota_dump] |
| 271 } {{*/quota2a/* 4000 800 {PWD/quota2a/x1/c.txt 500 1 0} {PWD/quota2a/x2/b.txt 20
0 1 0} {PWD/quota2a/x1/a.txt 100 0 0}}} |
| 272 do_test quota2-3.7 { |
| 273 file exists quota2a/x1/a.txt |
| 274 } {1} |
| 275 do_test quota2-3.8 { |
| 276 file exists quota2a/x2/b.txt |
| 277 } {1} |
| 278 do_test quota2-3.9 { |
| 279 file exists quota2a/x1/c.txt |
| 280 } {1} |
| 281 do_test quota2-3.10 { |
| 282 sqlite3_quota_remove quota2a/x1 |
| 283 standard_path [sqlite3_quota_dump] |
| 284 } {{*/quota2a/* 4000 700 {PWD/quota2a/x1/c.txt 500 1 1} {PWD/quota2a/x2/b.txt 20
0 1 0}}} |
| 285 do_test quota2-3.11 { |
| 286 sqlite3_quota_fclose $::h2 |
| 287 sqlite3_quota_fclose $::h3 |
| 288 standard_path [sqlite3_quota_dump] |
| 289 } {{*/quota2a/* 4000 200 {PWD/quota2a/x2/b.txt 200 0 0}}} |
| 290 do_test quota2-3.12 { |
| 291 file exists quota2a/x1/a.txt |
| 292 } {0} |
| 293 do_test quota2-3.13 { |
| 294 file exists quota2a/x2/b.txt |
| 295 } {1} |
| 296 do_test quota2-3.14 { |
| 297 file exists quota2a/x1/c.txt |
| 298 } {0} |
| 299 |
| 300 catch { sqlite3_quota_shutdown } |
| 301 catch { unset quota_request_ok } |
| 302 finish_test |
OLD | NEW |