mirror of https://github.com/trapexit/mergerfs.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.2 KiB
78 lines
2.2 KiB
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
|
|
from posix_parity import compare_calls
|
|
from posix_parity import fail
|
|
from posix_parity import join
|
|
from posix_parity import touch
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
print("usage: TEST_posix_copy_file_range <mountpoint>", file=sys.stderr)
|
|
return 1
|
|
|
|
if not hasattr(os, "copy_file_range"):
|
|
return 0
|
|
|
|
mount = sys.argv[1]
|
|
|
|
with tempfile.TemporaryDirectory() as native:
|
|
merge_src = join(mount, "posix-cfr/src")
|
|
merge_dst = join(mount, "posix-cfr/dst")
|
|
native_src = join(native, "posix-cfr/src")
|
|
native_dst = join(native, "posix-cfr/dst")
|
|
|
|
payload = b"0123456789abcdefghijklmnopqrstuvwxyz"
|
|
touch(merge_src, payload)
|
|
touch(native_src, payload)
|
|
touch(merge_dst, b"")
|
|
touch(native_dst, b"")
|
|
|
|
msfd = os.open(merge_src, os.O_RDONLY)
|
|
mdfd = os.open(merge_dst, os.O_WRONLY)
|
|
nsfd = os.open(native_src, os.O_RDONLY)
|
|
ndfd = os.open(native_dst, os.O_WRONLY)
|
|
try:
|
|
err = compare_calls(
|
|
"copy_file_range success",
|
|
lambda: os.copy_file_range(msfd, mdfd, 16),
|
|
lambda: os.copy_file_range(nsfd, ndfd, 16),
|
|
lambda a, b: a == b,
|
|
)
|
|
if err:
|
|
return fail(err)
|
|
finally:
|
|
os.close(msfd)
|
|
os.close(mdfd)
|
|
os.close(nsfd)
|
|
os.close(ndfd)
|
|
|
|
with open(merge_dst, "rb") as mf, open(native_dst, "rb") as nf:
|
|
mdata = mf.read()
|
|
ndata = nf.read()
|
|
if mdata != ndata:
|
|
return fail(f"copy_file_range dst mismatch mergerfs={mdata!r} native={ndata!r}")
|
|
|
|
mdfd = os.open(merge_dst, os.O_WRONLY)
|
|
ndfd = os.open(native_dst, os.O_WRONLY)
|
|
try:
|
|
err = compare_calls(
|
|
"copy_file_range EBADF src",
|
|
lambda: os.copy_file_range(-1, mdfd, 1),
|
|
lambda: os.copy_file_range(-1, ndfd, 1),
|
|
)
|
|
if err:
|
|
return fail(err)
|
|
finally:
|
|
os.close(mdfd)
|
|
os.close(ndfd)
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|