tools/testing/selftests/ntb/ntb_test.sh - linux-imx - Git at Google

 #!/bin/bash
 # Copyright (c) 2016 Microsemi. All Rights Reserved.
 #
 # This program is free software; you can redistribute it and/or
 # modify it under the terms of the GNU General Public License as
 # published by the Free Software Foundation; either version 2 of
 # the License, or (at your option) any later version.
 #
 # This program is distributed in the hope that it would be useful,
 # but WITHOUT ANY WARRANTY; without even the implied warranty of
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 # GNU General Public License for more details.
 #
 # Author: Logan Gunthorpe <logang@deltatee.com>

 REMOTE_HOST=
 LIST_DEVS=FALSE

 DEBUGFS=${DEBUGFS-/sys/kernel/debug}

 PERF_RUN_ORDER=32
 MAX_MW_SIZE=0
 RUN_DMA_TESTS=
 DONT_CLEANUP=
 MW_SIZE=65536

 function show_help()
 {
 	echo "Usage: $0 [OPTIONS] LOCAL_DEV REMOTE_DEV"
 	echo "Run tests on a pair of NTB endpoints."
 	echo
 	echo "If the NTB device loops back to the same host then,"
 	echo "just specifying the two PCI ids on the command line is"
 	echo "sufficient. Otherwise, if the NTB link spans two hosts"
 	echo "use the -r option to specify the hostname for the remote"
 	echo "device. SSH will then be used to test the remote side."
 	echo "An SSH key between the root users of the host would then"
 	echo "be highly recommended."
 	echo
 	echo "Options:"
 	echo "  -C              don't cleanup ntb modules on exit"
 	echo "  -d              run dma tests"
 	echo "  -h              show this help message"
 	echo "  -l              list available local and remote PCI ids"
 	echo "  -r REMOTE_HOST  specify the remote's hostname to connect"
         echo "                  to for the test (using ssh)"
 	echo "  -p NUM          ntb_perf run order (default: $PERF_RUN_ORDER)"
 	echo "  -w max_mw_size  maxmium memory window size"
 	echo
 }

 function parse_args()
 {
 	OPTIND=0
 	while getopts "Cdhlm:r:p:w:" opt; do
 		case "$opt" in
 		C)  DONT_CLEANUP=1 ;;
 		d)  RUN_DMA_TESTS=1 ;;
 		h)  show_help; exit 0 ;;
 		l)  LIST_DEVS=TRUE ;;
 		m)  MW_SIZE=${OPTARG} ;;
 		r)  REMOTE_HOST=${OPTARG} ;;
 		p)  PERF_RUN_ORDER=${OPTARG} ;;
 		w)  MAX_MW_SIZE=${OPTARG} ;;
 		\?)
 		    echo "Invalid option: -$OPTARG" >&2
 		    exit 1
 		    ;;
 		esac
 	done
 }

 parse_args "$@"
 shift $((OPTIND-1))
 LOCAL_DEV=$1
 shift
 parse_args "$@"
 shift $((OPTIND-1))
 REMOTE_DEV=$1
 shift
 parse_args "$@"

 set -e

 function _modprobe()
 {
         modprobe "$@"
 }

 function split_remote()
 {
 	VPATH=$1
 	REMOTE=

 	if [[ "$VPATH" == *":/"* ]]; then
 		REMOTE=${VPATH%%:*}
 		VPATH=${VPATH#*:}
 	fi
 }

 function read_file()
 {
 	split_remote $1
 	if [[ "$REMOTE" != "" ]]; then
 		ssh "$REMOTE" cat "$VPATH"
 	else
 		cat "$VPATH"
 	fi
 }

 function write_file()
 {
 	split_remote $2
 	VALUE=$1

 	if [[ "$REMOTE" != "" ]]; then
 		ssh "$REMOTE" "echo \"$VALUE\" > \"$VPATH\""
 	else
 		echo "$VALUE" > "$VPATH"
 	fi
 }

 function link_test()
 {
 	LOC=$1
 	REM=$2
 	EXP=0

 	echo "Running link tests on: $(basename $LOC) / $(basename $REM)"

 	if ! write_file "N" "$LOC/link" 2> /dev/null; then
 		echo "  Unsupported"
 		return
 	fi

 	write_file "N" "$LOC/link_event"

 	if [[ $(read_file "$REM/link") != "N" ]]; then
 		echo "Expected remote link to be down in $REM/link" >&2
 		exit -1
 	fi

 	write_file "Y" "$LOC/link"
 	write_file "Y" "$LOC/link_event"

 	echo "  Passed"
 }

 function doorbell_test()
 {
 	LOC=$1
 	REM=$2
 	EXP=0

 	echo "Running db tests on: $(basename $LOC) / $(basename $REM)"

 	write_file "c 0xFFFFFFFF" "$REM/db"

 	for ((i=1; i <= 8; i++)); do
 		let DB=$(read_file "$REM/db") || true
 		if [[ "$DB" != "$EXP" ]]; then
 			echo "Doorbell doesn't match expected value $EXP " \
 			     "in $REM/db" >&2
 			exit -1
 		fi

 		let "MASK=1 << ($i-1)" || true
 		let "EXP=$EXP | $MASK" || true
 		write_file "s $MASK" "$LOC/peer_db"
 	done

 	echo "  Passed"
 }

 function read_spad()
 {
        VPATH=$1
        IDX=$2

        ROW=($(read_file "$VPATH" | grep -e "^$IDX"))
        let VAL=${ROW[1]} || true
        echo $VAL
 }

 function scratchpad_test()
 {
 	LOC=$1
 	REM=$2
 	CNT=$(read_file "$LOC/spad" | wc -l)

 	echo "Running spad tests on: $(basename $LOC) / $(basename $REM)"

 	for ((i = 0; i < $CNT; i++)); do
 		VAL=$RANDOM
 		write_file "$i $VAL" "$LOC/peer_spad"
 		RVAL=$(read_spad "$REM/spad" $i)

 		if [[ "$VAL" != "$RVAL" ]]; then
 			echo "Scratchpad doesn't match expected value $VAL " \
 			     "in $REM/spad, got $RVAL" >&2
 			exit -1
 		fi

 	done

 	echo "  Passed"
 }

 function write_mw()
 {
 	split_remote $2

 	if [[ "$REMOTE" != "" ]]; then
 		ssh "$REMOTE" \
 			dd if=/dev/urandom "of=$VPATH" 2> /dev/null || true
 	else
 		dd if=/dev/urandom "of=$VPATH" 2> /dev/null || true
 	fi
 }

 function mw_test()
 {
 	IDX=$1
 	LOC=$2
 	REM=$3

 	echo "Running $IDX tests on: $(basename $LOC) / $(basename $REM)"

 	write_mw "$LOC/$IDX"

 	split_remote "$LOC/$IDX"
 	if [[ "$REMOTE" == "" ]]; then
 		A=$VPATH
 	else
 		A=/tmp/ntb_test.$$.A
 		ssh "$REMOTE" cat "$VPATH" > "$A"
 	fi

 	split_remote "$REM/peer_$IDX"
 	if [[ "$REMOTE" == "" ]]; then
 		B=$VPATH
 	else
 		B=/tmp/ntb_test.$$.B
 		ssh "$REMOTE" cat "$VPATH" > "$B"
 	fi

 	cmp -n $MW_SIZE "$A" "$B"
 	if [[ $? != 0 ]]; then
 		echo "Memory window $MW did not match!" >&2
 	fi

 	if [[ "$A" == "/tmp/*" ]]; then
 		rm "$A"
 	fi

 	if [[ "$B" == "/tmp/*" ]]; then
 		rm "$B"
 	fi

 	echo "  Passed"
 }

 function pingpong_test()
 {
 	LOC=$1
 	REM=$2

 	echo "Running ping pong tests on: $(basename $LOC) / $(basename $REM)"

 	LOC_START=$(read_file $LOC/count)
 	REM_START=$(read_file $REM/count)

 	sleep 7

 	LOC_END=$(read_file $LOC/count)
 	REM_END=$(read_file $REM/count)

 	if [[ $LOC_START == $LOC_END ]] || [[ $REM_START == $REM_END ]]; then
 		echo "Ping pong counter not incrementing!" >&2
 		exit 1
 	fi

 	echo "  Passed"
 }

 function perf_test()
 {
 	USE_DMA=$1

 	if [[ $USE_DMA == "1" ]]; then
 		WITH="with"
 	else
 		WITH="without"
 	fi

 	_modprobe ntb_perf run_order=$PERF_RUN_ORDER \
 		max_mw_size=$MAX_MW_SIZE use_dma=$USE_DMA

 	echo "Running local perf test $WITH DMA"
 	write_file "" $LOCAL_PERF/run
 	echo -n "  "
 	read_file $LOCAL_PERF/run
 	echo "  Passed"

 	echo "Running remote perf test $WITH DMA"
 	write_file "" $REMOTE_PERF/run
 	echo -n "  "
 	read_file $REMOTE_PERF/run
 	echo "  Passed"

 	_modprobe -r ntb_perf
 }

 function ntb_tool_tests()
 {
 	LOCAL_TOOL=$DEBUGFS/ntb_tool/$LOCAL_DEV
 	REMOTE_TOOL=$REMOTE_HOST:$DEBUGFS/ntb_tool/$REMOTE_DEV

 	echo "Starting ntb_tool tests..."

 	_modprobe ntb_tool

 	write_file Y $LOCAL_TOOL/link_event
 	write_file Y $REMOTE_TOOL/link_event

 	link_test $LOCAL_TOOL $REMOTE_TOOL
 	link_test $REMOTE_TOOL $LOCAL_TOOL

 	#Ensure the link is up on both sides before continuing
 	write_file Y $LOCAL_TOOL/link_event
 	write_file Y $REMOTE_TOOL/link_event

 	for PEER_TRANS in $(ls $LOCAL_TOOL/peer_trans*); do
 		PT=$(basename $PEER_TRANS)
 		write_file $MW_SIZE $LOCAL_TOOL/$PT
 		write_file $MW_SIZE $REMOTE_TOOL/$PT
 	done

 	doorbell_test $LOCAL_TOOL $REMOTE_TOOL
 	doorbell_test $REMOTE_TOOL $LOCAL_TOOL
 	scratchpad_test $LOCAL_TOOL $REMOTE_TOOL
 	scratchpad_test $REMOTE_TOOL $LOCAL_TOOL

 	for MW in $(ls $LOCAL_TOOL/mw*); do
 		MW=$(basename $MW)

 		mw_test $MW $LOCAL_TOOL $REMOTE_TOOL
 		mw_test $MW $REMOTE_TOOL $LOCAL_TOOL
 	done

 	_modprobe -r ntb_tool
 }

 function ntb_pingpong_tests()
 {
 	LOCAL_PP=$DEBUGFS/ntb_pingpong/$LOCAL_DEV
 	REMOTE_PP=$REMOTE_HOST:$DEBUGFS/ntb_pingpong/$REMOTE_DEV

 	echo "Starting ntb_pingpong tests..."

 	_modprobe ntb_pingpong

 	pingpong_test $LOCAL_PP $REMOTE_PP

 	_modprobe -r ntb_pingpong
 }

 function ntb_perf_tests()
 {
 	LOCAL_PERF=$DEBUGFS/ntb_perf/$LOCAL_DEV
 	REMOTE_PERF=$REMOTE_HOST:$DEBUGFS/ntb_perf/$REMOTE_DEV

 	echo "Starting ntb_perf tests..."

 	perf_test 0

 	if [[ $RUN_DMA_TESTS ]]; then
 		perf_test 1
 	fi
 }

 function cleanup()
 {
 	set +e
 	_modprobe -r ntb_tool 2> /dev/null
 	_modprobe -r ntb_perf 2> /dev/null
 	_modprobe -r ntb_pingpong 2> /dev/null
 	_modprobe -r ntb_transport 2> /dev/null
 	set -e
 }

 cleanup

 if ! [[ $$DONT_CLEANUP ]]; then
 	trap cleanup EXIT
 fi

 if [ "$(id -u)" != "0" ]; then
 	echo "This script must be run as root" 1>&2
 	exit 1
 fi

 if [[ "$LIST_DEVS" == TRUE ]]; then
 	echo "Local Devices:"
 	ls -1 /sys/bus/ntb/devices
 	echo

 	if [[ "$REMOTE_HOST" != "" ]]; then
 		echo "Remote Devices:"
 		ssh $REMOTE_HOST ls -1 /sys/bus/ntb/devices
 	fi

 	exit 0
 fi

 if [[ "$LOCAL_DEV" == $"" ]] || [[ "$REMOTE_DEV" == $"" ]]; then
 	show_help
 	exit 1
 fi

 ntb_tool_tests
 echo
 ntb_pingpong_tests
 echo
 ntb_perf_tests
 echo
	#!/bin/bash
	# Copyright (c) 2016 Microsemi. All Rights Reserved.
	#
	# This program is free software; you can redistribute it and/or
	# modify it under the terms of the GNU General Public License as
	# published by the Free Software Foundation; either version 2 of
	# the License, or (at your option) any later version.
	#
	# This program is distributed in the hope that it would be useful,
	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	# GNU General Public License for more details.
	#
	# Author: Logan Gunthorpe <logang@deltatee.com>

	REMOTE_HOST=
	LIST_DEVS=FALSE

	DEBUGFS=${DEBUGFS-/sys/kernel/debug}

	PERF_RUN_ORDER=32
	MAX_MW_SIZE=0
	RUN_DMA_TESTS=
	DONT_CLEANUP=
	MW_SIZE=65536

	function show_help()
	{
	echo "Usage: $0 [OPTIONS] LOCAL_DEV REMOTE_DEV"
	echo "Run tests on a pair of NTB endpoints."
	echo
	echo "If the NTB device loops back to the same host then,"
	echo "just specifying the two PCI ids on the command line is"
	echo "sufficient. Otherwise, if the NTB link spans two hosts"
	echo "use the -r option to specify the hostname for the remote"
	echo "device. SSH will then be used to test the remote side."
	echo "An SSH key between the root users of the host would then"
	echo "be highly recommended."
	echo
	echo "Options:"
	echo " -C don't cleanup ntb modules on exit"
	echo " -d run dma tests"
	echo " -h show this help message"
	echo " -l list available local and remote PCI ids"
	echo " -r REMOTE_HOST specify the remote's hostname to connect"
	echo " to for the test (using ssh)"
	echo " -p NUM ntb_perf run order (default: $PERF_RUN_ORDER)"
	echo " -w max_mw_size maxmium memory window size"
	echo
	}

	function parse_args()
	{
	OPTIND=0
	while getopts "Cdhlm:r:p:w:" opt; do
	case "$opt" in
	C) DONT_CLEANUP=1 ;;
	d) RUN_DMA_TESTS=1 ;;
	h) show_help; exit 0 ;;
	l) LIST_DEVS=TRUE ;;
	m) MW_SIZE=${OPTARG} ;;
	r) REMOTE_HOST=${OPTARG} ;;
	p) PERF_RUN_ORDER=${OPTARG} ;;
	w) MAX_MW_SIZE=${OPTARG} ;;
	\?)
	echo "Invalid option: -$OPTARG" >&2
	exit 1
	;;
	esac
	done
	}

	parse_args "$@"
	shift $((OPTIND-1))
	LOCAL_DEV=$1
	shift
	parse_args "$@"
	shift $((OPTIND-1))
	REMOTE_DEV=$1
	shift
	parse_args "$@"

	set -e

	function _modprobe()
	{
	modprobe "$@"
	}

	function split_remote()
	{
	VPATH=$1
	REMOTE=

	if [[ "$VPATH" == ":/" ]]; then
	REMOTE=${VPATH%%:*}
	VPATH=${VPATH#*:}
	fi
	}

	function read_file()
	{
	split_remote $1
	if [[ "$REMOTE" != "" ]]; then
	ssh "$REMOTE" cat "$VPATH"
	else
	cat "$VPATH"
	fi
	}

	function write_file()
	{
	split_remote $2
	VALUE=$1

	if [[ "$REMOTE" != "" ]]; then
	ssh "$REMOTE" "echo \"$VALUE\" > \"$VPATH\""
	else
	echo "$VALUE" > "$VPATH"
	fi
	}

	function link_test()
	{
	LOC=$1
	REM=$2
	EXP=0

	echo "Running link tests on: $(basename $LOC) / $(basename $REM)"

	if ! write_file "N" "$LOC/link" 2> /dev/null; then
	echo " Unsupported"
	return
	fi

	write_file "N" "$LOC/link_event"

	if [[ $(read_file "$REM/link") != "N" ]]; then
	echo "Expected remote link to be down in $REM/link" >&2
	exit -1
	fi

	write_file "Y" "$LOC/link"
	write_file "Y" "$LOC/link_event"

	echo " Passed"
	}

	function doorbell_test()
	{
	LOC=$1
	REM=$2
	EXP=0

	echo "Running db tests on: $(basename $LOC) / $(basename $REM)"

	write_file "c 0xFFFFFFFF" "$REM/db"

	for ((i=1; i <= 8; i++)); do
	let DB=$(read_file "$REM/db") \|\| true
	if [[ "$DB" != "$EXP" ]]; then
	echo "Doorbell doesn't match expected value $EXP " \
	"in $REM/db" >&2
	exit -1
	fi

	let "MASK=1 << ($i-1)" \|\| true
	let "EXP=$EXP \| $MASK" \|\| true
	write_file "s $MASK" "$LOC/peer_db"
	done

	echo " Passed"
	}

	function read_spad()
	{
	VPATH=$1
	IDX=$2

	ROW=($(read_file "$VPATH" \| grep -e "^$IDX"))
	let VAL=${ROW[1]} \|\| true
	echo $VAL
	}

	function scratchpad_test()
	{
	LOC=$1
	REM=$2
	CNT=$(read_file "$LOC/spad" \| wc -l)

	echo "Running spad tests on: $(basename $LOC) / $(basename $REM)"

	for ((i = 0; i < $CNT; i++)); do
	VAL=$RANDOM
	write_file "$i $VAL" "$LOC/peer_spad"
	RVAL=$(read_spad "$REM/spad" $i)

	if [[ "$VAL" != "$RVAL" ]]; then
	echo "Scratchpad doesn't match expected value $VAL " \
	"in $REM/spad, got $RVAL" >&2
	exit -1
	fi

	done

	echo " Passed"
	}

	function write_mw()
	{
	split_remote $2

	if [[ "$REMOTE" != "" ]]; then
	ssh "$REMOTE" \
	dd if=/dev/urandom "of=$VPATH" 2> /dev/null \|\| true
	else
	dd if=/dev/urandom "of=$VPATH" 2> /dev/null \|\| true
	fi
	}

	function mw_test()
	{
	IDX=$1
	LOC=$2
	REM=$3

	echo "Running $IDX tests on: $(basename $LOC) / $(basename $REM)"

	write_mw "$LOC/$IDX"

	split_remote "$LOC/$IDX"
	if [[ "$REMOTE" == "" ]]; then
	A=$VPATH
	else
	A=/tmp/ntb_test.$$.A
	ssh "$REMOTE" cat "$VPATH" > "$A"
	fi

	split_remote "$REM/peer_$IDX"
	if [[ "$REMOTE" == "" ]]; then
	B=$VPATH
	else
	B=/tmp/ntb_test.$$.B
	ssh "$REMOTE" cat "$VPATH" > "$B"
	fi

	cmp -n $MW_SIZE "$A" "$B"
	if [[ $? != 0 ]]; then
	echo "Memory window $MW did not match!" >&2
	fi

	if [[ "$A" == "/tmp/*" ]]; then
	rm "$A"
	fi

	if [[ "$B" == "/tmp/*" ]]; then
	rm "$B"
	fi

	echo " Passed"
	}

	function pingpong_test()
	{
	LOC=$1
	REM=$2

	echo "Running ping pong tests on: $(basename $LOC) / $(basename $REM)"

	LOC_START=$(read_file $LOC/count)
	REM_START=$(read_file $REM/count)

	sleep 7

	LOC_END=$(read_file $LOC/count)
	REM_END=$(read_file $REM/count)

	if [[ $LOC_START == $LOC_END ]] \|\| [[ $REM_START == $REM_END ]]; then
	echo "Ping pong counter not incrementing!" >&2
	exit 1
	fi

	echo " Passed"
	}

	function perf_test()
	{
	USE_DMA=$1

	if [[ $USE_DMA == "1" ]]; then
	WITH="with"
	else
	WITH="without"
	fi

	_modprobe ntb_perf run_order=$PERF_RUN_ORDER \
	max_mw_size=$MAX_MW_SIZE use_dma=$USE_DMA

	echo "Running local perf test $WITH DMA"
	write_file "" $LOCAL_PERF/run
	echo -n " "
	read_file $LOCAL_PERF/run
	echo " Passed"

	echo "Running remote perf test $WITH DMA"
	write_file "" $REMOTE_PERF/run
	echo -n " "
	read_file $REMOTE_PERF/run
	echo " Passed"

	_modprobe -r ntb_perf
	}

	function ntb_tool_tests()
	{
	LOCAL_TOOL=$DEBUGFS/ntb_tool/$LOCAL_DEV
	REMOTE_TOOL=$REMOTE_HOST:$DEBUGFS/ntb_tool/$REMOTE_DEV

	echo "Starting ntb_tool tests..."

	_modprobe ntb_tool

	write_file Y $LOCAL_TOOL/link_event
	write_file Y $REMOTE_TOOL/link_event

	link_test $LOCAL_TOOL $REMOTE_TOOL
	link_test $REMOTE_TOOL $LOCAL_TOOL

	#Ensure the link is up on both sides before continuing
	write_file Y $LOCAL_TOOL/link_event
	write_file Y $REMOTE_TOOL/link_event

	for PEER_TRANS in $(ls $LOCAL_TOOL/peer_trans*); do
	PT=$(basename $PEER_TRANS)
	write_file $MW_SIZE $LOCAL_TOOL/$PT
	write_file $MW_SIZE $REMOTE_TOOL/$PT
	done

	doorbell_test $LOCAL_TOOL $REMOTE_TOOL
	doorbell_test $REMOTE_TOOL $LOCAL_TOOL
	scratchpad_test $LOCAL_TOOL $REMOTE_TOOL
	scratchpad_test $REMOTE_TOOL $LOCAL_TOOL

	for MW in $(ls $LOCAL_TOOL/mw*); do
	MW=$(basename $MW)

	mw_test $MW $LOCAL_TOOL $REMOTE_TOOL
	mw_test $MW $REMOTE_TOOL $LOCAL_TOOL
	done

	_modprobe -r ntb_tool
	}

	function ntb_pingpong_tests()
	{
	LOCAL_PP=$DEBUGFS/ntb_pingpong/$LOCAL_DEV
	REMOTE_PP=$REMOTE_HOST:$DEBUGFS/ntb_pingpong/$REMOTE_DEV

	echo "Starting ntb_pingpong tests..."

	_modprobe ntb_pingpong

	pingpong_test $LOCAL_PP $REMOTE_PP

	_modprobe -r ntb_pingpong
	}

	function ntb_perf_tests()
	{
	LOCAL_PERF=$DEBUGFS/ntb_perf/$LOCAL_DEV
	REMOTE_PERF=$REMOTE_HOST:$DEBUGFS/ntb_perf/$REMOTE_DEV

	echo "Starting ntb_perf tests..."

	perf_test 0

	if [[ $RUN_DMA_TESTS ]]; then
	perf_test 1
	fi
	}

	function cleanup()
	{
	set +e
	_modprobe -r ntb_tool 2> /dev/null
	_modprobe -r ntb_perf 2> /dev/null
	_modprobe -r ntb_pingpong 2> /dev/null
	_modprobe -r ntb_transport 2> /dev/null
	set -e
	}

	cleanup

	if ! [[ $$DONT_CLEANUP ]]; then
	trap cleanup EXIT
	fi

	if [ "$(id -u)" != "0" ]; then
	echo "This script must be run as root" 1>&2
	exit 1
	fi

	if [[ "$LIST_DEVS" == TRUE ]]; then
	echo "Local Devices:"
	ls -1 /sys/bus/ntb/devices
	echo

	if [[ "$REMOTE_HOST" != "" ]]; then
	echo "Remote Devices:"
	ssh $REMOTE_HOST ls -1 /sys/bus/ntb/devices
	fi

	exit 0
	fi

	if [[ "$LOCAL_DEV" == $"" ]] \|\| [[ "$REMOTE_DEV" == $"" ]]; then
	show_help
	exit 1
	fi

	ntb_tool_tests
	echo
	ntb_pingpong_tests
	echo
	ntb_perf_tests
	echo