#!/usr/bin/env bash

rootdir=$(readlink -f $(dirname $0))/../..
source $rootdir/test/common/autotest_common.sh
source "$rootdir/scripts/common.sh"

TEST_TIMEOUT=1200

VHOST_APP="$rootdir/app/vhost/vhost -p 0"
FUZZ_RPC_SOCK="/var/tmp/spdk_fuzz.sock"
FUZZ_APP="$rootdir/test/app/fuzz/vhost_fuzz/vhost_fuzz -r $FUZZ_RPC_SOCK --wait-for-rpc"

vhost_rpc_py="$rootdir/scripts/rpc.py"
fuzz_generic_rpc_py="$rootdir/scripts/rpc.py -s $FUZZ_RPC_SOCK"
fuzz_specific_rpc_py="$rootdir/test/app/fuzz/common/fuzz_rpc.py -s $FUZZ_RPC_SOCK"

# This argument is used in addition to the test arguments in autotest_common.sh
for i in "$@"; do
    case "$i" in
        --timeout=*)
            TEST_TIMEOUT="${i#*=}"
    esac
done

timing_enter vhost_fuzz_test

#todo refactor this to use the vhosttestinit function when it becomes available.
timing_enter setup
$rootdir/scripts/setup.sh
timing_exit setup

$VHOST_APP &>$output_dir/vhost_fuzz_tgt_output.txt &
vhostpid=$!
waitforlisten $vhostpid

trap 'killprocess $vhostpid; exit 1' SIGINT SIGTERM exit

$FUZZ_APP -t $TEST_TIMEOUT 2>$output_dir/vhost_autofuzz_output1.txt &
fuzzpid=$!
waitforlisten $fuzzpid $FUZZ_RPC_SOCK

trap 'killprocess $vhostpid; killprocess $fuzzpid; exit 1' SIGINT SIGTERM exit

if [ "$TEST_TRANSPORT" == "bdev" ] || [ "$TEST_TRANSPORT" == "all" ]; then
    $vhost_rpc_py bdev_malloc_create -b Malloc0 64 512
    $vhost_rpc_py construct_vhost_blk_controller Vhost.1 Malloc0

    # test the vhost blk controller with valid data buffers.
    $fuzz_specific_rpc_py fuzz_vhost_create_dev -s $(pwd)/Vhost.1 -b -v
fi

if [ "$TEST_TRANSPORT" == "scsi" ] || [ "$TEST_TRANSPORT" == "all" ]; then
    $vhost_rpc_py bdev_malloc_create -b Malloc1 64 512
    $vhost_rpc_py construct_vhost_scsi_controller naa.VhostScsi0.1
    $vhost_rpc_py add_vhost_scsi_lun naa.VhostScsi0.1 0 Malloc1

    $vhost_rpc_py bdev_malloc_create -b Malloc2 64 512
    $vhost_rpc_py construct_vhost_scsi_controller naa.VhostScsi0.2
    $vhost_rpc_py add_vhost_scsi_lun naa.VhostScsi0.2 0 Malloc2

    # test the vhost scsi I/O queue with valid data buffers on a valid lun.
    $fuzz_specific_rpc_py fuzz_vhost_create_dev -s $(pwd)/naa.VhostScsi0.1 -l -v
    # test the vhost scsi management queue with valid data buffers.
    $fuzz_specific_rpc_py fuzz_vhost_create_dev -s $(pwd)/naa.VhostScsi0.2 -v -m
fi

# The test won't actually begin until this option is passed in.
$fuzz_generic_rpc_py start_subsystem_init

wait $fuzzpid

trap - SIGINT SIGTERM exit
set +e
killprocess $vhostpid
echo $?
set -e
killprocess $fuzzpid
timing_exit vhost_fuzz_test