mirror of
https://github.com/RPCS3/llvm.git
synced 2024-12-27 22:55:15 +00:00
d04a8d4b33
Sooooo many of these had incorrect or strange main module includes. I have manually inspected all of these, and fixed the main module include to be the nearest plausible thing I could find. If you own or care about any of these source files, I encourage you to take some time and check that these edits were sensible. I can't have broken anything (I strictly added headers, and reordered them, never removed), but they may not be the headers you'd really like to identify as containing the API being implemented. Many forward declarations and missing includes were added to a header files to allow them to parse cleanly when included first. The main module rule does in fact have its merits. =] git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@169131 91177308-0d34-0410-b5e6-96231b3b80d8
927 lines
26 KiB
C++
927 lines
26 KiB
C++
//===- llvm/Support/Win32/Path.cpp - Win32 Path Implementation ---*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file provides the Win32 specific implementation of the Path class.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
//=== WARNING: Implementation here must contain only generic Win32 code that
|
|
//=== is guaranteed to work on *all* Win32 variants.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "Windows.h"
|
|
#include <cstdio>
|
|
#include <malloc.h>
|
|
|
|
// We need to undo a macro defined in Windows.h, otherwise we won't compile:
|
|
#undef CopyFile
|
|
#undef GetCurrentDirectory
|
|
|
|
// Windows happily accepts either forward or backward slashes, though any path
|
|
// returned by a Win32 API will have backward slashes. As LLVM code basically
|
|
// assumes forward slashes are used, backward slashs are converted where they
|
|
// can be introduced into a path.
|
|
//
|
|
// Another invariant is that a path ends with a slash if and only if the path
|
|
// is a root directory. Any other use of a trailing slash is stripped. Unlike
|
|
// in Unix, Windows has a rather complicated notion of a root path and this
|
|
// invariant helps simply the code.
|
|
|
|
static void FlipBackSlashes(std::string& s) {
|
|
for (size_t i = 0; i < s.size(); i++)
|
|
if (s[i] == '\\')
|
|
s[i] = '/';
|
|
}
|
|
|
|
namespace llvm {
|
|
namespace sys {
|
|
|
|
const char PathSeparator = ';';
|
|
|
|
StringRef Path::GetEXESuffix() {
|
|
return "exe";
|
|
}
|
|
|
|
Path::Path(llvm::StringRef p)
|
|
: path(p) {
|
|
FlipBackSlashes(path);
|
|
}
|
|
|
|
Path::Path(const char *StrStart, unsigned StrLen)
|
|
: path(StrStart, StrLen) {
|
|
FlipBackSlashes(path);
|
|
}
|
|
|
|
Path&
|
|
Path::operator=(StringRef that) {
|
|
path.assign(that.data(), that.size());
|
|
FlipBackSlashes(path);
|
|
return *this;
|
|
}
|
|
|
|
bool
|
|
Path::isValid() const {
|
|
if (path.empty())
|
|
return false;
|
|
|
|
size_t len = path.size();
|
|
// If there is a null character, it and all its successors are ignored.
|
|
size_t pos = path.find_first_of('\0');
|
|
if (pos != std::string::npos)
|
|
len = pos;
|
|
|
|
// If there is a colon, it must be the second character, preceded by a letter
|
|
// and followed by something.
|
|
pos = path.rfind(':',len);
|
|
size_t rootslash = 0;
|
|
if (pos != std::string::npos) {
|
|
if (pos != 1 || !isalpha(path[0]) || len < 3)
|
|
return false;
|
|
rootslash = 2;
|
|
}
|
|
|
|
// Look for a UNC path, and if found adjust our notion of the root slash.
|
|
if (len > 3 && path[0] == '/' && path[1] == '/') {
|
|
rootslash = path.find('/', 2);
|
|
if (rootslash == std::string::npos)
|
|
rootslash = 0;
|
|
}
|
|
|
|
// Check for illegal characters.
|
|
if (path.find_first_of("\\<>\"|\001\002\003\004\005\006\007\010\011\012"
|
|
"\013\014\015\016\017\020\021\022\023\024\025\026"
|
|
"\027\030\031\032\033\034\035\036\037")
|
|
!= std::string::npos)
|
|
return false;
|
|
|
|
// Remove trailing slash, unless it's a root slash.
|
|
if (len > rootslash+1 && path[len-1] == '/')
|
|
path.erase(--len);
|
|
|
|
// Check each component for legality.
|
|
for (pos = 0; pos < len; ++pos) {
|
|
// A component may not end in a space.
|
|
if (path[pos] == ' ') {
|
|
if (pos+1 == len || path[pos+1] == '/' || path[pos+1] == '\0')
|
|
return false;
|
|
}
|
|
|
|
// A component may not end in a period.
|
|
if (path[pos] == '.') {
|
|
if (pos+1 == len || path[pos+1] == '/') {
|
|
// Unless it is the pseudo-directory "."...
|
|
if (pos == 0 || path[pos-1] == '/' || path[pos-1] == ':')
|
|
return true;
|
|
// or "..".
|
|
if (pos > 0 && path[pos-1] == '.') {
|
|
if (pos == 1 || path[pos-2] == '/' || path[pos-2] == ':')
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
void Path::makeAbsolute() {
|
|
TCHAR FullPath[MAX_PATH + 1] = {0};
|
|
LPTSTR FilePart = NULL;
|
|
|
|
DWORD RetLength = ::GetFullPathNameA(path.c_str(),
|
|
sizeof(FullPath)/sizeof(FullPath[0]),
|
|
FullPath, &FilePart);
|
|
|
|
if (0 == RetLength) {
|
|
// FIXME: Report the error GetLastError()
|
|
assert(0 && "Unable to make absolute path!");
|
|
} else if (RetLength > MAX_PATH) {
|
|
// FIXME: Report too small buffer (needed RetLength bytes).
|
|
assert(0 && "Unable to make absolute path!");
|
|
} else {
|
|
path = FullPath;
|
|
}
|
|
}
|
|
|
|
bool
|
|
Path::isAbsolute(const char *NameStart, unsigned NameLen) {
|
|
assert(NameStart);
|
|
// FIXME: This does not handle correctly an absolute path starting from
|
|
// a drive letter or in UNC format.
|
|
switch (NameLen) {
|
|
case 0:
|
|
return false;
|
|
case 1:
|
|
case 2:
|
|
return NameStart[0] == '/';
|
|
default:
|
|
return
|
|
(NameStart[0] == '/' || (NameStart[1] == ':' && NameStart[2] == '/')) ||
|
|
(NameStart[0] == '\\' || (NameStart[1] == ':' && NameStart[2] == '\\'));
|
|
}
|
|
}
|
|
|
|
bool
|
|
Path::isAbsolute() const {
|
|
// FIXME: This does not handle correctly an absolute path starting from
|
|
// a drive letter or in UNC format.
|
|
switch (path.length()) {
|
|
case 0:
|
|
return false;
|
|
case 1:
|
|
case 2:
|
|
return path[0] == '/';
|
|
default:
|
|
return path[0] == '/' || (path[1] == ':' && path[2] == '/');
|
|
}
|
|
}
|
|
|
|
static Path *TempDirectory;
|
|
|
|
Path
|
|
Path::GetTemporaryDirectory(std::string* ErrMsg) {
|
|
if (TempDirectory) {
|
|
#if defined(_MSC_VER)
|
|
// Visual Studio gets confused and emits a diagnostic about calling exists,
|
|
// even though this is the implementation for PathV1. Temporarily
|
|
// disable the deprecated warning message
|
|
#pragma warning(push)
|
|
#pragma warning(disable:4996)
|
|
#endif
|
|
assert(TempDirectory->exists() && "Who has removed TempDirectory?");
|
|
#if defined(_MSC_VER)
|
|
#pragma warning(pop)
|
|
#endif
|
|
return *TempDirectory;
|
|
}
|
|
|
|
char pathname[MAX_PATH];
|
|
if (!GetTempPath(MAX_PATH, pathname)) {
|
|
if (ErrMsg)
|
|
*ErrMsg = "Can't determine temporary directory";
|
|
return Path();
|
|
}
|
|
|
|
Path result;
|
|
result.set(pathname);
|
|
|
|
// Append a subdirectory based on our process id so multiple LLVMs don't
|
|
// step on each other's toes.
|
|
#ifdef __MINGW32__
|
|
// Mingw's Win32 header files are broken.
|
|
sprintf(pathname, "LLVM_%u", unsigned(GetCurrentProcessId()));
|
|
#else
|
|
sprintf(pathname, "LLVM_%u", GetCurrentProcessId());
|
|
#endif
|
|
result.appendComponent(pathname);
|
|
|
|
// If there's a directory left over from a previous LLVM execution that
|
|
// happened to have the same process id, get rid of it.
|
|
result.eraseFromDisk(true);
|
|
|
|
// And finally (re-)create the empty directory.
|
|
result.createDirectoryOnDisk(false);
|
|
TempDirectory = new Path(result);
|
|
return *TempDirectory;
|
|
}
|
|
|
|
// FIXME: the following set of functions don't map to Windows very well.
|
|
Path
|
|
Path::GetRootDirectory() {
|
|
// This is the only notion that that Windows has of a root directory. Nothing
|
|
// is here except for drives.
|
|
return Path("file:///");
|
|
}
|
|
|
|
void
|
|
Path::GetSystemLibraryPaths(std::vector<sys::Path>& Paths) {
|
|
char buff[MAX_PATH];
|
|
// Generic form of C:\Windows\System32
|
|
HRESULT res = SHGetFolderPathA(NULL,
|
|
CSIDL_FLAG_CREATE | CSIDL_SYSTEM,
|
|
NULL,
|
|
SHGFP_TYPE_CURRENT,
|
|
buff);
|
|
if (res != S_OK) {
|
|
assert(0 && "Failed to get system directory");
|
|
return;
|
|
}
|
|
Paths.push_back(sys::Path(buff));
|
|
|
|
// Reset buff.
|
|
buff[0] = 0;
|
|
// Generic form of C:\Windows
|
|
res = SHGetFolderPathA(NULL,
|
|
CSIDL_FLAG_CREATE | CSIDL_WINDOWS,
|
|
NULL,
|
|
SHGFP_TYPE_CURRENT,
|
|
buff);
|
|
if (res != S_OK) {
|
|
assert(0 && "Failed to get windows directory");
|
|
return;
|
|
}
|
|
Paths.push_back(sys::Path(buff));
|
|
}
|
|
|
|
void
|
|
Path::GetBitcodeLibraryPaths(std::vector<sys::Path>& Paths) {
|
|
char * env_var = getenv("LLVM_LIB_SEARCH_PATH");
|
|
if (env_var != 0) {
|
|
getPathList(env_var,Paths);
|
|
}
|
|
#ifdef LLVM_LIBDIR
|
|
{
|
|
Path tmpPath;
|
|
if (tmpPath.set(LLVM_LIBDIR))
|
|
if (tmpPath.canRead())
|
|
Paths.push_back(tmpPath);
|
|
}
|
|
#endif
|
|
GetSystemLibraryPaths(Paths);
|
|
}
|
|
|
|
Path
|
|
Path::GetUserHomeDirectory() {
|
|
char buff[MAX_PATH];
|
|
HRESULT res = SHGetFolderPathA(NULL,
|
|
CSIDL_FLAG_CREATE | CSIDL_APPDATA,
|
|
NULL,
|
|
SHGFP_TYPE_CURRENT,
|
|
buff);
|
|
if (res != S_OK)
|
|
assert(0 && "Failed to get user home directory");
|
|
return Path(buff);
|
|
}
|
|
|
|
Path
|
|
Path::GetCurrentDirectory() {
|
|
char pathname[MAX_PATH];
|
|
::GetCurrentDirectoryA(MAX_PATH,pathname);
|
|
return Path(pathname);
|
|
}
|
|
|
|
/// GetMainExecutable - Return the path to the main executable, given the
|
|
/// value of argv[0] from program startup.
|
|
Path Path::GetMainExecutable(const char *argv0, void *MainAddr) {
|
|
char pathname[MAX_PATH];
|
|
DWORD ret = ::GetModuleFileNameA(NULL, pathname, MAX_PATH);
|
|
return ret != MAX_PATH ? Path(pathname) : Path();
|
|
}
|
|
|
|
|
|
// FIXME: the above set of functions don't map to Windows very well.
|
|
|
|
|
|
StringRef Path::getDirname() const {
|
|
return getDirnameCharSep(path, "/");
|
|
}
|
|
|
|
StringRef
|
|
Path::getBasename() const {
|
|
// Find the last slash
|
|
size_t slash = path.rfind('/');
|
|
if (slash == std::string::npos)
|
|
slash = 0;
|
|
else
|
|
slash++;
|
|
|
|
size_t dot = path.rfind('.');
|
|
if (dot == std::string::npos || dot < slash)
|
|
return StringRef(path).substr(slash);
|
|
else
|
|
return StringRef(path).substr(slash, dot - slash);
|
|
}
|
|
|
|
StringRef
|
|
Path::getSuffix() const {
|
|
// Find the last slash
|
|
size_t slash = path.rfind('/');
|
|
if (slash == std::string::npos)
|
|
slash = 0;
|
|
else
|
|
slash++;
|
|
|
|
size_t dot = path.rfind('.');
|
|
if (dot == std::string::npos || dot < slash)
|
|
return StringRef("");
|
|
else
|
|
return StringRef(path).substr(dot + 1);
|
|
}
|
|
|
|
bool
|
|
Path::exists() const {
|
|
DWORD attr = GetFileAttributes(path.c_str());
|
|
return attr != INVALID_FILE_ATTRIBUTES;
|
|
}
|
|
|
|
bool
|
|
Path::isDirectory() const {
|
|
DWORD attr = GetFileAttributes(path.c_str());
|
|
return (attr != INVALID_FILE_ATTRIBUTES) &&
|
|
(attr & FILE_ATTRIBUTE_DIRECTORY);
|
|
}
|
|
|
|
bool
|
|
Path::isSymLink() const {
|
|
DWORD attributes = GetFileAttributes(path.c_str());
|
|
|
|
if (attributes == INVALID_FILE_ATTRIBUTES)
|
|
// There's no sane way to report this :(.
|
|
assert(0 && "GetFileAttributes returned INVALID_FILE_ATTRIBUTES");
|
|
|
|
// This isn't exactly what defines a NTFS symlink, but it is only true for
|
|
// paths that act like a symlink.
|
|
return attributes & FILE_ATTRIBUTE_REPARSE_POINT;
|
|
}
|
|
|
|
bool
|
|
Path::canRead() const {
|
|
// FIXME: take security attributes into account.
|
|
DWORD attr = GetFileAttributes(path.c_str());
|
|
return attr != INVALID_FILE_ATTRIBUTES;
|
|
}
|
|
|
|
bool
|
|
Path::canWrite() const {
|
|
// FIXME: take security attributes into account.
|
|
DWORD attr = GetFileAttributes(path.c_str());
|
|
return (attr != INVALID_FILE_ATTRIBUTES) && !(attr & FILE_ATTRIBUTE_READONLY);
|
|
}
|
|
|
|
bool
|
|
Path::canExecute() const {
|
|
// FIXME: take security attributes into account.
|
|
DWORD attr = GetFileAttributes(path.c_str());
|
|
return attr != INVALID_FILE_ATTRIBUTES;
|
|
}
|
|
|
|
bool
|
|
Path::isRegularFile() const {
|
|
bool res;
|
|
if (fs::is_regular_file(path, res))
|
|
return false;
|
|
return res;
|
|
}
|
|
|
|
StringRef
|
|
Path::getLast() const {
|
|
// Find the last slash
|
|
size_t pos = path.rfind('/');
|
|
|
|
// Handle the corner cases
|
|
if (pos == std::string::npos)
|
|
return path;
|
|
|
|
// If the last character is a slash, we have a root directory
|
|
if (pos == path.length()-1)
|
|
return path;
|
|
|
|
// Return everything after the last slash
|
|
return StringRef(path).substr(pos+1);
|
|
}
|
|
|
|
const FileStatus *
|
|
PathWithStatus::getFileStatus(bool update, std::string *ErrStr) const {
|
|
if (!fsIsValid || update) {
|
|
WIN32_FILE_ATTRIBUTE_DATA fi;
|
|
if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
|
|
MakeErrMsg(ErrStr, "getStatusInfo():" + std::string(path) +
|
|
": Can't get status: ");
|
|
return 0;
|
|
}
|
|
|
|
status.fileSize = fi.nFileSizeHigh;
|
|
status.fileSize <<= sizeof(fi.nFileSizeHigh)*8;
|
|
status.fileSize += fi.nFileSizeLow;
|
|
|
|
status.mode = fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY ? 0555 : 0777;
|
|
status.user = 9999; // Not applicable to Windows, so...
|
|
status.group = 9999; // Not applicable to Windows, so...
|
|
|
|
// FIXME: this is only unique if the file is accessed by the same file path.
|
|
// How do we do this for C:\dir\file and ..\dir\file ? Unix has inode
|
|
// numbers, but the concept doesn't exist in Windows.
|
|
status.uniqueID = 0;
|
|
for (unsigned i = 0; i < path.length(); ++i)
|
|
status.uniqueID += path[i];
|
|
|
|
ULARGE_INTEGER ui;
|
|
ui.LowPart = fi.ftLastWriteTime.dwLowDateTime;
|
|
ui.HighPart = fi.ftLastWriteTime.dwHighDateTime;
|
|
status.modTime.fromWin32Time(ui.QuadPart);
|
|
|
|
status.isDir = fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
|
|
fsIsValid = true;
|
|
}
|
|
return &status;
|
|
}
|
|
|
|
bool Path::makeReadableOnDisk(std::string* ErrMsg) {
|
|
// All files are readable on Windows (ignoring security attributes).
|
|
return false;
|
|
}
|
|
|
|
bool Path::makeWriteableOnDisk(std::string* ErrMsg) {
|
|
DWORD attr = GetFileAttributes(path.c_str());
|
|
|
|
// If it doesn't exist, we're done.
|
|
if (attr == INVALID_FILE_ATTRIBUTES)
|
|
return false;
|
|
|
|
if (attr & FILE_ATTRIBUTE_READONLY) {
|
|
if (!SetFileAttributes(path.c_str(), attr & ~FILE_ATTRIBUTE_READONLY)) {
|
|
MakeErrMsg(ErrMsg, std::string(path) + ": Can't make file writable: ");
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool Path::makeExecutableOnDisk(std::string* ErrMsg) {
|
|
// All files are executable on Windows (ignoring security attributes).
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
Path::getDirectoryContents(std::set<Path>& result, std::string* ErrMsg) const {
|
|
WIN32_FILE_ATTRIBUTE_DATA fi;
|
|
if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
|
|
MakeErrMsg(ErrMsg, path + ": can't get status of file");
|
|
return true;
|
|
}
|
|
|
|
if (!(fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
|
|
if (ErrMsg)
|
|
*ErrMsg = path + ": not a directory";
|
|
return true;
|
|
}
|
|
|
|
result.clear();
|
|
WIN32_FIND_DATA fd;
|
|
std::string searchpath = path;
|
|
if (path.size() == 0 || searchpath[path.size()-1] == '/')
|
|
searchpath += "*";
|
|
else
|
|
searchpath += "/*";
|
|
|
|
HANDLE h = FindFirstFile(searchpath.c_str(), &fd);
|
|
if (h == INVALID_HANDLE_VALUE) {
|
|
if (GetLastError() == ERROR_FILE_NOT_FOUND)
|
|
return true; // not really an error, now is it?
|
|
MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
|
|
return true;
|
|
}
|
|
|
|
do {
|
|
if (fd.cFileName[0] == '.')
|
|
continue;
|
|
Path aPath(path);
|
|
aPath.appendComponent(&fd.cFileName[0]);
|
|
result.insert(aPath);
|
|
} while (FindNextFile(h, &fd));
|
|
|
|
DWORD err = GetLastError();
|
|
FindClose(h);
|
|
if (err != ERROR_NO_MORE_FILES) {
|
|
SetLastError(err);
|
|
MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
Path::set(StringRef a_path) {
|
|
if (a_path.empty())
|
|
return false;
|
|
std::string save(path);
|
|
path = a_path;
|
|
FlipBackSlashes(path);
|
|
if (!isValid()) {
|
|
path = save;
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Path::appendComponent(StringRef name) {
|
|
if (name.empty())
|
|
return false;
|
|
std::string save(path);
|
|
if (!path.empty()) {
|
|
size_t last = path.size() - 1;
|
|
if (path[last] != '/')
|
|
path += '/';
|
|
}
|
|
path += name;
|
|
if (!isValid()) {
|
|
path = save;
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Path::eraseComponent() {
|
|
size_t slashpos = path.rfind('/',path.size());
|
|
if (slashpos == path.size() - 1 || slashpos == std::string::npos)
|
|
return false;
|
|
std::string save(path);
|
|
path.erase(slashpos);
|
|
if (!isValid()) {
|
|
path = save;
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Path::eraseSuffix() {
|
|
size_t dotpos = path.rfind('.',path.size());
|
|
size_t slashpos = path.rfind('/',path.size());
|
|
if (dotpos != std::string::npos) {
|
|
if (slashpos == std::string::npos || dotpos > slashpos+1) {
|
|
std::string save(path);
|
|
path.erase(dotpos, path.size()-dotpos);
|
|
if (!isValid()) {
|
|
path = save;
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
inline bool PathMsg(std::string* ErrMsg, const char* pathname, const char*msg) {
|
|
if (ErrMsg)
|
|
*ErrMsg = std::string(pathname) + ": " + std::string(msg);
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Path::createDirectoryOnDisk(bool create_parents, std::string* ErrMsg) {
|
|
// Get a writeable copy of the path name
|
|
size_t len = path.length();
|
|
char *pathname = reinterpret_cast<char *>(_alloca(len+2));
|
|
path.copy(pathname, len);
|
|
pathname[len] = 0;
|
|
|
|
// Make sure it ends with a slash.
|
|
if (len == 0 || pathname[len - 1] != '/') {
|
|
pathname[len] = '/';
|
|
pathname[++len] = 0;
|
|
}
|
|
|
|
// Determine starting point for initial / search.
|
|
char *next = pathname;
|
|
if (pathname[0] == '/' && pathname[1] == '/') {
|
|
// Skip host name.
|
|
next = strchr(pathname+2, '/');
|
|
if (next == NULL)
|
|
return PathMsg(ErrMsg, pathname, "badly formed remote directory");
|
|
|
|
// Skip share name.
|
|
next = strchr(next+1, '/');
|
|
if (next == NULL)
|
|
return PathMsg(ErrMsg, pathname,"badly formed remote directory");
|
|
|
|
next++;
|
|
if (*next == 0)
|
|
return PathMsg(ErrMsg, pathname, "badly formed remote directory");
|
|
|
|
} else {
|
|
if (pathname[1] == ':')
|
|
next += 2; // skip drive letter
|
|
if (*next == '/')
|
|
next++; // skip root directory
|
|
}
|
|
|
|
// If we're supposed to create intermediate directories
|
|
if (create_parents) {
|
|
// Loop through the directory components until we're done
|
|
while (*next) {
|
|
next = strchr(next, '/');
|
|
*next = 0;
|
|
if (!CreateDirectory(pathname, NULL) &&
|
|
GetLastError() != ERROR_ALREADY_EXISTS)
|
|
return MakeErrMsg(ErrMsg,
|
|
std::string(pathname) + ": Can't create directory: ");
|
|
*next++ = '/';
|
|
}
|
|
} else {
|
|
// Drop trailing slash.
|
|
pathname[len-1] = 0;
|
|
if (!CreateDirectory(pathname, NULL) &&
|
|
GetLastError() != ERROR_ALREADY_EXISTS) {
|
|
return MakeErrMsg(ErrMsg, std::string(pathname) +
|
|
": Can't create directory: ");
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
Path::createFileOnDisk(std::string* ErrMsg) {
|
|
// Create the file
|
|
HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
|
|
FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (h == INVALID_HANDLE_VALUE)
|
|
return MakeErrMsg(ErrMsg, path + ": Can't create file: ");
|
|
|
|
CloseHandle(h);
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
Path::eraseFromDisk(bool remove_contents, std::string *ErrStr) const {
|
|
WIN32_FILE_ATTRIBUTE_DATA fi;
|
|
if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi))
|
|
return true;
|
|
|
|
if (fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
|
|
// If it doesn't exist, we're done.
|
|
bool Exists;
|
|
if (fs::exists(path, Exists) || !Exists)
|
|
return false;
|
|
|
|
char *pathname = reinterpret_cast<char *>(_alloca(path.length()+3));
|
|
int lastchar = path.length() - 1 ;
|
|
path.copy(pathname, lastchar+1);
|
|
|
|
// Make path end with '/*'.
|
|
if (pathname[lastchar] != '/')
|
|
pathname[++lastchar] = '/';
|
|
pathname[lastchar+1] = '*';
|
|
pathname[lastchar+2] = 0;
|
|
|
|
if (remove_contents) {
|
|
WIN32_FIND_DATA fd;
|
|
HANDLE h = FindFirstFile(pathname, &fd);
|
|
|
|
// It's a bad idea to alter the contents of a directory while enumerating
|
|
// its contents. So build a list of its contents first, then destroy them.
|
|
|
|
if (h != INVALID_HANDLE_VALUE) {
|
|
std::vector<Path> list;
|
|
|
|
do {
|
|
if (strcmp(fd.cFileName, ".") == 0)
|
|
continue;
|
|
if (strcmp(fd.cFileName, "..") == 0)
|
|
continue;
|
|
|
|
Path aPath(path);
|
|
aPath.appendComponent(&fd.cFileName[0]);
|
|
list.push_back(aPath);
|
|
} while (FindNextFile(h, &fd));
|
|
|
|
DWORD err = GetLastError();
|
|
FindClose(h);
|
|
if (err != ERROR_NO_MORE_FILES) {
|
|
SetLastError(err);
|
|
return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
|
|
}
|
|
|
|
for (std::vector<Path>::iterator I = list.begin(); I != list.end();
|
|
++I) {
|
|
Path &aPath = *I;
|
|
aPath.eraseFromDisk(true);
|
|
}
|
|
} else {
|
|
if (GetLastError() != ERROR_FILE_NOT_FOUND)
|
|
return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
|
|
}
|
|
}
|
|
|
|
pathname[lastchar] = 0;
|
|
if (!RemoveDirectory(pathname))
|
|
return MakeErrMsg(ErrStr,
|
|
std::string(pathname) + ": Can't destroy directory: ");
|
|
return false;
|
|
} else {
|
|
// Read-only files cannot be deleted on Windows. Must remove the read-only
|
|
// attribute first.
|
|
if (fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
|
|
if (!SetFileAttributes(path.c_str(),
|
|
fi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
|
|
return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
|
|
}
|
|
|
|
if (!DeleteFile(path.c_str()))
|
|
return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool Path::getMagicNumber(std::string& Magic, unsigned len) const {
|
|
assert(len < 1024 && "Request for magic string too long");
|
|
char* buf = reinterpret_cast<char*>(alloca(len));
|
|
|
|
HANDLE h = CreateFile(path.c_str(),
|
|
GENERIC_READ,
|
|
FILE_SHARE_READ,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL);
|
|
if (h == INVALID_HANDLE_VALUE)
|
|
return false;
|
|
|
|
DWORD nRead = 0;
|
|
BOOL ret = ReadFile(h, buf, len, &nRead, NULL);
|
|
CloseHandle(h);
|
|
|
|
if (!ret || nRead != len)
|
|
return false;
|
|
|
|
Magic = std::string(buf, len);
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Path::renamePathOnDisk(const Path& newName, std::string* ErrMsg) {
|
|
if (!MoveFileEx(path.c_str(), newName.c_str(), MOVEFILE_REPLACE_EXISTING))
|
|
return MakeErrMsg(ErrMsg, "Can't move '" + path + "' to '" + newName.path
|
|
+ "': ");
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
Path::setStatusInfoOnDisk(const FileStatus &si, std::string *ErrMsg) const {
|
|
// FIXME: should work on directories also.
|
|
if (!si.isFile) {
|
|
return true;
|
|
}
|
|
|
|
HANDLE h = CreateFile(path.c_str(),
|
|
FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES,
|
|
FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL);
|
|
if (h == INVALID_HANDLE_VALUE)
|
|
return true;
|
|
|
|
BY_HANDLE_FILE_INFORMATION bhfi;
|
|
if (!GetFileInformationByHandle(h, &bhfi)) {
|
|
DWORD err = GetLastError();
|
|
CloseHandle(h);
|
|
SetLastError(err);
|
|
return MakeErrMsg(ErrMsg, path + ": GetFileInformationByHandle: ");
|
|
}
|
|
|
|
ULARGE_INTEGER ui;
|
|
ui.QuadPart = si.modTime.toWin32Time();
|
|
FILETIME ft;
|
|
ft.dwLowDateTime = ui.LowPart;
|
|
ft.dwHighDateTime = ui.HighPart;
|
|
BOOL ret = SetFileTime(h, NULL, &ft, &ft);
|
|
DWORD err = GetLastError();
|
|
CloseHandle(h);
|
|
if (!ret) {
|
|
SetLastError(err);
|
|
return MakeErrMsg(ErrMsg, path + ": SetFileTime: ");
|
|
}
|
|
|
|
// Best we can do with Unix permission bits is to interpret the owner
|
|
// writable bit.
|
|
if (si.mode & 0200) {
|
|
if (bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
|
|
if (!SetFileAttributes(path.c_str(),
|
|
bhfi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
|
|
return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
|
|
}
|
|
} else {
|
|
if (!(bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) {
|
|
if (!SetFileAttributes(path.c_str(),
|
|
bhfi.dwFileAttributes | FILE_ATTRIBUTE_READONLY))
|
|
return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
CopyFile(const sys::Path &Dest, const sys::Path &Src, std::string* ErrMsg) {
|
|
// Can't use CopyFile macro defined in Windows.h because it would mess up the
|
|
// above line. We use the expansion it would have in a non-UNICODE build.
|
|
if (!::CopyFileA(Src.c_str(), Dest.c_str(), false))
|
|
return MakeErrMsg(ErrMsg, "Can't copy '" + Src.str() +
|
|
"' to '" + Dest.str() + "': ");
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
Path::makeUnique(bool reuse_current, std::string* ErrMsg) {
|
|
bool Exists;
|
|
if (reuse_current && (fs::exists(path, Exists) || !Exists))
|
|
return false; // File doesn't exist already, just use it!
|
|
|
|
// Reserve space for -XXXXXX at the end.
|
|
char *FNBuffer = (char*) alloca(path.size()+8);
|
|
unsigned offset = path.size();
|
|
path.copy(FNBuffer, offset);
|
|
|
|
// Find a numeric suffix that isn't used by an existing file. Assume there
|
|
// won't be more than 1 million files with the same prefix. Probably a safe
|
|
// bet.
|
|
static int FCounter = -1;
|
|
if (FCounter < 0) {
|
|
// Give arbitrary initial seed.
|
|
// FIXME: We should use sys::fs::unique_file() in future.
|
|
LARGE_INTEGER cnt64;
|
|
DWORD x = GetCurrentProcessId();
|
|
x = (x << 16) | (x >> 16);
|
|
if (QueryPerformanceCounter(&cnt64)) // RDTSC
|
|
x ^= cnt64.HighPart ^ cnt64.LowPart;
|
|
FCounter = x % 1000000;
|
|
}
|
|
do {
|
|
sprintf(FNBuffer+offset, "-%06u", FCounter);
|
|
if (++FCounter > 999999)
|
|
FCounter = 0;
|
|
path = FNBuffer;
|
|
} while (!fs::exists(path, Exists) && Exists);
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
Path::createTemporaryFileOnDisk(bool reuse_current, std::string* ErrMsg) {
|
|
// Make this into a unique file name
|
|
makeUnique(reuse_current, ErrMsg);
|
|
|
|
// Now go and create it
|
|
HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
|
|
FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (h == INVALID_HANDLE_VALUE)
|
|
return MakeErrMsg(ErrMsg, path + ": can't create file");
|
|
|
|
CloseHandle(h);
|
|
return false;
|
|
}
|
|
|
|
/// MapInFilePages - Not yet implemented on win32.
|
|
const char *Path::MapInFilePages(int FD, size_t FileSize, off_t Offset) {
|
|
return 0;
|
|
}
|
|
|
|
/// MapInFilePages - Not yet implemented on win32.
|
|
void Path::UnMapFilePages(const char *Base, size_t FileSize) {
|
|
assert(0 && "NOT IMPLEMENTED");
|
|
}
|
|
|
|
}
|
|
}
|