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` 28status=1 # failure is the default! 29 30# get standard environment, filters and checks 31. ./common.rc 32. ./common.filter 33 34_supported_fmt generic 35_supported_proto nbd 36_supported_os Linux 37 38# Pick a TCP port based on our pid. This way multiple instances of this test 39# can run in parallel without conflicting. 40choose_tcp_port() { 41 echo $((($$ % 31744) + 1024)) # 1024 <= port < 32768 42} 43 44wait_for_tcp_port() { 45 while ! (netstat --tcp --listening --numeric | \ 46 grep "$1.*0\\.0\\.0\\.0:\\*.*LISTEN") >/dev/null 2>&1; do 47 sleep 0.1 48 done 49} 50 51check_disconnect() { 52 event=$1 53 when=$2 54 negotiation=$3 55 echo "=== Check disconnect $when $event ===" 56 echo 57 58 port=$(choose_tcp_port) 59 60 cat > "$TEST_DIR/nbd-fault-injector.conf" <<EOF 61[inject-error] 62event=$event 63when=$when 64EOF 65 66 if [ "$negotiation" = "--classic-negotiation" ]; then 67 extra_args=--classic-negotiation 68 nbd_url="nbd:127.0.0.1:$port" 69 else 70 nbd_url="nbd:127.0.0.1:$port:exportname=foo" 71 fi 72 73 $PYTHON nbd-fault-injector.py $extra_args "127.0.0.1:$port" "$TEST_DIR/nbd-fault-injector.conf" >/dev/null 2>&1 & 74 wait_for_tcp_port "127\\.0\\.0\\.1:$port" 75 $QEMU_IO -c "read 0 512" "$nbd_url" 2>&1 | _filter_qemu_io | _filter_nbd 76 77 echo 78} 79 80for event in neg1 "export" neg2 request reply data; do 81 for when in before after; do 82 check_disconnect "$event" "$when" 83 done 84 85 # Also inject short replies from the NBD server 86 case "$event" in 87 neg1) 88 for when in 8 16; do 89 check_disconnect "$event" "$when" 90 done 91 ;; 92 "export") 93 for when in 4 12 16; do 94 check_disconnect "$event" "$when" 95 done 96 ;; 97 neg2) 98 for when in 8 10; do 99 check_disconnect "$event" "$when" 100 done 101 ;; 102 reply) 103 for when in 4 8; do 104 check_disconnect "$event" "$when" 105 done 106 ;; 107 esac 108done 109 110# Also check classic negotiation without export information 111for when in before 8 16 24 28 after; do 112 check_disconnect "neg-classic" "$when" --classic-negotiation 113done 114 115# success, all done 116echo "*** done" 117rm -f $seq.full 118status=0 119