xref: /qemu/tests/qemu-iotests/083 (revision 7a4e543d) 
1#!/bin/bash
2#
3# Test NBD client unexpected disconnect
4#
5# Copyright Red Hat, Inc. 2014
6#
7# This program is free software; you can redistribute it and/or modify
8# it under the terms of the GNU General Public License as published by
9# the Free Software Foundation; either version 2 of the License, or
10# (at your option) any later version.
11#
12# This program is distributed in the hope that it will be useful,
13# but WITHOUT ANY WARRANTY; without even the implied warranty of
14# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15# GNU General Public License for more details.
16#
17# You should have received a copy of the GNU General Public License
18# along with this program.  If not, see <http://www.gnu.org/licenses/>.
19#
20
21# creator
22owner=stefanha@redhat.com
23
24seq=`basename $0`
25echo "QA output created by $seq"
26
27here=`pwd`
28tmp=/tmp/$$
29status=1	# failure is the default!
30
31# get standard environment, filters and checks
32. ./common.rc
33. ./common.filter
34
35_supported_fmt generic
36_supported_proto nbd
37_supported_os Linux
38
39# Pick a TCP port based on our pid.  This way multiple instances of this test
40# can run in parallel without conflicting.
41choose_tcp_port() {
42	echo $((($$ % 31744) + 1024)) # 1024 <= port < 32768
43}
44
45wait_for_tcp_port() {
46	while ! (netstat --tcp --listening --numeric | \
47		 grep "$1.*0\\.0\\.0\\.0:\\*.*LISTEN") 2>&1 >/dev/null; do
48		sleep 0.1
49	done
50}
51
52check_disconnect() {
53	event=$1
54	when=$2
55	negotiation=$3
56	echo "=== Check disconnect $when $event ==="
57	echo
58
59	port=$(choose_tcp_port)
60
61	cat > "$TEST_DIR/nbd-fault-injector.conf" <<EOF
62[inject-error]
63event=$event
64when=$when
65EOF
66
67	if [ "$negotiation" = "--classic-negotiation" ]; then
68		extra_args=--classic-negotiation
69		nbd_url="nbd:127.0.0.1:$port"
70	else
71		nbd_url="nbd:127.0.0.1:$port:exportname=foo"
72	fi
73
74	$PYTHON nbd-fault-injector.py $extra_args "127.0.0.1:$port" "$TEST_DIR/nbd-fault-injector.conf" 2>&1 >/dev/null &
75	wait_for_tcp_port "127\\.0\\.0\\.1:$port"
76	$QEMU_IO -c "read 0 512" "$nbd_url" 2>&1 | _filter_qemu_io | _filter_nbd
77
78	echo
79}
80
81for event in neg1 "export" neg2 request reply data; do
82	for when in before after; do
83		check_disconnect "$event" "$when"
84	done
85
86	# Also inject short replies from the NBD server
87	case "$event" in
88	neg1)
89		for when in 8 16; do
90			check_disconnect "$event" "$when"
91		done
92		;;
93	"export")
94		for when in 4 12 16; do
95			check_disconnect "$event" "$when"
96		done
97		;;
98	neg2)
99		for when in 8 10; do
100			check_disconnect "$event" "$when"
101		done
102		;;
103	reply)
104		for when in 4 8; do
105			check_disconnect "$event" "$when"
106		done
107		;;
108	esac
109done
110
111# Also check classic negotiation without export information
112for when in before 8 16 24 28 after; do
113	check_disconnect "neg-classic" "$when" --classic-negotiation
114done
115
116# success, all done
117echo "*** done"
118rm -f $seq.full
119status=0
120