]>
git.proxmox.com Git - mirror_qemu.git/blob - tests/qemu-iotests/147
4 # Test case for NBD's blockdev-add interface
6 # Copyright (C) 2016 Red Hat, Inc.
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.
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.
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/>.
28 from iotests
import cachemode
, aiomode
, imgfmt
, qemu_img
, qemu_nbd
, qemu_nbd_early_pipe
30 NBD_PORT_START
= 32768
31 NBD_PORT_END
= NBD_PORT_START
+ 1024
32 NBD_IPV6_PORT_START
= NBD_PORT_END
33 NBD_IPV6_PORT_END
= NBD_IPV6_PORT_START
+ 1024
35 test_img
= os
.path
.join(iotests
.test_dir
, 'test.img')
36 unix_socket
= os
.path
.join(iotests
.sock_dir
, 'nbd.socket')
39 def flatten_sock_addr(crumpled_address
):
40 result
= { 'type': crumpled_address
['type'] }
41 result
.update(crumpled_address
['data'])
45 class NBDBlockdevAddBase(iotests
.QMPTestCase
):
46 def blockdev_add_options(self
, address
, export
, node_name
):
47 options
= { 'node-name': node_name
,
54 if export
is not None:
55 options
['file']['export'] = export
58 def client_test(self
, filename
, address
, export
=None,
59 node_name
='nbd-blockdev', delete
=True):
60 bao
= self
.blockdev_add_options(address
, export
, node_name
)
61 result
= self
.vm
.qmp('blockdev-add', **bao
)
62 self
.assert_qmp(result
, 'return', {})
65 result
= self
.vm
.qmp('query-named-block-nodes')
66 for node
in result
['return']:
67 if node
['node-name'] == node_name
:
69 if isinstance(filename
, str):
70 self
.assert_qmp(node
, 'image/filename', filename
)
72 self
.assert_json_filename_equal(node
['image']['filename'],
75 self
.assertTrue(found
)
78 result
= self
.vm
.qmp('blockdev-del', node_name
=node_name
)
79 self
.assert_qmp(result
, 'return', {})
82 class QemuNBD(NBDBlockdevAddBase
):
84 qemu_img('create', '-f', iotests
.imgfmt
, test_img
, '64k')
85 self
.vm
= iotests
.VM()
92 os
.remove(unix_socket
)
96 def _try_server_up(self
, *args
):
97 status
, msg
= qemu_nbd_early_pipe('-f', imgfmt
, test_img
, *args
)
100 if 'Address already in use' in msg
:
104 def _server_up(self
, *args
):
105 self
.assertTrue(self
._try
_server
_up
(*args
))
109 nbd_port
= random
.randrange(NBD_PORT_START
, NBD_PORT_END
)
110 if self
._try
_server
_up
('-b', 'localhost', '-p', str(nbd_port
)):
113 address
= { 'type': 'inet',
116 'port': str(nbd_port
)
118 self
.client_test('nbd://localhost:%i' % nbd_port
,
119 flatten_sock_addr(address
))
122 self
._server
_up
('-k', unix_socket
)
123 address
= { 'type': 'unix',
124 'data': { 'path': unix_socket
} }
125 self
.client_test('nbd+unix://?socket=' + unix_socket
,
126 flatten_sock_addr(address
))
129 class BuiltinNBD(NBDBlockdevAddBase
):
131 qemu_img('create', '-f', iotests
.imgfmt
, test_img
, '64k')
132 self
.vm
= iotests
.VM()
134 self
.server
= iotests
.VM('.server')
135 self
.server
.add_drive_raw('if=none,id=nbd-export,' +
136 'file=%s,' % test_img
+
137 'format=%s,' % imgfmt
+
138 'cache=%s,' % cachemode
+
144 self
.server
.shutdown()
147 os
.remove(unix_socket
)
151 # Returns False on EADDRINUSE; fails an assertion on other errors.
152 # Returns True on success.
153 def _try_server_up(self
, address
, export_name
=None, export_name2
=None):
154 result
= self
.server
.qmp('nbd-server-start', addr
=address
)
155 if 'error' in result
and \
156 'Address already in use' in result
['error']['desc']:
158 self
.assert_qmp(result
, 'return', {})
160 if export_name
is None:
161 result
= self
.server
.qmp('nbd-server-add', device
='nbd-export')
163 result
= self
.server
.qmp('nbd-server-add', device
='nbd-export',
165 self
.assert_qmp(result
, 'return', {})
167 if export_name2
is not None:
168 result
= self
.server
.qmp('nbd-server-add', device
='nbd-export',
170 self
.assert_qmp(result
, 'return', {})
174 def _server_up(self
, address
, export_name
=None, export_name2
=None):
175 self
.assertTrue(self
._try
_server
_up
(address
, export_name
, export_name2
))
177 def _server_down(self
):
178 result
= self
.server
.qmp('nbd-server-stop')
179 self
.assert_qmp(result
, 'return', {})
181 def do_test_inet(self
, export_name
=None):
183 nbd_port
= random
.randrange(NBD_PORT_START
, NBD_PORT_END
)
184 address
= { 'type': 'inet',
187 'port': str(nbd_port
)
189 if self
._try
_server
_up
(address
, export_name
):
192 export_name
= export_name
or 'nbd-export'
193 self
.client_test('nbd://localhost:%i/%s' % (nbd_port
, export_name
),
194 flatten_sock_addr(address
), export_name
)
197 def test_inet_default_export_name(self
):
200 def test_inet_same_export_name(self
):
201 self
.do_test_inet('nbd-export')
203 def test_inet_different_export_name(self
):
204 self
.do_test_inet('shadow')
206 def test_inet_two_exports(self
):
208 nbd_port
= random
.randrange(NBD_PORT_START
, NBD_PORT_END
)
209 address
= { 'type': 'inet',
212 'port': str(nbd_port
)
214 if self
._try
_server
_up
(address
, 'exp1', 'exp2'):
217 self
.client_test('nbd://localhost:%i/%s' % (nbd_port
, 'exp1'),
218 flatten_sock_addr(address
), 'exp1', 'node1', False)
219 self
.client_test('nbd://localhost:%i/%s' % (nbd_port
, 'exp2'),
220 flatten_sock_addr(address
), 'exp2', 'node2', False)
221 result
= self
.vm
.qmp('blockdev-del', node_name
='node1')
222 self
.assert_qmp(result
, 'return', {})
223 result
= self
.vm
.qmp('blockdev-del', node_name
='node2')
224 self
.assert_qmp(result
, 'return', {})
227 def test_inet6(self
):
229 socket
.getaddrinfo("::0", "0", socket
.AF_INET6
,
230 socket
.SOCK_STREAM
, socket
.IPPROTO_TCP
,
231 socket
.AI_ADDRCONFIG | socket
.AI_CANONNAME
)
232 except socket
.gaierror
:
233 # IPv6 not available, skip
237 nbd_port
= random
.randrange(NBD_IPV6_PORT_START
, NBD_IPV6_PORT_END
)
238 address
= { 'type': 'inet',
241 'port': str(nbd_port
),
245 if self
._try
_server
_up
(address
):
248 filename
= { 'driver': 'raw',
251 'export': 'nbd-export',
252 'server': flatten_sock_addr(address
)
254 self
.client_test(filename
, flatten_sock_addr(address
), 'nbd-export')
258 address
= { 'type': 'unix',
259 'data': { 'path': unix_socket
} }
260 self
._server
_up
(address
)
261 self
.client_test('nbd+unix:///nbd-export?socket=' + unix_socket
,
262 flatten_sock_addr(address
), 'nbd-export')
266 self
._server
_up
({ 'type': 'unix',
267 'data': { 'path': unix_socket
} })
269 sockfd
= socket
.socket(socket
.AF_UNIX
, socket
.SOCK_STREAM
)
270 sockfd
.connect(unix_socket
)
272 result
= self
.vm
.send_fd_scm(fd
=sockfd
.fileno())
273 self
.assertEqual(result
, 0, 'Failed to send socket FD')
275 result
= self
.vm
.qmp('getfd', fdname
='nbd-fifo')
276 self
.assert_qmp(result
, 'return', {})
278 address
= { 'type': 'fd',
279 'data': { 'str': 'nbd-fifo' } }
280 filename
= { 'driver': 'raw',
283 'export': 'nbd-export',
284 'server': flatten_sock_addr(address
)
286 self
.client_test(filename
, flatten_sock_addr(address
), 'nbd-export')
291 if __name__
== '__main__':
292 iotests
.main(supported_fmts
=['raw'],
293 supported_protocols
=['nbd'])