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"
110d955d0bSDavid Majnemer #include "llvm/ADT/STLExtras.h"
129a67b073SChandler Carruth #include "llvm/ADT/SmallVector.h"
13e8bb3920SSimon Dardis #include "llvm/ADT/Triple.h"
14264b5d9eSZachary Turner #include "llvm/BinaryFormat/Magic.h"
15432a3883SNico Weber #include "llvm/Config/llvm-config.h"
167187e4bbSPawel Bylica #include "llvm/Support/ConvertUTF.h"
172a826e40SRafael Espindola #include "llvm/Support/Errc.h"
187ecd94ccSMichael J. Spencer #include "llvm/Support/ErrorHandling.h"
19130cec21SChandler Carruth #include "llvm/Support/FileSystem.h"
2075e557f1SReid Kleckner #include "llvm/Support/FileUtilities.h"
21e8bb3920SSimon Dardis #include "llvm/Support/Host.h"
2284ab9b3bSRafael Espindola #include "llvm/Support/MemoryBuffer.h"
231f063600SMichael J. Spencer #include "llvm/Support/raw_ostream.h"
243ef91c55SMichael J. Spencer #include "gtest/gtest.h"
2560b7132aSPavel Labath #include "gmock/gmock.h"
263ef91c55SMichael J. Spencer 
27712e8d29SNico Weber #ifdef _WIN32
28dd2ca837SNico Weber #include "llvm/ADT/ArrayRef.h"
29089303d8SAndrew Ng #include "llvm/Support/Chrono.h"
30b94f0522SNAKAMURA Takumi #include <windows.h>
31a813d608SRafael Espindola #include <winerror.h>
32a813d608SRafael Espindola #endif
33a813d608SRafael Espindola 
346b9396c0SFrederic Riss #ifdef LLVM_ON_UNIX
35d8350cf3SZachary Turner #include <pwd.h>
366b9396c0SFrederic Riss #include <sys/stat.h>
376b9396c0SFrederic Riss #endif
386b9396c0SFrederic Riss 
393ef91c55SMichael J. Spencer using namespace llvm;
404571040eSMichael J. Spencer using namespace llvm::sys;
413ef91c55SMichael J. Spencer 
423b264ba9SMichael J. Spencer #define ASSERT_NO_ERROR(x)                                                     \
43c049c657SRafael Espindola   if (std::error_code ASSERT_NO_ERROR_ec = x) {                                \
441f063600SMichael J. Spencer     SmallString<128> MessageStorage;                                           \
451f063600SMichael J. Spencer     raw_svector_ostream Message(MessageStorage);                               \
461f063600SMichael J. Spencer     Message << #x ": did not return errc::success.\n"                          \
471f063600SMichael J. Spencer             << "error number: " << ASSERT_NO_ERROR_ec.value() << "\n"          \
481f063600SMichael J. Spencer             << "error message: " << ASSERT_NO_ERROR_ec.message() << "\n";      \
491f063600SMichael J. Spencer     GTEST_FATAL_FAILURE_(MessageStorage.c_str());                              \
50c049c657SRafael Espindola   } else {                                                                     \
51c049c657SRafael Espindola   }
523b264ba9SMichael J. Spencer 
531f67a3cbSZachary Turner #define ASSERT_ERROR(x)                                                        \
541f67a3cbSZachary Turner   if (!x) {                                                                    \
551f67a3cbSZachary Turner     SmallString<128> MessageStorage;                                           \
561f67a3cbSZachary Turner     raw_svector_ostream Message(MessageStorage);                               \
571f67a3cbSZachary Turner     Message << #x ": did not return a failure error code.\n";                  \
581f67a3cbSZachary Turner     GTEST_FATAL_FAILURE_(MessageStorage.c_str());                              \
591f67a3cbSZachary Turner   }
601f67a3cbSZachary Turner 
613ef91c55SMichael J. Spencer namespace {
623ef91c55SMichael J. Spencer 
631f67a3cbSZachary Turner struct FileDescriptorCloser {
641f67a3cbSZachary Turner   explicit FileDescriptorCloser(int FD) : FD(FD) {}
651f67a3cbSZachary Turner   ~FileDescriptorCloser() { ::close(FD); }
661f67a3cbSZachary Turner   int FD;
671f67a3cbSZachary Turner };
681f67a3cbSZachary Turner 
69606bb1a2SZhanyong Wan TEST(is_separator, Works) {
70606bb1a2SZhanyong Wan   EXPECT_TRUE(path::is_separator('/'));
71606bb1a2SZhanyong Wan   EXPECT_FALSE(path::is_separator('\0'));
72606bb1a2SZhanyong Wan   EXPECT_FALSE(path::is_separator('-'));
73606bb1a2SZhanyong Wan   EXPECT_FALSE(path::is_separator(' '));
74606bb1a2SZhanyong Wan 
755c5091fcSZachary Turner   EXPECT_TRUE(path::is_separator('\\', path::Style::windows));
765c5091fcSZachary Turner   EXPECT_FALSE(path::is_separator('\\', path::Style::posix));
775c5091fcSZachary Turner 
78712e8d29SNico Weber #ifdef _WIN32
79606bb1a2SZhanyong Wan   EXPECT_TRUE(path::is_separator('\\'));
80606bb1a2SZhanyong Wan #else
8178d15af1SZachary Turner   EXPECT_FALSE(path::is_separator('\\'));
82606bb1a2SZhanyong Wan #endif
83606bb1a2SZhanyong Wan }
84606bb1a2SZhanyong Wan 
853ef91c55SMichael J. Spencer TEST(Support, Path) {
863ef91c55SMichael J. Spencer   SmallVector<StringRef, 40> paths;
873ef91c55SMichael J. Spencer   paths.push_back("");
883ef91c55SMichael J. Spencer   paths.push_back(".");
893ef91c55SMichael J. Spencer   paths.push_back("..");
903ef91c55SMichael J. Spencer   paths.push_back("foo");
913ef91c55SMichael J. Spencer   paths.push_back("/");
923ef91c55SMichael J. Spencer   paths.push_back("/foo");
933ef91c55SMichael J. Spencer   paths.push_back("foo/");
943ef91c55SMichael J. Spencer   paths.push_back("/foo/");
953ef91c55SMichael J. Spencer   paths.push_back("foo/bar");
963ef91c55SMichael J. Spencer   paths.push_back("/foo/bar");
973ef91c55SMichael J. Spencer   paths.push_back("//net");
9860b7132aSPavel Labath   paths.push_back("//net/");
993ef91c55SMichael J. Spencer   paths.push_back("//net/foo");
1003ef91c55SMichael J. Spencer   paths.push_back("///foo///");
1013ef91c55SMichael J. Spencer   paths.push_back("///foo///bar");
1023ef91c55SMichael J. Spencer   paths.push_back("/.");
1033ef91c55SMichael J. Spencer   paths.push_back("./");
1043ef91c55SMichael J. Spencer   paths.push_back("/..");
1053ef91c55SMichael J. Spencer   paths.push_back("../");
1063ef91c55SMichael J. Spencer   paths.push_back("foo/.");
1073ef91c55SMichael J. Spencer   paths.push_back("foo/..");
1083ef91c55SMichael J. Spencer   paths.push_back("foo/./");
1093ef91c55SMichael J. Spencer   paths.push_back("foo/./bar");
1103ef91c55SMichael J. Spencer   paths.push_back("foo/..");
1113ef91c55SMichael J. Spencer   paths.push_back("foo/../");
1123ef91c55SMichael J. Spencer   paths.push_back("foo/../bar");
1133ef91c55SMichael J. Spencer   paths.push_back("c:");
1143ef91c55SMichael J. Spencer   paths.push_back("c:/");
1153ef91c55SMichael J. Spencer   paths.push_back("c:foo");
1163ef91c55SMichael J. Spencer   paths.push_back("c:/foo");
1173ef91c55SMichael J. Spencer   paths.push_back("c:foo/");
1183ef91c55SMichael J. Spencer   paths.push_back("c:/foo/");
1193ef91c55SMichael J. Spencer   paths.push_back("c:/foo/bar");
1203ef91c55SMichael J. Spencer   paths.push_back("prn:");
1213ef91c55SMichael J. Spencer   paths.push_back("c:\\");
1223ef91c55SMichael J. Spencer   paths.push_back("c:foo");
1233ef91c55SMichael J. Spencer   paths.push_back("c:\\foo");
1243ef91c55SMichael J. Spencer   paths.push_back("c:foo\\");
1253ef91c55SMichael J. Spencer   paths.push_back("c:\\foo\\");
1263ef91c55SMichael J. Spencer   paths.push_back("c:\\foo/");
1273ef91c55SMichael J. Spencer   paths.push_back("c:/foo\\bar");
1283ef91c55SMichael J. Spencer 
1293ef91c55SMichael J. Spencer   for (SmallVector<StringRef, 40>::const_iterator i = paths.begin(),
1303ef91c55SMichael J. Spencer                                                   e = paths.end();
1313ef91c55SMichael J. Spencer                                                   i != e;
1323ef91c55SMichael J. Spencer                                                   ++i) {
13360b7132aSPavel Labath     SCOPED_TRACE(*i);
13460b7132aSPavel Labath     SmallVector<StringRef, 5> ComponentStack;
1353ef91c55SMichael J. Spencer     for (sys::path::const_iterator ci = sys::path::begin(*i),
1363ef91c55SMichael J. Spencer                                    ce = sys::path::end(*i);
1373ef91c55SMichael J. Spencer                                    ci != ce;
1383ef91c55SMichael J. Spencer                                    ++ci) {
13960b7132aSPavel Labath       EXPECT_FALSE(ci->empty());
1400c274aedSJustin Bogner       ComponentStack.push_back(*ci);
1413ef91c55SMichael J. Spencer     }
1423ef91c55SMichael J. Spencer 
14360b7132aSPavel Labath     SmallVector<StringRef, 5> ReverseComponentStack;
144545cbdf0SMichael J. Spencer     for (sys::path::reverse_iterator ci = sys::path::rbegin(*i),
145545cbdf0SMichael J. Spencer                                      ce = sys::path::rend(*i);
146545cbdf0SMichael J. Spencer                                      ci != ce;
147545cbdf0SMichael J. Spencer                                      ++ci) {
14860b7132aSPavel Labath       EXPECT_FALSE(ci->empty());
14960b7132aSPavel Labath       ReverseComponentStack.push_back(*ci);
150545cbdf0SMichael J. Spencer     }
15160b7132aSPavel Labath     std::reverse(ReverseComponentStack.begin(), ReverseComponentStack.end());
15260b7132aSPavel Labath     EXPECT_THAT(ComponentStack, testing::ContainerEq(ReverseComponentStack));
153545cbdf0SMichael J. Spencer 
1540db71d9eSJustin Bogner     // Crash test most of the API - since we're iterating over all of our paths
1550db71d9eSJustin Bogner     // here there isn't really anything reasonable to assert on in the results.
1560db71d9eSJustin Bogner     (void)path::has_root_path(*i);
1570db71d9eSJustin Bogner     (void)path::root_path(*i);
1580db71d9eSJustin Bogner     (void)path::has_root_name(*i);
1590db71d9eSJustin Bogner     (void)path::root_name(*i);
1600db71d9eSJustin Bogner     (void)path::has_root_directory(*i);
1610db71d9eSJustin Bogner     (void)path::root_directory(*i);
1620db71d9eSJustin Bogner     (void)path::has_parent_path(*i);
1630db71d9eSJustin Bogner     (void)path::parent_path(*i);
1640db71d9eSJustin Bogner     (void)path::has_filename(*i);
1650db71d9eSJustin Bogner     (void)path::filename(*i);
1660db71d9eSJustin Bogner     (void)path::has_stem(*i);
1670db71d9eSJustin Bogner     (void)path::stem(*i);
1680db71d9eSJustin Bogner     (void)path::has_extension(*i);
1690db71d9eSJustin Bogner     (void)path::extension(*i);
1700db71d9eSJustin Bogner     (void)path::is_absolute(*i);
1710db71d9eSJustin Bogner     (void)path::is_relative(*i);
1723ef91c55SMichael J. Spencer 
1733b264ba9SMichael J. Spencer     SmallString<128> temp_store;
174b5ca6447SMichael J. Spencer     temp_store = *i;
1753b264ba9SMichael J. Spencer     ASSERT_NO_ERROR(fs::make_absolute(temp_store));
176b5ca6447SMichael J. Spencer     temp_store = *i;
1771e090f0bSMichael J. Spencer     path::remove_filename(temp_store);
1784d0c6fddSMichael J. Spencer 
179b5ca6447SMichael J. Spencer     temp_store = *i;
1801e090f0bSMichael J. Spencer     path::replace_extension(temp_store, "ext");
1814d0c6fddSMichael J. Spencer     StringRef filename(temp_store.begin(), temp_store.size()), stem, ext;
182f616b218SMichael J. Spencer     stem = path::stem(filename);
183f616b218SMichael J. Spencer     ext  = path::extension(filename);
184487e764bSJustin Bogner     EXPECT_EQ(*sys::path::rbegin(filename), (stem + ext).str());
1854d0c6fddSMichael J. Spencer 
1861e090f0bSMichael J. Spencer     path::native(*i, temp_store);
1873ef91c55SMichael J. Spencer   }
188ae1d5996SBenjamin Kramer 
189ae1d5996SBenjamin Kramer   SmallString<32> Relative("foo.cpp");
190ae1d5996SBenjamin Kramer   ASSERT_NO_ERROR(sys::fs::make_absolute("/root", Relative));
1912b4e14edSBenjamin Kramer   Relative[5] = '/'; // Fix up windows paths.
192ae1d5996SBenjamin Kramer   ASSERT_EQ("/root/foo.cpp", Relative);
193346a1338SMichael J. Spencer }
1944571040eSMichael J. Spencer 
19560b7132aSPavel Labath TEST(Support, FilenameParent) {
19660b7132aSPavel Labath   EXPECT_EQ("/", path::filename("/"));
19760b7132aSPavel Labath   EXPECT_EQ("", path::parent_path("/"));
19873537eacSTareq A. Siraj 
19960b7132aSPavel Labath   EXPECT_EQ("\\", path::filename("c:\\", path::Style::windows));
20060b7132aSPavel Labath   EXPECT_EQ("c:", path::parent_path("c:\\", path::Style::windows));
20173537eacSTareq A. Siraj 
202d20289b3SPavel Labath   EXPECT_EQ("/", path::filename("///"));
203d20289b3SPavel Labath   EXPECT_EQ("", path::parent_path("///"));
204d20289b3SPavel Labath 
205d20289b3SPavel Labath   EXPECT_EQ("\\", path::filename("c:\\\\", path::Style::windows));
206d20289b3SPavel Labath   EXPECT_EQ("c:", path::parent_path("c:\\\\", path::Style::windows));
207d20289b3SPavel Labath 
20860b7132aSPavel Labath   EXPECT_EQ("bar", path::filename("/foo/bar"));
20960b7132aSPavel Labath   EXPECT_EQ("/foo", path::parent_path("/foo/bar"));
21060b7132aSPavel Labath 
21160b7132aSPavel Labath   EXPECT_EQ("foo", path::filename("/foo"));
21260b7132aSPavel Labath   EXPECT_EQ("/", path::parent_path("/foo"));
21360b7132aSPavel Labath 
21460b7132aSPavel Labath   EXPECT_EQ("foo", path::filename("foo"));
21560b7132aSPavel Labath   EXPECT_EQ("", path::parent_path("foo"));
21660b7132aSPavel Labath 
21760b7132aSPavel Labath   EXPECT_EQ(".", path::filename("foo/"));
21860b7132aSPavel Labath   EXPECT_EQ("foo", path::parent_path("foo/"));
21960b7132aSPavel Labath 
22060b7132aSPavel Labath   EXPECT_EQ("//net", path::filename("//net"));
22160b7132aSPavel Labath   EXPECT_EQ("", path::parent_path("//net"));
22260b7132aSPavel Labath 
22360b7132aSPavel Labath   EXPECT_EQ("/", path::filename("//net/"));
22460b7132aSPavel Labath   EXPECT_EQ("//net", path::parent_path("//net/"));
22560b7132aSPavel Labath 
22660b7132aSPavel Labath   EXPECT_EQ("foo", path::filename("//net/foo"));
22760b7132aSPavel Labath   EXPECT_EQ("//net/", path::parent_path("//net/foo"));
228d20289b3SPavel Labath 
229d20289b3SPavel Labath   // These checks are just to make sure we do something reasonable with the
230d20289b3SPavel Labath   // paths below. They are not meant to prescribe the one true interpretation of
231d20289b3SPavel Labath   // these paths. Other decompositions (e.g. "//" -> "" + "//") are also
232d20289b3SPavel Labath   // possible.
233d20289b3SPavel Labath   EXPECT_EQ("/", path::filename("//"));
234d20289b3SPavel Labath   EXPECT_EQ("", path::parent_path("//"));
235d20289b3SPavel Labath 
236d20289b3SPavel Labath   EXPECT_EQ("\\", path::filename("\\\\", path::Style::windows));
237d20289b3SPavel Labath   EXPECT_EQ("", path::parent_path("\\\\", path::Style::windows));
238d20289b3SPavel Labath 
239d20289b3SPavel Labath   EXPECT_EQ("\\", path::filename("\\\\\\", path::Style::windows));
240d20289b3SPavel Labath   EXPECT_EQ("", path::parent_path("\\\\\\", path::Style::windows));
24173537eacSTareq A. Siraj }
24273537eacSTareq A. Siraj 
24360b7132aSPavel Labath static std::vector<StringRef>
24460b7132aSPavel Labath GetComponents(StringRef Path, path::Style S = path::Style::native) {
24560b7132aSPavel Labath   return {path::begin(Path, S), path::end(Path)};
24673537eacSTareq A. Siraj }
24773537eacSTareq A. Siraj 
24860b7132aSPavel Labath TEST(Support, PathIterator) {
24960b7132aSPavel Labath   EXPECT_THAT(GetComponents("/foo"), testing::ElementsAre("/", "foo"));
25060b7132aSPavel Labath   EXPECT_THAT(GetComponents("/"), testing::ElementsAre("/"));
251d20289b3SPavel Labath   EXPECT_THAT(GetComponents("//"), testing::ElementsAre("/"));
252d20289b3SPavel Labath   EXPECT_THAT(GetComponents("///"), testing::ElementsAre("/"));
25360b7132aSPavel Labath   EXPECT_THAT(GetComponents("c/d/e/foo.txt"),
25460b7132aSPavel Labath               testing::ElementsAre("c", "d", "e", "foo.txt"));
25560b7132aSPavel Labath   EXPECT_THAT(GetComponents(".c/.d/../."),
25660b7132aSPavel Labath               testing::ElementsAre(".c", ".d", "..", "."));
25760b7132aSPavel Labath   EXPECT_THAT(GetComponents("/c/d/e/foo.txt"),
25860b7132aSPavel Labath               testing::ElementsAre("/", "c", "d", "e", "foo.txt"));
25960b7132aSPavel Labath   EXPECT_THAT(GetComponents("/.c/.d/../."),
26060b7132aSPavel Labath               testing::ElementsAre("/", ".c", ".d", "..", "."));
26160b7132aSPavel Labath   EXPECT_THAT(GetComponents("c:\\c\\e\\foo.txt", path::Style::windows),
26260b7132aSPavel Labath               testing::ElementsAre("c:", "\\", "c", "e", "foo.txt"));
26360b7132aSPavel Labath   EXPECT_THAT(GetComponents("//net/"), testing::ElementsAre("//net", "/"));
26460b7132aSPavel Labath   EXPECT_THAT(GetComponents("//net/c/foo.txt"),
26560b7132aSPavel Labath               testing::ElementsAre("//net", "/", "c", "foo.txt"));
26673537eacSTareq A. Siraj }
26773537eacSTareq A. Siraj 
2688d116396SBen Langmuir TEST(Support, AbsolutePathIteratorEnd) {
2698d116396SBen Langmuir   // Trailing slashes are converted to '.' unless they are part of the root path.
2705c5091fcSZachary Turner   SmallVector<std::pair<StringRef, path::Style>, 4> Paths;
2715c5091fcSZachary Turner   Paths.emplace_back("/foo/", path::Style::native);
2725c5091fcSZachary Turner   Paths.emplace_back("/foo//", path::Style::native);
27360b7132aSPavel Labath   Paths.emplace_back("//net/foo/", path::Style::native);
27460b7132aSPavel Labath   Paths.emplace_back("c:\\foo\\", path::Style::windows);
2758d116396SBen Langmuir 
2765c5091fcSZachary Turner   for (auto &Path : Paths) {
27760b7132aSPavel Labath     SCOPED_TRACE(Path.first);
2785c5091fcSZachary Turner     StringRef LastComponent = *path::rbegin(Path.first, Path.second);
2798d116396SBen Langmuir     EXPECT_EQ(".", LastComponent);
2808d116396SBen Langmuir   }
2818d116396SBen Langmuir 
2825c5091fcSZachary Turner   SmallVector<std::pair<StringRef, path::Style>, 3> RootPaths;
2835c5091fcSZachary Turner   RootPaths.emplace_back("/", path::Style::native);
2845c5091fcSZachary Turner   RootPaths.emplace_back("//net/", path::Style::native);
2855c5091fcSZachary Turner   RootPaths.emplace_back("c:\\", path::Style::windows);
286d20289b3SPavel Labath   RootPaths.emplace_back("//net//", path::Style::native);
287d20289b3SPavel Labath   RootPaths.emplace_back("c:\\\\", path::Style::windows);
2888d116396SBen Langmuir 
2895c5091fcSZachary Turner   for (auto &Path : RootPaths) {
29060b7132aSPavel Labath     SCOPED_TRACE(Path.first);
2915c5091fcSZachary Turner     StringRef LastComponent = *path::rbegin(Path.first, Path.second);
2928d116396SBen Langmuir     EXPECT_EQ(1u, LastComponent.size());
2935c5091fcSZachary Turner     EXPECT_TRUE(path::is_separator(LastComponent[0], Path.second));
2948d116396SBen Langmuir   }
2958d116396SBen Langmuir }
2968d116396SBen Langmuir 
297f7d41011SPeter Collingbourne TEST(Support, HomeDirectory) {
298cc275e42SNAKAMURA Takumi   std::string expected;
299712e8d29SNico Weber #ifdef _WIN32
3002fa025cdSPawel Bylica   if (wchar_t const *path = ::_wgetenv(L"USERPROFILE")) {
3017187e4bbSPawel Bylica     auto pathLen = ::wcslen(path);
3022fa025cdSPawel Bylica     ArrayRef<char> ref{reinterpret_cast<char const *>(path),
3032fa025cdSPawel Bylica                        pathLen * sizeof(wchar_t)};
3047187e4bbSPawel Bylica     convertUTF16ToUTF8String(ref, expected);
3052fa025cdSPawel Bylica   }
3067187e4bbSPawel Bylica #else
3072fa025cdSPawel Bylica   if (char const *path = ::getenv("HOME"))
3082fa025cdSPawel Bylica     expected = path;
309f7d41011SPeter Collingbourne #endif
3102fa025cdSPawel Bylica   // Do not try to test it if we don't know what to expect.
3112fa025cdSPawel Bylica   // On Windows we use something better than env vars.
3122fa025cdSPawel Bylica   if (!expected.empty()) {
313f7d41011SPeter Collingbourne     SmallString<128> HomeDir;
3147187e4bbSPawel Bylica     auto status = path::home_directory(HomeDir);
3152fa025cdSPawel Bylica     EXPECT_TRUE(status);
3167187e4bbSPawel Bylica     EXPECT_EQ(expected, HomeDir);
317f7d41011SPeter Collingbourne   }
318cc275e42SNAKAMURA Takumi }
319f7d41011SPeter Collingbourne 
320d8350cf3SZachary Turner #ifdef LLVM_ON_UNIX
321a3cf70bfSZachary Turner TEST(Support, HomeDirectoryWithNoEnv) {
322d8350cf3SZachary Turner   std::string OriginalStorage;
323d8350cf3SZachary Turner   char const *OriginalEnv = ::getenv("HOME");
324d8350cf3SZachary Turner   if (OriginalEnv) {
325d8350cf3SZachary Turner     // We're going to unset it, so make a copy and save a pointer to the copy
326d8350cf3SZachary Turner     // so that we can reset it at the end of the test.
327d8350cf3SZachary Turner     OriginalStorage = OriginalEnv;
328d8350cf3SZachary Turner     OriginalEnv = OriginalStorage.c_str();
329d8350cf3SZachary Turner   }
330d8350cf3SZachary Turner 
331d8350cf3SZachary Turner   // Don't run the test if we have nothing to compare against.
332d8350cf3SZachary Turner   struct passwd *pw = getpwuid(getuid());
333d8350cf3SZachary Turner   if (!pw || !pw->pw_dir) return;
334d8350cf3SZachary Turner 
335a3cf70bfSZachary Turner   ::unsetenv("HOME");
336d8350cf3SZachary Turner   EXPECT_EQ(nullptr, ::getenv("HOME"));
337d8350cf3SZachary Turner   std::string PwDir = pw->pw_dir;
338a3cf70bfSZachary Turner 
339a3cf70bfSZachary Turner   SmallString<128> HomeDir;
340a3cf70bfSZachary Turner   auto status = path::home_directory(HomeDir);
341a3cf70bfSZachary Turner   EXPECT_TRUE(status);
342d8350cf3SZachary Turner   EXPECT_EQ(PwDir, HomeDir);
343a3cf70bfSZachary Turner 
344d8350cf3SZachary Turner   // Now put the environment back to its original state (meaning that if it was
345d8350cf3SZachary Turner   // unset before, we don't reset it).
346d8350cf3SZachary Turner   if (OriginalEnv) ::setenv("HOME", OriginalEnv, 1);
347a3cf70bfSZachary Turner }
348a3cf70bfSZachary Turner #endif
349a3cf70bfSZachary Turner 
350a90e7451SPawel Bylica TEST(Support, TempDirectory) {
351a90e7451SPawel Bylica   SmallString<32> TempDir;
352a90e7451SPawel Bylica   path::system_temp_directory(false, TempDir);
353a90e7451SPawel Bylica   EXPECT_TRUE(!TempDir.empty());
354a90e7451SPawel Bylica   TempDir.clear();
355a90e7451SPawel Bylica   path::system_temp_directory(true, TempDir);
356a90e7451SPawel Bylica   EXPECT_TRUE(!TempDir.empty());
357a90e7451SPawel Bylica }
358a90e7451SPawel Bylica 
359712e8d29SNico Weber #ifdef _WIN32
360a90e7451SPawel Bylica static std::string path2regex(std::string Path) {
361a90e7451SPawel Bylica   size_t Pos = 0;
362a90e7451SPawel Bylica   while ((Pos = Path.find('\\', Pos)) != std::string::npos) {
363a90e7451SPawel Bylica     Path.replace(Pos, 1, "\\\\");
364a90e7451SPawel Bylica     Pos += 2;
365a90e7451SPawel Bylica   }
366a90e7451SPawel Bylica   return Path;
367a90e7451SPawel Bylica }
368a90e7451SPawel Bylica 
369a90e7451SPawel Bylica /// Helper for running temp dir test in separated process. See below.
370a90e7451SPawel Bylica #define EXPECT_TEMP_DIR(prepare, expected)                                     \
371a90e7451SPawel Bylica   EXPECT_EXIT(                                                                 \
372a90e7451SPawel Bylica       {                                                                        \
373a90e7451SPawel Bylica         prepare;                                                               \
374a90e7451SPawel Bylica         SmallString<300> TempDir;                                              \
375a90e7451SPawel Bylica         path::system_temp_directory(true, TempDir);                            \
376a90e7451SPawel Bylica         raw_os_ostream(std::cerr) << TempDir;                                  \
377a90e7451SPawel Bylica         std::exit(0);                                                          \
378a90e7451SPawel Bylica       },                                                                       \
379a90e7451SPawel Bylica       ::testing::ExitedWithCode(0), path2regex(expected))
380a90e7451SPawel Bylica 
381a90e7451SPawel Bylica TEST(SupportDeathTest, TempDirectoryOnWindows) {
382a90e7451SPawel Bylica   // In this test we want to check how system_temp_directory responds to
383a90e7451SPawel Bylica   // different values of specific env vars. To prevent corrupting env vars of
384a90e7451SPawel Bylica   // the current process all checks are done in separated processes.
385a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_wputenv_s(L"TMP", L"C:\\OtherFolder"), "C:\\OtherFolder");
386a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_wputenv_s(L"TMP", L"C:/Unix/Path/Seperators"),
387a90e7451SPawel Bylica                   "C:\\Unix\\Path\\Seperators");
388a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_wputenv_s(L"TMP", L"Local Path"), ".+\\Local Path$");
389a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_wputenv_s(L"TMP", L"F:\\TrailingSep\\"), "F:\\TrailingSep");
390a90e7451SPawel Bylica   EXPECT_TEMP_DIR(
391a90e7451SPawel Bylica       _wputenv_s(L"TMP", L"C:\\2\x03C0r-\x00B5\x00B3\\\x2135\x2080"),
392a90e7451SPawel Bylica       "C:\\2\xCF\x80r-\xC2\xB5\xC2\xB3\\\xE2\x84\xB5\xE2\x82\x80");
393a90e7451SPawel Bylica 
394a90e7451SPawel Bylica   // Test $TMP empty, $TEMP set.
395a90e7451SPawel Bylica   EXPECT_TEMP_DIR(
396a90e7451SPawel Bylica       {
397a90e7451SPawel Bylica         _wputenv_s(L"TMP", L"");
398a90e7451SPawel Bylica         _wputenv_s(L"TEMP", L"C:\\Valid\\Path");
399a90e7451SPawel Bylica       },
400a90e7451SPawel Bylica       "C:\\Valid\\Path");
401a90e7451SPawel Bylica 
402a90e7451SPawel Bylica   // All related env vars empty
403a90e7451SPawel Bylica   EXPECT_TEMP_DIR(
404a90e7451SPawel Bylica   {
405a90e7451SPawel Bylica     _wputenv_s(L"TMP", L"");
406a90e7451SPawel Bylica     _wputenv_s(L"TEMP", L"");
407a90e7451SPawel Bylica     _wputenv_s(L"USERPROFILE", L"");
408a90e7451SPawel Bylica   },
409a90e7451SPawel Bylica     "C:\\Temp");
410a90e7451SPawel Bylica 
411a90e7451SPawel Bylica   // Test evn var / path with 260 chars.
412a90e7451SPawel Bylica   SmallString<270> Expected{"C:\\Temp\\AB\\123456789"};
413a90e7451SPawel Bylica   while (Expected.size() < 260)
414a90e7451SPawel Bylica     Expected.append("\\DirNameWith19Charss");
41588e9ff62SReid Kleckner   ASSERT_EQ(260U, Expected.size());
416a90e7451SPawel Bylica   EXPECT_TEMP_DIR(_putenv_s("TMP", Expected.c_str()), Expected.c_str());
417a90e7451SPawel Bylica }
418a90e7451SPawel Bylica #endif
419a90e7451SPawel Bylica 
420346a1338SMichael J. Spencer class FileSystemTest : public testing::Test {
421346a1338SMichael J. Spencer protected:
422346a1338SMichael J. Spencer   /// Unique temporary directory in which all created filesystem entities must
423d9c4a9afSPaul Robinson   /// be placed. It is removed at the end of each test (must be empty).
424346a1338SMichael J. Spencer   SmallString<128> TestDirectory;
4251f67a3cbSZachary Turner   SmallString<128> NonExistantFile;
426346a1338SMichael J. Spencer 
427f817c1cbSAlexander Kornienko   void SetUp() override {
428346a1338SMichael J. Spencer     ASSERT_NO_ERROR(
4297ffacc49SRafael Espindola         fs::createUniqueDirectory("file-system-test", TestDirectory));
430346a1338SMichael J. Spencer     // We don't care about this specific file.
43142368cc1SMichael J. Spencer     errs() << "Test Directory: " << TestDirectory << '\n';
43242368cc1SMichael J. Spencer     errs().flush();
4331f67a3cbSZachary Turner     NonExistantFile = TestDirectory;
4341f67a3cbSZachary Turner 
4351f67a3cbSZachary Turner     // Even though this value is hardcoded, is a 128-bit GUID, so we should be
4361f67a3cbSZachary Turner     // guaranteed that this file will never exist.
4371f67a3cbSZachary Turner     sys::path::append(NonExistantFile, "1B28B495C16344CB9822E588CD4C3EF0");
438346a1338SMichael J. Spencer   }
439346a1338SMichael J. Spencer 
440f817c1cbSAlexander Kornienko   void TearDown() override { ASSERT_NO_ERROR(fs::remove(TestDirectory.str())); }
441346a1338SMichael J. Spencer };
442346a1338SMichael J. Spencer 
4439fbe530dSAaron Ballman TEST_F(FileSystemTest, Unique) {
4449fbe530dSAaron Ballman   // Create a temp file.
4459fbe530dSAaron Ballman   int FileDescriptor;
4469fbe530dSAaron Ballman   SmallString<64> TempPath;
4479fbe530dSAaron Ballman   ASSERT_NO_ERROR(
448155cf0f3SRafael Espindola       fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
4499fbe530dSAaron Ballman 
4509fbe530dSAaron Ballman   // The same file should return an identical unique id.
4517f822a93SRafael Espindola   fs::UniqueID F1, F2;
4527cf7c51cSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), F1));
4537cf7c51cSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), F2));
4549fbe530dSAaron Ballman   ASSERT_EQ(F1, F2);
4559fbe530dSAaron Ballman 
4569fbe530dSAaron Ballman   // Different files should return different unique ids.
4579fbe530dSAaron Ballman   int FileDescriptor2;
4589fbe530dSAaron Ballman   SmallString<64> TempPath2;
4599fbe530dSAaron Ballman   ASSERT_NO_ERROR(
460155cf0f3SRafael Espindola       fs::createTemporaryFile("prefix", "temp", FileDescriptor2, TempPath2));
4619fbe530dSAaron Ballman 
4627f822a93SRafael Espindola   fs::UniqueID D;
4637cf7c51cSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath2), D));
4649fbe530dSAaron Ballman   ASSERT_NE(D, F1);
4659fbe530dSAaron Ballman   ::close(FileDescriptor2);
4669fbe530dSAaron Ballman 
4679fbe530dSAaron Ballman   ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
4689fbe530dSAaron Ballman 
4699fbe530dSAaron Ballman   // Two paths representing the same file on disk should still provide the
4709fbe530dSAaron Ballman   // same unique id.  We can test this by making a hard link.
47183f858e5SRafael Espindola   ASSERT_NO_ERROR(fs::create_link(Twine(TempPath), Twine(TempPath2)));
4727f822a93SRafael Espindola   fs::UniqueID D2;
4737cf7c51cSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath2), D2));
4749fbe530dSAaron Ballman   ASSERT_EQ(D2, F1);
4759fbe530dSAaron Ballman 
4769fbe530dSAaron Ballman   ::close(FileDescriptor);
477a5932afeSRafael Espindola 
478a5932afeSRafael Espindola   SmallString<128> Dir1;
479a5932afeSRafael Espindola   ASSERT_NO_ERROR(
480a5932afeSRafael Espindola      fs::createUniqueDirectory("dir1", Dir1));
481a5932afeSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Dir1.c_str(), F1));
482a5932afeSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Dir1.c_str(), F2));
483a5932afeSRafael Espindola   ASSERT_EQ(F1, F2);
484a5932afeSRafael Espindola 
485a5932afeSRafael Espindola   SmallString<128> Dir2;
486a5932afeSRafael Espindola   ASSERT_NO_ERROR(
487a5932afeSRafael Espindola      fs::createUniqueDirectory("dir2", Dir2));
488a5932afeSRafael Espindola   ASSERT_NO_ERROR(fs::getUniqueID(Dir2.c_str(), F2));
489a5932afeSRafael Espindola   ASSERT_NE(F1, F2);
4901a4398a1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(Dir1));
4911a4398a1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(Dir2));
49275e557f1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(TempPath2));
49375e557f1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(TempPath));
4949fbe530dSAaron Ballman }
4959fbe530dSAaron Ballman 
496e48ace6aSZachary Turner TEST_F(FileSystemTest, RealPath) {
497e48ace6aSZachary Turner   ASSERT_NO_ERROR(
498e48ace6aSZachary Turner       fs::create_directories(Twine(TestDirectory) + "/test1/test2/test3"));
499e48ace6aSZachary Turner   ASSERT_TRUE(fs::exists(Twine(TestDirectory) + "/test1/test2/test3"));
500e48ace6aSZachary Turner 
501e48ace6aSZachary Turner   SmallString<64> RealBase;
502e48ace6aSZachary Turner   SmallString<64> Expected;
503e48ace6aSZachary Turner   SmallString<64> Actual;
504e48ace6aSZachary Turner 
505e48ace6aSZachary Turner   // TestDirectory itself might be under a symlink or have been specified with
506e48ace6aSZachary Turner   // a different case than the existing temp directory.  In such cases real_path
507e48ace6aSZachary Turner   // on the concatenated path will differ in the TestDirectory portion from
508e48ace6aSZachary Turner   // how we specified it.  Make sure to compare against the real_path of the
509e48ace6aSZachary Turner   // TestDirectory, and not just the value of TestDirectory.
510e48ace6aSZachary Turner   ASSERT_NO_ERROR(fs::real_path(TestDirectory, RealBase));
511e48ace6aSZachary Turner   path::native(Twine(RealBase) + "/test1/test2", Expected);
512e48ace6aSZachary Turner 
513e48ace6aSZachary Turner   ASSERT_NO_ERROR(fs::real_path(
514e48ace6aSZachary Turner       Twine(TestDirectory) + "/././test1/../test1/test2/./test3/..", Actual));
515e48ace6aSZachary Turner 
516e48ace6aSZachary Turner   EXPECT_EQ(Expected, Actual);
517e48ace6aSZachary Turner 
518e48ace6aSZachary Turner   SmallString<64> HomeDir;
519*b23f430eSJonas Devlieghere 
520*b23f430eSJonas Devlieghere   // This can fail if $HOME is not set and getpwuid fails.
521aeab48abSZachary Turner   bool Result = llvm::sys::path::home_directory(HomeDir);
522aeab48abSZachary Turner   if (Result) {
523e48ace6aSZachary Turner     ASSERT_NO_ERROR(fs::real_path(HomeDir, Expected));
524e48ace6aSZachary Turner     ASSERT_NO_ERROR(fs::real_path("~", Actual, true));
525e48ace6aSZachary Turner     EXPECT_EQ(Expected, Actual);
526e48ace6aSZachary Turner     ASSERT_NO_ERROR(fs::real_path("~/", Actual, true));
527e48ace6aSZachary Turner     EXPECT_EQ(Expected, Actual);
528aeab48abSZachary Turner   }
529e48ace6aSZachary Turner 
530e48ace6aSZachary Turner   ASSERT_NO_ERROR(fs::remove_directories(Twine(TestDirectory) + "/test1"));
531e48ace6aSZachary Turner }
532e48ace6aSZachary Turner 
533*b23f430eSJonas Devlieghere TEST_F(FileSystemTest, ExpandTilde) {
534*b23f430eSJonas Devlieghere   SmallString<64> Expected;
535*b23f430eSJonas Devlieghere   SmallString<64> Actual;
536*b23f430eSJonas Devlieghere   SmallString<64> HomeDir;
537*b23f430eSJonas Devlieghere 
538*b23f430eSJonas Devlieghere   // This can fail if $HOME is not set and getpwuid fails.
539*b23f430eSJonas Devlieghere   bool Result = llvm::sys::path::home_directory(HomeDir);
540*b23f430eSJonas Devlieghere   if (Result) {
541*b23f430eSJonas Devlieghere     fs::expand_tilde(HomeDir, Expected);
542*b23f430eSJonas Devlieghere 
543*b23f430eSJonas Devlieghere     fs::expand_tilde("~", Actual);
544*b23f430eSJonas Devlieghere     EXPECT_EQ(Expected, Actual);
545*b23f430eSJonas Devlieghere 
546*b23f430eSJonas Devlieghere     path::append(Expected, "foo");
547*b23f430eSJonas Devlieghere     fs::expand_tilde("~/foo", Actual);
548*b23f430eSJonas Devlieghere     EXPECT_EQ(Expected, Actual);
549*b23f430eSJonas Devlieghere   }
550*b23f430eSJonas Devlieghere }
551*b23f430eSJonas Devlieghere 
5524762c069SDavide Italiano #ifdef LLVM_ON_UNIX
5534762c069SDavide Italiano TEST_F(FileSystemTest, RealPathNoReadPerm) {
5544762c069SDavide Italiano   SmallString<64> Expanded;
5554762c069SDavide Italiano 
5564762c069SDavide Italiano   ASSERT_NO_ERROR(
5574762c069SDavide Italiano     fs::create_directories(Twine(TestDirectory) + "/noreadperm"));
5584762c069SDavide Italiano   ASSERT_TRUE(fs::exists(Twine(TestDirectory) + "/noreadperm"));
5594762c069SDavide Italiano 
5604762c069SDavide Italiano   fs::setPermissions(Twine(TestDirectory) + "/noreadperm", fs::no_perms);
5614762c069SDavide Italiano   fs::setPermissions(Twine(TestDirectory) + "/noreadperm", fs::all_exe);
5624762c069SDavide Italiano 
5634762c069SDavide Italiano   ASSERT_NO_ERROR(fs::real_path(Twine(TestDirectory) + "/noreadperm", Expanded,
5644762c069SDavide Italiano                                 false));
5654762c069SDavide Italiano 
5664762c069SDavide Italiano   ASSERT_NO_ERROR(fs::remove_directories(Twine(TestDirectory) + "/noreadperm"));
5674762c069SDavide Italiano }
5684762c069SDavide Italiano #endif
5694762c069SDavide Italiano 
5704762c069SDavide Italiano 
571fe161b9dSRafael Espindola TEST_F(FileSystemTest, TempFileKeepDiscard) {
572fe161b9dSRafael Espindola   // We can keep then discard.
573fe161b9dSRafael Espindola   auto TempFileOrError = fs::TempFile::create(TestDirectory + "/test-%%%%");
574fe161b9dSRafael Espindola   ASSERT_TRUE((bool)TempFileOrError);
575fe161b9dSRafael Espindola   fs::TempFile File = std::move(*TempFileOrError);
576fe161b9dSRafael Espindola   ASSERT_FALSE((bool)File.keep(TestDirectory + "/keep"));
577fe161b9dSRafael Espindola   ASSERT_FALSE((bool)File.discard());
578fe161b9dSRafael Espindola   ASSERT_TRUE(fs::exists(TestDirectory + "/keep"));
579fe161b9dSRafael Espindola   ASSERT_NO_ERROR(fs::remove(TestDirectory + "/keep"));
580fe161b9dSRafael Espindola }
581fe161b9dSRafael Espindola 
582fe161b9dSRafael Espindola TEST_F(FileSystemTest, TempFileDiscardDiscard) {
583fe161b9dSRafael Espindola   // We can discard twice.
584fe161b9dSRafael Espindola   auto TempFileOrError = fs::TempFile::create(TestDirectory + "/test-%%%%");
585fe161b9dSRafael Espindola   ASSERT_TRUE((bool)TempFileOrError);
586fe161b9dSRafael Espindola   fs::TempFile File = std::move(*TempFileOrError);
587fe161b9dSRafael Espindola   ASSERT_FALSE((bool)File.discard());
588fe161b9dSRafael Espindola   ASSERT_FALSE((bool)File.discard());
589fe161b9dSRafael Espindola   ASSERT_FALSE(fs::exists(TestDirectory + "/keep"));
590fe161b9dSRafael Espindola }
591fe161b9dSRafael Espindola 
592346a1338SMichael J. Spencer TEST_F(FileSystemTest, TempFiles) {
5934fb115d2SMichael J. Spencer   // Create a temp file.
5944571040eSMichael J. Spencer   int FileDescriptor;
5954571040eSMichael J. Spencer   SmallString<64> TempPath;
5963b264ba9SMichael J. Spencer   ASSERT_NO_ERROR(
597155cf0f3SRafael Espindola       fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
5984571040eSMichael J. Spencer 
5994fb115d2SMichael J. Spencer   // Make sure it exists.
600c1599157SRafael Espindola   ASSERT_TRUE(sys::fs::exists(Twine(TempPath)));
6014571040eSMichael J. Spencer 
6024fb115d2SMichael J. Spencer   // Create another temp tile.
6034fb115d2SMichael J. Spencer   int FD2;
6044fb115d2SMichael J. Spencer   SmallString<64> TempPath2;
605155cf0f3SRafael Espindola   ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "temp", FD2, TempPath2));
606d3c89041SRafael Espindola   ASSERT_TRUE(TempPath2.endswith(".temp"));
6074fb115d2SMichael J. Spencer   ASSERT_NE(TempPath.str(), TempPath2.str());
6084fb115d2SMichael J. Spencer 
609203d7807SMichael J. Spencer   fs::file_status A, B;
610203d7807SMichael J. Spencer   ASSERT_NO_ERROR(fs::status(Twine(TempPath), A));
611203d7807SMichael J. Spencer   ASSERT_NO_ERROR(fs::status(Twine(TempPath2), B));
612203d7807SMichael J. Spencer   EXPECT_FALSE(fs::equivalent(A, B));
613203d7807SMichael J. Spencer 
6144fb115d2SMichael J. Spencer   ::close(FD2);
615213c4cb1SRafael Espindola 
6164fb115d2SMichael J. Spencer   // Remove Temp2.
6175c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
6185c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
6195c20ac01SRafael Espindola   ASSERT_EQ(fs::remove(Twine(TempPath2), false),
6202a826e40SRafael Espindola             errc::no_such_file_or_directory);
6214fb115d2SMichael J. Spencer 
622c049c657SRafael Espindola   std::error_code EC = fs::status(TempPath2.c_str(), B);
6232a826e40SRafael Espindola   EXPECT_EQ(EC, errc::no_such_file_or_directory);
624107b74c6SRafael Espindola   EXPECT_EQ(B.type(), fs::file_type::file_not_found);
625107b74c6SRafael Espindola 
6264fb115d2SMichael J. Spencer   // Make sure Temp2 doesn't exist.
627281f23adSRafael Espindola   ASSERT_EQ(fs::access(Twine(TempPath2), sys::fs::AccessMode::Exist),
628281f23adSRafael Espindola             errc::no_such_file_or_directory);
6294fb115d2SMichael J. Spencer 
630d3c89041SRafael Espindola   SmallString<64> TempPath3;
631d3c89041SRafael Espindola   ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "", TempPath3));
632d3c89041SRafael Espindola   ASSERT_FALSE(TempPath3.endswith("."));
63375e557f1SReid Kleckner   FileRemover Cleanup3(TempPath3);
634d3c89041SRafael Espindola 
6354fb115d2SMichael J. Spencer   // Create a hard link to Temp1.
63683f858e5SRafael Espindola   ASSERT_NO_ERROR(fs::create_link(Twine(TempPath), Twine(TempPath2)));
6374fb115d2SMichael J. Spencer   bool equal;
6383b264ba9SMichael J. Spencer   ASSERT_NO_ERROR(fs::equivalent(Twine(TempPath), Twine(TempPath2), equal));
6394fb115d2SMichael J. Spencer   EXPECT_TRUE(equal);
640203d7807SMichael J. Spencer   ASSERT_NO_ERROR(fs::status(Twine(TempPath), A));
641203d7807SMichael J. Spencer   ASSERT_NO_ERROR(fs::status(Twine(TempPath2), B));
642203d7807SMichael J. Spencer   EXPECT_TRUE(fs::equivalent(A, B));
6434fb115d2SMichael J. Spencer 
6444fb115d2SMichael J. Spencer   // Remove Temp1.
6454571040eSMichael J. Spencer   ::close(FileDescriptor);
6465c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TempPath)));
6474571040eSMichael J. Spencer 
6484fb115d2SMichael J. Spencer   // Remove the hard link.
6495c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
6504fb115d2SMichael J. Spencer 
6514fb115d2SMichael J. Spencer   // Make sure Temp1 doesn't exist.
652281f23adSRafael Espindola   ASSERT_EQ(fs::access(Twine(TempPath), sys::fs::AccessMode::Exist),
653281f23adSRafael Espindola             errc::no_such_file_or_directory);
654fcdf9a82SAaron Ballman 
655712e8d29SNico Weber #ifdef _WIN32
656fcdf9a82SAaron Ballman   // Path name > 260 chars should get an error.
657fcdf9a82SAaron Ballman   const char *Path270 =
658fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz9abcdefghijklmnopqrstuvwxyz8"
659fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz7abcdefghijklmnopqrstuvwxyz6"
660fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz5abcdefghijklmnopqrstuvwxyz4"
661fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz3abcdefghijklmnopqrstuvwxyz2"
662fcdf9a82SAaron Ballman     "abcdefghijklmnopqrstuvwxyz1abcdefghijklmnopqrstuvwxyz0";
6631b6c7347SPaul Robinson   EXPECT_EQ(fs::createUniqueFile(Path270, FileDescriptor, TempPath),
664d9c4a9afSPaul Robinson             errc::invalid_argument);
665d9c4a9afSPaul Robinson   // Relative path < 247 chars, no problem.
666d9c4a9afSPaul Robinson   const char *Path216 =
667d9c4a9afSPaul Robinson     "abcdefghijklmnopqrstuvwxyz7abcdefghijklmnopqrstuvwxyz6"
668d9c4a9afSPaul Robinson     "abcdefghijklmnopqrstuvwxyz5abcdefghijklmnopqrstuvwxyz4"
669d9c4a9afSPaul Robinson     "abcdefghijklmnopqrstuvwxyz3abcdefghijklmnopqrstuvwxyz2"
670d9c4a9afSPaul Robinson     "abcdefghijklmnopqrstuvwxyz1abcdefghijklmnopqrstuvwxyz0";
6711b6c7347SPaul Robinson   ASSERT_NO_ERROR(fs::createTemporaryFile(Path216, "", TempPath));
672d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::remove(Twine(TempPath)));
673fcdf9a82SAaron Ballman #endif
674346a1338SMichael J. Spencer }
6757ecd94ccSMichael J. Spencer 
6769b36f51aSBob Haarman TEST_F(FileSystemTest, TempFileCollisions) {
6779b36f51aSBob Haarman   SmallString<128> TestDirectory;
6789b36f51aSBob Haarman   ASSERT_NO_ERROR(
6799b36f51aSBob Haarman       fs::createUniqueDirectory("CreateUniqueFileTest", TestDirectory));
6809b36f51aSBob Haarman   FileRemover Cleanup(TestDirectory);
6819b36f51aSBob Haarman   SmallString<128> Model = TestDirectory;
6829b36f51aSBob Haarman   path::append(Model, "%.tmp");
6839b36f51aSBob Haarman   SmallString<128> Path;
6849b36f51aSBob Haarman   std::vector<fs::TempFile> TempFiles;
6859b36f51aSBob Haarman 
6869b36f51aSBob Haarman   auto TryCreateTempFile = [&]() {
6879b36f51aSBob Haarman     Expected<fs::TempFile> T = fs::TempFile::create(Model);
6889b36f51aSBob Haarman     if (T) {
6899b36f51aSBob Haarman       TempFiles.push_back(std::move(*T));
6909b36f51aSBob Haarman       return true;
6919b36f51aSBob Haarman     } else {
6929b36f51aSBob Haarman       logAllUnhandledErrors(T.takeError(), errs(),
6939b36f51aSBob Haarman                             "Failed to create temporary file: ");
6949b36f51aSBob Haarman       return false;
6959b36f51aSBob Haarman     }
6969b36f51aSBob Haarman   };
6979b36f51aSBob Haarman 
6989b36f51aSBob Haarman   // We should be able to create exactly 16 temporary files.
6999b36f51aSBob Haarman   for (int i = 0; i < 16; ++i)
7009b36f51aSBob Haarman     EXPECT_TRUE(TryCreateTempFile());
7019b36f51aSBob Haarman   EXPECT_FALSE(TryCreateTempFile());
7029b36f51aSBob Haarman 
7039b36f51aSBob Haarman   for (fs::TempFile &T : TempFiles)
7049b36f51aSBob Haarman     cantFail(T.discard());
7059b36f51aSBob Haarman }
7069b36f51aSBob Haarman 
7075c20ac01SRafael Espindola TEST_F(FileSystemTest, CreateDir) {
7085c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::create_directory(Twine(TestDirectory) + "foo"));
7095c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::create_directory(Twine(TestDirectory) + "foo"));
7105c20ac01SRafael Espindola   ASSERT_EQ(fs::create_directory(Twine(TestDirectory) + "foo", false),
7112a826e40SRafael Espindola             errc::file_exists);
7125c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "foo"));
713d9c4a9afSPaul Robinson 
7146b9396c0SFrederic Riss #ifdef LLVM_ON_UNIX
7156b9396c0SFrederic Riss   // Set a 0000 umask so that we can test our directory permissions.
7166b9396c0SFrederic Riss   mode_t OldUmask = ::umask(0000);
7176b9396c0SFrederic Riss 
7186b9396c0SFrederic Riss   fs::file_status Status;
7196b9396c0SFrederic Riss   ASSERT_NO_ERROR(
7206b9396c0SFrederic Riss       fs::create_directory(Twine(TestDirectory) + "baz500", false,
7216b9396c0SFrederic Riss                            fs::perms::owner_read | fs::perms::owner_exe));
7226b9396c0SFrederic Riss   ASSERT_NO_ERROR(fs::status(Twine(TestDirectory) + "baz500", Status));
7236b9396c0SFrederic Riss   ASSERT_EQ(Status.permissions() & fs::perms::all_all,
7246b9396c0SFrederic Riss             fs::perms::owner_read | fs::perms::owner_exe);
7256b9396c0SFrederic Riss   ASSERT_NO_ERROR(fs::create_directory(Twine(TestDirectory) + "baz777", false,
7266b9396c0SFrederic Riss                                        fs::perms::all_all));
7276b9396c0SFrederic Riss   ASSERT_NO_ERROR(fs::status(Twine(TestDirectory) + "baz777", Status));
7286b9396c0SFrederic Riss   ASSERT_EQ(Status.permissions() & fs::perms::all_all, fs::perms::all_all);
7296b9396c0SFrederic Riss 
7306b9396c0SFrederic Riss   // Restore umask to be safe.
7316b9396c0SFrederic Riss   ::umask(OldUmask);
7326b9396c0SFrederic Riss #endif
7336b9396c0SFrederic Riss 
734712e8d29SNico Weber #ifdef _WIN32
735d9c4a9afSPaul Robinson   // Prove that create_directories() can handle a pathname > 248 characters,
736d9c4a9afSPaul Robinson   // which is the documented limit for CreateDirectory().
737d9c4a9afSPaul Robinson   // (248 is MAX_PATH subtracting room for an 8.3 filename.)
738d9c4a9afSPaul Robinson   // Generate a directory path guaranteed to fall into that range.
739d9c4a9afSPaul Robinson   size_t TmpLen = TestDirectory.size();
740d9c4a9afSPaul Robinson   const char *OneDir = "\\123456789";
741d9c4a9afSPaul Robinson   size_t OneDirLen = strlen(OneDir);
742493456e7SAaron Ballman   ASSERT_LT(OneDirLen, 12U);
743d9c4a9afSPaul Robinson   size_t NLevels = ((248 - TmpLen) / OneDirLen) + 1;
744d9c4a9afSPaul Robinson   SmallString<260> LongDir(TestDirectory);
745d9c4a9afSPaul Robinson   for (size_t I = 0; I < NLevels; ++I)
746d9c4a9afSPaul Robinson     LongDir.append(OneDir);
747d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::create_directories(Twine(LongDir)));
748d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::create_directories(Twine(LongDir)));
749d9c4a9afSPaul Robinson   ASSERT_EQ(fs::create_directories(Twine(LongDir), false),
750d9c4a9afSPaul Robinson             errc::file_exists);
751d9c4a9afSPaul Robinson   // Tidy up, "recursively" removing the directories.
752d9c4a9afSPaul Robinson   StringRef ThisDir(LongDir);
753d9c4a9afSPaul Robinson   for (size_t J = 0; J < NLevels; ++J) {
754d9c4a9afSPaul Robinson     ASSERT_NO_ERROR(fs::remove(ThisDir));
755d9c4a9afSPaul Robinson     ThisDir = path::parent_path(ThisDir);
756d9c4a9afSPaul Robinson   }
757d9c4a9afSPaul Robinson 
7583d48bb5fSPirama Arumuga Nainar   // Also verify that paths with Unix separators are handled correctly.
7593d48bb5fSPirama Arumuga Nainar   std::string LongPathWithUnixSeparators(TestDirectory.str());
7603d48bb5fSPirama Arumuga Nainar   // Add at least one subdirectory to TestDirectory, and replace slashes with
7613d48bb5fSPirama Arumuga Nainar   // backslashes
7623d48bb5fSPirama Arumuga Nainar   do {
7633d48bb5fSPirama Arumuga Nainar     LongPathWithUnixSeparators.append("/DirNameWith19Charss");
7643d48bb5fSPirama Arumuga Nainar   } while (LongPathWithUnixSeparators.size() < 260);
7653d48bb5fSPirama Arumuga Nainar   std::replace(LongPathWithUnixSeparators.begin(),
7663d48bb5fSPirama Arumuga Nainar                LongPathWithUnixSeparators.end(),
7673d48bb5fSPirama Arumuga Nainar                '\\', '/');
7683d48bb5fSPirama Arumuga Nainar   ASSERT_NO_ERROR(fs::create_directories(Twine(LongPathWithUnixSeparators)));
7693d48bb5fSPirama Arumuga Nainar   // cleanup
7703d48bb5fSPirama Arumuga Nainar   ASSERT_NO_ERROR(fs::remove_directories(Twine(TestDirectory) +
7713d48bb5fSPirama Arumuga Nainar                                          "/DirNameWith19Charss"));
7723d48bb5fSPirama Arumuga Nainar 
773d9c4a9afSPaul Robinson   // Similarly for a relative pathname.  Need to set the current directory to
774d9c4a9afSPaul Robinson   // TestDirectory so that the one we create ends up in the right place.
775d9c4a9afSPaul Robinson   char PreviousDir[260];
776d9c4a9afSPaul Robinson   size_t PreviousDirLen = ::GetCurrentDirectoryA(260, PreviousDir);
777493456e7SAaron Ballman   ASSERT_GT(PreviousDirLen, 0U);
778493456e7SAaron Ballman   ASSERT_LT(PreviousDirLen, 260U);
779d9c4a9afSPaul Robinson   ASSERT_NE(::SetCurrentDirectoryA(TestDirectory.c_str()), 0);
780d9c4a9afSPaul Robinson   LongDir.clear();
781d9c4a9afSPaul Robinson   // Generate a relative directory name with absolute length > 248.
782d9c4a9afSPaul Robinson   size_t LongDirLen = 249 - TestDirectory.size();
783d9c4a9afSPaul Robinson   LongDir.assign(LongDirLen, 'a');
784d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::create_directory(Twine(LongDir)));
785d9c4a9afSPaul Robinson   // While we're here, prove that .. and . handling works in these long paths.
786d9c4a9afSPaul Robinson   const char *DotDotDirs = "\\..\\.\\b";
787d9c4a9afSPaul Robinson   LongDir.append(DotDotDirs);
7881b6c7347SPaul Robinson   ASSERT_NO_ERROR(fs::create_directory("b"));
789d9c4a9afSPaul Robinson   ASSERT_EQ(fs::create_directory(Twine(LongDir), false), errc::file_exists);
790d9c4a9afSPaul Robinson   // And clean up.
7911b6c7347SPaul Robinson   ASSERT_NO_ERROR(fs::remove("b"));
792d9c4a9afSPaul Robinson   ASSERT_NO_ERROR(fs::remove(
793d9c4a9afSPaul Robinson     Twine(LongDir.substr(0, LongDir.size() - strlen(DotDotDirs)))));
794d9c4a9afSPaul Robinson   ASSERT_NE(::SetCurrentDirectoryA(PreviousDir), 0);
795d9c4a9afSPaul Robinson #endif
7965c20ac01SRafael Espindola }
7975c20ac01SRafael Espindola 
798346a1338SMichael J. Spencer TEST_F(FileSystemTest, DirectoryIteration) {
799c049c657SRafael Espindola   std::error_code ec;
8001f063600SMichael J. Spencer   for (fs::directory_iterator i(".", ec), e; i != e; i.increment(ec))
8011f063600SMichael J. Spencer     ASSERT_NO_ERROR(ec);
8020a7625d6SMichael J. Spencer 
8030a7625d6SMichael J. Spencer   // Create a known hierarchy to recurse over.
8045c20ac01SRafael Espindola   ASSERT_NO_ERROR(
8055c20ac01SRafael Espindola       fs::create_directories(Twine(TestDirectory) + "/recursive/a0/aa1"));
8065c20ac01SRafael Espindola   ASSERT_NO_ERROR(
8075c20ac01SRafael Espindola       fs::create_directories(Twine(TestDirectory) + "/recursive/a0/ab1"));
8085c20ac01SRafael Espindola   ASSERT_NO_ERROR(fs::create_directories(Twine(TestDirectory) +
8095c20ac01SRafael Espindola                                          "/recursive/dontlookhere/da1"));
8105c20ac01SRafael Espindola   ASSERT_NO_ERROR(
8115c20ac01SRafael Espindola       fs::create_directories(Twine(TestDirectory) + "/recursive/z0/za1"));
8125c20ac01SRafael Espindola   ASSERT_NO_ERROR(
8135c20ac01SRafael Espindola       fs::create_directories(Twine(TestDirectory) + "/recursive/pop/p1"));
8140a7625d6SMichael J. Spencer   typedef std::vector<std::string> v_t;
8150a7625d6SMichael J. Spencer   v_t visited;
8160a7625d6SMichael J. Spencer   for (fs::recursive_directory_iterator i(Twine(TestDirectory)
8170a7625d6SMichael J. Spencer          + "/recursive", ec), e; i != e; i.increment(ec)){
8180a7625d6SMichael J. Spencer     ASSERT_NO_ERROR(ec);
8190cfb3675SNAKAMURA Takumi     if (path::filename(i->path()) == "p1") {
8200a7625d6SMichael J. Spencer       i.pop();
8210cfb3675SNAKAMURA Takumi       // FIXME: recursive_directory_iterator should be more robust.
8220cfb3675SNAKAMURA Takumi       if (i == e) break;
8230cfb3675SNAKAMURA Takumi     }
8249b54a3e2SMichael J. Spencer     if (path::filename(i->path()) == "dontlookhere")
8259b54a3e2SMichael J. Spencer       i.no_push();
8260a7625d6SMichael J. Spencer     visited.push_back(path::filename(i->path()));
8270a7625d6SMichael J. Spencer   }
8280d955d0bSDavid Majnemer   v_t::const_iterator a0 = find(visited, "a0");
8290d955d0bSDavid Majnemer   v_t::const_iterator aa1 = find(visited, "aa1");
8300d955d0bSDavid Majnemer   v_t::const_iterator ab1 = find(visited, "ab1");
8310d955d0bSDavid Majnemer   v_t::const_iterator dontlookhere = find(visited, "dontlookhere");
8320d955d0bSDavid Majnemer   v_t::const_iterator da1 = find(visited, "da1");
8330d955d0bSDavid Majnemer   v_t::const_iterator z0 = find(visited, "z0");
8340d955d0bSDavid Majnemer   v_t::const_iterator za1 = find(visited, "za1");
8350d955d0bSDavid Majnemer   v_t::const_iterator pop = find(visited, "pop");
8360d955d0bSDavid Majnemer   v_t::const_iterator p1 = find(visited, "p1");
8370a7625d6SMichael J. Spencer 
8380a7625d6SMichael J. Spencer   // Make sure that each path was visited correctly.
8390a7625d6SMichael J. Spencer   ASSERT_NE(a0, visited.end());
8400a7625d6SMichael J. Spencer   ASSERT_NE(aa1, visited.end());
8410a7625d6SMichael J. Spencer   ASSERT_NE(ab1, visited.end());
8420a7625d6SMichael J. Spencer   ASSERT_NE(dontlookhere, visited.end());
8430a7625d6SMichael J. Spencer   ASSERT_EQ(da1, visited.end()); // Not visited.
8440a7625d6SMichael J. Spencer   ASSERT_NE(z0, visited.end());
8450a7625d6SMichael J. Spencer   ASSERT_NE(za1, visited.end());
8460a7625d6SMichael J. Spencer   ASSERT_NE(pop, visited.end());
8470a7625d6SMichael J. Spencer   ASSERT_EQ(p1, visited.end()); // Not visited.
8480a7625d6SMichael J. Spencer 
8490a7625d6SMichael J. Spencer   // Make sure that parents were visited before children. No other ordering
8500a7625d6SMichael J. Spencer   // guarantees can be made across siblings.
8510a7625d6SMichael J. Spencer   ASSERT_LT(a0, aa1);
8520a7625d6SMichael J. Spencer   ASSERT_LT(a0, ab1);
8530a7625d6SMichael J. Spencer   ASSERT_LT(z0, za1);
85478dcc03cSRafael Espindola 
85578dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/a0/aa1"));
85678dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/a0/ab1"));
85778dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/a0"));
85878dcc03cSRafael Espindola   ASSERT_NO_ERROR(
85978dcc03cSRafael Espindola       fs::remove(Twine(TestDirectory) + "/recursive/dontlookhere/da1"));
86078dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/dontlookhere"));
86178dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/pop/p1"));
86278dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/pop"));
86378dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/z0/za1"));
86478dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/z0"));
86578dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive"));
866ead771cbSBruno Cardoso Lopes 
867ead771cbSBruno Cardoso Lopes   // Test recursive_directory_iterator level()
868ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(
869ead771cbSBruno Cardoso Lopes       fs::create_directories(Twine(TestDirectory) + "/reclevel/a/b/c"));
870ead771cbSBruno Cardoso Lopes   fs::recursive_directory_iterator I(Twine(TestDirectory) + "/reclevel", ec), E;
871ead771cbSBruno Cardoso Lopes   for (int l = 0; I != E; I.increment(ec), ++l) {
872ead771cbSBruno Cardoso Lopes     ASSERT_NO_ERROR(ec);
873ead771cbSBruno Cardoso Lopes     EXPECT_EQ(I.level(), l);
874ead771cbSBruno Cardoso Lopes   }
875ead771cbSBruno Cardoso Lopes   EXPECT_EQ(I, E);
876ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel/a/b/c"));
877ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel/a/b"));
878ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel/a"));
879ead771cbSBruno Cardoso Lopes   ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/reclevel"));
8807ecd94ccSMichael J. Spencer }
8811d3c4a7bSMichael J. Spencer 
882e7d0fa68SJuergen Ributzka #ifdef LLVM_ON_UNIX
883e7d0fa68SJuergen Ributzka TEST_F(FileSystemTest, BrokenSymlinkDirectoryIteration) {
884e7d0fa68SJuergen Ributzka   // Create a known hierarchy to recurse over.
885e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(fs::create_directories(Twine(TestDirectory) + "/symlink"));
886e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(
887e7d0fa68SJuergen Ributzka       fs::create_link("no_such_file", Twine(TestDirectory) + "/symlink/a"));
888e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(
889e7d0fa68SJuergen Ributzka       fs::create_directories(Twine(TestDirectory) + "/symlink/b/bb"));
890e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(
891e7d0fa68SJuergen Ributzka       fs::create_link("no_such_file", Twine(TestDirectory) + "/symlink/b/ba"));
892e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(
893e7d0fa68SJuergen Ributzka       fs::create_link("no_such_file", Twine(TestDirectory) + "/symlink/b/bc"));
894e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(
895e7d0fa68SJuergen Ributzka       fs::create_link("no_such_file", Twine(TestDirectory) + "/symlink/c"));
896e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(
897e7d0fa68SJuergen Ributzka       fs::create_directories(Twine(TestDirectory) + "/symlink/d/dd/ddd"));
898e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(fs::create_link(Twine(TestDirectory) + "/symlink/d/dd",
899e7d0fa68SJuergen Ributzka                                   Twine(TestDirectory) + "/symlink/d/da"));
900e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(
901e7d0fa68SJuergen Ributzka       fs::create_link("no_such_file", Twine(TestDirectory) + "/symlink/e"));
902e7d0fa68SJuergen Ributzka 
903e7d0fa68SJuergen Ributzka   typedef std::vector<std::string> v_t;
904650fd6c3SMax Moroz   v_t VisitedNonBrokenSymlinks;
905650fd6c3SMax Moroz   v_t VisitedBrokenSymlinks;
906e7d0fa68SJuergen Ributzka   std::error_code ec;
90779c995c0SSam McCall   using testing::UnorderedElementsAre;
90879c995c0SSam McCall   using testing::UnorderedElementsAreArray;
909650fd6c3SMax Moroz 
910650fd6c3SMax Moroz   // Broken symbol links are expected to throw an error.
911e7d0fa68SJuergen Ributzka   for (fs::directory_iterator i(Twine(TestDirectory) + "/symlink", ec), e;
912e7d0fa68SJuergen Ributzka        i != e; i.increment(ec)) {
91379c995c0SSam McCall     ASSERT_NO_ERROR(ec);
91479c995c0SSam McCall     if (i->status().getError() ==
91579c995c0SSam McCall         std::make_error_code(std::errc::no_such_file_or_directory)) {
916650fd6c3SMax Moroz       VisitedBrokenSymlinks.push_back(path::filename(i->path()));
917e7d0fa68SJuergen Ributzka       continue;
918e7d0fa68SJuergen Ributzka     }
919650fd6c3SMax Moroz     VisitedNonBrokenSymlinks.push_back(path::filename(i->path()));
920e7d0fa68SJuergen Ributzka   }
92179c995c0SSam McCall   EXPECT_THAT(VisitedNonBrokenSymlinks, UnorderedElementsAre("b", "d"));
922650fd6c3SMax Moroz   VisitedNonBrokenSymlinks.clear();
923650fd6c3SMax Moroz 
92479c995c0SSam McCall   EXPECT_THAT(VisitedBrokenSymlinks, UnorderedElementsAre("a", "c", "e"));
925650fd6c3SMax Moroz   VisitedBrokenSymlinks.clear();
926650fd6c3SMax Moroz 
927650fd6c3SMax Moroz   // Broken symbol links are expected to throw an error.
928650fd6c3SMax Moroz   for (fs::recursive_directory_iterator i(
929650fd6c3SMax Moroz       Twine(TestDirectory) + "/symlink", ec), e; i != e; i.increment(ec)) {
93079c995c0SSam McCall     ASSERT_NO_ERROR(ec);
93179c995c0SSam McCall     if (i->status().getError() ==
93279c995c0SSam McCall         std::make_error_code(std::errc::no_such_file_or_directory)) {
933650fd6c3SMax Moroz       VisitedBrokenSymlinks.push_back(path::filename(i->path()));
934650fd6c3SMax Moroz       continue;
935650fd6c3SMax Moroz     }
936650fd6c3SMax Moroz     VisitedNonBrokenSymlinks.push_back(path::filename(i->path()));
937650fd6c3SMax Moroz   }
93879c995c0SSam McCall   EXPECT_THAT(VisitedNonBrokenSymlinks,
93979c995c0SSam McCall               UnorderedElementsAre("b", "bb", "d", "da", "dd", "ddd", "ddd"));
940650fd6c3SMax Moroz   VisitedNonBrokenSymlinks.clear();
941650fd6c3SMax Moroz 
94279c995c0SSam McCall   EXPECT_THAT(VisitedBrokenSymlinks,
94379c995c0SSam McCall               UnorderedElementsAre("a", "ba", "bc", "c", "e"));
944650fd6c3SMax Moroz   VisitedBrokenSymlinks.clear();
945650fd6c3SMax Moroz 
946650fd6c3SMax Moroz   for (fs::recursive_directory_iterator i(
947650fd6c3SMax Moroz       Twine(TestDirectory) + "/symlink", ec, /*follow_symlinks=*/false), e;
948650fd6c3SMax Moroz        i != e; i.increment(ec)) {
94979c995c0SSam McCall     ASSERT_NO_ERROR(ec);
95079c995c0SSam McCall     if (i->status().getError() ==
95179c995c0SSam McCall         std::make_error_code(std::errc::no_such_file_or_directory)) {
952650fd6c3SMax Moroz       VisitedBrokenSymlinks.push_back(path::filename(i->path()));
953650fd6c3SMax Moroz       continue;
954650fd6c3SMax Moroz     }
955650fd6c3SMax Moroz     VisitedNonBrokenSymlinks.push_back(path::filename(i->path()));
956650fd6c3SMax Moroz   }
95779c995c0SSam McCall   EXPECT_THAT(VisitedNonBrokenSymlinks,
95879c995c0SSam McCall               UnorderedElementsAreArray({"a", "b", "ba", "bb", "bc", "c", "d",
95979c995c0SSam McCall                                          "da", "dd", "ddd", "e"}));
960650fd6c3SMax Moroz   VisitedNonBrokenSymlinks.clear();
961650fd6c3SMax Moroz 
96279c995c0SSam McCall   EXPECT_THAT(VisitedBrokenSymlinks, UnorderedElementsAre());
963650fd6c3SMax Moroz   VisitedBrokenSymlinks.clear();
964e7d0fa68SJuergen Ributzka 
965e7d0fa68SJuergen Ributzka   ASSERT_NO_ERROR(fs::remove_directories(Twine(TestDirectory) + "/symlink"));
966e7d0fa68SJuergen Ributzka }
967e7d0fa68SJuergen Ributzka #endif
968e7d0fa68SJuergen Ributzka 
969260bda3fSZachary Turner TEST_F(FileSystemTest, Remove) {
970260bda3fSZachary Turner   SmallString<64> BaseDir;
971260bda3fSZachary Turner   SmallString<64> Paths[4];
972260bda3fSZachary Turner   int fds[4];
973260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueDirectory("fs_remove", BaseDir));
974260bda3fSZachary Turner 
975260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::create_directories(Twine(BaseDir) + "/foo/bar/baz"));
976260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::create_directories(Twine(BaseDir) + "/foo/bar/buzz"));
977260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueFile(
978260bda3fSZachary Turner       Twine(BaseDir) + "/foo/bar/baz/%%%%%%.tmp", fds[0], Paths[0]));
979260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueFile(
980260bda3fSZachary Turner       Twine(BaseDir) + "/foo/bar/baz/%%%%%%.tmp", fds[1], Paths[1]));
981260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueFile(
982260bda3fSZachary Turner       Twine(BaseDir) + "/foo/bar/buzz/%%%%%%.tmp", fds[2], Paths[2]));
983260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::createUniqueFile(
984260bda3fSZachary Turner       Twine(BaseDir) + "/foo/bar/buzz/%%%%%%.tmp", fds[3], Paths[3]));
985260bda3fSZachary Turner 
986260bda3fSZachary Turner   for (int fd : fds)
987260bda3fSZachary Turner     ::close(fd);
988260bda3fSZachary Turner 
989260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Twine(BaseDir) + "/foo/bar/baz"));
990260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Twine(BaseDir) + "/foo/bar/buzz"));
991260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Paths[0]));
992260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Paths[1]));
993260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Paths[2]));
994260bda3fSZachary Turner   EXPECT_TRUE(fs::exists(Paths[3]));
995260bda3fSZachary Turner 
996260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::remove_directories("D:/footest"));
997260bda3fSZachary Turner 
998260bda3fSZachary Turner   ASSERT_NO_ERROR(fs::remove_directories(BaseDir));
999260bda3fSZachary Turner   ASSERT_FALSE(fs::exists(BaseDir));
1000260bda3fSZachary Turner }
1001260bda3fSZachary Turner 
1002712e8d29SNico Weber #ifdef _WIN32
100384ab9b3bSRafael Espindola TEST_F(FileSystemTest, CarriageReturn) {
100484ab9b3bSRafael Espindola   SmallString<128> FilePathname(TestDirectory);
1005d83c63b7SReid Kleckner   std::error_code EC;
100684ab9b3bSRafael Espindola   path::append(FilePathname, "test");
100784ab9b3bSRafael Espindola 
100884ab9b3bSRafael Espindola   {
1009d83c63b7SReid Kleckner     raw_fd_ostream File(FilePathname, EC, sys::fs::F_Text);
1010d83c63b7SReid Kleckner     ASSERT_NO_ERROR(EC);
101184ab9b3bSRafael Espindola     File << '\n';
101284ab9b3bSRafael Espindola   }
101384ab9b3bSRafael Espindola   {
1014d83c63b7SReid Kleckner     auto Buf = MemoryBuffer::getFile(FilePathname.str());
1015586ee604SAaron Ballman     EXPECT_TRUE((bool)Buf);
1016e56fe8afSAaron Ballman     EXPECT_EQ(Buf.get()->getBuffer(), "\r\n");
101784ab9b3bSRafael Espindola   }
101884ab9b3bSRafael Espindola 
101984ab9b3bSRafael Espindola   {
1020d83c63b7SReid Kleckner     raw_fd_ostream File(FilePathname, EC, sys::fs::F_None);
1021d83c63b7SReid Kleckner     ASSERT_NO_ERROR(EC);
102284ab9b3bSRafael Espindola     File << '\n';
102384ab9b3bSRafael Espindola   }
102484ab9b3bSRafael Espindola   {
1025d83c63b7SReid Kleckner     auto Buf = MemoryBuffer::getFile(FilePathname.str());
1026586ee604SAaron Ballman     EXPECT_TRUE((bool)Buf);
1027e56fe8afSAaron Ballman     EXPECT_EQ(Buf.get()->getBuffer(), "\n");
102884ab9b3bSRafael Espindola   }
102978dcc03cSRafael Espindola   ASSERT_NO_ERROR(fs::remove(Twine(FilePathname)));
103084ab9b3bSRafael Espindola }
103184ab9b3bSRafael Espindola #endif
103284ab9b3bSRafael Espindola 
103359aaa6c0SRafael Espindola TEST_F(FileSystemTest, Resize) {
103459aaa6c0SRafael Espindola   int FD;
103559aaa6c0SRafael Espindola   SmallString<64> TempPath;
103659aaa6c0SRafael Espindola   ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "temp", FD, TempPath));
103759aaa6c0SRafael Espindola   ASSERT_NO_ERROR(fs::resize_file(FD, 123));
103859aaa6c0SRafael Espindola   fs::file_status Status;
103959aaa6c0SRafael Espindola   ASSERT_NO_ERROR(fs::status(FD, Status));
104059aaa6c0SRafael Espindola   ASSERT_EQ(Status.getSize(), 123U);
104175e557f1SReid Kleckner   ::close(FD);
104275e557f1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(TempPath));
104359aaa6c0SRafael Espindola }
104459aaa6c0SRafael Espindola 
104582a0c97bSZachary Turner TEST_F(FileSystemTest, MD5) {
104682a0c97bSZachary Turner   int FD;
104782a0c97bSZachary Turner   SmallString<64> TempPath;
104882a0c97bSZachary Turner   ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "temp", FD, TempPath));
104982a0c97bSZachary Turner   StringRef Data("abcdefghijklmnopqrstuvwxyz");
10504f5771d7SGalina Kistanova   ASSERT_EQ(write(FD, Data.data(), Data.size()), static_cast<ssize_t>(Data.size()));
105182a0c97bSZachary Turner   lseek(FD, 0, SEEK_SET);
105282a0c97bSZachary Turner   auto Hash = fs::md5_contents(FD);
105382a0c97bSZachary Turner   ::close(FD);
105482a0c97bSZachary Turner   ASSERT_NO_ERROR(Hash.getError());
105582a0c97bSZachary Turner 
105682a0c97bSZachary Turner   EXPECT_STREQ("c3fcd3d76192e4007dfb496cca67e13b", Hash->digest().c_str());
105782a0c97bSZachary Turner }
105882a0c97bSZachary Turner 
105918497e92SNick Kledzik TEST_F(FileSystemTest, FileMapping) {
106018497e92SNick Kledzik   // Create a temp file.
106118497e92SNick Kledzik   int FileDescriptor;
106218497e92SNick Kledzik   SmallString<64> TempPath;
106318497e92SNick Kledzik   ASSERT_NO_ERROR(
1064155cf0f3SRafael Espindola       fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
1065c69f13bfSRafael Espindola   unsigned Size = 4096;
1066c69f13bfSRafael Espindola   ASSERT_NO_ERROR(fs::resize_file(FileDescriptor, Size));
1067c69f13bfSRafael Espindola 
106818497e92SNick Kledzik   // Map in temp file and add some content
1069c049c657SRafael Espindola   std::error_code EC;
1070ef2284fbSMichael J. Spencer   StringRef Val("hello there");
1071ef2284fbSMichael J. Spencer   {
1072ef2284fbSMichael J. Spencer     fs::mapped_file_region mfr(FileDescriptor,
1073c69f13bfSRafael Espindola                                fs::mapped_file_region::readwrite, Size, 0, EC);
1074ef2284fbSMichael J. Spencer     ASSERT_NO_ERROR(EC);
1075ef2284fbSMichael J. Spencer     std::copy(Val.begin(), Val.end(), mfr.data());
1076ef2284fbSMichael J. Spencer     // Explicitly add a 0.
1077ef2284fbSMichael J. Spencer     mfr.data()[Val.size()] = 0;
107818497e92SNick Kledzik     // Unmap temp file
1079ef2284fbSMichael J. Spencer   }
108075e557f1SReid Kleckner   ASSERT_EQ(close(FileDescriptor), 0);
108118497e92SNick Kledzik 
108218497e92SNick Kledzik   // Map it back in read-only
108375e557f1SReid Kleckner   {
108471bc507cSRafael Espindola     int FD;
108571bc507cSRafael Espindola     EC = fs::openFileForRead(Twine(TempPath), FD);
108671bc507cSRafael Espindola     ASSERT_NO_ERROR(EC);
1087c69f13bfSRafael Espindola     fs::mapped_file_region mfr(FD, fs::mapped_file_region::readonly, Size, 0, EC);
1088ef2284fbSMichael J. Spencer     ASSERT_NO_ERROR(EC);
108918497e92SNick Kledzik 
109018497e92SNick Kledzik     // Verify content
1091ef2284fbSMichael J. Spencer     EXPECT_EQ(StringRef(mfr.const_data()), Val);
109218497e92SNick Kledzik 
109318497e92SNick Kledzik     // Unmap temp file
1094c69f13bfSRafael Espindola     fs::mapped_file_region m(FD, fs::mapped_file_region::readonly, Size, 0, EC);
1095ef2284fbSMichael J. Spencer     ASSERT_NO_ERROR(EC);
109671bc507cSRafael Espindola     ASSERT_EQ(close(FD), 0);
1097ef2284fbSMichael J. Spencer   }
109875e557f1SReid Kleckner   ASSERT_NO_ERROR(fs::remove(TempPath));
109975e557f1SReid Kleckner }
1100afc50b3eSSaleem Abdulrasool 
1101afc50b3eSSaleem Abdulrasool TEST(Support, NormalizePath) {
11028fb09e3eSZachary Turner   using TestTuple = std::tuple<const char *, const char *, const char *>;
11038fb09e3eSZachary Turner   std::vector<TestTuple> Tests;
11048fb09e3eSZachary Turner   Tests.emplace_back("a", "a", "a");
11058fb09e3eSZachary Turner   Tests.emplace_back("a/b", "a\\b", "a/b");
11068fb09e3eSZachary Turner   Tests.emplace_back("a\\b", "a\\b", "a/b");
11078fb09e3eSZachary Turner   Tests.emplace_back("a\\\\b", "a\\\\b", "a\\\\b");
11088fb09e3eSZachary Turner   Tests.emplace_back("\\a", "\\a", "/a");
11098fb09e3eSZachary Turner   Tests.emplace_back("a\\", "a\\", "a/");
11108fb09e3eSZachary Turner 
11115c5091fcSZachary Turner   for (auto &T : Tests) {
111237350060SZachary Turner     SmallString<64> Win(std::get<0>(T));
111337350060SZachary Turner     SmallString<64> Posix(Win);
11145c5091fcSZachary Turner     path::native(Win, path::Style::windows);
11155c5091fcSZachary Turner     path::native(Posix, path::Style::posix);
11165c5091fcSZachary Turner     EXPECT_EQ(std::get<1>(T), Win);
11175c5091fcSZachary Turner     EXPECT_EQ(std::get<2>(T), Posix);
11185c5091fcSZachary Turner   }
11199c761a36SSerge Pavlov 
1120712e8d29SNico Weber #if defined(_WIN32)
11219c761a36SSerge Pavlov   SmallString<64> PathHome;
11229c761a36SSerge Pavlov   path::home_directory(PathHome);
11239c761a36SSerge Pavlov 
11249c761a36SSerge Pavlov   const char *Path7a = "~/aaa";
11259c761a36SSerge Pavlov   SmallString<64> Path7(Path7a);
11269c761a36SSerge Pavlov   path::native(Path7);
11279c761a36SSerge Pavlov   EXPECT_TRUE(Path7.endswith("\\aaa"));
11289c761a36SSerge Pavlov   EXPECT_TRUE(Path7.startswith(PathHome));
11299c761a36SSerge Pavlov   EXPECT_EQ(Path7.size(), PathHome.size() + strlen(Path7a + 1));
11309c761a36SSerge Pavlov 
11319c761a36SSerge Pavlov   const char *Path8a = "~";
11329c761a36SSerge Pavlov   SmallString<64> Path8(Path8a);
11339c761a36SSerge Pavlov   path::native(Path8);
11349c761a36SSerge Pavlov   EXPECT_EQ(Path8, PathHome);
11359c761a36SSerge Pavlov 
11369c761a36SSerge Pavlov   const char *Path9a = "~aaa";
11379c761a36SSerge Pavlov   SmallString<64> Path9(Path9a);
11389c761a36SSerge Pavlov   path::native(Path9);
11399c761a36SSerge Pavlov   EXPECT_EQ(Path9, "~aaa");
11409c761a36SSerge Pavlov 
11419c761a36SSerge Pavlov   const char *Path10a = "aaa/~/b";
11429c761a36SSerge Pavlov   SmallString<64> Path10(Path10a);
11439c761a36SSerge Pavlov   path::native(Path10);
11449c761a36SSerge Pavlov   EXPECT_EQ(Path10, "aaa\\~\\b");
11459c761a36SSerge Pavlov #endif
1146afc50b3eSSaleem Abdulrasool }
1147a26be4a9SDouglas Katzman 
1148a26be4a9SDouglas Katzman TEST(Support, RemoveLeadingDotSlash) {
1149a26be4a9SDouglas Katzman   StringRef Path1("././/foolz/wat");
1150a26be4a9SDouglas Katzman   StringRef Path2("./////");
1151a26be4a9SDouglas Katzman 
1152a26be4a9SDouglas Katzman   Path1 = path::remove_leading_dotslash(Path1);
1153a26be4a9SDouglas Katzman   EXPECT_EQ(Path1, "foolz/wat");
1154a26be4a9SDouglas Katzman   Path2 = path::remove_leading_dotslash(Path2);
1155a26be4a9SDouglas Katzman   EXPECT_EQ(Path2, "");
1156a26be4a9SDouglas Katzman }
1157662b4fd3SMike Aizatsky 
11585c5091fcSZachary Turner static std::string remove_dots(StringRef path, bool remove_dot_dot,
11595c5091fcSZachary Turner                                path::Style style) {
1160662b4fd3SMike Aizatsky   SmallString<256> buffer(path);
11615c5091fcSZachary Turner   path::remove_dots(buffer, remove_dot_dot, style);
1162662b4fd3SMike Aizatsky   return buffer.str();
1163662b4fd3SMike Aizatsky }
1164662b4fd3SMike Aizatsky 
1165662b4fd3SMike Aizatsky TEST(Support, RemoveDots) {
11665c5091fcSZachary Turner   EXPECT_EQ("foolz\\wat",
11675c5091fcSZachary Turner             remove_dots(".\\.\\\\foolz\\wat", false, path::Style::windows));
1168f81f3a83SGreg Clayton   EXPECT_EQ("", remove_dots(".\\\\\\\\\\", false, path::Style::windows));
1169f8cf7133SMike Aizatsky 
11705c5091fcSZachary Turner   EXPECT_EQ("a\\..\\b\\c",
11715c5091fcSZachary Turner             remove_dots(".\\a\\..\\b\\c", false, path::Style::windows));
11725c5091fcSZachary Turner   EXPECT_EQ("b\\c", remove_dots(".\\a\\..\\b\\c", true, path::Style::windows));
11735c5091fcSZachary Turner   EXPECT_EQ("c", remove_dots(".\\.\\c", true, path::Style::windows));
11745c5091fcSZachary Turner   EXPECT_EQ("..\\a\\c",
11755c5091fcSZachary Turner             remove_dots("..\\a\\b\\..\\c", true, path::Style::windows));
11765c5091fcSZachary Turner   EXPECT_EQ("..\\..\\a\\c",
11775c5091fcSZachary Turner             remove_dots("..\\..\\a\\b\\..\\c", true, path::Style::windows));
1178f8cf7133SMike Aizatsky 
1179f8cf7133SMike Aizatsky   SmallString<64> Path1(".\\.\\c");
11805c5091fcSZachary Turner   EXPECT_TRUE(path::remove_dots(Path1, true, path::Style::windows));
1181f8cf7133SMike Aizatsky   EXPECT_EQ("c", Path1);
1182662b4fd3SMike Aizatsky 
11835c5091fcSZachary Turner   EXPECT_EQ("foolz/wat",
11845c5091fcSZachary Turner             remove_dots("././/foolz/wat", false, path::Style::posix));
1185f81f3a83SGreg Clayton   EXPECT_EQ("", remove_dots("./////", false, path::Style::posix));
1186662b4fd3SMike Aizatsky 
11875c5091fcSZachary Turner   EXPECT_EQ("a/../b/c", remove_dots("./a/../b/c", false, path::Style::posix));
11885c5091fcSZachary Turner   EXPECT_EQ("b/c", remove_dots("./a/../b/c", true, path::Style::posix));
11895c5091fcSZachary Turner   EXPECT_EQ("c", remove_dots("././c", true, path::Style::posix));
11905c5091fcSZachary Turner   EXPECT_EQ("../a/c", remove_dots("../a/b/../c", true, path::Style::posix));
11915c5091fcSZachary Turner   EXPECT_EQ("../../a/c",
11925c5091fcSZachary Turner             remove_dots("../../a/b/../c", true, path::Style::posix));
11935c5091fcSZachary Turner   EXPECT_EQ("/a/c", remove_dots("/../../a/c", true, path::Style::posix));
11945c5091fcSZachary Turner   EXPECT_EQ("/a/c",
11955c5091fcSZachary Turner             remove_dots("/../a/b//../././/c", true, path::Style::posix));
11965c5091fcSZachary Turner 
11975c5091fcSZachary Turner   SmallString<64> Path2("././c");
11985c5091fcSZachary Turner   EXPECT_TRUE(path::remove_dots(Path2, true, path::Style::posix));
11995c5091fcSZachary Turner   EXPECT_EQ("c", Path2);
1200662b4fd3SMike Aizatsky }
1201bbd10b45STeresa Johnson 
1202bbd10b45STeresa Johnson TEST(Support, ReplacePathPrefix) {
1203bbd10b45STeresa Johnson   SmallString<64> Path1("/foo");
1204bbd10b45STeresa Johnson   SmallString<64> Path2("/old/foo");
1205bbd10b45STeresa Johnson   SmallString<64> OldPrefix("/old");
1206bbd10b45STeresa Johnson   SmallString<64> NewPrefix("/new");
1207bbd10b45STeresa Johnson   SmallString<64> NewPrefix2("/longernew");
1208bbd10b45STeresa Johnson   SmallString<64> EmptyPrefix("");
1209bbd10b45STeresa Johnson 
1210bbd10b45STeresa Johnson   SmallString<64> Path = Path1;
1211bbd10b45STeresa Johnson   path::replace_path_prefix(Path, OldPrefix, NewPrefix);
1212bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/foo");
1213bbd10b45STeresa Johnson   Path = Path2;
1214bbd10b45STeresa Johnson   path::replace_path_prefix(Path, OldPrefix, NewPrefix);
1215bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/new/foo");
1216bbd10b45STeresa Johnson   Path = Path2;
1217bbd10b45STeresa Johnson   path::replace_path_prefix(Path, OldPrefix, NewPrefix2);
1218bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/longernew/foo");
1219bbd10b45STeresa Johnson   Path = Path1;
1220bbd10b45STeresa Johnson   path::replace_path_prefix(Path, EmptyPrefix, NewPrefix);
1221bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/new/foo");
1222bbd10b45STeresa Johnson   Path = Path2;
1223bbd10b45STeresa Johnson   path::replace_path_prefix(Path, OldPrefix, EmptyPrefix);
1224bbd10b45STeresa Johnson   EXPECT_EQ(Path, "/foo");
1225bbd10b45STeresa Johnson }
1226d9153272STaewook Oh 
1227d9153272STaewook Oh TEST_F(FileSystemTest, OpenFileForRead) {
1228d9153272STaewook Oh   // Create a temp file.
1229d9153272STaewook Oh   int FileDescriptor;
1230d9153272STaewook Oh   SmallString<64> TempPath;
1231d9153272STaewook Oh   ASSERT_NO_ERROR(
1232d9153272STaewook Oh       fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
123375e557f1SReid Kleckner   FileRemover Cleanup(TempPath);
1234d9153272STaewook Oh 
1235d9153272STaewook Oh   // Make sure it exists.
1236d9153272STaewook Oh   ASSERT_TRUE(sys::fs::exists(Twine(TempPath)));
1237d9153272STaewook Oh 
1238d9153272STaewook Oh   // Open the file for read
1239d9153272STaewook Oh   int FileDescriptor2;
1240d9153272STaewook Oh   SmallString<64> ResultPath;
12411f67a3cbSZachary Turner   ASSERT_NO_ERROR(fs::openFileForRead(Twine(TempPath), FileDescriptor2,
12421f67a3cbSZachary Turner                                       fs::OF_None, &ResultPath))
1243d9153272STaewook Oh 
1244d9153272STaewook Oh   // If we succeeded, check that the paths are the same (modulo case):
1245d9153272STaewook Oh   if (!ResultPath.empty()) {
1246d9153272STaewook Oh     // The paths returned by createTemporaryFile and getPathFromOpenFD
1247d9153272STaewook Oh     // should reference the same file on disk.
1248d9153272STaewook Oh     fs::UniqueID D1, D2;
1249d9153272STaewook Oh     ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), D1));
1250d9153272STaewook Oh     ASSERT_NO_ERROR(fs::getUniqueID(Twine(ResultPath), D2));
1251d9153272STaewook Oh     ASSERT_EQ(D1, D2);
1252d9153272STaewook Oh   }
1253d9153272STaewook Oh   ::close(FileDescriptor);
1254089303d8SAndrew Ng   ::close(FileDescriptor2);
1255089303d8SAndrew Ng 
1256089303d8SAndrew Ng #ifdef _WIN32
1257089303d8SAndrew Ng   // Since Windows Vista, file access time is not updated by default.
1258089303d8SAndrew Ng   // This is instead updated manually by openFileForRead.
1259089303d8SAndrew Ng   // https://blogs.technet.microsoft.com/filecab/2006/11/07/disabling-last-access-time-in-windows-vista-to-improve-ntfs-performance/
1260089303d8SAndrew Ng   // This part of the unit test is Windows specific as the updating of
1261089303d8SAndrew Ng   // access times can be disabled on Linux using /etc/fstab.
1262089303d8SAndrew Ng 
1263089303d8SAndrew Ng   // Set access time to UNIX epoch.
1264089303d8SAndrew Ng   ASSERT_NO_ERROR(sys::fs::openFileForWrite(Twine(TempPath), FileDescriptor,
1265089303d8SAndrew Ng                                             fs::CD_OpenExisting));
1266089303d8SAndrew Ng   TimePoint<> Epoch(std::chrono::milliseconds(0));
126797ea4850SJordan Rupprecht   ASSERT_NO_ERROR(fs::setLastAccessAndModificationTime(FileDescriptor, Epoch));
1268089303d8SAndrew Ng   ::close(FileDescriptor);
1269089303d8SAndrew Ng 
1270089303d8SAndrew Ng   // Open the file and ensure access time is updated, when forced.
1271089303d8SAndrew Ng   ASSERT_NO_ERROR(fs::openFileForRead(Twine(TempPath), FileDescriptor,
1272089303d8SAndrew Ng                                       fs::OF_UpdateAtime, &ResultPath));
1273089303d8SAndrew Ng 
1274089303d8SAndrew Ng   sys::fs::file_status Status;
1275089303d8SAndrew Ng   ASSERT_NO_ERROR(sys::fs::status(FileDescriptor, Status));
1276089303d8SAndrew Ng   auto FileAccessTime = Status.getLastAccessedTime();
1277089303d8SAndrew Ng 
1278089303d8SAndrew Ng   ASSERT_NE(Epoch, FileAccessTime);
1279089303d8SAndrew Ng   ::close(FileDescriptor);
1280089303d8SAndrew Ng 
1281089303d8SAndrew Ng   // Ideally this test would include a case when ATime is not forced to update,
1282089303d8SAndrew Ng   // however the expected behaviour will differ depending on the configuration
1283089303d8SAndrew Ng   // of the Windows file system.
1284089303d8SAndrew Ng #endif
1285d9153272STaewook Oh }
12862f096097SPavel Labath 
12871f67a3cbSZachary Turner static void createFileWithData(const Twine &Path, bool ShouldExistBefore,
12881f67a3cbSZachary Turner                                fs::CreationDisposition Disp, StringRef Data) {
12891f67a3cbSZachary Turner   int FD;
12901f67a3cbSZachary Turner   ASSERT_EQ(ShouldExistBefore, fs::exists(Path));
12911f67a3cbSZachary Turner   ASSERT_NO_ERROR(fs::openFileForWrite(Path, FD, Disp));
12921f67a3cbSZachary Turner   FileDescriptorCloser Closer(FD);
12931f67a3cbSZachary Turner   ASSERT_TRUE(fs::exists(Path));
12941f67a3cbSZachary Turner 
12951f67a3cbSZachary Turner   ASSERT_EQ(Data.size(), (size_t)write(FD, Data.data(), Data.size()));
12961f67a3cbSZachary Turner }
12971f67a3cbSZachary Turner 
12981f67a3cbSZachary Turner static void verifyFileContents(const Twine &Path, StringRef Contents) {
12991f67a3cbSZachary Turner   auto Buffer = MemoryBuffer::getFile(Path);
13001f67a3cbSZachary Turner   ASSERT_TRUE((bool)Buffer);
13011f67a3cbSZachary Turner   StringRef Data = Buffer.get()->getBuffer();
13021f67a3cbSZachary Turner   ASSERT_EQ(Data, Contents);
13031f67a3cbSZachary Turner }
13041f67a3cbSZachary Turner 
13051f67a3cbSZachary Turner TEST_F(FileSystemTest, CreateNew) {
13061f67a3cbSZachary Turner   int FD;
13071f67a3cbSZachary Turner   Optional<FileDescriptorCloser> Closer;
13081f67a3cbSZachary Turner 
13091f67a3cbSZachary Turner   // Succeeds if the file does not exist.
13101f67a3cbSZachary Turner   ASSERT_FALSE(fs::exists(NonExistantFile));
13111f67a3cbSZachary Turner   ASSERT_NO_ERROR(fs::openFileForWrite(NonExistantFile, FD, fs::CD_CreateNew));
13121f67a3cbSZachary Turner   ASSERT_TRUE(fs::exists(NonExistantFile));
13131f67a3cbSZachary Turner 
13141f67a3cbSZachary Turner   FileRemover Cleanup(NonExistantFile);
13151f67a3cbSZachary Turner   Closer.emplace(FD);
13161f67a3cbSZachary Turner 
13171f67a3cbSZachary Turner   // And creates a file of size 0.
13181f67a3cbSZachary Turner   sys::fs::file_status Status;
13191f67a3cbSZachary Turner   ASSERT_NO_ERROR(sys::fs::status(FD, Status));
13201f67a3cbSZachary Turner   EXPECT_EQ(0ULL, Status.getSize());
13211f67a3cbSZachary Turner 
13221f67a3cbSZachary Turner   // Close this first, before trying to re-open the file.
13231f67a3cbSZachary Turner   Closer.reset();
13241f67a3cbSZachary Turner 
13251f67a3cbSZachary Turner   // But fails if the file does exist.
13261f67a3cbSZachary Turner   ASSERT_ERROR(fs::openFileForWrite(NonExistantFile, FD, fs::CD_CreateNew));
13271f67a3cbSZachary Turner }
13281f67a3cbSZachary Turner 
13291f67a3cbSZachary Turner TEST_F(FileSystemTest, CreateAlways) {
13301f67a3cbSZachary Turner   int FD;
13311f67a3cbSZachary Turner   Optional<FileDescriptorCloser> Closer;
13321f67a3cbSZachary Turner 
13331f67a3cbSZachary Turner   // Succeeds if the file does not exist.
13341f67a3cbSZachary Turner   ASSERT_FALSE(fs::exists(NonExistantFile));
13351f67a3cbSZachary Turner   ASSERT_NO_ERROR(
13361f67a3cbSZachary Turner       fs::openFileForWrite(NonExistantFile, FD, fs::CD_CreateAlways));
13371f67a3cbSZachary Turner 
13381f67a3cbSZachary Turner   Closer.emplace(FD);
13391f67a3cbSZachary Turner 
13401f67a3cbSZachary Turner   ASSERT_TRUE(fs::exists(NonExistantFile));
13411f67a3cbSZachary Turner 
13421f67a3cbSZachary Turner   FileRemover Cleanup(NonExistantFile);
13431f67a3cbSZachary Turner 
13441f67a3cbSZachary Turner   // And creates a file of size 0.
13451f67a3cbSZachary Turner   uint64_t FileSize;
13461f67a3cbSZachary Turner   ASSERT_NO_ERROR(sys::fs::file_size(NonExistantFile, FileSize));
13471f67a3cbSZachary Turner   ASSERT_EQ(0ULL, FileSize);
13481f67a3cbSZachary Turner 
13491f67a3cbSZachary Turner   // If we write some data to it re-create it with CreateAlways, it succeeds and
13501f67a3cbSZachary Turner   // truncates to 0 bytes.
13511f67a3cbSZachary Turner   ASSERT_EQ(4, write(FD, "Test", 4));
13521f67a3cbSZachary Turner 
13531f67a3cbSZachary Turner   Closer.reset();
13541f67a3cbSZachary Turner 
13551f67a3cbSZachary Turner   ASSERT_NO_ERROR(sys::fs::file_size(NonExistantFile, FileSize));
13561f67a3cbSZachary Turner   ASSERT_EQ(4ULL, FileSize);
13571f67a3cbSZachary Turner 
13581f67a3cbSZachary Turner   ASSERT_NO_ERROR(
13591f67a3cbSZachary Turner       fs::openFileForWrite(NonExistantFile, FD, fs::CD_CreateAlways));
13601f67a3cbSZachary Turner   Closer.emplace(FD);
13611f67a3cbSZachary Turner   ASSERT_NO_ERROR(sys::fs::file_size(NonExistantFile, FileSize));
13621f67a3cbSZachary Turner   ASSERT_EQ(0ULL, FileSize);
13631f67a3cbSZachary Turner }
13641f67a3cbSZachary Turner 
13651f67a3cbSZachary Turner TEST_F(FileSystemTest, OpenExisting) {
13661f67a3cbSZachary Turner   int FD;
13671f67a3cbSZachary Turner 
13681f67a3cbSZachary Turner   // Fails if the file does not exist.
13691f67a3cbSZachary Turner   ASSERT_FALSE(fs::exists(NonExistantFile));
13701f67a3cbSZachary Turner   ASSERT_ERROR(fs::openFileForWrite(NonExistantFile, FD, fs::CD_OpenExisting));
13711f67a3cbSZachary Turner   ASSERT_FALSE(fs::exists(NonExistantFile));
13721f67a3cbSZachary Turner 
13731f67a3cbSZachary Turner   // Make a dummy file now so that we can try again when the file does exist.
13741f67a3cbSZachary Turner   createFileWithData(NonExistantFile, false, fs::CD_CreateNew, "Fizz");
13751f67a3cbSZachary Turner   FileRemover Cleanup(NonExistantFile);
13761f67a3cbSZachary Turner   uint64_t FileSize;
13771f67a3cbSZachary Turner   ASSERT_NO_ERROR(sys::fs::file_size(NonExistantFile, FileSize));
13781f67a3cbSZachary Turner   ASSERT_EQ(4ULL, FileSize);
13791f67a3cbSZachary Turner 
13801f67a3cbSZachary Turner   // If we re-create it with different data, it overwrites rather than
13811f67a3cbSZachary Turner   // appending.
13821f67a3cbSZachary Turner   createFileWithData(NonExistantFile, true, fs::CD_OpenExisting, "Buzz");
13831f67a3cbSZachary Turner   verifyFileContents(NonExistantFile, "Buzz");
13841f67a3cbSZachary Turner }
13851f67a3cbSZachary Turner 
13861f67a3cbSZachary Turner TEST_F(FileSystemTest, OpenAlways) {
13871f67a3cbSZachary Turner   // Succeeds if the file does not exist.
13881f67a3cbSZachary Turner   createFileWithData(NonExistantFile, false, fs::CD_OpenAlways, "Fizz");
13891f67a3cbSZachary Turner   FileRemover Cleanup(NonExistantFile);
13901f67a3cbSZachary Turner   uint64_t FileSize;
13911f67a3cbSZachary Turner   ASSERT_NO_ERROR(sys::fs::file_size(NonExistantFile, FileSize));
13921f67a3cbSZachary Turner   ASSERT_EQ(4ULL, FileSize);
13931f67a3cbSZachary Turner 
13941f67a3cbSZachary Turner   // Now re-open it and write again, verifying the contents get over-written.
13951f67a3cbSZachary Turner   createFileWithData(NonExistantFile, true, fs::CD_OpenAlways, "Bu");
13961f67a3cbSZachary Turner   verifyFileContents(NonExistantFile, "Buzz");
13971f67a3cbSZachary Turner }
13981f67a3cbSZachary Turner 
13991f67a3cbSZachary Turner TEST_F(FileSystemTest, AppendSetsCorrectFileOffset) {
14001f67a3cbSZachary Turner   fs::CreationDisposition Disps[] = {fs::CD_CreateAlways, fs::CD_OpenAlways,
14011f67a3cbSZachary Turner                                      fs::CD_OpenExisting};
14021f67a3cbSZachary Turner 
14031f67a3cbSZachary Turner   // Write some data and re-open it with every possible disposition (this is a
14041f67a3cbSZachary Turner   // hack that shouldn't work, but is left for compatibility.  F_Append
14051f67a3cbSZachary Turner   // overrides
14061f67a3cbSZachary Turner   // the specified disposition.
14071f67a3cbSZachary Turner   for (fs::CreationDisposition Disp : Disps) {
14081f67a3cbSZachary Turner     int FD;
14091f67a3cbSZachary Turner     Optional<FileDescriptorCloser> Closer;
14101f67a3cbSZachary Turner 
14111f67a3cbSZachary Turner     createFileWithData(NonExistantFile, false, fs::CD_CreateNew, "Fizz");
14121f67a3cbSZachary Turner 
14131f67a3cbSZachary Turner     FileRemover Cleanup(NonExistantFile);
14141f67a3cbSZachary Turner 
14151f67a3cbSZachary Turner     uint64_t FileSize;
14161f67a3cbSZachary Turner     ASSERT_NO_ERROR(sys::fs::file_size(NonExistantFile, FileSize));
14171f67a3cbSZachary Turner     ASSERT_EQ(4ULL, FileSize);
14181f67a3cbSZachary Turner     ASSERT_NO_ERROR(
14191f67a3cbSZachary Turner         fs::openFileForWrite(NonExistantFile, FD, Disp, fs::OF_Append));
14201f67a3cbSZachary Turner     Closer.emplace(FD);
14211f67a3cbSZachary Turner     ASSERT_NO_ERROR(sys::fs::file_size(NonExistantFile, FileSize));
14221f67a3cbSZachary Turner     ASSERT_EQ(4ULL, FileSize);
14231f67a3cbSZachary Turner 
14241f67a3cbSZachary Turner     ASSERT_EQ(4, write(FD, "Buzz", 4));
14251f67a3cbSZachary Turner     Closer.reset();
14261f67a3cbSZachary Turner 
14271f67a3cbSZachary Turner     verifyFileContents(NonExistantFile, "FizzBuzz");
14281f67a3cbSZachary Turner   }
14291f67a3cbSZachary Turner }
14301f67a3cbSZachary Turner 
14311f67a3cbSZachary Turner static void verifyRead(int FD, StringRef Data, bool ShouldSucceed) {
14321f67a3cbSZachary Turner   std::vector<char> Buffer;
14331f67a3cbSZachary Turner   Buffer.resize(Data.size());
14341f67a3cbSZachary Turner   int Result = ::read(FD, Buffer.data(), Buffer.size());
14351f67a3cbSZachary Turner   if (ShouldSucceed) {
14361f67a3cbSZachary Turner     ASSERT_EQ((size_t)Result, Data.size());
14371f67a3cbSZachary Turner     ASSERT_EQ(Data, StringRef(Buffer.data(), Buffer.size()));
14381f67a3cbSZachary Turner   } else {
14391f67a3cbSZachary Turner     ASSERT_EQ(-1, Result);
14401f67a3cbSZachary Turner     ASSERT_EQ(EBADF, errno);
14411f67a3cbSZachary Turner   }
14421f67a3cbSZachary Turner }
14431f67a3cbSZachary Turner 
14441f67a3cbSZachary Turner static void verifyWrite(int FD, StringRef Data, bool ShouldSucceed) {
14451f67a3cbSZachary Turner   int Result = ::write(FD, Data.data(), Data.size());
14461f67a3cbSZachary Turner   if (ShouldSucceed)
14471f67a3cbSZachary Turner     ASSERT_EQ((size_t)Result, Data.size());
14481f67a3cbSZachary Turner   else {
14491f67a3cbSZachary Turner     ASSERT_EQ(-1, Result);
14501f67a3cbSZachary Turner     ASSERT_EQ(EBADF, errno);
14511f67a3cbSZachary Turner   }
14521f67a3cbSZachary Turner }
14531f67a3cbSZachary Turner 
14541f67a3cbSZachary Turner TEST_F(FileSystemTest, ReadOnlyFileCantWrite) {
14551f67a3cbSZachary Turner   createFileWithData(NonExistantFile, false, fs::CD_CreateNew, "Fizz");
14561f67a3cbSZachary Turner   FileRemover Cleanup(NonExistantFile);
14571f67a3cbSZachary Turner 
14581f67a3cbSZachary Turner   int FD;
14591f67a3cbSZachary Turner   ASSERT_NO_ERROR(fs::openFileForRead(NonExistantFile, FD));
14601f67a3cbSZachary Turner   FileDescriptorCloser Closer(FD);
14611f67a3cbSZachary Turner 
14621f67a3cbSZachary Turner   verifyWrite(FD, "Buzz", false);
14631f67a3cbSZachary Turner   verifyRead(FD, "Fizz", true);
14641f67a3cbSZachary Turner }
14651f67a3cbSZachary Turner 
14661f67a3cbSZachary Turner TEST_F(FileSystemTest, WriteOnlyFileCantRead) {
14671f67a3cbSZachary Turner   createFileWithData(NonExistantFile, false, fs::CD_CreateNew, "Fizz");
14681f67a3cbSZachary Turner   FileRemover Cleanup(NonExistantFile);
14691f67a3cbSZachary Turner 
14701f67a3cbSZachary Turner   int FD;
14711f67a3cbSZachary Turner   ASSERT_NO_ERROR(
14721f67a3cbSZachary Turner       fs::openFileForWrite(NonExistantFile, FD, fs::CD_OpenExisting));
14731f67a3cbSZachary Turner   FileDescriptorCloser Closer(FD);
14741f67a3cbSZachary Turner   verifyRead(FD, "Fizz", false);
14751f67a3cbSZachary Turner   verifyWrite(FD, "Buzz", true);
14761f67a3cbSZachary Turner }
14771f67a3cbSZachary Turner 
14781f67a3cbSZachary Turner TEST_F(FileSystemTest, ReadWriteFileCanReadOrWrite) {
14791f67a3cbSZachary Turner   createFileWithData(NonExistantFile, false, fs::CD_CreateNew, "Fizz");
14801f67a3cbSZachary Turner   FileRemover Cleanup(NonExistantFile);
14811f67a3cbSZachary Turner 
14821f67a3cbSZachary Turner   int FD;
14831f67a3cbSZachary Turner   ASSERT_NO_ERROR(fs::openFileForReadWrite(NonExistantFile, FD,
14841f67a3cbSZachary Turner                                            fs::CD_OpenExisting, fs::OF_None));
14851f67a3cbSZachary Turner   FileDescriptorCloser Closer(FD);
14861f67a3cbSZachary Turner   verifyRead(FD, "Fizz", true);
14871f67a3cbSZachary Turner   verifyWrite(FD, "Buzz", true);
14881f67a3cbSZachary Turner }
14891f67a3cbSZachary Turner 
14902f096097SPavel Labath TEST_F(FileSystemTest, set_current_path) {
14912f096097SPavel Labath   SmallString<128> path;
14922f096097SPavel Labath 
14932f096097SPavel Labath   ASSERT_NO_ERROR(fs::current_path(path));
14942f096097SPavel Labath   ASSERT_NE(TestDirectory, path);
14952f096097SPavel Labath 
14962f096097SPavel Labath   struct RestorePath {
14972f096097SPavel Labath     SmallString<128> path;
14982f096097SPavel Labath     RestorePath(const SmallString<128> &path) : path(path) {}
14992f096097SPavel Labath     ~RestorePath() { fs::set_current_path(path); }
15002f096097SPavel Labath   } restore_path(path);
15012f096097SPavel Labath 
15022f096097SPavel Labath   ASSERT_NO_ERROR(fs::set_current_path(TestDirectory));
15032f096097SPavel Labath 
15042f096097SPavel Labath   ASSERT_NO_ERROR(fs::current_path(path));
150550440097SPavel Labath 
150650440097SPavel Labath   fs::UniqueID D1, D2;
150750440097SPavel Labath   ASSERT_NO_ERROR(fs::getUniqueID(TestDirectory, D1));
150850440097SPavel Labath   ASSERT_NO_ERROR(fs::getUniqueID(path, D2));
150950440097SPavel Labath   ASSERT_EQ(D1, D2) << "D1: " << TestDirectory << "\nD2: " << path;
15102f096097SPavel Labath }
15112f096097SPavel Labath 
1512566fdf4aSJames Henderson TEST_F(FileSystemTest, permissions) {
1513566fdf4aSJames Henderson   int FD;
1514566fdf4aSJames Henderson   SmallString<64> TempPath;
1515566fdf4aSJames Henderson   ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "temp", FD, TempPath));
1516566fdf4aSJames Henderson   FileRemover Cleanup(TempPath);
1517566fdf4aSJames Henderson 
1518566fdf4aSJames Henderson   // Make sure it exists.
1519566fdf4aSJames Henderson   ASSERT_TRUE(fs::exists(Twine(TempPath)));
1520566fdf4aSJames Henderson 
1521566fdf4aSJames Henderson   auto CheckPermissions = [&](fs::perms Expected) {
1522566fdf4aSJames Henderson     ErrorOr<fs::perms> Actual = fs::getPermissions(TempPath);
1523566fdf4aSJames Henderson     return Actual && *Actual == Expected;
1524566fdf4aSJames Henderson   };
1525566fdf4aSJames Henderson 
1526566fdf4aSJames Henderson   std::error_code NoError;
1527566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_all), NoError);
1528566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_all));
1529566fdf4aSJames Henderson 
1530566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_read | fs::all_exe), NoError);
1531566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_read | fs::all_exe));
1532566fdf4aSJames Henderson 
1533712e8d29SNico Weber #if defined(_WIN32)
1534566fdf4aSJames Henderson   fs::perms ReadOnly = fs::all_read | fs::all_exe;
1535566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::no_perms), NoError);
1536566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1537566fdf4aSJames Henderson 
1538566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::owner_read), NoError);
1539566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1540566fdf4aSJames Henderson 
1541566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::owner_write), NoError);
1542566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_all));
1543566fdf4aSJames Henderson 
1544566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::owner_exe), NoError);
1545566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1546566fdf4aSJames Henderson 
1547566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::owner_all), NoError);
1548566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_all));
1549566fdf4aSJames Henderson 
1550566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::group_read), NoError);
1551566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1552566fdf4aSJames Henderson 
1553566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::group_write), NoError);
1554566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_all));
1555566fdf4aSJames Henderson 
1556566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::group_exe), NoError);
1557566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1558566fdf4aSJames Henderson 
1559566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::group_all), NoError);
1560566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_all));
1561566fdf4aSJames Henderson 
1562566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::others_read), NoError);
1563566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1564566fdf4aSJames Henderson 
1565566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::others_write), NoError);
1566566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_all));
1567566fdf4aSJames Henderson 
1568566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::others_exe), NoError);
1569566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1570566fdf4aSJames Henderson 
1571566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::others_all), NoError);
1572566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_all));
1573566fdf4aSJames Henderson 
1574566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_read), NoError);
1575566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1576566fdf4aSJames Henderson 
1577566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_write), NoError);
1578566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_all));
1579566fdf4aSJames Henderson 
1580566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_exe), NoError);
1581566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1582566fdf4aSJames Henderson 
1583566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::set_uid_on_exe), NoError);
1584566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1585566fdf4aSJames Henderson 
1586566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::set_gid_on_exe), NoError);
1587566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1588566fdf4aSJames Henderson 
1589566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::sticky_bit), NoError);
1590566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1591566fdf4aSJames Henderson 
1592566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::set_uid_on_exe |
1593566fdf4aSJames Henderson                                              fs::set_gid_on_exe |
1594566fdf4aSJames Henderson                                              fs::sticky_bit),
1595566fdf4aSJames Henderson             NoError);
1596566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1597566fdf4aSJames Henderson 
1598566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, ReadOnly | fs::set_uid_on_exe |
1599566fdf4aSJames Henderson                                              fs::set_gid_on_exe |
1600566fdf4aSJames Henderson                                              fs::sticky_bit),
1601566fdf4aSJames Henderson             NoError);
1602566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(ReadOnly));
1603566fdf4aSJames Henderson 
1604566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_perms), NoError);
1605566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_all));
1606566fdf4aSJames Henderson #else
1607566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::no_perms), NoError);
1608566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::no_perms));
1609566fdf4aSJames Henderson 
1610566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::owner_read), NoError);
1611566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::owner_read));
1612566fdf4aSJames Henderson 
1613566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::owner_write), NoError);
1614566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::owner_write));
1615566fdf4aSJames Henderson 
1616566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::owner_exe), NoError);
1617566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::owner_exe));
1618566fdf4aSJames Henderson 
1619566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::owner_all), NoError);
1620566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::owner_all));
1621566fdf4aSJames Henderson 
1622566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::group_read), NoError);
1623566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::group_read));
1624566fdf4aSJames Henderson 
1625566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::group_write), NoError);
1626566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::group_write));
1627566fdf4aSJames Henderson 
1628566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::group_exe), NoError);
1629566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::group_exe));
1630566fdf4aSJames Henderson 
1631566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::group_all), NoError);
1632566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::group_all));
1633566fdf4aSJames Henderson 
1634566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::others_read), NoError);
1635566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::others_read));
1636566fdf4aSJames Henderson 
1637566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::others_write), NoError);
1638566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::others_write));
1639566fdf4aSJames Henderson 
1640566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::others_exe), NoError);
1641566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::others_exe));
1642566fdf4aSJames Henderson 
1643566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::others_all), NoError);
1644566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::others_all));
1645566fdf4aSJames Henderson 
1646566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_read), NoError);
1647566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_read));
1648566fdf4aSJames Henderson 
1649566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_write), NoError);
1650566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_write));
1651566fdf4aSJames Henderson 
1652566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_exe), NoError);
1653566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_exe));
1654566fdf4aSJames Henderson 
1655566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::set_uid_on_exe), NoError);
1656566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::set_uid_on_exe));
1657566fdf4aSJames Henderson 
1658566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::set_gid_on_exe), NoError);
1659566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::set_gid_on_exe));
1660566fdf4aSJames Henderson 
166149e033f4SDimitry Andric   // Modern BSDs require root to set the sticky bit on files.
166249e033f4SDimitry Andric #if !defined(__FreeBSD__) && !defined(__NetBSD__) && !defined(__OpenBSD__)
1663566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::sticky_bit), NoError);
1664566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::sticky_bit));
1665566fdf4aSJames Henderson 
1666566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::set_uid_on_exe |
1667566fdf4aSJames Henderson                                              fs::set_gid_on_exe |
1668566fdf4aSJames Henderson                                              fs::sticky_bit),
1669566fdf4aSJames Henderson             NoError);
1670566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::set_uid_on_exe | fs::set_gid_on_exe |
1671566fdf4aSJames Henderson                                fs::sticky_bit));
1672566fdf4aSJames Henderson 
1673566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_read | fs::set_uid_on_exe |
1674566fdf4aSJames Henderson                                              fs::set_gid_on_exe |
1675566fdf4aSJames Henderson                                              fs::sticky_bit),
1676566fdf4aSJames Henderson             NoError);
1677566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_read | fs::set_uid_on_exe |
1678566fdf4aSJames Henderson                                fs::set_gid_on_exe | fs::sticky_bit));
1679566fdf4aSJames Henderson 
1680566fdf4aSJames Henderson   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_perms), NoError);
1681566fdf4aSJames Henderson   EXPECT_TRUE(CheckPermissions(fs::all_perms));
168249e033f4SDimitry Andric #endif // !FreeBSD && !NetBSD && !OpenBSD
168349e033f4SDimitry Andric 
168449e033f4SDimitry Andric   EXPECT_EQ(fs::setPermissions(TempPath, fs::all_perms & ~fs::sticky_bit),
168549e033f4SDimitry Andric                                NoError);
168649e033f4SDimitry Andric   EXPECT_TRUE(CheckPermissions(fs::all_perms & ~fs::sticky_bit));
1687566fdf4aSJames Henderson #endif
1688566fdf4aSJames Henderson }
1689566fdf4aSJames Henderson 
16903ef91c55SMichael J. Spencer } // anonymous namespace
1691