xref: /qemu/tests/qemu-iotests/083 (revision d072cdf3) 
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
52filter_nbd() {
53	# nbd.c error messages contain function names and line numbers that are prone
54	# to change.  Message ordering depends on timing between send and receive
55	# callbacks sometimes, making them unreliable.
56	#
57	# Filter out the TCP port number since this changes between runs.
58	sed -e 's#^.*nbd\.c:.*##g' \
59	    -e 's#nbd:127\.0\.0\.1:[^:]*:#nbd:127\.0\.0\.1:PORT:#g'
60}
61
62check_disconnect() {
63	event=$1
64	when=$2
65	negotiation=$3
66	echo "=== Check disconnect $when $event ==="
67	echo
68
69	port=$(choose_tcp_port)
70
71	cat > "$TEST_DIR/nbd-fault-injector.conf" <<EOF
72[inject-error]
73event=$event
74when=$when
75EOF
76
77	if [ "$negotiation" = "--classic-negotiation" ]; then
78		extra_args=--classic-negotiation
79		nbd_url="nbd:127.0.0.1:$port"
80	else
81		nbd_url="nbd:127.0.0.1:$port:exportname=foo"
82	fi
83
84	$PYTHON nbd-fault-injector.py $extra_args "127.0.0.1:$port" "$TEST_DIR/nbd-fault-injector.conf" 2>&1 >/dev/null &
85	wait_for_tcp_port "127\\.0\\.0\\.1:$port"
86	$QEMU_IO -c "read 0 512" "$nbd_url" 2>&1 | _filter_qemu_io | filter_nbd
87
88	echo
89}
90
91for event in neg1 "export" neg2 request reply data; do
92	for when in before after; do
93		check_disconnect "$event" "$when"
94	done
95
96	# Also inject short replies from the NBD server
97	case "$event" in
98	neg1)
99		for when in 8 16; do
100			check_disconnect "$event" "$when"
101		done
102		;;
103	"export")
104		for when in 4 12 16; do
105			check_disconnect "$event" "$when"
106		done
107		;;
108	neg2)
109		for when in 8 10; do
110			check_disconnect "$event" "$when"
111		done
112		;;
113	reply)
114		for when in 4 8; do
115			check_disconnect "$event" "$when"
116		done
117		;;
118	esac
119done
120
121# Also check classic negotiation without export information
122for when in before 8 16 24 28 after; do
123	check_disconnect "neg-classic" "$when" --classic-negotiation
124done
125
126# success, all done
127echo "*** done"
128rm -f $seq.full
129status=0
130