blob: ea383895f2a65b0afa074cad05b55cee83cbd23a [file] [log] [blame]
Michael J. Spencerae7de7e2010-12-09 17:37:42 +00001//===- llvm/Support/Windows/PathV2.inc - Windows Path Impl ------*- C++ -*-===//
Michael J. Spencerebad2f92010-11-29 22:28:51 +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//
10// This file implements the Windows specific implementation of the PathV2 API.
11//
12//===----------------------------------------------------------------------===//
13
14//===----------------------------------------------------------------------===//
15//=== WARNING: Implementation here must contain only generic Windows code that
16//=== is guaranteed to work on *all* Windows variants.
17//===----------------------------------------------------------------------===//
18
19#include "Windows.h"
Michael J. Spencerc20a0322010-12-03 17:54:07 +000020#include <fcntl.h>
Michael J. Spencer45710402010-12-03 01:21:28 +000021#include <io.h>
Michael J. Spencerc20a0322010-12-03 17:54:07 +000022#include <sys/stat.h>
23#include <sys/types.h>
Michael J. Spencerebad2f92010-11-29 22:28:51 +000024
Michael J. Spenceref2284f2012-08-15 19:05:47 +000025#undef max
26
Michael J. Spencer60252472010-12-03 18:03:28 +000027// MinGW doesn't define this.
Michael J. Spencer521c3212010-12-03 18:04:11 +000028#ifndef _ERRNO_T_DEFINED
29#define _ERRNO_T_DEFINED
30typedef int errno_t;
Michael J. Spencer60252472010-12-03 18:03:28 +000031#endif
32
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000033using namespace llvm;
34
35namespace {
Michael J. Spencer772caff52010-12-03 17:53:23 +000036 typedef BOOLEAN (WINAPI *PtrCreateSymbolicLinkW)(
37 /*__in*/ LPCWSTR lpSymlinkFileName,
38 /*__in*/ LPCWSTR lpTargetFileName,
39 /*__in*/ DWORD dwFlags);
40
41 PtrCreateSymbolicLinkW create_symbolic_link_api = PtrCreateSymbolicLinkW(
Michael J. Spencer7ee6d5d2010-12-03 07:41:25 +000042 ::GetProcAddress(::GetModuleHandleA("kernel32.dll"),
43 "CreateSymbolicLinkW"));
44
Benjamin Kramercb520cd2010-12-17 18:59:09 +000045 error_code UTF8ToUTF16(StringRef utf8, SmallVectorImpl<wchar_t> &utf16) {
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000046 int len = ::MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS,
47 utf8.begin(), utf8.size(),
48 utf16.begin(), 0);
49
50 if (len == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +000051 return windows_error(::GetLastError());
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000052
53 utf16.reserve(len + 1);
54 utf16.set_size(len);
55
56 len = ::MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS,
57 utf8.begin(), utf8.size(),
58 utf16.begin(), utf16.size());
59
60 if (len == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +000061 return windows_error(::GetLastError());
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000062
63 // Make utf16 null terminated.
64 utf16.push_back(0);
65 utf16.pop_back();
66
David Blaikie18544b92012-02-09 19:24:12 +000067 return error_code::success();
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000068 }
Michael J. Spencer45710402010-12-03 01:21:28 +000069
70 error_code UTF16ToUTF8(const wchar_t *utf16, size_t utf16_len,
71 SmallVectorImpl<char> &utf8) {
72 // Get length.
Michael J. Spencer962fbc42010-12-04 00:32:14 +000073 int len = ::WideCharToMultiByte(CP_UTF8, 0,
Michael J. Spencer45710402010-12-03 01:21:28 +000074 utf16, utf16_len,
75 utf8.begin(), 0,
76 NULL, NULL);
77
78 if (len == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +000079 return windows_error(::GetLastError());
Michael J. Spencer45710402010-12-03 01:21:28 +000080
81 utf8.reserve(len);
82 utf8.set_size(len);
83
84 // Now do the actual conversion.
Michael J. Spencer962fbc42010-12-04 00:32:14 +000085 len = ::WideCharToMultiByte(CP_UTF8, 0,
Michael J. Spencer45710402010-12-03 01:21:28 +000086 utf16, utf16_len,
87 utf8.data(), utf8.size(),
88 NULL, NULL);
89
90 if (len == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +000091 return windows_error(::GetLastError());
Michael J. Spencer45710402010-12-03 01:21:28 +000092
93 // Make utf8 null terminated.
94 utf8.push_back(0);
95 utf8.pop_back();
96
David Blaikie18544b92012-02-09 19:24:12 +000097 return error_code::success();
Michael J. Spencer45710402010-12-03 01:21:28 +000098 }
99
100 error_code TempDir(SmallVectorImpl<wchar_t> &result) {
101 retry_temp_dir:
102 DWORD len = ::GetTempPathW(result.capacity(), result.begin());
103
104 if (len == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000105 return windows_error(::GetLastError());
Michael J. Spencer45710402010-12-03 01:21:28 +0000106
107 if (len > result.capacity()) {
108 result.reserve(len);
109 goto retry_temp_dir;
110 }
111
112 result.set_size(len);
David Blaikie18544b92012-02-09 19:24:12 +0000113 return error_code::success();
Michael J. Spencer45710402010-12-03 01:21:28 +0000114 }
115
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000116 bool is_separator(const wchar_t value) {
117 switch (value) {
118 case L'\\':
119 case L'/':
120 return true;
121 default:
122 return false;
123 }
124 }
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +0000125}
126
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000127namespace llvm {
128namespace sys {
Michael J. Spencer20daa282010-12-07 01:22:31 +0000129namespace fs {
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000130
131error_code current_path(SmallVectorImpl<char> &result) {
132 SmallVector<wchar_t, 128> cur_path;
133 cur_path.reserve(128);
134retry_cur_dir:
135 DWORD len = ::GetCurrentDirectoryW(cur_path.capacity(), cur_path.data());
136
137 // A zero return value indicates a failure other than insufficient space.
138 if (len == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000139 return windows_error(::GetLastError());
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000140
141 // If there's insufficient space, the len returned is larger than the len
142 // given.
143 if (len > cur_path.capacity()) {
144 cur_path.reserve(len);
145 goto retry_cur_dir;
146 }
147
148 cur_path.set_size(len);
149 // cur_path now holds the current directory in utf-16. Convert to utf-8.
150
151 // Find out how much space we need. Sadly, this function doesn't return the
152 // size needed unless you tell it the result size is 0, which means you
153 // _always_ have to call it twice.
Michael J. Spencer962fbc42010-12-04 00:32:14 +0000154 len = ::WideCharToMultiByte(CP_UTF8, 0,
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000155 cur_path.data(), cur_path.size(),
156 result.data(), 0,
157 NULL, NULL);
158
159 if (len == 0)
160 return make_error_code(windows_error(::GetLastError()));
161
162 result.reserve(len);
163 result.set_size(len);
164 // Now do the actual conversion.
Michael J. Spencer962fbc42010-12-04 00:32:14 +0000165 len = ::WideCharToMultiByte(CP_UTF8, 0,
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000166 cur_path.data(), cur_path.size(),
167 result.data(), result.size(),
168 NULL, NULL);
169 if (len == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000170 return windows_error(::GetLastError());
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000171
David Blaikie18544b92012-02-09 19:24:12 +0000172 return error_code::success();
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000173}
174
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +0000175error_code copy_file(const Twine &from, const Twine &to, copy_option copt) {
176 // Get arguments.
177 SmallString<128> from_storage;
178 SmallString<128> to_storage;
Michael J. Spencera41772a2010-12-03 01:21:38 +0000179 StringRef f = from.toStringRef(from_storage);
180 StringRef t = to.toStringRef(to_storage);
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +0000181
182 // Convert to utf-16.
183 SmallVector<wchar_t, 128> wide_from;
184 SmallVector<wchar_t, 128> wide_to;
185 if (error_code ec = UTF8ToUTF16(f, wide_from)) return ec;
186 if (error_code ec = UTF8ToUTF16(t, wide_to)) return ec;
187
188 // Copy the file.
189 BOOL res = ::CopyFileW(wide_from.begin(), wide_to.begin(),
190 copt != copy_option::overwrite_if_exists);
191
192 if (res == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000193 return windows_error(::GetLastError());
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +0000194
David Blaikie18544b92012-02-09 19:24:12 +0000195 return error_code::success();
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +0000196}
197
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000198error_code create_directory(const Twine &path, bool &existed) {
199 SmallString<128> path_storage;
200 SmallVector<wchar_t, 128> path_utf16;
201
202 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
203 path_utf16))
204 return ec;
205
206 if (!::CreateDirectoryW(path_utf16.begin(), NULL)) {
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000207 error_code ec = windows_error(::GetLastError());
208 if (ec == windows_error::already_exists)
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000209 existed = true;
210 else
211 return ec;
212 } else
213 existed = false;
214
David Blaikie18544b92012-02-09 19:24:12 +0000215 return error_code::success();
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000216}
217
Michael J. Spencere0c45602010-12-03 05:58:41 +0000218error_code create_hard_link(const Twine &to, const Twine &from) {
219 // Get arguments.
220 SmallString<128> from_storage;
221 SmallString<128> to_storage;
222 StringRef f = from.toStringRef(from_storage);
223 StringRef t = to.toStringRef(to_storage);
224
225 // Convert to utf-16.
226 SmallVector<wchar_t, 128> wide_from;
227 SmallVector<wchar_t, 128> wide_to;
228 if (error_code ec = UTF8ToUTF16(f, wide_from)) return ec;
229 if (error_code ec = UTF8ToUTF16(t, wide_to)) return ec;
230
231 if (!::CreateHardLinkW(wide_from.begin(), wide_to.begin(), NULL))
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000232 return windows_error(::GetLastError());
Michael J. Spencere0c45602010-12-03 05:58:41 +0000233
David Blaikie18544b92012-02-09 19:24:12 +0000234 return error_code::success();
Michael J. Spencere0c45602010-12-03 05:58:41 +0000235}
236
Michael J. Spencer7ee6d5d2010-12-03 07:41:25 +0000237error_code create_symlink(const Twine &to, const Twine &from) {
238 // Only do it if the function is available at runtime.
239 if (!create_symbolic_link_api)
240 return make_error_code(errc::function_not_supported);
241
242 // Get arguments.
243 SmallString<128> from_storage;
244 SmallString<128> to_storage;
245 StringRef f = from.toStringRef(from_storage);
246 StringRef t = to.toStringRef(to_storage);
247
248 // Convert to utf-16.
249 SmallVector<wchar_t, 128> wide_from;
250 SmallVector<wchar_t, 128> wide_to;
251 if (error_code ec = UTF8ToUTF16(f, wide_from)) return ec;
252 if (error_code ec = UTF8ToUTF16(t, wide_to)) return ec;
253
Michael J. Spencer962fbc42010-12-04 00:32:14 +0000254 if (!create_symbolic_link_api(wide_from.begin(), wide_to.begin(), 0))
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000255 return windows_error(::GetLastError());
Michael J. Spencer7ee6d5d2010-12-03 07:41:25 +0000256
David Blaikie18544b92012-02-09 19:24:12 +0000257 return error_code::success();
Michael J. Spencer7ee6d5d2010-12-03 07:41:25 +0000258}
259
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000260error_code remove(const Twine &path, bool &existed) {
261 SmallString<128> path_storage;
262 SmallVector<wchar_t, 128> path_utf16;
263
Michael J. Spencer153749b2011-01-05 16:39:22 +0000264 file_status st;
265 if (error_code ec = status(path, st))
266 return ec;
267
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000268 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
269 path_utf16))
270 return ec;
271
Michael J. Spencer153749b2011-01-05 16:39:22 +0000272 if (st.type() == file_type::directory_file) {
273 if (!::RemoveDirectoryW(c_str(path_utf16))) {
274 error_code ec = windows_error(::GetLastError());
275 if (ec != windows_error::file_not_found)
276 return ec;
277 existed = false;
278 } else
279 existed = true;
280 } else {
281 if (!::DeleteFileW(c_str(path_utf16))) {
282 error_code ec = windows_error(::GetLastError());
283 if (ec != windows_error::file_not_found)
284 return ec;
285 existed = false;
286 } else
287 existed = true;
288 }
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000289
David Blaikie18544b92012-02-09 19:24:12 +0000290 return error_code::success();
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000291}
292
Michael J. Spencer409f5562010-12-03 17:53:55 +0000293error_code rename(const Twine &from, const Twine &to) {
294 // Get arguments.
295 SmallString<128> from_storage;
296 SmallString<128> to_storage;
297 StringRef f = from.toStringRef(from_storage);
298 StringRef t = to.toStringRef(to_storage);
299
300 // Convert to utf-16.
301 SmallVector<wchar_t, 128> wide_from;
302 SmallVector<wchar_t, 128> wide_to;
303 if (error_code ec = UTF8ToUTF16(f, wide_from)) return ec;
304 if (error_code ec = UTF8ToUTF16(t, wide_to)) return ec;
305
NAKAMURA Takumi3b7f9952012-05-08 14:31:46 +0000306 error_code ec = error_code::success();
307 for (int i = 0; i < 2000; i++) {
308 if (::MoveFileExW(wide_from.begin(), wide_to.begin(),
309 MOVEFILE_COPY_ALLOWED | MOVEFILE_REPLACE_EXISTING))
310 return error_code::success();
311 ec = windows_error(::GetLastError());
312 if (ec != windows_error::access_denied)
313 break;
314 // Retry MoveFile() at ACCESS_DENIED.
315 // System scanners (eg. indexer) might open the source file when
316 // It is written and closed.
317 ::Sleep(1);
318 }
Michael J. Spencer409f5562010-12-03 17:53:55 +0000319
NAKAMURA Takumi3b7f9952012-05-08 14:31:46 +0000320 return ec;
Michael J. Spencer409f5562010-12-03 17:53:55 +0000321}
322
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000323error_code resize_file(const Twine &path, uint64_t size) {
324 SmallString<128> path_storage;
325 SmallVector<wchar_t, 128> path_utf16;
326
327 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
328 path_utf16))
329 return ec;
330
Michael J. Spencer20abb202012-12-03 22:09:31 +0000331 int fd = ::_wopen(path_utf16.begin(), O_BINARY | _O_RDWR, S_IWRITE);
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000332 if (fd == -1)
333 return error_code(errno, generic_category());
Michael J. Spencerca242f22010-12-03 18:48:56 +0000334#ifdef HAVE__CHSIZE_S
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000335 errno_t error = ::_chsize_s(fd, size);
Michael J. Spencerca242f22010-12-03 18:48:56 +0000336#else
337 errno_t error = ::_chsize(fd, size);
338#endif
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000339 ::close(fd);
340 return error_code(error, generic_category());
341}
342
Michael J. Spencer45710402010-12-03 01:21:28 +0000343error_code exists(const Twine &path, bool &result) {
344 SmallString<128> path_storage;
345 SmallVector<wchar_t, 128> path_utf16;
346
347 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
348 path_utf16))
349 return ec;
350
351 DWORD attributes = ::GetFileAttributesW(path_utf16.begin());
352
353 if (attributes == INVALID_FILE_ATTRIBUTES) {
354 // See if the file didn't actually exist.
355 error_code ec = make_error_code(windows_error(::GetLastError()));
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000356 if (ec != windows_error::file_not_found &&
357 ec != windows_error::path_not_found)
Michael J. Spencer45710402010-12-03 01:21:28 +0000358 return ec;
359 result = false;
360 } else
361 result = true;
David Blaikie18544b92012-02-09 19:24:12 +0000362 return error_code::success();
Michael J. Spencer45710402010-12-03 01:21:28 +0000363}
364
Rafael Espindolab0a5c962013-06-14 19:38:45 +0000365bool can_execute(const Twine &Path) {
366 SmallString<128> PathStorage;
367 SmallVector<wchar_t, 128> PathUtf16;
368
369 if (UTF8ToUTF16(Path.toStringRef(PathStorage), PathUtf16))
370 return false;
371
372 DWORD Attr = ::GetFileAttributesW(PathUtf16.begin());
373 return Attr != INVALID_FILE_ATTRIBUTES;
374}
375
Michael J. Spencer203d7802011-12-12 06:04:28 +0000376bool equivalent(file_status A, file_status B) {
377 assert(status_known(A) && status_known(B));
378 return A.FileIndexHigh == B.FileIndexHigh &&
379 A.FileIndexLow == B.FileIndexLow &&
380 A.FileSizeHigh == B.FileSizeHigh &&
381 A.FileSizeLow == B.FileSizeLow &&
382 A.LastWriteTimeHigh == B.LastWriteTimeHigh &&
383 A.LastWriteTimeLow == B.LastWriteTimeLow &&
384 A.VolumeSerialNumber == B.VolumeSerialNumber;
385}
386
Michael J. Spencer376d3872010-12-03 18:49:13 +0000387error_code equivalent(const Twine &A, const Twine &B, bool &result) {
Michael J. Spencer203d7802011-12-12 06:04:28 +0000388 file_status fsA, fsB;
389 if (error_code ec = status(A, fsA)) return ec;
390 if (error_code ec = status(B, fsB)) return ec;
391 result = equivalent(fsA, fsB);
David Blaikie18544b92012-02-09 19:24:12 +0000392 return error_code::success();
Michael J. Spencer376d3872010-12-03 18:49:13 +0000393}
394
Michael J. Spencer818ab4a2010-12-04 00:31:48 +0000395error_code file_size(const Twine &path, uint64_t &result) {
396 SmallString<128> path_storage;
397 SmallVector<wchar_t, 128> path_utf16;
398
399 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
400 path_utf16))
401 return ec;
402
403 WIN32_FILE_ATTRIBUTE_DATA FileData;
404 if (!::GetFileAttributesExW(path_utf16.begin(),
405 ::GetFileExInfoStandard,
406 &FileData))
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000407 return windows_error(::GetLastError());
Michael J. Spencer818ab4a2010-12-04 00:31:48 +0000408
409 result =
410 (uint64_t(FileData.nFileSizeHigh) << (sizeof(FileData.nFileSizeLow) * 8))
411 + FileData.nFileSizeLow;
412
David Blaikie18544b92012-02-09 19:24:12 +0000413 return error_code::success();
Michael J. Spencer818ab4a2010-12-04 00:31:48 +0000414}
415
Benjamin Kramer58994ec2011-08-20 21:36:38 +0000416static bool isReservedName(StringRef path) {
417 // This list of reserved names comes from MSDN, at:
418 // http://msdn.microsoft.com/en-us/library/aa365247%28v=vs.85%29.aspx
419 static const char *sReservedNames[] = { "nul", "con", "prn", "aux",
420 "com1", "com2", "com3", "com4", "com5", "com6",
421 "com7", "com8", "com9", "lpt1", "lpt2", "lpt3",
422 "lpt4", "lpt5", "lpt6", "lpt7", "lpt8", "lpt9" };
423
424 // First, check to see if this is a device namespace, which always
425 // starts with \\.\, since device namespaces are not legal file paths.
426 if (path.startswith("\\\\.\\"))
427 return true;
428
429 // Then compare against the list of ancient reserved names
430 for (size_t i = 0; i < sizeof(sReservedNames) / sizeof(const char *); ++i) {
431 if (path.equals_lower(sReservedNames[i]))
432 return true;
433 }
434
435 // The path isn't what we consider reserved.
436 return false;
437}
438
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000439error_code status(const Twine &path, file_status &result) {
440 SmallString<128> path_storage;
441 SmallVector<wchar_t, 128> path_utf16;
442
NAKAMURA Takumia3d47492011-03-16 02:53:32 +0000443 StringRef path8 = path.toStringRef(path_storage);
Benjamin Kramer58994ec2011-08-20 21:36:38 +0000444 if (isReservedName(path8)) {
NAKAMURA Takumia3d47492011-03-16 02:53:32 +0000445 result = file_status(file_type::character_file);
David Blaikie18544b92012-02-09 19:24:12 +0000446 return error_code::success();
NAKAMURA Takumia3d47492011-03-16 02:53:32 +0000447 }
448
Michael J. Spencer203d7802011-12-12 06:04:28 +0000449 if (error_code ec = UTF8ToUTF16(path8, path_utf16))
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000450 return ec;
451
452 DWORD attr = ::GetFileAttributesW(path_utf16.begin());
453 if (attr == INVALID_FILE_ATTRIBUTES)
454 goto handle_status_error;
455
456 // Handle reparse points.
457 if (attr & FILE_ATTRIBUTE_REPARSE_POINT) {
Michael J. Spencer513f1b62011-12-12 06:03:33 +0000458 ScopedFileHandle h(
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000459 ::CreateFileW(path_utf16.begin(),
460 0, // Attributes only.
461 FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
462 NULL,
463 OPEN_EXISTING,
464 FILE_FLAG_BACKUP_SEMANTICS,
465 0));
Michael J. Spencer513f1b62011-12-12 06:03:33 +0000466 if (!h)
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000467 goto handle_status_error;
468 }
469
470 if (attr & FILE_ATTRIBUTE_DIRECTORY)
471 result = file_status(file_type::directory_file);
Michael J. Spencer203d7802011-12-12 06:04:28 +0000472 else {
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000473 result = file_status(file_type::regular_file);
Michael J. Spencer203d7802011-12-12 06:04:28 +0000474 ScopedFileHandle h(
475 ::CreateFileW(path_utf16.begin(),
476 0, // Attributes only.
477 FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
478 NULL,
479 OPEN_EXISTING,
480 FILE_FLAG_BACKUP_SEMANTICS,
481 0));
482 if (!h)
483 goto handle_status_error;
484 BY_HANDLE_FILE_INFORMATION Info;
485 if (!::GetFileInformationByHandle(h, &Info))
486 goto handle_status_error;
487 result.FileIndexHigh = Info.nFileIndexHigh;
488 result.FileIndexLow = Info.nFileIndexLow;
489 result.FileSizeHigh = Info.nFileSizeHigh;
490 result.FileSizeLow = Info.nFileSizeLow;
491 result.LastWriteTimeHigh = Info.ftLastWriteTime.dwHighDateTime;
492 result.LastWriteTimeLow = Info.ftLastWriteTime.dwLowDateTime;
493 result.VolumeSerialNumber = Info.dwVolumeSerialNumber;
494 }
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000495
David Blaikie18544b92012-02-09 19:24:12 +0000496 return error_code::success();
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000497
498handle_status_error:
499 error_code ec = windows_error(::GetLastError());
500 if (ec == windows_error::file_not_found ||
501 ec == windows_error::path_not_found)
502 result = file_status(file_type::file_not_found);
503 else if (ec == windows_error::sharing_violation)
504 result = file_status(file_type::type_unknown);
505 else {
506 result = file_status(file_type::status_error);
507 return ec;
508 }
509
David Blaikie18544b92012-02-09 19:24:12 +0000510 return error_code::success();
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000511}
512
Nick Kledzik18497e92012-06-20 00:28:54 +0000513
514// Modifies permissions on a file.
515error_code permissions(const Twine &path, perms prms) {
516#if 0 // verify code below before enabling:
517 // If the permissions bits are not trying to modify
518 // "write" permissions, there is nothing to do.
519 if (!(prms & (owner_write|group_write|others_write)))
520 return error_code::success();
521
522 SmallString<128> path_storage;
523 SmallVector<wchar_t, 128> path_utf16;
524
525 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
526 path_utf16))
527 return ec;
528
529 DWORD attributes = ::GetFileAttributesW(path_utf16.begin());
530
531 if (prms & add_perms) {
532 attributes &= ~FILE_ATTRIBUTE_READONLY;
533 }
534 else if (prms & remove_perms) {
535 attributes |= FILE_ATTRIBUTE_READONLY;
536 }
537 else {
538 assert(0 && "neither add_perms or remove_perms is set");
539 }
540
541 if ( ! ::SetFileAttributesW(path_utf16.begin(), attributes))
542 return windows_error(::GetLastError());
543#endif
544 return error_code::success();
545}
546
547
Eric Christopherb6148ed2012-05-11 00:07:44 +0000548// FIXME: mode should be used here and default to user r/w only,
549// it currently comes in as a UNIX mode.
Michael J. Spencer45710402010-12-03 01:21:28 +0000550error_code unique_file(const Twine &model, int &result_fd,
Eric Christopherb6148ed2012-05-11 00:07:44 +0000551 SmallVectorImpl<char> &result_path,
552 bool makeAbsolute, unsigned mode) {
Michael J. Spencer45710402010-12-03 01:21:28 +0000553 // Use result_path as temp storage.
554 result_path.set_size(0);
555 StringRef m = model.toStringRef(result_path);
556
557 SmallVector<wchar_t, 128> model_utf16;
558 if (error_code ec = UTF8ToUTF16(m, model_utf16)) return ec;
559
Argyrios Kyrtzidis348937d2011-07-28 00:29:20 +0000560 if (makeAbsolute) {
561 // Make model absolute by prepending a temp directory if it's not already.
562 bool absolute = path::is_absolute(m);
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000563
Argyrios Kyrtzidis348937d2011-07-28 00:29:20 +0000564 if (!absolute) {
565 SmallVector<wchar_t, 64> temp_dir;
566 if (error_code ec = TempDir(temp_dir)) return ec;
567 // Handle c: by removing it.
568 if (model_utf16.size() > 2 && model_utf16[1] == L':') {
569 model_utf16.erase(model_utf16.begin(), model_utf16.begin() + 2);
570 }
571 model_utf16.insert(model_utf16.begin(), temp_dir.begin(), temp_dir.end());
Michael J. Spencer45710402010-12-03 01:21:28 +0000572 }
Michael J. Spencer45710402010-12-03 01:21:28 +0000573 }
574
575 // Replace '%' with random chars. From here on, DO NOT modify model. It may be
576 // needed if the randomly chosen path already exists.
577 SmallVector<wchar_t, 128> random_path_utf16;
578
579 // Get a Crypto Provider for CryptGenRandom.
Michael J. Spencer39c46212010-12-06 04:28:13 +0000580 HCRYPTPROV HCPC;
581 if (!::CryptAcquireContextW(&HCPC,
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000582 NULL,
583 NULL,
584 PROV_RSA_FULL,
NAKAMURA Takumi8a074512011-01-17 22:41:34 +0000585 CRYPT_VERIFYCONTEXT))
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000586 return windows_error(::GetLastError());
Michael J. Spencer39c46212010-12-06 04:28:13 +0000587 ScopedCryptContext CryptoProvider(HCPC);
Michael J. Spencer45710402010-12-03 01:21:28 +0000588
589retry_random_path:
590 random_path_utf16.set_size(0);
591 for (SmallVectorImpl<wchar_t>::const_iterator i = model_utf16.begin(),
592 e = model_utf16.end();
593 i != e; ++i) {
594 if (*i == L'%') {
595 BYTE val = 0;
596 if (!::CryptGenRandom(CryptoProvider, 1, &val))
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000597 return windows_error(::GetLastError());
Michael J. Spencer45710402010-12-03 01:21:28 +0000598 random_path_utf16.push_back("0123456789abcdef"[val & 15]);
599 }
600 else
601 random_path_utf16.push_back(*i);
602 }
603 // Make random_path_utf16 null terminated.
604 random_path_utf16.push_back(0);
605 random_path_utf16.pop_back();
606
Michael J. Spencer20a11b22013-03-12 22:32:39 +0000607 // Make sure we don't fall into an infinite loop by constantly trying
608 // to create the parent path.
609 bool TriedToCreateParent = false;
610
Michael J. Spencer45710402010-12-03 01:21:28 +0000611 // Try to create + open the path.
612retry_create_file:
613 HANDLE TempFileHandle = ::CreateFileW(random_path_utf16.begin(),
614 GENERIC_READ | GENERIC_WRITE,
615 FILE_SHARE_READ,
616 NULL,
617 // Return ERROR_FILE_EXISTS if the file
618 // already exists.
619 CREATE_NEW,
620 FILE_ATTRIBUTE_TEMPORARY,
621 NULL);
622 if (TempFileHandle == INVALID_HANDLE_VALUE) {
623 // If the file existed, try again, otherwise, error.
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000624 error_code ec = windows_error(::GetLastError());
625 if (ec == windows_error::file_exists)
Michael J. Spencer45710402010-12-03 01:21:28 +0000626 goto retry_random_path;
627 // Check for non-existing parent directories.
Michael J. Spencer20a11b22013-03-12 22:32:39 +0000628 if (ec == windows_error::path_not_found && !TriedToCreateParent) {
629 TriedToCreateParent = true;
630
Michael J. Spencer45710402010-12-03 01:21:28 +0000631 // Create the directories using result_path as temp storage.
632 if (error_code ec = UTF16ToUTF8(random_path_utf16.begin(),
633 random_path_utf16.size(), result_path))
634 return ec;
635 StringRef p(result_path.begin(), result_path.size());
636 SmallString<64> dir_to_create;
637 for (path::const_iterator i = path::begin(p),
638 e = --path::end(p); i != e; ++i) {
Michael J. Spencer1e090f02010-12-07 03:57:37 +0000639 path::append(dir_to_create, *i);
Michael J. Spencer45710402010-12-03 01:21:28 +0000640 bool Exists;
641 if (error_code ec = exists(Twine(dir_to_create), Exists)) return ec;
642 if (!Exists) {
643 // If c: doesn't exist, bail.
644 if (i->endswith(":"))
645 return ec;
646
647 SmallVector<wchar_t, 64> dir_to_create_utf16;
648 if (error_code ec = UTF8ToUTF16(dir_to_create, dir_to_create_utf16))
649 return ec;
650
651 // Create the directory.
652 if (!::CreateDirectoryW(dir_to_create_utf16.begin(), NULL))
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000653 return windows_error(::GetLastError());
Michael J. Spencer45710402010-12-03 01:21:28 +0000654 }
655 }
656 goto retry_create_file;
657 }
658 return ec;
659 }
660
661 // Set result_path to the utf-8 representation of the path.
662 if (error_code ec = UTF16ToUTF8(random_path_utf16.begin(),
663 random_path_utf16.size(), result_path)) {
664 ::CloseHandle(TempFileHandle);
665 ::DeleteFileW(random_path_utf16.begin());
666 return ec;
667 }
668
669 // Convert the Windows API file handle into a C-runtime handle.
670 int fd = ::_open_osfhandle(intptr_t(TempFileHandle), 0);
671 if (fd == -1) {
672 ::CloseHandle(TempFileHandle);
673 ::DeleteFileW(random_path_utf16.begin());
674 // MSDN doesn't say anything about _open_osfhandle setting errno or
675 // GetLastError(), so just return invalid_handle.
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000676 return windows_error::invalid_handle;
Michael J. Spencer45710402010-12-03 01:21:28 +0000677 }
678
679 result_fd = fd;
David Blaikie18544b92012-02-09 19:24:12 +0000680 return error_code::success();
Michael J. Spencer45710402010-12-03 01:21:28 +0000681}
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000682
Michael J. Spenceree1699c2011-01-15 18:52:33 +0000683error_code get_magic(const Twine &path, uint32_t len,
684 SmallVectorImpl<char> &result) {
685 SmallString<128> path_storage;
686 SmallVector<wchar_t, 128> path_utf16;
687 result.set_size(0);
688
689 // Convert path to UTF-16.
690 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
691 path_utf16))
692 return ec;
693
694 // Open file.
695 HANDLE file = ::CreateFileW(c_str(path_utf16),
696 GENERIC_READ,
697 FILE_SHARE_READ,
698 NULL,
699 OPEN_EXISTING,
700 FILE_ATTRIBUTE_READONLY,
701 NULL);
702 if (file == INVALID_HANDLE_VALUE)
703 return windows_error(::GetLastError());
704
705 // Allocate buffer.
706 result.reserve(len);
707
708 // Get magic!
709 DWORD bytes_read = 0;
710 BOOL read_success = ::ReadFile(file, result.data(), len, &bytes_read, NULL);
711 error_code ec = windows_error(::GetLastError());
712 ::CloseHandle(file);
713 if (!read_success || (bytes_read != len)) {
714 // Set result size to the number of bytes read if it's valid.
NAKAMURA Takumi3e76b502011-03-07 00:12:53 +0000715 if (bytes_read <= len)
Michael J. Spenceree1699c2011-01-15 18:52:33 +0000716 result.set_size(bytes_read);
717 // ERROR_HANDLE_EOF is mapped to errc::value_too_large.
718 return ec;
719 }
720
721 result.set_size(len);
David Blaikie18544b92012-02-09 19:24:12 +0000722 return error_code::success();
Michael J. Spenceree1699c2011-01-15 18:52:33 +0000723}
724
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000725error_code mapped_file_region::init(int FD, bool CloseFD, uint64_t Offset) {
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000726 FileDescriptor = FD;
727 // Make sure that the requested size fits within SIZE_T.
728 if (Size > std::numeric_limits<SIZE_T>::max()) {
Michael J. Spencerd932d412013-03-15 19:25:47 +0000729 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000730 if (CloseFD)
731 _close(FileDescriptor);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000732 } else
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000733 ::CloseHandle(FileHandle);
734 return make_error_code(errc::invalid_argument);
735 }
736
737 DWORD flprotect;
738 switch (Mode) {
739 case readonly: flprotect = PAGE_READONLY; break;
740 case readwrite: flprotect = PAGE_READWRITE; break;
741 case priv: flprotect = PAGE_WRITECOPY; break;
742 default: llvm_unreachable("invalid mapping mode");
743 }
744
745 FileMappingHandle = ::CreateFileMapping(FileHandle,
746 0,
747 flprotect,
748 Size >> 32,
749 Size & 0xffffffff,
750 0);
751 if (FileMappingHandle == NULL) {
752 error_code ec = windows_error(GetLastError());
Michael J. Spencerd932d412013-03-15 19:25:47 +0000753 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000754 if (CloseFD)
755 _close(FileDescriptor);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000756 } else
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000757 ::CloseHandle(FileHandle);
758 return ec;
759 }
760
761 DWORD dwDesiredAccess;
762 switch (Mode) {
763 case readonly: dwDesiredAccess = FILE_MAP_READ; break;
764 case readwrite: dwDesiredAccess = FILE_MAP_WRITE; break;
765 case priv: dwDesiredAccess = FILE_MAP_COPY; break;
766 default: llvm_unreachable("invalid mapping mode");
767 }
768 Mapping = ::MapViewOfFile(FileMappingHandle,
769 dwDesiredAccess,
770 Offset >> 32,
771 Offset & 0xffffffff,
772 Size);
773 if (Mapping == NULL) {
774 error_code ec = windows_error(GetLastError());
775 ::CloseHandle(FileMappingHandle);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000776 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000777 if (CloseFD)
778 _close(FileDescriptor);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000779 } else
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000780 ::CloseHandle(FileHandle);
781 return ec;
782 }
783
784 if (Size == 0) {
785 MEMORY_BASIC_INFORMATION mbi;
786 SIZE_T Result = VirtualQuery(Mapping, &mbi, sizeof(mbi));
787 if (Result == 0) {
788 error_code ec = windows_error(GetLastError());
789 ::UnmapViewOfFile(Mapping);
790 ::CloseHandle(FileMappingHandle);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000791 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000792 if (CloseFD)
793 _close(FileDescriptor);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000794 } else
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000795 ::CloseHandle(FileHandle);
796 return ec;
797 }
798 Size = mbi.RegionSize;
799 }
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000800
801 // Close all the handles except for the view. It will keep the other handles
802 // alive.
803 ::CloseHandle(FileMappingHandle);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000804 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000805 if (CloseFD)
806 _close(FileDescriptor); // Also closes FileHandle.
Michael J. Spencerd932d412013-03-15 19:25:47 +0000807 } else
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000808 ::CloseHandle(FileHandle);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000809 return error_code::success();
810}
811
812mapped_file_region::mapped_file_region(const Twine &path,
813 mapmode mode,
814 uint64_t length,
815 uint64_t offset,
816 error_code &ec)
817 : Mode(mode)
818 , Size(length)
819 , Mapping()
820 , FileDescriptor()
821 , FileHandle(INVALID_HANDLE_VALUE)
822 , FileMappingHandle() {
823 SmallString<128> path_storage;
824 SmallVector<wchar_t, 128> path_utf16;
825
826 // Convert path to UTF-16.
Nico Webereada11b2012-09-25 05:24:16 +0000827 if ((ec = UTF8ToUTF16(path.toStringRef(path_storage), path_utf16)))
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000828 return;
829
830 // Get file handle for creating a file mapping.
831 FileHandle = ::CreateFileW(c_str(path_utf16),
832 Mode == readonly ? GENERIC_READ
833 : GENERIC_READ | GENERIC_WRITE,
834 Mode == readonly ? FILE_SHARE_READ
835 : 0,
836 0,
837 Mode == readonly ? OPEN_EXISTING
838 : OPEN_ALWAYS,
839 Mode == readonly ? FILE_ATTRIBUTE_READONLY
840 : FILE_ATTRIBUTE_NORMAL,
841 0);
842 if (FileHandle == INVALID_HANDLE_VALUE) {
843 ec = windows_error(::GetLastError());
844 return;
845 }
846
847 FileDescriptor = 0;
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000848 ec = init(FileDescriptor, true, offset);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000849 if (ec) {
850 Mapping = FileMappingHandle = 0;
851 FileHandle = INVALID_HANDLE_VALUE;
852 FileDescriptor = 0;
853 }
854}
855
856mapped_file_region::mapped_file_region(int fd,
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000857 bool closefd,
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000858 mapmode mode,
859 uint64_t length,
860 uint64_t offset,
861 error_code &ec)
862 : Mode(mode)
863 , Size(length)
864 , Mapping()
865 , FileDescriptor(fd)
866 , FileHandle(INVALID_HANDLE_VALUE)
867 , FileMappingHandle() {
868 FileHandle = reinterpret_cast<HANDLE>(_get_osfhandle(fd));
869 if (FileHandle == INVALID_HANDLE_VALUE) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000870 if (closefd)
871 _close(FileDescriptor);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000872 FileDescriptor = 0;
873 ec = make_error_code(errc::bad_file_descriptor);
874 return;
875 }
876
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000877 ec = init(FileDescriptor, closefd, offset);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000878 if (ec) {
879 Mapping = FileMappingHandle = 0;
880 FileHandle = INVALID_HANDLE_VALUE;
881 FileDescriptor = 0;
882 }
883}
884
885mapped_file_region::~mapped_file_region() {
886 if (Mapping)
887 ::UnmapViewOfFile(Mapping);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000888}
889
Chandler Carruthf12e3a62012-11-30 11:45:22 +0000890#if LLVM_HAS_RVALUE_REFERENCES
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000891mapped_file_region::mapped_file_region(mapped_file_region &&other)
892 : Mode(other.Mode)
893 , Size(other.Size)
894 , Mapping(other.Mapping)
895 , FileDescriptor(other.FileDescriptor)
896 , FileHandle(other.FileHandle)
897 , FileMappingHandle(other.FileMappingHandle) {
898 other.Mapping = other.FileMappingHandle = 0;
899 other.FileHandle = INVALID_HANDLE_VALUE;
900 other.FileDescriptor = 0;
901}
902#endif
903
904mapped_file_region::mapmode mapped_file_region::flags() const {
905 assert(Mapping && "Mapping failed but used anyway!");
906 return Mode;
907}
908
909uint64_t mapped_file_region::size() const {
910 assert(Mapping && "Mapping failed but used anyway!");
911 return Size;
912}
913
914char *mapped_file_region::data() const {
915 assert(Mode != readonly && "Cannot get non const data for readonly mapping!");
916 assert(Mapping && "Mapping failed but used anyway!");
917 return reinterpret_cast<char*>(Mapping);
918}
919
920const char *mapped_file_region::const_data() const {
921 assert(Mapping && "Mapping failed but used anyway!");
922 return reinterpret_cast<const char*>(Mapping);
923}
924
925int mapped_file_region::alignment() {
926 SYSTEM_INFO SysInfo;
927 ::GetSystemInfo(&SysInfo);
928 return SysInfo.dwAllocationGranularity;
929}
930
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000931error_code detail::directory_iterator_construct(detail::DirIterState &it,
932 StringRef path){
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000933 SmallVector<wchar_t, 128> path_utf16;
934
935 if (error_code ec = UTF8ToUTF16(path,
936 path_utf16))
937 return ec;
938
939 // Convert path to the format that Windows is happy with.
940 if (path_utf16.size() > 0 &&
941 !is_separator(path_utf16[path.size() - 1]) &&
942 path_utf16[path.size() - 1] != L':') {
943 path_utf16.push_back(L'\\');
944 path_utf16.push_back(L'*');
945 } else {
946 path_utf16.push_back(L'*');
947 }
948
949 // Get the first directory entry.
950 WIN32_FIND_DATAW FirstFind;
Michael J. Spencer751e9aa2010-12-09 17:37:18 +0000951 ScopedFindHandle FindHandle(::FindFirstFileW(c_str(path_utf16), &FirstFind));
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000952 if (!FindHandle)
953 return windows_error(::GetLastError());
954
Michael J. Spencer98879d72011-01-05 16:39:30 +0000955 size_t FilenameLen = ::wcslen(FirstFind.cFileName);
956 while ((FilenameLen == 1 && FirstFind.cFileName[0] == L'.') ||
957 (FilenameLen == 2 && FirstFind.cFileName[0] == L'.' &&
958 FirstFind.cFileName[1] == L'.'))
959 if (!::FindNextFileW(FindHandle, &FirstFind)) {
960 error_code ec = windows_error(::GetLastError());
961 // Check for end.
962 if (ec == windows_error::no_more_files)
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000963 return detail::directory_iterator_destruct(it);
Michael J. Spencer98879d72011-01-05 16:39:30 +0000964 return ec;
965 } else
966 FilenameLen = ::wcslen(FirstFind.cFileName);
967
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000968 // Construct the current directory entry.
Michael J. Spencer98879d72011-01-05 16:39:30 +0000969 SmallString<128> directory_entry_name_utf8;
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000970 if (error_code ec = UTF16ToUTF8(FirstFind.cFileName,
971 ::wcslen(FirstFind.cFileName),
Michael J. Spencer98879d72011-01-05 16:39:30 +0000972 directory_entry_name_utf8))
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000973 return ec;
974
975 it.IterationHandle = intptr_t(FindHandle.take());
Michael J. Spencer98879d72011-01-05 16:39:30 +0000976 SmallString<128> directory_entry_path(path);
977 path::append(directory_entry_path, directory_entry_name_utf8.str());
978 it.CurrentEntry = directory_entry(directory_entry_path.str());
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000979
David Blaikie18544b92012-02-09 19:24:12 +0000980 return error_code::success();
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000981}
982
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000983error_code detail::directory_iterator_destruct(detail::DirIterState &it) {
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000984 if (it.IterationHandle != 0)
985 // Closes the handle if it's valid.
986 ScopedFindHandle close(HANDLE(it.IterationHandle));
987 it.IterationHandle = 0;
988 it.CurrentEntry = directory_entry();
David Blaikie18544b92012-02-09 19:24:12 +0000989 return error_code::success();
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000990}
991
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000992error_code detail::directory_iterator_increment(detail::DirIterState &it) {
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000993 WIN32_FIND_DATAW FindData;
994 if (!::FindNextFileW(HANDLE(it.IterationHandle), &FindData)) {
995 error_code ec = windows_error(::GetLastError());
996 // Check for end.
997 if (ec == windows_error::no_more_files)
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000998 return detail::directory_iterator_destruct(it);
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000999 return ec;
1000 }
1001
Michael J. Spencer98879d72011-01-05 16:39:30 +00001002 size_t FilenameLen = ::wcslen(FindData.cFileName);
1003 if ((FilenameLen == 1 && FindData.cFileName[0] == L'.') ||
1004 (FilenameLen == 2 && FindData.cFileName[0] == L'.' &&
1005 FindData.cFileName[1] == L'.'))
1006 return directory_iterator_increment(it);
1007
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +00001008 SmallString<128> directory_entry_path_utf8;
1009 if (error_code ec = UTF16ToUTF8(FindData.cFileName,
1010 ::wcslen(FindData.cFileName),
1011 directory_entry_path_utf8))
1012 return ec;
1013
1014 it.CurrentEntry.replace_filename(Twine(directory_entry_path_utf8));
David Blaikie18544b92012-02-09 19:24:12 +00001015 return error_code::success();
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +00001016}
1017
Nick Kledzik18497e92012-06-20 00:28:54 +00001018error_code map_file_pages(const Twine &path, off_t file_offset, size_t size,
1019 bool map_writable, void *&result) {
1020 assert(0 && "NOT IMPLEMENTED");
Francois Pichet5dc987a2012-06-20 04:08:49 +00001021 return windows_error::invalid_function;
Nick Kledzik18497e92012-06-20 00:28:54 +00001022}
1023
1024error_code unmap_file_pages(void *base, size_t size) {
1025 assert(0 && "NOT IMPLEMENTED");
Francois Pichet5dc987a2012-06-20 04:08:49 +00001026 return windows_error::invalid_function;
Nick Kledzik18497e92012-06-20 00:28:54 +00001027}
1028
1029
1030
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +00001031} // end namespace fs
Michael J. Spencerebad2f92010-11-29 22:28:51 +00001032} // end namespace sys
1033} // end namespace llvm