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