#!/bin/bash
# SPDX-License-Identifier: GPL-2.0
# Copyright 2019 Google LLC
#
# Trigger blk_mq_update_nr_hw_queues().

. tests/block/rc
. common/null_blk

DESCRIPTION="trigger blk_mq_update_nr_hw_queues()"
QUICK=1

requires() {
	_have_fio && _have_null_blk
}

modify_nr_hw_queues() {
	local deadline num_cpus

	deadline=$(($(_uptime_s) + TIMEOUT))
	num_cpus=$(nproc)
	while [ "$(_uptime_s)" -lt "$deadline" ]; do
		sleep .1
		echo 1 > /sys/kernel/config/nullb/nullb1/submit_queues
		sleep .1
		echo "$num_cpus" > /sys/kernel/config/nullb/nullb1/submit_queues
	done
}

test() {
	local sq=/sys/kernel/config/nullb/nullb1/submit_queues

	: "${TIMEOUT:=30}"
	_configure_null_blk nullb1 completion_nsec=0 blocksize=512 \
			    size=16 memory_backed=1 power=1 &&
	if { echo 1 >$sq; } 2>/dev/null; then
		modify_nr_hw_queues &
		fio --rw=randwrite --bs=4K --loops=$((10**6)) \
		    --iodepth=64 --group_reporting --sync=1 --direct=1 \
		    --ioengine=libaio --filename="/dev/nullb1" \
		    --runtime="${TIMEOUT}" --name=nullb1 \
		    --output="${RESULTS_DIR}/block/fio-output-029.txt" \
		    >>"$FULL"
		wait
	else
		echo "Skipping test because $sq cannot be modified" >>"$FULL"
	fi
	_exit_null_blk
	echo Passed
}
