15713a05bSZachary Turner //===-- FileSpec.cpp --------------------------------------------*- C++ -*-===// 25713a05bSZachary Turner // 35713a05bSZachary Turner // The LLVM Compiler Infrastructure 45713a05bSZachary Turner // 55713a05bSZachary Turner // This file is distributed under the University of Illinois Open Source 65713a05bSZachary Turner // License. See LICENSE.TXT for details. 75713a05bSZachary Turner // 85713a05bSZachary Turner //===----------------------------------------------------------------------===// 95713a05bSZachary Turner 105713a05bSZachary Turner #include "lldb/Utility/FileSpec.h" 115713a05bSZachary Turner #include "lldb/Utility/RegularExpression.h" 125713a05bSZachary Turner #include "lldb/Utility/Stream.h" 135713a05bSZachary Turner #include "lldb/Utility/TildeExpressionResolver.h" 145713a05bSZachary Turner 154479ac15SZachary Turner #include "llvm/ADT/SmallString.h" // for SmallString 164479ac15SZachary Turner #include "llvm/ADT/SmallVector.h" // for SmallVectorTemplat... 175713a05bSZachary Turner #include "llvm/ADT/StringRef.h" 184479ac15SZachary Turner #include "llvm/ADT/Triple.h" // for Triple 194479ac15SZachary Turner #include "llvm/ADT/Twine.h" // for Twine 204479ac15SZachary Turner #include "llvm/Support/ErrorOr.h" // for ErrorOr 215713a05bSZachary Turner #include "llvm/Support/FileSystem.h" 225713a05bSZachary Turner #include "llvm/Support/Program.h" 234479ac15SZachary Turner #include "llvm/Support/raw_ostream.h" // for raw_ostream, fs 244479ac15SZachary Turner 254479ac15SZachary Turner #include <algorithm> // for replace, min, unique 264479ac15SZachary Turner #include <system_error> // for error_code 274479ac15SZachary Turner #include <vector> // for vector 284479ac15SZachary Turner 294479ac15SZachary Turner #include <assert.h> // for assert 304479ac15SZachary Turner #include <stdio.h> // for size_t, NULL, snpr... 314479ac15SZachary Turner #include <string.h> // for strcmp 325713a05bSZachary Turner 335713a05bSZachary Turner using namespace lldb; 345713a05bSZachary Turner using namespace lldb_private; 355713a05bSZachary Turner 365713a05bSZachary Turner namespace { 375713a05bSZachary Turner 38*2cb7cf8eSPavel Labath static constexpr FileSpec::Style GetNativeStyle() { 39b1cb0b79SNico Weber #if defined(_WIN32) 40*2cb7cf8eSPavel Labath return FileSpec::Style::windows; 415713a05bSZachary Turner #else 42*2cb7cf8eSPavel Labath return FileSpec::Style::posix; 435713a05bSZachary Turner #endif 445713a05bSZachary Turner } 455713a05bSZachary Turner 46*2cb7cf8eSPavel Labath bool PathStyleIsPosix(FileSpec::Style style) { 47*2cb7cf8eSPavel Labath return (style == FileSpec::Style::posix || 48*2cb7cf8eSPavel Labath (style == FileSpec::Style::native && 49*2cb7cf8eSPavel Labath GetNativeStyle() == FileSpec::Style::posix)); 505713a05bSZachary Turner } 515713a05bSZachary Turner 52*2cb7cf8eSPavel Labath const char *GetPathSeparators(FileSpec::Style style) { 53*2cb7cf8eSPavel Labath return PathStyleIsPosix(style) ? "/" : "\\/"; 545713a05bSZachary Turner } 555713a05bSZachary Turner 56*2cb7cf8eSPavel Labath char GetPreferredPathSeparator(FileSpec::Style style) { 57*2cb7cf8eSPavel Labath return GetPathSeparators(style)[0]; 585713a05bSZachary Turner } 595713a05bSZachary Turner 60*2cb7cf8eSPavel Labath bool IsPathSeparator(char value, FileSpec::Style style) { 61*2cb7cf8eSPavel Labath return value == '/' || (!PathStyleIsPosix(style) && value == '\\'); 625713a05bSZachary Turner } 635713a05bSZachary Turner 64*2cb7cf8eSPavel Labath void Denormalize(llvm::SmallVectorImpl<char> &path, FileSpec::Style style) { 65*2cb7cf8eSPavel Labath if (PathStyleIsPosix(style)) 665713a05bSZachary Turner return; 675713a05bSZachary Turner 685713a05bSZachary Turner std::replace(path.begin(), path.end(), '/', '\\'); 695713a05bSZachary Turner } 705713a05bSZachary Turner } // end anonymous namespace 715713a05bSZachary Turner 725713a05bSZachary Turner void FileSpec::Resolve(llvm::SmallVectorImpl<char> &path) { 735713a05bSZachary Turner if (path.empty()) 745713a05bSZachary Turner return; 755713a05bSZachary Turner 765713a05bSZachary Turner llvm::SmallString<32> Source(path.begin(), path.end()); 775713a05bSZachary Turner StandardTildeExpressionResolver Resolver; 785713a05bSZachary Turner Resolver.ResolveFullPath(Source, path); 795713a05bSZachary Turner 805713a05bSZachary Turner // Save a copy of the original path that's passed in 815713a05bSZachary Turner llvm::SmallString<128> original_path(path.begin(), path.end()); 825713a05bSZachary Turner 835713a05bSZachary Turner llvm::sys::fs::make_absolute(path); 845713a05bSZachary Turner if (!llvm::sys::fs::exists(path)) { 855713a05bSZachary Turner path.clear(); 865713a05bSZachary Turner path.append(original_path.begin(), original_path.end()); 875713a05bSZachary Turner } 885713a05bSZachary Turner } 895713a05bSZachary Turner 90*2cb7cf8eSPavel Labath FileSpec::FileSpec() : m_style(GetNativeStyle()) {} 915713a05bSZachary Turner 925713a05bSZachary Turner //------------------------------------------------------------------ 9305097246SAdrian Prantl // Default constructor that can take an optional full path to a file on disk. 945713a05bSZachary Turner //------------------------------------------------------------------ 95*2cb7cf8eSPavel Labath FileSpec::FileSpec(llvm::StringRef path, bool resolve_path, Style style) 96*2cb7cf8eSPavel Labath : m_style(style) { 97*2cb7cf8eSPavel Labath SetFile(path, resolve_path, style); 985713a05bSZachary Turner } 995713a05bSZachary Turner 1005713a05bSZachary Turner FileSpec::FileSpec(llvm::StringRef path, bool resolve_path, 1015713a05bSZachary Turner const llvm::Triple &Triple) 1025713a05bSZachary Turner : FileSpec{path, resolve_path, 103*2cb7cf8eSPavel Labath Triple.isOSWindows() ? Style::windows : Style::posix} {} 1045713a05bSZachary Turner 1055713a05bSZachary Turner //------------------------------------------------------------------ 1065713a05bSZachary Turner // Copy constructor 1075713a05bSZachary Turner //------------------------------------------------------------------ 1085713a05bSZachary Turner FileSpec::FileSpec(const FileSpec &rhs) 1095713a05bSZachary Turner : m_directory(rhs.m_directory), m_filename(rhs.m_filename), 110*2cb7cf8eSPavel Labath m_is_resolved(rhs.m_is_resolved), m_style(rhs.m_style) {} 1115713a05bSZachary Turner 1125713a05bSZachary Turner //------------------------------------------------------------------ 1135713a05bSZachary Turner // Copy constructor 1145713a05bSZachary Turner //------------------------------------------------------------------ 1155713a05bSZachary Turner FileSpec::FileSpec(const FileSpec *rhs) : m_directory(), m_filename() { 1165713a05bSZachary Turner if (rhs) 1175713a05bSZachary Turner *this = *rhs; 1185713a05bSZachary Turner } 1195713a05bSZachary Turner 1205713a05bSZachary Turner //------------------------------------------------------------------ 1215713a05bSZachary Turner // Virtual destructor in case anyone inherits from this class. 1225713a05bSZachary Turner //------------------------------------------------------------------ 1235713a05bSZachary Turner FileSpec::~FileSpec() {} 1245713a05bSZachary Turner 125776cd7adSGreg Clayton namespace { 126776cd7adSGreg Clayton //------------------------------------------------------------------ 127776cd7adSGreg Clayton /// Safely get a character at the specified index. 128776cd7adSGreg Clayton /// 129776cd7adSGreg Clayton /// @param[in] path 130776cd7adSGreg Clayton /// A full, partial, or relative path to a file. 131776cd7adSGreg Clayton /// 132776cd7adSGreg Clayton /// @param[in] i 133776cd7adSGreg Clayton /// An index into path which may or may not be valid. 134776cd7adSGreg Clayton /// 135776cd7adSGreg Clayton /// @return 136776cd7adSGreg Clayton /// The character at index \a i if the index is valid, or 0 if 137776cd7adSGreg Clayton /// the index is not valid. 138776cd7adSGreg Clayton //------------------------------------------------------------------ 139776cd7adSGreg Clayton inline char safeCharAtIndex(const llvm::StringRef &path, size_t i) { 140776cd7adSGreg Clayton if (i < path.size()) 141776cd7adSGreg Clayton return path[i]; 142776cd7adSGreg Clayton return 0; 143776cd7adSGreg Clayton } 144776cd7adSGreg Clayton 145776cd7adSGreg Clayton //------------------------------------------------------------------ 146776cd7adSGreg Clayton /// Check if a path needs to be normalized. 147776cd7adSGreg Clayton /// 148776cd7adSGreg Clayton /// Check if a path needs to be normalized. We currently consider a 149776cd7adSGreg Clayton /// path to need normalization if any of the following are true 150776cd7adSGreg Clayton /// - path contains "/./" 151776cd7adSGreg Clayton /// - path contains "/../" 152776cd7adSGreg Clayton /// - path contains "//" 153776cd7adSGreg Clayton /// - path ends with "/" 154776cd7adSGreg Clayton /// Paths that start with "./" or with "../" are not considered to 155776cd7adSGreg Clayton /// need normalization since we aren't trying to resolve the path, 156776cd7adSGreg Clayton /// we are just trying to remove redundant things from the path. 157776cd7adSGreg Clayton /// 158776cd7adSGreg Clayton /// @param[in] path 159776cd7adSGreg Clayton /// A full, partial, or relative path to a file. 160776cd7adSGreg Clayton /// 161776cd7adSGreg Clayton /// @return 162776cd7adSGreg Clayton /// Returns \b true if the path needs to be normalized. 163776cd7adSGreg Clayton //------------------------------------------------------------------ 164410c5acfSPavel Labath bool needsNormalization(const llvm::StringRef &path) { 16527a0e10aSGreg Clayton if (path.empty()) 16627a0e10aSGreg Clayton return false; 16727a0e10aSGreg Clayton // We strip off leading "." values so these paths need to be normalized 16827a0e10aSGreg Clayton if (path[0] == '.') 16927a0e10aSGreg Clayton return true; 17027a0e10aSGreg Clayton for (auto i = path.find_first_of("\\/"); i != llvm::StringRef::npos; 17127a0e10aSGreg Clayton i = path.find_first_of("\\/", i + 1)) { 172776cd7adSGreg Clayton const auto next = safeCharAtIndex(path, i+1); 173776cd7adSGreg Clayton switch (next) { 174776cd7adSGreg Clayton case 0: 175776cd7adSGreg Clayton // path separator char at the end of the string which should be 176776cd7adSGreg Clayton // stripped unless it is the one and only character 177776cd7adSGreg Clayton return i > 0; 178776cd7adSGreg Clayton case '/': 179776cd7adSGreg Clayton case '\\': 180776cd7adSGreg Clayton // two path separator chars in the middle of a path needs to be 181776cd7adSGreg Clayton // normalized 18227a0e10aSGreg Clayton if (i > 0) 183776cd7adSGreg Clayton return true; 184776cd7adSGreg Clayton ++i; 185776cd7adSGreg Clayton break; 186776cd7adSGreg Clayton 187776cd7adSGreg Clayton case '.': { 188776cd7adSGreg Clayton const auto next_next = safeCharAtIndex(path, i+2); 189776cd7adSGreg Clayton switch (next_next) { 190776cd7adSGreg Clayton default: break; 191776cd7adSGreg Clayton case 0: return true; // ends with "/." 192776cd7adSGreg Clayton case '/': 193776cd7adSGreg Clayton case '\\': 194776cd7adSGreg Clayton return true; // contains "/./" 195776cd7adSGreg Clayton case '.': { 196776cd7adSGreg Clayton const auto next_next_next = safeCharAtIndex(path, i+3); 197776cd7adSGreg Clayton switch (next_next_next) { 198776cd7adSGreg Clayton default: break; 199776cd7adSGreg Clayton case 0: return true; // ends with "/.." 200776cd7adSGreg Clayton case '/': 201776cd7adSGreg Clayton case '\\': 202776cd7adSGreg Clayton return true; // contains "/../" 203776cd7adSGreg Clayton } 204776cd7adSGreg Clayton break; 205776cd7adSGreg Clayton } 206776cd7adSGreg Clayton } 207776cd7adSGreg Clayton } 208776cd7adSGreg Clayton break; 209776cd7adSGreg Clayton 210776cd7adSGreg Clayton default: 211776cd7adSGreg Clayton break; 212776cd7adSGreg Clayton } 213776cd7adSGreg Clayton } 214776cd7adSGreg Clayton return false; 215776cd7adSGreg Clayton } 216776cd7adSGreg Clayton 217776cd7adSGreg Clayton 218776cd7adSGreg Clayton } 2195713a05bSZachary Turner //------------------------------------------------------------------ 2205713a05bSZachary Turner // Assignment operator. 2215713a05bSZachary Turner //------------------------------------------------------------------ 2225713a05bSZachary Turner const FileSpec &FileSpec::operator=(const FileSpec &rhs) { 2235713a05bSZachary Turner if (this != &rhs) { 2245713a05bSZachary Turner m_directory = rhs.m_directory; 2255713a05bSZachary Turner m_filename = rhs.m_filename; 2265713a05bSZachary Turner m_is_resolved = rhs.m_is_resolved; 227*2cb7cf8eSPavel Labath m_style = rhs.m_style; 2285713a05bSZachary Turner } 2295713a05bSZachary Turner return *this; 2305713a05bSZachary Turner } 2315713a05bSZachary Turner 2325713a05bSZachary Turner //------------------------------------------------------------------ 23305097246SAdrian Prantl // Update the contents of this object with a new path. The path will be split 23405097246SAdrian Prantl // up into a directory and filename and stored as uniqued string values for 23505097246SAdrian Prantl // quick comparison and efficient memory usage. 2365713a05bSZachary Turner //------------------------------------------------------------------ 237*2cb7cf8eSPavel Labath void FileSpec::SetFile(llvm::StringRef pathname, bool resolve, Style style) { 2385713a05bSZachary Turner m_filename.Clear(); 2395713a05bSZachary Turner m_directory.Clear(); 2405713a05bSZachary Turner m_is_resolved = false; 241*2cb7cf8eSPavel Labath m_style = (style == Style::native) ? GetNativeStyle() : style; 2425713a05bSZachary Turner 2435713a05bSZachary Turner if (pathname.empty()) 2445713a05bSZachary Turner return; 2455713a05bSZachary Turner 2465713a05bSZachary Turner llvm::SmallString<64> resolved(pathname); 2475713a05bSZachary Turner 2485713a05bSZachary Turner if (resolve) { 2495713a05bSZachary Turner FileSpec::Resolve(resolved); 2505713a05bSZachary Turner m_is_resolved = true; 2515713a05bSZachary Turner } 2525713a05bSZachary Turner 253776cd7adSGreg Clayton // Normalize the path by removing ".", ".." and other redundant components. 254410c5acfSPavel Labath if (needsNormalization(resolved)) 255*2cb7cf8eSPavel Labath llvm::sys::path::remove_dots(resolved, true, m_style); 256776cd7adSGreg Clayton 257776cd7adSGreg Clayton // Normalize back slashes to forward slashes 258*2cb7cf8eSPavel Labath if (m_style == Style::windows) 259776cd7adSGreg Clayton std::replace(resolved.begin(), resolved.end(), '\\', '/'); 2605713a05bSZachary Turner 261*2cb7cf8eSPavel Labath m_filename.SetString(llvm::sys::path::filename(resolved, m_style)); 262*2cb7cf8eSPavel Labath llvm::StringRef dir = llvm::sys::path::parent_path(resolved, m_style); 263e7306b10SPavel Labath if (!dir.empty()) 264e7306b10SPavel Labath m_directory.SetString(dir); 2655713a05bSZachary Turner } 2665713a05bSZachary Turner 2675713a05bSZachary Turner void FileSpec::SetFile(llvm::StringRef path, bool resolve, 2685713a05bSZachary Turner const llvm::Triple &Triple) { 2695713a05bSZachary Turner return SetFile(path, resolve, 270*2cb7cf8eSPavel Labath Triple.isOSWindows() ? Style::windows : Style::posix); 2715713a05bSZachary Turner } 2725713a05bSZachary Turner 2735713a05bSZachary Turner //---------------------------------------------------------------------- 27405097246SAdrian Prantl // Convert to pointer operator. This allows code to check any FileSpec objects 27505097246SAdrian Prantl // to see if they contain anything valid using code such as: 2765713a05bSZachary Turner // 2775713a05bSZachary Turner // if (file_spec) 2785713a05bSZachary Turner // {} 2795713a05bSZachary Turner //---------------------------------------------------------------------- 2805713a05bSZachary Turner FileSpec::operator bool() const { return m_filename || m_directory; } 2815713a05bSZachary Turner 2825713a05bSZachary Turner //---------------------------------------------------------------------- 28305097246SAdrian Prantl // Logical NOT operator. This allows code to check any FileSpec objects to see 28405097246SAdrian Prantl // if they are invalid using code such as: 2855713a05bSZachary Turner // 2865713a05bSZachary Turner // if (!file_spec) 2875713a05bSZachary Turner // {} 2885713a05bSZachary Turner //---------------------------------------------------------------------- 2895713a05bSZachary Turner bool FileSpec::operator!() const { return !m_directory && !m_filename; } 2905713a05bSZachary Turner 2915713a05bSZachary Turner bool FileSpec::DirectoryEquals(const FileSpec &rhs) const { 2925713a05bSZachary Turner const bool case_sensitive = IsCaseSensitive() || rhs.IsCaseSensitive(); 2935713a05bSZachary Turner return ConstString::Equals(m_directory, rhs.m_directory, case_sensitive); 2945713a05bSZachary Turner } 2955713a05bSZachary Turner 2965713a05bSZachary Turner bool FileSpec::FileEquals(const FileSpec &rhs) const { 2975713a05bSZachary Turner const bool case_sensitive = IsCaseSensitive() || rhs.IsCaseSensitive(); 2985713a05bSZachary Turner return ConstString::Equals(m_filename, rhs.m_filename, case_sensitive); 2995713a05bSZachary Turner } 3005713a05bSZachary Turner 3015713a05bSZachary Turner //------------------------------------------------------------------ 3025713a05bSZachary Turner // Equal to operator 3035713a05bSZachary Turner //------------------------------------------------------------------ 3045713a05bSZachary Turner bool FileSpec::operator==(const FileSpec &rhs) const { 3055713a05bSZachary Turner if (!FileEquals(rhs)) 3065713a05bSZachary Turner return false; 3075713a05bSZachary Turner if (DirectoryEquals(rhs)) 3085713a05bSZachary Turner return true; 3095713a05bSZachary Turner 3105713a05bSZachary Turner // TODO: determine if we want to keep this code in here. 31105097246SAdrian Prantl // The code below was added to handle a case where we were trying to set a 31205097246SAdrian Prantl // file and line breakpoint and one path was resolved, and the other not and 31305097246SAdrian Prantl // the directory was in a mount point that resolved to a more complete path: 31405097246SAdrian Prantl // "/tmp/a.c" == "/private/tmp/a.c". I might end up pulling this out... 3155713a05bSZachary Turner if (IsResolved() && rhs.IsResolved()) { 3165713a05bSZachary Turner // Both paths are resolved, no need to look further... 3175713a05bSZachary Turner return false; 3185713a05bSZachary Turner } 3195713a05bSZachary Turner 3205713a05bSZachary Turner FileSpec resolved_lhs(*this); 3215713a05bSZachary Turner 3225713a05bSZachary Turner // If "this" isn't resolved, resolve it 3235713a05bSZachary Turner if (!IsResolved()) { 3245713a05bSZachary Turner if (resolved_lhs.ResolvePath()) { 3255713a05bSZachary Turner // This path wasn't resolved but now it is. Check if the resolved 32605097246SAdrian Prantl // directory is the same as our unresolved directory, and if so, we can 32705097246SAdrian Prantl // mark this object as resolved to avoid more future resolves 3285713a05bSZachary Turner m_is_resolved = (m_directory == resolved_lhs.m_directory); 3295713a05bSZachary Turner } else 3305713a05bSZachary Turner return false; 3315713a05bSZachary Turner } 3325713a05bSZachary Turner 3335713a05bSZachary Turner FileSpec resolved_rhs(rhs); 3345713a05bSZachary Turner if (!rhs.IsResolved()) { 3355713a05bSZachary Turner if (resolved_rhs.ResolvePath()) { 3365713a05bSZachary Turner // rhs's path wasn't resolved but now it is. Check if the resolved 33705097246SAdrian Prantl // directory is the same as rhs's unresolved directory, and if so, we can 33805097246SAdrian Prantl // mark this object as resolved to avoid more future resolves 3395713a05bSZachary Turner rhs.m_is_resolved = (rhs.m_directory == resolved_rhs.m_directory); 3405713a05bSZachary Turner } else 3415713a05bSZachary Turner return false; 3425713a05bSZachary Turner } 3435713a05bSZachary Turner 34405097246SAdrian Prantl // If we reach this point in the code we were able to resolve both paths and 34505097246SAdrian Prantl // since we only resolve the paths if the basenames are equal, then we can 34605097246SAdrian Prantl // just check if both directories are equal... 3475713a05bSZachary Turner return DirectoryEquals(rhs); 3485713a05bSZachary Turner } 3495713a05bSZachary Turner 3505713a05bSZachary Turner //------------------------------------------------------------------ 3515713a05bSZachary Turner // Not equal to operator 3525713a05bSZachary Turner //------------------------------------------------------------------ 3535713a05bSZachary Turner bool FileSpec::operator!=(const FileSpec &rhs) const { return !(*this == rhs); } 3545713a05bSZachary Turner 3555713a05bSZachary Turner //------------------------------------------------------------------ 3565713a05bSZachary Turner // Less than operator 3575713a05bSZachary Turner //------------------------------------------------------------------ 3585713a05bSZachary Turner bool FileSpec::operator<(const FileSpec &rhs) const { 3595713a05bSZachary Turner return FileSpec::Compare(*this, rhs, true) < 0; 3605713a05bSZachary Turner } 3615713a05bSZachary Turner 3625713a05bSZachary Turner //------------------------------------------------------------------ 3635713a05bSZachary Turner // Dump a FileSpec object to a stream 3645713a05bSZachary Turner //------------------------------------------------------------------ 3655713a05bSZachary Turner Stream &lldb_private::operator<<(Stream &s, const FileSpec &f) { 3665713a05bSZachary Turner f.Dump(&s); 3675713a05bSZachary Turner return s; 3685713a05bSZachary Turner } 3695713a05bSZachary Turner 3705713a05bSZachary Turner //------------------------------------------------------------------ 37105097246SAdrian Prantl // Clear this object by releasing both the directory and filename string values 37205097246SAdrian Prantl // and making them both the empty string. 3735713a05bSZachary Turner //------------------------------------------------------------------ 3745713a05bSZachary Turner void FileSpec::Clear() { 3755713a05bSZachary Turner m_directory.Clear(); 3765713a05bSZachary Turner m_filename.Clear(); 3775713a05bSZachary Turner } 3785713a05bSZachary Turner 3795713a05bSZachary Turner //------------------------------------------------------------------ 38005097246SAdrian Prantl // Compare two FileSpec objects. If "full" is true, then both the directory and 38105097246SAdrian Prantl // the filename must match. If "full" is false, then the directory names for 38205097246SAdrian Prantl // "a" and "b" are only compared if they are both non-empty. This allows a 38305097246SAdrian Prantl // FileSpec object to only contain a filename and it can match FileSpec objects 38405097246SAdrian Prantl // that have matching filenames with different paths. 3855713a05bSZachary Turner // 38605097246SAdrian Prantl // Return -1 if the "a" is less than "b", 0 if "a" is equal to "b" and "1" if 38705097246SAdrian Prantl // "a" is greater than "b". 3885713a05bSZachary Turner //------------------------------------------------------------------ 3895713a05bSZachary Turner int FileSpec::Compare(const FileSpec &a, const FileSpec &b, bool full) { 3905713a05bSZachary Turner int result = 0; 3915713a05bSZachary Turner 3925713a05bSZachary Turner // case sensitivity of compare 3935713a05bSZachary Turner const bool case_sensitive = a.IsCaseSensitive() || b.IsCaseSensitive(); 3945713a05bSZachary Turner 3955713a05bSZachary Turner // If full is true, then we must compare both the directory and filename. 3965713a05bSZachary Turner 39705097246SAdrian Prantl // If full is false, then if either directory is empty, then we match on the 39805097246SAdrian Prantl // basename only, and if both directories have valid values, we still do a 39905097246SAdrian Prantl // full compare. This allows for matching when we just have a filename in one 40005097246SAdrian Prantl // of the FileSpec objects. 4015713a05bSZachary Turner 4025713a05bSZachary Turner if (full || (a.m_directory && b.m_directory)) { 4035713a05bSZachary Turner result = ConstString::Compare(a.m_directory, b.m_directory, case_sensitive); 4045713a05bSZachary Turner if (result) 4055713a05bSZachary Turner return result; 4065713a05bSZachary Turner } 4075713a05bSZachary Turner return ConstString::Compare(a.m_filename, b.m_filename, case_sensitive); 4085713a05bSZachary Turner } 4095713a05bSZachary Turner 410776cd7adSGreg Clayton bool FileSpec::Equal(const FileSpec &a, const FileSpec &b, bool full) { 41197e4f472SJim Ingham 4125713a05bSZachary Turner // case sensitivity of equality test 4135713a05bSZachary Turner const bool case_sensitive = a.IsCaseSensitive() || b.IsCaseSensitive(); 4145713a05bSZachary Turner 415776cd7adSGreg Clayton const bool filenames_equal = ConstString::Equals(a.m_filename, 41697e4f472SJim Ingham b.m_filename, 41797e4f472SJim Ingham case_sensitive); 41897e4f472SJim Ingham 419776cd7adSGreg Clayton if (!filenames_equal) 42097e4f472SJim Ingham return false; 42197e4f472SJim Ingham 4225713a05bSZachary Turner if (!full && (a.GetDirectory().IsEmpty() || b.GetDirectory().IsEmpty())) 42397e4f472SJim Ingham return filenames_equal; 4245713a05bSZachary Turner 4255713a05bSZachary Turner return a == b; 4265713a05bSZachary Turner } 4275713a05bSZachary Turner 4285713a05bSZachary Turner //------------------------------------------------------------------ 42905097246SAdrian Prantl // Dump the object to the supplied stream. If the object contains a valid 43005097246SAdrian Prantl // directory name, it will be displayed followed by a directory delimiter, and 43105097246SAdrian Prantl // the filename. 4325713a05bSZachary Turner //------------------------------------------------------------------ 4335713a05bSZachary Turner void FileSpec::Dump(Stream *s) const { 4345713a05bSZachary Turner if (s) { 4355713a05bSZachary Turner std::string path{GetPath(true)}; 4365713a05bSZachary Turner s->PutCString(path); 437*2cb7cf8eSPavel Labath char path_separator = GetPreferredPathSeparator(m_style); 4385713a05bSZachary Turner if (!m_filename && !path.empty() && path.back() != path_separator) 4395713a05bSZachary Turner s->PutChar(path_separator); 4405713a05bSZachary Turner } 4415713a05bSZachary Turner } 4425713a05bSZachary Turner 4435713a05bSZachary Turner //------------------------------------------------------------------ 4445713a05bSZachary Turner // Returns true if the file exists. 4455713a05bSZachary Turner //------------------------------------------------------------------ 4465713a05bSZachary Turner bool FileSpec::Exists() const { return llvm::sys::fs::exists(GetPath()); } 4475713a05bSZachary Turner 4485713a05bSZachary Turner bool FileSpec::Readable() const { 4495713a05bSZachary Turner return GetPermissions() & llvm::sys::fs::perms::all_read; 4505713a05bSZachary Turner } 4515713a05bSZachary Turner 4525713a05bSZachary Turner bool FileSpec::ResolveExecutableLocation() { 4535713a05bSZachary Turner // CLEANUP: Use StringRef for string handling. 4545713a05bSZachary Turner if (!m_directory) { 4555713a05bSZachary Turner const char *file_cstr = m_filename.GetCString(); 4565713a05bSZachary Turner if (file_cstr) { 4575713a05bSZachary Turner const std::string file_str(file_cstr); 4585713a05bSZachary Turner llvm::ErrorOr<std::string> error_or_path = 4595713a05bSZachary Turner llvm::sys::findProgramByName(file_str); 4605713a05bSZachary Turner if (!error_or_path) 4615713a05bSZachary Turner return false; 4625713a05bSZachary Turner std::string path = error_or_path.get(); 4635713a05bSZachary Turner llvm::StringRef dir_ref = llvm::sys::path::parent_path(path); 4645713a05bSZachary Turner if (!dir_ref.empty()) { 4655713a05bSZachary Turner // FindProgramByName returns "." if it can't find the file. 4665713a05bSZachary Turner if (strcmp(".", dir_ref.data()) == 0) 4675713a05bSZachary Turner return false; 4685713a05bSZachary Turner 4695713a05bSZachary Turner m_directory.SetCString(dir_ref.data()); 4705713a05bSZachary Turner if (Exists()) 4715713a05bSZachary Turner return true; 4725713a05bSZachary Turner else { 4735713a05bSZachary Turner // If FindProgramByName found the file, it returns the directory + 47405097246SAdrian Prantl // filename in its return results. We need to separate them. 4755713a05bSZachary Turner FileSpec tmp_file(dir_ref.data(), false); 4765713a05bSZachary Turner if (tmp_file.Exists()) { 4775713a05bSZachary Turner m_directory = tmp_file.m_directory; 4785713a05bSZachary Turner return true; 4795713a05bSZachary Turner } 4805713a05bSZachary Turner } 4815713a05bSZachary Turner } 4825713a05bSZachary Turner } 4835713a05bSZachary Turner } 4845713a05bSZachary Turner 4855713a05bSZachary Turner return false; 4865713a05bSZachary Turner } 4875713a05bSZachary Turner 4885713a05bSZachary Turner bool FileSpec::ResolvePath() { 4895713a05bSZachary Turner if (m_is_resolved) 4905713a05bSZachary Turner return true; // We have already resolved this path 4915713a05bSZachary Turner 4925713a05bSZachary Turner // SetFile(...) will set m_is_resolved correctly if it can resolve the path 4935713a05bSZachary Turner SetFile(GetPath(false), true); 4945713a05bSZachary Turner return m_is_resolved; 4955713a05bSZachary Turner } 4965713a05bSZachary Turner 4975713a05bSZachary Turner uint64_t FileSpec::GetByteSize() const { 4985713a05bSZachary Turner uint64_t Size = 0; 4995713a05bSZachary Turner if (llvm::sys::fs::file_size(GetPath(), Size)) 5005713a05bSZachary Turner return 0; 5015713a05bSZachary Turner return Size; 5025713a05bSZachary Turner } 5035713a05bSZachary Turner 504*2cb7cf8eSPavel Labath FileSpec::Style FileSpec::GetPathStyle() const { return m_style; } 5055713a05bSZachary Turner 5065713a05bSZachary Turner uint32_t FileSpec::GetPermissions() const { 5075713a05bSZachary Turner namespace fs = llvm::sys::fs; 5085713a05bSZachary Turner fs::file_status st; 5095713a05bSZachary Turner if (fs::status(GetPath(), st, false)) 5105713a05bSZachary Turner return fs::perms::perms_not_known; 5115713a05bSZachary Turner 5125713a05bSZachary Turner return st.permissions(); 5135713a05bSZachary Turner } 5145713a05bSZachary Turner 5155713a05bSZachary Turner //------------------------------------------------------------------ 5165713a05bSZachary Turner // Directory string get accessor. 5175713a05bSZachary Turner //------------------------------------------------------------------ 5185713a05bSZachary Turner ConstString &FileSpec::GetDirectory() { return m_directory; } 5195713a05bSZachary Turner 5205713a05bSZachary Turner //------------------------------------------------------------------ 5215713a05bSZachary Turner // Directory string const get accessor. 5225713a05bSZachary Turner //------------------------------------------------------------------ 5235713a05bSZachary Turner const ConstString &FileSpec::GetDirectory() const { return m_directory; } 5245713a05bSZachary Turner 5255713a05bSZachary Turner //------------------------------------------------------------------ 5265713a05bSZachary Turner // Filename string get accessor. 5275713a05bSZachary Turner //------------------------------------------------------------------ 5285713a05bSZachary Turner ConstString &FileSpec::GetFilename() { return m_filename; } 5295713a05bSZachary Turner 5305713a05bSZachary Turner //------------------------------------------------------------------ 5315713a05bSZachary Turner // Filename string const get accessor. 5325713a05bSZachary Turner //------------------------------------------------------------------ 5335713a05bSZachary Turner const ConstString &FileSpec::GetFilename() const { return m_filename; } 5345713a05bSZachary Turner 5355713a05bSZachary Turner //------------------------------------------------------------------ 53605097246SAdrian Prantl // Extract the directory and path into a fixed buffer. This is needed as the 53705097246SAdrian Prantl // directory and path are stored in separate string values. 5385713a05bSZachary Turner //------------------------------------------------------------------ 5395713a05bSZachary Turner size_t FileSpec::GetPath(char *path, size_t path_max_len, 5405713a05bSZachary Turner bool denormalize) const { 5415713a05bSZachary Turner if (!path) 5425713a05bSZachary Turner return 0; 5435713a05bSZachary Turner 5445713a05bSZachary Turner std::string result = GetPath(denormalize); 5455713a05bSZachary Turner ::snprintf(path, path_max_len, "%s", result.c_str()); 5465713a05bSZachary Turner return std::min(path_max_len - 1, result.length()); 5475713a05bSZachary Turner } 5485713a05bSZachary Turner 5495713a05bSZachary Turner std::string FileSpec::GetPath(bool denormalize) const { 5505713a05bSZachary Turner llvm::SmallString<64> result; 5515713a05bSZachary Turner GetPath(result, denormalize); 5525713a05bSZachary Turner return std::string(result.begin(), result.end()); 5535713a05bSZachary Turner } 5545713a05bSZachary Turner 5555713a05bSZachary Turner const char *FileSpec::GetCString(bool denormalize) const { 5565713a05bSZachary Turner return ConstString{GetPath(denormalize)}.AsCString(NULL); 5575713a05bSZachary Turner } 5585713a05bSZachary Turner 5595713a05bSZachary Turner void FileSpec::GetPath(llvm::SmallVectorImpl<char> &path, 5605713a05bSZachary Turner bool denormalize) const { 5615713a05bSZachary Turner path.append(m_directory.GetStringRef().begin(), 5625713a05bSZachary Turner m_directory.GetStringRef().end()); 563776cd7adSGreg Clayton // Since the path was normalized and all paths use '/' when stored in these 564776cd7adSGreg Clayton // objects, we don't need to look for the actual syntax specific path 565776cd7adSGreg Clayton // separator, we just look for and insert '/'. 566776cd7adSGreg Clayton if (m_directory && m_filename && m_directory.GetStringRef().back() != '/' && 567776cd7adSGreg Clayton m_filename.GetStringRef().back() != '/') 568776cd7adSGreg Clayton path.insert(path.end(), '/'); 5695713a05bSZachary Turner path.append(m_filename.GetStringRef().begin(), 5705713a05bSZachary Turner m_filename.GetStringRef().end()); 5715713a05bSZachary Turner if (denormalize && !path.empty()) 572*2cb7cf8eSPavel Labath Denormalize(path, m_style); 5735713a05bSZachary Turner } 5745713a05bSZachary Turner 5755713a05bSZachary Turner ConstString FileSpec::GetFileNameExtension() const { 5765713a05bSZachary Turner if (m_filename) { 5775713a05bSZachary Turner const char *filename = m_filename.GetCString(); 5785713a05bSZachary Turner const char *dot_pos = strrchr(filename, '.'); 5795713a05bSZachary Turner if (dot_pos && dot_pos[1] != '\0') 5805713a05bSZachary Turner return ConstString(dot_pos + 1); 5815713a05bSZachary Turner } 5825713a05bSZachary Turner return ConstString(); 5835713a05bSZachary Turner } 5845713a05bSZachary Turner 5855713a05bSZachary Turner ConstString FileSpec::GetFileNameStrippingExtension() const { 5865713a05bSZachary Turner const char *filename = m_filename.GetCString(); 5875713a05bSZachary Turner if (filename == NULL) 5885713a05bSZachary Turner return ConstString(); 5895713a05bSZachary Turner 5905713a05bSZachary Turner const char *dot_pos = strrchr(filename, '.'); 5915713a05bSZachary Turner if (dot_pos == NULL) 5925713a05bSZachary Turner return m_filename; 5935713a05bSZachary Turner 5945713a05bSZachary Turner return ConstString(filename, dot_pos - filename); 5955713a05bSZachary Turner } 5965713a05bSZachary Turner 5975713a05bSZachary Turner //------------------------------------------------------------------ 59805097246SAdrian Prantl // Return the size in bytes that this object takes in memory. This returns the 59905097246SAdrian Prantl // size in bytes of this object, not any shared string values it may refer to. 6005713a05bSZachary Turner //------------------------------------------------------------------ 6015713a05bSZachary Turner size_t FileSpec::MemorySize() const { 6025713a05bSZachary Turner return m_filename.MemorySize() + m_directory.MemorySize(); 6035713a05bSZachary Turner } 6045713a05bSZachary Turner 6055713a05bSZachary Turner void FileSpec::EnumerateDirectory(llvm::StringRef dir_path, 6065713a05bSZachary Turner bool find_directories, bool find_files, 6075713a05bSZachary Turner bool find_other, 6085713a05bSZachary Turner EnumerateDirectoryCallbackType callback, 6095713a05bSZachary Turner void *callback_baton) { 6105713a05bSZachary Turner namespace fs = llvm::sys::fs; 6115713a05bSZachary Turner std::error_code EC; 6125713a05bSZachary Turner fs::recursive_directory_iterator Iter(dir_path, EC); 6135713a05bSZachary Turner fs::recursive_directory_iterator End; 6145713a05bSZachary Turner for (; Iter != End && !EC; Iter.increment(EC)) { 6155713a05bSZachary Turner const auto &Item = *Iter; 6160dfdb447SPeter Collingbourne llvm::ErrorOr<fs::basic_file_status> Status = Item.status(); 6170dfdb447SPeter Collingbourne if (!Status) 6185713a05bSZachary Turner break; 6190dfdb447SPeter Collingbourne if (!find_files && fs::is_regular_file(*Status)) 6205713a05bSZachary Turner continue; 6210dfdb447SPeter Collingbourne if (!find_directories && fs::is_directory(*Status)) 6225713a05bSZachary Turner continue; 6230dfdb447SPeter Collingbourne if (!find_other && fs::is_other(*Status)) 6245713a05bSZachary Turner continue; 6255713a05bSZachary Turner 6265713a05bSZachary Turner FileSpec Spec(Item.path(), false); 6270dfdb447SPeter Collingbourne auto Result = callback(callback_baton, Status->type(), Spec); 6285713a05bSZachary Turner if (Result == eEnumerateDirectoryResultQuit) 6295713a05bSZachary Turner return; 6305713a05bSZachary Turner if (Result == eEnumerateDirectoryResultNext) { 6315713a05bSZachary Turner // Default behavior is to recurse. Opt out if the callback doesn't want 6325713a05bSZachary Turner // this behavior. 6335713a05bSZachary Turner Iter.no_push(); 6345713a05bSZachary Turner } 6355713a05bSZachary Turner } 6365713a05bSZachary Turner } 6375713a05bSZachary Turner 6385713a05bSZachary Turner FileSpec 6395713a05bSZachary Turner FileSpec::CopyByAppendingPathComponent(llvm::StringRef component) const { 6405713a05bSZachary Turner FileSpec ret = *this; 6415713a05bSZachary Turner ret.AppendPathComponent(component); 6425713a05bSZachary Turner return ret; 6435713a05bSZachary Turner } 6445713a05bSZachary Turner 6455713a05bSZachary Turner FileSpec FileSpec::CopyByRemovingLastPathComponent() const { 6465713a05bSZachary Turner // CLEANUP: Use StringRef for string handling. 6475713a05bSZachary Turner const bool resolve = false; 6485713a05bSZachary Turner if (m_filename.IsEmpty() && m_directory.IsEmpty()) 6495713a05bSZachary Turner return FileSpec("", resolve); 6505713a05bSZachary Turner if (m_directory.IsEmpty()) 6515713a05bSZachary Turner return FileSpec("", resolve); 6525713a05bSZachary Turner if (m_filename.IsEmpty()) { 6535713a05bSZachary Turner const char *dir_cstr = m_directory.GetCString(); 6545713a05bSZachary Turner const char *last_slash_ptr = ::strrchr(dir_cstr, '/'); 6555713a05bSZachary Turner 6565713a05bSZachary Turner // check for obvious cases before doing the full thing 6575713a05bSZachary Turner if (!last_slash_ptr) 6585713a05bSZachary Turner return FileSpec("", resolve); 6595713a05bSZachary Turner if (last_slash_ptr == dir_cstr) 6605713a05bSZachary Turner return FileSpec("/", resolve); 6615713a05bSZachary Turner 6625713a05bSZachary Turner size_t last_slash_pos = last_slash_ptr - dir_cstr + 1; 6635713a05bSZachary Turner ConstString new_path(dir_cstr, last_slash_pos); 6645713a05bSZachary Turner return FileSpec(new_path.GetCString(), resolve); 6655713a05bSZachary Turner } else 6665713a05bSZachary Turner return FileSpec(m_directory.GetCString(), resolve); 6675713a05bSZachary Turner } 6685713a05bSZachary Turner 6695713a05bSZachary Turner ConstString FileSpec::GetLastPathComponent() const { 6705713a05bSZachary Turner // CLEANUP: Use StringRef for string handling. 6715713a05bSZachary Turner if (m_filename) 6725713a05bSZachary Turner return m_filename; 6735713a05bSZachary Turner if (m_directory) { 6745713a05bSZachary Turner const char *dir_cstr = m_directory.GetCString(); 6755713a05bSZachary Turner const char *last_slash_ptr = ::strrchr(dir_cstr, '/'); 6765713a05bSZachary Turner if (last_slash_ptr == NULL) 6775713a05bSZachary Turner return m_directory; 6785713a05bSZachary Turner if (last_slash_ptr == dir_cstr) { 6795713a05bSZachary Turner if (last_slash_ptr[1] == 0) 6805713a05bSZachary Turner return ConstString(last_slash_ptr); 6815713a05bSZachary Turner else 6825713a05bSZachary Turner return ConstString(last_slash_ptr + 1); 6835713a05bSZachary Turner } 6845713a05bSZachary Turner if (last_slash_ptr[1] != 0) 6855713a05bSZachary Turner return ConstString(last_slash_ptr + 1); 6865713a05bSZachary Turner const char *penultimate_slash_ptr = last_slash_ptr; 6875713a05bSZachary Turner while (*penultimate_slash_ptr) { 6885713a05bSZachary Turner --penultimate_slash_ptr; 6895713a05bSZachary Turner if (penultimate_slash_ptr == dir_cstr) 6905713a05bSZachary Turner break; 6915713a05bSZachary Turner if (*penultimate_slash_ptr == '/') 6925713a05bSZachary Turner break; 6935713a05bSZachary Turner } 6945713a05bSZachary Turner ConstString result(penultimate_slash_ptr + 1, 6955713a05bSZachary Turner last_slash_ptr - penultimate_slash_ptr); 6965713a05bSZachary Turner return result; 6975713a05bSZachary Turner } 6985713a05bSZachary Turner return ConstString(); 6995713a05bSZachary Turner } 7005713a05bSZachary Turner 7015713a05bSZachary Turner static std::string 702*2cb7cf8eSPavel Labath join_path_components(FileSpec::Style style, 7035713a05bSZachary Turner const std::vector<llvm::StringRef> components) { 7045713a05bSZachary Turner std::string result; 7055713a05bSZachary Turner for (size_t i = 0; i < components.size(); ++i) { 7065713a05bSZachary Turner if (components[i].empty()) 7075713a05bSZachary Turner continue; 7085713a05bSZachary Turner result += components[i]; 7095713a05bSZachary Turner if (i != components.size() - 1 && 710*2cb7cf8eSPavel Labath !IsPathSeparator(components[i].back(), style)) 711*2cb7cf8eSPavel Labath result += GetPreferredPathSeparator(style); 7125713a05bSZachary Turner } 7135713a05bSZachary Turner 7145713a05bSZachary Turner return result; 7155713a05bSZachary Turner } 7165713a05bSZachary Turner 7175713a05bSZachary Turner void FileSpec::PrependPathComponent(llvm::StringRef component) { 7185713a05bSZachary Turner if (component.empty()) 7195713a05bSZachary Turner return; 7205713a05bSZachary Turner 7215713a05bSZachary Turner const bool resolve = false; 7225713a05bSZachary Turner if (m_filename.IsEmpty() && m_directory.IsEmpty()) { 7235713a05bSZachary Turner SetFile(component, resolve); 7245713a05bSZachary Turner return; 7255713a05bSZachary Turner } 7265713a05bSZachary Turner 7275713a05bSZachary Turner std::string result = 728*2cb7cf8eSPavel Labath join_path_components(m_style, {component, m_directory.GetStringRef(), 7295713a05bSZachary Turner m_filename.GetStringRef()}); 730*2cb7cf8eSPavel Labath SetFile(result, resolve, m_style); 7315713a05bSZachary Turner } 7325713a05bSZachary Turner 7335713a05bSZachary Turner void FileSpec::PrependPathComponent(const FileSpec &new_path) { 7345713a05bSZachary Turner return PrependPathComponent(new_path.GetPath(false)); 7355713a05bSZachary Turner } 7365713a05bSZachary Turner 7375713a05bSZachary Turner void FileSpec::AppendPathComponent(llvm::StringRef component) { 7385713a05bSZachary Turner if (component.empty()) 7395713a05bSZachary Turner return; 7405713a05bSZachary Turner 7415713a05bSZachary Turner component = component.drop_while( 742*2cb7cf8eSPavel Labath [this](char c) { return IsPathSeparator(c, m_style); }); 7435713a05bSZachary Turner 7445713a05bSZachary Turner std::string result = 745*2cb7cf8eSPavel Labath join_path_components(m_style, {m_directory.GetStringRef(), 7465713a05bSZachary Turner m_filename.GetStringRef(), component}); 7475713a05bSZachary Turner 748*2cb7cf8eSPavel Labath SetFile(result, false, m_style); 7495713a05bSZachary Turner } 7505713a05bSZachary Turner 7515713a05bSZachary Turner void FileSpec::AppendPathComponent(const FileSpec &new_path) { 7525713a05bSZachary Turner return AppendPathComponent(new_path.GetPath(false)); 7535713a05bSZachary Turner } 7545713a05bSZachary Turner 7555713a05bSZachary Turner void FileSpec::RemoveLastPathComponent() { 7565713a05bSZachary Turner // CLEANUP: Use StringRef for string handling. 7575713a05bSZachary Turner 7585713a05bSZachary Turner const bool resolve = false; 7595713a05bSZachary Turner if (m_filename.IsEmpty() && m_directory.IsEmpty()) { 7605713a05bSZachary Turner SetFile("", resolve); 7615713a05bSZachary Turner return; 7625713a05bSZachary Turner } 7635713a05bSZachary Turner if (m_directory.IsEmpty()) { 7645713a05bSZachary Turner SetFile("", resolve); 7655713a05bSZachary Turner return; 7665713a05bSZachary Turner } 7675713a05bSZachary Turner if (m_filename.IsEmpty()) { 7685713a05bSZachary Turner const char *dir_cstr = m_directory.GetCString(); 7695713a05bSZachary Turner const char *last_slash_ptr = ::strrchr(dir_cstr, '/'); 7705713a05bSZachary Turner 7715713a05bSZachary Turner // check for obvious cases before doing the full thing 7725713a05bSZachary Turner if (!last_slash_ptr) { 7735713a05bSZachary Turner SetFile("", resolve); 7745713a05bSZachary Turner return; 7755713a05bSZachary Turner } 7765713a05bSZachary Turner if (last_slash_ptr == dir_cstr) { 7775713a05bSZachary Turner SetFile("/", resolve); 7785713a05bSZachary Turner return; 7795713a05bSZachary Turner } 7805713a05bSZachary Turner size_t last_slash_pos = last_slash_ptr - dir_cstr + 1; 7815713a05bSZachary Turner ConstString new_path(dir_cstr, last_slash_pos); 7825713a05bSZachary Turner SetFile(new_path.GetCString(), resolve); 7835713a05bSZachary Turner } else 7845713a05bSZachary Turner SetFile(m_directory.GetCString(), resolve); 7855713a05bSZachary Turner } 7865713a05bSZachary Turner //------------------------------------------------------------------ 7875713a05bSZachary Turner /// Returns true if the filespec represents an implementation source 7885713a05bSZachary Turner /// file (files with a ".c", ".cpp", ".m", ".mm" (many more) 7895713a05bSZachary Turner /// extension). 7905713a05bSZachary Turner /// 7915713a05bSZachary Turner /// @return 7925713a05bSZachary Turner /// \b true if the filespec represents an implementation source 7935713a05bSZachary Turner /// file, \b false otherwise. 7945713a05bSZachary Turner //------------------------------------------------------------------ 7955713a05bSZachary Turner bool FileSpec::IsSourceImplementationFile() const { 7965713a05bSZachary Turner ConstString extension(GetFileNameExtension()); 7975713a05bSZachary Turner if (!extension) 7985713a05bSZachary Turner return false; 7995713a05bSZachary Turner 8005713a05bSZachary Turner static RegularExpression g_source_file_regex(llvm::StringRef( 8015713a05bSZachary Turner "^([cC]|[mM]|[mM][mM]|[cC][pP][pP]|[cC]\\+\\+|[cC][xX][xX]|[cC][cC]|[" 8025713a05bSZachary Turner "cC][pP]|[sS]|[aA][sS][mM]|[fF]|[fF]77|[fF]90|[fF]95|[fF]03|[fF][oO][" 8035713a05bSZachary Turner "rR]|[fF][tT][nN]|[fF][pP][pP]|[aA][dD][aA]|[aA][dD][bB]|[aA][dD][sS])" 8045713a05bSZachary Turner "$")); 8055713a05bSZachary Turner return g_source_file_regex.Execute(extension.GetStringRef()); 8065713a05bSZachary Turner } 8075713a05bSZachary Turner 8085713a05bSZachary Turner bool FileSpec::IsRelative() const { 8095713a05bSZachary Turner const char *dir = m_directory.GetCString(); 8105713a05bSZachary Turner llvm::StringRef directory(dir ? dir : ""); 8115713a05bSZachary Turner 8125713a05bSZachary Turner if (directory.size() > 0) { 813*2cb7cf8eSPavel Labath if (PathStyleIsPosix(m_style)) { 8145713a05bSZachary Turner // If the path doesn't start with '/' or '~', return true 8155713a05bSZachary Turner switch (directory[0]) { 8165713a05bSZachary Turner case '/': 8175713a05bSZachary Turner case '~': 8185713a05bSZachary Turner return false; 8195713a05bSZachary Turner default: 8205713a05bSZachary Turner return true; 8215713a05bSZachary Turner } 8225713a05bSZachary Turner } else { 8235713a05bSZachary Turner if (directory.size() >= 2 && directory[1] == ':') 8245713a05bSZachary Turner return false; 8255713a05bSZachary Turner if (directory[0] == '/') 8265713a05bSZachary Turner return false; 8275713a05bSZachary Turner return true; 8285713a05bSZachary Turner } 8295713a05bSZachary Turner } else if (m_filename) { 8305713a05bSZachary Turner // No directory, just a basename, return true 8315713a05bSZachary Turner return true; 8325713a05bSZachary Turner } 8335713a05bSZachary Turner return false; 8345713a05bSZachary Turner } 8355713a05bSZachary Turner 8365713a05bSZachary Turner bool FileSpec::IsAbsolute() const { return !FileSpec::IsRelative(); } 8375713a05bSZachary Turner 8385713a05bSZachary Turner void llvm::format_provider<FileSpec>::format(const FileSpec &F, 8395713a05bSZachary Turner raw_ostream &Stream, 8405713a05bSZachary Turner StringRef Style) { 8415713a05bSZachary Turner assert( 8425713a05bSZachary Turner (Style.empty() || Style.equals_lower("F") || Style.equals_lower("D")) && 8435713a05bSZachary Turner "Invalid FileSpec style!"); 8445713a05bSZachary Turner 8455713a05bSZachary Turner StringRef dir = F.GetDirectory().GetStringRef(); 8465713a05bSZachary Turner StringRef file = F.GetFilename().GetStringRef(); 8475713a05bSZachary Turner 8485713a05bSZachary Turner if (dir.empty() && file.empty()) { 8495713a05bSZachary Turner Stream << "(empty)"; 8505713a05bSZachary Turner return; 8515713a05bSZachary Turner } 8525713a05bSZachary Turner 8535713a05bSZachary Turner if (Style.equals_lower("F")) { 8545713a05bSZachary Turner Stream << (file.empty() ? "(empty)" : file); 8555713a05bSZachary Turner return; 8565713a05bSZachary Turner } 8575713a05bSZachary Turner 8585713a05bSZachary Turner // Style is either D or empty, either way we need to print the directory. 8595713a05bSZachary Turner if (!dir.empty()) { 86005097246SAdrian Prantl // Directory is stored in normalized form, which might be different than 86105097246SAdrian Prantl // preferred form. In order to handle this, we need to cut off the 86205097246SAdrian Prantl // filename, then denormalize, then write the entire denorm'ed directory. 8635713a05bSZachary Turner llvm::SmallString<64> denormalized_dir = dir; 864*2cb7cf8eSPavel Labath Denormalize(denormalized_dir, F.GetPathStyle()); 8655713a05bSZachary Turner Stream << denormalized_dir; 866*2cb7cf8eSPavel Labath Stream << GetPreferredPathSeparator(F.GetPathStyle()); 8675713a05bSZachary Turner } 8685713a05bSZachary Turner 8695713a05bSZachary Turner if (Style.equals_lower("D")) { 8705713a05bSZachary Turner // We only want to print the directory, so now just exit. 8715713a05bSZachary Turner if (dir.empty()) 8725713a05bSZachary Turner Stream << "(empty)"; 8735713a05bSZachary Turner return; 8745713a05bSZachary Turner } 8755713a05bSZachary Turner 8765713a05bSZachary Turner if (!file.empty()) 8775713a05bSZachary Turner Stream << file; 8785713a05bSZachary Turner } 879