+check_grants() {
+ local tgt=$1
+ local expected=$2
+ local msg=$3
+ local cur_grants=$($LCTL get_param -n osc.$tgt.cur_grant_bytes)
+
+ ((cur_grants == expected)) ||
+ error "$msg: grants mismatch: $cur_grants, expected $expected"
+}
+
+round_up_p2() {
+ echo $((($1 + $2 - 1) & ~($2 - 1)))
+}
+
+test_64e() {
+ [ $PARALLEL == "yes" ] && skip "skip parallel run"
+ [ $OST1_VERSION -ge $(version_code 2.11.56) ] ||
+ skip "Need OSS version at least 2.11.56"
+
+ # Remount client to reset grant
+ remount_client $MOUNT || error "failed to remount client"
+ local osc_tgt="$FSNAME-OST0000-osc-$($LFS getname -i $DIR)"
+
+ local init_grants=$(import_param $osc_tgt initial_grant)
+
+ check_grants $osc_tgt $init_grants "init grants"
+
+ local extent_tax=$(import_param $osc_tgt grant_extent_tax)
+ local max_brw_size=$(import_param $osc_tgt max_brw_size)
+ local gbs=$(import_param $osc_tgt grant_block_size)
+
+ # write random number of bytes from max_brw_size / 4 to max_brw_size
+ local write_bytes=$(shuf -i $((max_brw_size / 4))-$max_brw_size -n 1)
+ # align for direct io
+ write_bytes=$(round_up_p2 $write_bytes PAGE_SIZE)
+ # round to grant consumption unit
+ local wb_round_up=$(round_up_p2 $write_bytes gbs)
+
+ local grants=$((wb_round_up + extent_tax))
+
+ $LFS setstripe -c 1 -i 0 $DIR/$tfile || error "lfs setstripe failed"
+
+ # define OBD_FAIL_TGT_NO_GRANT 0x725
+ # make the server not grant more back
+ do_facet ost1 $LCTL set_param fail_loc=0x725
+ dd if=/dev/zero of=$DIR/$tfile bs=$write_bytes count=1 oflag=direct
+
+ do_facet ost1 $LCTL set_param fail_loc=0
+
+ check_grants $osc_tgt $((init_grants - grants)) "dio w/o grant alloc"
+
+ rm -f $DIR/$tfile || error "rm failed"
+
+ # Remount client to reset grant
+ remount_client $MOUNT || error "failed to remount client"
+ osc_tgt="$FSNAME-OST0000-osc-$($LFS getname -i $DIR)"
+
+ $LFS setstripe -c 1 -i 0 $DIR/$tfile || error "lfs setstripe failed"
+
+ # define OBD_FAIL_TGT_NO_GRANT 0x725
+ # make the server not grant more back
+ do_facet ost1 $LCTL set_param fail_loc=0x725
+ $MULTIOP $DIR/$tfile "oO_WRONLY:w${write_bytes}yc"
+ do_facet ost1 $LCTL set_param fail_loc=0
+
+ check_grants $osc_tgt $((init_grants - grants)) "buf io w/o grant alloc"
+}
+run_test 64e "check grant consumption (no grant allocation)"
+
+test_64f() {
+ [ $PARALLEL == "yes" ] && skip "skip parallel run"
+
+ # Remount client to reset grant
+ remount_client $MOUNT || error "failed to remount client"
+ local osc_tgt="$FSNAME-OST0000-osc-$($LFS getname -i $DIR)"
+
+ local init_grants=$(import_param $osc_tgt initial_grant)
+ local extent_tax=$(import_param $osc_tgt grant_extent_tax)
+ local max_brw_size=$(import_param $osc_tgt max_brw_size)
+ local gbs=$(import_param $osc_tgt grant_block_size)
+ local chunk=$(grant_chunk $osc_tgt)
+
+ # write random number of bytes from max_brw_size / 4 to max_brw_size
+ local write_bytes=$(shuf -i $((max_brw_size / 4))-$max_brw_size -n 1)
+ # align for direct io
+ write_bytes=$(round_up_p2 $write_bytes PAGE_SIZE)
+ # round to grant consumption unit
+ local wb_round_up=$(round_up_p2 $write_bytes gbs)
+
+ local grants=$((wb_round_up + extent_tax))
+
+ $LFS setstripe -c 1 -i 0 $DIR/$tfile || error "lfs setstripe failed"
+ dd if=/dev/zero of=$DIR/$tfile bs=$write_bytes count=1 oflag=direct ||
+ error "error writing to $DIR/$tfile"
+
+ check_grants $osc_tgt $((init_grants - grants + chunk)) \
+ "direct io with grant allocation"
+
+ rm -f $DIR/$tfile || error "rm failed"
+
+ # Remount client to reset grant
+ remount_client $MOUNT || error "failed to remount client"
+ osc_tgt="$FSNAME-OST0000-osc-$($LFS getname -i $DIR)"
+
+ $LFS setstripe -c 1 -i 0 $DIR/$tfile || error "lfs setstripe failed"
+
+ local cmd="oO_WRONLY:w${write_bytes}_yc"
+
+ $MULTIOP $DIR/$tfile $cmd &
+ MULTIPID=$!
+ sleep 1
+
+ check_grants $osc_tgt $((init_grants - grants)) \
+ "buffered io, not write rpc"
+
+ kill -USR1 $MULTIPID
+ wait
+
+ check_grants $osc_tgt $((init_grants - grants + chunk)) \
+ "buffered io, one RPC"
+}
+run_test 64f "check grant consumption (with grant allocation)"
+