| 1 | #!/usr/bin/env python3 |
| 2 | # group: rw quick |
| 3 | # |
| 4 | # Test NBD client reconnection |
| 5 | # |
| 6 | # Copyright (c) 2019 Virtuozzo International GmbH |
| 7 | # |
| 8 | # This program is free software; you can redistribute it and/or modify |
| 9 | # it under the terms of the GNU General Public License as published by |
| 10 | # the Free Software Foundation; either version 2 of the License, or |
| 11 | # (at your option) any later version. |
| 12 | # |
| 13 | # This program is distributed in the hope that it will be useful, |
| 14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 16 | # GNU General Public License for more details. |
| 17 | # |
| 18 | # You should have received a copy of the GNU General Public License |
| 19 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
| 20 | # |
| 21 | |
| 22 | import os |
| 23 | import subprocess |
| 24 | import iotests |
| 25 | from iotests import file_path, log |
| 26 | |
| 27 | iotests.script_initialize() |
| 28 | |
| 29 | |
| 30 | conf_file = file_path('nbd-fault-injector.conf') |
| 31 | nbd_sock = file_path('nbd-sock', base_dir=iotests.sock_dir) |
| 32 | |
| 33 | |
| 34 | def make_conf_file(event): |
| 35 | """ |
| 36 | Create configuration file for the nbd-fault-injector.py |
| 37 | |
| 38 | :param event: which event the server should close a connection on |
| 39 | """ |
| 40 | with open(conf_file, 'w') as conff: |
| 41 | conff.write('[inject-error]\nevent={}\nwhen=after'.format(event)) |
| 42 | |
| 43 | |
| 44 | def start_server_NBD(event): |
| 45 | make_conf_file(event) |
| 46 | |
| 47 | srv = subprocess.Popen(['./nbd-fault-injector.py', '--classic-negotiation', |
| 48 | nbd_sock, conf_file], stdout=subprocess.PIPE, |
| 49 | stderr=subprocess.STDOUT, universal_newlines=True) |
| 50 | line = srv.stdout.readline() |
| 51 | if 'Listening on ' in line: |
| 52 | log('NBD server: started') |
| 53 | else: |
| 54 | log('NBD server: ' + line.rstrip()) |
| 55 | |
| 56 | return srv |
| 57 | |
| 58 | |
| 59 | def start_client_NBD(): |
| 60 | log('NBD client: QEMU-IO write') |
| 61 | args = iotests.qemu_io_args_no_fmt + \ |
| 62 | ['-c', 'write -P 0x7 0 3M', '--image-opts', |
| 63 | 'driver=nbd,server.type=unix,server.path={},' |
| 64 | 'reconnect-delay=7'.format(nbd_sock)] |
| 65 | clt = subprocess.Popen(args, stdout=subprocess.PIPE, |
| 66 | stderr=subprocess.STDOUT, |
| 67 | universal_newlines=True) |
| 68 | return clt |
| 69 | |
| 70 | |
| 71 | def check_proc_NBD(proc, connector): |
| 72 | try: |
| 73 | outs, errs = proc.communicate(timeout=10) |
| 74 | |
| 75 | if proc.returncode < 0: |
| 76 | log('NBD {}: EXIT SIGNAL {}\n'.format(connector, proc.returncode)) |
| 77 | log(outs) |
| 78 | else: |
| 79 | msg = outs.split('\n', 1) |
| 80 | log('NBD {}: {}'.format(connector, msg[0])) |
| 81 | |
| 82 | except subprocess.TimeoutExpired: |
| 83 | proc.kill() |
| 84 | log('NBD {}: ERROR timeout expired'.format(connector)) |
| 85 | finally: |
| 86 | if connector == 'server': |
| 87 | os.remove(nbd_sock) |
| 88 | os.remove(conf_file) |
| 89 | |
| 90 | |
| 91 | srv = start_server_NBD('data') |
| 92 | clt = start_client_NBD() |
| 93 | # The server should close the connection after a client write request |
| 94 | check_proc_NBD(srv, 'server') |
| 95 | # Start the NBD server again |
| 96 | srv = start_server_NBD('reply') |
| 97 | # The client should reconnect and complete the write operation |
| 98 | check_proc_NBD(clt, 'client') |
| 99 | # Make it sure that server terminated |
| 100 | check_proc_NBD(srv, 'server') |