1#!/usr/bin/env 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 27status=1 # failure is the default! 28 29_cleanup() 30{ 31 rm -f "$SOCK_DIR/nbd.sock" 32 rm -f nbd-fault-injector.out 33 rm -f nbd-fault-injector.conf 34} 35trap "_cleanup; exit \$status" 0 1 2 3 15 36 37# get standard environment, filters and checks 38. ./common.rc 39. ./common.filter 40 41_supported_fmt raw 42_supported_proto nbd 43_supported_os Linux 44 45check_disconnect() { 46 local event export_name=foo extra_args nbd_addr nbd_url proto when 47 48 while true; do 49 case $1 in 50 --classic-negotiation) 51 shift 52 extra_args=--classic-negotiation 53 export_name= 54 ;; 55 --tcp) 56 shift 57 proto=tcp 58 ;; 59 --unix) 60 shift 61 proto=unix 62 ;; 63 *) 64 break 65 ;; 66 esac 67 done 68 69 event=$1 70 when=$2 71 echo "=== Check disconnect $when $event ===" 72 echo 73 74 cat > "$TEST_DIR/nbd-fault-injector.conf" <<EOF 75[inject-error] 76event=$event 77when=$when 78EOF 79 80 if [ "$proto" = "tcp" ]; then 81 nbd_addr="127.0.0.1:0" 82 else 83 nbd_addr="$SOCK_DIR/nbd.sock" 84 fi 85 86 rm -f "$SOCK_DIR/nbd.sock" 87 88 echo > "$TEST_DIR/nbd-fault-injector.out" 89 $PYTHON nbd-fault-injector.py $extra_args "$nbd_addr" "$TEST_DIR/nbd-fault-injector.conf" >"$TEST_DIR/nbd-fault-injector.out" 2>&1 & 90 91 # Wait for server to be ready 92 while ! grep -q 'Listening on ' "$TEST_DIR/nbd-fault-injector.out"; do 93 sleep 0.1 94 done 95 96 # Extract the final address (port number has now been assigned in tcp case) 97 nbd_addr=$(sed -n 's/^Listening on //p' \ 98 "$TEST_DIR/nbd-fault-injector.out") 99 100 if [ "$proto" = "tcp" ]; then 101 nbd_url="nbd+tcp://$nbd_addr/$export_name" 102 else 103 nbd_url="nbd+unix:///$export_name?socket=$nbd_addr" 104 fi 105 106 $QEMU_IO -c "read 0 512" "$nbd_url" 2>&1 | _filter_qemu_io | _filter_nbd 107 108 echo 109} 110 111for proto in tcp unix; do 112 for event in neg1 "export" neg2 request reply data; do 113 for when in before after; do 114 check_disconnect "--$proto" "$event" "$when" 115 done 116 117 # Also inject short replies from the NBD server 118 case "$event" in 119 neg1) 120 for when in 8 16; do 121 check_disconnect "--$proto" "$event" "$when" 122 done 123 ;; 124 "export") 125 for when in 4 12 16; do 126 check_disconnect "--$proto" "$event" "$when" 127 done 128 ;; 129 neg2) 130 for when in 8 10; do 131 check_disconnect "--$proto" "$event" "$when" 132 done 133 ;; 134 reply) 135 for when in 4 8; do 136 check_disconnect "--$proto" "$event" "$when" 137 done 138 ;; 139 esac 140 done 141 142 # Also check classic negotiation without export information 143 for when in before 8 16 24 28 after; do 144 check_disconnect "--$proto" --classic-negotiation "neg-classic" "$when" 145 done 146done 147 148# success, all done 149echo "*** done" 150rm -f $seq.full 151status=0 152