fc6e58d9e6
Add SPDX header to python files with the 'py' extension in the test directory. Signed-off-by: Albert Esteve <aesteve@redhat.com>
94 lines
2.1 KiB
Python
94 lines
2.1 KiB
Python
# SPDX-FileCopyrightText: Red Hat, Inc.
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
"""
|
|
Test script for file.BlockIO() and file.FileIO(), simulating fallocate
|
|
and blkdiscard --zeroout.
|
|
"""
|
|
|
|
import argparse
|
|
import time
|
|
import urllib.parse
|
|
|
|
from ovirt_imageio._internal.backends import file
|
|
|
|
|
|
def humansize(s):
|
|
if s.isdigit():
|
|
return int(s)
|
|
value, unit = s[:-1], s[-1]
|
|
value = int(value)
|
|
unit = unit.lower()
|
|
if unit == "k":
|
|
return value * 1024
|
|
elif unit == "m":
|
|
return value * 1024**2
|
|
elif unit == "g":
|
|
return value * 1024**3
|
|
else:
|
|
raise ValueError("Unsupported unit: %r" % unit)
|
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument(
|
|
"-l", "--length",
|
|
dest="length",
|
|
type=humansize,
|
|
help=("The number of bytes to zero (counting from the starting point) "
|
|
"(default entire device or file)"))
|
|
|
|
parser.add_argument(
|
|
"-o", "--offset",
|
|
dest="offset",
|
|
type=humansize,
|
|
default=0,
|
|
help="Byte offset into the device from which to start zeroing (default 0)")
|
|
|
|
parser.add_argument(
|
|
"-p", "--step",
|
|
dest="step",
|
|
type=humansize,
|
|
help=("The number of bytes to zero within one iteration. The default "
|
|
"is to discard all by one ioctl call"))
|
|
|
|
parser.add_argument(
|
|
"-s", "--sparse",
|
|
dest="sparse",
|
|
action="store_true",
|
|
help="Deallocate zeroed space (punch holes)")
|
|
|
|
parser.add_argument(
|
|
"filename",
|
|
help="file or block device to fill with zeros")
|
|
|
|
args = parser.parse_args()
|
|
|
|
url = urllib.parse.urlparse("file://" + args.filename)
|
|
|
|
start_time = time.monotonic()
|
|
|
|
with file.open(url, "r+", sparse=args.sparse) as f:
|
|
if args.length is None:
|
|
args.length = f.size() - args.offset
|
|
|
|
if args.step is None:
|
|
args.step = args.length
|
|
|
|
f.seek(args.offset)
|
|
|
|
count = args.length
|
|
while count:
|
|
step = min(args.step, count)
|
|
f.zero(step)
|
|
count -= step
|
|
|
|
f.flush()
|
|
|
|
elapsed_time = time.monotonic() - start_time
|
|
|
|
print("Zero %.2f GiB in %.3f seconds (%.2f GiB/s)" % (
|
|
float(args.length) / 1024**3,
|
|
elapsed_time,
|
|
float(args.length) / 1024**3 / elapsed_time))
|