def test_qcow2_write_read(tmpdir):
image = str(tmpdir.join("image"))
sock = nbd.UnixAddress(tmpdir.join("sock"))
offset = 1024**2
data = b"can read and write qcow2"
subprocess.check_call(["qemu-img", "create", "-f", "qcow2", image, "1g"])
with qemu_nbd.run(image, "qcow2", sock):
with nbd.Client(sock) as c:
c.write(offset, data)
c.flush()
with nbd.Client(sock) as c:
assert c.read(offset, len(data)) == data
def test_full_backup_single_image(tmpdir, fmt, transport):
chunk_size = 1024**2
disk_size = 5 * chunk_size
# Create disk
disk = str(tmpdir.join("disk." + fmt))
subprocess.check_call([
"qemu-img",
"create",
"-f",
fmt,
disk,
str(disk_size),
])
# Pupulate disk with data.
with qemu_nbd.open(disk, fmt) as d:
for i in range(0, disk_size, chunk_size):
d.write(i, b"%d\n" % i)
d.flush()
if transport == "unix":
sock = nbd.UnixAddress(tmpdir.join("sock"))
else:
sock = nbd.TCPAddress("localhost", testutil.random_tcp_port())
# Start full backup and copy the data, veifying what we read.
with backup.full_backup(tmpdir, disk, fmt, sock):
with nbd.Client(sock, "sda") as c:
log.debug("Backing up data with nbd client")
for i in range(0, disk_size, chunk_size):
data = c.read(i, chunk_size)
assert data.startswith(b"%d\n\0" % i)
def open(image, fmt, read_only=False):
"""
Open nbd client for accessing image using qemu-nbd.
"""
sock = nbd.UnixAddress(image + ".sock")
with run(image, fmt, sock, read_only=read_only):
with nbd.Client(sock) as c:
yield c
def test_handshake(tmpdir, export, fmt):
image = str(tmpdir.join("image"))
subprocess.check_call(["qemu-img", "create", "-f", fmt, image, "1g"])
sock = nbd.UnixAddress(tmpdir.join("sock"))
with qemu_nbd.run(image, fmt, sock, export_name=export):
if export:
c = nbd.Client(sock, export)
else:
c = nbd.Client(sock)
with c:
# TODO: test transmission_flags?
assert c.export_size == 1024**3
assert c.minimum_block_size == 1
assert c.preferred_block_size == 4096
assert c.maximum_block_size == 32 * 1024**2
assert c.base_allocation
def test_zero_min_block_size(tmpdir, format):
offset = 1024**2
image = str(tmpdir.join("image"))
sock = nbd.UnixAddress(tmpdir.join("sock"))
subprocess.check_call(["qemu-img", "create", "-f", format, image, "1g"])
with qemu_nbd.run(image, format, sock):
# Fill range with data
with nbd.Client(sock) as c:
size = c.minimum_block_size
c.write(offset, b"x" * size)
c.flush()
# Zero range using minimum block size
with nbd.Client(sock) as c:
c.zero(offset, size)
c.flush()
with nbd.Client(sock) as c:
assert c.read(offset, size) == b"\0" * size
def test_zero(tmpdir, format):
size = 2 * 1024**2
offset = 1024**2
image = str(tmpdir.join("image"))
sock = nbd.UnixAddress(tmpdir.join("sock"))
subprocess.check_call(
["qemu-img", "create", "-f", format, image,
str(size)])
with qemu_nbd.run(image, format, sock):
# Fill image with data
with nbd.Client(sock) as c:
c.write(0, b"x" * size)
c.flush()
# Zero a range
with nbd.Client(sock) as c:
c.zero(offset, 4096)
c.flush()
with nbd.Client(sock) as c:
assert c.read(offset, 4096) == b"\0" * 4096
def test_raw_read(tmpdir):
image = str(tmpdir.join("image"))
sock = nbd.UnixAddress(tmpdir.join("sock"))
offset = 1024**2
data = b"can read from raw"
with open(image, "wb") as f:
f.truncate(1024**3)
f.seek(offset)
f.write(data)
with qemu_nbd.run(image, "raw", sock):
with nbd.Client(sock) as c:
assert c.read(offset, len(data)) == data
def test_raw_write(tmpdir):
image = str(tmpdir.join("image"))
with open(image, "wb") as f:
f.truncate(1024**3)
sock = nbd.UnixAddress(tmpdir.join("sock"))
offset = 1024**2
data = b"can write to raw"
with qemu_nbd.run(image, "raw", sock):
with nbd.Client(sock) as c:
c.write(offset, data)
c.flush()
with open(image, "rb") as f:
f.seek(offset)
assert f.read(len(data)) == data
def test_full_backup_handshake(tmpdir, fmt, transport):
image = str(tmpdir.join("image"))
subprocess.check_call(["qemu-img", "create", "-f", fmt, image, "1g"])
if transport == "unix":
sock = nbd.UnixAddress(tmpdir.join("sock"))
else:
sock = nbd.TCPAddress("localhost", testutil.random_tcp_port())
with backup.full_backup(tmpdir, image, fmt, sock):
with nbd.Client(sock, "sda") as c:
# TODO: test transmission_flags?
assert c.export_size == 1024**3
assert c.minimum_block_size == 1
assert c.preferred_block_size == 4096
assert c.maximum_block_size == 32 * 1024**2
assert c.base_allocation
def test_full_backup_complete_chain(tmpdir, transport):
depth = 3
chunk_size = 1024**2
disk_size = depth * chunk_size
for i in range(depth):
# Create disk based on previous one.
disk = str(tmpdir.join("disk.%d" % i))
cmd = ["qemu-img", "create", "-f", "qcow2"]
if i > 0:
cmd.append("-b")
cmd.append("disk.%d" % (i - 1))
cmd.append(disk)
cmd.append(str(disk_size))
subprocess.check_call(cmd)
# This data can be read only from this disk.
with qemu_nbd.open(disk, "qcow2") as d:
d.write(i * chunk_size, b"%d\n" % i)
d.flush()
if transport == "unix":
sock = nbd.UnixAddress(tmpdir.join("sock"))
else:
sock = nbd.TCPAddress("localhost", testutil.random_tcp_port())
# Start full backup and copy the data, veifying what we read.
with backup.full_backup(tmpdir, disk, "qcow2", sock):
with nbd.Client(sock, "sda") as c:
log.debug("Backing up data with nbd client")
for i in range(depth):
# Every chunk comes from different image.
data = c.read(i * chunk_size, chunk_size)
assert data.startswith(b"%d\n\0" % i)
