Whamcloud - gitweb
b=14842
[fs/lustre-release.git] / lustre / tests / sanity-quota.sh
1 #!/bin/bash
2 #
3 # Run select tests by setting ONLY, or as arguments to the script.
4 # Skip specific tests by setting EXCEPT.
5 #
6 # Run test by setting NOSETUP=true when ltest has setup env for us
7 set -e
8
9 #kernel 2.4.x doesn't support quota
10 K_VER=`uname --kernel-release | cut -b 1-3`
11 if [ $K_VER = "2.4" ]; then
12     echo "Kernel 2.4 doesn't support quota"
13     exit 0
14 fi
15
16 SRCDIR=`dirname $0`
17 export PATH=$PWD/$SRCDIR:$SRCDIR:$PWD/$SRCDIR/../utils:$PATH:/sbin
18
19 ONLY=${ONLY:-"$*"}
20 ALWAYS_EXCEPT="$SANITY_QUOTA_EXCEPT"
21 # UPDATE THE COMMENT ABOVE WITH BUG NUMBERS WHEN CHANGING ALWAYS_EXCEPT!
22
23 case `uname -r` in
24 2.6*) FSTYPE=${FSTYPE:-ldiskfs};;
25 *) error "unsupported kernel" ;;
26 esac
27
28 [ "$ALWAYS_EXCEPT$EXCEPT" ] && \
29         echo "Skipping tests: `echo $ALWAYS_EXCEPT $EXCEPT`"
30
31 TMP=${TMP:-/tmp}
32
33 ORIG_PWD=${PWD}
34 TSTID=${TSTID:-60000}
35 TSTID2=${TSTID2:-60001}
36 TSTUSR=${TSTUSR:-"quota_usr"}
37 TSTUSR2=${TSTUSR2:-"quota_2usr"}
38 BLK_SZ=1024
39 BUNIT_SZ=${BUNIT_SZ:-1000}      # default 1000 quota blocks
40 BTUNE_SZ=${BTUNE_SZ:-500}       # default 50% of BUNIT_SZ
41 IUNIT_SZ=${IUNIT_SZ:-10}        # default 10 files
42 ITUNE_SZ=${ITUNE_SZ:-5}         # default 50% of IUNIT_SZ
43 MAX_DQ_TIME=604800
44 MAX_IQ_TIME=604800
45
46 TRACE=${TRACE:-""}
47 LUSTRE=${LUSTRE:-`dirname $0`/..}
48 . $LUSTRE/tests/test-framework.sh
49 init_test_env $@
50 . ${CONFIG:=$LUSTRE/tests/cfg/$NAME.sh}
51
52 [ "$SLOW" = "no" ] && EXCEPT_SLOW="9 10 11"
53
54 QUOTALOG=${TESTSUITELOG:-$TMP/$(basename $0 .sh).log}
55
56 [ "$QUOTALOG" ] && rm -f $QUOTALOG || true
57
58 DIR=${DIR:-$MOUNT}
59 DIR2=${DIR2:-$MOUNT2}
60
61 cleanup_and_setup_lustre
62
63 LOVNAME=`cat $LPROC/llite/*/lov/common_name | tail -n 1`
64 OSTCOUNT=`cat $LPROC/lov/$LOVNAME/numobd`
65
66 SHOW_QUOTA_USER="$LFS quota -u $TSTUSR $DIR"
67 SHOW_QUOTA_GROUP="$LFS quota -g $TSTUSR $DIR"
68 SHOW_QUOTA_INFO="$LFS quota -t $DIR"
69
70 # control the time of tests
71 cycle=30
72 [ "$SLOW" = "no" ] && cycle=10
73
74 build_test_filter
75
76 eval ONLY_0=true
77 eval ONLY_99=true
78
79 # set_blk_tunables(btune_sz)
80 set_blk_tunesz() {
81         # set btune size on all obdfilters
82         do_facet ost1 "set -x; for i in /proc/fs/lustre/obdfilter/*/quota_btune_sz; do
83                 echo $(($1 * BLK_SZ)) >> \\\$i;
84         done"
85         # set btune size on mds
86         do_facet $SINGLEMDS "for i in /proc/fs/lustre/mds/${FSNAME}-MDT*/quota_btune_sz; do
87                 echo $(($1 * BLK_SZ)) >> \\\$i;
88         done"
89 }
90
91 # set_blk_unitsz(bunit_sz)
92 set_blk_unitsz() {
93         do_facet ost1 "for i in /proc/fs/lustre/obdfilter/*/quota_bunit_sz; do
94                 echo $(($1 * BLK_SZ)) >> \\\$i;
95         done"
96         do_facet $SINGLEMDS "for i in /proc/fs/lustre/mds/${FSNAME}-MDT*/quota_bunit_sz; do
97                 echo $(($1 * BLK_SZ)) >> \\\$i;
98         done"
99 }
100
101 # set_file_tunesz(itune_sz)
102 set_file_tunesz() {
103         # set iunit and itune size on all obdfilters
104         do_facet ost1 "for i in /proc/fs/lustre/obdfilter/*/quota_itune_sz; do
105                 echo $1 >> \\\$i;
106         done"
107         # set iunit and itune size on mds
108         do_facet $SINGLEMDS "for i in /proc/fs/lustre/mds/${FSNAME}-MDT*/quota_itune_sz; do
109                 echo $1 >> \\\$i;
110         done"
111 }
112
113 # set_file_unitsz(iunit_sz)
114 set_file_unitsz() {
115         do_facet ost1 "for i in /proc/fs/lustre/obdfilter/*/quota_iunit_sz; do
116                 echo $1 >> \\\$i;
117         done"
118         do_facet $SINGLEMDS "for i in /proc/fs/lustre/mds/${FSNAME}-MDT*/quota_iunit_sz; do
119                 echo $1 >> \\\$i;
120         done"
121 }
122
123 # These are for test on local machine,if run sanity-quota.sh on 
124 # real cluster, ltest should have setup the test environment: 
125 #
126 # - create test user/group on all servers with same id.
127 # - set unit size/tune on all servers size to reasonable value.
128 pre_test() {
129         if [ -z "$NOSETUP" ]; then
130                 # set block tunables
131                 set_blk_tunesz $BTUNE_SZ
132                 set_blk_unitsz $BUNIT_SZ
133                 # set file tunables
134                 set_file_tunesz $ITUNE_SZ
135                 set_file_unitsz $IUNIT_SZ
136         fi
137 }
138 pre_test
139
140 post_test() {
141         if [ -z "$NOSETUP" ]; then
142                 # restore block tunables to default size
143                 set_blk_unitsz $((1024 * 100))
144                 set_blk_tunesz $((1024 * 50))
145                 # restore file tunables to default size
146                 set_file_unitsz 5000
147                 set_file_tunesz 2500
148         fi
149 }
150
151 RUNAS="runas -u $TSTID"
152 RUNAS2="runas -u $TSTID2"
153 FAIL_ON_ERROR=true check_runas_id $TSTID $RUNAS
154 FAIL_ON_ERROR=true check_runas_id $TSTID2 $RUNAS2
155
156 FAIL_ON_ERROR=false
157
158 # set quota
159 test_0() {
160         $LFS quotaoff -ug $DIR
161         $LFS quotacheck -ug $DIR
162
163         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR
164         $LFS setquota -g $TSTUSR 0 0 0 0 $DIR
165 }
166 run_test 0 "Set quota ============================="
167
168 # block hard limit (normal use and out of quota)
169 test_1() {
170         chmod 0777 $DIR/$tdir
171
172         LIMIT=$(( $BUNIT_SZ * $(($OSTCOUNT + 1)) * 5)) # 5 bunits each sever
173         TESTFILE=$DIR/$tdir/$tfile-0    
174         
175         echo "  User quota (limit: $LIMIT kbytes)"
176         $LFS setquota -u $TSTUSR 0 $LIMIT 0 0 $DIR
177         $SHOW_QUOTA_USER
178         
179         $LFS setstripe $TESTFILE -c 1
180         chown $TSTUSR.$TSTUSR $TESTFILE
181
182         echo "    Write ..."
183         $RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ count=$(($LIMIT/2)) || error "(usr) write failure, but expect success"
184         echo "    Done"
185         echo "    Write out of block quota ..."
186         # this time maybe cache write,  ignore it's failure
187         $RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ count=$(($LIMIT/2)) seek=$(($LIMIT/2)) || true
188         # flush cache, ensure noquota flag is setted on client
189         sync; sleep 1; sync;
190         $RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ count=$BUNIT_SZ seek=$LIMIT && error "(usr) write success, but expect EDQUOT"
191
192         rm -f $TESTFILE
193         
194         echo "  Group quota (limit: $LIMIT kbytes)"
195         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR           # clear user limit
196         $LFS setquota -g $TSTUSR 0 $LIMIT 0 0 $DIR
197         $SHOW_QUOTA_GROUP
198         TESTFILE=$DIR/$tdir/$tfile-1    
199
200         $LFS setstripe $TESTFILE -c 1
201         chown $TSTUSR.$TSTUSR $TESTFILE
202
203         echo "    Write ..."
204         $RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ count=$(($LIMIT/2)) || error "(grp) write failure, but expect success"
205         echo "    Done"
206         echo "    Write out of block quota ..."
207         # this time maybe cache write, ignore it's failure
208         $RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ count=$(($LIMIT/2)) seek=$(($LIMIT/2)) || true
209         sync; sleep 1; sync;
210         $RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ count=$BUNIT_SZ seek=$LIMIT && error "(grp) write success, but expect EDQUOT"
211
212         # cleanup
213         rm -f $TESTFILE
214         $LFS setquota -g $TSTUSR 0 0 0 0 $DIR
215 }
216 run_test 1 "Block hard limit (normal use and out of quota) ==="
217
218 # file hard limit (normal use and out of quota)
219 test_2() {
220         chmod 0777 $DIR/$tdir
221
222         LIMIT=$(($IUNIT_SZ * 10)) # 10 iunits on mds
223         TESTFILE=$DIR/$tdir/$tfile-0    
224
225         echo "  User quota (limit: $LIMIT files)"
226         $LFS setquota -u $TSTUSR 0 0 0 $LIMIT $DIR
227         $SHOW_QUOTA_USER
228
229         echo "    Create $LIMIT files ..."
230         $RUNAS createmany -m ${TESTFILE} $LIMIT || \
231             error "(usr) create failure, but except success"
232         echo "    Done"
233         echo "    Create out of file quota ..."
234         $RUNAS touch ${TESTFILE}_xxx && \
235                 error "(usr) touch success, but expect EDQUOT"
236
237         unlinkmany ${TESTFILE} $LIMIT
238         rm ${TESTFILE}_xxx
239
240         echo "  Group quota (limit: $LIMIT files)"
241         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR           # clear user limit
242         $LFS setquota -g $TSTUSR 0 0 0 $LIMIT $DIR
243         $SHOW_QUOTA_GROUP
244         TESTFILE=$DIR/$tdir/$tfile-1
245
246         echo "    Create $LIMIT files ..."
247         $RUNAS createmany -m ${TESTFILE} $LIMIT || \
248                 error "(grp) create failure, but expect success"
249
250         echo "    Done"
251         echo "    Create out of file quota ..."
252         $RUNAS touch ${TESTFILE}_xxx && \
253                 error "(grp) touch success, but expect EDQUOT"
254
255         $RUNAS touch ${TESTFILE}_xxx > /dev/null 2>&1 && error "(grp) touch success, but expect EDQUOT"
256
257         # cleanup
258         unlinkmany ${TESTFILE} $LIMIT
259         rm ${TESTFILE}_xxx
260
261         $LFS setquota -g $TSTUSR 0 0 0 0 $DIR
262 }
263 run_test 2 "File hard limit (normal use and out of quota) ==="
264
265 test_block_soft() {
266         TESTFILE=$1
267         TIMER=$(($2 * 3 / 2))
268         OFFSET=0
269
270         echo "    Write to exceed soft limit"
271         RUNDD="$RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ"
272         $RUNDD count=$((BUNIT_SZ+1)) || \
273                 error "write failure, but expect success"
274         OFFSET=$((OFFSET + BUNIT_SZ + 1))
275         sync; sleep 1; sync;
276
277         $SHOW_QUOTA_USER
278         $SHOW_QUOTA_GROUP
279         $SHOW_QUOTA_INFO
280
281         echo "    Write before timer goes off"
282         $RUNDD count=$BUNIT_SZ seek=$OFFSET || \
283                 error "write failure, but expect success"
284         OFFSET=$((OFFSET + BUNIT_SZ))
285         sync; sleep 1; sync;
286         echo "    Done"
287         
288         echo "    Sleep $TIMER seconds ..."
289         sleep $TIMER
290
291         $SHOW_QUOTA_USER
292         $SHOW_QUOTA_GROUP
293         $SHOW_QUOTA_INFO
294
295         echo "    Write after timer goes off"
296         # maybe cache write, ignore.
297         sync; sleep 1; sync;
298         $RUNDD count=$BUNIT_SZ seek=$OFFSET || true
299         OFFSET=$((OFFSET + BUNIT_SZ))
300         sync; sleep 1; sync;
301         $RUNDD count=$BUNIT_SZ seek=$OFFSET && \
302                 error "write success, but expect EDQUOT"
303
304         $SHOW_QUOTA_USER
305         $SHOW_QUOTA_GROUP
306         $SHOW_QUOTA_INFO
307
308         echo "    Unlink file to stop timer"
309         rm -f $TESTFILE
310         echo "    Done"
311
312         $SHOW_QUOTA_USER
313         $SHOW_QUOTA_GROUP
314         $SHOW_QUOTA_INFO
315
316         echo "    Write ..."
317         $RUNDD count=$BUNIT_SZ || error "write failure, but expect success"
318         echo "    Done"
319
320         # cleanup
321         rm -f $TESTFILE
322 }
323
324 # block soft limit (start timer, timer goes off, stop timer)
325 test_3() {
326         chmod 0777 $DIR/$tdir
327
328         LIMIT=$(( $BUNIT_SZ * 2 )) # 1 bunit on mds and 1 bunit on the ost
329         GRACE=10
330
331         echo "  User quota (soft limit: $LIMIT kbytes  grace: $GRACE seconds)"
332         TESTFILE=$DIR/$tdir/$tfile-0
333
334         $LFS setstripe $TESTFILE -c 1
335         chown $TSTUSR.$TSTUSR $TESTFILE
336
337         $LFS setquota -t -u $GRACE $MAX_IQ_TIME $DIR
338         $LFS setquota -u $TSTUSR $LIMIT 0 0 0 $DIR
339
340         test_block_soft $TESTFILE $GRACE
341         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR
342
343         echo "  Group quota (soft limit: $LIMIT kbytes  grace: $GRACE seconds)"
344         TESTFILE=$DIR/$tdir/$tfile-1
345
346         $LFS setstripe $TESTFILE -c 1
347         chown $TSTUSR.$TSTUSR $TESTFILE
348
349         $LFS setquota -t -g $GRACE $MAX_IQ_TIME $DIR
350         $LFS setquota -g $TSTUSR $LIMIT 0 0 0 $DIR
351
352         test_block_soft $TESTFILE $GRACE
353         $LFS setquota -g $TSTUSR 0 0 0 0 $DIR
354 }
355 run_test 3 "Block soft limit (start timer, timer goes off, stop timer) ==="
356
357 test_file_soft() {
358         TESTFILE=$1
359         LIMIT=$2
360         TIMER=$(($3 * 3 / 2))
361
362         echo "    Create files to exceed soft limit"
363         $RUNAS createmany -m ${TESTFILE}_ $((LIMIT + 1)) || \
364                 error "create failure, but expect success"
365         sync; sleep 1; sync
366         echo "    Done"
367
368         echo "    Create file before timer goes off"
369         $RUNAS touch ${TESTFILE}_before || \
370                 error "failed create before timer expired, but expect success"
371         sync; sleep 1; sync
372         echo "    Done"
373
374         echo "    Sleep $TIMER seconds ..."
375         sleep $TIMER
376         
377         $SHOW_QUOTA_USER
378         $SHOW_QUOTA_GROUP
379         $SHOW_QUOTA_INFO
380         
381         echo "    Create file after timer goes off"
382         $RUNAS createmany -m ${TESTFILE}_after_ $((IUNIT_SZ - 2)) || \
383                 error "create ${TESTFILE}_after failure, but expect success"
384         sync; sleep 1; sync
385         $RUNAS touch ${TESTFILE}_after && \
386                 error "create after timer expired, but expect EDQUOT"
387         sync; sleep 1; sync
388
389         $SHOW_QUOTA_USER
390         $SHOW_QUOTA_GROUP
391         $SHOW_QUOTA_INFO
392         
393         echo "    Unlink files to stop timer"
394         find `dirname $TESTFILE` -name "`basename ${TESTFILE}`*" | xargs rm -f
395         echo "    Done"
396
397         echo "    Create file"
398         $RUNAS touch ${TESTFILE}_xxx || \
399                 error "touch after timer stop failure, but expect success"
400         sync; sleep 1; sync
401         echo "    Done"
402
403         # cleanup
404         rm -f ${TESTFILE}_xxx
405 }
406
407 # file soft limit (start timer, timer goes off, stop timer)
408 test_4() {
409         chmod 0777 $DIR/$tdir
410         LIMIT=$(($IUNIT_SZ * 10))       # 10 iunits on mds
411         TESTFILE=$DIR/$tdir/$tfile-0
412
413         GRACE=5
414
415         echo "  User quota (soft limit: $LIMIT files  grace: $GRACE seconds)"
416         $LFS setquota -t -u $MAX_DQ_TIME $GRACE $DIR
417         $LFS setquota -u $TSTUSR 0 0 $LIMIT 0 $DIR
418         $SHOW_QUOTA_USER
419
420         test_file_soft $TESTFILE $LIMIT $GRACE
421         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR
422
423         echo "  Group quota (soft limit: $LIMIT files  grace: $GRACE seconds)"
424         $LFS setquota -t -g $MAX_DQ_TIME $GRACE $DIR
425         $LFS setquota -g $TSTUSR 0 0 $LIMIT 0 $DIR
426         $SHOW_QUOTA_GROUP
427         TESTFILE=$DIR/$tdir/$tfile-1
428
429         test_file_soft $TESTFILE $LIMIT $GRACE
430         $LFS setquota -g $TSTUSR 0 0 0 0 $DIR
431
432         # cleanup
433         $LFS setquota -t -u $MAX_DQ_TIME $MAX_IQ_TIME $DIR
434         $LFS setquota -t -g $MAX_DQ_TIME $MAX_IQ_TIME $DIR
435 }
436 run_test 4 "File soft limit (start timer, timer goes off, stop timer) ==="
437
438 test_4a() {
439         GR_STR1="1w3d"
440         GR_STR2="1000s"
441         GR_STR3="5s"
442         GR_STR4="1w2d3h4m5s"
443         GR_STR5="5c"
444         GR_STR6="1111111111111111"
445
446         # test of valid grace strings handling
447         echo "  Valid grace strings test"
448         $LFS setquota -t -u $GR_STR1 $GR_STR2 $DIR
449         $LFS quota -u -t $DIR | grep "Block grace time: $GR_STR1"
450         $LFS setquota -t -g $GR_STR3 $GR_STR4 $DIR
451         $LFS quota -g -t $DIR | grep "Inode grace time: $GR_STR4"
452
453         # test of invalid grace strings handling
454         echo "  Invalid grace strings test"
455         ! $LFS setquota -t -u $GR_STR4 $GR_STR5 $DIR
456         ! $LFS setquota -t -g $GR_STR4 $GR_STR6 $DIR
457
458         # cleanup
459         $LFS setquota -t -u $MAX_DQ_TIME $MAX_IQ_TIME $DIR
460         $LFS setquota -t -g $MAX_DQ_TIME $MAX_IQ_TIME $DIR
461 }
462 run_test 4a "Grace time strings handling ==="
463
464 # chown & chgrp (chown & chgrp successfully even out of block/file quota)
465 test_5() {
466         BLIMIT=$(( $BUNIT_SZ * $((OSTCOUNT + 1)) * 10)) # 10 bunits on each server
467         ILIMIT=$(( $IUNIT_SZ * 10 )) # 10 iunits on mds
468         
469         echo "  Set quota limit (0 $BLIMIT 0 $ILIMIT) for $TSTUSR.$TSTUSR"
470         $LFS setquota -u $TSTUSR 0 $BLIMIT 0 $ILIMIT $DIR
471         $LFS setquota -g $TSTUSR 0 $BLIMIT 0 $ILIMIT $DIR
472         $SHOW_QUOTA_USER
473         $SHOW_QUOTA_GROUP
474         
475         echo "  Create more than $ILIMIT files and more than $BLIMIT kbytes ..."
476         createmany -m $DIR/$tdir/$tfile-0_ $((ILIMIT + 1)) || \
477                 error "touch failure, expect success"
478         dd if=/dev/zero of=$DIR/$tdir/$tfile-0_1 bs=$BLK_SZ count=$((BLIMIT+1)) || error "write failure, expect success"
479
480         echo "  Chown files to $TSTUSR.$TSTUSR ..."
481         for i in `seq 0 $ILIMIT`; do
482         chown $TSTUSR.$TSTUSR $DIR/$tdir/$tfile-0_$i || \
483                         error "chown failure, but expect success"
484         done
485
486         # cleanup
487         unlinkmany $DIR/$tdir/$tfile-0_ $((ILIMIT + 1))
488
489         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR
490         $LFS setquota -g $TSTUSR 0 0 0 0 $DIR
491 }
492 run_test 5 "Chown & chgrp successfully even out of block/file quota ==="
493
494 # block quota acquire & release
495 test_6() {
496         if [ $OSTCOUNT -lt 2 ]; then
497                 skip "$OSTCOUNT < 2, too few osts"
498                 return 0;
499         fi
500
501         chmod 0777 $DIR/$tdir
502
503         LIMIT=$((BUNIT_SZ * (OSTCOUNT + 1) * 5)) # 5 bunits per server
504         FILEA="$DIR/$tdir/$tfile-0_a"
505         FILEB="$DIR/$tdir/$tfile-0_b"
506         
507         echo "  Set block limit $LIMIT kbytes to $TSTUSR.$TSTUSR"
508         $LFS setquota -u $TSTUSR 0 $LIMIT 0 0 $DIR
509         $LFS setquota -g $TSTUSR 0 $LIMIT 0 0 $DIR
510         $SHOW_QUOTA_USER
511         $SHOW_QUOTA_GROUP
512
513         echo "  Create filea on OST0 and fileb on OST1"
514         $LFS setstripe $FILEA -i 0 -c 1
515         $LFS setstripe $FILEB -i 1 -c 1
516         chown $TSTUSR.$TSTUSR $FILEA
517         chown $TSTUSR.$TSTUSR $FILEB
518
519         echo "  Exceed quota limit ..."
520         RUNDD="$RUNAS dd if=/dev/zero of=$FILEB bs=$BLK_SZ"
521         $RUNDD count=$((LIMIT - BUNIT_SZ * OSTCOUNT)) || \
522                 error "write fileb failure, but expect success"
523
524         sync; sleep 1; sync;
525         $SHOW_QUOTA_USER
526         $SHOW_QUOTA_GROUP
527         $RUNDD seek=$LIMIT count=$((BUNIT_SZ * OSTCOUNT)) && \
528                 error "write fileb success, but expect EDQUOT"
529         sync; sleep 1; sync;
530         echo "  Write to OST0 return EDQUOT"
531         # this write maybe cache write, ignore it's failure
532         RUNDD="$RUNAS dd if=/dev/zero of=$FILEA bs=$BLK_SZ"
533         $RUNDD count=$(($BUNIT_SZ * 2)) || true
534         sync; sleep 1; sync;
535         $SHOW_QUOTA_USER
536         $SHOW_QUOTA_GROUP
537         $RUNDD count=$((BUNIT_SZ * 2)) seek=$((BUNIT_SZ *2)) && \
538                 error "write filea success, but expect EDQUOT"
539
540         echo "  Remove fileb to let OST1 release quota"
541         rm -f $FILEB
542         sync; sleep 10; sync; # need to allow journal commit for small fs
543
544         echo "  Write to OST0"
545         $RUNDD count=$((LIMIT - BUNIT_SZ * OSTCOUNT)) || \
546                 error "write filea failure, expect success"
547         echo "  Done"
548
549         # cleanup
550         rm -f $FILEA
551         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR
552         $LFS setquota -g $TSTUSR 0 0 0 0 $DIR
553         return 0
554 }
555 run_test 6 "Block quota acquire & release ========="
556
557 # quota recovery (block quota only by now)
558 test_7()
559 {
560         chmod 0777 $DIR/$tdir
561         remote_mds && skip "remote mds" && return 0
562
563         LIMIT=$(( $BUNIT_SZ * $(($OSTCOUNT + 1)) * 10)) # 10 bunits each sever
564         TESTFILE="$DIR/$tdir/$tfile-0"
565         
566         $LFS setquota -u $TSTUSR 0 $LIMIT 0 0 $DIR
567         
568         $LFS setstripe $TESTFILE -c 1
569         chown $TSTUSR.$TSTUSR $TESTFILE
570
571         echo "  Write to OST0..."
572         $RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ count=$BUNIT_SZ || \
573                 error "write failure, but expect success"
574         
575         #define OBD_FAIL_OBD_DQACQ               0x604
576         echo 0x604 > /proc/sys/lustre/fail_loc
577         echo "  Remove files on OST0"
578         rm -f $TESTFILE
579         echo 0 > /proc/sys/lustre/fail_loc
580
581         echo "  Trigger recovery..."
582         OSC0_UUID="`$LCTL dl | awk '$3 ~ /osc/ { print $1 }'`"
583         for i in $OSC0_UUID; do
584                 $LCTL --device $i activate || error "activate osc failed!"
585         done
586
587         # sleep a while to wait for recovery done
588         sleep 20
589
590         # check limits
591         PATTERN="`echo $DIR | sed 's/\//\\\\\//g'`"
592         TOTAL_LIMIT="`$LFS quota -u $TSTUSR $DIR | awk '/^.*'$PATTERN'.*[[:digit:]+][[:space:]+]/ { print $4 }'`"
593         [ $TOTAL_LIMIT -eq $LIMIT ] || error "total limits not recovery!"
594         echo "  total limits = $TOTAL_LIMIT"
595         
596         OST0_UUID=`do_facet ost1 "$LCTL dl | grep -m1 obdfilter" | awk '{print $((NF-1))}'`
597         [ -z "$OST0_UUID" ] && OST0_UUID=`do_facet ost1 "$LCTL dl | grep -m1 obdfilter" | awk '{print $((NF-1))}'`
598         OST0_LIMIT="`$LFS quota -o $OST0_UUID -u $TSTUSR $DIR | awk '/^.*[[:digit:]+][[:space:]+]/ { print $3 }'`"
599         [ $OST0_LIMIT -eq $BUNIT_SZ ] || error "high limits not released!"
600         echo "  limits on $OST0_UUID = $OST0_LIMIT"
601
602         # cleanup
603         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR
604 }
605 run_test 7 "Quota recovery (only block limit) ======"
606
607 # run dbench with quota enabled
608 test_8() {
609         BLK_LIMIT=$((100 * 1024 * 1024)) # 100G
610         FILE_LIMIT=1000000
611         DBENCH_LIB=${DBENCH_LIB:-/usr/lib/dbench}
612         
613         [ ! -d $DBENCH_LIB ] && skip "dbench not installed" && return 0
614
615         wait_delete_completed
616         
617         echo "  Set enough high limit for user: $TSTUSR"
618         $LFS setquota -u $TSTUSR 0 $BLK_LIMIT 0 $FILE_LIMIT $DIR
619         echo "  Set enough high limit for group: $TSTUSR"
620         $LFS setquota -g $USER 0 $BLK_LIMIT 0 $FILE_LIMIT $DIR
621
622         TGT=$DIR/$tdir/client.txt
623         SRC=${SRC:-$DBENCH_LIB/client.txt}
624         [ ! -e $TGT -a -e $SRC ] && echo "copying $SRC to $TGT" && cp $SRC $TGT
625         SRC=$DBENCH_LIB/client_plain.txt
626         [ ! -e $TGT -a -e $SRC ] && echo "copying $SRC to $TGT" && cp $SRC $TGT
627
628         chmod 0777 $DIR/$tdir
629         SAVE_PWD=$PWD
630         cd $DIR/$tdir
631         local duration=""
632         [ "$SLOW" = "no" ] && duration=" -t 120"
633         $RUNAS dbench -c client.txt 3 $duration
634         RC=$?
635         [ $RC -ne 0 ] && killall -9 dbench
636         
637         rm -f client.txt
638         sync; sleep 3; sync;
639
640         cd $SAVE_PWD
641         return $RC
642 }
643 run_test 8 "Run dbench with quota enabled ==========="
644
645 # run for fixing bug10707, it needs a big room. test for 64bit
646 KB=1024
647 GB=$((KB * 1024 * 1024))
648 FSIZE=$((OSTCOUNT * 9 / 2))
649 # Use this as dd bs to decrease time
650 # inode->i_blkbits = min(PTLRPC_MAX_BRW_BITS+1, LL_MAX_BLKSIZE_BITS);
651 blksize=$((1 << 21)) # 2Mb
652
653 test_9() {
654         chmod 0777 $DIR/$tdir
655         lustrefs_size=`(echo 0; df -t lustre -P | awk '{print $4}') | tail -n 1`
656         size_file=$((FSIZE * GB))
657         echo "lustrefs_size:$lustrefs_size  size_file:$((size_file / KB))"
658         if [ $((lustrefs_size * KB)) -lt $size_file ]; then
659                 skip "less than $size_file bytes free"
660                 return 0;
661         fi
662
663         set_blk_unitsz $((1024 * 100))
664         set_blk_tunesz $((1024 * 50))
665
666         # set the D_QUOTA flag
667         debugsave
668         sysctl -w lnet.debug="+quota"
669
670         TESTFILE="$DIR/$tdir/$tfile-0"
671
672         BLK_LIMIT=$((100 * KB * KB)) # 100G
673         FILE_LIMIT=1000000
674
675         echo "  Set enough high limit(block:$BLK_LIMIT; file: $FILE_LIMIT) for user: $TSTUSR"
676         $LFS setquota -u $TSTUSR 0 $BLK_LIMIT 0 $FILE_LIMIT $DIR
677         echo "  Set enough high limit(block:$BLK_LIMIT; file: $FILE_LIMIT) for group: $TSTUSR"
678         $LFS setquota -g $TSTUSR 0 $BLK_LIMIT 0 $FILE_LIMIT $DIR
679
680         echo "  Set stripe"
681         [ $OSTCOUNT -ge 2 ] && $LFS setstripe $TESTFILE -c $OSTCOUNT
682         touch $TESTFILE
683         chown $TSTUSR.$TSTUSR $TESTFILE
684
685         $SHOW_QUOTA_USER
686         $SHOW_QUOTA_GROUP
687
688         echo "    Write the big file of $FSIZE G ..."
689         $RUNAS dd if=/dev/zero of=$TESTFILE  bs=$blksize count=$((size_file / blksize)) || \
690                error "(usr) write $FSIZE G file failure, but expect success"
691
692         $SHOW_QUOTA_USER
693         $SHOW_QUOTA_GROUP
694
695         echo "    delete the big file of $FSIZE G..." 
696         $RUNAS rm -f $TESTFILE
697
698         $SHOW_QUOTA_USER
699         $SHOW_QUOTA_GROUP
700
701         echo "    write the big file of 2 G..."
702         $RUNAS dd if=/dev/zero of=$TESTFILE  bs=$blksize count=$((2 * GB / blksize)) || \
703                error "(usr) write 2 G file failure, but expect seccess"
704
705         echo "    delete the big file of 2 G..."
706         $RUNAS rm -f $TESTFILE 
707         RC=$?
708
709         set_blk_tunesz $BTUNE_SZ
710         set_blk_unitsz $BUNIT_SZ
711
712         debugrestore
713         wait_delete_completed
714
715         return $RC
716 }
717 run_test 9 "run for fixing bug10707(64bit) ==========="
718
719 # run for fixing bug10707, it need a big room. test for 32bit
720 test_10() {
721         chmod 0777 $DIR/$tdir
722         lustrefs_size=`(echo 0; df -t lustre -P | awk '{print $4}') | tail -n 1`
723         size_file=$((FSIZE * GB))
724         echo "lustrefs_size:$lustrefs_size  size_file:$((size_file / KB))"
725         if [ $((lustrefs_size * KB)) -lt $size_file ]; then
726                 skip "less than $size_file bytes free"
727                 return 0;
728         fi
729
730         sync; sleep 10; sync;
731
732         set_blk_unitsz $((1024 * 100))
733         set_blk_tunesz $((1024 * 50))
734
735         # set the D_QUOTA flag
736         debugsave
737         sysctl -w lnet.debug="+quota"
738         
739         # make qd_count 32 bit
740         sysctl -w lustre.fail_loc=0xA00
741
742         TESTFILE="$DIR/$tdir/$tfile-0"
743
744         BLK_LIMIT=$((100 * KB * KB)) # 100G
745         FILE_LIMIT=1000000
746
747         echo "  Set enough high limit(block:$BLK_LIMIT; file: $FILE_LIMIT) for user: $TSTUSR"
748         $LFS setquota -u $TSTUSR 0 $BLK_LIMIT 0 $FILE_LIMIT $DIR
749         echo "  Set enough high limit(block:$BLK_LIMIT; file: $FILE_LIMIT) for group: $TSTUSR"
750         $LFS setquota -g $TSTUSR 0 $BLK_LIMIT 0 $FILE_LIMIT $DIR
751        
752         echo "  Set stripe"
753         [ $OSTCOUNT -ge 2 ] && $LFS setstripe $TESTFILE -c $OSTCOUNT
754         touch $TESTFILE
755         chown $TSTUSR.$TSTUSR $TESTFILE
756
757         $SHOW_QUOTA_USER
758         $SHOW_QUOTA_GROUP
759
760         echo "    Write the big file of $FSIZE G ..."
761         $RUNAS dd if=/dev/zero of=$TESTFILE  bs=$blksize count=$((size_file / blksize)) || \
762                 error "(usr) write $FSIZE G file failure, but expect success"
763  
764         $SHOW_QUOTA_USER
765         $SHOW_QUOTA_GROUP
766
767         echo "    delete the big file of $FSIZE G..."
768         $RUNAS rm -f $TESTFILE 
769
770         $SHOW_QUOTA_USER
771         $SHOW_QUOTA_GROUP
772
773         echo "    write the big file of 2 G..."
774         $RUNAS dd if=/dev/zero of=$TESTFILE  bs=$blksize count=$((2 * GB / blkzise)) || \
775                 error "(usr) write 2 G file failure, but expect success" 
776
777         echo "    delete the big file of 2 G..."
778         $RUNAS rm -f $TESTFILE 
779
780         RC=$?
781
782         # clear the flage
783         debugrestore
784
785         # make qd_count 64 bit
786         sysctl -w lustre.fail_loc=0
787
788         set_blk_tunesz $BTUNE_SZ
789         set_blk_unitsz $BUNIT_SZ
790
791         wait_delete_completed
792
793         return $RC
794 }
795 run_test 10 "run for fixing bug10707(32bit) ==========="
796
797 test_11() {
798        wait_delete_completed
799
800        #prepare the test
801        block_limit=`(echo 0; df -t lustre -P | awk '{print $(NF - 4)}') | tail -n 1`
802        echo $block_limit
803        orig_dbr=`cat /proc/sys/vm/dirty_background_ratio`
804        orig_dec=`cat /proc/sys/vm/dirty_expire_centisecs`
805        orig_dr=`cat /proc/sys/vm/dirty_ratio`
806        orig_dwc=`cat /proc/sys/vm/dirty_writeback_centisecs`
807        echo 1  > /proc/sys/vm/dirty_background_ratio
808        echo 30 > /proc/sys/vm/dirty_expire_centisecs
809        echo 1  > /proc/sys/vm/dirty_ratio
810        echo 50 > /proc/sys/vm/dirty_writeback_centisecs
811        TESTDIR="$DIR/$tdir"
812        local RV=0
813
814        #do the test
815        local SECS=0
816        local REPS=3
817        [ "$SLOW" = no ] && REPS=1
818        local sleep=20
819        local i=1
820        while [ $i -le $REPS ]; do
821            echo "test: cycle($i of $REPS) start at $(date)"
822            mkdir -p $DIR/$tdir && chmod 777 $DIR/$tdir
823            echo -n "    create a file for uid "
824            for j in `seq 1 30`; do
825                echo -n "$j "
826                # 30MB per dd for a total of 900MB (if space even permits)
827                runas -u $j dd if=/dev/zero of=$DIR/$tdir/$tfile  bs=$blksize count=15 > /dev/null 2>&1 &
828            done
829            echo ""
830            PROCS=$(ps -ef | grep -v grep | grep "dd if /dev/zero of $TESTDIR" | wc -l)
831            LAST_USED=0
832            while [ $PROCS -gt 0 ]; do 
833              sleep 20
834              SECS=$((SECS + sleep))
835              PROCS=$(ps -ef | grep -v grep | grep "dd if /dev/zero of $TESTDIR" | wc -l)
836              USED=$(du -s $DIR/$tdir | awk '{print $1}')
837              PCT=$(($USED * 100 / $block_limit))
838              echo "${i}/${REPS} ${PCT}% p${PROCS} t${SECS}  "
839              if [ $USED -le $LAST_USED ]; then
840                  kill -9 $(ps -ef | grep "dd if /dev/zero of $DIR/$tdir" | grep -v grep | awk '{ print $2 }')
841                  i=$REPS
842                  RV=2
843                  break
844              fi
845              LAST_USED=$USED
846            done
847            echo "    removing the test files..."
848            rm -f $DIR/$tdir/$tfile
849            echo "cycle $i done at $(date)"
850            i=$[$i+1]
851        done
852        echo "Test took $SECS sec"
853
854        #clean
855        echo $orig_dbr > /proc/sys/vm/dirty_background_ratio
856        echo $orig_dec > /proc/sys/vm/dirty_expire_centisecs
857        echo $orig_dr  > /proc/sys/vm/dirty_ratio
858        echo $orig_dwc > /proc/sys/vm/dirty_writeback_centisecs
859        if [ $RV -ne 0 ]; then
860            error "Nothing was written for $SECS sec ... aborting"
861        fi
862        return $RV
863 }
864 run_test 11 "run for fixing bug10912 ==========="
865
866
867 # test a deadlock between quota and journal b=11693
868 test_12() {
869         chmod 0777 $DIR/$tdir
870
871         [ "$(grep $DIR2 /proc/mounts)" ] || mount_client $DIR2 || \
872                 { skip "Need lustre mounted on $MOUNT2 " && retutn 0; }
873
874         LIMIT=$(( $BUNIT_SZ * $(($OSTCOUNT + 1)) * 10)) # 10 bunits each sever
875         TESTFILE="$DIR/$tdir/$tfile-0"
876         TESTFILE2="$DIR2/$tdir/$tfile-1"
877         
878         echo "   User quota (limit: $LIMIT kbytes)"
879         $LFS setquota -u $TSTUSR 0 $LIMIT 0 0 $DIR
880
881         $LFS setstripe $TESTFILE -i 0 -c 1 
882         chown $TSTUSR.$TSTUSR $TESTFILE
883         $LFS setstripe $TESTFILE2 -i 0 -c 1
884         chown $TSTUSR2.$TSTUSR2 $TESTFILE2
885
886         #define OBD_FAIL_OST_HOLD_WRITE_RPC      0x21f
887         sysctl -w lustre.fail_loc=0x0000021f        
888
889         echo "   step1: write out of block quota ..."
890         $RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ count=$(($LIMIT*2)) & 
891         DDPID=$!
892         sleep 5
893         $RUNAS2 dd if=/dev/zero of=$TESTFILE2 bs=$BLK_SZ count=102400 & 
894         DDPID1=$!
895
896         echo  "   step2: testing ......"
897         count=0
898         while [ true ]; do
899             if [ -z `ps -ef | awk '$2 == '${DDPID1}' { print $8 }'` ]; then break; fi
900             count=$[count+1]
901             if [ $count -gt 64 ]; then
902                 sysctl -w lustre.fail_loc=0
903                 error "dd should be finished!"
904             fi
905             sleep 1
906         done    
907         echo "(dd_pid=$DDPID1, time=$count)successful"
908
909         #Recover fail_loc and dd will finish soon
910         sysctl -w lustre.fail_loc=0
911
912         echo  "   step3: testing ......"
913         count=0
914         while [ true ]; do
915             if [ -z `ps -ef | awk '$2 == '${DDPID}' { print $8 }'` ]; then break; fi
916             count=$[count+1]
917             if [ $count -gt 100 ]; then
918                 error "dd should be finished!"
919             fi
920             sleep 1
921         done    
922         echo "(dd_pid=$DDPID, time=$count)successful"
923
924         rm -f $TESTFILE $TESTFILE2
925         
926         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR           # clear user limit
927 }
928 run_test 12 "test a deadlock between quota and journal ==="
929
930 # test multiple clients write block quota b=11693
931 test_13() {
932         # one OST * 10 + (mds + other OSTs)
933         LIMIT=$((BUNIT_SZ * 10 + (BUNIT_SZ * OSTCOUNT)))
934         TESTFILE="$DIR/$tdir/$tfile"
935         
936         echo "   User quota (limit: $LIMIT kbytes)"
937         $LFS setquota -u $TSTUSR 0 $LIMIT 0 0 $DIR
938         $SHOW_QUOTA_USER
939         
940         $LFS setstripe $TESTFILE -i 0 -c 1
941         chown $TSTUSR.$TSTUSR $TESTFILE
942         $LFS setstripe $TESTFILE.2 -i 0 -c 1
943         chown $TSTUSR.$TSTUSR $TESTFILE.2
944
945         echo "   step1: write out of block quota ..."
946         # one bunit will give mds
947         $RUNAS dd if=/dev/zero of=$TESTFILE bs=$BLK_SZ count=$[($LIMIT - $BUNIT_SZ) / 2] & 
948         DDPID=$!
949         $RUNAS dd if=/dev/zero of=$TESTFILE.2 bs=$BLK_SZ count=$[($LIMIT - $BUNIT_SZ) / 2] & 
950         DDPID1=$!
951
952         echo  "   step2: testing ......"
953         count=0
954         while [ true ]; do
955             if [ -z `ps -ef | awk '$2 == '${DDPID}' { print $8 }'` ]; then break; fi
956             count=$[count+1]
957             if [ $count -gt 64 ]; then
958                 error "dd should be finished!"
959             fi
960             sleep 1
961         done    
962         echo "(dd_pid=$DDPID, time=$count)successful"
963
964         count=0
965         while [ true ]; do
966             if [ -z `ps -ef | awk '$2 == '${DDPID1}' { print $8 }'` ]; then break; fi
967             count=$[count+1]
968             if [ $count -gt 64 ]; then
969                 error "dd should be finished!"
970             fi
971             sleep 1
972         done    
973         echo "(dd_pid=$DDPID1, time=$count)successful"
974
975         sync; sleep 5; sync;
976
977         echo  "   step3: checking ......"
978         fz=`stat -c %s $TESTFILE`
979         fz2=`stat -c %s $TESTFILE.2`
980         $SHOW_QUOTA_USER
981         [ $((fz + fz2)) -lt $((BUNIT_SZ * BLK_SZ * 10)) ] && \
982                 error "files too small $fz + $fz2 < $((BUNIT_SZ * BLK_SZ * 10))"
983
984         rm -f $TESTFILE $TESTFILE.2
985         
986         $LFS setquota -u $TSTUSR 0 0 0 0 $DIR           # clear user limit
987 }
988 run_test 13 "test multiple clients write block quota ==="
989
990 check_if_quota_zero(){
991         line=`$LFS quota -$1 $2 $DIR | wc -l`
992         for i in `seq 3 $line`; do
993             if [ $i -eq 3 ]; then
994                 field="3 4 6 7"
995             else
996                 field="3 5"
997             fi
998             for j in $field; do
999                 tmp=`$LFS quota -$1 $2 $DIR | sed -n ${i}p | 
1000                      awk  '{print $'"$j"'}'`
1001                 [ -n "$tmp" ] && [ $tmp -ne 0 ] && $LFS quota -$1 $2 $DIR && \
1002                     error "quota on $2 isn't clean"
1003             done
1004         done
1005         echo "pass check_if_quota_zero"
1006 }
1007
1008 pre_test_14 () {
1009         # reboot the lustre
1010         cd $T_PWD; sh llmountcleanup.sh || error "llmountcleanup failed"
1011         sh llmount.sh
1012         pre_test
1013         run_test 0 "reboot lustre"
1014 }
1015
1016 pre_test_14 
1017
1018 test_14(){ # b=12223 -- setting quota on root
1019         TESTFILE="$DIR/$tdir/$tfile"
1020
1021         # out of root's file and block quota
1022         $LFS setquota -u root 10 10 10 10 $DIR
1023         createmany -m ${TESTFILE} 20 || \
1024             error "unexpected: user(root) create files failly!"
1025         dd if=/dev/zero of=$TESTFILE bs=4k count=4096 || \
1026             error "unexpected: user(root) write files failly!"
1027         chmod 666 $TESTFILE
1028         $RUNAS dd if=/dev/zero of=${TESTFILE} seek=4096 bs=4k count=4096 && \
1029             error "unexpected: user(quota_usr) write a file successfully!"      
1030
1031         # trigger the llog
1032         chmod 777 $DIR
1033         for i in `seq 1 10`; do $RUNAS touch ${TESTFILE}a_$i; done 
1034         for i in `seq 1 10`; do $RUNAS rm -f ${TESTFILE}a_$i; done 
1035
1036         # do the check
1037         dmesg | tail | grep "\-122" |grep llog_obd_origin_add && error "err -122 not found in dmesg" 
1038         $LFS setquota -u root 0 0 0 0 $DIR
1039         #check_if_quota_zero u root
1040
1041         # clean 
1042         unlinkmany ${TESTFILE} 15
1043         rm -f $TESTFILE
1044 }
1045 run_test 14 "test setting quota on root ==="
1046
1047 # turn off quota
1048 test_99()
1049 {
1050         $LFS quotaoff $DIR
1051         return 0
1052 }
1053 run_test 99 "Quota off ==============================="
1054
1055
1056 log "cleanup: ======================================================"
1057 cd $ORIG_PWD
1058 post_test
1059 check_and_cleanup_lustre
1060 echo '=========================== finished ==============================='
1061 [ -f "$QUOTALOG" ] && cat $QUOTALOG && grep -q FAIL $QUOTALOG && exit 1 || true
1062 echo "$0: completed"