mirror of https://github.com/borgbackup/borg.git
85 lines
3.0 KiB
Python
85 lines
3.0 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
from ..platform.xattr import buffer, split_lstring
|
|
from ..xattr import is_enabled, getxattr, setxattr, listxattr
|
|
from ..platformflags import is_linux
|
|
|
|
|
|
@pytest.fixture()
|
|
def tempfile_symlink(tmp_path):
|
|
if not is_enabled(tmp_path):
|
|
pytest.skip("xattr not enabled on filesystem")
|
|
with open(os.fspath(tmp_path / "xattr"), "w") as temp_file:
|
|
symlink = temp_file.name + ".symlink"
|
|
os.symlink(temp_file.name, symlink)
|
|
yield temp_file, symlink
|
|
|
|
|
|
def assert_equal_se(is_x, want_x):
|
|
# check 2 xattr lists for equality, but ignore security.selinux attr
|
|
is_x = set(is_x) - {b"security.selinux"}
|
|
want_x = set(want_x)
|
|
assert is_x == want_x
|
|
|
|
|
|
def test(tempfile_symlink):
|
|
temp_file, symlink = tempfile_symlink
|
|
tmp_fn = os.fsencode(temp_file.name)
|
|
tmp_lfn = os.fsencode(symlink)
|
|
tmp_fd = temp_file.fileno()
|
|
assert_equal_se(listxattr(tmp_fn), [])
|
|
assert_equal_se(listxattr(tmp_fd), [])
|
|
assert_equal_se(listxattr(tmp_lfn), [])
|
|
setxattr(tmp_fn, b"user.foo", b"bar")
|
|
setxattr(tmp_fd, b"user.bar", b"foo")
|
|
setxattr(tmp_fn, b"user.empty", b"")
|
|
if not is_linux:
|
|
# linux does not allow setting user.* xattrs on symlinks
|
|
setxattr(tmp_lfn, b"user.linkxattr", b"baz")
|
|
assert_equal_se(listxattr(tmp_fn), [b"user.foo", b"user.bar", b"user.empty"])
|
|
assert_equal_se(listxattr(tmp_fd), [b"user.foo", b"user.bar", b"user.empty"])
|
|
assert_equal_se(listxattr(tmp_lfn, follow_symlinks=True), [b"user.foo", b"user.bar", b"user.empty"])
|
|
if not is_linux:
|
|
assert_equal_se(listxattr(tmp_lfn), [b"user.linkxattr"])
|
|
assert getxattr(tmp_fn, b"user.foo") == b"bar"
|
|
assert getxattr(tmp_fd, b"user.foo") == b"bar"
|
|
assert getxattr(tmp_lfn, b"user.foo", follow_symlinks=True) == b"bar"
|
|
if not is_linux:
|
|
assert getxattr(tmp_lfn, b"user.linkxattr") == b"baz"
|
|
assert getxattr(tmp_fn, b"user.empty") == b""
|
|
|
|
|
|
def test_listxattr_buffer_growth(tempfile_symlink):
|
|
temp_file, symlink = tempfile_symlink
|
|
tmp_fn = os.fsencode(temp_file.name)
|
|
# make it work even with ext4, which imposes rather low limits
|
|
buffer.resize(size=64, init=True)
|
|
# xattr raw key list will be > 64
|
|
keys = [b"user.attr%d" % i for i in range(20)]
|
|
for key in keys:
|
|
setxattr(tmp_fn, key, b"x")
|
|
got_keys = listxattr(tmp_fn)
|
|
assert_equal_se(got_keys, keys)
|
|
assert len(buffer) > 64
|
|
|
|
|
|
def test_getxattr_buffer_growth(tempfile_symlink):
|
|
temp_file, symlink = tempfile_symlink
|
|
tmp_fn = os.fsencode(temp_file.name)
|
|
# make it work even with ext4, which imposes rather low limits
|
|
buffer.resize(size=64, init=True)
|
|
value = b"x" * 126
|
|
setxattr(tmp_fn, b"user.big", value)
|
|
got_value = getxattr(tmp_fn, b"user.big")
|
|
assert value == got_value
|
|
assert len(buffer) == 128
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"lstring, splitted", [(b"", []), (b"\x00", [b""]), (b"\x01a", [b"a"]), (b"\x01a\x02cd", [b"a", b"cd"])]
|
|
)
|
|
def test_split_lstring(lstring, splitted):
|
|
assert split_lstring(lstring) == splitted
|