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