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