diff options
author | Paul Cristian Sarbu <paul.cristian.sarbu@huawei.com> | 2023-07-06 14:29:27 +0200 |
---|---|---|
committer | Paul Cristian Sarbu <paul.cristian.sarbu@huawei.com> | 2023-07-11 14:07:11 +0200 |
commit | 85938804a4a043c75cc4ca51a64ee789e55ba7cb (patch) | |
tree | ee04af0ecd39e8c9697d2405edc0c318ff906fed /src | |
parent | bec3322cfff305f13735c5f72febfd752b49fa10 (diff) | |
download | justbuild-85938804a4a043c75cc4ca51a64ee789e55ba7cb.tar.gz |
filesystem: Avoid unwanted indirections...
...that std::filesystem::* calls produce. This is because existence
and type checks use almost exclusively std::filesystem::status,
which follows symbolic links, when being called with path arguments.
Instead, one should instead use these methods with the value
returned by a call of std::filesystem::symlink_status.
This commit also streamlines the FileSystemManager tests, as well
as replace bare calls to std::filesystem with their FileSystemManager
counterparts (where suitable).
Diffstat (limited to 'src')
-rw-r--r-- | src/buildtool/file_system/file_system_manager.hpp | 36 | ||||
-rw-r--r-- | src/buildtool/main/main.cpp | 2 |
2 files changed, 27 insertions, 11 deletions
diff --git a/src/buildtool/file_system/file_system_manager.hpp b/src/buildtool/file_system/file_system_manager.hpp index 868acbcc..04f84cf4 100644 --- a/src/buildtool/file_system/file_system_manager.hpp +++ b/src/buildtool/file_system/file_system_manager.hpp @@ -437,10 +437,11 @@ class FileSystemManager { [[nodiscard]] static auto RemoveFile( std::filesystem::path const& file) noexcept -> bool { try { - if (!std::filesystem::exists(file)) { + auto status = std::filesystem::symlink_status(file); + if (!std::filesystem::exists(status)) { return true; } - if (!IsFile(file)) { + if (!std::filesystem::is_regular_file(status)) { return false; } return std::filesystem::remove(file); @@ -457,9 +458,13 @@ class FileSystemManager { bool recursively = false) noexcept -> bool { try { - if (!std::filesystem::exists(dir)) { + auto status = std::filesystem::symlink_status(dir); + if (!std::filesystem::exists(status)) { return true; } + if (!std::filesystem::is_directory(status)) { + return false; + } if (recursively) { return (std::filesystem::remove_all(dir) != static_cast<uintmax_t>(-1)); @@ -476,14 +481,16 @@ class FileSystemManager { /// \brief Returns if symlink is non-upwards, i.e., its string content path /// never passes itself in the directory tree. + /// \param non_strict if set, do not check non-upwardness. Use with care! [[nodiscard]] static auto IsNonUpwardsSymlink( - std::filesystem::path const& link) noexcept -> bool { + std::filesystem::path const& link, + bool non_strict = false) noexcept -> bool { try { if (not std::filesystem::is_symlink(link)) { return false; } - auto dest = std::filesystem::read_symlink(link); - return PathIsNonUpwards(dest); + return non_strict or + PathIsNonUpwards(std::filesystem::read_symlink(link)); } catch (std::exception const& e) { Logger::Log(LogLevel::Error, e.what()); return false; @@ -929,7 +936,8 @@ class FileSystemManager { [[nodiscard]] static auto CreateDirectoryImpl( std::filesystem::path const& dir) noexcept -> CreationStatus { try { - if (std::filesystem::is_directory(dir)) { + if (std::filesystem::is_directory( + std::filesystem::symlink_status(dir))) { return CreationStatus::Exists; } if (std::filesystem::create_directories(dir)) { @@ -939,7 +947,8 @@ class FileSystemManager { // after the current thread checked if it existed. For that reason, // we try to create it and check if it exists if create_directories // was not successful. - if (std::filesystem::is_directory(dir)) { + if (std::filesystem::is_directory( + std::filesystem::symlink_status(dir))) { return CreationStatus::Exists; } @@ -955,7 +964,8 @@ class FileSystemManager { [[nodiscard]] static auto CreateFileImpl( std::filesystem::path const& file) noexcept -> CreationStatus { try { - if (std::filesystem::is_regular_file(file)) { + if (std::filesystem::is_regular_file( + std::filesystem::symlink_status(file))) { return CreationStatus::Exists; } if (gsl::owner<FILE*> fp = std::fopen(file.c_str(), "wx")) { @@ -966,7 +976,8 @@ class FileSystemManager { // the current thread checked if it existed. For that reason, we try // to create it and check if it exists if fopen() with exclusive bit // was not successful. - if (std::filesystem::is_regular_file(file)) { + if (std::filesystem::is_regular_file( + std::filesystem::symlink_status(file))) { return CreationStatus::Exists; } return CreationStatus::Failed; @@ -983,6 +994,11 @@ class FileSystemManager { std::filesystem::copy_options::overwrite_existing) noexcept -> bool { try { + // src and dst should be actual files + if (std::filesystem::is_symlink(src) or + std::filesystem::is_symlink(dst)) { + return false; + } return std::filesystem::copy_file(src, dst, opt); } catch (std::exception const& e) { Logger::Log(LogLevel::Error, diff --git a/src/buildtool/main/main.cpp b/src/buildtool/main/main.cpp index 846fe257..8d2d4d2e 100644 --- a/src/buildtool/main/main.cpp +++ b/src/buildtool/main/main.cpp @@ -479,7 +479,7 @@ void SetupHashFunction() { -> Configuration { Configuration config{}; if (not clargs.config_file.empty()) { - if (not std::filesystem::exists(clargs.config_file)) { + if (not FileSystemManager::Exists(clargs.config_file)) { Logger::Log(LogLevel::Error, "Config file {} does not exist.", clargs.config_file.string()); |