From: strudelll <strudelll@etersoft.ru> Detect reflink-capable filesystems (btrfs or successful FICLONERANGE) and set FILE_SUPPORTS_BLOCK_REFCOUNTING in FileFsAttributeInformation. Add a conformance test that uses FSCTL_DUPLICATE_EXTENTS_TO_FILE and expects the flag when reflink works, skipping otherwise. --- dlls/kernel32/tests/volume.c | 86 ++++++++++++++++++++++++++++++++++++ dlls/ntdll/unix/file.c | 34 +++++++++++++- 2 files changed, 118 insertions(+), 2 deletions(-) diff --git a/dlls/kernel32/tests/volume.c b/dlls/kernel32/tests/volume.c index 975aa7bb69d..5573d5e1cb0 100644 --- a/dlls/kernel32/tests/volume.c +++ b/dlls/kernel32/tests/volume.c @@ -2014,6 +2014,91 @@ static void test_GetVolumeInformationByHandle(void) CloseHandle( file ); } +static void test_block_refcounting_flag(void) +{ + WCHAR tmp_path[MAX_PATH], src_path[MAX_PATH], dst_path[MAX_PATH]; + HANDLE src = INVALID_HANDLE_VALUE, dst = INVALID_HANDLE_VALUE; + DWORD sectors_per_cluster = 0, bytes_per_sector = 0, cluster_size, flags; + DWORD written, bytes_returned; + DUPLICATE_EXTENTS_DATA dup = {0}; + char *buffer = NULL; + BOOL ret; + + if (!GetTempPathW( ARRAY_SIZE(tmp_path), tmp_path ) || !tmp_path[0]) + { + skip("GetTempPathW failed, skipping block refcounting test.\n"); + return; + } + + if (!GetTempFileNameW( tmp_path, L"cln", 0, src_path )) + { + skip("GetTempFileNameW failed, skipping block refcounting test.\n"); + return; + } + + if (!GetTempFileNameW( tmp_path, L"cln", 0, dst_path )) + { + DeleteFileW( src_path ); + skip("GetTempFileNameW failed, skipping block refcounting test.\n"); + return; + } + + src = CreateFileW( src_path, GENERIC_READ | GENERIC_WRITE, + FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, + NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_TEMPORARY, NULL ); + ok(src != INVALID_HANDLE_VALUE, "failed to create source file, err %lu\n", GetLastError()); + if (src == INVALID_HANDLE_VALUE) + goto cleanup; + + dst = CreateFileW( dst_path, GENERIC_READ | GENERIC_WRITE, + FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, + NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_TEMPORARY, NULL ); + ok(dst != INVALID_HANDLE_VALUE, "failed to create dest file, err %lu\n", GetLastError()); + if (dst == INVALID_HANDLE_VALUE) + goto cleanup; + + ret = GetDiskFreeSpaceW( tmp_path, §ors_per_cluster, &bytes_per_sector, NULL, NULL ); + cluster_size = ret ? sectors_per_cluster * bytes_per_sector : 0; + if (!cluster_size) cluster_size = 4096; + + buffer = HeapAlloc( GetProcessHeap(), 0, cluster_size ); + ok(!!buffer, "failed to allocate %lu bytes\n", cluster_size); + if (!buffer) + goto cleanup; + + memset( buffer, 0x5a, cluster_size ); + ret = WriteFile( src, buffer, cluster_size, &written, NULL ); + ok(ret && written == cluster_size, "WriteFile failed, ret %d written %lu err %lu\n", ret, written, GetLastError()); + if (!ret) + goto cleanup; + + dup.FileHandle = src; + dup.SourceFileOffset.QuadPart = 0; + dup.TargetFileOffset.QuadPart = 0; + dup.ByteCount.QuadPart = cluster_size; + + ret = DeviceIoControl( dst, FSCTL_DUPLICATE_EXTENTS_TO_FILE, &dup, sizeof(dup), + NULL, 0, &bytes_returned, NULL ); + if (!ret) + { + skip("duplicate extents not supported on %s, err %lu\n", debugstr_w(tmp_path), GetLastError()); + goto cleanup; + } + + ret = GetVolumeInformationW( tmp_path, NULL, 0, NULL, NULL, &flags, NULL, 0 ); + ok(ret, "GetVolumeInformationW failed for %s, err %lu\n", debugstr_w(tmp_path), GetLastError()); + if (ret) + ok(flags & FILE_SUPPORTS_BLOCK_REFCOUNTING, + "expected FILE_SUPPORTS_BLOCK_REFCOUNTING for %s, got %#lx\n", debugstr_w(tmp_path), flags); + +cleanup: + if (src != INVALID_HANDLE_VALUE) CloseHandle( src ); + if (dst != INVALID_HANDLE_VALUE) CloseHandle( dst ); + DeleteFileW( src_path ); + DeleteFileW( dst_path ); + HeapFree( GetProcessHeap(), 0, buffer ); +} + static void test_mountmgr_query_points(void) { char input_buffer[64]; @@ -2217,6 +2302,7 @@ START_TEST(volume) test_cdrom_ioctl(); test_mounted_folder(); test_GetVolumeInformationByHandle(); + test_block_refcounting_flag(); test_mountmgr_query_points(); test_GetDiskSpaceInformationA(); } diff --git a/dlls/ntdll/unix/file.c b/dlls/ntdll/unix/file.c index b6641f31eb9..ad4dadc08fc 100644 --- a/dlls/ntdll/unix/file.c +++ b/dlls/ntdll/unix/file.c @@ -156,6 +156,10 @@ typedef struct /* Case-insensitivity attribute */ #define EXT4_CASEFOLD_FL 0x40000000 +#ifndef BTRFS_SUPER_MAGIC +#define BTRFS_SUPER_MAGIC 0x9123683e +#endif + #ifndef O_DIRECTORY # define O_DIRECTORY 0200000 /* must be directory */ #endif @@ -7465,8 +7469,10 @@ NTSTATUS WINAPI NtQueryVolumeInformationFile( HANDLE handle, IO_STATUS_BLOCK *io static const WCHAR udfW[] = {'U','D','F'}; FILE_FS_ATTRIBUTE_INFORMATION *info = buffer; + struct statfs stfs; struct mountmgr_unix_drive drive; enum mountmgr_fs_type fs_type = MOUNTMGR_FS_TYPE_NTFS; + BOOL supports_block_refcounting = FALSE; if (length < sizeof(FILE_FS_ATTRIBUTE_INFORMATION)) { @@ -7474,11 +7480,32 @@ NTSTATUS WINAPI NtQueryVolumeInformationFile( HANDLE handle, IO_STATUS_BLOCK *io break; } +#if defined(linux) && defined(HAVE_FSTATFS) + if (!fstatfs( fd, &stfs ) && stfs.f_type == BTRFS_SUPER_MAGIC) + supports_block_refcounting = TRUE; +#endif + +#ifdef FICLONERANGE + if (!supports_block_refcounting) + { + struct file_clone_range range = + { + .src_fd = fd, + .src_offset = 0, + .src_length = 0, + .dest_offset = 0, + }; + + if (!ioctl( fd, FICLONERANGE, &range )) + supports_block_refcounting = TRUE; + else if (errno != EOPNOTSUPP && errno != ENOTTY) + supports_block_refcounting = TRUE; + } +#endif + if (!get_mountmgr_fs_info( handle, fd, &drive, sizeof(drive) )) fs_type = drive.fs_type; else { - struct statfs stfs; - if (!fstatfs( fd, &stfs )) { #if defined(linux) && defined(HAVE_FSTATFS) @@ -7541,6 +7568,9 @@ NTSTATUS WINAPI NtQueryVolumeInformationFile( HANDLE handle, IO_STATUS_BLOCK *io break; } + if (supports_block_refcounting) + info->FileSystemAttributes |= FILE_SUPPORTS_BLOCK_REFCOUNTING; + io->Information = offsetof( FILE_FS_ATTRIBUTE_INFORMATION, FileSystemName ) + info->FileSystemNameLength; status = STATUS_SUCCESS; break; -- GitLab https://gitlab.winehq.org/wine/wine/-/merge_requests/9732