| // Copyright 2016 The Bazel Authors. All rights reserved. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| #include <ctype.h> // isalpha |
| #include <wchar.h> // wcslen |
| #include <wctype.h> // iswalpha |
| #include <windows.h> |
| |
| #include <memory> // unique_ptr |
| #include <sstream> |
| #include <vector> |
| |
| #include "src/main/cpp/util/errors.h" |
| #include "src/main/cpp/util/exit_code.h" |
| #include "src/main/cpp/util/file.h" |
| #include "src/main/cpp/util/logging.h" |
| #include "src/main/cpp/util/path.h" |
| #include "src/main/cpp/util/path_platform.h" |
| #include "src/main/cpp/util/strings.h" |
| #include "src/main/native/windows/file.h" |
| #include "src/main/native/windows/util.h" |
| |
| namespace blaze_util { |
| |
| using bazel::windows::AutoHandle; |
| using bazel::windows::GetLongPath; |
| using bazel::windows::HasUncPrefix; |
| using bazel::windows::OpenDirectory; |
| using std::basic_string; |
| using std::pair; |
| using std::string; |
| using std::unique_ptr; |
| using std::vector; |
| using std::wstring; |
| |
| // Returns true if `path` refers to a directory or (non-dangling) junction. |
| // `path` must be a normalized Windows path, with UNC prefix (and absolute) if |
| // necessary. |
| static bool IsDirectoryW(const wstring& path); |
| |
| // Returns true the file or junction at `path` is successfully deleted. |
| // Returns false otherwise, or if `path` doesn't exist or is a directory. |
| // `path` must be a normalized Windows path, with UNC prefix (and absolute) if |
| // necessary. |
| static bool UnlinkPathW(const wstring& path); |
| |
| static bool CanReadFileW(const wstring& path); |
| |
| template <typename char_type> |
| static bool IsPathSeparator(char_type ch) { |
| return ch == '/' || ch == '\\'; |
| } |
| |
| class WindowsPipe : public IPipe { |
| public: |
| WindowsPipe(const HANDLE& read_handle, const HANDLE& write_handle) |
| : _read_handle(read_handle), _write_handle(write_handle) {} |
| |
| WindowsPipe() = delete; |
| |
| bool Send(const void* buffer, int size) override { |
| DWORD actually_written = 0; |
| return ::WriteFile(_write_handle, buffer, size, &actually_written, |
| NULL) == TRUE; |
| } |
| |
| int Receive(void* buffer, int size, int* error) override { |
| DWORD actually_read = 0; |
| BOOL result = ::ReadFile(_read_handle, buffer, size, &actually_read, NULL); |
| if (error != nullptr) { |
| // TODO(laszlocsomor): handle the error mode that is errno=EINTR on Linux. |
| *error = result ? IPipe::SUCCESS : IPipe::OTHER_ERROR; |
| } |
| return result ? actually_read : -1; |
| } |
| |
| private: |
| AutoHandle _read_handle; |
| AutoHandle _write_handle; |
| }; |
| |
| IPipe* CreatePipe() { |
| // The pipe HANDLEs can be inherited. |
| SECURITY_ATTRIBUTES sa = {sizeof(SECURITY_ATTRIBUTES), NULL, TRUE}; |
| HANDLE read_handle = INVALID_HANDLE_VALUE; |
| HANDLE write_handle = INVALID_HANDLE_VALUE; |
| if (!CreatePipe(&read_handle, &write_handle, &sa, 0)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "CreatePipe failed: " << GetLastErrorString(); |
| } |
| return new WindowsPipe(read_handle, write_handle); |
| } |
| |
| class WindowsFileMtime : public IFileMtime { |
| public: |
| WindowsFileMtime() |
| : near_future_(GetFuture(9)), distant_future_(GetFuture(10)) {} |
| |
| bool IsUntampered(const string& path) override; |
| bool SetToNow(const string& path) override; |
| bool SetToDistantFuture(const string& path) override; |
| |
| private: |
| // 9 years in the future. |
| const FILETIME near_future_; |
| // 10 years in the future. |
| const FILETIME distant_future_; |
| |
| static FILETIME GetNow(); |
| static FILETIME GetFuture(WORD years); |
| static bool Set(const string& path, FILETIME time); |
| }; |
| |
| bool WindowsFileMtime::IsUntampered(const string& path) { |
| if (path.empty() || IsDevNull(path.c_str())) { |
| return false; |
| } |
| |
| wstring wpath; |
| string error; |
| if (!AsAbsoluteWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "WindowsFileMtime::IsUntampered(" << path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| } |
| |
| // Get attributes, to check if the file exists. (It may still be a dangling |
| // junction.) |
| DWORD attrs = GetFileAttributesW(wpath.c_str()); |
| if (attrs == INVALID_FILE_ATTRIBUTES) { |
| return false; |
| } |
| |
| bool is_directory = attrs & FILE_ATTRIBUTE_DIRECTORY; |
| AutoHandle handle(CreateFileW( |
| /* lpFileName */ wpath.c_str(), |
| /* dwDesiredAccess */ GENERIC_READ, |
| /* dwShareMode */ FILE_SHARE_READ, |
| /* lpSecurityAttributes */ NULL, |
| /* dwCreationDisposition */ OPEN_EXISTING, |
| /* dwFlagsAndAttributes */ |
| // Per CreateFile's documentation on MSDN, opening directories requires |
| // the FILE_FLAG_BACKUP_SEMANTICS flag. |
| is_directory ? FILE_FLAG_BACKUP_SEMANTICS : FILE_ATTRIBUTE_NORMAL, |
| /* hTemplateFile */ NULL)); |
| |
| if (!handle.IsValid()) { |
| return false; |
| } |
| |
| if (is_directory) { |
| return true; |
| } else { |
| BY_HANDLE_FILE_INFORMATION info; |
| if (!GetFileInformationByHandle(handle, &info)) { |
| return false; |
| } |
| |
| // Compare the mtime with `near_future_`, not with `GetNow()` or |
| // `distant_future_`. |
| // This way we don't need to call GetNow() every time we want to compare |
| // (and thus convert a SYSTEMTIME to FILETIME), and we also don't need to |
| // worry about potentially unreliable FILETIME equality check (in case it |
| // uses floats or something crazy). |
| return CompareFileTime(&near_future_, &info.ftLastWriteTime) == -1; |
| } |
| } |
| |
| bool WindowsFileMtime::SetToNow(const string& path) { |
| return Set(path, GetNow()); |
| } |
| |
| bool WindowsFileMtime::SetToDistantFuture(const string& path) { |
| return Set(path, distant_future_); |
| } |
| |
| bool WindowsFileMtime::Set(const string& path, FILETIME time) { |
| if (path.empty()) { |
| return false; |
| } |
| wstring wpath; |
| string error; |
| if (!AsAbsoluteWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "WindowsFileMtime::Set(" << path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| |
| AutoHandle handle(::CreateFileW( |
| /* lpFileName */ wpath.c_str(), |
| /* dwDesiredAccess */ FILE_WRITE_ATTRIBUTES, |
| /* dwShareMode */ FILE_SHARE_READ, |
| /* lpSecurityAttributes */ NULL, |
| /* dwCreationDisposition */ OPEN_EXISTING, |
| /* dwFlagsAndAttributes */ |
| IsDirectoryW(wpath) |
| ? (FILE_FLAG_OPEN_REPARSE_POINT | FILE_FLAG_BACKUP_SEMANTICS) |
| : FILE_ATTRIBUTE_NORMAL, |
| /* hTemplateFile */ NULL)); |
| if (!handle.IsValid()) { |
| return false; |
| } |
| return ::SetFileTime( |
| /* hFile */ handle, |
| /* lpCreationTime */ NULL, |
| /* lpLastAccessTime */ NULL, |
| /* lpLastWriteTime */ &time) == TRUE; |
| } |
| |
| FILETIME WindowsFileMtime::GetNow() { |
| FILETIME now; |
| GetSystemTimeAsFileTime(&now); |
| return now; |
| } |
| |
| FILETIME WindowsFileMtime::GetFuture(WORD years) { |
| FILETIME result; |
| GetSystemTimeAsFileTime(&result); |
| |
| // 1 year in FILETIME. |
| constexpr ULONGLONG kOneYear = 365ULL * 24 * 60 * 60 * 10'000'000; |
| |
| ULARGE_INTEGER result_value; |
| result_value.LowPart = result.dwLowDateTime; |
| result_value.HighPart = result.dwHighDateTime; |
| result_value.QuadPart += kOneYear * years; |
| result.dwLowDateTime = result_value.LowPart; |
| result.dwHighDateTime = result_value.HighPart; |
| return result; |
| } |
| |
| IFileMtime* CreateFileMtime() { return new WindowsFileMtime(); } |
| |
| static bool OpenFileForReading(const string& filename, HANDLE* result) { |
| if (filename.empty()) { |
| return false; |
| } |
| // TODO(laszlocsomor): remove the following check; it won't allow opening NUL. |
| if (IsDevNull(filename.c_str())) { |
| return true; |
| } |
| wstring wfilename; |
| string error; |
| if (!AsAbsoluteWindowsPath(filename, &wfilename, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "OpenFileForReading(" << filename |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| } |
| *result = ::CreateFileW( |
| /* lpFileName */ wfilename.c_str(), |
| /* dwDesiredAccess */ GENERIC_READ, |
| /* dwShareMode */ FILE_SHARE_READ, |
| /* lpSecurityAttributes */ NULL, |
| /* dwCreationDisposition */ OPEN_EXISTING, |
| /* dwFlagsAndAttributes */ FILE_ATTRIBUTE_NORMAL, |
| /* hTemplateFile */ NULL); |
| return true; |
| } |
| |
| int ReadFromHandle(file_handle_type handle, void* data, size_t size, |
| int* error) { |
| DWORD actually_read = 0; |
| bool success = ::ReadFile(handle, data, size, &actually_read, NULL); |
| if (error != nullptr) { |
| // TODO(laszlocsomor): handle the error cases that are errno=EINTR and |
| // errno=EAGAIN on Linux. |
| *error = success ? ReadFileResult::SUCCESS : ReadFileResult::OTHER_ERROR; |
| } |
| return success ? actually_read : -1; |
| } |
| |
| bool ReadFile(const string& filename, string* content, int max_size) { |
| if (IsDevNull(filename.c_str())) { |
| // mimic read(2) behavior: we can always read 0 bytes from /dev/null |
| content->clear(); |
| return true; |
| } |
| HANDLE handle; |
| if (!OpenFileForReading(filename, &handle)) { |
| return false; |
| } |
| |
| AutoHandle autohandle(handle); |
| if (!autohandle.IsValid()) { |
| return false; |
| } |
| content->clear(); |
| return ReadFrom(handle, content, max_size); |
| } |
| |
| bool ReadFile(const string& filename, void* data, size_t size) { |
| if (IsDevNull(filename.c_str())) { |
| // mimic read(2) behavior: we can always read 0 bytes from /dev/null |
| return true; |
| } |
| HANDLE handle; |
| if (!OpenFileForReading(filename, &handle)) { |
| return false; |
| } |
| |
| AutoHandle autohandle(handle); |
| if (!autohandle.IsValid()) { |
| return false; |
| } |
| return ReadFrom(handle, data, size); |
| } |
| |
| bool WriteFile(const void* data, size_t size, const string& filename, |
| unsigned int perm) { |
| if (IsDevNull(filename.c_str())) { |
| return true; // mimic write(2) behavior with /dev/null |
| } |
| wstring wpath; |
| string error; |
| if (!AsAbsoluteWindowsPath(filename, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "WriteFile(" << filename |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| |
| UnlinkPathW(wpath); // We don't care about the success of this. |
| AutoHandle handle(::CreateFileW( |
| /* lpFileName */ wpath.c_str(), |
| /* dwDesiredAccess */ GENERIC_WRITE, |
| /* dwShareMode */ FILE_SHARE_READ, |
| /* lpSecurityAttributes */ NULL, |
| /* dwCreationDisposition */ CREATE_ALWAYS, |
| /* dwFlagsAndAttributes */ FILE_ATTRIBUTE_NORMAL, |
| /* hTemplateFile */ NULL)); |
| if (!handle.IsValid()) { |
| return false; |
| } |
| |
| // TODO(laszlocsomor): respect `perm` and set the file permissions accordingly |
| DWORD actually_written = 0; |
| ::WriteFile(handle, data, size, &actually_written, NULL); |
| return actually_written == size; |
| } |
| |
| int WriteToStdOutErr(const void* data, size_t size, bool to_stdout) { |
| DWORD written = 0; |
| HANDLE h = ::GetStdHandle(to_stdout ? STD_OUTPUT_HANDLE : STD_ERROR_HANDLE); |
| if (h == INVALID_HANDLE_VALUE) { |
| return WriteResult::OTHER_ERROR; |
| } |
| |
| if (::WriteFile(h, data, size, &written, NULL)) { |
| return (written == size) ? WriteResult::SUCCESS : WriteResult::OTHER_ERROR; |
| } else { |
| return (GetLastError() == ERROR_NO_DATA) ? WriteResult::BROKEN_PIPE |
| : WriteResult::OTHER_ERROR; |
| } |
| } |
| |
| int RenameDirectory(const std::string& old_name, const std::string& new_name) { |
| wstring wold_name; |
| string error; |
| if (!AsAbsoluteWindowsPath(old_name, &wold_name, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "RenameDirectory(" << old_name << ", " << new_name |
| << "): AsAbsoluteWindowsPath(" << old_name << ") failed: " << error; |
| return kRenameDirectoryFailureOtherError; |
| } |
| |
| wstring wnew_name; |
| if (!AsAbsoluteWindowsPath(new_name, &wnew_name, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "RenameDirectory(" << old_name << ", " << new_name |
| << "): AsAbsoluteWindowsPath(" << new_name << ") failed: " << error; |
| return kRenameDirectoryFailureOtherError; |
| } |
| |
| if (!::MoveFileExW(wold_name.c_str(), wnew_name.c_str(), |
| MOVEFILE_COPY_ALLOWED | MOVEFILE_FAIL_IF_NOT_TRACKABLE | |
| MOVEFILE_WRITE_THROUGH)) { |
| return GetLastError() == ERROR_ALREADY_EXISTS |
| ? kRenameDirectoryFailureNotEmpty |
| : kRenameDirectoryFailureOtherError; |
| } |
| return kRenameDirectorySuccess; |
| } |
| |
| static bool UnlinkPathW(const wstring& path) { |
| DWORD attrs = ::GetFileAttributesW(path.c_str()); |
| if (attrs == INVALID_FILE_ATTRIBUTES) { |
| // Path does not exist. |
| return false; |
| } |
| if (attrs & FILE_ATTRIBUTE_DIRECTORY) { |
| if (!(attrs & FILE_ATTRIBUTE_REPARSE_POINT)) { |
| // Path is a directory; unlink(2) also cannot remove directories. |
| return false; |
| } |
| // Otherwise it's a junction, remove using RemoveDirectoryW. |
| return ::RemoveDirectoryW(path.c_str()) == TRUE; |
| } else { |
| // Otherwise it's a file, remove using DeleteFileW. |
| return ::DeleteFileW(path.c_str()) == TRUE; |
| } |
| } |
| |
| bool UnlinkPath(const string& file_path) { |
| if (IsDevNull(file_path.c_str())) { |
| return false; |
| } |
| |
| wstring wpath; |
| string error; |
| if (!AsAbsoluteWindowsPath(file_path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "UnlinkPath(" << file_path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| return UnlinkPathW(wpath); |
| } |
| |
| class JunctionResolver { |
| public: |
| JunctionResolver(); |
| |
| // Resolves junctions, or simply checks file existence (if not a junction). |
| // |
| // Returns true if `path` is not a junction and it exists. |
| // Returns true if `path` is a junction and can be successfully resolved and |
| // its target exists. |
| // Returns false otherwise. |
| // |
| // If `result` is not nullptr and the method returned false, then this will be |
| // reset to point to a new WCHAR buffer containing the final resolved path. |
| // If `path` was a junction, this will be the fully resolved path, otherwise |
| // it will be a copy of `path`. |
| bool Resolve(const WCHAR* path, std::unique_ptr<WCHAR[]>* result); |
| |
| private: |
| static const int kMaximumJunctionDepth; |
| |
| // This struct is a simplified version of REPARSE_DATA_BUFFER, defined by |
| // the <Ntifs.h> header file, which is not available on some systems. |
| // This struct removes the original one's union keeping only |
| // MountPointReparseBuffer, while also renames some fields to reflect how |
| // ::DeviceIoControl actually uses them when reading junction data. |
| typedef struct _ReparseMountPointData { |
| static const int kSize = MAXIMUM_REPARSE_DATA_BUFFER_SIZE; |
| |
| ULONG ReparseTag; |
| USHORT Dummy1; |
| USHORT Dummy2; |
| USHORT Dummy3; |
| USHORT Dummy4; |
| // Length of string in PathBuffer, in WCHARs, including the "\??\" prefix |
| // and the null-terminator. |
| // |
| // Reparse points use the "\??\" prefix instead of "\\?\", presumably |
| // because the junction is resolved by the kernel and it points to a Device |
| // Object path (which is what the kernel understands), and "\??" is a device |
| // path. ("\??" is shorthand for "\DosDevices" under which disk drives |
| // reside, e.g. "C:" is a symlink to "\DosDevices\C:" aka "\??\C:"). |
| // See (on 2017-01-04): |
| // https://msdn.microsoft.com/en-us/library/windows/hardware/ff565384(v=vs.85).aspx |
| // https://msdn.microsoft.com/en-us/library/windows/hardware/ff557762(v=vs.85).aspx |
| USHORT Size; |
| USHORT Dummy5; |
| // First character of the string returned by ::DeviceIoControl. The rest of |
| // the string follows this in memory, that's why the caller must allocate |
| // kSize bytes and cast that data to ReparseMountPointData. |
| WCHAR PathBuffer[1]; |
| } ReparseMountPointData; |
| |
| uint8_t reparse_buffer_bytes_[ReparseMountPointData::kSize]; |
| ReparseMountPointData* reparse_buffer_; |
| |
| bool Resolve(const WCHAR* path, std::unique_ptr<WCHAR[]>* result, |
| int max_junction_depth); |
| }; |
| |
| // Maximum reparse point depth on Windows 8 and above is 63. |
| // Source (on 2016-12-20): |
| // https://msdn.microsoft.com/en-us/library/windows/desktop/aa365503(v=vs.85).aspx |
| const int JunctionResolver::kMaximumJunctionDepth = 63; |
| |
| JunctionResolver::JunctionResolver() |
| : reparse_buffer_( |
| reinterpret_cast<ReparseMountPointData*>(reparse_buffer_bytes_)) { |
| reparse_buffer_->ReparseTag = IO_REPARSE_TAG_MOUNT_POINT; |
| } |
| |
| bool JunctionResolver::Resolve(const WCHAR* path, unique_ptr<WCHAR[]>* result, |
| int max_junction_depth) { |
| DWORD attributes = ::GetFileAttributesW(path); |
| if (attributes == INVALID_FILE_ATTRIBUTES) { |
| // `path` does not exist. |
| return false; |
| } else { |
| if ((attributes & FILE_ATTRIBUTE_DIRECTORY) != 0 && |
| (attributes & FILE_ATTRIBUTE_REPARSE_POINT) != 0) { |
| // `path` is a junction. GetFileAttributesW succeeds for these even if |
| // their target does not exist. We need to resolve the target and check if |
| // that exists. (There seems to be no API function for this.) |
| if (max_junction_depth <= 0) { |
| // Too many levels of junctions. Simply say this file doesn't exist. |
| return false; |
| } |
| // Get a handle to the directory. |
| AutoHandle handle(OpenDirectory(path, /* read_write */ false)); |
| if (!handle.IsValid()) { |
| // Opening the junction failed for whatever reason. For all intents and |
| // purposes we can treat this file as if it didn't exist. |
| return false; |
| } |
| // Read out the junction data. |
| DWORD bytes_returned; |
| BOOL ok = ::DeviceIoControl( |
| handle, FSCTL_GET_REPARSE_POINT, NULL, 0, reparse_buffer_, |
| MAXIMUM_REPARSE_DATA_BUFFER_SIZE, &bytes_returned, NULL); |
| if (!ok) { |
| // Reading the junction data failed. For all intents and purposes we can |
| // treat this file as if it didn't exist. |
| return false; |
| } |
| reparse_buffer_->PathBuffer[reparse_buffer_->Size - 1] = UNICODE_NULL; |
| // Check if the junction target exists. |
| return Resolve(reparse_buffer_->PathBuffer, result, |
| max_junction_depth - 1); |
| } |
| } |
| // `path` is a normal file or directory. |
| if (result) { |
| size_t len = wcslen(path) + 1; |
| result->reset(new WCHAR[len]); |
| memcpy(result->get(), path, len * sizeof(WCHAR)); |
| } |
| return true; |
| } |
| |
| bool JunctionResolver::Resolve(const WCHAR* path, unique_ptr<WCHAR[]>* result) { |
| return Resolve(path, result, kMaximumJunctionDepth); |
| } |
| |
| bool ReadDirectorySymlink(const string& name, string* result) { |
| wstring wname; |
| string error; |
| if (!AsAbsoluteWindowsPath(name, &wname, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "ReadDirectorySymlink(" << name |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| unique_ptr<WCHAR[]> result_ptr; |
| if (!JunctionResolver().Resolve(wname.c_str(), &result_ptr)) { |
| return false; |
| } |
| *result = WstringToCstring(RemoveUncPrefixMaybe(result_ptr.get())).get(); |
| return true; |
| } |
| |
| bool PathExists(const string& path) { |
| if (path.empty()) { |
| return false; |
| } |
| if (IsDevNull(path.c_str())) { |
| return true; |
| } |
| wstring wpath; |
| string error; |
| if (!AsAbsoluteWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "PathExists(" << path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| return JunctionResolver().Resolve(wpath.c_str(), nullptr); |
| } |
| |
| string MakeCanonical(const char* path) { |
| if (IsDevNull(path)) { |
| return "NUL"; |
| } |
| wstring wpath; |
| if (path == nullptr || path[0] == 0) { |
| return ""; |
| } |
| string error; |
| if (!AsAbsoluteWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "MakeCanonical(" << path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| } |
| |
| // Resolve all segments of the path. Do this from leaf to root, so we always |
| // know that the path's tail is resolved and junctions may be found only in |
| // its head. |
| std::vector<wstring> realpath_reversed; |
| while (true) { |
| // Resolve the last segment. |
| unique_ptr<WCHAR[]> realpath; |
| if (!JunctionResolver().Resolve(wpath.c_str(), &realpath)) { |
| // The path doesn't exist or there are too many levels of indirection, |
| // so just give up. |
| return ""; |
| } |
| // The last segment is surely not a junction anymore. Split it off the path |
| // and keep resolving its ancestors until we reach the root directory. |
| pair<wstring, wstring> split(SplitPathW(realpath.get())); |
| if (split.second.empty()) { |
| // `wpath` was a root directory, we're done. |
| realpath_reversed.push_back(split.first); |
| break; |
| } else { |
| // `wpath` was not yet a root directory, split off the last segment and |
| // store it in the stack, keep resolving the rest. |
| realpath_reversed.push_back(split.second); |
| wpath = split.first; |
| } |
| } |
| |
| // Concatenate the segments in reverse order. |
| int segment_cnt = 0; |
| std::wstringstream builder; |
| for (auto segment = realpath_reversed.crbegin(); |
| segment != realpath_reversed.crend(); ++segment) { |
| if (segment_cnt < 2) { |
| segment_cnt++; |
| } else { |
| // Start appending '\' separator after not the first but the second |
| // segment, since the first segment is a drive name "c:\" and already |
| // has the separator. |
| builder << L"\\"; |
| } |
| builder << *segment; |
| } |
| wstring realpath(builder.str()); |
| if (HasUncPrefix(realpath.c_str())) { |
| // `realpath` has an UNC prefix if `path` did, or if `path` contained |
| // junctions. |
| // In the first case, the UNC prefix is the usual "\\?\", but in the second |
| // case it is "\??\", because that's what the junction resolution yields, |
| // because that's the prefix the filesystem uses for storing junction |
| // values. |
| // Since "\??\" is only meaningful for the kernel and not for usermode |
| // Win32 API functions, we need to replace this prefix with the usual "\\?\" |
| // one. |
| realpath[1] = L'\\'; |
| } |
| |
| // Resolve all 8dot3 style segments of the path, if any. The input path may |
| // have had some. Junctions may also refer to 8dot3 names. |
| unique_ptr<WCHAR[]> long_realpath; |
| wstring werror(GetLongPath(realpath.c_str(), &long_realpath)); |
| if (!werror.empty()) { |
| // TODO(laszlocsomor): refactor MakeCanonical to return an error message, |
| // return `werror` here. |
| return ""; |
| } |
| |
| // Convert the path to lower-case. |
| size_t size = |
| wcslen(long_realpath.get()) - (HasUncPrefix(long_realpath.get()) ? 4 : 0); |
| unique_ptr<WCHAR[]> lcase_realpath(new WCHAR[size + 1]); |
| const WCHAR* p_from = RemoveUncPrefixMaybe(long_realpath.get()); |
| WCHAR* p_to = lcase_realpath.get(); |
| while (size-- > 0) { |
| *p_to++ = towlower(*p_from++); |
| } |
| *p_to = 0; |
| return string(WstringToCstring(lcase_realpath.get()).get()); |
| } |
| |
| static bool CanReadFileW(const wstring& path) { |
| DWORD attrs = ::GetFileAttributesW(path.c_str()); |
| if ((attrs == INVALID_FILE_ATTRIBUTES) || |
| (attrs & FILE_ATTRIBUTE_DIRECTORY)) { |
| // The path doesn't exist or is a directory/junction. |
| return false; |
| } |
| // The only easy way to find out if a file is readable is to attempt to open |
| // it for reading. |
| AutoHandle handle(::CreateFileW( |
| /* lpFileName */ path.c_str(), |
| /* dwDesiredAccess */ GENERIC_READ, |
| /* dwShareMode */ FILE_SHARE_READ, |
| /* lpSecurityAttributes */ NULL, |
| /* dwCreationDisposition */ OPEN_EXISTING, |
| /* dwFlagsAndAttributes */ FILE_ATTRIBUTE_NORMAL, |
| /* hTemplateFile */ NULL)); |
| return handle.IsValid(); |
| } |
| |
| bool CanReadFile(const std::string& path) { |
| wstring wpath; |
| string error; |
| if (!AsAbsoluteWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "CanReadFile(" << path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| return CanReadFileW(wpath); |
| } |
| |
| bool CanExecuteFile(const std::string& path) { |
| wstring wpath; |
| string error; |
| if (!AsAbsoluteWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "CanExecuteFile(" << path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| return CanReadFileW(wpath) && (ends_with(wpath, wstring(L".exe")) || |
| ends_with(wpath, wstring(L".com")) || |
| ends_with(wpath, wstring(L".cmd")) || |
| ends_with(wpath, wstring(L".bat"))); |
| } |
| |
| bool CanAccessDirectory(const std::string& path) { |
| wstring wpath; |
| string error; |
| if (!AsAbsoluteWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "CanAccessDirectory(" << path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| DWORD attr = ::GetFileAttributesW(wpath.c_str()); |
| if ((attr == INVALID_FILE_ATTRIBUTES) || !(attr & FILE_ATTRIBUTE_DIRECTORY)) { |
| // The path doesn't exist or is not a directory. |
| return false; |
| } |
| |
| // The only easy way to know if a directory is writable is by attempting to |
| // open a file for writing in it. |
| wstring dummy_path = wpath + L"\\bazel_directory_access_test"; |
| |
| // The path may have just became too long for MAX_PATH, so add the UNC prefix |
| // if necessary. |
| AddUncPrefixMaybe(&dummy_path); |
| |
| // Attempt to open the dummy file for read/write access. |
| // If the file happens to exist, no big deal, we won't overwrite it thanks to |
| // OPEN_ALWAYS. |
| HANDLE handle = ::CreateFileW( |
| /* lpFileName */ dummy_path.c_str(), |
| /* dwDesiredAccess */ GENERIC_WRITE | GENERIC_READ, |
| /* dwShareMode */ FILE_SHARE_READ | FILE_SHARE_WRITE, |
| /* lpSecurityAttributes */ NULL, |
| /* dwCreationDisposition */ OPEN_ALWAYS, |
| /* dwFlagsAndAttributes */ FILE_ATTRIBUTE_NORMAL, |
| /* hTemplateFile */ NULL); |
| DWORD err = GetLastError(); |
| if (handle == INVALID_HANDLE_VALUE && err != ERROR_ALREADY_EXISTS) { |
| // We couldn't open the file, and not because the dummy file already exists. |
| // Consequently it is because `wpath` doesn't exist. |
| return false; |
| } |
| // The fact that we could open the file, regardless of it existing beforehand |
| // or not, means the directory also exists and we can read/write in it. |
| CloseHandle(handle); |
| if (err != ERROR_ALREADY_EXISTS) { |
| // The file didn't exist before, but due to OPEN_ALWAYS we created it just |
| // now, so do delete it. |
| ::DeleteFileW(dummy_path.c_str()); |
| } // Otherwise the file existed before, leave it alone. |
| return true; |
| } |
| |
| static bool IsDirectoryW(const wstring& path) { |
| DWORD attrs = ::GetFileAttributesW(path.c_str()); |
| return (attrs != INVALID_FILE_ATTRIBUTES) && |
| (attrs & FILE_ATTRIBUTE_DIRECTORY) && |
| JunctionResolver().Resolve(path.c_str(), nullptr); |
| } |
| |
| bool IsDirectory(const string& path) { |
| if (path.empty() || IsDevNull(path.c_str())) { |
| return false; |
| } |
| wstring wpath; |
| string error; |
| if (!AsAbsoluteWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "IsDirectory(" << path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| return IsDirectoryW(wpath); |
| } |
| |
| void SyncFile(const string& path) { |
| // No-op on Windows native; unsupported by Cygwin. |
| // fsync always fails on Cygwin with "Permission denied" for some reason. |
| } |
| |
| bool MakeDirectoriesW(const wstring& path, unsigned int mode) { |
| if (path.empty()) { |
| return false; |
| } |
| if (IsRootDirectoryW(path) || IsDirectoryW(path)) { |
| return true; |
| } |
| wstring parent = SplitPathW(path).first; |
| if (parent.empty()) { |
| // Since `path` is not a root directory, there should have been at least one |
| // directory above it. |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "MakeDirectoriesW(" << blaze_util::WstringToString(path) |
| << ") could not find dirname: " << GetLastErrorString(); |
| } |
| return MakeDirectoriesW(parent, mode) && |
| ::CreateDirectoryW(path.c_str(), NULL) == TRUE; |
| } |
| |
| bool MakeDirectories(const string& path, unsigned int mode) { |
| // TODO(laszlocsomor): respect `mode` to the extent that it's possible on |
| // Windows; it's currently ignored. |
| if (path.empty() || IsDevNull(path.c_str())) { |
| return false; |
| } |
| wstring wpath; |
| // According to MSDN, CreateDirectory's limit without the UNC prefix is |
| // 248 characters (so it could fit another filename before reaching MAX_PATH). |
| string error; |
| if (!AsAbsoluteWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "MakeDirectories(" << path |
| << "): AsAbsoluteWindowsPath failed: " << error; |
| return false; |
| } |
| return MakeDirectoriesW(wpath, mode); |
| } |
| |
| std::wstring GetCwdW() { |
| DWORD len = ::GetCurrentDirectoryW(0, nullptr); |
| unique_ptr<WCHAR[]> cwd(new WCHAR[len]); |
| if (!::GetCurrentDirectoryW(len, cwd.get())) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "GetCurrentDirectoryW failed: " << GetLastErrorString(); |
| } |
| for (WCHAR* p = cwd.get(); *p != 0; ++p) { |
| *p = towlower(*p); |
| } |
| return std::wstring(cwd.get()); |
| } |
| |
| string GetCwd() { |
| return string( |
| WstringToCstring(RemoveUncPrefixMaybe(GetCwdW().c_str())).get()); |
| } |
| |
| bool ChangeDirectory(const string& path) { |
| string spath; |
| string error; |
| if (!AsShortWindowsPath(path, &spath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "ChangeDirectory(" << path << "): failed: " << error; |
| } |
| return ::SetCurrentDirectoryA(spath.c_str()) == TRUE; |
| } |
| |
| class DirectoryTreeWalkerW : public DirectoryEntryConsumerW { |
| public: |
| DirectoryTreeWalkerW(vector<wstring>* files, |
| _ForEachDirectoryEntryW walk_entries) |
| : _files(files), _walk_entries(walk_entries) {} |
| |
| void Consume(const wstring& path, bool follow_directory) override { |
| if (follow_directory) { |
| Walk(path); |
| } else { |
| _files->push_back(path); |
| } |
| } |
| |
| void Walk(const wstring& path) { _walk_entries(path, this); } |
| |
| private: |
| vector<wstring>* _files; |
| _ForEachDirectoryEntryW _walk_entries; |
| }; |
| |
| void ForEachDirectoryEntryW(const wstring& path, |
| DirectoryEntryConsumerW* consume) { |
| wstring wpath; |
| if (path.empty() || IsDevNull(path.c_str())) { |
| return; |
| } |
| string error; |
| if (!AsWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "ForEachDirectoryEntryW(" << path |
| << "): AsWindowsPath failed: " << GetLastErrorString(); |
| } |
| |
| static const wstring kUncPrefix(L"\\\\?\\"); |
| static const wstring kDot(L"."); |
| static const wstring kDotDot(L".."); |
| // Always add an UNC prefix to ensure we can work with long paths. |
| if (!HasUncPrefix(wpath.c_str())) { |
| wpath = kUncPrefix + wpath; |
| } |
| // Unconditionally add a trailing backslash. We know `wpath` has no trailing |
| // backslash because it comes from AsWindowsPath whose output is always |
| // normalized (see NormalizeWindowsPath). |
| wpath.append(L"\\"); |
| WIN32_FIND_DATAW metadata; |
| HANDLE handle = ::FindFirstFileW((wpath + L"*").c_str(), &metadata); |
| if (handle == INVALID_HANDLE_VALUE) { |
| return; // directory does not exist or is empty |
| } |
| |
| do { |
| if (kDot != metadata.cFileName && kDotDot != metadata.cFileName) { |
| wstring wname = wpath + metadata.cFileName; |
| wstring name(/* omit prefix */ 4 + wname.c_str()); |
| bool is_dir = (metadata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0; |
| bool is_junc = |
| (metadata.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) != 0; |
| consume->Consume(name, is_dir && !is_junc); |
| } |
| } while (::FindNextFileW(handle, &metadata)); |
| ::FindClose(handle); |
| } |
| |
| void GetAllFilesUnderW(const wstring& path, vector<wstring>* result) { |
| _GetAllFilesUnderW(path, result, &ForEachDirectoryEntryW); |
| } |
| |
| void _GetAllFilesUnderW(const wstring& path, vector<wstring>* result, |
| _ForEachDirectoryEntryW walk_entries) { |
| DirectoryTreeWalkerW(result, walk_entries).Walk(path); |
| } |
| |
| void ForEachDirectoryEntry(const string &path, |
| DirectoryEntryConsumer *consume) { |
| wstring wpath; |
| if (path.empty() || IsDevNull(path.c_str())) { |
| return; |
| } |
| string error; |
| if (!AsWindowsPath(path, &wpath, &error)) { |
| BAZEL_DIE(blaze_exit_code::LOCAL_ENVIRONMENTAL_ERROR) |
| << "ForEachDirectoryEntry(" << path |
| << "): AsWindowsPath failed: " << GetLastErrorString(); |
| } |
| |
| static const wstring kUncPrefix(L"\\\\?\\"); |
| static const wstring kDot(L"."); |
| static const wstring kDotDot(L".."); |
| // Always add an UNC prefix to ensure we can work with long paths. |
| if (!HasUncPrefix(wpath.c_str())) { |
| wpath = kUncPrefix + wpath; |
| } |
| // Unconditionally add a trailing backslash. We know `wpath` has no trailing |
| // backslash because it comes from AsWindowsPath whose output is always |
| // normalized (see NormalizeWindowsPath). |
| wpath.append(L"\\"); |
| WIN32_FIND_DATAW metadata; |
| HANDLE handle = ::FindFirstFileW((wpath + L"*").c_str(), &metadata); |
| if (handle == INVALID_HANDLE_VALUE) { |
| return; // directory does not exist or is empty |
| } |
| |
| do { |
| if (kDot != metadata.cFileName && kDotDot != metadata.cFileName) { |
| wstring wname = wpath + metadata.cFileName; |
| string name(WstringToCstring(/* omit prefix */ 4 + wname.c_str()).get()); |
| bool is_dir = (metadata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0; |
| bool is_junc = |
| (metadata.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) != 0; |
| consume->Consume(name, is_dir && !is_junc); |
| } |
| } while (::FindNextFileW(handle, &metadata)); |
| ::FindClose(handle); |
| } |
| |
| } // namespace blaze_util |