| //===-- Path.cpp - Implement OS Path Concept ------------------------------===// | |
| // | |
| // The LLVM Compiler Infrastructure | |
| // | |
| // This file is distributed under the University of Illinois Open Source | |
| // License. See LICENSE.TXT for details. | |
| // | |
| //===----------------------------------------------------------------------===// | |
| // | |
| // This file implements the operating system Path API. | |
| // | |
| //===----------------------------------------------------------------------===// | |
| #include "llvm/Support/COFF.h" | |
| #include "llvm/Support/Endian.h" | |
| #include "llvm/Support/Errc.h" | |
| #include "llvm/Support/ErrorHandling.h" | |
| #include "llvm/Support/FileSystem.h" | |
| #include "llvm/Support/MachO.h" | |
| #include "llvm/Support/Path.h" | |
| #include "llvm/Support/Process.h" | |
| #include <errno.h> | |
| #include <cctype> | |
| #include <cstring> | |
| #if !defined(_MSC_VER) && !defined(__MINGW32__) | |
| #include <unistd.h> | |
| #else | |
| #include <io.h> | |
| #endif | |
| using namespace llvm; | |
| using namespace llvm::support::endian; | |
| namespace { | |
| using llvm::StringRef; | |
| using llvm::sys::path::is_separator; | |
| #ifdef LLVM_ON_WIN32 | |
| const char *separators = "\\/"; | |
| const char preferred_separator = '\\'; | |
| #else | |
| const char separators = '/'; | |
| const char preferred_separator = '/'; | |
| #endif | |
| StringRef find_first_component(StringRef path) { | |
| // Look for this first component in the following order. | |
| // * empty (in this case we return an empty string) | |
| // * either C: or {//,\\}net. | |
| // * {/,\} | |
| // * {file,directory}name | |
| if (path.empty()) | |
| return path; | |
| #ifdef LLVM_ON_WIN32 | |
| // C: | |
| if (path.size() >= 2 && std::isalpha(static_cast<unsigned char>(path[0])) && | |
| path[1] == ':') | |
| return path.substr(0, 2); | |
| #endif | |
| // //net | |
| if ((path.size() > 2) && | |
| is_separator(path[0]) && | |
| path[0] == path[1] && | |
| !is_separator(path[2])) { | |
| // Find the next directory separator. | |
| size_t end = path.find_first_of(separators, 2); | |
| return path.substr(0, end); | |
| } | |
| // {/,\} | |
| if (is_separator(path[0])) | |
| return path.substr(0, 1); | |
| // * {file,directory}name | |
| size_t end = path.find_first_of(separators); | |
| return path.substr(0, end); | |
| } | |
| size_t filename_pos(StringRef str) { | |
| if (str.size() == 2 && | |
| is_separator(str[0]) && | |
| str[0] == str[1]) | |
| return 0; | |
| if (str.size() > 0 && is_separator(str[str.size() - 1])) | |
| return str.size() - 1; | |
| size_t pos = str.find_last_of(separators, str.size() - 1); | |
| #ifdef LLVM_ON_WIN32 | |
| if (pos == StringRef::npos) | |
| pos = str.find_last_of(':', str.size() - 2); | |
| #endif | |
| if (pos == StringRef::npos || | |
| (pos == 1 && is_separator(str[0]))) | |
| return 0; | |
| return pos + 1; | |
| } | |
| size_t root_dir_start(StringRef str) { | |
| // case "c:/" | |
| #ifdef LLVM_ON_WIN32 | |
| if (str.size() > 2 && | |
| str[1] == ':' && | |
| is_separator(str[2])) | |
| return 2; | |
| #endif | |
| // case "//" | |
| if (str.size() == 2 && | |
| is_separator(str[0]) && | |
| str[0] == str[1]) | |
| return StringRef::npos; | |
| // case "//net" | |
| if (str.size() > 3 && | |
| is_separator(str[0]) && | |
| str[0] == str[1] && | |
| !is_separator(str[2])) { | |
| return str.find_first_of(separators, 2); | |
| } | |
| // case "/" | |
| if (str.size() > 0 && is_separator(str[0])) | |
| return 0; | |
| return StringRef::npos; | |
| } | |
| size_t parent_path_end(StringRef path) { | |
| size_t end_pos = filename_pos(path); | |
| bool filename_was_sep = path.size() > 0 && is_separator(path[end_pos]); | |
| // Skip separators except for root dir. | |
| size_t root_dir_pos = root_dir_start(path.substr(0, end_pos)); | |
| while(end_pos > 0 && | |
| (end_pos - 1) != root_dir_pos && | |
| is_separator(path[end_pos - 1])) | |
| --end_pos; | |
| if (end_pos == 1 && root_dir_pos == 0 && filename_was_sep) | |
| return StringRef::npos; | |
| return end_pos; | |
| } | |
| } // end unnamed namespace | |
| enum FSEntity { | |
| FS_Dir, | |
| FS_File, | |
| FS_Name | |
| }; | |
| static std::error_code createUniqueEntity(const Twine &Model, int &ResultFD, | |
| SmallVectorImpl<char> &ResultPath, | |
| bool MakeAbsolute, unsigned Mode, | |
| FSEntity Type) { | |
| SmallString<128> ModelStorage; | |
| Model.toVector(ModelStorage); | |
| if (MakeAbsolute) { | |
| // Make model absolute by prepending a temp directory if it's not already. | |
| if (!sys::path::is_absolute(Twine(ModelStorage))) { | |
| SmallString<128> TDir; | |
| sys::path::system_temp_directory(true, TDir); | |
| sys::path::append(TDir, Twine(ModelStorage)); | |
| ModelStorage.swap(TDir); | |
| } | |
| } | |
| // From here on, DO NOT modify model. It may be needed if the randomly chosen | |
| // path already exists. | |
| ResultPath = ModelStorage; | |
| // Null terminate. | |
| ResultPath.push_back(0); | |
| ResultPath.pop_back(); | |
| retry_random_path: | |
| // Replace '%' with random chars. | |
| for (unsigned i = 0, e = ModelStorage.size(); i != e; ++i) { | |
| if (ModelStorage[i] == '%') | |
| ResultPath[i] = "0123456789abcdef"[sys::Process::GetRandomNumber() & 15]; | |
| } | |
| // Try to open + create the file. | |
| switch (Type) { | |
| case FS_File: { | |
| if (std::error_code EC = | |
| sys::fs::openFileForWrite(Twine(ResultPath.begin()), ResultFD, | |
| sys::fs::F_RW | sys::fs::F_Excl, Mode)) { | |
| if (EC == errc::file_exists) | |
| goto retry_random_path; | |
| return EC; | |
| } | |
| return std::error_code(); | |
| } | |
| case FS_Name: { | |
| std::error_code EC = | |
| sys::fs::access(ResultPath.begin(), sys::fs::AccessMode::Exist); | |
| if (EC == errc::no_such_file_or_directory) | |
| return std::error_code(); | |
| if (EC) | |
| return EC; | |
| goto retry_random_path; | |
| } | |
| case FS_Dir: { | |
| if (std::error_code EC = | |
| sys::fs::create_directory(ResultPath.begin(), false)) { | |
| if (EC == errc::file_exists) | |
| goto retry_random_path; | |
| return EC; | |
| } | |
| return std::error_code(); | |
| } | |
| } | |
| llvm_unreachable("Invalid Type"); | |
| } | |
| namespace llvm { | |
| namespace sys { | |
| namespace path { | |
| const_iterator begin(StringRef path) { | |
| const_iterator i; | |
| i.Path = path; | |
| i.Component = find_first_component(path); | |
| i.Position = 0; | |
| return i; | |
| } | |
| const_iterator end(StringRef path) { | |
| const_iterator i; | |
| i.Path = path; | |
| i.Position = path.size(); | |
| return i; | |
| } | |
| const_iterator &const_iterator::operator++() { | |
| assert(Position < Path.size() && "Tried to increment past end!"); | |
| // Increment Position to past the current component | |
| Position += Component.size(); | |
| // Check for end. | |
| if (Position == Path.size()) { | |
| Component = StringRef(); | |
| return *this; | |
| } | |
| // Both POSIX and Windows treat paths that begin with exactly two separators | |
| // specially. | |
| bool was_net = Component.size() > 2 && | |
| is_separator(Component[0]) && | |
| Component[1] == Component[0] && | |
| !is_separator(Component[2]); | |
| // Handle separators. | |
| if (is_separator(Path[Position])) { | |
| // Root dir. | |
| if (was_net | |
| #ifdef LLVM_ON_WIN32 | |
| // c:/ | |
| || Component.endswith(":") | |
| #endif | |
| ) { | |
| Component = Path.substr(Position, 1); | |
| return *this; | |
| } | |
| // Skip extra separators. | |
| while (Position != Path.size() && | |
| is_separator(Path[Position])) { | |
| ++Position; | |
| } | |
| // Treat trailing '/' as a '.'. | |
| if (Position == Path.size()) { | |
| --Position; | |
| Component = "."; | |
| return *this; | |
| } | |
| } | |
| // Find next component. | |
| size_t end_pos = Path.find_first_of(separators, Position); | |
| Component = Path.slice(Position, end_pos); | |
| return *this; | |
| } | |
| bool const_iterator::operator==(const const_iterator &RHS) const { | |
| return Path.begin() == RHS.Path.begin() && Position == RHS.Position; | |
| } | |
| ptrdiff_t const_iterator::operator-(const const_iterator &RHS) const { | |
| return Position - RHS.Position; | |
| } | |
| reverse_iterator rbegin(StringRef Path) { | |
| reverse_iterator I; | |
| I.Path = Path; | |
| I.Position = Path.size(); | |
| return ++I; | |
| } | |
| reverse_iterator rend(StringRef Path) { | |
| reverse_iterator I; | |
| I.Path = Path; | |
| I.Component = Path.substr(0, 0); | |
| I.Position = 0; | |
| return I; | |
| } | |
| reverse_iterator &reverse_iterator::operator++() { | |
| // If we're at the end and the previous char was a '/', return '.' unless | |
| // we are the root path. | |
| size_t root_dir_pos = root_dir_start(Path); | |
| if (Position == Path.size() && | |
| Path.size() > root_dir_pos + 1 && | |
| is_separator(Path[Position - 1])) { | |
| --Position; | |
| Component = "."; | |
| return *this; | |
| } | |
| // Skip separators unless it's the root directory. | |
| size_t end_pos = Position; | |
| while(end_pos > 0 && | |
| (end_pos - 1) != root_dir_pos && | |
| is_separator(Path[end_pos - 1])) | |
| --end_pos; | |
| // Find next separator. | |
| size_t start_pos = filename_pos(Path.substr(0, end_pos)); | |
| Component = Path.slice(start_pos, end_pos); | |
| Position = start_pos; | |
| return *this; | |
| } | |
| bool reverse_iterator::operator==(const reverse_iterator &RHS) const { | |
| return Path.begin() == RHS.Path.begin() && Component == RHS.Component && | |
| Position == RHS.Position; | |
| } | |
| ptrdiff_t reverse_iterator::operator-(const reverse_iterator &RHS) const { | |
| return Position - RHS.Position; | |
| } | |
| StringRef root_path(StringRef path) { | |
| const_iterator b = begin(path), | |
| pos = b, | |
| e = end(path); | |
| if (b != e) { | |
| bool has_net = b->size() > 2 && is_separator((*b)[0]) && (*b)[1] == (*b)[0]; | |
| bool has_drive = | |
| #ifdef LLVM_ON_WIN32 | |
| b->endswith(":"); | |
| #else | |
| false; | |
| #endif | |
| if (has_net || has_drive) { | |
| if ((++pos != e) && is_separator((*pos)[0])) { | |
| // {C:/,//net/}, so get the first two components. | |
| return path.substr(0, b->size() + pos->size()); | |
| } else { | |
| // just {C:,//net}, return the first component. | |
| return *b; | |
| } | |
| } | |
| // POSIX style root directory. | |
| if (is_separator((*b)[0])) { | |
| return *b; | |
| } | |
| } | |
| return StringRef(); | |
| } | |
| StringRef root_name(StringRef path) { | |
| const_iterator b = begin(path), | |
| e = end(path); | |
| if (b != e) { | |
| bool has_net = b->size() > 2 && is_separator((*b)[0]) && (*b)[1] == (*b)[0]; | |
| bool has_drive = | |
| #ifdef LLVM_ON_WIN32 | |
| b->endswith(":"); | |
| #else | |
| false; | |
| #endif | |
| if (has_net || has_drive) { | |
| // just {C:,//net}, return the first component. | |
| return *b; | |
| } | |
| } | |
| // No path or no name. | |
| return StringRef(); | |
| } | |
| StringRef root_directory(StringRef path) { | |
| const_iterator b = begin(path), | |
| pos = b, | |
| e = end(path); | |
| if (b != e) { | |
| bool has_net = b->size() > 2 && is_separator((*b)[0]) && (*b)[1] == (*b)[0]; | |
| bool has_drive = | |
| #ifdef LLVM_ON_WIN32 | |
| b->endswith(":"); | |
| #else | |
| false; | |
| #endif | |
| if ((has_net || has_drive) && | |
| // {C:,//net}, skip to the next component. | |
| (++pos != e) && is_separator((*pos)[0])) { | |
| return *pos; | |
| } | |
| // POSIX style root directory. | |
| if (!has_net && is_separator((*b)[0])) { | |
| return *b; | |
| } | |
| } | |
| // No path or no root. | |
| return StringRef(); | |
| } | |
| StringRef relative_path(StringRef path) { | |
| StringRef root = root_path(path); | |
| return path.substr(root.size()); | |
| } | |
| void append(SmallVectorImpl<char> &path, const Twine &a, | |
| const Twine &b, | |
| const Twine &c, | |
| const Twine &d) { | |
| SmallString<32> a_storage; | |
| SmallString<32> b_storage; | |
| SmallString<32> c_storage; | |
| SmallString<32> d_storage; | |
| SmallVector<StringRef, 4> components; | |
| if (!a.isTriviallyEmpty()) components.push_back(a.toStringRef(a_storage)); | |
| if (!b.isTriviallyEmpty()) components.push_back(b.toStringRef(b_storage)); | |
| if (!c.isTriviallyEmpty()) components.push_back(c.toStringRef(c_storage)); | |
| if (!d.isTriviallyEmpty()) components.push_back(d.toStringRef(d_storage)); | |
| for (auto &component : components) { | |
| bool path_has_sep = !path.empty() && is_separator(path[path.size() - 1]); | |
| bool component_has_sep = !component.empty() && is_separator(component[0]); | |
| bool is_root_name = has_root_name(component); | |
| if (path_has_sep) { | |
| // Strip separators from beginning of component. | |
| size_t loc = component.find_first_not_of(separators); | |
| StringRef c = component.substr(loc); | |
| // Append it. | |
| path.append(c.begin(), c.end()); | |
| continue; | |
| } | |
| if (!component_has_sep && !(path.empty() || is_root_name)) { | |
| // Add a separator. | |
| path.push_back(preferred_separator); | |
| } | |
| path.append(component.begin(), component.end()); | |
| } | |
| } | |
| void append(SmallVectorImpl<char> &path, | |
| const_iterator begin, const_iterator end) { | |
| for (; begin != end; ++begin) | |
| path::append(path, *begin); | |
| } | |
| StringRef parent_path(StringRef path) { | |
| size_t end_pos = parent_path_end(path); | |
| if (end_pos == StringRef::npos) | |
| return StringRef(); | |
| else | |
| return path.substr(0, end_pos); | |
| } | |
| void remove_filename(SmallVectorImpl<char> &path) { | |
| size_t end_pos = parent_path_end(StringRef(path.begin(), path.size())); | |
| if (end_pos != StringRef::npos) | |
| path.set_size(end_pos); | |
| } | |
| void replace_extension(SmallVectorImpl<char> &path, const Twine &extension) { | |
| StringRef p(path.begin(), path.size()); | |
| SmallString<32> ext_storage; | |
| StringRef ext = extension.toStringRef(ext_storage); | |
| // Erase existing extension. | |
| size_t pos = p.find_last_of('.'); | |
| if (pos != StringRef::npos && pos >= filename_pos(p)) | |
| path.set_size(pos); | |
| // Append '.' if needed. | |
| if (ext.size() > 0 && ext[0] != '.') | |
| path.push_back('.'); | |
| // Append extension. | |
| path.append(ext.begin(), ext.end()); | |
| } | |
| void replace_path_prefix(SmallVectorImpl<char> &Path, | |
| const StringRef &OldPrefix, | |
| const StringRef &NewPrefix) { | |
| if (OldPrefix.empty() && NewPrefix.empty()) | |
| return; | |
| StringRef OrigPath(Path.begin(), Path.size()); | |
| if (!OrigPath.startswith(OldPrefix)) | |
| return; | |
| // If prefixes have the same size we can simply copy the new one over. | |
| if (OldPrefix.size() == NewPrefix.size()) { | |
| std::copy(NewPrefix.begin(), NewPrefix.end(), Path.begin()); | |
| return; | |
| } | |
| StringRef RelPath = OrigPath.substr(OldPrefix.size()); | |
| SmallString<256> NewPath; | |
| path::append(NewPath, NewPrefix); | |
| path::append(NewPath, RelPath); | |
| Path.swap(NewPath); | |
| } | |
| void native(const Twine &path, SmallVectorImpl<char> &result) { | |
| assert((!path.isSingleStringRef() || | |
| path.getSingleStringRef().data() != result.data()) && | |
| "path and result are not allowed to overlap!"); | |
| // Clear result. | |
| result.clear(); | |
| path.toVector(result); | |
| native(result); | |
| } | |
| void native(SmallVectorImpl<char> &Path) { | |
| #ifdef LLVM_ON_WIN32 | |
| std::replace(Path.begin(), Path.end(), '/', '\\'); | |
| #else | |
| for (auto PI = Path.begin(), PE = Path.end(); PI < PE; ++PI) { | |
| if (*PI == '\\') { | |
| auto PN = PI + 1; | |
| if (PN < PE && *PN == '\\') | |
| ++PI; // increment once, the for loop will move over the escaped slash | |
| else | |
| *PI = '/'; | |
| } | |
| } | |
| #endif | |
| } | |
| StringRef filename(StringRef path) { | |
| return *rbegin(path); | |
| } | |
| StringRef stem(StringRef path) { | |
| StringRef fname = filename(path); | |
| size_t pos = fname.find_last_of('.'); | |
| if (pos == StringRef::npos) | |
| return fname; | |
| else | |
| if ((fname.size() == 1 && fname == ".") || | |
| (fname.size() == 2 && fname == "..")) | |
| return fname; | |
| else | |
| return fname.substr(0, pos); | |
| } | |
| StringRef extension(StringRef path) { | |
| StringRef fname = filename(path); | |
| size_t pos = fname.find_last_of('.'); | |
| if (pos == StringRef::npos) | |
| return StringRef(); | |
| else | |
| if ((fname.size() == 1 && fname == ".") || | |
| (fname.size() == 2 && fname == "..")) | |
| return StringRef(); | |
| else | |
| return fname.substr(pos); | |
| } | |
| bool is_separator(char value) { | |
| switch(value) { | |
| #ifdef LLVM_ON_WIN32 | |
| case '\\': // fall through | |
| #endif | |
| case '/': return true; | |
| default: return false; | |
| } | |
| } | |
| static const char preferred_separator_string[] = { preferred_separator, '\0' }; | |
| StringRef get_separator() { | |
| return preferred_separator_string; | |
| } | |
| bool has_root_name(const Twine &path) { | |
| SmallString<128> path_storage; | |
| StringRef p = path.toStringRef(path_storage); | |
| return !root_name(p).empty(); | |
| } | |
| bool has_root_directory(const Twine &path) { | |
| SmallString<128> path_storage; | |
| StringRef p = path.toStringRef(path_storage); | |
| return !root_directory(p).empty(); | |
| } | |
| bool has_root_path(const Twine &path) { | |
| SmallString<128> path_storage; | |
| StringRef p = path.toStringRef(path_storage); | |
| return !root_path(p).empty(); | |
| } | |
| bool has_relative_path(const Twine &path) { | |
| SmallString<128> path_storage; | |
| StringRef p = path.toStringRef(path_storage); | |
| return !relative_path(p).empty(); | |
| } | |
| bool has_filename(const Twine &path) { | |
| SmallString<128> path_storage; | |
| StringRef p = path.toStringRef(path_storage); | |
| return !filename(p).empty(); | |
| } | |
| bool has_parent_path(const Twine &path) { | |
| SmallString<128> path_storage; | |
| StringRef p = path.toStringRef(path_storage); | |
| return !parent_path(p).empty(); | |
| } | |
| bool has_stem(const Twine &path) { | |
| SmallString<128> path_storage; | |
| StringRef p = path.toStringRef(path_storage); | |
| return !stem(p).empty(); | |
| } | |
| bool has_extension(const Twine &path) { | |
| SmallString<128> path_storage; | |
| StringRef p = path.toStringRef(path_storage); | |
| return !extension(p).empty(); | |
| } | |
| bool is_absolute(const Twine &path) { | |
| SmallString<128> path_storage; | |
| StringRef p = path.toStringRef(path_storage); | |
| bool rootDir = has_root_directory(p), | |
| #ifdef LLVM_ON_WIN32 | |
| rootName = has_root_name(p); | |
| #else | |
| rootName = true; | |
| #endif | |
| return rootDir && rootName; | |
| } | |
| bool is_relative(const Twine &path) { return !is_absolute(path); } | |
| StringRef remove_leading_dotslash(StringRef Path) { | |
| // Remove leading "./" (or ".//" or "././" etc.) | |
| while (Path.size() > 2 && Path[0] == '.' && is_separator(Path[1])) { | |
| Path = Path.substr(2); | |
| while (Path.size() > 0 && is_separator(Path[0])) | |
| Path = Path.substr(1); | |
| } | |
| return Path; | |
| } | |
| static SmallString<256> remove_dots(StringRef path, bool remove_dot_dot) { | |
| SmallVector<StringRef, 16> components; | |
| // Skip the root path, then look for traversal in the components. | |
| StringRef rel = path::relative_path(path); | |
| for (StringRef C : llvm::make_range(path::begin(rel), path::end(rel))) { | |
| if (C == ".") | |
| continue; | |
| // Leading ".." will remain in the path unless it's at the root. | |
| if (remove_dot_dot && C == "..") { | |
| if (!components.empty() && components.back() != "..") { | |
| components.pop_back(); | |
| continue; | |
| } | |
| if (path::is_absolute(path)) | |
| continue; | |
| } | |
| components.push_back(C); | |
| } | |
| SmallString<256> buffer = path::root_path(path); | |
| for (StringRef C : components) | |
| path::append(buffer, C); | |
| return buffer; | |
| } | |
| bool remove_dots(SmallVectorImpl<char> &path, bool remove_dot_dot) { | |
| StringRef p(path.data(), path.size()); | |
| SmallString<256> result = remove_dots(p, remove_dot_dot); | |
| if (result == path) | |
| return false; | |
| path.swap(result); | |
| return true; | |
| } | |
| } // end namespace path | |
| namespace fs { | |
| std::error_code getUniqueID(const Twine Path, UniqueID &Result) { | |
| file_status Status; | |
| std::error_code EC = status(Path, Status); | |
| if (EC) | |
| return EC; | |
| Result = Status.getUniqueID(); | |
| return std::error_code(); | |
| } | |
| std::error_code createUniqueFile(const Twine &Model, int &ResultFd, | |
| SmallVectorImpl<char> &ResultPath, | |
| unsigned Mode) { | |
| return createUniqueEntity(Model, ResultFd, ResultPath, false, Mode, FS_File); | |
| } | |
| std::error_code createUniqueFile(const Twine &Model, | |
| SmallVectorImpl<char> &ResultPath) { | |
| int Dummy; | |
| return createUniqueEntity(Model, Dummy, ResultPath, false, 0, FS_Name); | |
| } | |
| static std::error_code | |
| createTemporaryFile(const Twine &Model, int &ResultFD, | |
| llvm::SmallVectorImpl<char> &ResultPath, FSEntity Type) { | |
| SmallString<128> Storage; | |
| StringRef P = Model.toNullTerminatedStringRef(Storage); | |
| assert(P.find_first_of(separators) == StringRef::npos && | |
| "Model must be a simple filename."); | |
| // Use P.begin() so that createUniqueEntity doesn't need to recreate Storage. | |
| return createUniqueEntity(P.begin(), ResultFD, ResultPath, | |
| true, owner_read | owner_write, Type); | |
| } | |
| static std::error_code | |
| createTemporaryFile(const Twine &Prefix, StringRef Suffix, int &ResultFD, | |
| llvm::SmallVectorImpl<char> &ResultPath, FSEntity Type) { | |
| const char *Middle = Suffix.empty() ? "-%%%%%%" : "-%%%%%%."; | |
| return createTemporaryFile(Prefix + Middle + Suffix, ResultFD, ResultPath, | |
| Type); | |
| } | |
| std::error_code createTemporaryFile(const Twine &Prefix, StringRef Suffix, | |
| int &ResultFD, | |
| SmallVectorImpl<char> &ResultPath) { | |
| return createTemporaryFile(Prefix, Suffix, ResultFD, ResultPath, FS_File); | |
| } | |
| std::error_code createTemporaryFile(const Twine &Prefix, StringRef Suffix, | |
| SmallVectorImpl<char> &ResultPath) { | |
| int Dummy; | |
| return createTemporaryFile(Prefix, Suffix, Dummy, ResultPath, FS_Name); | |
| } | |
| // This is a mkdtemp with a different pattern. We use createUniqueEntity mostly | |
| // for consistency. We should try using mkdtemp. | |
| std::error_code createUniqueDirectory(const Twine &Prefix, | |
| SmallVectorImpl<char> &ResultPath) { | |
| int Dummy; | |
| return createUniqueEntity(Prefix + "-%%%%%%", Dummy, ResultPath, | |
| true, 0, FS_Dir); | |
| } | |
| static std::error_code make_absolute(const Twine ¤t_directory, | |
| SmallVectorImpl<char> &path, | |
| bool use_current_directory) { | |
| StringRef p(path.data(), path.size()); | |
| bool rootDirectory = path::has_root_directory(p), | |
| #ifdef LLVM_ON_WIN32 | |
| rootName = path::has_root_name(p); | |
| #else | |
| rootName = true; | |
| #endif | |
| // Already absolute. | |
| if (rootName && rootDirectory) | |
| return std::error_code(); | |
| // All of the following conditions will need the current directory. | |
| SmallString<128> current_dir; | |
| if (use_current_directory) | |
| current_directory.toVector(current_dir); | |
| else if (std::error_code ec = current_path(current_dir)) | |
| return ec; | |
| // Relative path. Prepend the current directory. | |
| if (!rootName && !rootDirectory) { | |
| // Append path to the current directory. | |
| path::append(current_dir, p); | |
| // Set path to the result. | |
| path.swap(current_dir); | |
| return std::error_code(); | |
| } | |
| if (!rootName && rootDirectory) { | |
| StringRef cdrn = path::root_name(current_dir); | |
| SmallString<128> curDirRootName(cdrn.begin(), cdrn.end()); | |
| path::append(curDirRootName, p); | |
| // Set path to the result. | |
| path.swap(curDirRootName); | |
| return std::error_code(); | |
| } | |
| if (rootName && !rootDirectory) { | |
| StringRef pRootName = path::root_name(p); | |
| StringRef bRootDirectory = path::root_directory(current_dir); | |
| StringRef bRelativePath = path::relative_path(current_dir); | |
| StringRef pRelativePath = path::relative_path(p); | |
| SmallString<128> res; | |
| path::append(res, pRootName, bRootDirectory, bRelativePath, pRelativePath); | |
| path.swap(res); | |
| return std::error_code(); | |
| } | |
| llvm_unreachable("All rootName and rootDirectory combinations should have " | |
| "occurred above!"); | |
| } | |
| std::error_code make_absolute(const Twine ¤t_directory, | |
| SmallVectorImpl<char> &path) { | |
| return make_absolute(current_directory, path, true); | |
| } | |
| std::error_code make_absolute(SmallVectorImpl<char> &path) { | |
| return make_absolute(Twine(), path, false); | |
| } | |
| std::error_code create_directories(const Twine &Path, bool IgnoreExisting, | |
| perms Perms) { | |
| SmallString<128> PathStorage; | |
| StringRef P = Path.toStringRef(PathStorage); | |
| // Be optimistic and try to create the directory | |
| std::error_code EC = create_directory(P, IgnoreExisting, Perms); | |
| // If we succeeded, or had any error other than the parent not existing, just | |
| // return it. | |
| if (EC != errc::no_such_file_or_directory) | |
| return EC; | |
| // We failed because of a no_such_file_or_directory, try to create the | |
| // parent. | |
| StringRef Parent = path::parent_path(P); | |
| if (Parent.empty()) | |
| return EC; | |
| if ((EC = create_directories(Parent, IgnoreExisting, Perms))) | |
| return EC; | |
| return create_directory(P, IgnoreExisting, Perms); | |
| } | |
| std::error_code copy_file(const Twine &From, const Twine &To) { | |
| int ReadFD, WriteFD; | |
| if (std::error_code EC = openFileForRead(From, ReadFD)) | |
| return EC; | |
| if (std::error_code EC = openFileForWrite(To, WriteFD, F_None)) { | |
| close(ReadFD); | |
| return EC; | |
| } | |
| const size_t BufSize = 4096; | |
| char *Buf = new char[BufSize]; | |
| int BytesRead = 0, BytesWritten = 0; | |
| for (;;) { | |
| BytesRead = read(ReadFD, Buf, BufSize); | |
| if (BytesRead <= 0) | |
| break; | |
| while (BytesRead) { | |
| BytesWritten = write(WriteFD, Buf, BytesRead); | |
| if (BytesWritten < 0) | |
| break; | |
| BytesRead -= BytesWritten; | |
| } | |
| if (BytesWritten < 0) | |
| break; | |
| } | |
| close(ReadFD); | |
| close(WriteFD); | |
| delete[] Buf; | |
| if (BytesRead < 0 || BytesWritten < 0) | |
| return std::error_code(errno, std::generic_category()); | |
| return std::error_code(); | |
| } | |
| bool exists(file_status status) { | |
| return status_known(status) && status.type() != file_type::file_not_found; | |
| } | |
| bool status_known(file_status s) { | |
| return s.type() != file_type::status_error; | |
| } | |
| bool is_directory(file_status status) { | |
| return status.type() == file_type::directory_file; | |
| } | |
| std::error_code is_directory(const Twine &path, bool &result) { | |
| file_status st; | |
| if (std::error_code ec = status(path, st)) | |
| return ec; | |
| result = is_directory(st); | |
| return std::error_code(); | |
| } | |
| bool is_regular_file(file_status status) { | |
| return status.type() == file_type::regular_file; | |
| } | |
| std::error_code is_regular_file(const Twine &path, bool &result) { | |
| file_status st; | |
| if (std::error_code ec = status(path, st)) | |
| return ec; | |
| result = is_regular_file(st); | |
| return std::error_code(); | |
| } | |
| bool is_other(file_status status) { | |
| return exists(status) && | |
| !is_regular_file(status) && | |
| !is_directory(status); | |
| } | |
| std::error_code is_other(const Twine &Path, bool &Result) { | |
| file_status FileStatus; | |
| if (std::error_code EC = status(Path, FileStatus)) | |
| return EC; | |
| Result = is_other(FileStatus); | |
| return std::error_code(); | |
| } | |
| void directory_entry::replace_filename(const Twine &filename, file_status st) { | |
| SmallString<128> path = path::parent_path(Path); | |
| path::append(path, filename); | |
| Path = path.str(); | |
| Status = st; | |
| } | |
| template <size_t N> | |
| static bool startswith(StringRef Magic, const char (&S)[N]) { | |
| return Magic.startswith(StringRef(S, N - 1)); | |
| } | |
| /// @brief Identify the magic in magic. | |
| file_magic identify_magic(StringRef Magic) { | |
| if (Magic.size() < 4) | |
| return file_magic::unknown; | |
| switch ((unsigned char)Magic[0]) { | |
| case 0x00: { | |
| // COFF bigobj, CL.exe's LTO object file, or short import library file | |
| if (startswith(Magic, "\0\0\xFF\xFF")) { | |
| size_t MinSize = offsetof(COFF::BigObjHeader, UUID) + sizeof(COFF::BigObjMagic); | |
| if (Magic.size() < MinSize) | |
| return file_magic::coff_import_library; | |
| const char *Start = Magic.data() + offsetof(COFF::BigObjHeader, UUID); | |
| if (memcmp(Start, COFF::BigObjMagic, sizeof(COFF::BigObjMagic)) == 0) | |
| return file_magic::coff_object; | |
| if (memcmp(Start, COFF::ClGlObjMagic, sizeof(COFF::BigObjMagic)) == 0) | |
| return file_magic::coff_cl_gl_object; | |
| return file_magic::coff_import_library; | |
| } | |
| // Windows resource file | |
| if (startswith(Magic, "\0\0\0\0\x20\0\0\0\xFF")) | |
| return file_magic::windows_resource; | |
| // 0x0000 = COFF unknown machine type | |
| if (Magic[1] == 0) | |
| return file_magic::coff_object; | |
| if (startswith(Magic, "\0asm")) | |
| return file_magic::wasm_object; | |
| break; | |
| } | |
| case 0xDE: // 0x0B17C0DE = BC wraper | |
| if (startswith(Magic, "\xDE\xC0\x17\x0B")) | |
| return file_magic::bitcode; | |
| break; | |
| case 'B': | |
| if (startswith(Magic, "BC\xC0\xDE")) | |
| return file_magic::bitcode; | |
| break; | |
| case '!': | |
| if (startswith(Magic, "!<arch>\n") || startswith(Magic, "!<thin>\n")) | |
| return file_magic::archive; | |
| break; | |
| case '\177': | |
| if (startswith(Magic, "\177ELF") && Magic.size() >= 18) { | |
| bool Data2MSB = Magic[5] == 2; | |
| unsigned high = Data2MSB ? 16 : 17; | |
| unsigned low = Data2MSB ? 17 : 16; | |
| if (Magic[high] == 0) { | |
| switch (Magic[low]) { | |
| default: return file_magic::elf; | |
| case 1: return file_magic::elf_relocatable; | |
| case 2: return file_magic::elf_executable; | |
| case 3: return file_magic::elf_shared_object; | |
| case 4: return file_magic::elf_core; | |
| } | |
| } | |
| // It's still some type of ELF file. | |
| return file_magic::elf; | |
| } | |
| break; | |
| case 0xCA: | |
| if (startswith(Magic, "\xCA\xFE\xBA\xBE") || | |
| startswith(Magic, "\xCA\xFE\xBA\xBF")) { | |
| // This is complicated by an overlap with Java class files. | |
| // See the Mach-O section in /usr/share/file/magic for details. | |
| if (Magic.size() >= 8 && Magic[7] < 43) | |
| return file_magic::macho_universal_binary; | |
| } | |
| break; | |
| // The two magic numbers for mach-o are: | |
| // 0xfeedface - 32-bit mach-o | |
| // 0xfeedfacf - 64-bit mach-o | |
| case 0xFE: | |
| case 0xCE: | |
| case 0xCF: { | |
| uint16_t type = 0; | |
| if (startswith(Magic, "\xFE\xED\xFA\xCE") || | |
| startswith(Magic, "\xFE\xED\xFA\xCF")) { | |
| /* Native endian */ | |
| size_t MinSize; | |
| if (Magic[3] == char(0xCE)) | |
| MinSize = sizeof(MachO::mach_header); | |
| else | |
| MinSize = sizeof(MachO::mach_header_64); | |
| if (Magic.size() >= MinSize) | |
| type = Magic[12] << 24 | Magic[13] << 12 | Magic[14] << 8 | Magic[15]; | |
| } else if (startswith(Magic, "\xCE\xFA\xED\xFE") || | |
| startswith(Magic, "\xCF\xFA\xED\xFE")) { | |
| /* Reverse endian */ | |
| size_t MinSize; | |
| if (Magic[0] == char(0xCE)) | |
| MinSize = sizeof(MachO::mach_header); | |
| else | |
| MinSize = sizeof(MachO::mach_header_64); | |
| if (Magic.size() >= MinSize) | |
| type = Magic[15] << 24 | Magic[14] << 12 |Magic[13] << 8 | Magic[12]; | |
| } | |
| switch (type) { | |
| default: break; | |
| case 1: return file_magic::macho_object; | |
| case 2: return file_magic::macho_executable; | |
| case 3: return file_magic::macho_fixed_virtual_memory_shared_lib; | |
| case 4: return file_magic::macho_core; | |
| case 5: return file_magic::macho_preload_executable; | |
| case 6: return file_magic::macho_dynamically_linked_shared_lib; | |
| case 7: return file_magic::macho_dynamic_linker; | |
| case 8: return file_magic::macho_bundle; | |
| case 9: return file_magic::macho_dynamically_linked_shared_lib_stub; | |
| case 10: return file_magic::macho_dsym_companion; | |
| case 11: return file_magic::macho_kext_bundle; | |
| } | |
| break; | |
| } | |
| case 0xF0: // PowerPC Windows | |
| case 0x83: // Alpha 32-bit | |
| case 0x84: // Alpha 64-bit | |
| case 0x66: // MPS R4000 Windows | |
| case 0x50: // mc68K | |
| case 0x4c: // 80386 Windows | |
| case 0xc4: // ARMNT Windows | |
| if (Magic[1] == 0x01) | |
| return file_magic::coff_object; | |
| case 0x90: // PA-RISC Windows | |
| case 0x68: // mc68K Windows | |
| if (Magic[1] == 0x02) | |
| return file_magic::coff_object; | |
| break; | |
| case 'M': // Possible MS-DOS stub on Windows PE file | |
| if (startswith(Magic, "MZ")) { | |
| uint32_t off = read32le(Magic.data() + 0x3c); | |
| // PE/COFF file, either EXE or DLL. | |
| if (off < Magic.size() && | |
| memcmp(Magic.data()+off, COFF::PEMagic, sizeof(COFF::PEMagic)) == 0) | |
| return file_magic::pecoff_executable; | |
| } | |
| break; | |
| case 0x64: // x86-64 Windows. | |
| if (Magic[1] == char(0x86)) | |
| return file_magic::coff_object; | |
| break; | |
| default: | |
| break; | |
| } | |
| return file_magic::unknown; | |
| } | |
| std::error_code identify_magic(const Twine &Path, file_magic &Result) { | |
| int FD; | |
| if (std::error_code EC = openFileForRead(Path, FD)) | |
| return EC; | |
| char Buffer[32]; | |
| int Length = read(FD, Buffer, sizeof(Buffer)); | |
| if (close(FD) != 0 || Length < 0) | |
| return std::error_code(errno, std::generic_category()); | |
| Result = identify_magic(StringRef(Buffer, Length)); | |
| return std::error_code(); | |
| } | |
| std::error_code directory_entry::status(file_status &result) const { | |
| return fs::status(Path, result); | |
| } | |
| } // end namespace fs | |
| } // end namespace sys | |
| } // end namespace llvm | |
| // Include the truly platform-specific parts. | |
| #if defined(LLVM_ON_UNIX) | |
| #include "Unix/Path.inc" | |
| #endif | |
| #if defined(LLVM_ON_WIN32) | |
| #include "Windows/Path.inc" | |
| #endif | |
| namespace llvm { | |
| namespace sys { | |
| namespace path { | |
| bool user_cache_directory(SmallVectorImpl<char> &Result, const Twine &Path1, | |
| const Twine &Path2, const Twine &Path3) { | |
| if (getUserCacheDir(Result)) { | |
| append(Result, Path1, Path2, Path3); | |
| return true; | |
| } | |
| return false; | |
| } | |
| } // end namespace path | |
| } // end namsspace sys | |
| } // end namespace llvm |