tests/qemu-iotests/277

   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 nbd_sock, conf_file = file_path('nbd-sock', 'nbd-fault-injector.conf')
  31
  32
  33 def make_conf_file(event):
  34     """
  35     Create configuration file for the nbd-fault-injector.py
  36
  37     :param event: which event the server should close a connection on
  38     """
  39     with open(conf_file, 'w') as conff:
  40         conff.write('[inject-error]\nevent={}\nwhen=after'.format(event))
  41
  42
  43 def start_server_NBD(event):
  44     make_conf_file(event)
  45
  46     srv = subprocess.Popen(['./nbd-fault-injector.py', '--classic-negotiation',
  47                            nbd_sock, conf_file], stdout=subprocess.PIPE,
  48                            stderr=subprocess.STDOUT, universal_newlines=True)
  49     line = srv.stdout.readline()
  50     if 'Listening on ' in line:
  51         log('NBD server: started')
  52     else:
  53         log('NBD server: ' + line.rstrip())
  54
  55     return srv
  56
  57
  58 def start_client_NBD():
  59     log('NBD client: QEMU-IO write')
  60     args = iotests.qemu_io_args_no_fmt + \
  61         ['-c', 'write -P 0x7 0 3M', '--image-opts',
  62          'driver=nbd,server.type=unix,server.path={},'
  63          'reconnect-delay=7'.format(nbd_sock)]
  64     clt = subprocess.Popen(args, stdout=subprocess.PIPE,
  65                            stderr=subprocess.STDOUT,
  66                            universal_newlines=True)
  67     return clt
  68
  69
  70 def check_proc_NBD(proc, connector):
  71     try:
  72         outs, errs = proc.communicate(timeout=10)
  73
  74         if proc.returncode < 0:
  75             log('NBD {}: EXIT SIGNAL {}\n'.format(connector, proc.returncode))
  76             log(outs)
  77         else:
  78             msg = outs.split('\n', 1)
  79             log('NBD {}: {}'.format(connector, msg[0]))
  80
  81     except subprocess.TimeoutExpired:
  82         proc.kill()
  83         log('NBD {}: ERROR timeout expired'.format(connector))
  84     finally:
  85         if connector == 'server':
  86             os.remove(nbd_sock)
  87             os.remove(conf_file)
  88
  89
  90 srv = start_server_NBD('data')
  91 clt = start_client_NBD()
  92 # The server should close the connection after a client write request
  93 check_proc_NBD(srv, 'server')
  94 # Start the NBD server again
  95 srv = start_server_NBD('reply')
  96 # The client should reconnect and complete the write operation
  97 check_proc_NBD(clt, 'client')
  98 # Make it sure that server terminated
  99 check_proc_NBD(srv, 'server')