blob: 660ea3663acfda093c3591defb0f97d393a8a550 [file] [log] [blame]
Michael J. Spencer3ef91c52010-11-29 22:29:04 +00001//===- llvm/unittest/Support/Path.cpp - Path tests ------------------------===//
Michael J. Spencer98847782010-11-24 19:20:05 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
Rafael Espindola3bc8e712013-06-11 22:21:28 +000010#include "llvm/Support/Path.h"
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +000011#include "llvm/Support/ErrorHandling.h"
Chandler Carruth130cec22012-12-04 10:23:08 +000012#include "llvm/Support/FileSystem.h"
Rafael Espindola84ab9b32013-07-19 14:41:25 +000013#include "llvm/Support/MemoryBuffer.h"
Michael J. Spencer1f063602011-01-06 05:57:54 +000014#include "llvm/Support/raw_ostream.h"
Michael J. Spencer98847782010-11-24 19:20:05 +000015#include "gtest/gtest.h"
16
Rafael Espindolaa813d602014-06-11 03:58:34 +000017#ifdef LLVM_ON_WIN32
18#include <winerror.h>
19#endif
20
Michael J. Spencerebad2f92010-11-29 22:28:51 +000021using namespace llvm;
Michael J. Spencer45710402010-12-03 01:21:28 +000022using namespace llvm::sys;
Michael J. Spencerebad2f92010-11-29 22:28:51 +000023
Michael J. Spencer3b264ba2011-01-04 17:00:18 +000024#define ASSERT_NO_ERROR(x) \
Michael J. Spencer1f063602011-01-06 05:57:54 +000025 if (error_code ASSERT_NO_ERROR_ec = x) { \
26 SmallString<128> MessageStorage; \
27 raw_svector_ostream Message(MessageStorage); \
28 Message << #x ": did not return errc::success.\n" \
29 << "error number: " << ASSERT_NO_ERROR_ec.value() << "\n" \
30 << "error message: " << ASSERT_NO_ERROR_ec.message() << "\n"; \
31 GTEST_FATAL_FAILURE_(MessageStorage.c_str()); \
Michael J. Spencer3b264ba2011-01-04 17:00:18 +000032 } else {}
33
Michael J. Spencer98847782010-11-24 19:20:05 +000034namespace {
35
Zhanyong Wan606bb1a2011-02-11 21:24:40 +000036TEST(is_separator, Works) {
37 EXPECT_TRUE(path::is_separator('/'));
38 EXPECT_FALSE(path::is_separator('\0'));
39 EXPECT_FALSE(path::is_separator('-'));
40 EXPECT_FALSE(path::is_separator(' '));
41
42#ifdef LLVM_ON_WIN32
43 EXPECT_TRUE(path::is_separator('\\'));
44#else
45 EXPECT_FALSE(path::is_separator('\\'));
46#endif
47}
48
Michael J. Spencer3ef91c52010-11-29 22:29:04 +000049TEST(Support, Path) {
Michael J. Spencerebad2f92010-11-29 22:28:51 +000050 SmallVector<StringRef, 40> paths;
51 paths.push_back("");
52 paths.push_back(".");
53 paths.push_back("..");
54 paths.push_back("foo");
55 paths.push_back("/");
56 paths.push_back("/foo");
57 paths.push_back("foo/");
58 paths.push_back("/foo/");
59 paths.push_back("foo/bar");
60 paths.push_back("/foo/bar");
61 paths.push_back("//net");
62 paths.push_back("//net/foo");
63 paths.push_back("///foo///");
64 paths.push_back("///foo///bar");
65 paths.push_back("/.");
66 paths.push_back("./");
67 paths.push_back("/..");
68 paths.push_back("../");
69 paths.push_back("foo/.");
70 paths.push_back("foo/..");
71 paths.push_back("foo/./");
72 paths.push_back("foo/./bar");
73 paths.push_back("foo/..");
74 paths.push_back("foo/../");
75 paths.push_back("foo/../bar");
76 paths.push_back("c:");
77 paths.push_back("c:/");
78 paths.push_back("c:foo");
79 paths.push_back("c:/foo");
80 paths.push_back("c:foo/");
81 paths.push_back("c:/foo/");
82 paths.push_back("c:/foo/bar");
83 paths.push_back("prn:");
84 paths.push_back("c:\\");
85 paths.push_back("c:foo");
86 paths.push_back("c:\\foo");
87 paths.push_back("c:foo\\");
88 paths.push_back("c:\\foo\\");
89 paths.push_back("c:\\foo/");
90 paths.push_back("c:/foo\\bar");
91
92 for (SmallVector<StringRef, 40>::const_iterator i = paths.begin(),
93 e = paths.end();
94 i != e;
95 ++i) {
Michael J. Spencerebad2f92010-11-29 22:28:51 +000096 for (sys::path::const_iterator ci = sys::path::begin(*i),
97 ce = sys::path::end(*i);
98 ci != ce;
99 ++ci) {
Michael J. Spencerb5ca6442010-12-03 02:22:34 +0000100 ASSERT_FALSE(ci->empty());
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000101 }
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000102
Michael J. Spencer75942f92010-12-01 22:28:42 +0000103#if 0 // Valgrind is whining about this.
Michael J. Spencer545cbdf2010-11-30 23:28:07 +0000104 outs() << " Reverse Iteration: [";
105 for (sys::path::reverse_iterator ci = sys::path::rbegin(*i),
106 ce = sys::path::rend(*i);
107 ci != ce;
108 ++ci) {
109 outs() << *ci << ',';
110 }
111 outs() << "]\n";
Michael J. Spencer75942f92010-12-01 22:28:42 +0000112#endif
Michael J. Spencer545cbdf2010-11-30 23:28:07 +0000113
Michael J. Spencerf616b212010-12-07 17:04:04 +0000114 path::has_root_path(*i);
115 path::root_path(*i);
116 path::has_root_name(*i);
117 path::root_name(*i);
118 path::has_root_directory(*i);
119 path::root_directory(*i);
120 path::has_parent_path(*i);
121 path::parent_path(*i);
122 path::has_filename(*i);
123 path::filename(*i);
124 path::has_stem(*i);
125 path::stem(*i);
126 path::has_extension(*i);
127 path::extension(*i);
128 path::is_absolute(*i);
129 path::is_relative(*i);
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000130
Michael J. Spencer3b264ba2011-01-04 17:00:18 +0000131 SmallString<128> temp_store;
Michael J. Spencerb5ca6442010-12-03 02:22:34 +0000132 temp_store = *i;
Michael J. Spencer3b264ba2011-01-04 17:00:18 +0000133 ASSERT_NO_ERROR(fs::make_absolute(temp_store));
Michael J. Spencerb5ca6442010-12-03 02:22:34 +0000134 temp_store = *i;
Michael J. Spencer1e090f02010-12-07 03:57:37 +0000135 path::remove_filename(temp_store);
Michael J. Spencer4d0c6fd2010-12-01 06:03:33 +0000136
Michael J. Spencerb5ca6442010-12-03 02:22:34 +0000137 temp_store = *i;
Michael J. Spencer1e090f02010-12-07 03:57:37 +0000138 path::replace_extension(temp_store, "ext");
Michael J. Spencer4d0c6fd2010-12-01 06:03:33 +0000139 StringRef filename(temp_store.begin(), temp_store.size()), stem, ext;
Michael J. Spencerf616b212010-12-07 17:04:04 +0000140 stem = path::stem(filename);
141 ext = path::extension(filename);
Michael J. Spencer4d0c6fd2010-12-01 06:03:33 +0000142 EXPECT_EQ(*(--sys::path::end(filename)), (stem + ext).str());
143
Michael J. Spencer1e090f02010-12-07 03:57:37 +0000144 path::native(*i, temp_store);
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000145 }
Michael J. Spencer346a1332011-01-05 16:39:05 +0000146}
Michael J. Spencer45710402010-12-03 01:21:28 +0000147
Tareq A. Siraj73537ea2013-08-12 17:10:49 +0000148TEST(Support, RelativePathIterator) {
149 SmallString<64> Path(StringRef("c/d/e/foo.txt"));
150 typedef SmallVector<StringRef, 4> PathComponents;
151 PathComponents ExpectedPathComponents;
152 PathComponents ActualPathComponents;
153
154 StringRef(Path).split(ExpectedPathComponents, "/");
155
156 for (path::const_iterator I = path::begin(Path), E = path::end(Path); I != E;
157 ++I) {
158 ActualPathComponents.push_back(*I);
159 }
160
161 ASSERT_EQ(ExpectedPathComponents.size(), ActualPathComponents.size());
162
163 for (size_t i = 0; i <ExpectedPathComponents.size(); ++i) {
164 EXPECT_EQ(ExpectedPathComponents[i].str(), ActualPathComponents[i].str());
165 }
166}
167
168TEST(Support, AbsolutePathIterator) {
169 SmallString<64> Path(StringRef("/c/d/e/foo.txt"));
170 typedef SmallVector<StringRef, 4> PathComponents;
171 PathComponents ExpectedPathComponents;
172 PathComponents ActualPathComponents;
173
174 StringRef(Path).split(ExpectedPathComponents, "/");
175
176 // The root path will also be a component when iterating
177 ExpectedPathComponents[0] = "/";
178
179 for (path::const_iterator I = path::begin(Path), E = path::end(Path); I != E;
180 ++I) {
181 ActualPathComponents.push_back(*I);
182 }
183
184 ASSERT_EQ(ExpectedPathComponents.size(), ActualPathComponents.size());
185
186 for (size_t i = 0; i <ExpectedPathComponents.size(); ++i) {
187 EXPECT_EQ(ExpectedPathComponents[i].str(), ActualPathComponents[i].str());
188 }
189}
190
191#ifdef LLVM_ON_WIN32
192TEST(Support, AbsolutePathIteratorWin32) {
193 SmallString<64> Path(StringRef("c:\\c\\e\\foo.txt"));
194 typedef SmallVector<StringRef, 4> PathComponents;
195 PathComponents ExpectedPathComponents;
196 PathComponents ActualPathComponents;
197
198 StringRef(Path).split(ExpectedPathComponents, "\\");
199
200 // The root path (which comes after the drive name) will also be a component
201 // when iterating.
202 ExpectedPathComponents.insert(ExpectedPathComponents.begin()+1, "\\");
203
204 for (path::const_iterator I = path::begin(Path), E = path::end(Path); I != E;
205 ++I) {
206 ActualPathComponents.push_back(*I);
207 }
208
209 ASSERT_EQ(ExpectedPathComponents.size(), ActualPathComponents.size());
210
211 for (size_t i = 0; i <ExpectedPathComponents.size(); ++i) {
212 EXPECT_EQ(ExpectedPathComponents[i].str(), ActualPathComponents[i].str());
213 }
214}
215#endif // LLVM_ON_WIN32
216
Ben Langmuir8d116392014-03-05 19:56:30 +0000217TEST(Support, AbsolutePathIteratorEnd) {
218 // Trailing slashes are converted to '.' unless they are part of the root path.
219 SmallVector<StringRef, 4> Paths;
220 Paths.push_back("/foo/");
221 Paths.push_back("/foo//");
222 Paths.push_back("//net//");
223#ifdef LLVM_ON_WIN32
224 Paths.push_back("c:\\\\");
225#endif
226
227 for (StringRef Path : Paths) {
228 StringRef LastComponent = *--path::end(Path);
229 EXPECT_EQ(".", LastComponent);
230 }
231
232 SmallVector<StringRef, 3> RootPaths;
233 RootPaths.push_back("/");
234 RootPaths.push_back("//net/");
235#ifdef LLVM_ON_WIN32
236 RootPaths.push_back("c:\\");
237#endif
238
239 for (StringRef Path : RootPaths) {
240 StringRef LastComponent = *--path::end(Path);
241 EXPECT_EQ(1u, LastComponent.size());
242 EXPECT_TRUE(path::is_separator(LastComponent[0]));
243 }
244}
245
Peter Collingbournef7d41012014-01-31 23:46:06 +0000246TEST(Support, HomeDirectory) {
247#ifdef LLVM_ON_UNIX
248 // This test only makes sense on Unix if $HOME is set.
249 if (::getenv("HOME")) {
250#endif
251 SmallString<128> HomeDir;
252 EXPECT_TRUE(path::home_directory(HomeDir));
253 EXPECT_FALSE(HomeDir.empty());
254#ifdef LLVM_ON_UNIX
255 }
256#endif
257}
258
Michael J. Spencer346a1332011-01-05 16:39:05 +0000259class FileSystemTest : public testing::Test {
260protected:
261 /// Unique temporary directory in which all created filesystem entities must
262 /// be placed. It is recursively removed at the end of each test.
263 SmallString<128> TestDirectory;
264
265 virtual void SetUp() {
Michael J. Spencer346a1332011-01-05 16:39:05 +0000266 ASSERT_NO_ERROR(
Rafael Espindola7ffacc42013-06-27 03:45:31 +0000267 fs::createUniqueDirectory("file-system-test", TestDirectory));
Michael J. Spencer346a1332011-01-05 16:39:05 +0000268 // We don't care about this specific file.
Michael J. Spencer42368cc2011-01-05 16:39:46 +0000269 errs() << "Test Directory: " << TestDirectory << '\n';
270 errs().flush();
Michael J. Spencer346a1332011-01-05 16:39:05 +0000271 }
272
273 virtual void TearDown() {
Rafael Espindola78dcc032014-01-10 20:36:42 +0000274 ASSERT_NO_ERROR(fs::remove(TestDirectory.str()));
Michael J. Spencer346a1332011-01-05 16:39:05 +0000275 }
276};
277
Aaron Ballman9fbe5302013-06-19 21:03:50 +0000278TEST_F(FileSystemTest, Unique) {
279 // Create a temp file.
280 int FileDescriptor;
281 SmallString<64> TempPath;
282 ASSERT_NO_ERROR(
Rafael Espindola155cf0f2013-07-05 20:14:52 +0000283 fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
Aaron Ballman9fbe5302013-06-19 21:03:50 +0000284
285 // The same file should return an identical unique id.
Rafael Espindola7f822a92013-07-29 21:26:49 +0000286 fs::UniqueID F1, F2;
Rafael Espindola7cf7c512013-06-20 15:06:35 +0000287 ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), F1));
288 ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath), F2));
Aaron Ballman9fbe5302013-06-19 21:03:50 +0000289 ASSERT_EQ(F1, F2);
290
291 // Different files should return different unique ids.
292 int FileDescriptor2;
293 SmallString<64> TempPath2;
294 ASSERT_NO_ERROR(
Rafael Espindola155cf0f2013-07-05 20:14:52 +0000295 fs::createTemporaryFile("prefix", "temp", FileDescriptor2, TempPath2));
296
Rafael Espindola7f822a92013-07-29 21:26:49 +0000297 fs::UniqueID D;
Rafael Espindola7cf7c512013-06-20 15:06:35 +0000298 ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath2), D));
Aaron Ballman9fbe5302013-06-19 21:03:50 +0000299 ASSERT_NE(D, F1);
300 ::close(FileDescriptor2);
301
302 ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
303
304 // Two paths representing the same file on disk should still provide the
305 // same unique id. We can test this by making a hard link.
Rafael Espindola83f858e2014-03-11 18:40:24 +0000306 ASSERT_NO_ERROR(fs::create_link(Twine(TempPath), Twine(TempPath2)));
Rafael Espindola7f822a92013-07-29 21:26:49 +0000307 fs::UniqueID D2;
Rafael Espindola7cf7c512013-06-20 15:06:35 +0000308 ASSERT_NO_ERROR(fs::getUniqueID(Twine(TempPath2), D2));
Aaron Ballman9fbe5302013-06-19 21:03:50 +0000309 ASSERT_EQ(D2, F1);
310
311 ::close(FileDescriptor);
Rafael Espindolaa5932af2013-07-30 20:25:53 +0000312
313 SmallString<128> Dir1;
314 ASSERT_NO_ERROR(
315 fs::createUniqueDirectory("dir1", Dir1));
316 ASSERT_NO_ERROR(fs::getUniqueID(Dir1.c_str(), F1));
317 ASSERT_NO_ERROR(fs::getUniqueID(Dir1.c_str(), F2));
318 ASSERT_EQ(F1, F2);
319
320 SmallString<128> Dir2;
321 ASSERT_NO_ERROR(
322 fs::createUniqueDirectory("dir2", Dir2));
323 ASSERT_NO_ERROR(fs::getUniqueID(Dir2.c_str(), F2));
324 ASSERT_NE(F1, F2);
Aaron Ballman9fbe5302013-06-19 21:03:50 +0000325}
326
Michael J. Spencer346a1332011-01-05 16:39:05 +0000327TEST_F(FileSystemTest, TempFiles) {
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000328 // Create a temp file.
Michael J. Spencer45710402010-12-03 01:21:28 +0000329 int FileDescriptor;
330 SmallString<64> TempPath;
Michael J. Spencer3b264ba2011-01-04 17:00:18 +0000331 ASSERT_NO_ERROR(
Rafael Espindola155cf0f2013-07-05 20:14:52 +0000332 fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
Michael J. Spencer45710402010-12-03 01:21:28 +0000333
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000334 // Make sure it exists.
Michael J. Spencer45710402010-12-03 01:21:28 +0000335 bool TempFileExists;
Michael J. Spencer3b264ba2011-01-04 17:00:18 +0000336 ASSERT_NO_ERROR(sys::fs::exists(Twine(TempPath), TempFileExists));
Michael J. Spencer45710402010-12-03 01:21:28 +0000337 EXPECT_TRUE(TempFileExists);
338
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000339 // Create another temp tile.
340 int FD2;
341 SmallString<64> TempPath2;
Rafael Espindola155cf0f2013-07-05 20:14:52 +0000342 ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "temp", FD2, TempPath2));
Rafael Espindolad3c89042013-07-25 15:00:17 +0000343 ASSERT_TRUE(TempPath2.endswith(".temp"));
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000344 ASSERT_NE(TempPath.str(), TempPath2.str());
345
Michael J. Spencer203d7802011-12-12 06:04:28 +0000346 fs::file_status A, B;
347 ASSERT_NO_ERROR(fs::status(Twine(TempPath), A));
348 ASSERT_NO_ERROR(fs::status(Twine(TempPath2), B));
349 EXPECT_FALSE(fs::equivalent(A, B));
350
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000351 ::close(FD2);
Rafael Espindola213c4cb2013-07-18 03:29:51 +0000352
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000353 // Remove Temp2.
Rafael Espindola5c20ac02014-02-23 13:56:14 +0000354 ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
355 ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
356 ASSERT_EQ(fs::remove(Twine(TempPath2), false),
357 errc::no_such_file_or_directory);
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000358
Rafael Espindola107b74c2013-07-31 00:10:25 +0000359 error_code EC = fs::status(TempPath2.c_str(), B);
360 EXPECT_EQ(EC, errc::no_such_file_or_directory);
361 EXPECT_EQ(B.type(), fs::file_type::file_not_found);
362
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000363 // Make sure Temp2 doesn't exist.
Michael J. Spencer3b264ba2011-01-04 17:00:18 +0000364 ASSERT_NO_ERROR(fs::exists(Twine(TempPath2), TempFileExists));
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000365 EXPECT_FALSE(TempFileExists);
366
Rafael Espindolad3c89042013-07-25 15:00:17 +0000367 SmallString<64> TempPath3;
368 ASSERT_NO_ERROR(fs::createTemporaryFile("prefix", "", TempPath3));
369 ASSERT_FALSE(TempPath3.endswith("."));
370
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000371 // Create a hard link to Temp1.
Rafael Espindola83f858e2014-03-11 18:40:24 +0000372 ASSERT_NO_ERROR(fs::create_link(Twine(TempPath), Twine(TempPath2)));
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000373 bool equal;
Michael J. Spencer3b264ba2011-01-04 17:00:18 +0000374 ASSERT_NO_ERROR(fs::equivalent(Twine(TempPath), Twine(TempPath2), equal));
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000375 EXPECT_TRUE(equal);
Michael J. Spencer203d7802011-12-12 06:04:28 +0000376 ASSERT_NO_ERROR(fs::status(Twine(TempPath), A));
377 ASSERT_NO_ERROR(fs::status(Twine(TempPath2), B));
378 EXPECT_TRUE(fs::equivalent(A, B));
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000379
380 // Remove Temp1.
Michael J. Spencer45710402010-12-03 01:21:28 +0000381 ::close(FileDescriptor);
Rafael Espindola5c20ac02014-02-23 13:56:14 +0000382 ASSERT_NO_ERROR(fs::remove(Twine(TempPath)));
Michael J. Spencer45710402010-12-03 01:21:28 +0000383
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000384 // Remove the hard link.
Rafael Espindola5c20ac02014-02-23 13:56:14 +0000385 ASSERT_NO_ERROR(fs::remove(Twine(TempPath2)));
Michael J. Spencer4fb115d2010-12-04 03:18:42 +0000386
387 // Make sure Temp1 doesn't exist.
Michael J. Spencer3b264ba2011-01-04 17:00:18 +0000388 ASSERT_NO_ERROR(fs::exists(Twine(TempPath), TempFileExists));
Benjamin Kramer23e6bdf2010-12-03 12:33:32 +0000389 EXPECT_FALSE(TempFileExists);
Aaron Ballmanfcdf9a82013-03-16 15:00:51 +0000390
391#ifdef LLVM_ON_WIN32
392 // Path name > 260 chars should get an error.
393 const char *Path270 =
394 "abcdefghijklmnopqrstuvwxyz9abcdefghijklmnopqrstuvwxyz8"
395 "abcdefghijklmnopqrstuvwxyz7abcdefghijklmnopqrstuvwxyz6"
396 "abcdefghijklmnopqrstuvwxyz5abcdefghijklmnopqrstuvwxyz4"
397 "abcdefghijklmnopqrstuvwxyz3abcdefghijklmnopqrstuvwxyz2"
398 "abcdefghijklmnopqrstuvwxyz1abcdefghijklmnopqrstuvwxyz0";
Rafael Espindola2eb1ae42013-07-05 22:32:33 +0000399 EXPECT_EQ(fs::createUniqueFile(Twine(Path270), FileDescriptor, TempPath),
Rafael Espindolaa813d602014-06-11 03:58:34 +0000400 error_code(ERROR_PATH_NOT_FOUND, system_category()));
Aaron Ballmanfcdf9a82013-03-16 15:00:51 +0000401#endif
Michael J. Spencer346a1332011-01-05 16:39:05 +0000402}
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000403
Rafael Espindola5c20ac02014-02-23 13:56:14 +0000404TEST_F(FileSystemTest, CreateDir) {
405 ASSERT_NO_ERROR(fs::create_directory(Twine(TestDirectory) + "foo"));
406 ASSERT_NO_ERROR(fs::create_directory(Twine(TestDirectory) + "foo"));
407 ASSERT_EQ(fs::create_directory(Twine(TestDirectory) + "foo", false),
408 errc::file_exists);
409 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "foo"));
410}
411
Michael J. Spencer346a1332011-01-05 16:39:05 +0000412TEST_F(FileSystemTest, DirectoryIteration) {
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000413 error_code ec;
Michael J. Spencer1f063602011-01-06 05:57:54 +0000414 for (fs::directory_iterator i(".", ec), e; i != e; i.increment(ec))
415 ASSERT_NO_ERROR(ec);
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000416
417 // Create a known hierarchy to recurse over.
Rafael Espindola5c20ac02014-02-23 13:56:14 +0000418 ASSERT_NO_ERROR(
419 fs::create_directories(Twine(TestDirectory) + "/recursive/a0/aa1"));
420 ASSERT_NO_ERROR(
421 fs::create_directories(Twine(TestDirectory) + "/recursive/a0/ab1"));
422 ASSERT_NO_ERROR(fs::create_directories(Twine(TestDirectory) +
423 "/recursive/dontlookhere/da1"));
424 ASSERT_NO_ERROR(
425 fs::create_directories(Twine(TestDirectory) + "/recursive/z0/za1"));
426 ASSERT_NO_ERROR(
427 fs::create_directories(Twine(TestDirectory) + "/recursive/pop/p1"));
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000428 typedef std::vector<std::string> v_t;
429 v_t visited;
430 for (fs::recursive_directory_iterator i(Twine(TestDirectory)
431 + "/recursive", ec), e; i != e; i.increment(ec)){
432 ASSERT_NO_ERROR(ec);
NAKAMURA Takumi0cfb3672011-12-09 23:20:03 +0000433 if (path::filename(i->path()) == "p1") {
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000434 i.pop();
NAKAMURA Takumi0cfb3672011-12-09 23:20:03 +0000435 // FIXME: recursive_directory_iterator should be more robust.
436 if (i == e) break;
437 }
Michael J. Spencer9b54a3e2011-12-09 01:14:41 +0000438 if (path::filename(i->path()) == "dontlookhere")
439 i.no_push();
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000440 visited.push_back(path::filename(i->path()));
441 }
442 v_t::const_iterator a0 = std::find(visited.begin(), visited.end(), "a0");
443 v_t::const_iterator aa1 = std::find(visited.begin(), visited.end(), "aa1");
444 v_t::const_iterator ab1 = std::find(visited.begin(), visited.end(), "ab1");
445 v_t::const_iterator dontlookhere = std::find(visited.begin(), visited.end(),
446 "dontlookhere");
447 v_t::const_iterator da1 = std::find(visited.begin(), visited.end(), "da1");
448 v_t::const_iterator z0 = std::find(visited.begin(), visited.end(), "z0");
449 v_t::const_iterator za1 = std::find(visited.begin(), visited.end(), "za1");
450 v_t::const_iterator pop = std::find(visited.begin(), visited.end(), "pop");
451 v_t::const_iterator p1 = std::find(visited.begin(), visited.end(), "p1");
452
453 // Make sure that each path was visited correctly.
454 ASSERT_NE(a0, visited.end());
455 ASSERT_NE(aa1, visited.end());
456 ASSERT_NE(ab1, visited.end());
457 ASSERT_NE(dontlookhere, visited.end());
458 ASSERT_EQ(da1, visited.end()); // Not visited.
459 ASSERT_NE(z0, visited.end());
460 ASSERT_NE(za1, visited.end());
461 ASSERT_NE(pop, visited.end());
462 ASSERT_EQ(p1, visited.end()); // Not visited.
463
464 // Make sure that parents were visited before children. No other ordering
465 // guarantees can be made across siblings.
466 ASSERT_LT(a0, aa1);
467 ASSERT_LT(a0, ab1);
468 ASSERT_LT(z0, za1);
Rafael Espindola78dcc032014-01-10 20:36:42 +0000469
470 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/a0/aa1"));
471 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/a0/ab1"));
472 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/a0"));
473 ASSERT_NO_ERROR(
474 fs::remove(Twine(TestDirectory) + "/recursive/dontlookhere/da1"));
475 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/dontlookhere"));
476 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/pop/p1"));
477 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/pop"));
478 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/z0/za1"));
479 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive/z0"));
480 ASSERT_NO_ERROR(fs::remove(Twine(TestDirectory) + "/recursive"));
Michael J. Spencer1f063602011-01-06 05:57:54 +0000481}
Michael J. Spencer1d3c4a72011-01-06 05:58:02 +0000482
Rui Ueyama5e3de7a2013-11-13 21:55:41 +0000483const char archive[] = "!<arch>\x0A";
484const char bitcode[] = "\xde\xc0\x17\x0b";
Rui Ueyama829c4392013-11-14 22:09:08 +0000485const char coff_object[] = "\x00\x00......";
Rui Ueyamae448f9e2013-11-15 21:22:02 +0000486const char coff_import_library[] = "\x00\x00\xff\xff....";
Rui Ueyama5e3de7a2013-11-13 21:55:41 +0000487const char elf_relocatable[] = { 0x7f, 'E', 'L', 'F', 1, 2, 1, 0, 0,
488 0, 0, 0, 0, 0, 0, 0, 0, 1 };
489const char macho_universal_binary[] = "\xca\xfe\xba\xbe...\0x00";
490const char macho_object[] = "\xfe\xed\xfa\xce..........\x00\x01";
491const char macho_executable[] = "\xfe\xed\xfa\xce..........\x00\x02";
492const char macho_fixed_virtual_memory_shared_lib[] =
493 "\xfe\xed\xfa\xce..........\x00\x03";
494const char macho_core[] = "\xfe\xed\xfa\xce..........\x00\x04";
495const char macho_preload_executable[] = "\xfe\xed\xfa\xce..........\x00\x05";
496const char macho_dynamically_linked_shared_lib[] =
497 "\xfe\xed\xfa\xce..........\x00\x06";
498const char macho_dynamic_linker[] = "\xfe\xed\xfa\xce..........\x00\x07";
499const char macho_bundle[] = "\xfe\xed\xfa\xce..........\x00\x08";
500const char macho_dsym_companion[] = "\xfe\xed\xfa\xce..........\x00\x0a";
501const char windows_resource[] = "\x00\x00\x00\x00\x020\x00\x00\x00\xff";
Michael J. Spencerb8055cb2013-04-05 20:10:04 +0000502
Michael J. Spencer1d3c4a72011-01-06 05:58:02 +0000503TEST_F(FileSystemTest, Magic) {
504 struct type {
505 const char *filename;
506 const char *magic_str;
Michael J. Spencerb8055cb2013-04-05 20:10:04 +0000507 size_t magic_str_len;
508 fs::file_magic magic;
Rui Ueyama5e3de7a2013-11-13 21:55:41 +0000509 } types[] = {
510#define DEFINE(magic) \
511 { #magic, magic, sizeof(magic), fs::file_magic::magic }
512 DEFINE(archive),
513 DEFINE(bitcode),
Rui Ueyama829c4392013-11-14 22:09:08 +0000514 DEFINE(coff_object),
Rui Ueyamae448f9e2013-11-15 21:22:02 +0000515 DEFINE(coff_import_library),
Rui Ueyama5e3de7a2013-11-13 21:55:41 +0000516 DEFINE(elf_relocatable),
517 DEFINE(macho_universal_binary),
518 DEFINE(macho_object),
519 DEFINE(macho_executable),
520 DEFINE(macho_fixed_virtual_memory_shared_lib),
521 DEFINE(macho_core),
522 DEFINE(macho_preload_executable),
523 DEFINE(macho_dynamically_linked_shared_lib),
524 DEFINE(macho_dynamic_linker),
525 DEFINE(macho_bundle),
526 DEFINE(macho_dsym_companion),
527 DEFINE(windows_resource)
528#undef DEFINE
529 };
Michael J. Spencer1d3c4a72011-01-06 05:58:02 +0000530
531 // Create some files filled with magic.
532 for (type *i = types, *e = types + (sizeof(types) / sizeof(type)); i != e;
533 ++i) {
534 SmallString<128> file_pathname(TestDirectory);
535 path::append(file_pathname, i->filename);
536 std::string ErrMsg;
Rafael Espindola90c7f1c2014-02-24 18:20:12 +0000537 raw_fd_ostream file(file_pathname.c_str(), ErrMsg, sys::fs::F_None);
Michael J. Spencer1d3c4a72011-01-06 05:58:02 +0000538 ASSERT_FALSE(file.has_error());
539 StringRef magic(i->magic_str, i->magic_str_len);
540 file << magic;
Michael J. Spencerb5871802011-01-15 18:52:49 +0000541 file.close();
Michael J. Spencer1d3c4a72011-01-06 05:58:02 +0000542 bool res = false;
543 ASSERT_NO_ERROR(fs::has_magic(file_pathname.c_str(), magic, res));
544 EXPECT_TRUE(res);
Michael J. Spencerb8055cb2013-04-05 20:10:04 +0000545 EXPECT_EQ(i->magic, fs::identify_magic(magic));
Rafael Espindola78dcc032014-01-10 20:36:42 +0000546 ASSERT_NO_ERROR(fs::remove(Twine(file_pathname)));
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000547 }
Michael J. Spencer98847782010-11-24 19:20:05 +0000548}
549
Rafael Espindola84ab9b32013-07-19 14:41:25 +0000550#ifdef LLVM_ON_WIN32
551TEST_F(FileSystemTest, CarriageReturn) {
552 SmallString<128> FilePathname(TestDirectory);
553 std::string ErrMsg;
554 path::append(FilePathname, "test");
555
556 {
Rafael Espindola90c7f1c2014-02-24 18:20:12 +0000557 raw_fd_ostream File(FilePathname.c_str(), ErrMsg, sys::fs::F_Text);
Rafael Espindola84ab9b32013-07-19 14:41:25 +0000558 EXPECT_EQ(ErrMsg, "");
559 File << '\n';
560 }
561 {
Ahmed Charles56440fd2014-03-06 05:51:42 +0000562 std::unique_ptr<MemoryBuffer> Buf;
Rafael Espindolae5bf2462013-10-25 19:47:55 +0000563 MemoryBuffer::getFile(FilePathname.c_str(), Buf);
Rafael Espindola84ab9b32013-07-19 14:41:25 +0000564 EXPECT_EQ(Buf->getBuffer(), "\r\n");
565 }
566
567 {
Rafael Espindola90c7f1c2014-02-24 18:20:12 +0000568 raw_fd_ostream File(FilePathname.c_str(), ErrMsg, sys::fs::F_None);
Rafael Espindola84ab9b32013-07-19 14:41:25 +0000569 EXPECT_EQ(ErrMsg, "");
570 File << '\n';
571 }
572 {
Ahmed Charles56440fd2014-03-06 05:51:42 +0000573 std::unique_ptr<MemoryBuffer> Buf;
Rafael Espindolae5bf2462013-10-25 19:47:55 +0000574 MemoryBuffer::getFile(FilePathname.c_str(), Buf);
Rafael Espindola84ab9b32013-07-19 14:41:25 +0000575 EXPECT_EQ(Buf->getBuffer(), "\n");
576 }
Rafael Espindola78dcc032014-01-10 20:36:42 +0000577 ASSERT_NO_ERROR(fs::remove(Twine(FilePathname)));
Rafael Espindola84ab9b32013-07-19 14:41:25 +0000578}
579#endif
580
Nick Kledzik18497e92012-06-20 00:28:54 +0000581TEST_F(FileSystemTest, FileMapping) {
582 // Create a temp file.
583 int FileDescriptor;
584 SmallString<64> TempPath;
585 ASSERT_NO_ERROR(
Rafael Espindola155cf0f2013-07-05 20:14:52 +0000586 fs::createTemporaryFile("prefix", "temp", FileDescriptor, TempPath));
Nick Kledzik18497e92012-06-20 00:28:54 +0000587 // Map in temp file and add some content
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000588 error_code EC;
589 StringRef Val("hello there");
590 {
591 fs::mapped_file_region mfr(FileDescriptor,
Michael J. Spencer01ac34e2013-03-14 00:33:37 +0000592 true,
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000593 fs::mapped_file_region::readwrite,
594 4096,
595 0,
596 EC);
597 ASSERT_NO_ERROR(EC);
598 std::copy(Val.begin(), Val.end(), mfr.data());
599 // Explicitly add a 0.
600 mfr.data()[Val.size()] = 0;
601 // Unmap temp file
602 }
Rui Ueyama89d1bdb2013-11-13 20:31:21 +0000603
Nick Kledzik18497e92012-06-20 00:28:54 +0000604 // Map it back in read-only
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000605 fs::mapped_file_region mfr(Twine(TempPath),
606 fs::mapped_file_region::readonly,
607 0,
608 0,
609 EC);
610 ASSERT_NO_ERROR(EC);
Rui Ueyama89d1bdb2013-11-13 20:31:21 +0000611
Nick Kledzik18497e92012-06-20 00:28:54 +0000612 // Verify content
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000613 EXPECT_EQ(StringRef(mfr.const_data()), Val);
Rui Ueyama89d1bdb2013-11-13 20:31:21 +0000614
Nick Kledzik18497e92012-06-20 00:28:54 +0000615 // Unmap temp file
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000616
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000617 fs::mapped_file_region m(Twine(TempPath),
618 fs::mapped_file_region::readonly,
619 0,
620 0,
621 EC);
622 ASSERT_NO_ERROR(EC);
623 const char *Data = m.const_data();
Chandler Carruth002da5d2014-03-02 04:08:41 +0000624 fs::mapped_file_region mfrrv(std::move(m));
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000625 EXPECT_EQ(mfrrv.const_data(), Data);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000626}
Saleem Abdulrasoolafc50b3ed2014-03-11 22:05:42 +0000627
628TEST(Support, NormalizePath) {
629#if defined(LLVM_ON_WIN32)
630#define EXPECT_PATH_IS(path__, windows__, not_windows__) \
631 EXPECT_EQ(path__, windows__);
632#else
633#define EXPECT_PATH_IS(path__, windows__, not_windows__) \
634 EXPECT_EQ(path__, not_windows__);
635#endif
636
637 SmallString<64> Path1("a");
638 SmallString<64> Path2("a/b");
639 SmallString<64> Path3("a\\b");
640 SmallString<64> Path4("a\\\\b");
641 SmallString<64> Path5("\\a");
642 SmallString<64> Path6("a\\");
643
644 ASSERT_NO_ERROR(fs::normalize_separators(Path1));
645 EXPECT_PATH_IS(Path1, "a", "a");
646
647 ASSERT_NO_ERROR(fs::normalize_separators(Path2));
648 EXPECT_PATH_IS(Path2, "a/b", "a/b");
649
650 ASSERT_NO_ERROR(fs::normalize_separators(Path3));
651 EXPECT_PATH_IS(Path3, "a\\b", "a/b");
652
653 ASSERT_NO_ERROR(fs::normalize_separators(Path4));
654 EXPECT_PATH_IS(Path4, "a\\\\b", "a\\\\b");
655
656 ASSERT_NO_ERROR(fs::normalize_separators(Path5));
657 EXPECT_PATH_IS(Path5, "\\a", "/a");
658
659 ASSERT_NO_ERROR(fs::normalize_separators(Path6));
660 EXPECT_PATH_IS(Path6, "a\\", "a/");
661
662#undef EXPECT_PATH_IS
663}
Michael J. Spencer98847782010-11-24 19:20:05 +0000664} // anonymous namespace