mirror of
https://github.com/mhx/dwarfs.git
synced 2025-09-10 13:04:15 -04:00
refactor: split xattr implementation to avoid #ifdef hell
This commit is contained in:
Marcus Holland-Moritz
parent
aa000939a6
commit
a4bd54c04f
@ -515,6 +515,12 @@ list(APPEND LIBDWARFS_SRC
|
|||||||
src/dwarfs/xattr.cpp
|
src/dwarfs/xattr.cpp
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if(WIN32)
|
||||||
|
list(APPEND LIBDWARFS_SRC src/dwarfs/xattr_win.cpp)
|
||||||
|
else()
|
||||||
|
list(APPEND LIBDWARFS_SRC src/dwarfs/xattr_posix.cpp)
|
||||||
|
endif()
|
||||||
|
|
||||||
if(WITH_MAN_OPTION)
|
if(WITH_MAN_OPTION)
|
||||||
include(${CMAKE_SOURCE_DIR}/cmake/render_manpage.cmake)
|
include(${CMAKE_SOURCE_DIR}/cmake/render_manpage.cmake)
|
||||||
|
|
||||||
|
|||||||
@ -19,369 +19,10 @@
|
|||||||
* along with dwarfs. If not, see <https://www.gnu.org/licenses/>.
|
* along with dwarfs. If not, see <https://www.gnu.org/licenses/>.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
|
|
||||||
#include <vector>
|
|
||||||
|
|
||||||
#include <ntstatus.h>
|
|
||||||
#include <tchar.h>
|
|
||||||
#include <windows.h>
|
|
||||||
#include <winternl.h>
|
|
||||||
|
|
||||||
#include <boost/algorithm/string.hpp>
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
#include <sys/xattr.h>
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#include <folly/ScopeGuard.h>
|
|
||||||
#include <folly/String.h>
|
|
||||||
|
|
||||||
#include <fmt/format.h>
|
|
||||||
|
|
||||||
#include "dwarfs/xattr.h"
|
#include "dwarfs/xattr.h"
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
|
|
||||||
extern "C" {
|
|
||||||
|
|
||||||
typedef struct _FILE_FULL_EA_INFORMATION {
|
|
||||||
ULONG NextEntryOffset;
|
|
||||||
UCHAR Flags;
|
|
||||||
UCHAR EaNameLength;
|
|
||||||
USHORT EaValueLength;
|
|
||||||
CHAR EaName[1];
|
|
||||||
} FILE_FULL_EA_INFORMATION, *PFILE_FULL_EA_INFORMATION;
|
|
||||||
|
|
||||||
typedef struct _FILE_GET_EA_INFORMATION {
|
|
||||||
ULONG NextEntryOffset;
|
|
||||||
UCHAR EaNameLength;
|
|
||||||
CHAR EaName[1];
|
|
||||||
} FILE_GET_EA_INFORMATION, *PFILE_GET_EA_INFORMATION;
|
|
||||||
|
|
||||||
NTSYSAPI NTSTATUS NTAPI RtlDosPathNameToNtPathName_U_WithStatus(
|
|
||||||
PCWSTR DosFileName, PUNICODE_STRING NtFileName, PWSTR* FilePart,
|
|
||||||
PVOID RelativeName);
|
|
||||||
|
|
||||||
VOID NTAPI RtlFreeUnicodeString(PUNICODE_STRING UnicodeString);
|
|
||||||
|
|
||||||
NTSYSAPI NTSTATUS NTAPI NtQueryEaFile(HANDLE FileHandle,
|
|
||||||
PIO_STATUS_BLOCK IoStatusBlock,
|
|
||||||
PVOID Buffer, ULONG Length,
|
|
||||||
BOOLEAN ReturnSingleEntry, PVOID EaList,
|
|
||||||
ULONG EaListLength, PULONG EaIndex,
|
|
||||||
BOOLEAN RestartScan);
|
|
||||||
|
|
||||||
NTSYSAPI NTSTATUS NTAPI NtSetEaFile(HANDLE FileHandle,
|
|
||||||
PIO_STATUS_BLOCK IoStatusBlock,
|
|
||||||
PVOID Buffer, ULONG Length);
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
namespace dwarfs {
|
namespace dwarfs {
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
|
|
||||||
constexpr size_t kMaxFullEaBufferSize{
|
|
||||||
offsetof(FILE_FULL_EA_INFORMATION, EaName) + 256 + 65536};
|
|
||||||
constexpr size_t kMaxGetEaBufferSize{offsetof(FILE_GET_EA_INFORMATION, EaName) +
|
|
||||||
256};
|
|
||||||
|
|
||||||
HANDLE open_file(std::filesystem::path const& path, bool writeable,
|
|
||||||
std::error_code& ec) {
|
|
||||||
UNICODE_STRING nt_path;
|
|
||||||
|
|
||||||
if (auto r = ::RtlDosPathNameToNtPathName_U_WithStatus(
|
|
||||||
path.wstring().c_str(), &nt_path, nullptr, nullptr);
|
|
||||||
r != 0) {
|
|
||||||
ec = std::error_code(r, std::system_category());
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
SCOPE_EXIT { ::RtlFreeUnicodeString(&nt_path); };
|
|
||||||
|
|
||||||
HANDLE fh;
|
|
||||||
IO_STATUS_BLOCK iosb;
|
|
||||||
OBJECT_ATTRIBUTES attr;
|
|
||||||
ACCESS_MASK desired_access = FILE_READ_EA;
|
|
||||||
|
|
||||||
if (writeable) {
|
|
||||||
desired_access |= FILE_WRITE_EA;
|
|
||||||
}
|
|
||||||
|
|
||||||
InitializeObjectAttributes(&attr, &nt_path, 0, nullptr, nullptr);
|
|
||||||
|
|
||||||
if (auto r = ::NtCreateFile(
|
|
||||||
&fh, desired_access, &attr, &iosb, nullptr, FILE_ATTRIBUTE_NORMAL,
|
|
||||||
FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_OPEN, 0,
|
|
||||||
nullptr, 0);
|
|
||||||
r != 0) {
|
|
||||||
ec = std::error_code(::RtlNtStatusToDosError(r), std::system_category());
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
return fh;
|
|
||||||
}
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
ssize_t portable_getxattr(const char* path, const char* name, void* value,
|
|
||||||
size_t size) {
|
|
||||||
#ifdef __APPLE__
|
|
||||||
return ::getxattr(path, name, value, size, 0, 0);
|
|
||||||
#else
|
|
||||||
return ::getxattr(path, name, value, size);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
int portable_setxattr(const char* path, const char* name, const void* value,
|
|
||||||
size_t size, int flags) {
|
|
||||||
#ifdef __APPLE__
|
|
||||||
return ::setxattr(path, name, value, size, 0, flags);
|
|
||||||
#else
|
|
||||||
return ::setxattr(path, name, value, size, flags);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
int portable_removexattr(const char* path, const char* name) {
|
|
||||||
#ifdef __APPLE__
|
|
||||||
return ::removexattr(path, name, 0);
|
|
||||||
#else
|
|
||||||
return ::removexattr(path, name);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
ssize_t portable_listxattr(const char* path, char* list, size_t size) {
|
|
||||||
#ifdef __APPLE__
|
|
||||||
return ::listxattr(path, list, size, 0);
|
|
||||||
#else
|
|
||||||
return ::listxattr(path, list, size);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
constexpr size_t kExtraSize{1024};
|
|
||||||
|
|
||||||
} // namespace
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
|
|
||||||
std::string getxattr(std::filesystem::path const& path, std::string const& name,
|
|
||||||
std::error_code& ec) {
|
|
||||||
ec.clear();
|
|
||||||
|
|
||||||
if (name.size() > std::numeric_limits<UCHAR>::max()) {
|
|
||||||
ec = std::error_code(ERROR_INVALID_EA_NAME, std::system_category());
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
auto fh = open_file(path, false, ec);
|
|
||||||
|
|
||||||
if (!fh) {
|
|
||||||
// error code already set
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
SCOPE_EXIT { ::NtClose(fh); };
|
|
||||||
|
|
||||||
CHAR getea_buf[kMaxGetEaBufferSize];
|
|
||||||
ULONG getea_len =
|
|
||||||
FIELD_OFFSET(FILE_GET_EA_INFORMATION, EaName) + name.size() + 1;
|
|
||||||
auto getea = reinterpret_cast<PFILE_GET_EA_INFORMATION>(getea_buf);
|
|
||||||
|
|
||||||
getea->NextEntryOffset = 0;
|
|
||||||
getea->EaNameLength = static_cast<UCHAR>(name.size());
|
|
||||||
std::memcpy(getea->EaName, name.data(), name.size());
|
|
||||||
getea->EaName[name.size()] = '\0';
|
|
||||||
|
|
||||||
std::vector<CHAR> ea_buf(kMaxFullEaBufferSize);
|
|
||||||
PFILE_FULL_EA_INFORMATION ea;
|
|
||||||
IO_STATUS_BLOCK iosb;
|
|
||||||
|
|
||||||
ea = reinterpret_cast<PFILE_FULL_EA_INFORMATION>(ea_buf.data());
|
|
||||||
|
|
||||||
auto res = ::NtQueryEaFile(fh, &iosb, ea, ea_buf.size(), FALSE, getea,
|
|
||||||
getea_len, nullptr, FALSE);
|
|
||||||
|
|
||||||
if (res != STATUS_SUCCESS) {
|
|
||||||
ec = std::error_code(::RtlNtStatusToDosError(res), std::system_category());
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
if (ea->EaValueLength == 0) {
|
|
||||||
ec = std::error_code(ENODATA, std::generic_category());
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
return {ea->EaName + ea->EaNameLength + 1, ea->EaValueLength};
|
|
||||||
}
|
|
||||||
|
|
||||||
void setxattr(std::filesystem::path const& path, std::string const& name,
|
|
||||||
std::string_view value, std::error_code& ec) {
|
|
||||||
// TODO
|
|
||||||
}
|
|
||||||
|
|
||||||
void removexattr(std::filesystem::path const& path, std::string const& name,
|
|
||||||
std::error_code& ec) {
|
|
||||||
// TODO
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<std::string>
|
|
||||||
listxattr(std::filesystem::path const& path, std::error_code& ec) {
|
|
||||||
ec.clear();
|
|
||||||
|
|
||||||
auto fh = open_file(path, false, ec);
|
|
||||||
|
|
||||||
if (!fh) {
|
|
||||||
// error code already set
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
SCOPE_EXIT { ::NtClose(fh); };
|
|
||||||
|
|
||||||
std::vector<std::string> names;
|
|
||||||
std::vector<CHAR> ea_buf(kMaxFullEaBufferSize);
|
|
||||||
BOOLEAN restart = TRUE;
|
|
||||||
|
|
||||||
for (;;) {
|
|
||||||
IO_STATUS_BLOCK iosb;
|
|
||||||
|
|
||||||
auto ea = reinterpret_cast<PFILE_FULL_EA_INFORMATION>(ea_buf.data());
|
|
||||||
auto res = ::NtQueryEaFile(fh, &iosb, ea, ea_buf.size(), FALSE, nullptr, 0,
|
|
||||||
nullptr, restart);
|
|
||||||
|
|
||||||
if (res != STATUS_SUCCESS && res != STATUS_BUFFER_OVERFLOW) {
|
|
||||||
ec =
|
|
||||||
std::error_code(::RtlNtStatusToDosError(res), std::system_category());
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
for (;;) {
|
|
||||||
std::string name(ea->EaName, ea->EaNameLength);
|
|
||||||
boost::algorithm::to_lower(name);
|
|
||||||
names.push_back(std::move(name));
|
|
||||||
|
|
||||||
if (ea->NextEntryOffset == 0) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
ea = reinterpret_cast<PFILE_FULL_EA_INFORMATION>(
|
|
||||||
reinterpret_cast<PCHAR>(ea) + ea->NextEntryOffset);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (res == STATUS_SUCCESS) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
restart = FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
return names;
|
|
||||||
}
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
std::string getxattr(std::filesystem::path const& path, std::string const& name,
|
|
||||||
std::error_code& ec) {
|
|
||||||
ec.clear();
|
|
||||||
|
|
||||||
auto cpath = path.c_str();
|
|
||||||
auto cname = name.c_str();
|
|
||||||
|
|
||||||
for (;;) {
|
|
||||||
ssize_t size = portable_getxattr(cpath, cname, nullptr, 0);
|
|
||||||
|
|
||||||
if (size < 0) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string value;
|
|
||||||
value.resize(size + kExtraSize);
|
|
||||||
|
|
||||||
size = portable_getxattr(cpath, cname, value.data(), value.size());
|
|
||||||
|
|
||||||
if (size >= 0) {
|
|
||||||
value.resize(size);
|
|
||||||
return value;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (errno != ERANGE) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
ec = std::error_code(errno, std::generic_category());
|
|
||||||
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
void setxattr(std::filesystem::path const& path, std::string const& name,
|
|
||||||
std::string_view value, std::error_code& ec) {
|
|
||||||
ec.clear();
|
|
||||||
|
|
||||||
if (portable_setxattr(path.c_str(), name.c_str(), value.data(), value.size(),
|
|
||||||
0) < 0) {
|
|
||||||
ec = std::error_code(errno, std::generic_category());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void removexattr(std::filesystem::path const& path, std::string const& name,
|
|
||||||
std::error_code& ec) {
|
|
||||||
ec.clear();
|
|
||||||
|
|
||||||
if (portable_removexattr(path.c_str(), name.c_str()) < 0) {
|
|
||||||
ec = std::error_code(errno, std::generic_category());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<std::string>
|
|
||||||
listxattr(std::filesystem::path const& path, std::error_code& ec) {
|
|
||||||
ec.clear();
|
|
||||||
|
|
||||||
auto cpath = path.c_str();
|
|
||||||
|
|
||||||
for (;;) {
|
|
||||||
ssize_t size = portable_listxattr(cpath, nullptr, 0);
|
|
||||||
|
|
||||||
if (size < 0) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string list;
|
|
||||||
list.resize(size + kExtraSize);
|
|
||||||
|
|
||||||
size = portable_listxattr(cpath, list.data(), list.size());
|
|
||||||
|
|
||||||
if (size >= 0) {
|
|
||||||
std::vector<std::string> names;
|
|
||||||
|
|
||||||
if (size > 0) {
|
|
||||||
// drop the last '\0'
|
|
||||||
list.resize(size - 1);
|
|
||||||
folly::split('\0', list, names);
|
|
||||||
}
|
|
||||||
|
|
||||||
return names;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (errno != ERANGE) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
ec = std::error_code(errno, std::generic_category());
|
|
||||||
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
std::string
|
std::string
|
||||||
getxattr(std::filesystem::path const& path, std::string const& name) {
|
getxattr(std::filesystem::path const& path, std::string const& name) {
|
||||||
std::error_code ec;
|
std::error_code ec;
|
||||||
|
|||||||
163
src/dwarfs/xattr_posix.cpp
Normal file
163
src/dwarfs/xattr_posix.cpp
Normal file
@ -0,0 +1,163 @@
|
|||||||
|
/* vim:set ts=2 sw=2 sts=2 et: */
|
||||||
|
/**
|
||||||
|
* \author Marcus Holland-Moritz (github@mhxnet.de)
|
||||||
|
* \copyright Copyright (c) Marcus Holland-Moritz
|
||||||
|
*
|
||||||
|
* This file is part of dwarfs.
|
||||||
|
*
|
||||||
|
* dwarfs is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* dwarfs is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with dwarfs. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <sys/xattr.h>
|
||||||
|
|
||||||
|
#include <folly/String.h>
|
||||||
|
|
||||||
|
#include "dwarfs/xattr.h"
|
||||||
|
|
||||||
|
namespace dwarfs {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
ssize_t portable_getxattr(const char* path, const char* name, void* value,
|
||||||
|
size_t size) {
|
||||||
|
#ifdef __APPLE__
|
||||||
|
return ::getxattr(path, name, value, size, 0, 0);
|
||||||
|
#else
|
||||||
|
return ::getxattr(path, name, value, size);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
int portable_setxattr(const char* path, const char* name, const void* value,
|
||||||
|
size_t size, int flags) {
|
||||||
|
#ifdef __APPLE__
|
||||||
|
return ::setxattr(path, name, value, size, 0, flags);
|
||||||
|
#else
|
||||||
|
return ::setxattr(path, name, value, size, flags);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
int portable_removexattr(const char* path, const char* name) {
|
||||||
|
#ifdef __APPLE__
|
||||||
|
return ::removexattr(path, name, 0);
|
||||||
|
#else
|
||||||
|
return ::removexattr(path, name);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
ssize_t portable_listxattr(const char* path, char* list, size_t size) {
|
||||||
|
#ifdef __APPLE__
|
||||||
|
return ::listxattr(path, list, size, 0);
|
||||||
|
#else
|
||||||
|
return ::listxattr(path, list, size);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
constexpr size_t kExtraSize{1024};
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
std::string getxattr(std::filesystem::path const& path, std::string const& name,
|
||||||
|
std::error_code& ec) {
|
||||||
|
ec.clear();
|
||||||
|
|
||||||
|
auto cpath = path.c_str();
|
||||||
|
auto cname = name.c_str();
|
||||||
|
|
||||||
|
for (;;) {
|
||||||
|
ssize_t size = portable_getxattr(cpath, cname, nullptr, 0);
|
||||||
|
|
||||||
|
if (size < 0) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string value;
|
||||||
|
value.resize(size + kExtraSize);
|
||||||
|
|
||||||
|
size = portable_getxattr(cpath, cname, value.data(), value.size());
|
||||||
|
|
||||||
|
if (size >= 0) {
|
||||||
|
value.resize(size);
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (errno != ERANGE) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ec = std::error_code(errno, std::generic_category());
|
||||||
|
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
void setxattr(std::filesystem::path const& path, std::string const& name,
|
||||||
|
std::string_view value, std::error_code& ec) {
|
||||||
|
ec.clear();
|
||||||
|
|
||||||
|
if (portable_setxattr(path.c_str(), name.c_str(), value.data(), value.size(),
|
||||||
|
0) < 0) {
|
||||||
|
ec = std::error_code(errno, std::generic_category());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void removexattr(std::filesystem::path const& path, std::string const& name,
|
||||||
|
std::error_code& ec) {
|
||||||
|
ec.clear();
|
||||||
|
|
||||||
|
if (portable_removexattr(path.c_str(), name.c_str()) < 0) {
|
||||||
|
ec = std::error_code(errno, std::generic_category());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<std::string>
|
||||||
|
listxattr(std::filesystem::path const& path, std::error_code& ec) {
|
||||||
|
ec.clear();
|
||||||
|
|
||||||
|
auto cpath = path.c_str();
|
||||||
|
|
||||||
|
for (;;) {
|
||||||
|
ssize_t size = portable_listxattr(cpath, nullptr, 0);
|
||||||
|
|
||||||
|
if (size < 0) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string list;
|
||||||
|
list.resize(size + kExtraSize);
|
||||||
|
|
||||||
|
size = portable_listxattr(cpath, list.data(), list.size());
|
||||||
|
|
||||||
|
if (size >= 0) {
|
||||||
|
std::vector<std::string> names;
|
||||||
|
|
||||||
|
if (size > 0) {
|
||||||
|
// drop the last '\0'
|
||||||
|
list.resize(size - 1);
|
||||||
|
folly::split('\0', list, names);
|
||||||
|
}
|
||||||
|
|
||||||
|
return names;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (errno != ERANGE) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ec = std::error_code(errno, std::generic_category());
|
||||||
|
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace dwarfs
|
||||||
229
src/dwarfs/xattr_win.cpp
Normal file
229
src/dwarfs/xattr_win.cpp
Normal file
@ -0,0 +1,229 @@
|
|||||||
|
/* vim:set ts=2 sw=2 sts=2 et: */
|
||||||
|
/**
|
||||||
|
* \author Marcus Holland-Moritz (github@mhxnet.de)
|
||||||
|
* \copyright Copyright (c) Marcus Holland-Moritz
|
||||||
|
*
|
||||||
|
* This file is part of dwarfs.
|
||||||
|
*
|
||||||
|
* dwarfs is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* dwarfs is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with dwarfs. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include <ntstatus.h>
|
||||||
|
#include <tchar.h>
|
||||||
|
#include <windows.h>
|
||||||
|
#include <winternl.h>
|
||||||
|
|
||||||
|
#include <boost/algorithm/string.hpp>
|
||||||
|
|
||||||
|
#include <folly/ScopeGuard.h>
|
||||||
|
|
||||||
|
#include "dwarfs/xattr.h"
|
||||||
|
|
||||||
|
extern "C" {
|
||||||
|
|
||||||
|
typedef struct _FILE_FULL_EA_INFORMATION {
|
||||||
|
ULONG NextEntryOffset;
|
||||||
|
UCHAR Flags;
|
||||||
|
UCHAR EaNameLength;
|
||||||
|
USHORT EaValueLength;
|
||||||
|
CHAR EaName[1];
|
||||||
|
} FILE_FULL_EA_INFORMATION, *PFILE_FULL_EA_INFORMATION;
|
||||||
|
|
||||||
|
typedef struct _FILE_GET_EA_INFORMATION {
|
||||||
|
ULONG NextEntryOffset;
|
||||||
|
UCHAR EaNameLength;
|
||||||
|
CHAR EaName[1];
|
||||||
|
} FILE_GET_EA_INFORMATION, *PFILE_GET_EA_INFORMATION;
|
||||||
|
|
||||||
|
NTSYSAPI NTSTATUS NTAPI RtlDosPathNameToNtPathName_U_WithStatus(
|
||||||
|
PCWSTR DosFileName, PUNICODE_STRING NtFileName, PWSTR* FilePart,
|
||||||
|
PVOID RelativeName);
|
||||||
|
|
||||||
|
VOID NTAPI RtlFreeUnicodeString(PUNICODE_STRING UnicodeString);
|
||||||
|
|
||||||
|
NTSYSAPI NTSTATUS NTAPI NtQueryEaFile(HANDLE FileHandle,
|
||||||
|
PIO_STATUS_BLOCK IoStatusBlock,
|
||||||
|
PVOID Buffer, ULONG Length,
|
||||||
|
BOOLEAN ReturnSingleEntry, PVOID EaList,
|
||||||
|
ULONG EaListLength, PULONG EaIndex,
|
||||||
|
BOOLEAN RestartScan);
|
||||||
|
|
||||||
|
NTSYSAPI NTSTATUS NTAPI NtSetEaFile(HANDLE FileHandle,
|
||||||
|
PIO_STATUS_BLOCK IoStatusBlock,
|
||||||
|
PVOID Buffer, ULONG Length);
|
||||||
|
}
|
||||||
|
|
||||||
|
namespace dwarfs {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
constexpr size_t kMaxFullEaBufferSize{
|
||||||
|
offsetof(FILE_FULL_EA_INFORMATION, EaName) + 256 + 65536};
|
||||||
|
constexpr size_t kMaxGetEaBufferSize{offsetof(FILE_GET_EA_INFORMATION, EaName) +
|
||||||
|
256};
|
||||||
|
|
||||||
|
HANDLE open_file(std::filesystem::path const& path, bool writeable,
|
||||||
|
std::error_code& ec) {
|
||||||
|
UNICODE_STRING nt_path;
|
||||||
|
|
||||||
|
if (auto r = ::RtlDosPathNameToNtPathName_U_WithStatus(
|
||||||
|
path.wstring().c_str(), &nt_path, nullptr, nullptr);
|
||||||
|
r != 0) {
|
||||||
|
ec = std::error_code(r, std::system_category());
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
SCOPE_EXIT { ::RtlFreeUnicodeString(&nt_path); };
|
||||||
|
|
||||||
|
HANDLE fh;
|
||||||
|
IO_STATUS_BLOCK iosb;
|
||||||
|
OBJECT_ATTRIBUTES attr;
|
||||||
|
ACCESS_MASK desired_access = FILE_READ_EA;
|
||||||
|
|
||||||
|
if (writeable) {
|
||||||
|
desired_access |= FILE_WRITE_EA;
|
||||||
|
}
|
||||||
|
|
||||||
|
InitializeObjectAttributes(&attr, &nt_path, 0, nullptr, nullptr);
|
||||||
|
|
||||||
|
if (auto r = ::NtCreateFile(
|
||||||
|
&fh, desired_access, &attr, &iosb, nullptr, FILE_ATTRIBUTE_NORMAL,
|
||||||
|
FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_OPEN, 0,
|
||||||
|
nullptr, 0);
|
||||||
|
r != 0) {
|
||||||
|
ec = std::error_code(::RtlNtStatusToDosError(r), std::system_category());
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
return fh;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
std::string getxattr(std::filesystem::path const& path, std::string const& name,
|
||||||
|
std::error_code& ec) {
|
||||||
|
ec.clear();
|
||||||
|
|
||||||
|
if (name.size() > std::numeric_limits<UCHAR>::max()) {
|
||||||
|
ec = std::error_code(ERROR_INVALID_EA_NAME, std::system_category());
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
auto fh = open_file(path, false, ec);
|
||||||
|
|
||||||
|
if (!fh) {
|
||||||
|
// error code already set
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
SCOPE_EXIT { ::NtClose(fh); };
|
||||||
|
|
||||||
|
CHAR getea_buf[kMaxGetEaBufferSize];
|
||||||
|
ULONG getea_len =
|
||||||
|
FIELD_OFFSET(FILE_GET_EA_INFORMATION, EaName) + name.size() + 1;
|
||||||
|
auto getea = reinterpret_cast<PFILE_GET_EA_INFORMATION>(getea_buf);
|
||||||
|
|
||||||
|
getea->NextEntryOffset = 0;
|
||||||
|
getea->EaNameLength = static_cast<UCHAR>(name.size());
|
||||||
|
std::memcpy(getea->EaName, name.data(), name.size());
|
||||||
|
getea->EaName[name.size()] = '\0';
|
||||||
|
|
||||||
|
std::vector<CHAR> ea_buf(kMaxFullEaBufferSize);
|
||||||
|
PFILE_FULL_EA_INFORMATION ea;
|
||||||
|
IO_STATUS_BLOCK iosb;
|
||||||
|
|
||||||
|
ea = reinterpret_cast<PFILE_FULL_EA_INFORMATION>(ea_buf.data());
|
||||||
|
|
||||||
|
auto res = ::NtQueryEaFile(fh, &iosb, ea, ea_buf.size(), FALSE, getea,
|
||||||
|
getea_len, nullptr, FALSE);
|
||||||
|
|
||||||
|
if (res != STATUS_SUCCESS) {
|
||||||
|
ec = std::error_code(::RtlNtStatusToDosError(res), std::system_category());
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
if (ea->EaValueLength == 0) {
|
||||||
|
ec = std::error_code(ENODATA, std::generic_category());
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
return {ea->EaName + ea->EaNameLength + 1, ea->EaValueLength};
|
||||||
|
}
|
||||||
|
|
||||||
|
void setxattr(std::filesystem::path const& path, std::string const& name,
|
||||||
|
std::string_view value, std::error_code& ec) {
|
||||||
|
// TODO
|
||||||
|
}
|
||||||
|
|
||||||
|
void removexattr(std::filesystem::path const& path, std::string const& name,
|
||||||
|
std::error_code& ec) {
|
||||||
|
// TODO
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<std::string>
|
||||||
|
listxattr(std::filesystem::path const& path, std::error_code& ec) {
|
||||||
|
ec.clear();
|
||||||
|
|
||||||
|
auto fh = open_file(path, false, ec);
|
||||||
|
|
||||||
|
if (!fh) {
|
||||||
|
// error code already set
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
SCOPE_EXIT { ::NtClose(fh); };
|
||||||
|
|
||||||
|
std::vector<std::string> names;
|
||||||
|
std::vector<CHAR> ea_buf(kMaxFullEaBufferSize);
|
||||||
|
BOOLEAN restart = TRUE;
|
||||||
|
|
||||||
|
for (;;) {
|
||||||
|
IO_STATUS_BLOCK iosb;
|
||||||
|
|
||||||
|
auto ea = reinterpret_cast<PFILE_FULL_EA_INFORMATION>(ea_buf.data());
|
||||||
|
auto res = ::NtQueryEaFile(fh, &iosb, ea, ea_buf.size(), FALSE, nullptr, 0,
|
||||||
|
nullptr, restart);
|
||||||
|
|
||||||
|
if (res != STATUS_SUCCESS && res != STATUS_BUFFER_OVERFLOW) {
|
||||||
|
ec =
|
||||||
|
std::error_code(::RtlNtStatusToDosError(res), std::system_category());
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
for (;;) {
|
||||||
|
std::string name(ea->EaName, ea->EaNameLength);
|
||||||
|
boost::algorithm::to_lower(name);
|
||||||
|
names.push_back(std::move(name));
|
||||||
|
|
||||||
|
if (ea->NextEntryOffset == 0) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
ea = reinterpret_cast<PFILE_FULL_EA_INFORMATION>(
|
||||||
|
reinterpret_cast<PCHAR>(ea) + ea->NextEntryOffset);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (res == STATUS_SUCCESS) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
restart = FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
return names;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace dwarfs
|
||||||
Loading…
x
Reference in New Issue
Block a user