13ef91c55SMichael J. Spencer //===- llvm/unittest/Support/Path.cpp - Path tests ------------------------===//
23ef91c55SMichael J. Spencer //
33ef91c55SMichael J. Spencer //                     The LLVM Compiler Infrastructure
43ef91c55SMichael J. Spencer //
53ef91c55SMichael J. Spencer // This file is distributed under the University of Illinois Open Source
63ef91c55SMichael J. Spencer // License. See LICENSE.TXT for details.
73ef91c55SMichael J. Spencer //
83ef91c55SMichael J. Spencer //===----------------------------------------------------------------------===//
93ef91c55SMichael J. Spencer 
103bc8e719SRafael Espindola #include "llvm/Support/Path.h"
11e8bb3920SSimon Dardis #include "llvm/ADT/SmallVector.h"
120d955d0bSDavid Majnemer #include "llvm/ADT/STLExtras.h"
13e8bb3920SSimon Dardis #include "llvm/ADT/Triple.h"
147187e4bbSPawel Bylica #include "llvm/Support/ConvertUTF.h"
152a826e40SRafael Espindola #include "llvm/Support/Errc.h"
167ecd94ccSMichael J. Spencer #include "llvm/Support/ErrorHandling.h"
17130cec21SChandler Carruth #include "llvm/Support/FileSystem.h"
1875e557f1SReid Kleckner #include "llvm/Support/FileUtilities.h"
19e8bb3920SSimon Dardis #include "llvm/Support/Host.h"
2084ab9b3bSRafael Espindola #include "llvm/Support/MemoryBuffer.h"
211f063600SMichael J. Spencer #include "llvm/Support/raw_ostream.h"
223ef91c55SMichael J. Spencer #include "gtest/gtest.h"
233ef91c55SMichael J. Spencer 
24a813d608SRafael Espindola #ifdef LLVM_ON_WIN32
25dd2ca837SNico Weber #include "llvm/ADT/ArrayRef.h"
26b94f0522SNAKAMURA Takumi #include <windows.h>
27a813d608SRafael Espindola #include <winerror.h>
28a813d608SRafael Espindola #endif
29a813d608SRafael Espindola 
306b9396c0SFrederic Riss #ifdef LLVM_ON_UNIX
316b9396c0SFrederic Riss #include <sys/stat.h>
326b9396c0SFrederic Riss #endif
336b9396c0SFrederic Riss 
343ef91c55SMichael J. Spencer using namespace llvm;
354571040eSMichael J. Spencer using namespace llvm::sys;
363ef91c55SMichael J. Spencer 
373b264ba9SMichael J. Spencer #define ASSERT_NO_ERROR(x)                                                     \
38c049c657SRafael Espindola   if (std::error_code ASSERT_NO_ERROR_ec = x) {                                \
391f063600SMichael J. Spencer     SmallString<128> MessageStorage;                                           \
401f063600SMichael J. Spencer     raw_svector_ostream Message(MessageStorage);                               \
411f063600SMichael J. Spencer     Message << #x ": did not return errc::success.\n"                          \
421f063600SMichael J. Spencer             << "error number: " << ASSERT_NO_ERROR_ec.value() << "\n"          \
431f063600SMichael J. Spencer             << "error message: " << ASSERT_NO_ERROR_ec.message() << "\n";      \
441f063600SMichael J. Spencer     GTEST_FATAL_FAILURE_(MessageStorage.c_str());                              \
45c049c657SRafael Espindola   } else {                                                                     \
46c049c657SRafael Espindola   }
473b264ba9SMichael J. Spencer 
483ef91c55SMichael J. Spencer namespace {
493ef91c55SMichael J. Spencer 
50606bb1a2SZhanyong Wan TEST(is_separator, Works) {
51606bb1a2SZhanyong Wan   EXPECT_TRUE(path::is_separator('/'));
52606bb1a2SZhanyong Wan   EXPECT_FALSE(path::is_separator('\0'));
53606bb1a2SZhanyong Wan   EXPECT_FALSE(path::is_separator('-'));
54606bb1a2SZhanyong Wan   EXPECT_FALSE(path::is_separator(' '));
55606bb1a2SZhanyong Wan 
56606bb1a2SZhanyong Wan #ifdef LLVM_ON_WIN32
57606bb1a2SZhanyong Wan   EXPECT_TRUE(path::is_separator('\\'));
58606bb1a2SZhanyong Wan #else
59606bb1a2SZhanyong Wan   EXPECT_FALSE(path::is_separator('\\'));
60606bb1a2SZhanyong Wan #endif
61606bb1a2SZhanyong Wan }
62606bb1a2SZhanyong Wan 
633ef91c55SMichael J. Spencer TEST(Support, Path) {
643ef91c55SMichael J. Spencer   SmallVector<StringRef, 40> paths;
653ef91c55SMichael J. Spencer   paths.push_back("");
663ef91c55SMichael J. Spencer   paths.push_back(".");
673ef91c55SMichael J. Spencer   paths.push_back("..");
683ef91c55SMichael J. Spencer   paths.push_back("foo");
693ef91c55SMichael J. Spencer   paths.push_back("/");
703ef91c55SMichael J. Spencer   paths.push_back("/foo");
713ef91c55SMichael J. Spencer   paths.push_back("foo/");
723ef91c55SMichael J. Spencer   paths.push_back("/foo/");
733ef91c55SMichael J. Spencer   paths.push_back("foo/bar");
743ef91c55SMichael J. Spencer   paths.push_back("/foo/bar");
753ef91c55SMichael J. Spencer   paths.push_back("//net");
763ef91c55SMichael J. Spencer   paths.push_back("//net/foo");
773ef91c55SMichael J. Spencer   paths.push_back("///foo///");
783ef91c55SMichael J. Spencer   paths.push_back("///foo///bar");
793ef91c55SMichael J. Spencer   paths.push_back("/.");
803ef91c55SMichael J. Spencer   paths.push_back("./");
813ef91c55SMichael J. Spencer   paths.push_back("/..");
823ef91c55SMichael J. Spencer   paths.push_back("../");
833ef91c55SMichael J. Spencer   paths.push_back("foo/.");
843ef91c55SMichael J. Spencer   paths.push_back("foo/..");
853ef91c55SMichael J. Spencer   paths.push_back("foo/./");
863ef91c55SMichael J. Spencer   paths.push_back("foo/./bar");
873ef91c55SMichael J. Spencer   paths.push_back("foo/..");
883ef91c55SMichael J. Spencer   paths.push_back("foo/../");
893ef91c55SMichael J. Spencer   paths.push_back("foo/../bar");
903ef91c55SMichael J. Spencer   paths.push_back("c:");
913ef91c55SMichael J. Spencer   paths.push_back("c:/");
923ef91c55SMichael J. Spencer   paths.push_back("c:foo");
933ef91c55SMichael J. Spencer   paths.push_back("c:/foo");
943ef91c55SMichael J. Spencer   paths.push_back("c:foo/");
953ef91c55SMichael J. Spencer   paths.push_back("c:/foo/");
963ef91c55SMichael J. Spencer   paths.push_back("c:/foo/bar");
973ef91c55SMichael J. Spencer   paths.push_back("prn:");
983ef91c55SMichael J. Spencer   paths.push_back("c:\\");
993ef91c55SMichael J. Spencer   paths.push_back("c:foo");
1003ef91c55SMichael J. Spencer   paths.push_back("c:\\foo");
1013ef91c55SMichael J. Spencer   paths.push_back("c:foo\\");
1023ef91c55SMichael J. Spencer   paths.push_back("c:\\foo\\");
1033ef91c55SMichael J. Spencer   paths.push_back("c:\\foo/");
1043ef91c55SMichael J. Spencer   paths.push_back("c:/foo\\bar");
1053ef91c55SMichael J. Spencer 
1060c274aedSJustin Bogner   SmallVector<StringRef, 5> ComponentStack;
1073ef91c55SMichael J. Spencer   for (SmallVector<StringRef, 40>::const_iterator i = paths.begin(),
1083ef91c55SMichael J. Spencer                                                   e = paths.end();
1093ef91c55SMichael J. Spencer                                                   i != e;
1103ef91c55SMichael J. Spencer                                                   ++i) {
1113ef91c55SMichael J. Spencer     for (sys::path::const_iterator ci = sys::path::begin(*i),
1123ef91c55SMichael J. Spencer                                    ce = sys::path::end(*i);
1133ef91c55SMichael J. Spencer                                    ci != ce;
1143ef91c55SMichael J. Spencer                                    ++ci) {
115b5ca6447SMichael J. Spencer       ASSERT_FALSE(ci->empty());
1160c274aedSJustin Bogner       ComponentStack.push_back(*ci);
1173ef91c55SMichael J. Spencer     }
1183ef91c55SMichael J. Spencer 
119545cbdf0SMichael J. Spencer     for (sys::path::reverse_iterator ci = sys::path::rbegin(*i),
120545cbdf0SMichael J. Spencer                                      ce = sys::path::rend(*i);
121545cbdf0SMichael J. Spencer                                      ci != ce;
122545cbdf0SMichael J. Spencer                                      ++ci) {
1230c274aedSJustin Bogner       ASSERT_TRUE(*ci == ComponentStack.back());
1240c274aedSJustin Bogner       ComponentStack.pop_back();
125545cbdf0SMichael J. Spencer     }
1260c274aedSJustin Bogner     ASSERT_TRUE(ComponentStack.empty());
127545cbdf0SMichael J. Spencer 
1280db71d9eSJustin Bogner     // Crash test most of the API - since we're iterating over all of our paths
1290db71d9eSJustin Bogner     // here there isn't really anything reasonable to assert on in the results.
1300db71d9eSJustin Bogner     (void)path::has_root_path(*i);
1310db71d9eSJustin Bogner     (void)path::root_path(*i);
1320db71d9eSJustin Bogner     (void)path::has_root_name(*i);
1330db71d9eSJustin Bogner     (void)path::root_name(*i);
1340db71d9eSJustin Bogner     (void)path::has_root_directory(*i);
1350db71d9eSJustin Bogner     (void)path::root_directory(*i);
1360db71d9eSJustin Bogner     (void)path::has_parent_path(*i);
1370db71d9eSJustin Bogner     (void)path::parent_path(*i);
1380db71d9eSJustin Bogner     (void)path::has_filename(*i);
1390db71d9eSJustin Bogner     (void)path::filename(*i);
1400db71d9eSJustin Bogner     (void)path::has_stem(*i);
1410db71d9eSJustin Bogner     (void)path::stem(*i);
1420db71d9eSJustin Bogner     (void)path::has_extension(*i);
1430db71d9eSJustin Bogner     (void)path::extension(*i);
1440db71d9eSJustin Bogner     (void)path::is_absolute(*i);
1450db71d9eSJustin Bogner     (void)path::is_relative(*i);
1463ef91c55SMichael J. Spencer 
1473b264ba9SMichael J. Spencer     SmallString<128> temp_store;
148b5ca6447SMichael J. Spencer     temp_store = *i;
1493b264ba9SMichael J. Spencer     ASSERT_NO_ERROR(fs::make_absolute(temp_store));
150b5ca6447SMichael J. Spencer     temp_store = *i;
1511e090f0bSMichael J. Spencer     path::remove_filename(temp_store);
1524d0c6fddSMichael J. Spencer 
153b5ca6447SMichael J. Spencer     temp_store = *i;
1541e090f0bSMichael J. Spencer     path::replace_extension(temp_store, "ext");
1554d0c6fddSMichael J. Spencer     StringRef filename(temp_store.begin(), temp_store.size()), stem, ext;
156f616b218SMichael J. Spencer     stem = path::stem(filename);
157f616b218SMichael J. Spencer     ext  = path::extension(filename);
158487e764bSJustin Bogner     EXPECT_EQ(*sys::path::rbegin(filename), (stem + ext).str());
1594d0c6fddSMichael J. Spencer 
1601e090f0bSMichael J. Spencer     path::native(*i, temp_store);
1613ef91c55SMichael J. Spencer   }
162ae1d5996SBenjamin Kramer 
163ae1d5996SBenjamin Kramer   SmallString<32> Relative("foo.cpp");
164ae1d5996SBenjamin Kramer   ASSERT_NO_ERROR(sys::fs::make_absolute("/root", Relative));
1652b4e14edSBenjamin Kramer   Relative[5] = '/'; // Fix up windows paths.
166ae1d5996SBenjamin Kramer   ASSERT_EQ("/root/foo.cpp", Relative);
167346a1338SMichael J. Spencer }
1684571040eSMichael J. Spencer 
16973537eacSTareq A. Siraj TEST(Support, RelativePathIterator) {
17073537eacSTareq A. Siraj   SmallString<64> Path(StringRef("c/d/e/foo.txt"));
17173537eacSTareq A. Siraj   typedef SmallVector<StringRef, 4> PathComponents;
17273537eacSTareq A. Siraj   PathComponents ExpectedPathComponents;
17373537eacSTareq A. Siraj   PathComponents ActualPathComponents;
17473537eacSTareq A. Siraj 
175e4405e94SChandler Carruth   StringRef(Path).split(ExpectedPathComponents, '/');
17673537eacSTareq A. Siraj 
17773537eacSTareq A. Siraj   for (path::const_iterator I = path::begin(Path), E = path::end(Path); I != E;
17873537eacSTareq A. Siraj        ++I) {
17973537eacSTareq A. Siraj     ActualPathComponents.push_back(*I);
18073537eacSTareq A. Siraj   }
18173537eacSTareq A. Siraj 
18273537eacSTareq A. Siraj   ASSERT_EQ(ExpectedPathComponents.size(), ActualPathComponents.size());
18373537eacSTareq A. Siraj 
18473537eacSTareq A. Siraj   for (size_t i = 0; i <ExpectedPathComponents.size(); ++i) {
18573537eacSTareq A. Siraj     EXPECT_EQ(ExpectedPathComponents[i].str(), ActualPathComponents[i].str());
18673537eacSTareq A. Siraj   }
18773537eacSTareq A. Siraj }
18873537eacSTareq A. Siraj 
1892547f93eSBen Langmuir TEST(Support, RelativePathDotIterator) {
1902547f93eSBen Langmuir   SmallString<64> Path(StringRef(".c/.d/../."));
1912547f93eSBen Langmuir   typedef SmallVector<StringRef, 4> PathComponents;
1922547f93eSBen Langmuir   PathComponents ExpectedPathComponents;
1932547f93eSBen Langmuir   PathComponents ActualPathComponents;
1942547f93eSBen Langmuir 
195e4405e94SChandler Carruth   StringRef(Path).split(ExpectedPathComponents, '/');
1962547f93eSBen Langmuir 
1972547f93eSBen Langmuir   for (path::const_iterator I = path::begin(Path), E = path::end(Path); I != E;
1982547f93eSBen Langmuir        ++I) {
1992547f93eSBen Langmuir     ActualPathComponents.push_back(*I);
2002547f93eSBen Langmuir   }
2012547f93eSBen Langmuir 
2022547f93eSBen Langmuir   ASSERT_EQ(ExpectedPathComponents.size(), ActualPathComponents.size());
2032547f93eSBen Langmuir 
2042547f93eSBen Langmuir   for (size_t i = 0; i <ExpectedPathComponents.size(); ++i) {
2052547f93eSBen Langmuir     EXPECT_EQ(ExpectedPathComponents[i].str(), ActualPathComponents[i].str());
2062547f93eSBen Langmuir   }
2072547f93eSBen Langmuir }
2082547f93eSBen Langmuir 
20973537eacSTareq A. Siraj TEST(Support, AbsolutePathIterator) {
21073537eacSTareq A. Siraj   SmallString<64> Path(StringRef("/c/d/e/foo.txt"));
21173537eacSTareq A. Siraj   typedef SmallVector<StringRef, 4> PathComponents;
21273537eacSTareq A. Siraj   PathComponents ExpectedPathComponents;
21373537eacSTareq A. Siraj   PathComponents ActualPathComponents;
21473537eacSTareq A. Siraj 
215e4405e94SChandler Carruth   StringRef(Path).split(ExpectedPathComponents, '/');
21673537eacSTareq A. Siraj 
21773537eacSTareq A. Siraj   // The root path will also be a component when iterating
21873537eacSTareq A. Siraj   ExpectedPathComponents[0] = "/";
21973537eacSTareq A. Siraj 
22073537eacSTareq A. Siraj   for (path::const_iterator I = path::begin(Path), E = path::end(Path); I != E;
22173537eacSTareq A. Siraj        ++I) {
22273537eacSTareq A. Siraj     ActualPathComponents.push_back(*I);
22373537eacSTareq A. Siraj   }
22473537eacSTareq A. Siraj 
22573537eacSTareq A. Siraj   ASSERT_EQ(ExpectedPathComponents.size(), ActualPathComponents.size());
22673537eacSTareq A. Siraj 
22773537eacSTareq A. Siraj   for (size_t i = 0; i <ExpectedPathComponents.size(); ++i) {
22873537eacSTareq A. Siraj     EXPECT_EQ(ExpectedPathComponents[i].str(), ActualPathComponents[i].str());
22973537eacSTareq A. Siraj   }
23073537eacSTareq A. Siraj }
23173537eacSTareq A. Siraj 
2322547f93eSBen Langmuir TEST(Support, AbsolutePathDotIterator) {
2332547f93eSBen Langmuir   SmallString<64> Path(StringRef("/.c/.d/../."));
2342547f93eSBen Langmuir   typedef SmallVector<StringRef, 4> PathComponents;
2352547f93eSBen Langmuir   PathComponents ExpectedPathComponents;
2362547f93eSBen Langmuir   PathComponents ActualPathComponents;
2372547f93eSBen Langmuir 
238e4405e94SChandler Carruth   StringRef(Path).split(ExpectedPathComponents, '/');
2392547f93eSBen Langmuir 
2402547f93eSBen Langmuir   // The root path will also be a component when iterating
2412547f93eSBen Langmuir   ExpectedPathComponents[0] = "/";
2422547f93eSBen Langmuir 
2432547f93eSBen Langmuir   for (path::const_iterator I = path::begin(Path), E = path::end(Path); I != E;
2442547f93eSBen Langmuir        ++I) {
2452547f93eSBen Langmuir     ActualPathComponents.push_back(*I);
2462547f93eSBen Langmuir   }
2472547f93eSBen Langmuir 
2482547f93eSBen Langmuir   ASSERT_EQ(ExpectedPathComponents.size(), ActualPathComponents.size());
2492547f93eSBen Langmuir 
2502547f93eSBen Langmuir   for (size_t i = 0; i <ExpectedPathComponents.size(); ++i) {
2512547f93eSBen Langmuir     EXPECT_EQ(ExpectedPathComponents[i].str(), ActualPathComponents[i].str());
2522547f93eSBen Langmuir   }
2532547f93eSBen Langmuir }
2542547f93eSBen Langmuir 
25573537eacSTareq A. Siraj #ifdef LLVM_ON_WIN32
25673537eacSTareq A. Siraj TEST(Support, AbsolutePathIteratorWin32) {
25773537eacSTareq A. Siraj   SmallString<64> Path(StringRef("c:\\c\\e\\foo.txt"));
25873537eacSTareq A. Siraj   typedef SmallVector<StringRef, 4> PathComponents;
25973537eacSTareq A. Siraj   PathComponents ExpectedPathComponents;
26073537eacSTareq A. Siraj   PathComponents ActualPathComponents;
26173537eacSTareq A. Siraj 
26273537eacSTareq A. Siraj   StringRef(Path).split(ExpectedPathComponents, "\\");
26373537eacSTareq A. Siraj 
26473537eacSTareq A. Siraj   // The root path (which comes after the drive name) will also be a component
26573537eacSTareq A. Siraj   // when iterating.
26673537eacSTareq A. Siraj   ExpectedPathComponents.insert(ExpectedPathComponents.begin()+1, "\\");
26773537eacSTareq A. Siraj 
26873537eacSTareq A. Siraj   for (path::const_iterator I = path::begin(Path), E = path::end(Path); I != E;
26973537eacSTareq A. Siraj        ++I) {
27073537eacSTareq A. Siraj     ActualPathComponents.push_back(*I);
27173537eacSTareq A. Siraj   }
27273537eacSTareq A. Siraj 
27373537eacSTareq A. Siraj   ASSERT_EQ(ExpectedPathComponents.size(), ActualPathComponents.size());
27473537eacSTareq A. Siraj 
27573537eacSTareq A. Siraj   for (size_t i = 0; i <ExpectedPathComponents.size(); ++i) {
27673537eacSTareq A. Siraj     EXPECT_EQ(ExpectedPathComponents[i].str(), ActualPathComponents[i].str());
27773537eacSTareq A. Siraj   }
27873537eacSTareq A. Siraj }
27973537eacSTareq A. Siraj #endif // LLVM_ON_WIN32
28073537eacSTareq A. Siraj 
2818d116396SBen Langmuir TEST(Support, AbsolutePathIteratorEnd) {
2828d116396SBen Langmuir   // Trailing slashes are converted to '.' unless they are part of the root path.
2838d116396SBen Langmuir   SmallVector<StringRef, 4> Paths;
2848d116396SBen Langmuir   Paths.push_back("/foo/");
2858d116396SBen Langmuir   Paths.push_back("/foo//");
2868d116396SBen Langmuir   Paths.push_back("//net//");
2878d116396SBen Langmuir #ifdef LLVM_ON_WIN32
2888d116396SBen Langmuir   Paths.push_back("c:\\\\");
2898d116396SBen Langmuir #endif
2908d116396SBen Langmuir 
2918d116396SBen Langmuir   for (StringRef Path : Paths) {
292487e764bSJustin Bogner     StringRef LastComponent = *path::rbegin(Path);
2938d116396SBen Langmuir     EXPECT_EQ(".", LastComponent);
2948d116396SBen Langmuir   }
2958d116396SBen Langmuir 
2968d116396SBen Langmuir   SmallVector<StringRef, 3> RootPaths;
2978d116396SBen Langmuir   RootPaths.push_back("/");
2988d116396SBen Langmuir   RootPaths.push_back("//net/");
2998d116396SBen Langmuir #ifdef LLVM_ON_WIN32
3008d116396SBen Langmuir   RootPaths.push_back("c:\\");
3018d116396SBen Langmuir #endif
3028d116396SBen Langmuir 
3038d116396SBen Langmuir   for (StringRef Path : RootPaths) {
304487e764bSJustin Bogner     StringRef LastComponent = *path::rbegin(Path);
3058d116396SBen Langmuir     EXPECT_EQ(1u, LastComponent.size());
3068d116396SBen Langmuir     EXPECT_TRUE(path::is_separator(LastComponent[0]));
3078d116396SBen Langmuir   }
3088d116396SBen Langmuir }
3098d116396SBen Langmuir 
310f7d41011SPeter Collingbourne TEST(Support, HomeDirectory) {
311cc275e42SNAKAMURA Takumi   std::string expected;
3127187e4bbSPawel Bylica #ifdef LLVM_ON_WIN32
3132fa025cdSPawel Bylica   if (wchar_t const *path = ::_wgetenv(L"USERPROFILE")) {
3147187e4bbSPawel Bylica     auto pathLen = ::wcslen(path);
3152fa025cdSPawel Bylica     ArrayRef<char> ref{reinterpret_cast<char const *>(path),
3162fa025cdSPawel Bylica                        pathLen * sizeof(wchar_t)};
3177187e4bbSPawel Bylica     convertUTF16ToUTF8String(ref, expected);
3182fa025cdSPawel Bylica   }
3197187e4bbSPawel Bylica #else
3202fa025cdSPawel Bylica   if (char const *path = ::getenv("HOME"))
3212fa025cdSPawel Bylica     expected = path;
322f7d41011SPeter Collingbourne #endif
3232fa025cdSPawel Bylica   // Do not try to test it if we don't know what to expect.
3242fa025cdSPawel Bylica   // On Windows we use something better than env vars.
3252fa025cdSPawel Bylica   if (!expected.empty()) {
326f7d41011SPeter Collingbourne     SmallString<128> HomeDir;
3277187e4bbSPawel Bylica     auto status = path::home_directory(HomeDir);
3282fa025cdSPawel Bylica     EXPECT_TRUE(status);
3297187e4bbSPawel Bylica     EXPECT_EQ(expected, HomeDir);
330f7d41011SPeter Collingbourne   }
331cc275e42SNAKAMURA Takumi }
332f7d41011SPeter Collingbourne 
3330e97e5cbSPawel Bylica TEST(Support, UserCacheDirectory) {
3340e97e5cbSPawel Bylica   SmallString<13> CacheDir;
3350e97e5cbSPawel Bylica   SmallString<20> CacheDir2;
3360e97e5cbSPawel Bylica   auto Status = path::user_cache_directory(CacheDir, "");
3370e97e5cbSPawel Bylica   EXPECT_TRUE(Status ^ CacheDir.empty());
3380e97e5cbSPawel Bylica 
3390e97e5cbSPawel Bylica   if (Status) {
3400e97e5cbSPawel Bylica     EXPECT_TRUE(path::user_cache_directory(CacheDir2, "")); // should succeed
3410e97e5cbSPawel Bylica     EXPECT_EQ(CacheDir, CacheDir2); // and return same paths
3420e97e5cbSPawel Bylica 
3430e97e5cbSPawel Bylica     EXPECT_TRUE(path::user_cache_directory(CacheDir, "A", "B", "file.c"));
3440e97e5cbSPawel Bylica     auto It = path::rbegin(CacheDir);
3450e97e5cbSPawel Bylica     EXPECT_EQ("file.c", *It);
3460e97e5cbSPawel Bylica     EXPECT_EQ("B", *++It);
3470e97e5cbSPawel Bylica     EXPECT_EQ("A", *++It);
3480e97e5cbSPawel Bylica     auto ParentDir = *++It;
3490e97e5cbSPawel Bylica 
3500e97e5cbSPawel Bylica     // Test Unicode: "<user_cache_dir>/(pi)r^2/aleth.0"
3510e97e5cbSPawel Bylica     EXPECT_TRUE(path::user_cache_directory(CacheDir2, "\xCF\x80r\xC2\xB2",
3520e97e5cbSPawel Bylica                                            "\xE2\x84\xB5.0"));
3530e97e5cbSPawel Bylica     auto It2 = path::rbegin(CacheDir2);
3540e97e5cbSPawel Bylica     EXPECT_EQ("\xE2\x84\xB5.0", *It2);
3550e97e5cbSPawel Bylica     EXPECT_EQ("\xCF\x80r\xC2\xB2", *++It2);
3560e97e5cbSPawel Bylica     auto ParentDir2 = *++It2;
3570e97e5cbSPawel Bylica 
3580e97e5cbSPawel Bylica     EXPECT_EQ(ParentDir, ParentDir2);
3590e97e5cbSPawel Bylica   }
3600e97e5cbSPawel Bylica }
3610e97e5cbSPawel Bylica 
362a90e7451SPawel Bylica TEST(Support, TempDirectory) {
363a90e7451SPawel Bylica   SmallString<32> TempDir;
364a90e7451SPawel Bylica   path::system_temp_directory(false, TempDir);
365a90e7451SPawel Bylica   EXPECT_TRUE(!TempDir.empty());
366a90e7451SPawel Bylica   TempDir.clear();
367a90e7451SPawel Bylica   path::system_temp_directory(true, TempDir);
368a90e7451SPawel Bylica   EXPECT_TRUE(!TempDir.empty());
369a90e7451SPawel Bylica }
370a90e7451SPawel Bylica 
37106d5618cSDavid Blaikie #ifdef LLVM_ON_WIN32
372a90e7451SPawel Bylica static std::string path2regex(std::string Path) {
373a90e7451SPawel Bylica   size_t Pos = 0;
374a90e7451SPawel Bylica   while ((Pos = Path.find('\\', Pos)) != std::string::npos) {
375a90e7451SPawel Bylica     Path.replace(Pos, 1, "\\\\");
376a90e7451SPawel Bylica     Pos += 2;
377a90e7451SPawel Bylica   }
378a90e7451SPawel Bylica   return Path;
379a90e7451SPawel Bylica }
380a90e7451SPawel Bylica 
381a90e7451SPawel Bylica /// Helper for running temp dir test in separated process. See below.
382a90e7451SPawel Bylica #define EXPECT_TEMP_DIR(prepare, expected)                                     \
383a90e7451SPawel Bylica   EXPECT_EXIT(                                                                 \
384a90e7451SPawel Bylica       {                                                                        \
385a90e7451SPawel Bylica         prepare;                                                               \
386a90e7451SPawel Bylica         SmallString<300> TempDir;                                              \
387a90e7451SPawel Bylica         path::system_temp_directory(true, TempDir);                            \
388a90e7451SPawel Bylica         raw_os_ostream(std::cerr) << TempDir;                                  \
389a90e7451SPawel Bylica         std::exit(0);                                                          \
390a90e7451SPawel Bylica       },                                                                       \
391a90e7451SPawel Bylica       ::testing::ExitedWithCode(0), path2regex(expected))
392a90e7451SPawel Bylica 
393a90e7451SPawel Bylica TEST(SupportDeathTest, TempDirectoryOnWindows) {
394a90e7451SPawel Bylica   // In this test we want to check how system_temp_directory responds to
395a90e7451SPawel Bylica   // different values of specific env vars. To prevent corrupting env vars of
396a90e7451SPawel Bylica   // the current process all checks are done in separated processes.
397a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_wputenv_s(L"TMP", L"C:\\OtherFolder"), "C:\\OtherFolder");
398a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_wputenv_s(L"TMP", L"C:/Unix/Path/Seperators"),
399a90e7451SPawel Bylica                   "C:\\Unix\\Path\\Seperators");
400a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_wputenv_s(L"TMP", L"Local Path"), ".+\\Local Path$");
401a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_wputenv_s(L"TMP", L"F:\\TrailingSep\\"), "F:\\TrailingSep");
402a90e7451SPawel Bylica   EXPECT_TEMP_DIR(
403a90e7451SPawel Bylica       _wputenv_s(L"TMP", L"C:\\2\x03C0r-\x00B5\x00B3\\\x2135\x2080"),
404a90e7451SPawel Bylica       "C:\\2\xCF\x80r-\xC2\xB5\xC2\xB3\\\xE2\x84\xB5\xE2\x82\x80");
405a90e7451SPawel Bylica 
406a90e7451SPawel Bylica   // Test $TMP empty, $TEMP set.
407a90e7451SPawel Bylica   EXPECT_TEMP_DIR(
408a90e7451SPawel Bylica       {
409a90e7451SPawel Bylica         _wputenv_s(L"TMP", L"");
410a90e7451SPawel Bylica         _wputenv_s(L"TEMP", L"C:\\Valid\\Path");
411a90e7451SPawel Bylica       },
412a90e7451SPawel Bylica       "C:\\Valid\\Path");
413a90e7451SPawel Bylica 
414a90e7451SPawel Bylica   // All related env vars empty
415a90e7451SPawel Bylica   EXPECT_TEMP_DIR(
416a90e7451SPawel Bylica   {
417a90e7451SPawel Bylica     _wputenv_s(L"TMP", L"");
418a90e7451SPawel Bylica     _wputenv_s(L"TEMP", L"");
419a90e7451SPawel Bylica     _wputenv_s(L"USERPROFILE", L"");
420a90e7451SPawel Bylica   },
421a90e7451SPawel Bylica     "C:\\Temp");
422a90e7451SPawel Bylica 
423a90e7451SPawel Bylica   // Test evn var / path with 260 chars.
424a90e7451SPawel Bylica   SmallString<270> Expected{"C:\\Temp\\AB\\123456789"};
425a90e7451SPawel Bylica   while (Expected.size() < 260)
426a90e7451SPawel Bylica     Expected.append("\\DirNameWith19Charss");
42788e9ff62SReid Kleckner   ASSERT_EQ(260U, Expected.size());
428a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_putenv_s("TMP", Expected.c_str()), Expected.c_str());
429a90e7451SPawel Bylica }
430a90e7451SPawel Bylica #endif
431a90e7451SPawel Bylica 
432346a1338SMichael J. Spencer class FileSystemTest : public testing::Test {
433346a1338SMichael J. Spencer protected:
434346a1338SMichael J. Spencer   /// Unique temporary directory in which all created filesystem entities must
435d9c4a9afSPaul Robinson   /// be placed. It is removed at the end of each test (must be empty).
436346a1338SMichael J. Spencer   SmallString<128> TestDirectory;
437346a1338SMichael J. Spencer 
438f817c1cbSAlexander Kornienko   void SetUp() override {
439346a1338SMichael J. Spencer     ASSERT_NO_ERROR(
4407ffacc49SRafael Espindola         fs::createUniqueDirectory("file-system-test", TestDirectory));
441346a1338SMichael J. Spencer     // We don't care about this specific file.
44242368cc1SMichael J. Spencer     errs() << "Test Directory: " << TestDirectory << '\n';
44342368cc1SMichael J. Spencer     errs().flush();
444346a1338SMichael J. Spencer   }
445346a1338SMichael J. Spencer 
446f817c1cbSAlexander Kornienko   void TearDown() override { ASSERT_NO_ERROR(fs::remove(TestDirectory.str())); }
447346a1338SMichael J. Spencer };
448346a1338SMichael J. Spencer 
4499fbe530dSAaron Ballman TEST_F(FileSystemTest, Unique) {
4509fbe530dSAaron Ballman   // Create a temp file.
4519fbe530dSAaron Ballman   int FileDescriptor;
4529fbe530dSAaron Ballman   SmallString<64> TempPath;
4539fbe530dSAaron Ballman   ASSERT_NO_ERROR(
454155cf0f3SRafael Espindola       fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
4559fbe530dSAaron Ballman 
4569fbe530dSAaron Ballman   // The same file should return an identical unique id.
4577f822a93SRafael Espindola   fs::UniqueID F1, F2;
4587cf7c51cSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), F1));
4597cf7c51cSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), F2));
4609fbe530dSAaron Ballman   ASSERT_EQ(F1, F2);
4619fbe530dSAaron Ballman 
4629fbe530dSAaron Ballman   // Different files should return different unique ids.
4639fbe530dSAaron Ballman   int FileDescriptor2;
4649fbe530dSAaron Ballman   SmallString<64> TempPath2;
4659fbe530dSAaron Ballman   ASSERT_NO_ERROR(
466155cf0f3SRafael Espindola       fs::createTemporaryFile("prefix", "temp", FileDescriptor2, TempPath2));
4679fbe530dSAaron Ballman 
4687f822a93SRafael Espindola   fs::UniqueID D;
4697cf7c51cSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath2), D));
4709fbe530dSAaron Ballman   ASSERT_NE(D, F1);
4719fbe530dSAaron Ballman   ::close(FileDescriptor2);
4729fbe530dSAaron Ballman 
4739fbe530dSAaron Ballman   ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
4749fbe530dSAaron Ballman 
4759fbe530dSAaron Ballman   // Two paths representing the same file on disk should still provide the
4769fbe530dSAaron Ballman   // same unique id.  We can test this by making a hard link.
47783f858e5SRafael Espindola   ASSERT_NO_ERROR(fs::create_link(Twine(TempPath), Twine(TempPath2)));
4787f822a93SRafael Espindola   fs::UniqueID D2;
4797cf7c51cSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath2), D2));
4809fbe530dSAaron Ballman   ASSERT_EQ(D2, F1);
4819fbe530dSAaron Ballman 
4829fbe530dSAaron Ballman   ::close(FileDescriptor);
483a5932afeSRafael Espindola 
484a5932afeSRafael Espindola   SmallString<128> Dir1;
485a5932afeSRafael Espindola   ASSERT_NO_ERROR(
486a5932afeSRafael Espindola      fs::createUniqueDirectory("dir1", Dir1));
487a5932afeSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Dir1.c_str(), F1));
488a5932afeSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Dir1.c_str(), F2));
489a5932afeSRafael Espindola   ASSERT_EQ(F1, F2);
490a5932afeSRafael Espindola 
491a5932afeSRafael Espindola   SmallString<128> Dir2;
492a5932afeSRafael Espindola   ASSERT_NO_ERROR(
493a5932afeSRafael Espindola      fs::createUniqueDirectory("dir2", Dir2));
494a5932afeSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Dir2.c_str(), F2));
495a5932afeSRafael Espindola   ASSERT_NE(F1, F2);
4961a4398a1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(Dir1));
4971a4398a1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(Dir2));
49875e557f1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(TempPath2));
49975e557f1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(TempPath));
5009fbe530dSAaron Ballman }
5019fbe530dSAaron Ballman 
502346a1338SMichael J. Spencer TEST_F(FileSystemTest, TempFiles) {
5034fb115d2SMichael J. Spencer   // Create a temp file.
5044571040eSMichael J. Spencer   int FileDescriptor;
5054571040eSMichael J. Spencer   SmallString<64> TempPath;
5063b264ba9SMichael J. Spencer   ASSERT_NO_ERROR(
507155cf0f3SRafael Espindola       fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
5084571040eSMichael J. Spencer 
5094fb115d2SMichael J. Spencer   // Make sure it exists.
510c1599157SRafael Espindola   ASSERT_TRUE(sys::fs::exists(Twine(TempPath)));
5114571040eSMichael J. Spencer 
5124fb115d2SMichael J. Spencer   // Create another temp tile.
5134fb115d2SMichael J. Spencer   int FD2;
5144fb115d2SMichael J. Spencer   SmallString<64> TempPath2;
515155cf0f3SRafael Espindola   ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "temp", FD2, TempPath2));
516d3c89041SRafael Espindola   ASSERT_TRUE(TempPath2.endswith(".temp"));
5174fb115d2SMichael J. Spencer   ASSERT_NE(TempPath.str(), TempPath2.str());
5184fb115d2SMichael J. Spencer 
519203d7807SMichael J. Spencer   fs::file_status A, B;
520203d7807SMichael J. Spencer   ASSERT_NO_ERROR(fs::status(Twine(TempPath), A));
521203d7807SMichael J. Spencer   ASSERT_NO_ERROR(fs::status(Twine(TempPath2), B));
522203d7807SMichael J. Spencer   EXPECT_FALSE(fs::equivalent(A, B));
523203d7807SMichael J. Spencer 
5244fb115d2SMichael J. Spencer   ::close(FD2);
525213c4cb1SRafael Espindola 
5264fb115d2SMichael J. Spencer   // Remove Temp2.
5275c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
5285c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
5295c20ac01SRafael Espindola   ASSERT_EQ(fs::remove(Twine(TempPath2), false),
5302a826e40SRafael Espindola             errc::no_such_file_or_directory);
5314fb115d2SMichael J. Spencer 
532c049c657SRafael Espindola   std::error_code EC = fs::status(TempPath2.c_str(), B);
5332a826e40SRafael Espindola   EXPECT_EQ(EC, errc::no_such_file_or_directory);
534107b74c6SRafael Espindola   EXPECT_EQ(B.type(), fs::file_type::file_not_found);
535107b74c6SRafael Espindola 
5364fb115d2SMichael J. Spencer   // Make sure Temp2 doesn't exist.
537281f23adSRafael Espindola   ASSERT_EQ(fs::access(Twine(TempPath2), sys::fs::AccessMode::Exist),
538281f23adSRafael Espindola             errc::no_such_file_or_directory);
5394fb115d2SMichael J. Spencer 
540d3c89041SRafael Espindola   SmallString<64> TempPath3;
541d3c89041SRafael Espindola   ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "", TempPath3));
542d3c89041SRafael Espindola   ASSERT_FALSE(TempPath3.endswith("."));
54375e557f1SReid Kleckner   FileRemover Cleanup3(TempPath3);
544d3c89041SRafael Espindola 
5454fb115d2SMichael J. Spencer   // Create a hard link to Temp1.
54683f858e5SRafael Espindola   ASSERT_NO_ERROR(fs::create_link(Twine(TempPath), Twine(TempPath2)));
5474fb115d2SMichael J. Spencer   bool equal;
5483b264ba9SMichael J. Spencer   ASSERT_NO_ERROR(fs::equivalent(Twine(TempPath), Twine(TempPath2), equal));
5494fb115d2SMichael J. Spencer   EXPECT_TRUE(equal);
550203d7807SMichael J. Spencer   ASSERT_NO_ERROR(fs::status(Twine(TempPath), A));
551203d7807SMichael J. Spencer   ASSERT_NO_ERROR(fs::status(Twine(TempPath2), B));
552203d7807SMichael J. Spencer   EXPECT_TRUE(fs::equivalent(A, B));
5534fb115d2SMichael J. Spencer 
5544fb115d2SMichael J. Spencer   // Remove Temp1.
5554571040eSMichael J. Spencer   ::close(FileDescriptor);
5565c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TempPath)));
5574571040eSMichael J. Spencer 
5584fb115d2SMichael J. Spencer   // Remove the hard link.
5595c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
5604fb115d2SMichael J. Spencer 
5614fb115d2SMichael J. Spencer   // Make sure Temp1 doesn't exist.
562281f23adSRafael Espindola   ASSERT_EQ(fs::access(Twine(TempPath), sys::fs::AccessMode::Exist),
563281f23adSRafael Espindola             errc::no_such_file_or_directory);
564fcdf9a82SAaron Ballman 
565fcdf9a82SAaron Ballman #ifdef LLVM_ON_WIN32
566fcdf9a82SAaron Ballman   // Path name > 260 chars should get an error.
567fcdf9a82SAaron Ballman   const char *Path270 =
568fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz9abcdefghijklmnopqrstuvwxyz8"
569fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz7abcdefghijklmnopqrstuvwxyz6"
570fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz5abcdefghijklmnopqrstuvwxyz4"
571fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz3abcdefghijklmnopqrstuvwxyz2"
572fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz1abcdefghijklmnopqrstuvwxyz0";
5731b6c7347SPaul Robinson   EXPECT_EQ(fs::createUniqueFile(Path270, FileDescriptor, TempPath),
574d9c4a9afSPaul Robinson             errc::invalid_argument);
575d9c4a9afSPaul Robinson   // Relative path < 247 chars, no problem.
576d9c4a9afSPaul Robinson   const char *Path216 =
577d9c4a9afSPaul Robinson     "abcdefghijklmnopqrstuvwxyz7abcdefghijklmnopqrstuvwxyz6"
578d9c4a9afSPaul Robinson     "abcdefghijklmnopqrstuvwxyz5abcdefghijklmnopqrstuvwxyz4"
579d9c4a9afSPaul Robinson     "abcdefghijklmnopqrstuvwxyz3abcdefghijklmnopqrstuvwxyz2"
580d9c4a9afSPaul Robinson     "abcdefghijklmnopqrstuvwxyz1abcdefghijklmnopqrstuvwxyz0";
5811b6c7347SPaul Robinson   ASSERT_NO_ERROR(fs::createTemporaryFile(Path216, "", TempPath));
582d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::remove(Twine(TempPath)));
583fcdf9a82SAaron Ballman #endif
584346a1338SMichael J. Spencer }
5857ecd94ccSMichael J. Spencer 
5865c20ac01SRafael Espindola TEST_F(FileSystemTest, CreateDir) {
5875c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::create_directory(Twine(TestDirectory) + "foo"));
5885c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::create_directory(Twine(TestDirectory) + "foo"));
5895c20ac01SRafael Espindola   ASSERT_EQ(fs::create_directory(Twine(TestDirectory) + "foo", false),
5902a826e40SRafael Espindola             errc::file_exists);
5915c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "foo"));
592d9c4a9afSPaul Robinson 
5936b9396c0SFrederic Riss #ifdef LLVM_ON_UNIX
5946b9396c0SFrederic Riss   // Set a 0000 umask so that we can test our directory permissions.
5956b9396c0SFrederic Riss   mode_t OldUmask = ::umask(0000);
5966b9396c0SFrederic Riss 
5976b9396c0SFrederic Riss   fs::file_status Status;
5986b9396c0SFrederic Riss   ASSERT_NO_ERROR(
5996b9396c0SFrederic Riss       fs::create_directory(Twine(TestDirectory) + "baz500", false,
6006b9396c0SFrederic Riss                            fs::perms::owner_read | fs::perms::owner_exe));
6016b9396c0SFrederic Riss   ASSERT_NO_ERROR(fs::status(Twine(TestDirectory) + "baz500", Status));
6026b9396c0SFrederic Riss   ASSERT_EQ(Status.permissions() & fs::perms::all_all,
6036b9396c0SFrederic Riss             fs::perms::owner_read | fs::perms::owner_exe);
6046b9396c0SFrederic Riss   ASSERT_NO_ERROR(fs::create_directory(Twine(TestDirectory) + "baz777", false,
6056b9396c0SFrederic Riss                                        fs::perms::all_all));
6066b9396c0SFrederic Riss   ASSERT_NO_ERROR(fs::status(Twine(TestDirectory) + "baz777", Status));
6076b9396c0SFrederic Riss   ASSERT_EQ(Status.permissions() & fs::perms::all_all, fs::perms::all_all);
6086b9396c0SFrederic Riss 
6096b9396c0SFrederic Riss   // Restore umask to be safe.
6106b9396c0SFrederic Riss   ::umask(OldUmask);
6116b9396c0SFrederic Riss #endif
6126b9396c0SFrederic Riss 
613d9c4a9afSPaul Robinson #ifdef LLVM_ON_WIN32
614d9c4a9afSPaul Robinson   // Prove that create_directories() can handle a pathname > 248 characters,
615d9c4a9afSPaul Robinson   // which is the documented limit for CreateDirectory().
616d9c4a9afSPaul Robinson   // (248 is MAX_PATH subtracting room for an 8.3 filename.)
617d9c4a9afSPaul Robinson   // Generate a directory path guaranteed to fall into that range.
618d9c4a9afSPaul Robinson   size_t TmpLen = TestDirectory.size();
619d9c4a9afSPaul Robinson   const char *OneDir = "\\123456789";
620d9c4a9afSPaul Robinson   size_t OneDirLen = strlen(OneDir);
621493456e7SAaron Ballman   ASSERT_LT(OneDirLen, 12U);
622d9c4a9afSPaul Robinson   size_t NLevels = ((248 - TmpLen) / OneDirLen) + 1;
623d9c4a9afSPaul Robinson   SmallString<260> LongDir(TestDirectory);
624d9c4a9afSPaul Robinson   for (size_t I = 0; I < NLevels; ++I)
625d9c4a9afSPaul Robinson     LongDir.append(OneDir);
626d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::create_directories(Twine(LongDir)));
627d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::create_directories(Twine(LongDir)));
628d9c4a9afSPaul Robinson   ASSERT_EQ(fs::create_directories(Twine(LongDir), false),
629d9c4a9afSPaul Robinson             errc::file_exists);
630d9c4a9afSPaul Robinson   // Tidy up, "recursively" removing the directories.
631d9c4a9afSPaul Robinson   StringRef ThisDir(LongDir);
632d9c4a9afSPaul Robinson   for (size_t J = 0; J < NLevels; ++J) {
633d9c4a9afSPaul Robinson     ASSERT_NO_ERROR(fs::remove(ThisDir));
634d9c4a9afSPaul Robinson     ThisDir = path::parent_path(ThisDir);
635d9c4a9afSPaul Robinson   }
636d9c4a9afSPaul Robinson 
637d9c4a9afSPaul Robinson   // Similarly for a relative pathname.  Need to set the current directory to
638d9c4a9afSPaul Robinson   // TestDirectory so that the one we create ends up in the right place.
639d9c4a9afSPaul Robinson   char PreviousDir[260];
640d9c4a9afSPaul Robinson   size_t PreviousDirLen = ::GetCurrentDirectoryA(260, PreviousDir);
641493456e7SAaron Ballman   ASSERT_GT(PreviousDirLen, 0U);
642493456e7SAaron Ballman   ASSERT_LT(PreviousDirLen, 260U);
643d9c4a9afSPaul Robinson   ASSERT_NE(::SetCurrentDirectoryA(TestDirectory.c_str()), 0);
644d9c4a9afSPaul Robinson   LongDir.clear();
645d9c4a9afSPaul Robinson   // Generate a relative directory name with absolute length > 248.
646d9c4a9afSPaul Robinson   size_t LongDirLen = 249 - TestDirectory.size();
647d9c4a9afSPaul Robinson   LongDir.assign(LongDirLen, 'a');
648d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::create_directory(Twine(LongDir)));
649d9c4a9afSPaul Robinson   // While we're here, prove that .. and . handling works in these long paths.
650d9c4a9afSPaul Robinson   const char *DotDotDirs = "\\..\\.\\b";
651d9c4a9afSPaul Robinson   LongDir.append(DotDotDirs);
6521b6c7347SPaul Robinson   ASSERT_NO_ERROR(fs::create_directory("b"));
653d9c4a9afSPaul Robinson   ASSERT_EQ(fs::create_directory(Twine(LongDir), false), errc::file_exists);
654d9c4a9afSPaul Robinson   // And clean up.
6551b6c7347SPaul Robinson   ASSERT_NO_ERROR(fs::remove("b"));
656d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::remove(
657d9c4a9afSPaul Robinson     Twine(LongDir.substr(0, LongDir.size() - strlen(DotDotDirs)))));
658d9c4a9afSPaul Robinson   ASSERT_NE(::SetCurrentDirectoryA(PreviousDir), 0);
659d9c4a9afSPaul Robinson #endif
6605c20ac01SRafael Espindola }
6615c20ac01SRafael Espindola 
662346a1338SMichael J. Spencer TEST_F(FileSystemTest, DirectoryIteration) {
663c049c657SRafael Espindola   std::error_code ec;
6641f063600SMichael J. Spencer   for (fs::directory_iterator i(".", ec), e; i != e; i.increment(ec))
6651f063600SMichael J. Spencer     ASSERT_NO_ERROR(ec);
6660a7625d6SMichael J. Spencer 
6670a7625d6SMichael J. Spencer   // Create a known hierarchy to recurse over.
6685c20ac01SRafael Espindola   ASSERT_NO_ERROR(
6695c20ac01SRafael Espindola       fs::create_directories(Twine(TestDirectory) + "/recursive/a0/aa1"));
6705c20ac01SRafael Espindola   ASSERT_NO_ERROR(
6715c20ac01SRafael Espindola       fs::create_directories(Twine(TestDirectory) + "/recursive/a0/ab1"));
6725c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::create_directories(Twine(TestDirectory) +
6735c20ac01SRafael Espindola                                          "/recursive/dontlookhere/da1"));
6745c20ac01SRafael Espindola   ASSERT_NO_ERROR(
6755c20ac01SRafael Espindola       fs::create_directories(Twine(TestDirectory) + "/recursive/z0/za1"));
6765c20ac01SRafael Espindola   ASSERT_NO_ERROR(
6775c20ac01SRafael Espindola       fs::create_directories(Twine(TestDirectory) + "/recursive/pop/p1"));
6780a7625d6SMichael J. Spencer   typedef std::vector<std::string> v_t;
6790a7625d6SMichael J. Spencer   v_t visited;
6800a7625d6SMichael J. Spencer   for (fs::recursive_directory_iterator i(Twine(TestDirectory)
6810a7625d6SMichael J. Spencer          + "/recursive", ec), e; i != e; i.increment(ec)){
6820a7625d6SMichael J. Spencer     ASSERT_NO_ERROR(ec);
6830cfb3675SNAKAMURA Takumi     if (path::filename(i->path()) == "p1") {
6840a7625d6SMichael J. Spencer       i.pop();
6850cfb3675SNAKAMURA Takumi       // FIXME: recursive_directory_iterator should be more robust.
6860cfb3675SNAKAMURA Takumi       if (i == e) break;
6870cfb3675SNAKAMURA Takumi     }
6889b54a3e2SMichael J. Spencer     if (path::filename(i->path()) == "dontlookhere")
6899b54a3e2SMichael J. Spencer       i.no_push();
6900a7625d6SMichael J. Spencer     visited.push_back(path::filename(i->path()));
6910a7625d6SMichael J. Spencer   }
6920d955d0bSDavid Majnemer   v_t::const_iterator a0 = find(visited, "a0");
6930d955d0bSDavid Majnemer   v_t::const_iterator aa1 = find(visited, "aa1");
6940d955d0bSDavid Majnemer   v_t::const_iterator ab1 = find(visited, "ab1");
6950d955d0bSDavid Majnemer   v_t::const_iterator dontlookhere = find(visited, "dontlookhere");
6960d955d0bSDavid Majnemer   v_t::const_iterator da1 = find(visited, "da1");
6970d955d0bSDavid Majnemer   v_t::const_iterator z0 = find(visited, "z0");
6980d955d0bSDavid Majnemer   v_t::const_iterator za1 = find(visited, "za1");
6990d955d0bSDavid Majnemer   v_t::const_iterator pop = find(visited, "pop");
7000d955d0bSDavid Majnemer   v_t::const_iterator p1 = find(visited, "p1");
7010a7625d6SMichael J. Spencer 
7020a7625d6SMichael J. Spencer   // Make sure that each path was visited correctly.
7030a7625d6SMichael J. Spencer   ASSERT_NE(a0, visited.end());
7040a7625d6SMichael J. Spencer   ASSERT_NE(aa1, visited.end());
7050a7625d6SMichael J. Spencer   ASSERT_NE(ab1, visited.end());
7060a7625d6SMichael J. Spencer   ASSERT_NE(dontlookhere, visited.end());
7070a7625d6SMichael J. Spencer   ASSERT_EQ(da1, visited.end()); // Not visited.
7080a7625d6SMichael J. Spencer   ASSERT_NE(z0, visited.end());
7090a7625d6SMichael J. Spencer   ASSERT_NE(za1, visited.end());
7100a7625d6SMichael J. Spencer   ASSERT_NE(pop, visited.end());
7110a7625d6SMichael J. Spencer   ASSERT_EQ(p1, visited.end()); // Not visited.
7120a7625d6SMichael J. Spencer 
7130a7625d6SMichael J. Spencer   // Make sure that parents were visited before children. No other ordering
7140a7625d6SMichael J. Spencer   // guarantees can be made across siblings.
7150a7625d6SMichael J. Spencer   ASSERT_LT(a0, aa1);
7160a7625d6SMichael J. Spencer   ASSERT_LT(a0, ab1);
7170a7625d6SMichael J. Spencer   ASSERT_LT(z0, za1);
71878dcc03cSRafael Espindola 
71978dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/a0/aa1"));
72078dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/a0/ab1"));
72178dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/a0"));
72278dcc03cSRafael Espindola   ASSERT_NO_ERROR(
72378dcc03cSRafael Espindola       fs::remove(Twine(TestDirectory) + "/recursive/dontlookhere/da1"));
72478dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/dontlookhere"));
72578dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/pop/p1"));
72678dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/pop"));
72778dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/z0/za1"));
72878dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/z0"));
72978dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive"));
730ead771cbSBruno Cardoso Lopes 
731ead771cbSBruno Cardoso Lopes   // Test recursive_directory_iterator level()
732ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(
733ead771cbSBruno Cardoso Lopes       fs::create_directories(Twine(TestDirectory) + "/reclevel/a/b/c"));
734ead771cbSBruno Cardoso Lopes   fs::recursive_directory_iterator I(Twine(TestDirectory) + "/reclevel", ec), E;
735ead771cbSBruno Cardoso Lopes   for (int l = 0; I != E; I.increment(ec), ++l) {
736ead771cbSBruno Cardoso Lopes     ASSERT_NO_ERROR(ec);
737ead771cbSBruno Cardoso Lopes     EXPECT_EQ(I.level(), l);
738ead771cbSBruno Cardoso Lopes   }
739ead771cbSBruno Cardoso Lopes   EXPECT_EQ(I, E);
740ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel/a/b/c"));
741ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel/a/b"));
742ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel/a"));
743ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel"));
7447ecd94ccSMichael J. Spencer }
7451d3c4a7bSMichael J. Spencer 
746*260bda3fSZachary Turner TEST_F(FileSystemTest, Remove) {
747*260bda3fSZachary Turner   SmallString<64> BaseDir;
748*260bda3fSZachary Turner   SmallString<64> Paths[4];
749*260bda3fSZachary Turner   int fds[4];
750*260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueDirectory("fs_remove", BaseDir));
751*260bda3fSZachary Turner 
752*260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::create_directories(Twine(BaseDir) + "/foo/bar/baz"));
753*260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::create_directories(Twine(BaseDir) + "/foo/bar/buzz"));
754*260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueFile(
755*260bda3fSZachary Turner       Twine(BaseDir) + "/foo/bar/baz/%%%%%%.tmp", fds[0], Paths[0]));
756*260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueFile(
757*260bda3fSZachary Turner       Twine(BaseDir) + "/foo/bar/baz/%%%%%%.tmp", fds[1], Paths[1]));
758*260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueFile(
759*260bda3fSZachary Turner       Twine(BaseDir) + "/foo/bar/buzz/%%%%%%.tmp", fds[2], Paths[2]));
760*260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueFile(
761*260bda3fSZachary Turner       Twine(BaseDir) + "/foo/bar/buzz/%%%%%%.tmp", fds[3], Paths[3]));
762*260bda3fSZachary Turner 
763*260bda3fSZachary Turner   for (int fd : fds)
764*260bda3fSZachary Turner     ::close(fd);
765*260bda3fSZachary Turner 
766*260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Twine(BaseDir) + "/foo/bar/baz"));
767*260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Twine(BaseDir) + "/foo/bar/buzz"));
768*260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Paths[0]));
769*260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Paths[1]));
770*260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Paths[2]));
771*260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Paths[3]));
772*260bda3fSZachary Turner 
773*260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::remove_directories("D:/footest"));
774*260bda3fSZachary Turner 
775*260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::remove_directories(BaseDir));
776*260bda3fSZachary Turner   ASSERT_FALSE(fs::exists(BaseDir));
777*260bda3fSZachary Turner }
778*260bda3fSZachary Turner 
7795e3de7a7SRui Ueyama const char archive[] = "!<arch>\x0A";
7805e3de7a7SRui Ueyama const char bitcode[] = "\xde\xc0\x17\x0b";
781829c4392SRui Ueyama const char coff_object[] = "\x00\x00......";
7825c69ff5cSRui Ueyama const char coff_bigobj[] = "\x00\x00\xff\xff\x00\x02......"
7832acb0584SRui Ueyama     "\xc7\xa1\xba\xd1\xee\xba\xa9\x4b\xaf\x20\xfa\xf6\x6a\xa4\xdc\xb8";
784e448f9e4SRui Ueyama const char coff_import_library[] = "\x00\x00\xff\xff....";
7855e3de7a7SRui Ueyama const char elf_relocatable[] = { 0x7f, 'E', 'L', 'F', 1, 2, 1, 0, 0,
7865e3de7a7SRui Ueyama                                  0,    0,   0,   0,   0, 0, 0, 0, 1 };
7871562f69fSEtienne Bergeron const char macho_universal_binary[] = "\xca\xfe\xba\xbe...\x00";
78887c85b7eSKevin Enderby const char macho_object[] =
78987c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x01............";
79087c85b7eSKevin Enderby const char macho_executable[] =
79187c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x02............";
7925e3de7a7SRui Ueyama const char macho_fixed_virtual_memory_shared_lib[] =
79387c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x03............";
79487c85b7eSKevin Enderby const char macho_core[] =
79587c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x04............";
79687c85b7eSKevin Enderby const char macho_preload_executable[] =
79787c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x05............";
7985e3de7a7SRui Ueyama const char macho_dynamically_linked_shared_lib[] =
79987c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x06............";
80087c85b7eSKevin Enderby const char macho_dynamic_linker[] =
80187c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x07............";
80287c85b7eSKevin Enderby const char macho_bundle[] =
80387c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x08............";
80487c85b7eSKevin Enderby const char macho_dsym_companion[] =
80587c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x0a............";
80687c85b7eSKevin Enderby const char macho_kext_bundle[] =
80787c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x0b............";
8085e3de7a7SRui Ueyama const char windows_resource[] = "\x00\x00\x00\x00\x020\x00\x00\x00\xff";
8092d2b254eSNick Kledzik const char macho_dynamically_linked_shared_lib_stub[] =
81087c85b7eSKevin Enderby     "\xfe\xed\xfa\xce........\x00\x00\x00\x09............";
811b8055cbcSMichael J. Spencer 
8121d3c4a7bSMichael J. Spencer TEST_F(FileSystemTest, Magic) {
8131d3c4a7bSMichael J. Spencer   struct type {
8141d3c4a7bSMichael J. Spencer     const char *filename;
8151d3c4a7bSMichael J. Spencer     const char *magic_str;
8161d3c4a7bSMichael J. Spencer     size_t magic_str_len;
817b8055cbcSMichael J. Spencer     fs::file_magic magic;
818b8055cbcSMichael J. Spencer   } types[] = {
8195e3de7a7SRui Ueyama #define DEFINE(magic)                                           \
8205e3de7a7SRui Ueyama     { #magic, magic, sizeof(magic), fs::file_magic::magic }
8215e3de7a7SRui Ueyama     DEFINE(archive),
8225e3de7a7SRui Ueyama     DEFINE(bitcode),
823829c4392SRui Ueyama     DEFINE(coff_object),
8242acb0584SRui Ueyama     { "coff_bigobj", coff_bigobj, sizeof(coff_bigobj), fs::file_magic::coff_object },
825e448f9e4SRui Ueyama     DEFINE(coff_import_library),
8265e3de7a7SRui Ueyama     DEFINE(elf_relocatable),
8275e3de7a7SRui Ueyama     DEFINE(macho_universal_binary),
8285e3de7a7SRui Ueyama     DEFINE(macho_object),
8295e3de7a7SRui Ueyama     DEFINE(macho_executable),
8305e3de7a7SRui Ueyama     DEFINE(macho_fixed_virtual_memory_shared_lib),
8315e3de7a7SRui Ueyama     DEFINE(macho_core),
8325e3de7a7SRui Ueyama     DEFINE(macho_preload_executable),
8335e3de7a7SRui Ueyama     DEFINE(macho_dynamically_linked_shared_lib),
8345e3de7a7SRui Ueyama     DEFINE(macho_dynamic_linker),
8355e3de7a7SRui Ueyama     DEFINE(macho_bundle),
8362d2b254eSNick Kledzik     DEFINE(macho_dynamically_linked_shared_lib_stub),
8375e3de7a7SRui Ueyama     DEFINE(macho_dsym_companion),
838a7ad4b3fSJustin Bogner     DEFINE(macho_kext_bundle),
8395e3de7a7SRui Ueyama     DEFINE(windows_resource)
8405e3de7a7SRui Ueyama #undef DEFINE
841b8055cbcSMichael J. Spencer     };
8421d3c4a7bSMichael J. Spencer 
8431d3c4a7bSMichael J. Spencer   // Create some files filled with magic.
8441d3c4a7bSMichael J. Spencer   for (type *i = types, *e = types + (sizeof(types) / sizeof(type)); i != e;
8451d3c4a7bSMichael J. Spencer                                                                      ++i) {
8461d3c4a7bSMichael J. Spencer     SmallString<128> file_pathname(TestDirectory);
8471d3c4a7bSMichael J. Spencer     path::append(file_pathname, i->filename);
8483fd1e993SRafael Espindola     std::error_code EC;
8493fd1e993SRafael Espindola     raw_fd_ostream file(file_pathname, EC, sys::fs::F_None);
8501d3c4a7bSMichael J. Spencer     ASSERT_FALSE(file.has_error());
8511d3c4a7bSMichael J. Spencer     StringRef magic(i->magic_str, i->magic_str_len);
8521d3c4a7bSMichael J. Spencer     file << magic;
853b587180fSMichael J. Spencer     file.close();
854b8055cbcSMichael J. Spencer     EXPECT_EQ(i->magic, fs::identify_magic(magic));
85578dcc03cSRafael Espindola     ASSERT_NO_ERROR(fs::remove(Twine(file_pathname)));
8567ecd94ccSMichael J. Spencer   }
8573ef91c55SMichael J. Spencer }
8583ef91c55SMichael J. Spencer 
85984ab9b3bSRafael Espindola #ifdef LLVM_ON_WIN32
86084ab9b3bSRafael Espindola TEST_F(FileSystemTest, CarriageReturn) {
86184ab9b3bSRafael Espindola   SmallString<128> FilePathname(TestDirectory);
862d83c63b7SReid Kleckner   std::error_code EC;
86384ab9b3bSRafael Espindola   path::append(FilePathname, "test");
86484ab9b3bSRafael Espindola 
86584ab9b3bSRafael Espindola   {
866d83c63b7SReid Kleckner     raw_fd_ostream File(FilePathname, EC, sys::fs::F_Text);
867d83c63b7SReid Kleckner     ASSERT_NO_ERROR(EC);
86884ab9b3bSRafael Espindola     File << '\n';
86984ab9b3bSRafael Espindola   }
87084ab9b3bSRafael Espindola   {
871d83c63b7SReid Kleckner     auto Buf = MemoryBuffer::getFile(FilePathname.str());
872586ee604SAaron Ballman     EXPECT_TRUE((bool)Buf);
873e56fe8afSAaron Ballman     EXPECT_EQ(Buf.get()->getBuffer(), "\r\n");
87484ab9b3bSRafael Espindola   }
87584ab9b3bSRafael Espindola 
87684ab9b3bSRafael Espindola   {
877d83c63b7SReid Kleckner     raw_fd_ostream File(FilePathname, EC, sys::fs::F_None);
878d83c63b7SReid Kleckner     ASSERT_NO_ERROR(EC);
87984ab9b3bSRafael Espindola     File << '\n';
88084ab9b3bSRafael Espindola   }
88184ab9b3bSRafael Espindola   {
882d83c63b7SReid Kleckner     auto Buf = MemoryBuffer::getFile(FilePathname.str());
883586ee604SAaron Ballman     EXPECT_TRUE((bool)Buf);
884e56fe8afSAaron Ballman     EXPECT_EQ(Buf.get()->getBuffer(), "\n");
88584ab9b3bSRafael Espindola   }
88678dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(FilePathname)));
88784ab9b3bSRafael Espindola }
88884ab9b3bSRafael Espindola #endif
88984ab9b3bSRafael Espindola 
89059aaa6c0SRafael Espindola TEST_F(FileSystemTest, Resize) {
89159aaa6c0SRafael Espindola   int FD;
89259aaa6c0SRafael Espindola   SmallString<64> TempPath;
89359aaa6c0SRafael Espindola   ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "temp", FD, TempPath));
89459aaa6c0SRafael Espindola   ASSERT_NO_ERROR(fs::resize_file(FD, 123));
89559aaa6c0SRafael Espindola   fs::file_status Status;
89659aaa6c0SRafael Espindola   ASSERT_NO_ERROR(fs::status(FD, Status));
89759aaa6c0SRafael Espindola   ASSERT_EQ(Status.getSize(), 123U);
89875e557f1SReid Kleckner   ::close(FD);
89975e557f1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(TempPath));
90059aaa6c0SRafael Espindola }
90159aaa6c0SRafael Espindola 
90218497e92SNick Kledzik TEST_F(FileSystemTest, FileMapping) {
90318497e92SNick Kledzik   // Create a temp file.
90418497e92SNick Kledzik   int FileDescriptor;
90518497e92SNick Kledzik   SmallString<64> TempPath;
90618497e92SNick Kledzik   ASSERT_NO_ERROR(
907155cf0f3SRafael Espindola       fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
908c69f13bfSRafael Espindola   unsigned Size = 4096;
909c69f13bfSRafael Espindola   ASSERT_NO_ERROR(fs::resize_file(FileDescriptor, Size));
910c69f13bfSRafael Espindola 
91118497e92SNick Kledzik   // Map in temp file and add some content
912c049c657SRafael Espindola   std::error_code EC;
913ef2284fbSMichael J. Spencer   StringRef Val("hello there");
914ef2284fbSMichael J. Spencer   {
915ef2284fbSMichael J. Spencer     fs::mapped_file_region mfr(FileDescriptor,
916c69f13bfSRafael Espindola                                fs::mapped_file_region::readwrite, Size, 0, EC);
917ef2284fbSMichael J. Spencer     ASSERT_NO_ERROR(EC);
918ef2284fbSMichael J. Spencer     std::copy(Val.begin(), Val.end(), mfr.data());
919ef2284fbSMichael J. Spencer     // Explicitly add a 0.
920ef2284fbSMichael J. Spencer     mfr.data()[Val.size()] = 0;
92118497e92SNick Kledzik     // Unmap temp file
922ef2284fbSMichael J. Spencer   }
92375e557f1SReid Kleckner   ASSERT_EQ(close(FileDescriptor), 0);
92418497e92SNick Kledzik 
92518497e92SNick Kledzik   // Map it back in read-only
92675e557f1SReid Kleckner   {
92771bc507cSRafael Espindola     int FD;
92871bc507cSRafael Espindola     EC = fs::openFileForRead(Twine(TempPath), FD);
92971bc507cSRafael Espindola     ASSERT_NO_ERROR(EC);
930c69f13bfSRafael Espindola     fs::mapped_file_region mfr(FD, fs::mapped_file_region::readonly, Size, 0, EC);
931ef2284fbSMichael J. Spencer     ASSERT_NO_ERROR(EC);
93218497e92SNick Kledzik 
93318497e92SNick Kledzik     // Verify content
934ef2284fbSMichael J. Spencer     EXPECT_EQ(StringRef(mfr.const_data()), Val);
93518497e92SNick Kledzik 
93618497e92SNick Kledzik     // Unmap temp file
937c69f13bfSRafael Espindola     fs::mapped_file_region m(FD, fs::mapped_file_region::readonly, Size, 0, EC);
938ef2284fbSMichael J. Spencer     ASSERT_NO_ERROR(EC);
93971bc507cSRafael Espindola     ASSERT_EQ(close(FD), 0);
940ef2284fbSMichael J. Spencer   }
94175e557f1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(TempPath));
94275e557f1SReid Kleckner }
943afc50b3eSSaleem Abdulrasool 
944afc50b3eSSaleem Abdulrasool TEST(Support, NormalizePath) {
945afc50b3eSSaleem Abdulrasool #if defined(LLVM_ON_WIN32)
946afc50b3eSSaleem Abdulrasool #define EXPECT_PATH_IS(path__, windows__, not_windows__)                        \
947afc50b3eSSaleem Abdulrasool   EXPECT_EQ(path__, windows__);
948afc50b3eSSaleem Abdulrasool #else
949afc50b3eSSaleem Abdulrasool #define EXPECT_PATH_IS(path__, windows__, not_windows__)                        \
950afc50b3eSSaleem Abdulrasool   EXPECT_EQ(path__, not_windows__);
951afc50b3eSSaleem Abdulrasool #endif
952afc50b3eSSaleem Abdulrasool 
953afc50b3eSSaleem Abdulrasool   SmallString<64> Path1("a");
954afc50b3eSSaleem Abdulrasool   SmallString<64> Path2("a/b");
955afc50b3eSSaleem Abdulrasool   SmallString<64> Path3("a\\b");
956afc50b3eSSaleem Abdulrasool   SmallString<64> Path4("a\\\\b");
957afc50b3eSSaleem Abdulrasool   SmallString<64> Path5("\\a");
958afc50b3eSSaleem Abdulrasool   SmallString<64> Path6("a\\");
959afc50b3eSSaleem Abdulrasool 
9607e774c24SRafael Espindola   path::native(Path1);
961afc50b3eSSaleem Abdulrasool   EXPECT_PATH_IS(Path1, "a", "a");
962afc50b3eSSaleem Abdulrasool 
9637e774c24SRafael Espindola   path::native(Path2);
964ffbabb79SRafael Espindola   EXPECT_PATH_IS(Path2, "a\\b", "a/b");
965afc50b3eSSaleem Abdulrasool 
9667e774c24SRafael Espindola   path::native(Path3);
967afc50b3eSSaleem Abdulrasool   EXPECT_PATH_IS(Path3, "a\\b", "a/b");
968afc50b3eSSaleem Abdulrasool 
9697e774c24SRafael Espindola   path::native(Path4);
970afc50b3eSSaleem Abdulrasool   EXPECT_PATH_IS(Path4, "a\\\\b", "a\\\\b");
971afc50b3eSSaleem Abdulrasool 
9727e774c24SRafael Espindola   path::native(Path5);
973afc50b3eSSaleem Abdulrasool   EXPECT_PATH_IS(Path5, "\\a", "/a");
974afc50b3eSSaleem Abdulrasool 
9757e774c24SRafael Espindola   path::native(Path6);
976afc50b3eSSaleem Abdulrasool   EXPECT_PATH_IS(Path6, "a\\", "a/");
977afc50b3eSSaleem Abdulrasool 
978afc50b3eSSaleem Abdulrasool #undef EXPECT_PATH_IS
9799c761a36SSerge Pavlov 
9809c761a36SSerge Pavlov #if defined(LLVM_ON_WIN32)
9819c761a36SSerge Pavlov   SmallString<64> PathHome;
9829c761a36SSerge Pavlov   path::home_directory(PathHome);
9839c761a36SSerge Pavlov 
9849c761a36SSerge Pavlov   const char *Path7a = "~/aaa";
9859c761a36SSerge Pavlov   SmallString<64> Path7(Path7a);
9869c761a36SSerge Pavlov   path::native(Path7);
9879c761a36SSerge Pavlov   EXPECT_TRUE(Path7.endswith("\\aaa"));
9889c761a36SSerge Pavlov   EXPECT_TRUE(Path7.startswith(PathHome));
9899c761a36SSerge Pavlov   EXPECT_EQ(Path7.size(), PathHome.size() + strlen(Path7a + 1));
9909c761a36SSerge Pavlov 
9919c761a36SSerge Pavlov   const char *Path8a = "~";
9929c761a36SSerge Pavlov   SmallString<64> Path8(Path8a);
9939c761a36SSerge Pavlov   path::native(Path8);
9949c761a36SSerge Pavlov   EXPECT_EQ(Path8, PathHome);
9959c761a36SSerge Pavlov 
9969c761a36SSerge Pavlov   const char *Path9a = "~aaa";
9979c761a36SSerge Pavlov   SmallString<64> Path9(Path9a);
9989c761a36SSerge Pavlov   path::native(Path9);
9999c761a36SSerge Pavlov   EXPECT_EQ(Path9, "~aaa");
10009c761a36SSerge Pavlov 
10019c761a36SSerge Pavlov   const char *Path10a = "aaa/~/b";
10029c761a36SSerge Pavlov   SmallString<64> Path10(Path10a);
10039c761a36SSerge Pavlov   path::native(Path10);
10049c761a36SSerge Pavlov   EXPECT_EQ(Path10, "aaa\\~\\b");
10059c761a36SSerge Pavlov #endif
1006afc50b3eSSaleem Abdulrasool }
1007a26be4a9SDouglas Katzman 
1008a26be4a9SDouglas Katzman TEST(Support, RemoveLeadingDotSlash) {
1009a26be4a9SDouglas Katzman   StringRef Path1("././/foolz/wat");
1010a26be4a9SDouglas Katzman   StringRef Path2("./////");
1011a26be4a9SDouglas Katzman 
1012a26be4a9SDouglas Katzman   Path1 = path::remove_leading_dotslash(Path1);
1013a26be4a9SDouglas Katzman   EXPECT_EQ(Path1, "foolz/wat");
1014a26be4a9SDouglas Katzman   Path2 = path::remove_leading_dotslash(Path2);
1015a26be4a9SDouglas Katzman   EXPECT_EQ(Path2, "");
1016a26be4a9SDouglas Katzman }
1017662b4fd3SMike Aizatsky 
1018662b4fd3SMike Aizatsky static std::string remove_dots(StringRef path,
1019662b4fd3SMike Aizatsky     bool remove_dot_dot) {
1020662b4fd3SMike Aizatsky   SmallString<256> buffer(path);
1021662b4fd3SMike Aizatsky   path::remove_dots(buffer, remove_dot_dot);
1022662b4fd3SMike Aizatsky   return buffer.str();
1023662b4fd3SMike Aizatsky }
1024662b4fd3SMike Aizatsky 
1025662b4fd3SMike Aizatsky TEST(Support, RemoveDots) {
1026f8cf7133SMike Aizatsky #if defined(LLVM_ON_WIN32)
1027f8cf7133SMike Aizatsky   EXPECT_EQ("foolz\\wat", remove_dots(".\\.\\\\foolz\\wat", false));
1028f8cf7133SMike Aizatsky   EXPECT_EQ("", remove_dots(".\\\\\\\\\\", false));
1029f8cf7133SMike Aizatsky 
1030f8cf7133SMike Aizatsky   EXPECT_EQ("a\\..\\b\\c", remove_dots(".\\a\\..\\b\\c", false));
1031f8cf7133SMike Aizatsky   EXPECT_EQ("b\\c", remove_dots(".\\a\\..\\b\\c", true));
1032f8cf7133SMike Aizatsky   EXPECT_EQ("c", remove_dots(".\\.\\c", true));
1033af5a28feSEric Liu   EXPECT_EQ("..\\a\\c", remove_dots("..\\a\\b\\..\\c", true));
1034af5a28feSEric Liu   EXPECT_EQ("..\\..\\a\\c", remove_dots("..\\..\\a\\b\\..\\c", true));
1035f8cf7133SMike Aizatsky 
1036f8cf7133SMike Aizatsky   SmallString<64> Path1(".\\.\\c");
1037f8cf7133SMike Aizatsky   EXPECT_TRUE(path::remove_dots(Path1, true));
1038f8cf7133SMike Aizatsky   EXPECT_EQ("c", Path1);
1039f8cf7133SMike Aizatsky #else
1040662b4fd3SMike Aizatsky   EXPECT_EQ("foolz/wat", remove_dots("././/foolz/wat", false));
1041662b4fd3SMike Aizatsky   EXPECT_EQ("", remove_dots("./////", false));
1042662b4fd3SMike Aizatsky 
1043662b4fd3SMike Aizatsky   EXPECT_EQ("a/../b/c", remove_dots("./a/../b/c", false));
1044662b4fd3SMike Aizatsky   EXPECT_EQ("b/c", remove_dots("./a/../b/c", true));
1045662b4fd3SMike Aizatsky   EXPECT_EQ("c", remove_dots("././c", true));
1046af5a28feSEric Liu   EXPECT_EQ("../a/c", remove_dots("../a/b/../c", true));
1047af5a28feSEric Liu   EXPECT_EQ("../../a/c", remove_dots("../../a/b/../c", true));
1048937dd7afSBenjamin Kramer   EXPECT_EQ("/a/c", remove_dots("/../../a/c", true));
1049937dd7afSBenjamin Kramer   EXPECT_EQ("/a/c", remove_dots("/../a/b//../././/c", true));
1050662b4fd3SMike Aizatsky 
1051662b4fd3SMike Aizatsky   SmallString<64> Path1("././c");
1052662b4fd3SMike Aizatsky   EXPECT_TRUE(path::remove_dots(Path1, true));
1053662b4fd3SMike Aizatsky   EXPECT_EQ("c", Path1);
1054f8cf7133SMike Aizatsky #endif
1055662b4fd3SMike Aizatsky }
1056bbd10b45STeresa Johnson 
1057bbd10b45STeresa Johnson TEST(Support, ReplacePathPrefix) {
1058bbd10b45STeresa Johnson   SmallString<64> Path1("/foo");
1059bbd10b45STeresa Johnson   SmallString<64> Path2("/old/foo");
1060bbd10b45STeresa Johnson   SmallString<64> OldPrefix("/old");
1061bbd10b45STeresa Johnson   SmallString<64> NewPrefix("/new");
1062bbd10b45STeresa Johnson   SmallString<64> NewPrefix2("/longernew");
1063bbd10b45STeresa Johnson   SmallString<64> EmptyPrefix("");
1064bbd10b45STeresa Johnson 
1065bbd10b45STeresa Johnson   SmallString<64> Path = Path1;
1066bbd10b45STeresa Johnson   path::replace_path_prefix(Path, OldPrefix, NewPrefix);
1067bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/foo");
1068bbd10b45STeresa Johnson   Path = Path2;
1069bbd10b45STeresa Johnson   path::replace_path_prefix(Path, OldPrefix, NewPrefix);
1070bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/new/foo");
1071bbd10b45STeresa Johnson   Path = Path2;
1072bbd10b45STeresa Johnson   path::replace_path_prefix(Path, OldPrefix, NewPrefix2);
1073bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/longernew/foo");
1074bbd10b45STeresa Johnson   Path = Path1;
1075bbd10b45STeresa Johnson   path::replace_path_prefix(Path, EmptyPrefix, NewPrefix);
1076bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/new/foo");
1077bbd10b45STeresa Johnson   Path = Path2;
1078bbd10b45STeresa Johnson   path::replace_path_prefix(Path, OldPrefix, EmptyPrefix);
1079bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/foo");
1080bbd10b45STeresa Johnson }
1081d9153272STaewook Oh 
1082d9153272STaewook Oh TEST_F(FileSystemTest, PathFromFD) {
1083d9153272STaewook Oh   // Create a temp file.
1084d9153272STaewook Oh   int FileDescriptor;
1085d9153272STaewook Oh   SmallString<64> TempPath;
1086d9153272STaewook Oh   ASSERT_NO_ERROR(
1087d9153272STaewook Oh       fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
108875e557f1SReid Kleckner   FileRemover Cleanup(TempPath);
1089d9153272STaewook Oh 
1090d9153272STaewook Oh   // Make sure it exists.
1091d9153272STaewook Oh   ASSERT_TRUE(sys::fs::exists(Twine(TempPath)));
1092d9153272STaewook Oh 
1093d9153272STaewook Oh   // Try to get the path from the file descriptor
1094d9153272STaewook Oh   SmallString<64> ResultPath;
1095d9153272STaewook Oh   std::error_code ErrorCode =
1096d9153272STaewook Oh       fs::getPathFromOpenFD(FileDescriptor, ResultPath);
1097d9153272STaewook Oh 
1098d9153272STaewook Oh   // If we succeeded, check that the paths are the same (modulo case):
1099d9153272STaewook Oh   if (!ErrorCode) {
1100d9153272STaewook Oh     // The paths returned by createTemporaryFile and getPathFromOpenFD
1101d9153272STaewook Oh     // should reference the same file on disk.
1102d9153272STaewook Oh     fs::UniqueID D1, D2;
1103d9153272STaewook Oh     ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), D1));
1104d9153272STaewook Oh     ASSERT_NO_ERROR(fs::getUniqueID(Twine(ResultPath), D2));
1105d9153272STaewook Oh     ASSERT_EQ(D1, D2);
1106d9153272STaewook Oh   }
1107d9153272STaewook Oh 
1108d9153272STaewook Oh   ::close(FileDescriptor);
1109d9153272STaewook Oh }
1110d9153272STaewook Oh 
11113dd74b8eSAaron Ballman TEST_F(FileSystemTest, PathFromFDWin32) {
11123dd74b8eSAaron Ballman   // Create a temp file.
11133dd74b8eSAaron Ballman   int FileDescriptor;
11143dd74b8eSAaron Ballman   SmallString<64> TempPath;
11153dd74b8eSAaron Ballman   ASSERT_NO_ERROR(
11163dd74b8eSAaron Ballman     fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
111775e557f1SReid Kleckner   FileRemover Cleanup(TempPath);
11183dd74b8eSAaron Ballman 
11193dd74b8eSAaron Ballman   // Make sure it exists.
11203dd74b8eSAaron Ballman   ASSERT_TRUE(sys::fs::exists(Twine(TempPath)));
11213dd74b8eSAaron Ballman 
11223dd74b8eSAaron Ballman   SmallVector<char, 8> ResultPath;
11233dd74b8eSAaron Ballman   std::error_code ErrorCode =
11243dd74b8eSAaron Ballman     fs::getPathFromOpenFD(FileDescriptor, ResultPath);
11253dd74b8eSAaron Ballman 
11263dd74b8eSAaron Ballman   if (!ErrorCode) {
11273dd74b8eSAaron Ballman     // Now that we know how much space is required for the path, create a path
11283dd74b8eSAaron Ballman     // buffer with exactly enough space (sans null terminator, which should not
11293dd74b8eSAaron Ballman     // be present), and call getPathFromOpenFD again to ensure that the API
11303dd74b8eSAaron Ballman     // properly handles exactly-sized buffers.
11313dd74b8eSAaron Ballman     SmallVector<char, 8> ExactSizedPath(ResultPath.size());
11323dd74b8eSAaron Ballman     ErrorCode = fs::getPathFromOpenFD(FileDescriptor, ExactSizedPath);
11333dd74b8eSAaron Ballman     ResultPath = ExactSizedPath;
11343dd74b8eSAaron Ballman   }
11353dd74b8eSAaron Ballman 
11363dd74b8eSAaron Ballman   if (!ErrorCode) {
11373dd74b8eSAaron Ballman     fs::UniqueID D1, D2;
11383dd74b8eSAaron Ballman     ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), D1));
11393dd74b8eSAaron Ballman     ASSERT_NO_ERROR(fs::getUniqueID(Twine(ResultPath), D2));
11403dd74b8eSAaron Ballman     ASSERT_EQ(D1, D2);
11413dd74b8eSAaron Ballman   }
11423dd74b8eSAaron Ballman   ::close(FileDescriptor);
11433dd74b8eSAaron Ballman }
11443dd74b8eSAaron Ballman 
11450ad00462SAaron Ballman TEST_F(FileSystemTest, PathFromFDUnicode) {
11460ad00462SAaron Ballman   // Create a temp file.
11470ad00462SAaron Ballman   int FileDescriptor;
11480ad00462SAaron Ballman   SmallString<64> TempPath;
11490ad00462SAaron Ballman 
11500ad00462SAaron Ballman   // Test Unicode: "<temp directory>/(pi)r^2<temp rand chars>.aleth.0"
11510ad00462SAaron Ballman   ASSERT_NO_ERROR(
11520ad00462SAaron Ballman     fs::createTemporaryFile("\xCF\x80r\xC2\xB2",
11530ad00462SAaron Ballman                             "\xE2\x84\xB5.0", FileDescriptor, TempPath));
115475e557f1SReid Kleckner   FileRemover Cleanup(TempPath);
11550ad00462SAaron Ballman 
11560ad00462SAaron Ballman   // Make sure it exists.
11570ad00462SAaron Ballman   ASSERT_TRUE(sys::fs::exists(Twine(TempPath)));
11580ad00462SAaron Ballman 
11590ad00462SAaron Ballman   SmallVector<char, 8> ResultPath;
11600ad00462SAaron Ballman   std::error_code ErrorCode =
11610ad00462SAaron Ballman     fs::getPathFromOpenFD(FileDescriptor, ResultPath);
11620ad00462SAaron Ballman 
11630ad00462SAaron Ballman   if (!ErrorCode) {
11640ad00462SAaron Ballman     fs::UniqueID D1, D2;
11650ad00462SAaron Ballman     ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), D1));
11660ad00462SAaron Ballman     ASSERT_NO_ERROR(fs::getUniqueID(Twine(ResultPath), D2));
11670ad00462SAaron Ballman     ASSERT_EQ(D1, D2);
11680ad00462SAaron Ballman   }
11690ad00462SAaron Ballman   ::close(FileDescriptor);
11700ad00462SAaron Ballman }
11710ad00462SAaron Ballman 
1172d9153272STaewook Oh TEST_F(FileSystemTest, OpenFileForRead) {
1173d9153272STaewook Oh   // Create a temp file.
1174d9153272STaewook Oh   int FileDescriptor;
1175d9153272STaewook Oh   SmallString<64> TempPath;
1176d9153272STaewook Oh   ASSERT_NO_ERROR(
1177d9153272STaewook Oh       fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
117875e557f1SReid Kleckner   FileRemover Cleanup(TempPath);
1179d9153272STaewook Oh 
1180d9153272STaewook Oh   // Make sure it exists.
1181d9153272STaewook Oh   ASSERT_TRUE(sys::fs::exists(Twine(TempPath)));
1182d9153272STaewook Oh 
1183d9153272STaewook Oh   // Open the file for read
1184d9153272STaewook Oh   int FileDescriptor2;
1185d9153272STaewook Oh   SmallString<64> ResultPath;
1186d9153272STaewook Oh   ASSERT_NO_ERROR(
1187d9153272STaewook Oh       fs::openFileForRead(Twine(TempPath), FileDescriptor2, &ResultPath))
1188d9153272STaewook Oh 
1189d9153272STaewook Oh   // If we succeeded, check that the paths are the same (modulo case):
1190d9153272STaewook Oh   if (!ResultPath.empty()) {
1191d9153272STaewook Oh     // The paths returned by createTemporaryFile and getPathFromOpenFD
1192d9153272STaewook Oh     // should reference the same file on disk.
1193d9153272STaewook Oh     fs::UniqueID D1, D2;
1194d9153272STaewook Oh     ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), D1));
1195d9153272STaewook Oh     ASSERT_NO_ERROR(fs::getUniqueID(Twine(ResultPath), D2));
1196d9153272STaewook Oh     ASSERT_EQ(D1, D2);
1197d9153272STaewook Oh   }
1198d9153272STaewook Oh 
1199d9153272STaewook Oh   ::close(FileDescriptor);
1200d9153272STaewook Oh }
12012f096097SPavel Labath 
12022f096097SPavel Labath TEST_F(FileSystemTest, set_current_path) {
12032f096097SPavel Labath   SmallString<128> path;
12042f096097SPavel Labath 
12052f096097SPavel Labath   ASSERT_NO_ERROR(fs::current_path(path));
12062f096097SPavel Labath   ASSERT_NE(TestDirectory, path);
12072f096097SPavel Labath 
12082f096097SPavel Labath   struct RestorePath {
12092f096097SPavel Labath     SmallString<128> path;
12102f096097SPavel Labath     RestorePath(const SmallString<128> &path) : path(path) {}
12112f096097SPavel Labath     ~RestorePath() { fs::set_current_path(path); }
12122f096097SPavel Labath   } restore_path(path);
12132f096097SPavel Labath 
12142f096097SPavel Labath   ASSERT_NO_ERROR(fs::set_current_path(TestDirectory));
12152f096097SPavel Labath 
12162f096097SPavel Labath   ASSERT_NO_ERROR(fs::current_path(path));
121750440097SPavel Labath 
121850440097SPavel Labath   fs::UniqueID D1, D2;
121950440097SPavel Labath   ASSERT_NO_ERROR(fs::getUniqueID(TestDirectory, D1));
122050440097SPavel Labath   ASSERT_NO_ERROR(fs::getUniqueID(path, D2));
122150440097SPavel Labath   ASSERT_EQ(D1, D2) << "D1: " << TestDirectory << "\nD2: " << path;
12222f096097SPavel Labath }
12232f096097SPavel Labath 
12243ef91c55SMichael J. Spencer } // anonymous namespace
1225