Whamcloud - gitweb
LU-2211 quota: cap how long a thread can wait for quota
[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 SRCDIR=`dirname $0`
10 export PATH=$PWD/$SRCDIR:$SRCDIR:$PWD/$SRCDIR/../utils:$PATH:/sbin
11
12 ONLY=${ONLY:-"$*"}
13 ALWAYS_EXCEPT="$SANITY_QUOTA_EXCEPT"
14 # UPDATE THE COMMENT ABOVE WITH BUG NUMBERS WHEN CHANGING ALWAYS_EXCEPT!
15
16 [ "$ALWAYS_EXCEPT$EXCEPT" ] &&
17         echo "Skipping tests: `echo $ALWAYS_EXCEPT $EXCEPT`"
18
19 TMP=${TMP:-/tmp}
20
21 ORIG_PWD=${PWD}
22 TSTID=${TSTID:-60000}
23 TSTID2=${TSTID2:-60001}
24 TSTUSR=${TSTUSR:-"quota_usr"}
25 TSTUSR2=${TSTUSR2:-"quota_2usr"}
26 BLK_SZ=1024
27 MAX_DQ_TIME=604800
28 MAX_IQ_TIME=604800
29
30 LUSTRE=${LUSTRE:-$(cd $(dirname $0)/..; echo $PWD)}
31 . $LUSTRE/tests/test-framework.sh
32 init_test_env $@
33 . ${CONFIG:=$LUSTRE/tests/cfg/$NAME.sh}
34 init_logging
35 DIRECTIO=${DIRECTIO:-$LUSTRE/tests/directio}
36
37 [ $MDSCOUNT -gt 1 ] && skip "CMD case" && exit 0
38
39 require_dsh_mds || exit 0
40 require_dsh_ost || exit 0
41
42 # XXX Once we drop the interoperability with old server (< 2.3.50), the
43 #     sanity-quota-old.sh should be removed.
44 if [ $(lustre_version_code $SINGLEMDS) -lt $(version_code 2.3.50) ]; then
45         exec $LUSTRE/tests/sanity-quota-old.sh
46 fi
47
48 # if e2fsprogs support quota feature?
49 if [ $(facet_fstype $SINGLEMDS) == ldiskfs ] && \
50         ! $DEBUGFS -c -R supported_features | grep -q 'quota'; then
51         skip "e2fsprogs doesn't support quota" && exit 0
52 fi
53
54 [ "$SLOW" = "no" ] && EXCEPT_SLOW="9 18 21"
55
56 QUOTALOG=${TESTSUITELOG:-$TMP/$(basename $0 .sh).log}
57
58 [ "$QUOTALOG" ] && rm -f $QUOTALOG || true
59
60 DIR=${DIR:-$MOUNT}
61 DIR2=${DIR2:-$MOUNT2}
62
63 QUOTA_AUTO_OLD=$QUOTA_AUTO
64 export QUOTA_AUTO=0
65
66 check_and_setup_lustre
67
68 LOVNAME=`lctl get_param -n llite.*.lov.common_name | tail -n 1`
69 OSTCOUNT=`lctl get_param -n lov.$LOVNAME.numobd`
70
71 SHOW_QUOTA_USER="$LFS quota -v -u $TSTUSR $DIR"
72 SHOW_QUOTA_USERID="$LFS quota -v -u $TSTID $DIR"
73 SHOW_QUOTA_USER2="$LFS quota -v -u $TSTUSR2 $DIR"
74 SHOW_QUOTA_GROUP="$LFS quota -v -g $TSTUSR $DIR"
75 SHOW_QUOTA_GROUPID="$LFS quota -v -g $TSTID $DIR"
76 SHOW_QUOTA_GROUP2="$LFS quota -v -g $TSTUSR2 $DIR"
77 SHOW_QUOTA_INFO_USER="$LFS quota -t -u $DIR"
78 SHOW_QUOTA_INFO_GROUP="$LFS quota -t -g $DIR"
79
80 build_test_filter
81
82 lustre_fail() {
83         local fail_node=$1
84         local fail_loc=$2
85         local fail_val=${3:-0}
86
87         if [ $fail_node == "mds" ] || [ $fail_node == "mds_ost" ]; then
88                 do_facet $SINGLEMDS "lctl set_param fail_val=$fail_val"
89                 do_facet $SINGLEMDS "lctl set_param fail_loc=$fail_loc"
90         fi
91
92         if [ $fail_node == "ost" ] || [ $fail_node == "mds_ost" ]; then
93                 for num in `seq $OSTCOUNT`; do
94                         do_facet ost$num "lctl set_param fail_val=$fail_val"
95                         do_facet ost$num "lctl set_param fail_loc=$fail_loc"
96                 done
97         fi
98 }
99
100 RUNAS="runas -u $TSTID -g $TSTID"
101 RUNAS2="runas -u $TSTID2 -g $TSTID2"
102 DD="dd if=/dev/zero bs=1M"
103
104 FAIL_ON_ERROR=false
105
106 check_runas_id_ret $TSTUSR $TSTUSR $RUNAS ||
107         error "Please create user $TSTUSR($TSTID) and group $TSTUSR($TSTID)"
108 check_runas_id_ret $TSTUSR2 $TSTUSR2 $RUNAS2 ||
109         error "Please create user $TSTUSR2($TSTID2) and group $TSTUSR2($TSTID2)"
110
111 # clear quota limits for a user or a group
112 # usage: resetquota -u username
113 #        resetquota -g groupname
114
115 resetquota() {
116         [ "$#" != 2 ] && error "resetquota: wrong number of arguments: $#"
117         [ "$1" != "-u" -a "$1" != "-g" ] &&
118                 error "resetquota: wrong specifier $1 passed"
119
120         $LFS setquota "$1" "$2" -b 0 -B 0 -i 0 -I 0 $MOUNT
121         # give a chance to slave to release space
122         sleep 1
123 }
124
125 quota_scan() {
126         local LOCAL_UG=$1
127         local LOCAL_ID=$2
128
129         if [ "$LOCAL_UG" == "a" -o "$LOCAL_UG" == "u" ]; then
130                 $LFS quota -v -u $LOCAL_ID $DIR
131                 log "Files for user ($LOCAL_ID):"
132                 ($LFS find -user $LOCAL_ID $DIR | head -n 4 |
133                         xargs stat 2>/dev/null)
134         fi
135
136         if [ "$LOCAL_UG" == "a" -o "$LOCAL_UG" == "g" ]; then
137                 $LFS quota -v -u $LOCAL_ID $DIR
138                 log "Files for group ($LOCAL_ID):"
139                 ($LFS find -group $LOCAL_ID $DIR | head -n 4 |
140                         xargs stat 2>/dev/null)
141         fi
142 }
143
144 quota_error() {
145         quota_scan $1 $2
146         shift 2
147         error "$*"
148 }
149
150 quota_log() {
151         quota_scan $1 $2
152         shift 2
153         log "$*"
154 }
155
156 # get quota for a user or a group
157 # usage: getquota -u|-g <username>|<groupname> global|<obd_uuid> \
158 #                 bhardlimit|bsoftlimit|bgrace|ihardlimit|isoftlimit|igrace
159 #
160 getquota() {
161         local spec
162         local uuid
163
164         [ "$#" != 4 ] && error "getquota: wrong number of arguments: $#"
165         [ "$1" != "-u" -a "$1" != "-g" ] &&
166                 error "getquota: wrong u/g specifier $1 passed"
167
168         uuid="$3"
169
170         case "$4" in
171                 curspace)   spec=1;;
172                 bsoftlimit) spec=2;;
173                 bhardlimit) spec=3;;
174                 bgrace)     spec=4;;
175                 curinodes)  spec=5;;
176                 isoftlimit) spec=6;;
177                 ihardlimit) spec=7;;
178                 igrace)     spec=8;;
179                 *)          error "unknown quota parameter $4";;
180         esac
181
182         [ "$uuid" = "global" ] && uuid=$DIR
183
184         $LFS quota -v "$1" "$2" $DIR |
185                 awk 'BEGIN { num='$spec' } { if ($1 == "'$uuid'") \
186                 { if (NF == 1) { getline } else { num++ } ; print $num;} }' \
187                 | tr -d "*"
188 }
189
190 # set mdt quota type
191 # usage: set_mdt_qtype ug|u|g|none
192 set_mdt_qtype() {
193         local qtype=$1
194         local varsvc=${SINGLEMDS}_svc
195         do_facet mgs $LCTL conf_param $FSNAME.quota.mdt=$qtype
196         if $(facet_up $SINGLEMDS); then
197                 wait_update_facet $SINGLEMDS "$LCTL get_param -n \
198                         osd-$FSTYPE.${!varsvc}.quota_slave.enabled" "$qtype" ||
199                                 return 1
200         fi
201         return 0
202 }
203
204 # set ost quota type
205 # usage: set_ost_quota_type ug|u|g|none
206 set_ost_qtype() {
207         local qtype=$1
208         local varsvc
209         local osts=$(get_facets OST)
210         do_facet mgs $LCTL conf_param $FSNAME.quota.ost=$qtype
211         # we have to make sure each OST received config changes
212         for ost in ${osts//,/ }; do
213                 varsvc=${ost}_svc
214                 if $(facet_up $ost); then
215                         wait_update_facet $ost "$LCTL get_param -n \
216                                 osd-$FSTYPE.${!varsvc}.quota_slave.enabled" \
217                                         "$qtype" || return 1
218                 fi
219         done
220         return 0
221 }
222
223 wait_reintegration() {
224         local ntype=$1
225         local qtype=$2
226         local result="glb[1],slv[1],reint[0]"
227         local varsvc
228
229         if [ $ntype == "mdt" ]; then
230                 varsvc=${SINGLEMDS}_svc
231                 if $(facet_up $SINGLEMDS); then
232                         wait_update_facet $SINGLEMDS "$LCTL get_param -n \
233                         osd-$FSTYPE.${!varsvc}.quota_slave.info |
234                         grep "$qtype" | awk '{ print \\\$3 }'" "$result" ||
235                                 return 1
236                 fi
237         else
238                 local osts=$(get_facets OST)
239                 for ost in ${osts//,/ }; do
240                         varsvc=${ost}_svc
241                         if $(facet_up $ost); then
242                                 wait_update_facet $ost "$LCTL get_param -n \
243                                 osd-$FSTYPE.${!varsvc}.quota_slave.info |
244                                 grep "$qtype" | awk '{ print \\\$3 }'" \
245                                         "$result" || return 1
246                         fi
247                 done
248         fi
249         return 0
250 }
251
252 wait_mdt_reint() {
253         local qtype=$1
254
255         if [ $qtype == "u" ] || [ $qtype == "ug" ]; then
256                 wait_reintegration "mdt" "user" || return 1
257         fi
258
259         if [ $qtype == "g" ] || [ $qtype == "ug" ]; then
260                 wait_reintegration "mdt" "group" || return 1
261         fi
262         return 0
263 }
264
265 wait_ost_reint() {
266         local qtype=$1
267
268         if [ $qtype == "u" ] || [ $qtype == "ug" ]; then
269                 wait_reintegration "ost" "user" || return 1
270         fi
271
272         if [ $qtype == "g" ] || [ $qtype == "ug" ]; then
273                 wait_reintegration "ost" "group" || return 1
274         fi
275         return 0
276 }
277
278 setup_quota_test() {
279         rm -rf $DIR/$tdir
280         wait_delete_completed
281         echo "Creating test directory"
282         mkdir -p $DIR/$tdir
283         chmod 077 $DIR/$tdir
284         # always clear fail_loc in case of fail_loc isn't cleared
285         # properly when previous test failed
286         lustre_fail mds_ost 0
287 }
288
289 cleanup_quota_test() {
290         trap 0
291         echo "Delete files..."
292         rm -rf $DIR/$tdir
293         echo "Wait for unlink objects finished..."
294         wait_delete_completed
295         sync_all_data || true
296 }
297
298 quota_show_check() {
299         local bf=$1
300         local ug=$2
301         local qid=$3
302         local usage
303
304         $LFS quota -v -$ug $qid $DIR
305
306         if [ "$bf" == "a" -o "$bf" == "b" ]; then
307                 usage=$(getquota -$ug $qid global curspace)
308                 if [ -z $usage ]; then
309                         quota_error $ug $qid \
310                                 "Query block quota failed ($ug:$qid)."
311                 else
312                         [ $usage -ne 0 ] && quota_log $ug $qid \
313                                 "Block quota isn't 0 ($ug:$qid:$usage)."
314                 fi
315         fi
316
317         if [ "$bf" == "a" -o "$bf" == "f" ]; then
318                 usage=$(getquota -$ug $qid global curinodes)
319                 if [ -z $usage ]; then
320                         quota_error $ug $qid \
321                                 "Query file quota failed ($ug:$qid)."
322                 else
323                         [ $usage -ne 0 ] && quota_log $ug $qid \
324                                 "File quota isn't 0 ($ug:$qid:$usage)."
325                 fi
326         fi
327 }
328
329 # enable quota debug
330 quota_init() {
331         do_nodes $(comma_list $(nodes_list)) "lctl set_param debug=+quota"
332 }
333 quota_init
334
335 test_quota_performance() {
336         local TESTFILE="$DIR/$tdir/$tfile-0"
337         local size=$1 # in MB
338         local stime=$(date +%s)
339         $RUNAS $DD of=$TESTFILE count=$size conv=fsync ||
340                 quota_error u $TSTUSR "write failure"
341         local etime=$(date +%s)
342         delta=$((etime - stime))
343         if [ $delta -gt 0 ]; then
344             rate=$((size * 1024 / delta))
345             [ $rate -gt 1024 ] ||
346                 error "SLOW IO for $TSTUSR (user): $rate KB/sec"
347         fi
348         rm -f $TESTFILE
349 }
350
351 # test basic quota performance b=21696
352 test_0() {
353         local MB=100 # 100M
354         [ "$SLOW" = "no" ] && MB=10
355
356         local free_space=$(lfs df | grep "filesystem summary" | \
357                                 awk '{print $5}')
358         [ $free_space -le $((MB * 1024)) ] &&
359                 skip "not enough space ${free_space} KB, " \
360                         "required $((MB * 1024)) KB" && return
361         setup_quota_test
362         trap cleanup_quota_test EXIT
363
364         set_ost_qtype "none" || error "disable ost quota failed"
365         test_quota_performance $MB
366
367         set_ost_qtype "ug" || error "enable ost quota failed"
368         $LFS setquota -u $TSTUSR -b 0 -B 10G -i 0 -I 0 $DIR
369         test_quota_performance $MB
370
371         cleanup_quota_test
372         resetquota -u $TSTUSR
373 }
374 run_test 0 "Test basic quota performance"
375
376 # test block hardlimit
377 test_1() {
378         local LIMIT=10  # 10M
379         local TESTFILE="$DIR/$tdir/$tfile-0"
380
381         setup_quota_test
382         trap cleanup_quota_test EXIT
383
384         # enable ost quota
385         set_ost_qtype "ug" || "enable ost quota failed"
386
387         # test for user
388         log "User quota (block hardlimit:$LIMIT MB)"
389         $LFS setquota -u $TSTUSR -b 0 -B ${LIMIT}M -i 0 -I 0 $DIR
390
391         # make sure the system is clean
392         local USED=$(getquota -u $TSTUSR global curspace)
393         [ $USED -ne 0 ] && error "Used space($USED) for user $TSTUSR isn't 0."
394
395         $LFS setstripe $TESTFILE -c 1
396         chown $TSTUSR.$TSTUSR $TESTFILE
397
398         log "Write..."
399         $RUNAS $DD of=$TESTFILE count=$((LIMIT/2)) ||
400                 quota_error u $TSTUSR "user write failure, but expect success"
401         log "Write out of block quota ..."
402         # this time maybe cache write,  ignore it's failure
403         $RUNAS $DD of=$TESTFILE count=$((LIMIT/2)) seek=$((LIMIT/2)) || true
404         # flush cache, ensure noquota flag is set on client
405         cancel_lru_locks osc
406         $RUNAS $DD of=$TESTFILE count=1 seek=$LIMIT &&
407                 quota_error u $TSTUSR "user write success, but expect EDQUOT"
408
409         rm -f $TESTFILE
410         wait_delete_completed
411         sync_all_data || true
412         USED=$(getquota -u $TSTUSR global curspace)
413         [ $USED -ne 0 ] && quota_error u $TSTUSR \
414                 "user quota isn't released after deletion"
415         resetquota -u $TSTUSR
416
417         # test for group
418         log "--------------------------------------"
419         log "Group quota (block hardlimit:$LIMIT MB)"
420         $LFS setquota -g $TSTUSR -b 0 -B ${LIMIT}M -i 0 -I 0 $DIR
421
422         TESTFILE="$DIR/$tdir/$tfile-1"
423         # make sure the system is clean
424         USED=$(getquota -g $TSTUSR global curspace)
425         [ $USED -ne 0 ] && error "Used space($USED) for group $TSTUSR isn't 0"
426
427         $LFS setstripe $TESTFILE -c 1
428         chown $TSTUSR.$TSTUSR $TESTFILE
429
430         log "Write ..."
431         $RUNAS $DD of=$TESTFILE count=$((LIMIT/2)) ||
432                 quota_error g $TSTUSR "group write failure, but expect success"
433         log "Write out of block quota ..."
434         # this time maybe cache write, ignore it's failure
435         $RUNAS $DD of=$TESTFILE count=$((LIMIT/2)) seek=$((LIMIT/2)) || true
436         cancel_lru_locks osc
437         $RUNAS $DD of=$TESTFILE count=10 seek=$LIMIT &&
438                 quota_error g $TSTUSR "group write success, but expect EDQUOT"
439
440         # cleanup
441         cleanup_quota_test
442
443         USED=$(getquota -g $TSTUSR global curspace)
444         [ $USED -ne 0 ] && quota_error g $TSTUSR \
445                 "group quota isn't released after deletion"
446
447         resetquota -g $TSTUSR
448 }
449 run_test 1 "Block hard limit (normal use and out of quota)"
450
451 # test inode hardlimit
452 test_2() {
453         local LIMIT=$((1024 * 1024)) # 1M inodes
454         local TESTFILE="$DIR/$tdir/$tfile-0"
455
456         [ "$SLOW" = "no" ] && LIMIT=1024 # 1k inodes
457
458         setup_quota_test
459         trap cleanup_quota_test EXIT
460
461         # enable mdt quota
462         set_mdt_qtype "ug" || "enable mdt quota failed"
463
464         # test for user
465         log "User quota (inode hardlimit:$LIMIT files)"
466         $LFS setquota -u $TSTUSR -b 0 -B 0 -i 0 -I $LIMIT $DIR
467
468         # make sure the system is clean
469         local USED=$(getquota -u $TSTUSR global curinodes)
470         [ $USED -ne 0 ] && error "Used inodes($USED) for user $TSTUSR isn't 0."
471
472         log "Create $LIMIT files ..."
473         $RUNAS createmany -m ${TESTFILE} $LIMIT || \
474                 quota_error u $TSTUSR "user create failure, but expect success"
475         log "Create out of file quota ..."
476         $RUNAS touch ${TESTFILE}_xxx && \
477                 quota_error u $TSTUSR "user create success, but expect EDQUOT"
478
479         # cleanup
480         unlinkmany ${TESTFILE} $LIMIT
481         rm -f ${TESTFILE}_xxx
482         wait_delete_completed
483
484         USED=$(getquota -u $TSTUSR global curinodes)
485         [ $USED -ne 0 ] && quota_error u $TSTUSR \
486                 "user quota isn't released after deletion"
487         resetquota -u $TSTUSR
488
489         # test for group
490         log "--------------------------------------"
491         log "Group quota (inode hardlimit:$LIMIT files)"
492         $LFS setquota -g $TSTUSR -b 0 -B 0 -i 0 -I $LIMIT $DIR
493
494         TESTFILE=$DIR/$tdir/$tfile-1
495         # make sure the system is clean
496         USED=$(getquota -g $TSTUSR global curinodes)
497         [ $USED -ne 0 ] && error "Used inodes($USED) for group $TSTUSR isn't 0."
498
499         log "Create $LIMIT files ..."
500         $RUNAS createmany -m ${TESTFILE} $LIMIT ||
501                 quota_error g $TSTUSR "group create failure, but expect success"
502         log "Create out of file quota ..."
503         $RUNAS touch ${TESTFILE}_xxx && \
504                 quota_error g $TSTUSR "group create success, but expect EDQUOT"
505
506         # cleanup
507         unlinkmany ${TESTFILE} $LIMIT
508         rm -f ${TESTFILE}_xxx
509         wait_delete_completed
510
511         USED=$(getquota -g $TSTUSR global curinodes)
512         [ $USED -ne 0 ] && quota_error g $TSTUSR \
513                 "user quota isn't released after deletion"
514
515         cleanup_quota_test
516         resetquota -g $TSTUSR
517 }
518 run_test 2 "File hard limit (normal use and out of quota)"
519
520 test_block_soft() {
521         local TESTFILE=$1
522         local TIMER=$(($2 * 3 / 2))
523         local LIMIT=$3
524         local OFFSET=0
525
526         setup_quota_test
527         trap cleanup_quota_test EXIT
528
529         $LFS setstripe $TESTFILE -c 1 -i 0
530         chown $TSTUSR.$TSTUSR $TESTFILE
531
532         echo "Write up to soft limit"
533         $RUNAS $DD of=$TESTFILE count=$LIMIT ||
534                 quota_error a $TSTUSR "write failure, but expect success"
535         OFFSET=$((LIMIT * 1024))
536         cancel_lru_locks osc
537
538         echo "Write to exceed soft limit"
539         $RUNAS dd if=/dev/zero of=$TESTFILE bs=1K count=10 seek=$OFFSET ||
540                 quota_error a $TSTUSR "write failure, but expect success"
541         OFFSET=$((OFFSET + 1024)) # make sure we don't write to same block
542         cancel_lru_locks osc
543
544         $SHOW_QUOTA_USER
545         $SHOW_QUOTA_GROUP
546         $SHOW_QUOTA_INFO_USER
547         $SHOW_QUOTA_INFO_GROUP
548
549         echo "Write before timer goes off"
550         $RUNAS dd if=/dev/zero of=$TESTFILE bs=1K count=10 seek=$OFFSET ||
551                 quota_error a $TSTUSR "write failure, but expect success"
552         OFFSET=$((OFFSET + 1024))
553         cancel_lru_locks osc
554
555         echo "Sleep $TIMER seconds ..."
556         sleep $TIMER
557
558         $SHOW_QUOTA_USER
559         $SHOW_QUOTA_GROUP
560         $SHOW_QUOTA_INFO_USER
561         $SHOW_QUOTA_INFO_GROUP
562
563         echo "Write after timer goes off"
564         # maybe cache write, ignore.
565         $RUNAS dd if=/dev/zero of=$TESTFILE bs=1K count=10 seek=$OFFSET || true
566         OFFSET=$((OFFSET + 1024))
567         cancel_lru_locks osc
568         $RUNAS dd if=/dev/zero of=$TESTFILE bs=1K count=10 seek=$OFFSET &&
569                 quota_error a $TSTUSR "write success, but expect EDQUOT"
570
571         $SHOW_QUOTA_USER
572         $SHOW_QUOTA_GROUP
573         $SHOW_QUOTA_INFO_USER
574         $SHOW_QUOTA_INFO_GROUP
575
576         echo "Unlink file to stop timer"
577         rm -f $TESTFILE
578         wait_delete_completed
579         sync_all_data || true
580
581         $SHOW_QUOTA_USER
582         $SHOW_QUOTA_GROUP
583         $SHOW_QUOTA_INFO_USER
584         $SHOW_QUOTA_INFO_GROUP
585
586         $LFS setstripe $TESTFILE -c 1 -i 0
587         chown $TSTUSR.$TSTUSR $TESTFILE
588
589         echo "Write ..."
590         $RUNAS $DD of=$TESTFILE count=$LIMIT ||
591                 quota_error a $TSTUSR "write failure, but expect success"
592         # cleanup
593         cleanup_quota_test
594 }
595
596 # block soft limit
597 test_3() {
598         local LIMIT=1  # 1MB
599         local GRACE=10 # 10s
600         local TESTFILE=$DIR/$tdir/$tfile-0
601
602         set_ost_qtype "ug" || error "enable ost quota failed"
603
604         echo "User quota (soft limit:$LIMIT MB  grace:$GRACE seconds)"
605         # make sure the system is clean
606         local USED=$(getquota -u $TSTUSR global curspace)
607         [ $USED -ne 0 ] && error "Used space($USED) for user $TSTUSR isn't 0."
608
609         $LFS setquota -t -u --block-grace $GRACE --inode-grace $MAX_IQ_TIME $DIR
610         $LFS setquota -u $TSTUSR -b ${LIMIT}M -B 0 -i 0 -I 0 $DIR
611
612         test_block_soft $TESTFILE $GRACE $LIMIT
613         resetquota -u $TSTUSR
614
615         echo "Group quota (soft limit:$LIMIT MB  grace:$GRACE seconds)"
616         TESTFILE=$DIR/$tdir/$tfile-1
617         # make sure the system is clean
618         USED=$(getquota -g $TSTUSR global curspace)
619         [ $USED -ne 0 ] && error "Used space($USED) for group $TSTUSR isn't 0."
620
621         $LFS setquota -t -g --block-grace $GRACE --inode-grace $MAX_IQ_TIME $DIR
622         $LFS setquota -g $TSTUSR -b ${LIMIT}M -B 0 -i 0 -I 0 $DIR
623
624         test_block_soft $TESTFILE $GRACE $LIMIT
625         resetquota -g $TSTUSR
626
627         # cleanup
628         $LFS setquota -t -u --block-grace $MAX_DQ_TIME --inode-grace \
629                 $MAX_IQ_TIME $DIR
630         $LFS setquota -t -g --block-grace $MAX_DQ_TIME --inode-grace \
631                 $MAX_IQ_TIME $DIR
632 }
633 run_test 3 "Block soft limit (start timer, timer goes off, stop timer)"
634
635 test_file_soft() {
636         local TESTFILE=$1
637         local LIMIT=$2
638         local TIMER=$(($3 * 3 / 2))
639
640         setup_quota_test
641         trap cleanup_quota_test EXIT
642
643         echo "Create files to exceed soft limit"
644         $RUNAS createmany -m ${TESTFILE}_ $((LIMIT + 1)) ||
645                 quota_error a $TSTUSR "create failure, but expect success"
646         sync; sleep 1; sync
647
648         echo "Create file before timer goes off"
649         $RUNAS touch ${TESTFILE}_before ||
650                 quota_error a $TSTUSR "failed create before timer expired," \
651                         "but expect success"
652         sync; sleep 1; sync
653
654         echo "Sleep $TIMER seconds ..."
655         sleep $TIMER
656
657         $SHOW_QUOTA_USER
658         $SHOW_QUOTA_GROUP
659         $SHOW_QUOTA_INFO_USER
660         $SHOW_QUOTA_INFO_GROUP
661
662         echo "Create file after timer goes off"
663         # There is a window that space is accounted in the quota usage but
664         # hasn't been decreased from the pending write, if we acquire quota
665         # in this window, we'll acquire more than we needed.
666         $RUNAS touch ${TESTFILE}_after_1 ${TESTFILE}_after_2 || true
667         sync; sleep 1; sync
668         $RUNAS touch ${TESTFILE}_after_3 &&
669                 quota_error a $TSTUSR "create after timer expired," \
670                         "but expect EDQUOT"
671         sync; sleep 1; sync
672
673         $SHOW_QUOTA_USER
674         $SHOW_QUOTA_GROUP
675         $SHOW_QUOTA_INFO_USER
676         $SHOW_QUOTA_INFO_GROUP
677
678         echo "Unlink files to stop timer"
679         find $(dirname $TESTFILE) -name "$(basename ${TESTFILE})*" | xargs rm -f
680         wait_delete_completed
681
682         echo "Create file"
683         $RUNAS touch ${TESTFILE}_xxx ||
684                 quota_error a $TSTUSR "touch after timer stop failure," \
685                         "but expect success"
686         sync; sleep 1; sync
687
688         # cleanup
689         cleanup_quota_test
690 }
691
692 # file soft limit
693 test_4a() {
694         local LIMIT=10 # inodes
695         local TESTFILE=$DIR/$tdir/$tfile-0
696         local GRACE=5
697
698         set_mdt_qtype "ug" || error "enable mdt quota failed"
699
700         echo "User quota (soft limit:$LIMIT files  grace:$GRACE seconds)"
701         # make sure the system is clean
702         local USED=$(getquota -u $TSTUSR global curinodes)
703         [ $USED -ne 0 ] && error "Used space($USED) for user $TSTUSR isn't 0."
704
705         $LFS setquota -t -u --block-grace $MAX_DQ_TIME --inode-grace $GRACE $DIR
706         $LFS setquota -u $TSTUSR -b 0 -B 0 -i $LIMIT -I 0 $DIR
707
708         test_file_soft $TESTFILE $LIMIT $GRACE
709         resetquota -u $TSTUSR
710
711         echo "Group quota (soft limit:$LIMIT files  grace:$GRACE seconds)"
712         # make sure the system is clean
713         USED=$(getquota -g $TSTUSR global curinodes)
714         [ $USED -ne 0 ] && error "Used space($USED) for group $TSTUSR isn't 0."
715
716         $LFS setquota -t -g --block-grace $MAX_DQ_TIME --inode-grace $GRACE $DIR
717         $LFS setquota -g $TSTUSR -b 0 -B 0 -i $LIMIT -I 0 $DIR
718         TESTFILE=$DIR/$tdir/$tfile-1
719
720         test_file_soft $TESTFILE $LIMIT $GRACE
721         resetquota -g $TSTUSR
722
723         # cleanup
724         $LFS setquota -t -u --block-grace $MAX_DQ_TIME --inode-grace \
725                 $MAX_IQ_TIME $DIR
726         $LFS setquota -t -g --block-grace $MAX_DQ_TIME --inode-grace \
727                 $MAX_IQ_TIME $DIR
728 }
729 run_test 4a "File soft limit (start timer, timer goes off, stop timer)"
730
731 test_4b() {
732         local GR_STR1="1w3d"
733         local GR_STR2="1000s"
734         local GR_STR3="5s"
735         local GR_STR4="1w2d3h4m5s"
736         local GR_STR5="5c"
737         local GR_STR6="1111111111111111"
738
739         wait_delete_completed
740
741         # test of valid grace strings handling
742         echo "Valid grace strings test"
743         $LFS setquota -t -u --block-grace $GR_STR1 --inode-grace $GR_STR2 $DIR
744         $LFS quota -u -t $DIR | grep "Block grace time: $GR_STR1"
745         $LFS setquota -t -g --block-grace $GR_STR3 --inode-grace $GR_STR4 $DIR
746         $LFS quota -g -t $DIR | grep "Inode grace time: $GR_STR4"
747
748         # test of invalid grace strings handling
749         echo "  Invalid grace strings test"
750         ! $LFS setquota -t -u --block-grace $GR_STR4 --inode-grace $GR_STR5 $DIR
751         ! $LFS setquota -t -g --block-grace $GR_STR4 --inode-grace $GR_STR6 $DIR
752
753         # cleanup
754         $LFS setquota -t -u --block-grace $MAX_DQ_TIME --inode-grace \
755                 $MAX_IQ_TIME $DIR
756         $LFS setquota -t -g --block-grace $MAX_DQ_TIME --inode-grace \
757                 $MAX_IQ_TIME $DIR
758 }
759 run_test 4b "Grace time strings handling"
760
761 # chown & chgrp (chown & chgrp successfully even out of block/file quota)
762 test_5() {
763         local BLIMIT=10 # 10M
764         local ILIMIT=10 # 10 inodes
765
766         setup_quota_test
767         trap cleanup_quota_test EXIT
768
769         set_mdt_qtype "ug" || error "enable mdt quota failed"
770         set_ost_qtype "ug" || error "enable ost quota failed"
771
772         echo "Set quota limit (0 ${BLIMIT}M 0 $ILIMIT) for $TSTUSR.$TSTUSR"
773         $LFS setquota -u $TSTUSR -b 0 -B ${BLIMIT}M -i 0 -I $ILIMIT $DIR
774         $LFS setquota -g $TSTUSR -b 0 -B ${BLIMIT}M -i 0 -I $ILIMIT $DIR
775
776         # make sure the system is clean
777         local USED=$(getquota -u $TSTUSR global curinodes)
778         [ $USED -ne 0 ] && error "Used inode($USED) for user $TSTUSR isn't 0."
779         USED=$(getquota -g $TSTUSR global curinodes)
780         [ $USED -ne 0 ] && error "Used inode($USED) for group $TSTUSR isn't 0."
781         USED=$(getquota -u $TSTUSR global curspace)
782         [ $USED -ne 0 ] && error "Used block($USED) for user $TSTUSR isn't 0."
783         USED=$(getquota -g $TSTUSR global curspace)
784         [ $USED -ne 0 ] && error "Used block($USED) for group $TSTUSR isn't 0."
785
786         echo "Create more than $ILIMIT files and more than $BLIMIT MB ..."
787         createmany -m $DIR/$tdir/$tfile-0_ $((ILIMIT + 1)) ||
788                 error "create failure, expect success"
789         $DD of=$DIR/$tdir/$tfile-0_1 count=$((BLIMIT+1)) ||
790                 error "write failure, expect success"
791
792         echo "Chown files to $TSTUSR.$TSTUSR ..."
793         for i in `seq 0 $ILIMIT`; do
794                 chown $TSTUSR.$TSTUSR $DIR/$tdir/$tfile-0_$i ||
795                         quota_error a $TSTUSR "chown failure, expect success"
796         done
797
798         # cleanup
799         unlinkmany $DIR/$tdir/$tfile-0_ $((ILIMIT + 1))
800         cleanup_quota_test
801
802         resetquota -u $TSTUSR
803         resetquota -g $TSTUSR
804 }
805 run_test 5 "Chown & chgrp successfully even out of block/file quota"
806
807 # test dropping acquire request on master
808 test_6() {
809         local LIMIT=3 # 3M
810
811         setup_quota_test
812         trap cleanup_quota_test EXIT
813
814         # make sure the system is clean
815         local USED=$(getquota -u $TSTUSR global curspace)
816         [ $USED -ne 0 ] && error "Used space($USED) for user $TSTUSR isn't 0."
817
818         # make sure no granted quota on ost
819         set_ost_qtype "ug" || error "enable ost quota failed"
820         resetquota -u $TSTUSR
821
822         # create file for $TSTUSR
823         local TESTFILE=$DIR/$tdir/$tfile-$TSTUSR
824         $LFS setstripe $TESTFILE -c 1 -i 0
825         chown $TSTUSR.$TSTUSR $TESTFILE
826
827         # create file for $TSTUSR2
828         local TESTFILE2=$DIR/$tdir/$tfile-$TSTUSR2
829         $LFS setstripe $TESTFILE2 -c 1 -i 0
830         chown $TSTUSR2.$TSTUSR2 $TESTFILE2
831
832         # cache per-ID lock for $TSTUSR on slave
833         $LFS setquota -u $TSTUSR -b 0 -B ${LIMIT}M -i 0 -I 0 $DIR
834         $RUNAS $DD of=$TESTFILE count=1 ||
835                 error "write $TESTFILE failure, expect success"
836         $RUNAS2 $DD of=$TESTFILE2 count=1 ||
837                 error "write $TESTFILE2 failure, expect success"
838         sync; sync
839         sync_all_data || true
840
841         #define QUOTA_DQACQ 601
842         #define OBD_FAIL_PTLRPC_DROP_REQ_OPC 0x513
843         lustre_fail mds 0x513 601
844
845         # write to un-enforced ID ($TSTUSR2) should succeed
846         $RUNAS2 $DD of=$TESTFILE2 count=$LIMIT seek=1 oflag=sync conv=notrunc ||
847                 error "write failure, expect success"
848
849         # write to enforced ID ($TSTUSR) in background, exceeding limit
850         # to make sure DQACQ is sent
851         $RUNAS $DD of=$TESTFILE count=$LIMIT seek=1 oflag=sync conv=notrunc &
852         DDPID=$!
853
854         # watchdog timer uses a factor of 2
855         echo "Sleep for $((TIMEOUT * 2 + 1)) seconds ..."
856         sleep $((TIMEOUT * 2 + 1))
857
858         # write should be blocked and never finished
859         if ! ps -p $DDPID  > /dev/null 2>&1; then
860                 lustre_fail mds 0 0
861                 error "write finished incorrectly!"
862         fi
863
864         lustre_fail mds 0 0
865
866         # no watchdog is triggered
867         do_facet ost1 dmesg > $TMP/lustre-log-${TESTNAME}.log
868         watchdog=$(awk '/sanity-quota test 6/ {start = 1;}
869                        /Service thread pid/ && /was inactive/ {
870                                if (start) {
871                                        print;
872                                }
873                        }' $TMP/lustre-log-${TESTNAME}.log)
874         [ -z "$watchdog" ] || error "$watchdog"
875
876         rm -f $TMP/lustre-log-${TESTNAME}.log
877
878         # write should continue & succeed
879         local count=0
880         while [ true ]; do
881                 if ! ps -p ${DDPID} > /dev/null 2>&1; then break; fi
882                 if [ $count -ge 120 ]; then
883                         quota_error u $TSTUSR "dd not finished in $count secs"
884                 fi
885                 count=$((count + 1))
886                 [ $((count % 10)) -eq 0 ] && echo "Waiting $count secs"
887                 sleep 1
888         done
889
890         cleanup_quota_test
891         resetquota -u $TSTUSR
892 }
893 run_test 6 "Test dropping acquire request on master"
894
895 # quota reintegration (global index)
896 test_7a() {
897         local TESTFILE=$DIR/$tdir/$tfile
898         local LIMIT=20 # 20M
899
900         [ "$SLOW" = "no" ] && LIMIT=5
901
902         setup_quota_test
903         trap cleanup_quota_test EXIT
904
905         # make sure the system is clean
906         local USED=$(getquota -u $TSTUSR global curspace)
907         [ $USED -ne 0 ] && error "Used space($USED) for user $TSTUSR isn't 0."
908
909         # make sure no granted quota on ost1
910         set_ost_qtype "ug" || error "enable ost quota failed"
911         resetquota -u $TSTUSR
912         set_ost_qtype "none" || error "disable ost quota failed"
913
914         local OSTUUID=$(ostuuid_from_index 0)
915         USED=$(getquota -u $TSTUSR $OSTUUID bhardlimit)
916         [ $USED -ne 0 ] && error "limit($USED) on $OSTUUID for user" \
917                 "$TSTUSR isn't 0."
918
919         # create test file
920         $LFS setstripe $TESTFILE -c 1 -i 0
921         chown $TSTUSR.$TSTUSR $TESTFILE
922
923         echo "Stop ost1..."
924         stop ost1
925
926         echo "Enable quota & set quota limit for $TSTUSR"
927         set_ost_qtype "ug" || error "enable ost quota failed"
928         $LFS setquota -u $TSTUSR -b 0 -B ${LIMIT}M -i 0 -I 0 $DIR
929
930         echo "Start ost1..."
931         start ost1 $(ostdevname 1) $OST_MOUNT_OPTS
932
933         wait_ost_reint "ug" || error "reintegration failed"
934
935         # hardlimit should have been fetched by slave during global
936         # reintegration, write will exceed quota
937         $RUNAS $DD of=$TESTFILE count=$((LIMIT + 1)) oflag=sync &&
938                 quota_error u $TSTUSR "write success, but expect EDQUOT"
939
940         cleanup_quota_test
941         resetquota -u $TSTUSR
942 }
943 run_test 7a "Quota reintegration (global index)"
944
945 # quota reintegration (slave index)
946 test_7b() {
947         local LIMIT="100G"
948         local TESTFILE=$DIR/$tdir/$tfile
949
950         setup_quota_test
951         trap cleanup_quota_test EXIT
952
953         # make sure the system is clean
954         local USED=$(getquota -u $TSTUSR global curspace)
955         [ $USED -ne 0 ] && error "Used space($USED) for user $TSTUSR isn't 0."
956
957         # make sure no granted quota on ost1
958         set_ost_qtype "ug" || error "enable ost quota failed"
959         resetquota -u $TSTUSR
960         set_ost_qtype "none" || error "disable ost quota failed"
961
962         local OSTUUID=$(ostuuid_from_index 0)
963         USED=$(getquota -u $TSTUSR $OSTUUID bhardlimit)
964         [ $USED -ne 0 ] && error "limit($USED) on $OSTUUID for user" \
965                 "$TSTUSR isn't 0."
966
967         # create test file
968         $LFS setstripe $TESTFILE -c 1 -i 0
969         chown $TSTUSR.$TSTUSR $TESTFILE
970
971         # consume some space to make sure the granted space will not
972         # be released during reconciliation
973         $RUNAS $DD of=$TESTFILE count=1 oflag=sync ||
974                 error "consume space failure, expect success"
975
976         # define OBD_FAIL_QUOTA_EDQUOT 0xa02
977         lustre_fail mds 0xa02
978
979         set_ost_qtype "ug" || error "enable ost quota failed"
980         $LFS setquota -u $TSTUSR -b 0 -B $LIMIT -i 0 -I 0 $DIR
981
982         # ignore the write error
983         $RUNAS $DD of=$TESTFILE count=1 seek=1 oflag=sync conv=notrunc
984
985         local old_used=$(getquota -u $TSTUSR $OSTUUID bhardlimit)
986
987         lustre_fail mds 0
988
989         echo "Restart ost to trigger reintegration..."
990         stop ost1
991         start ost1 $(ostdevname 1) $OST_MOUNT_OPTS
992
993         wait_ost_reint "ug" || error "reintegration failed"
994
995         USED=$(getquota -u $TSTUSR $OSTUUID bhardlimit)
996         [ $USED -gt $old_used ] || error "limit on $OSTUUID $USED <= $old_used"
997
998         cleanup_quota_test
999         resetquota -u $TSTUSR
1000         $SHOW_QUOTA_USER
1001 }
1002 run_test 7b "Quota reintegration (slave index)"
1003
1004 # quota reintegration (restart mds during reintegration)
1005 test_7c() {
1006         local LIMIT=20 # 20M
1007         local TESTFILE=$DIR/$tdir/$tfile
1008
1009         [ "$SLOW" = "no" ] && LIMIT=5
1010
1011         setup_quota_test
1012         trap cleanup_quota_test EXIT
1013
1014         # make sure the system is clean
1015         local USED=$(getquota -u $TSTUSR global curspace)
1016         [ $USED -ne 0 ] && error "Used space($USED) for user $TSTUSR isn't 0."
1017
1018         set_ost_qtype "none" || error "disable ost quota failed"
1019         $LFS setquota -u $TSTUSR -b 0 -B ${LIMIT}M -i 0 -I 0 $DIR
1020
1021         # define OBD_FAIL_QUOTA_DELAY_REINT 0xa03
1022         lustre_fail ost 0xa03
1023
1024         # enable ost quota to trigger reintegration
1025         set_ost_qtype "ug" || error "enable ost quota failed"
1026
1027         echo "Stop mds..."
1028         stop mds1
1029
1030         lustre_fail ost 0
1031
1032         echo "Start mds..."
1033         start mds1 $(mdsdevname 1) $MDS_MOUNT_OPTS
1034
1035         wait_ost_reint "ug" || error "reintegration failed"
1036
1037         # hardlimit should have been fetched by slave during global
1038         # reintegration, write will exceed quota
1039         $RUNAS $DD of=$TESTFILE count=$((LIMIT + 1)) oflag=sync &&
1040                 quota_error u $TSTUSR "write success, but expect EDQUOT"
1041
1042         cleanup_quota_test
1043         resetquota -u $TSTUSR
1044 }
1045 run_test 7c "Quota reintegration (restart mds during reintegration)"
1046
1047 # run dbench with quota enabled
1048 test_8() {
1049         local BLK_LIMIT="100g" #100G
1050         local FILE_LIMIT=1000000
1051
1052         setup_quota_test
1053         trap cleanup_quota_test EXIT
1054
1055         set_mdt_qtype "ug" || error "enable mdt quota failed"
1056         set_ost_qtype "ug" || error "enable ost quota failed"
1057
1058         echo "Set enough high limit for user: $TSTUSR"
1059         $LFS setquota -u $TSTUSR -b 0 -B $BLK_LIMIT -i 0 -I $FILE_LIMIT $DIR
1060         echo "Set enough high limit for group: $TSTUSR"
1061         $LFS setquota -g $TSTUSR -b 0 -B $BLK_LIMIT -i 0 -I $FILE_LIMIT $DIR
1062
1063         local duration=""
1064         [ "$SLOW" = "no" ] && duration=" -t 120"
1065         $RUNAS bash rundbench -D $DIR/$tdir 3 $duration ||
1066                 quota_error a $TSTUSR "dbench failed!"
1067
1068         cleanup_quota_test
1069         resetquota -u $TSTUSR
1070         resetquota -g $TSTUSR
1071 }
1072 run_test 8 "Run dbench with quota enabled"
1073
1074 # this check is just for test_9
1075 OST0_MIN=4900000 #4.67G
1076
1077 check_whether_skip () {
1078         local OST0_SIZE=$($LFS df $DIR | awk '/\[OST:0\]/ {print $4}')
1079         log "OST0_SIZE: $OST0_SIZE  required: $OST0_MIN"
1080         if [ $OST0_SIZE -lt $OST0_MIN ]; then
1081                 echo "WARN: OST0 has less than $OST0_MIN free, skip this test."
1082                 return 0
1083         else
1084                 return 1
1085         fi
1086 }
1087
1088 # run for fixing bug10707, it needs a big room. test for 64bit
1089 test_9() {
1090         local filesize=$((1024 * 1024 * 1024 * 9 / 2)) # 4.5G
1091
1092         check_whether_skip && return 0
1093
1094         setup_quota_test
1095         trap cleanup_quota_test EXIT
1096
1097         set_ost_qtype "ug" || error "enable ost quota failed"
1098
1099         local TESTFILE="$DIR/$tdir/$tfile-0"
1100         local BLK_LIMIT=100G #100G
1101         local FILE_LIMIT=1000000
1102
1103         echo "Set block limit $BLK_LIMIT bytes to $TSTUSR.$TSTUSR"
1104
1105         log "Set enough high limit(block:$BLK_LIMIT; file: $FILE_LIMIT)" \
1106                 "for user: $TSTUSR"
1107         $LFS setquota -u $TSTUSR -b 0 -B $BLK_LIMIT -i 0 -I $FILE_LIMIT $DIR
1108
1109         log "Set enough high limit(block:$BLK_LIMIT; file: $FILE_LIMIT)" \
1110                 "for group: $TSTUSR"
1111         $LFS setquota -g $TSTUSR -b 0 -B $BLK_LIMIT -i 0 -I $FILE_LIMIT $DIR
1112
1113         quota_show_check a u $TSTUSR
1114         quota_show_check a g $TSTUSR
1115
1116         echo "Create test file"
1117         $LFS setstripe $TESTFILE -c 1 -i 0
1118         chown $TSTUSR.$TSTUSR $TESTFILE
1119
1120         log "Write the big file of 4.5G ..."
1121         $RUNAS $DD of=$TESTFILE count=$filesize ||
1122                 quota_error a $TSTUSR "write 4.5G file failure, expect success"
1123
1124         $SHOW_QUOTA_USER
1125         $SHOW_QUOTA_GROUP
1126
1127         cleanup_quota_test
1128         resetquota -u $TSTUSR
1129         resetquota -g $TSTUSR
1130
1131         $SHOW_QUOTA_USER
1132         $SHOW_QUOTA_GROUP
1133 }
1134 run_test 9 "Block limit larger than 4GB (b10707)"
1135
1136 test_10() {
1137         local TESTFILE=$DIR/$tdir/$tfile
1138
1139         setup_quota_test
1140         trap cleanup_quota_test EXIT
1141
1142         # set limit to root user should fail
1143         $LFS setquota -u root -b 100G -B 500G -i 1K -I 1M $DIR &&
1144                 error "set limit for root user successfully, expect failure"
1145         $LFS setquota -g root -b 1T -B 10T -i 5K -I 100M $DIR &&
1146                 error "set limit for root group successfully, expect failure"
1147
1148         # root user can overrun quota
1149         set_ost_qtype "ug" || "enable ost quota failed"
1150
1151         $LFS setquota -u $TSTUSR -b 0 -B 2M -i 0 -I 0 $DIR
1152         quota_show_check b u $TSTUSR
1153
1154         $LFS setstripe $TESTFILE -c 1
1155         chown $TSTUSR.$TSTUSR $TESTFILE
1156
1157         runas -u 0 -g 0 $DD of=$TESTFILE count=3 oflag=sync ||
1158                 error "write failure, expect success"
1159
1160         cleanup_quota_test
1161         resetquota -u $TSTUSR
1162 }
1163 run_test 10 "Test quota for root user"
1164
1165 test_11() {
1166         local TESTFILE=$DIR/$tdir/$tfile
1167         setup_quota_test
1168         trap cleanup_quota_test EXIT
1169
1170         set_mdt_qtype "ug" || "enable mdt quota failed"
1171         $LFS setquota -u $TSTUSR -b 0 -B 0 -i 0 -I 1 $DIR
1172
1173         touch "$TESTFILE"-0
1174         touch "$TESTFILE"-1
1175
1176         chown $TSTUSR.$TSTUSR "$TESTFILE"-0
1177         chown $TSTUSR.$TSTUSR "$TESTFILE"-1
1178
1179         $SHOW_QUOTA_USER
1180         local USED=$(getquota -u $TSTUSR global curinodes)
1181         [ $USED -ge 2 ] || error "Used inodes($USED) is less than 2"
1182
1183         cleanup_quota_test
1184         resetquota -u $TSTUSR
1185 }
1186 run_test 11 "Chown/chgrp ignores quota"
1187
1188 test_12() {
1189         [ "$OSTCOUNT" -lt "2" ] && skip "skipping rebalancing test" && return
1190
1191         local blimit=22 # 22M
1192         local blk_cnt=$((blimit - 5))
1193         local TESTFILE0="$DIR/$tdir/$tfile"-0
1194         local TESTFILE1="$DIR/$tdir/$tfile"-1
1195
1196         setup_quota_test
1197         trap cleanup_quota_test EXIT
1198
1199         set_ost_qtype "u" || "enable ost quota failed"
1200         quota_show_check b u $TSTUSR
1201
1202         $LFS setquota -u $TSTUSR -b 0 -B "$blimit"M -i 0 -I 0 $DIR
1203
1204         $LFS setstripe $TESTFILE0 -c 1 -i 0
1205         $LFS setstripe $TESTFILE1 -c 1 -i 1
1206         chown $TSTUSR.$TSTUSR $TESTFILE0
1207         chown $TSTUSR.$TSTUSR $TESTFILE1
1208
1209         echo "Write to ost0..."
1210         $RUNAS $DD of=$TESTFILE0 count=$blk_cnt oflag=sync ||
1211                 quota_error a $TSTUSR "dd failed"
1212
1213         echo "Write to ost1..."
1214         $RUNAS $DD of=$TESTFILE1 count=$blk_cnt oflag=sync &&
1215                 quota_error a $TSTUSR "dd succeed, expect EDQUOT"
1216
1217         echo "Free space from ost0..."
1218         rm -f $TESTFILE0
1219         wait_delete_completed
1220         sync_all_data || true
1221
1222         echo "Write to ost1 after space freed from ost0..."
1223         $RUNAS $DD of=$TESTFILE1 count=$blk_cnt oflag=sync ||
1224                 quota_error a $TSTUSR "rebalancing failed"
1225
1226         cleanup_quota_test
1227         resetquota -u $TSTUSR
1228 }
1229 run_test 12 "Block quota rebalancing"
1230
1231 test_13(){
1232         local TESTFILE=$DIR/$tdir/$tfile
1233         # the name of osp on ost1 name is MDT0000-osp-OST0000
1234         local procf="ldlm.namespaces.*MDT0000-osp-OST0000.lru_size"
1235
1236         setup_quota_test
1237         trap cleanup_quota_test EXIT
1238
1239         set_ost_qtype "u" || "enable ost quota failed"
1240         quota_show_check b u $TSTUSR
1241
1242         $LFS setquota -u $TSTUSR -b 0 -B 10M -i 0 -I 0 $DIR
1243         $LFS setstripe $TESTFILE -c 1 -i 0
1244         chown $TSTUSR.$TSTUSR $TESTFILE
1245
1246         # clear the locks in cache first
1247         do_facet ost1 $LCTL set_param -n $procf=clear
1248         local nlock=$(do_facet ost1 $LCTL get_param -n $procf)
1249         [ $nlock -eq 0 ] || error "$nlock cached locks"
1250
1251         # write to acquire the per-ID lock
1252         $RUNAS $DD of=$TESTFILE count=1 oflag=sync ||
1253                 quota_error a $TSTUSR "dd failed"
1254
1255         nlock=$(do_facet ost1 $LCTL get_param -n $procf)
1256         [ $nlock -eq 1 ] || error "lock count($nlock) isn't 1"
1257
1258         # clear quota doesn't trigger per-ID lock cancellation
1259         resetquota -u $TSTUSR
1260         nlock=$(do_facet ost1 $LCTL get_param -n $procf)
1261         [ $nlock -eq 1 ] || error "per-ID lock is lost on quota clear"
1262
1263         # clear the per-ID lock
1264         do_facet ost1 $LCTL set_param -n $procf=clear
1265         nlock=$(do_facet ost1 $LCTL get_param -n $procf)
1266         [ $nlock -eq 0 ] || error "per-ID lock isn't cleared"
1267
1268         # spare quota should be released
1269         local OSTUUID=$(ostuuid_from_index 0)
1270         local limit=$(getquota -u $TSTUSR $OSTUUID bhardlimit)
1271         local space=$(getquota -u $TSTUSR $OSTUUID curspace)
1272         [ $limit -le $space ] ||
1273                 error "spare quota isn't released, limit:$limit, space:$space"
1274
1275         cleanup_quota_test
1276 }
1277 run_test 13 "Cancel per-ID lock in the LRU list"
1278
1279 test_15(){
1280         local LIMIT=$((24 * 1024 * 1024 * 1024 * 1024)) # 24 TB
1281
1282         wait_delete_completed
1283         sync_all_data || true
1284
1285         # test for user
1286         $LFS setquota -u $TSTUSR -b 0 -B $LIMIT -i 0 -I 0 $DIR
1287         local TOTAL_LIMIT=$(getquota -u $TSTUSR global bhardlimit)
1288         [ $TOTAL_LIMIT -eq $LIMIT ] ||
1289                 error "(user) limit:$TOTAL_LIMIT, expect:$LIMIT, failed!"
1290         resetquota -u $TSTUSR
1291
1292         # test for group
1293         $LFS setquota -g $TSTUSR -b 0 -B $LIMIT -i 0 -I 0 $DIR
1294         TOTAL_LIMIT=$(getquota -g $TSTUSR global bhardlimit)
1295         [ $TOTAL_LIMIT -eq $LIMIT ] ||
1296                 error "(group) limits:$TOTAL_LIMIT, expect:$LIMIT, failed!"
1297         resetquota -g $TSTUSR
1298 }
1299 run_test 15 "Set over 4T block quota"
1300
1301 test_17sub() {
1302         local err_code=$1
1303         local BLKS=1    # 1M less than limit
1304         local TESTFILE=$DIR/$tdir/$tfile
1305
1306         setup_quota_test
1307         trap cleanup_quota_test EXIT
1308
1309         # make sure the system is clean
1310         local USED=$(getquota -u $TSTUSR global curspace)
1311         [ $USED -ne 0 ] && error "Used space($USED) for user $TSTUSR isn't 0."
1312
1313         set_ost_qtype "ug" || error "enable ost quota failed"
1314         # make sure no granted quota on ost
1315         resetquota -u $TSTUSR
1316         $LFS setquota -u $TSTUSR -b 0 -B 10M -i 0 -I 0 $DIR
1317
1318         quota_show_check b u $TSTUSR
1319
1320         #define OBD_FAIL_QUOTA_RECOVERABLE_ERR 0xa04
1321         lustre_fail mds 0xa04 $err_code
1322
1323         # write in background
1324         $RUNAS $DD of=$TESTFILE count=$BLKS oflag=direct &
1325         local DDPID=$!
1326
1327         sleep 2
1328         # write should be blocked and never finished
1329         if ! ps -p $DDPID  > /dev/null 2>&1; then
1330                 lustre_fail mds 0 0
1331                 quota_error u $TSTUSR "write finished incorrectly!"
1332         fi
1333
1334         lustre_fail mds 0 0
1335
1336         local count=0
1337         local timeout=30
1338         while [ true ]; do
1339                 if ! ps -p ${DDPID} > /dev/null 2>&1; then break; fi
1340                 count=$((count+1))
1341                 if [ $count -gt $timeout ]; then
1342                         quota_error u $TSTUSR "dd is not finished!"
1343                 fi
1344                 sleep 1
1345         done
1346
1347         sync; sync_all_data || true
1348
1349         USED=$(getquota -u $TSTUSR global curspace)
1350         [ $USED -ge $(($BLKS * 1024)) ] || quota_error u $TSTUSR \
1351                 "Used space(${USED}K) is less than ${BLKS}M"
1352
1353         cleanup_quota_test
1354         resetquota -u $TSTUSR
1355 }
1356
1357 # DQACQ return recoverable error
1358 test_17() {
1359         echo "DQACQ return -ENOLCK"
1360         #define ENOLCK  37
1361         test_17sub 37 || error "Handle -ENOLCK failed"
1362
1363         echo "DQACQ return -EAGAIN"
1364         #define EAGAIN  11
1365         test_17sub 11 || error "Handle -EAGAIN failed"
1366
1367         echo "DQACQ return -ETIMEDOUT"
1368         #define ETIMEDOUT 110
1369         test_17sub 110 || error "Handle -ETIMEDOUT failed"
1370
1371         echo "DQACQ return -ENOTCONN"
1372         #define ENOTCONN 107
1373         test_17sub 107 || error "Handle -ENOTCONN failed"
1374 }
1375
1376 run_test 17 "DQACQ return recoverable error"
1377
1378 test_18_sub () {
1379         local io_type=$1
1380         local blimit="200m" # 200M
1381         local TESTFILE="$DIR/$tdir/$tfile"
1382
1383         setup_quota_test
1384         trap cleanup_quota_test EXIT
1385
1386         set_ost_qtype "u" || error "enable ost quota failed"
1387         log "User quota (limit: $blimit)"
1388         $LFS setquota -u $TSTUSR -b 0 -B $blimit -i 0 -I 0 $MOUNT
1389         quota_show_check b u $TSTUSR
1390
1391         $LFS setstripe $TESTFILE -i 0 -c 1
1392         chown $TSTUSR.$TSTUSR $TESTFILE
1393
1394         local timeout=$(sysctl -n lustre.timeout)
1395
1396         if [ $io_type = "directio" ]; then
1397                 log "Write 100M (directio) ..."
1398                 $RUNAS $DD of=$TESTFILE count=100 oflag=direct &
1399         else
1400                 log "Write 100M (buffered) ..."
1401                 $RUNAS $DD of=$TESTFILE count=100 &
1402         fi
1403         local DDPID=$!
1404
1405         replay_barrier $SINGLEMDS
1406         log "Fail mds for $((2 * timeout)) seconds"
1407         fail $SINGLEMDS $((2 * timeout))
1408
1409         local count=0
1410         if at_is_enabled; then
1411                 timeout=$(at_max_get mds)
1412         else
1413                 timeout=$(lctl get_param -n timeout)
1414         fi
1415
1416         while [ true ]; do
1417                 if ! ps -p ${DDPID} > /dev/null 2>&1; then break; fi
1418                 if [ $((++count % (2 * timeout) )) -eq 0 ]; then
1419                         log "it took $count second"
1420                 fi
1421                 sleep 1
1422         done
1423
1424         log "(dd_pid=$DDPID, time=$count, timeout=$timeout)"
1425         sync
1426         cancel_lru_locks mdc
1427         cancel_lru_locks osc
1428         $SHOW_QUOTA_USER
1429
1430         local testfile_size=$(stat -c %s $TESTFILE)
1431         if [ $testfile_size -ne $((BLK_SZ * 1024 * 100)) ] ; then
1432                 quota_error u $TSTUSR "expect $((BLK_SZ * 1024 * 100)),"
1433                         "got ${testfile_size}. Verifying file failed!"
1434         fi
1435         cleanup_quota_test
1436         resetquota -u $TSTUSR
1437 }
1438
1439 # test when mds does failover, the ost still could work well
1440 # this test shouldn't trigger watchdog b=14840
1441 test_18() {
1442         test_18_sub normal
1443         test_18_sub directio
1444
1445         # check if watchdog is triggered
1446         do_facet ost1 dmesg > $TMP/lustre-log-${TESTNAME}.log
1447         local watchdog=$(awk '/test 18b/ {start = 1;}
1448                        /Service thread pid/ && /was inactive/ {
1449                                if (start) {
1450                                        print;
1451                                }
1452                        }' $TMP/lustre-log-${TESTNAME}.log)
1453         [ -z "$watchdog" ] || error "$watchdog"
1454         rm -f $TMP/lustre-log-${TESTNAME}.log
1455 }
1456 run_test 18 "MDS failover while writing, no watchdog triggered (b14840)"
1457
1458 test_19() {
1459         local blimit=5 # 5M
1460         local TESTFILE=$DIR/$tdir/$tfile
1461
1462         setup_quota_test
1463         trap cleanup_quota_test EXIT
1464
1465         set_ost_qtype "ug" || error "enable ost quota failed"
1466
1467         # bind file to a single OST
1468         $LFS setstripe -c 1 $TESTFILE
1469         chown $TSTUSR.$TSTUSR $TESTFILE
1470
1471         echo "Set user quota (limit: "$blimit"M)"
1472         $LFS setquota -u $TSTUSR -b 0 -B "$blimit"M -i 0 -I 0 $MOUNT
1473         quota_show_check b u $TSTUSR
1474         echo "Update quota limits"
1475         $LFS setquota -u $TSTUSR -b 0 -B "$blimit"M -i 0 -I 0 $MOUNT
1476         quota_show_check b u $TSTUSR
1477
1478         # first wirte might be cached
1479         $RUNAS $DD of=$TESTFILE count=$((blimit + 1))
1480         cancel_lru_locks osc
1481         $SHOW_QUOTA_USER
1482         $RUNAS $DD of=$TESTFILE count=$((blimit + 1)) seek=$((blimit + 1)) &&
1483                 quota_error u $TSTUSR "Write success, expect failure"
1484         $SHOW_QUOTA_USER
1485
1486         cleanup_quota_test
1487         resetquota -u $TSTUSR
1488 }
1489 run_test 19 "Updating admin limits doesn't zero operational limits(b14790)"
1490
1491 test_20() { # b15754
1492         local LSTR=(2g 1t 4k 3m) # limits strings
1493         # limits values
1494         local LVAL=($[2*1024*1024] $[1*1024*1024*1024] $[4*1024] $[3*1024*1024])
1495
1496         resetquota -u $TSTUSR
1497
1498         $LFS setquota -u $TSTUSR --block-softlimit ${LSTR[0]} \
1499                 $MOUNT || error "could not set quota limits"
1500         $LFS setquota -u $TSTUSR --block-hardlimit ${LSTR[1]} \
1501                                 --inode-softlimit ${LSTR[2]} \
1502                                 --inode-hardlimit ${LSTR[3]} \
1503                                 $MOUNT || error "could not set quota limits"
1504
1505         [ "$(getquota -u $TSTUSR global bsoftlimit)" = "${LVAL[0]}" ] ||
1506                 error "bsoftlimit was not set properly"
1507         [ "$(getquota -u $TSTUSR global bhardlimit)" = "${LVAL[1]}" ] ||
1508                 error "bhardlimit was not set properly"
1509         [ "$(getquota -u $TSTUSR global isoftlimit)" = "${LVAL[2]}" ] ||
1510                 error "isoftlimit was not set properly"
1511         [ "$(getquota -u $TSTUSR global ihardlimit)" = "${LVAL[3]}" ] ||
1512                 error "ihardlimit was not set properly"
1513
1514         resetquota -u $TSTUSR
1515 }
1516 run_test 20 "Test if setquota specifiers work properly (b15754)"
1517
1518 test_21_sub() {
1519         local testfile=$1
1520         local blk_number=$2
1521         local seconds=$3
1522
1523         local time=$(($(date +%s) + seconds))
1524         while [ $(date +%s) -lt $time ]; do
1525                 $RUNAS $DD of=$testfile count=$blk_number > /dev/null 2>&1
1526         done
1527 }
1528
1529 # run for fixing bug16053, setquota shouldn't fail when writing and
1530 # deleting are happening
1531 test_21() {
1532         local TESTFILE="$DIR/$tdir/$tfile"
1533         local BLIMIT=10 # 10G
1534         local ILIMIT=1000000
1535
1536         setup_quota_test
1537         trap cleanup_quota_test EXIT
1538
1539         set_ost_qtype "ug" || error "Enable ost quota failed"
1540
1541         log "Set limit(block:${BLIMIT}G; file:$ILIMIT) for user: $TSTUSR"
1542         $LFS setquota -u $TSTUSR -b 0 -B ${BLIMIT}G -i 0 -I $ILIMIT $MOUNT
1543         log "Set limit(block:${BLIMIT}G; file:$ILIMIT) for group: $TSTUSR"
1544         $LFS setquota -g $TSTUSR -b 0 -B $BLIMIT -i 0 -I $ILIMIT $MOUNT
1545
1546         # repeat writing on a 1M file
1547         test_21_sub ${TESTFILE}_1 1 30 &
1548         local DDPID1=$!
1549         # repeat writing on a 128M file
1550         test_21_sub ${TESTFILE}_2 128 30 &
1551         local DDPID2=$!
1552
1553         local time=$(($(date +%s) + 30))
1554         local i=1
1555         while [ $(date +%s) -lt $time ]; do
1556                 log "Set quota for $i times"
1557                 $LFS setquota -u $TSTUSR -b 0 -B "$((BLIMIT + i))G" -i 0 \
1558                         -I $((ILIMIT + i)) $MOUNT ||
1559                                 error "Set user quota failed"
1560                 $LFS setquota -g $TSTUSR -b 0 -B "$((BLIMIT + i))G" -i 0 \
1561                         -I $((ILIMIT + i)) $MOUNT ||
1562                                 error "Set group quota failed"
1563                 i=$((i+1))
1564                 sleep 1
1565         done
1566
1567         local count=0
1568         while [ true ]; do
1569                 if ! ps -p ${DDPID1} > /dev/null 2>&1; then break; fi
1570                 count=$((count+1))
1571                 if [ $count -gt 60 ]; then
1572                         quota_error a $TSTUSR "dd should be finished!"
1573                 fi
1574                 sleep 1
1575         done
1576         echo "(dd_pid=$DDPID1, time=$count)successful"
1577
1578         count=0
1579         while [ true ]; do
1580                 if ! ps -p ${DDPID2} > /dev/null 2>&1; then break; fi
1581                 count=$((count+1))
1582                 if [ $count -gt 60 ]; then
1583                         quota_error a $TSTUSR "dd should be finished!"
1584                 fi
1585                 sleep 1
1586         done
1587         echo "(dd_pid=$DDPID2, time=$count)successful"
1588
1589         cleanup_quota_test
1590         resetquota -u $TSTUSR
1591         resetquota -g $TSTUSR
1592 }
1593 run_test 21 "Setquota while writing & deleting (b16053)"
1594
1595 # enable/disable quota enforcement permanently
1596 test_22() {
1597         echo "Set both mdt & ost quota type as ug"
1598         set_mdt_qtype "ug" || error "enable mdt quota failed"
1599         set_ost_qtype "ug" || error "enable ost quota failed"
1600
1601         echo "Restart..."
1602         stopall
1603         mount
1604         setupall
1605
1606         echo "Verify if quota is enabled"
1607         local qtype=$(mdt_quota_type)
1608         [ $qtype != "ug" ] && error "mdt quota setting is lost"
1609         qtype=$(ost_quota_type)
1610         [ $qtype != "ug" ] && error "ost quota setting is lost"
1611
1612         echo "Set both mdt & ost quota type as none"
1613         set_mdt_qtype "none" || error "disable mdt quota failed"
1614         set_ost_qtype "none" || error "disable ost quota failed"
1615
1616         echo "Restart..."
1617         stopall
1618         mount
1619         setupall
1620
1621         echo "Verify if quota is disabled"
1622         qtype=$(mdt_quota_type)
1623         [ $qtype != "none" ] && error "mdt quota setting is lost"
1624         qtype=$(ost_quota_type)
1625         [ $qtype != "none" ] && error "ost quota setting is lost"
1626
1627         quota_init
1628 }
1629 run_test 22 "enable/disable quota by 'lctl conf_param'"
1630
1631 test_23_sub() {
1632         local TESTFILE="$DIR/$tdir/$tfile"
1633         local LIMIT=$1
1634
1635         setup_quota_test
1636         trap cleanup_quota_test EXIT
1637
1638         set_ost_qtype "ug" || error "Enable ost quota failed"
1639
1640         # test for user
1641         log "User quota (limit: $LIMIT MB)"
1642         $LFS setquota -u $TSTUSR -b 0 -B "$LIMIT"M -i 0 -I 0 $DIR
1643         quota_show_check b u $TSTUSR
1644
1645         $LFS setstripe $TESTFILE -c 1 -i 0
1646         chown $TSTUSR.$TSTUSR $TESTFILE
1647
1648         log "Step1: trigger EDQUOT with O_DIRECT"
1649         log "Write half of file"
1650         $RUNAS $DD of=$TESTFILE count=$((LIMIT/2)) oflag=direct ||
1651                 quota_error u $TSTUSR "(1) Write failure, expect success." \
1652                         "limit=$LIMIT"
1653         log "Write out of block quota ..."
1654         $RUNAS $DD of=$TESTFILE count=$((LIMIT/2 + 1)) seek=$((LIMIT/2)) \
1655                 oflag=direct conv=notrunc &&
1656                 quota_error u $TSTUSR "(2) Write success, expect EDQUOT." \
1657                         "limit=$LIMIT"
1658         log "Step1: done"
1659
1660         log "Step2: rewrite should succeed"
1661         $RUNAS $DD of=$TESTFILE count=1 oflag=direct conv=notrunc||
1662                 quota_error u $TSTUSR "(3) Write failure, expect success." \
1663                         "limit=$LIMIT"
1664         log "Step2: done"
1665
1666         cleanup_quota_test
1667
1668         local OST0_UUID=$(ostuuid_from_index 0)
1669         local OST0_QUOTA_USED=$(getquota -u $TSTUSR $OST0_UUID curspace)
1670         [ $OST0_QUOTA_USED -ne 0 ] &&
1671                 ($SHOW_QUOTA_USER; \
1672                 quota_error u $TSTUSR "quota isn't released")
1673         $SHOW_QUOTA_USER
1674         resetquota -u $TSTUSR
1675 }
1676
1677 test_23() {
1678         local OST0_MIN=$((6 * 1024)) # 6MB, extra space for meta blocks.
1679         check_whether_skip && return 0
1680         log "run for 4MB test file"
1681         test_23_sub 4
1682
1683         OST0_MIN=$((60 * 1024)) # 60MB, extra space for meta blocks.
1684         check_whether_skip && return 0
1685         log "run for 40MB test file"
1686         test_23_sub 40
1687 }
1688 run_test 23 "Quota should be honored with directIO (b16125)"
1689
1690 test_24() {
1691         local blimit=5 # 5M
1692         local TESTFILE="$DIR/$tdir/$tfile"
1693
1694         setup_quota_test
1695         trap cleanup_quota_test EXIT
1696
1697         set_ost_qtype "ug" || error "enable ost quota failed"
1698
1699         # bind file to a single OST
1700         $LFS setstripe -c 1 $TESTFILE
1701         chown $TSTUSR.$TSTUSR $TESTFILE
1702
1703         echo "Set user quota (limit: "$blimit"M)"
1704         $LFS setquota -u $TSTUSR -b 0 -B "$blimit"M -i 0 -I 0 $MOUNT
1705
1706         # overrun quota by root user
1707         runas -u 0 -g 0 $DD of=$TESTFILE count=$((blimit + 1)) ||
1708                 error "write failure, expect success"
1709         cancel_lru_locks osc
1710         sync_all_data || true
1711
1712         $SHOW_QUOTA_USER | grep '*' || error "no matching *"
1713
1714         cleanup_quota_test
1715         resetquota -u $TSTUSR
1716 }
1717 run_test 24 "lfs draws an asterix when limit is reached (b16646)"
1718
1719 test_27a() { # b19612
1720         $LFS quota $TSTUSR $DIR &&
1721                 error "lfs succeeded with no type, but should have failed"
1722         $LFS setquota $TSTUSR $DIR &&
1723                 error "lfs succeeded with no type, but should have failed"
1724         return 0
1725 }
1726 run_test 27a "lfs quota/setquota should handle wrong arguments (b19612)"
1727
1728 test_27b() { # b20200
1729         $LFS setquota -u $TSTID -b 1000 -B 1000 -i 1000 -I 1000 $DIR ||
1730                 error "lfs setquota failed with uid argument"
1731         $LFS setquota -g $TSTID -b 1000 -B 1000 -i 1000 -I 1000 $DIR ||
1732                 error "lfs stequota failed with gid argument"
1733         $SHOW_QUOTA_USERID || error "lfs quota failed with uid argument"
1734         $SHOW_QUOTA_GROUPID || error "lfs quota failed with gid argument"
1735         resetquota -u $TSTUSR
1736         resetquota -g $TSTUSR
1737         return 0
1738 }
1739 run_test 27b "lfs quota/setquota should handle user/group ID (b20200)"
1740
1741 test_30() {
1742         local output
1743         local LIMIT=4 # 4MB
1744         local TESTFILE="$DIR/$tdir/$tfile"
1745         local GRACE=10
1746
1747         setup_quota_test
1748         trap cleanup_quota_test EXIT
1749
1750         set_ost_qtype "u" || error "enable ost quota failed"
1751
1752         $LFS setstripe $TESTFILE -i 0 -c 1
1753         chown $TSTUSR.$TSTUSR $TESTFILE
1754
1755         $LFS setquota -t -u --block-grace $GRACE --inode-grace $MAX_IQ_TIME $DIR
1756         $LFS setquota -u $TSTUSR -b ${LIMIT}M -B 0 -i 0 -I 0 $DIR
1757         $RUNAS $DD of=$TESTFILE count=$((LIMIT * 2)) || true
1758         cancel_lru_locks osc
1759         sleep $GRACE
1760         $LFS setquota -u $TSTUSR -B 0 $DIR
1761         # over-quota flag has not yet settled since we do not trigger async
1762         # events based on grace time period expiration
1763         $SHOW_QUOTA_USER
1764         $RUNAS $DD of=$TESTFILE conv=notrunc oflag=append count=1 || true
1765         cancel_lru_locks osc
1766         # now over-quota flag should be settled and further writes should fail
1767         $SHOW_QUOTA_USER
1768         $RUNAS $DD of=$TESTFILE conv=notrunc oflag=append count=1 &&
1769                 error "grace times were reset"
1770         # cleanup
1771         cleanup_quota_test
1772         resetquota -u $TSTUSR
1773         $LFS setquota -t -u --block-grace $MAX_DQ_TIME --inode-grace \
1774                 $MAX_IQ_TIME $DIR
1775 }
1776 run_test 30 "Hard limit updates should not reset grace times"
1777
1778 # basic usage tracking for user & group
1779 test_33() {
1780         local INODES=10 # 10 files
1781         local BLK_CNT=2 # of 2M each
1782         local TOTAL_BLKS=$((INODES * BLK_CNT * 1024))
1783
1784         setup_quota_test
1785         trap cleanup_quota_test EXIT
1786
1787         # make sure the system is clean
1788         local USED=$(getquota -u $TSTID global curspace)
1789         [ $USED -ne 0 ] &&
1790                 error "Used space ($USED) for user $TSTID isn't 0."
1791         USED=$(getquota -g $TSTID global curspace)
1792         [ $USED -ne 0 ] &&
1793                 error "Used space ($USED) for group $TSTID isn't 0."
1794
1795         echo "Write files..."
1796         for i in `seq 0 $INODES`; do
1797                 $RUNAS $DD of=$DIR/$tdir/$tfile-$i count=$BLK_CNT 2>/dev/null ||
1798                         error "write failed"
1799                 echo "Iteration $i/$INODES completed"
1800         done
1801         cancel_lru_locks osc
1802         sync; sync_all_data || true
1803
1804         echo "Verify disk usage after write"
1805         USED=$(getquota -u $TSTID global curspace)
1806         [ $USED -lt $TOTAL_BLKS ] &&
1807                 error "Used space for user $TSTID:$USED, expected:$TOTAL_BLKS"
1808         USED=$(getquota -g $TSTID global curspace)
1809         [ $USED -lt $TOTAL_BLKS ] &&
1810                 error "Used space for group $TSTID:$USED, expected:$TOTAL_BLKS"
1811
1812         echo "Verify inode usage after write"
1813         USED=$(getquota -u $TSTID global curinodes)
1814         [ $USED -lt $INODES ] &&
1815                 error "Used inode for user $TSTID is $USED, expected $INODES"
1816         USED=$(getquota -g $TSTID global curinodes)
1817         [ $USED -lt $INODES ] &&
1818                 error "Used inode for group $TSTID is $USED, expected $INODES"
1819
1820         cleanup_quota_test
1821
1822         echo "Verify disk usage after delete"
1823         USED=$(getquota -u $TSTID global curspace)
1824         [ $USED -eq 0 ] || error "Used space for user $TSTID isn't 0. $USED"
1825         USED=$(getquota -u $TSTID global curinodes)
1826         [ $USED -eq 0 ] || error "Used inodes for user $TSTID isn't 0. $USED"
1827         USED=$(getquota -g $TSTID global curspace)
1828         [ $USED -eq 0 ] || error "Used space for group $TSTID isn't 0. $USED"
1829         USED=$(getquota -g $TSTID global curinodes)
1830         [ $USED -eq 0 ] || error "Used inodes for group $TSTID isn't 0. $USED"
1831 }
1832 run_test 33 "Basic usage tracking for user & group"
1833
1834 # usage transfer test for user & group
1835 test_34() {
1836         local BLK_CNT=2 # 2MB
1837
1838         setup_quota_test
1839         trap cleanup_quota_test EXIT
1840
1841         # make sure the system is clean
1842         local USED=$(getquota -u $TSTID global curspace)
1843         [ $USED -ne 0 ] && error "Used space ($USED) for user $TSTID isn't 0."
1844         USED=$(getquota -g $TSTID global curspace)
1845         [ $USED -ne 0 ] && error "Used space ($USED) for group $TSTID isn't 0."
1846
1847         echo "Write file..."
1848         $DD of=$DIR/$tdir/$tfile count=$BLK_CNT 2>/dev/null ||
1849                 error "write failed"
1850         cancel_lru_locks osc
1851         sync; sync_all_data || true
1852
1853         echo "chown the file to user $TSTID"
1854         chown $TSTID $DIR/$tdir/$tfile || error "chown failed"
1855
1856         echo "Wait for setattr on objects finished..."
1857         wait_delete_completed
1858
1859         BLK_CNT=$((BLK_CNT * 1024))
1860
1861         echo "Verify disk usage for user $TSTID"
1862         USED=$(getquota -u $TSTID global curspace)
1863         [ $USED -lt $BLK_CNT ] &&
1864                 error "Used space for user $TSTID is ${USED}, expected $BLK_CNT"
1865         USED=$(getquota -u $TSTID global curinodes)
1866         [ $USED -ne 1 ] &&
1867                 error "Used inodes for user $TSTID is $USED, expected 1"
1868
1869         echo "chgrp the file to group $TSTID"
1870         chgrp $TSTID $DIR/$tdir/$tfile || error "chgrp failed"
1871
1872         echo "Wait for setattr on objects finished..."
1873         wait_delete_completed
1874
1875         echo "Verify disk usage for group $TSTID"
1876         USED=$(getquota -g $TSTID global curspace)
1877         [ $USED -ge $BLK_CNT ] ||
1878                 error "Used space for group $TSTID is $USED, expected $BLK_CNT"
1879         USED=$(getquota -g $TSTID global curinodes)
1880         [ $USED -eq 1 ] ||
1881                 error "Used inodes for group $TSTID is $USED, expected 1"
1882
1883         cleanup_quota_test
1884 }
1885 run_test 34 "Usage transfer for user & group"
1886
1887 # usage is still accessible across restart
1888 test_35() {
1889         local BLK_CNT=2 # 2 MB
1890
1891         setup_quota_test
1892         trap cleanup_quota_test EXIT
1893
1894         echo "Write file..."
1895         $RUNAS $DD of=$DIR/$tdir/$tfile count=$BLK_CNT 2>/dev/null ||
1896                 error "write failed"
1897         cancel_lru_locks osc
1898         sync; sync_all_data || true
1899
1900         echo "Save disk usage before restart"
1901         local ORIG_USR_SPACE=$(getquota -u $TSTID global curspace)
1902         [ $ORIG_USR_SPACE -eq 0 ] &&
1903                 error "Used space for user $TSTID is 0, expected ${BLK_CNT}M"
1904         local ORIG_USR_INODES=$(getquota -u $TSTID global curinodes)
1905         [ $ORIG_USR_INODES -eq 0 ] &&
1906                 error "Used inodes for user $TSTID is 0, expected 1"
1907         echo "User $TSTID: ${ORIG_USR_SPACE}KB $ORIG_USR_INODES inodes"
1908         local ORIG_GRP_SPACE=$(getquota -g $TSTID global curspace)
1909         [ $ORIG_GRP_SPACE -eq 0 ] &&
1910                 error "Used space for group $TSTID is 0, expected ${BLK_CNT}M"
1911         local ORIG_GRP_INODES=$(getquota -g $TSTID global curinodes)
1912         [ $ORIG_GRP_INODES -eq 0 ] &&
1913                 error "Used inodes for group $TSTID is 0, expected 1"
1914         echo "Group $TSTID: ${ORIG_GRP_SPACE}KB $ORIG_GRP_INODES inodes"
1915
1916         log "Restart..."
1917         local ORIG_REFORMAT=$REFORMAT
1918         REFORMAT=""
1919         cleanup_and_setup_lustre
1920         REFORMAT=$ORIG_REFORMAT
1921         quota_init
1922
1923         echo "Verify disk usage after restart"
1924         local USED=$(getquota -u $TSTID global curspace)
1925         [ $USED -eq $ORIG_USR_SPACE ] ||
1926                 error "Used space for user $TSTID changed from " \
1927                         "$ORIG_USR_SPACE to $USED"
1928         USED=$(getquota -u $TSTID global curinodes)
1929         [ $USED -eq $ORIG_USR_INODES ] ||
1930                 error "Used inodes for user $TSTID changed from " \
1931                         "$ORIG_USR_INODES to $USED"
1932         USED=$(getquota -g $TSTID global curspace)
1933         [ $USED -eq $ORIG_GRP_SPACE ] ||
1934                 error "Used space for group $TSTID changed from " \
1935                         "$ORIG_GRP_SPACE to $USED"
1936         USED=$(getquota -g $TSTID global curinodes)
1937         [ $USED -eq $ORIG_GRP_INODES ] ||
1938                 error "Used inodes for group $TSTID changed from " \
1939                         "$ORIG_GRP_INODES to $USED"
1940
1941         # check if the vfs_dq_init() is called before writing
1942         echo "Append to the same file..."
1943         $RUNAS $DD of=$DIR/$tdir/$tfile count=$BLK_CNT seek=1 2>/dev/null ||
1944                 error "write failed"
1945         cancel_lru_locks osc
1946         sync; sync_all_data || true
1947
1948         echo "Verify space usage is increased"
1949         USED=$(getquota -u $TSTID global curspace)
1950         [ $USED -gt $ORIG_USR_SPACE ] ||
1951                 error "Used space for user $TSTID isn't increased" \
1952                         "orig:$ORIG_USR_SPACE, now:$USED"
1953         USED=$(getquota -g $TSTID global curspace)
1954         [ $USED -gt $ORIG_GRP_SPACE ] ||
1955                 error "Used space for group $TSTID isn't increased" \
1956                         "orig:$ORIG_GRP_SPACE, now:$USED"
1957
1958         cleanup_quota_test
1959 }
1960 run_test 35 "Usage is still accessible across reboot"
1961
1962 # test migrating old amdin quota files (in Linux quota file format v2) into new
1963 # quota global index (in IAM format)
1964 test_36() {
1965         [ $(facet_fstype $SINGLEMDS) != ldiskfs ] && \
1966                 skip "skipping migration test" && return
1967
1968         # get the mdt0 device name
1969         local mdt0_node=$(facet_active_host $SINGLEMDS)
1970         local mdt0_dev=$(mdsdevname ${SINGLEMDS//mds/})
1971
1972         echo "Reformat..."
1973         formatall
1974
1975         echo "Copy admin quota files into MDT0..."
1976         local mntpt=$(facet_mntpt $SINGLEMDS)
1977         local mdt0_fstype=$(facet_fstype $SINGLEMDS)
1978         local opt
1979         if ! do_node $mdt0_node test -b $mdt0_fstype; then
1980                 opt="-o loop"
1981         fi
1982         echo "$mdt0_node, $mdt0_dev, $mntpt, $opt"
1983         do_node $mdt0_node mount -t $mdt0_fstype $opt $mdt0_dev $mntpt
1984         do_node $mdt0_node mkdir $mntpt/OBJECTS
1985         do_node $mdt0_node cp $LUSTRE/tests/admin_quotafile_v2.usr $mntpt/OBJECTS
1986         do_node $mdt0_node cp $LUSTRE/tests/admin_quotafile_v2.grp $mntpt/OBJECTS
1987         do_node $mdt0_node umount -f $mntpt
1988
1989         echo "Setup all..."
1990         setupall
1991
1992         echo "Verify global limits..."
1993         local id_cnt
1994         local limit
1995
1996         local proc="qmt.*.md-0x0.glb-usr"
1997         id_cnt=$(do_node $mdt0_node $LCTL get_param -n $proc | wc -l)
1998         [ $id_cnt -eq 401 ] || error "Migrate inode user limit failed: $id_cnt"
1999         limit=$(getquota -u 1 global isoftlimit)
2000         [ $limit -eq 1024 ] || error "User inode softlimit: $limit"
2001         limit=$(getquota -u 1 global ihardlimit)
2002         [ $limit -eq 2048 ] || error "User inode hardlimit: $limit"
2003
2004         proc="qmt.*.md-0x0.glb-grp"
2005         id_cnt=$(do_node $mdt0_node $LCTL get_param -n $proc | wc -l)
2006         [ $id_cnt -eq 401 ] || error "Migrate inode group limit failed: $id_cnt"
2007         limit=$(getquota -g 1 global isoftlimit)
2008         [ $limit -eq 1024 ] || error "Group inode softlimit: $limit"
2009         limit=$(getquota -g 1 global ihardlimit)
2010         [ $limit -eq 2048 ] || error "Group inode hardlimit: $limit"
2011
2012         proc=" qmt.*.dt-0x0.glb-usr"
2013         id_cnt=$(do_node $mdt0_node $LCTL get_param -n $proc | wc -l)
2014         [ $id_cnt -eq 401 ] || error "Migrate block user limit failed: $id_cnt"
2015         limit=$(getquota -u 60001 global bsoftlimit)
2016         [ $limit -eq 10485760 ] || error "User block softlimit: $limit"
2017         limit=$(getquota -u 60001 global bhardlimit)
2018         [ $limit -eq 20971520 ] || error "User block hardlimit: $limit"
2019
2020         proc="qmt.*.dt-0x0.glb-grp"
2021         id_cnt=$(do_node $mdt0_node $LCTL get_param -n $proc | wc -l)
2022         [ $id_cnt -eq 401 ] || error "Migrate block user limit failed: $id_cnt"
2023         limit=$(getquota -g 60001 global bsoftlimit)
2024         [ $limit -eq 10485760 ] || error "Group block softlimit: $limit"
2025         limit=$(getquota -g 60001 global bhardlimit)
2026         [ $limit -eq 20971520 ] || error "Group block hardlimit: $limit"
2027
2028         echo "Cleanup..."
2029         formatall
2030         setupall
2031 }
2032 run_test 36 "Migrate old admin files into new global indexes"
2033
2034 quota_fini()
2035 {
2036         do_nodes $(comma_list $(nodes_list)) "lctl set_param debug=-quota"
2037 }
2038 quota_fini
2039
2040 cd $ORIG_PWD
2041 complete $SECONDS
2042 check_and_cleanup_lustre
2043 export QUOTA_AUTO=$QUOTA_AUTO_OLD
2044 exit_status