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