blob: 94a501b39ac782a711b850a83b5d55482e288c82 [file] [log] [blame]
Rafael Espindolaf1fc3822013-06-26 19:33:03 +00001//===- llvm/Support/Windows/Path.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//
Rafael Espindolaf1fc3822013-06-26 19:33:03 +000010// This file implements the Windows specific implementation of the Path API.
Michael J. Spencerebad2f92010-11-29 22:28:51 +000011//
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
Craig Topperf18edae2013-07-15 07:15:05 +000019#include "llvm/ADT/STLExtras.h"
Michael J. Spencerebad2f92010-11-29 22:28:51 +000020#include "Windows.h"
Michael J. Spencerc20a0322010-12-03 17:54:07 +000021#include <fcntl.h>
Michael J. Spencer45710402010-12-03 01:21:28 +000022#include <io.h>
Michael J. Spencerc20a0322010-12-03 17:54:07 +000023#include <sys/stat.h>
24#include <sys/types.h>
Michael J. Spencerebad2f92010-11-29 22:28:51 +000025
Michael J. Spenceref2284f2012-08-15 19:05:47 +000026#undef max
27
Michael J. Spencer60252472010-12-03 18:03:28 +000028// MinGW doesn't define this.
Michael J. Spencer521c3212010-12-03 18:04:11 +000029#ifndef _ERRNO_T_DEFINED
30#define _ERRNO_T_DEFINED
31typedef int errno_t;
Michael J. Spencer60252472010-12-03 18:03:28 +000032#endif
33
Reid Kleckner11da0042013-08-07 20:19:31 +000034#ifdef _MSC_VER
35# pragma comment(lib, "advapi32.lib") // This provides CryptAcquireContextW.
36#endif
37
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000038using namespace llvm;
39
Rui Ueyama471d0c52013-09-10 19:45:51 +000040using llvm::sys::windows::UTF8ToUTF16;
41using llvm::sys::windows::UTF16ToUTF8;
42
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000043namespace {
Michael J. Spencer772caff52010-12-03 17:53:23 +000044 typedef BOOLEAN (WINAPI *PtrCreateSymbolicLinkW)(
45 /*__in*/ LPCWSTR lpSymlinkFileName,
46 /*__in*/ LPCWSTR lpTargetFileName,
47 /*__in*/ DWORD dwFlags);
48
49 PtrCreateSymbolicLinkW create_symbolic_link_api = PtrCreateSymbolicLinkW(
Michael J. Spencer7ee6d5d2010-12-03 07:41:25 +000050 ::GetProcAddress(::GetModuleHandleA("kernel32.dll"),
51 "CreateSymbolicLinkW"));
52
Michael J. Spencer45710402010-12-03 01:21:28 +000053 error_code TempDir(SmallVectorImpl<wchar_t> &result) {
54 retry_temp_dir:
55 DWORD len = ::GetTempPathW(result.capacity(), result.begin());
56
57 if (len == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +000058 return windows_error(::GetLastError());
Michael J. Spencer45710402010-12-03 01:21:28 +000059
60 if (len > result.capacity()) {
61 result.reserve(len);
62 goto retry_temp_dir;
63 }
64
65 result.set_size(len);
David Blaikie18544b92012-02-09 19:24:12 +000066 return error_code::success();
Michael J. Spencer45710402010-12-03 01:21:28 +000067 }
68
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +000069 bool is_separator(const wchar_t value) {
70 switch (value) {
71 case L'\\':
72 case L'/':
73 return true;
74 default:
75 return false;
76 }
77 }
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000078}
79
Rafael Espindolae79a8722013-06-28 03:48:47 +000080// FIXME: mode should be used here and default to user r/w only,
81// it currently comes in as a UNIX mode.
82static error_code createUniqueEntity(const Twine &model, int &result_fd,
83 SmallVectorImpl<char> &result_path,
84 bool makeAbsolute, unsigned mode,
85 FSEntity Type) {
86 // Use result_path as temp storage.
87 result_path.set_size(0);
88 StringRef m = model.toStringRef(result_path);
89
90 SmallVector<wchar_t, 128> model_utf16;
91 if (error_code ec = UTF8ToUTF16(m, model_utf16)) return ec;
92
93 if (makeAbsolute) {
94 // Make model absolute by prepending a temp directory if it's not already.
95 bool absolute = sys::path::is_absolute(m);
96
97 if (!absolute) {
98 SmallVector<wchar_t, 64> temp_dir;
99 if (error_code ec = TempDir(temp_dir)) return ec;
100 // Handle c: by removing it.
101 if (model_utf16.size() > 2 && model_utf16[1] == L':') {
102 model_utf16.erase(model_utf16.begin(), model_utf16.begin() + 2);
103 }
104 model_utf16.insert(model_utf16.begin(), temp_dir.begin(), temp_dir.end());
105 }
106 }
107
108 // Replace '%' with random chars. From here on, DO NOT modify model. It may be
109 // needed if the randomly chosen path already exists.
110 SmallVector<wchar_t, 128> random_path_utf16;
111
112 // Get a Crypto Provider for CryptGenRandom.
113 HCRYPTPROV HCPC;
114 if (!::CryptAcquireContextW(&HCPC,
115 NULL,
116 NULL,
117 PROV_RSA_FULL,
118 CRYPT_VERIFYCONTEXT))
119 return windows_error(::GetLastError());
120 ScopedCryptContext CryptoProvider(HCPC);
121
122retry_random_path:
123 random_path_utf16.set_size(0);
124 for (SmallVectorImpl<wchar_t>::const_iterator i = model_utf16.begin(),
125 e = model_utf16.end();
126 i != e; ++i) {
127 if (*i == L'%') {
128 BYTE val = 0;
129 if (!::CryptGenRandom(CryptoProvider, 1, &val))
130 return windows_error(::GetLastError());
131 random_path_utf16.push_back("0123456789abcdef"[val & 15]);
132 }
133 else
134 random_path_utf16.push_back(*i);
135 }
136 // Make random_path_utf16 null terminated.
137 random_path_utf16.push_back(0);
138 random_path_utf16.pop_back();
139
Serge Pavlov4c2a09d2013-07-19 16:23:54 +0000140 HANDLE TempFileHandle = INVALID_HANDLE_VALUE;
Rafael Espindolae79a8722013-06-28 03:48:47 +0000141
142 switch (Type) {
143 case FS_File: {
144 // Try to create + open the path.
145 TempFileHandle =
146 ::CreateFileW(random_path_utf16.begin(), GENERIC_READ | GENERIC_WRITE,
147 FILE_SHARE_READ, NULL,
148 // Return ERROR_FILE_EXISTS if the file
149 // already exists.
150 CREATE_NEW, FILE_ATTRIBUTE_TEMPORARY, NULL);
151 if (TempFileHandle == INVALID_HANDLE_VALUE) {
152 // If the file existed, try again, otherwise, error.
153 error_code ec = windows_error(::GetLastError());
154 if (ec == windows_error::file_exists)
155 goto retry_random_path;
156
157 return ec;
158 }
159
160 // Convert the Windows API file handle into a C-runtime handle.
161 int fd = ::_open_osfhandle(intptr_t(TempFileHandle), 0);
162 if (fd == -1) {
163 ::CloseHandle(TempFileHandle);
164 ::DeleteFileW(random_path_utf16.begin());
165 // MSDN doesn't say anything about _open_osfhandle setting errno or
166 // GetLastError(), so just return invalid_handle.
167 return windows_error::invalid_handle;
168 }
169
170 result_fd = fd;
171 break;
172 }
173
174 case FS_Name: {
175 DWORD attributes = ::GetFileAttributesW(random_path_utf16.begin());
176 if (attributes != INVALID_FILE_ATTRIBUTES)
177 goto retry_random_path;
178 error_code EC = make_error_code(windows_error(::GetLastError()));
179 if (EC != windows_error::file_not_found &&
180 EC != windows_error::path_not_found)
181 return EC;
182 break;
183 }
184
185 case FS_Dir:
186 if (!::CreateDirectoryW(random_path_utf16.begin(), NULL)) {
187 error_code EC = windows_error(::GetLastError());
188 if (EC != windows_error::already_exists)
189 return EC;
190 goto retry_random_path;
191 }
192 break;
193 }
194
195 // Set result_path to the utf-8 representation of the path.
196 if (error_code ec = UTF16ToUTF8(random_path_utf16.begin(),
197 random_path_utf16.size(), result_path)) {
198 switch (Type) {
199 case FS_File:
200 ::CloseHandle(TempFileHandle);
201 ::DeleteFileW(random_path_utf16.begin());
202 case FS_Name:
203 break;
204 case FS_Dir:
205 ::RemoveDirectoryW(random_path_utf16.begin());
206 break;
207 }
208 return ec;
209 }
210
211 return error_code::success();
212}
213
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000214namespace llvm {
215namespace sys {
Michael J. Spencer20daa282010-12-07 01:22:31 +0000216namespace fs {
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000217
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000218std::string getMainExecutable(const char *argv0, void *MainExecAddr) {
219 char pathname[MAX_PATH];
220 DWORD ret = ::GetModuleFileNameA(NULL, pathname, MAX_PATH);
221 return ret != MAX_PATH ? pathname : "";
222}
223
Rafael Espindolad1230992013-07-29 21:55:38 +0000224UniqueID file_status::getUniqueID() const {
Rafael Espindola7f822a92013-07-29 21:26:49 +0000225 // The file is uniquely identified by the volume serial number along
226 // with the 64-bit file identifier.
227 uint64_t FileID = (static_cast<uint64_t>(FileIndexHigh) << 32ULL) |
228 static_cast<uint64_t>(FileIndexLow);
229
230 return UniqueID(VolumeSerialNumber, FileID);
231}
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000232
Rafael Espindolabe3ede72013-06-20 21:51:49 +0000233TimeValue file_status::getLastModificationTime() const {
Rafael Espindoladb5d8fe2013-06-20 18:42:04 +0000234 ULARGE_INTEGER UI;
235 UI.LowPart = LastWriteTimeLow;
236 UI.HighPart = LastWriteTimeHigh;
237
238 TimeValue Ret;
239 Ret.fromWin32Time(UI.QuadPart);
240 return Ret;
241}
242
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000243error_code current_path(SmallVectorImpl<char> &result) {
244 SmallVector<wchar_t, 128> cur_path;
245 cur_path.reserve(128);
246retry_cur_dir:
247 DWORD len = ::GetCurrentDirectoryW(cur_path.capacity(), cur_path.data());
248
249 // A zero return value indicates a failure other than insufficient space.
250 if (len == 0)
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000251 return windows_error(::GetLastError());
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000252
253 // If there's insufficient space, the len returned is larger than the len
254 // given.
255 if (len > cur_path.capacity()) {
256 cur_path.reserve(len);
257 goto retry_cur_dir;
258 }
259
260 cur_path.set_size(len);
Aaron Ballmanb16cf532013-08-16 17:53:28 +0000261 return UTF16ToUTF8(cur_path.begin(), cur_path.size(), result);
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000262}
263
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000264error_code create_directory(const Twine &path, bool &existed) {
265 SmallString<128> path_storage;
266 SmallVector<wchar_t, 128> path_utf16;
267
268 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
269 path_utf16))
270 return ec;
271
272 if (!::CreateDirectoryW(path_utf16.begin(), NULL)) {
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000273 error_code ec = windows_error(::GetLastError());
274 if (ec == windows_error::already_exists)
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000275 existed = true;
276 else
277 return ec;
278 } else
279 existed = false;
280
David Blaikie18544b92012-02-09 19:24:12 +0000281 return error_code::success();
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000282}
283
Michael J. Spencere0c45602010-12-03 05:58:41 +0000284error_code create_hard_link(const Twine &to, const Twine &from) {
285 // Get arguments.
286 SmallString<128> from_storage;
287 SmallString<128> to_storage;
288 StringRef f = from.toStringRef(from_storage);
289 StringRef t = to.toStringRef(to_storage);
290
291 // Convert to utf-16.
292 SmallVector<wchar_t, 128> wide_from;
293 SmallVector<wchar_t, 128> wide_to;
294 if (error_code ec = UTF8ToUTF16(f, wide_from)) return ec;
295 if (error_code ec = UTF8ToUTF16(t, wide_to)) return ec;
296
297 if (!::CreateHardLinkW(wide_from.begin(), wide_to.begin(), NULL))
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000298 return windows_error(::GetLastError());
Michael J. Spencere0c45602010-12-03 05:58:41 +0000299
David Blaikie18544b92012-02-09 19:24:12 +0000300 return error_code::success();
Michael J. Spencere0c45602010-12-03 05:58:41 +0000301}
302
Michael J. Spencer7ee6d5d2010-12-03 07:41:25 +0000303error_code create_symlink(const Twine &to, const Twine &from) {
304 // Only do it if the function is available at runtime.
305 if (!create_symbolic_link_api)
306 return make_error_code(errc::function_not_supported);
307
308 // Get arguments.
309 SmallString<128> from_storage;
310 SmallString<128> to_storage;
311 StringRef f = from.toStringRef(from_storage);
312 StringRef t = to.toStringRef(to_storage);
313
314 // Convert to utf-16.
315 SmallVector<wchar_t, 128> wide_from;
316 SmallVector<wchar_t, 128> wide_to;
317 if (error_code ec = UTF8ToUTF16(f, wide_from)) return ec;
318 if (error_code ec = UTF8ToUTF16(t, wide_to)) return ec;
319
Michael J. Spencer962fbc42010-12-04 00:32:14 +0000320 if (!create_symbolic_link_api(wide_from.begin(), wide_to.begin(), 0))
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000321 return windows_error(::GetLastError());
Michael J. Spencer7ee6d5d2010-12-03 07:41:25 +0000322
David Blaikie18544b92012-02-09 19:24:12 +0000323 return error_code::success();
Michael J. Spencer7ee6d5d2010-12-03 07:41:25 +0000324}
325
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000326error_code remove(const Twine &path, bool &existed) {
327 SmallString<128> path_storage;
328 SmallVector<wchar_t, 128> path_utf16;
329
Michael J. Spencer153749b2011-01-05 16:39:22 +0000330 file_status st;
Rafael Espindola107b74c2013-07-31 00:10:25 +0000331 error_code EC = status(path, st);
332 if (EC) {
333 if (EC == windows_error::file_not_found ||
334 EC == windows_error::path_not_found) {
335 existed = false;
336 return error_code::success();
337 }
338 return EC;
339 }
Michael J. Spencer153749b2011-01-05 16:39:22 +0000340
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000341 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
342 path_utf16))
343 return ec;
344
Michael J. Spencer153749b2011-01-05 16:39:22 +0000345 if (st.type() == file_type::directory_file) {
346 if (!::RemoveDirectoryW(c_str(path_utf16))) {
347 error_code ec = windows_error(::GetLastError());
348 if (ec != windows_error::file_not_found)
349 return ec;
350 existed = false;
351 } else
352 existed = true;
353 } else {
354 if (!::DeleteFileW(c_str(path_utf16))) {
355 error_code ec = windows_error(::GetLastError());
356 if (ec != windows_error::file_not_found)
357 return ec;
358 existed = false;
359 } else
360 existed = true;
361 }
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000362
David Blaikie18544b92012-02-09 19:24:12 +0000363 return error_code::success();
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000364}
365
Michael J. Spencer409f5562010-12-03 17:53:55 +0000366error_code rename(const Twine &from, const Twine &to) {
367 // Get arguments.
368 SmallString<128> from_storage;
369 SmallString<128> to_storage;
370 StringRef f = from.toStringRef(from_storage);
371 StringRef t = to.toStringRef(to_storage);
372
373 // Convert to utf-16.
374 SmallVector<wchar_t, 128> wide_from;
375 SmallVector<wchar_t, 128> wide_to;
376 if (error_code ec = UTF8ToUTF16(f, wide_from)) return ec;
377 if (error_code ec = UTF8ToUTF16(t, wide_to)) return ec;
378
NAKAMURA Takumi3b7f9952012-05-08 14:31:46 +0000379 error_code ec = error_code::success();
380 for (int i = 0; i < 2000; i++) {
381 if (::MoveFileExW(wide_from.begin(), wide_to.begin(),
382 MOVEFILE_COPY_ALLOWED | MOVEFILE_REPLACE_EXISTING))
383 return error_code::success();
384 ec = windows_error(::GetLastError());
385 if (ec != windows_error::access_denied)
386 break;
387 // Retry MoveFile() at ACCESS_DENIED.
388 // System scanners (eg. indexer) might open the source file when
389 // It is written and closed.
390 ::Sleep(1);
391 }
Michael J. Spencer409f5562010-12-03 17:53:55 +0000392
NAKAMURA Takumi3b7f9952012-05-08 14:31:46 +0000393 return ec;
Michael J. Spencer409f5562010-12-03 17:53:55 +0000394}
395
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000396error_code resize_file(const Twine &path, uint64_t size) {
397 SmallString<128> path_storage;
398 SmallVector<wchar_t, 128> path_utf16;
399
400 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
401 path_utf16))
402 return ec;
403
Michael J. Spencer20abb202012-12-03 22:09:31 +0000404 int fd = ::_wopen(path_utf16.begin(), O_BINARY | _O_RDWR, S_IWRITE);
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000405 if (fd == -1)
406 return error_code(errno, generic_category());
Michael J. Spencerca242f22010-12-03 18:48:56 +0000407#ifdef HAVE__CHSIZE_S
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000408 errno_t error = ::_chsize_s(fd, size);
Michael J. Spencerca242f22010-12-03 18:48:56 +0000409#else
410 errno_t error = ::_chsize(fd, size);
411#endif
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000412 ::close(fd);
413 return error_code(error, generic_category());
414}
415
Michael J. Spencer45710402010-12-03 01:21:28 +0000416error_code exists(const Twine &path, bool &result) {
417 SmallString<128> path_storage;
418 SmallVector<wchar_t, 128> path_utf16;
419
420 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
421 path_utf16))
422 return ec;
423
424 DWORD attributes = ::GetFileAttributesW(path_utf16.begin());
425
426 if (attributes == INVALID_FILE_ATTRIBUTES) {
427 // See if the file didn't actually exist.
428 error_code ec = make_error_code(windows_error(::GetLastError()));
Michael J. Spencer66a1f862010-12-04 18:45:32 +0000429 if (ec != windows_error::file_not_found &&
430 ec != windows_error::path_not_found)
Michael J. Spencer45710402010-12-03 01:21:28 +0000431 return ec;
432 result = false;
433 } else
434 result = true;
David Blaikie18544b92012-02-09 19:24:12 +0000435 return error_code::success();
Michael J. Spencer45710402010-12-03 01:21:28 +0000436}
437
Rafael Espindolaa1280c12013-06-18 20:56:38 +0000438bool can_write(const Twine &Path) {
439 // FIXME: take security attributes into account.
440 SmallString<128> PathStorage;
441 SmallVector<wchar_t, 128> PathUtf16;
442
443 if (UTF8ToUTF16(Path.toStringRef(PathStorage), PathUtf16))
444 return false;
445
446 DWORD Attr = ::GetFileAttributesW(PathUtf16.begin());
447 return (Attr != INVALID_FILE_ATTRIBUTES) && !(Attr & FILE_ATTRIBUTE_READONLY);
448}
449
Rafael Espindolab0a5c962013-06-14 19:38:45 +0000450bool can_execute(const Twine &Path) {
451 SmallString<128> PathStorage;
452 SmallVector<wchar_t, 128> PathUtf16;
453
454 if (UTF8ToUTF16(Path.toStringRef(PathStorage), PathUtf16))
455 return false;
456
457 DWORD Attr = ::GetFileAttributesW(PathUtf16.begin());
458 return Attr != INVALID_FILE_ATTRIBUTES;
459}
460
Michael J. Spencer203d7802011-12-12 06:04:28 +0000461bool equivalent(file_status A, file_status B) {
462 assert(status_known(A) && status_known(B));
463 return A.FileIndexHigh == B.FileIndexHigh &&
464 A.FileIndexLow == B.FileIndexLow &&
465 A.FileSizeHigh == B.FileSizeHigh &&
466 A.FileSizeLow == B.FileSizeLow &&
467 A.LastWriteTimeHigh == B.LastWriteTimeHigh &&
468 A.LastWriteTimeLow == B.LastWriteTimeLow &&
469 A.VolumeSerialNumber == B.VolumeSerialNumber;
470}
471
Michael J. Spencer376d3872010-12-03 18:49:13 +0000472error_code equivalent(const Twine &A, const Twine &B, bool &result) {
Michael J. Spencer203d7802011-12-12 06:04:28 +0000473 file_status fsA, fsB;
474 if (error_code ec = status(A, fsA)) return ec;
475 if (error_code ec = status(B, fsB)) return ec;
476 result = equivalent(fsA, fsB);
David Blaikie18544b92012-02-09 19:24:12 +0000477 return error_code::success();
Michael J. Spencer376d3872010-12-03 18:49:13 +0000478}
479
Benjamin Kramer58994ec2011-08-20 21:36:38 +0000480static bool isReservedName(StringRef path) {
481 // This list of reserved names comes from MSDN, at:
482 // http://msdn.microsoft.com/en-us/library/aa365247%28v=vs.85%29.aspx
483 static const char *sReservedNames[] = { "nul", "con", "prn", "aux",
484 "com1", "com2", "com3", "com4", "com5", "com6",
485 "com7", "com8", "com9", "lpt1", "lpt2", "lpt3",
486 "lpt4", "lpt5", "lpt6", "lpt7", "lpt8", "lpt9" };
487
488 // First, check to see if this is a device namespace, which always
489 // starts with \\.\, since device namespaces are not legal file paths.
490 if (path.startswith("\\\\.\\"))
491 return true;
492
493 // Then compare against the list of ancient reserved names
Craig Topper58713212013-07-15 04:27:47 +0000494 for (size_t i = 0; i < array_lengthof(sReservedNames); ++i) {
Benjamin Kramer58994ec2011-08-20 21:36:38 +0000495 if (path.equals_lower(sReservedNames[i]))
496 return true;
497 }
498
499 // The path isn't what we consider reserved.
500 return false;
501}
502
Rafael Espindola77021c92013-07-16 03:20:13 +0000503static error_code getStatus(HANDLE FileHandle, file_status &Result) {
504 if (FileHandle == INVALID_HANDLE_VALUE)
505 goto handle_status_error;
506
NAKAMURA Takumi8b01da42013-07-18 17:00:54 +0000507 switch (::GetFileType(FileHandle)) {
508 default:
Rafael Espindola81177c52013-07-18 18:42:52 +0000509 llvm_unreachable("Don't know anything about this file type");
510 case FILE_TYPE_UNKNOWN: {
511 DWORD Err = ::GetLastError();
512 if (Err != NO_ERROR)
513 return windows_error(Err);
514 Result = file_status(file_type::type_unknown);
515 return error_code::success();
516 }
NAKAMURA Takumi8b01da42013-07-18 17:00:54 +0000517 case FILE_TYPE_DISK:
518 break;
519 case FILE_TYPE_CHAR:
520 Result = file_status(file_type::character_file);
521 return error_code::success();
522 case FILE_TYPE_PIPE:
523 Result = file_status(file_type::fifo_file);
524 return error_code::success();
525 }
526
Rafael Espindola77021c92013-07-16 03:20:13 +0000527 BY_HANDLE_FILE_INFORMATION Info;
528 if (!::GetFileInformationByHandle(FileHandle, &Info))
529 goto handle_status_error;
530
Rafael Espindolaa5932af2013-07-30 20:25:53 +0000531 {
532 file_type Type = (Info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
533 ? file_type::directory_file
534 : file_type::regular_file;
535 Result =
536 file_status(Type, Info.ftLastWriteTime.dwHighDateTime,
537 Info.ftLastWriteTime.dwLowDateTime,
538 Info.dwVolumeSerialNumber, Info.nFileSizeHigh,
539 Info.nFileSizeLow, Info.nFileIndexHigh, Info.nFileIndexLow);
540 return error_code::success();
541 }
Rafael Espindola77021c92013-07-16 03:20:13 +0000542
543handle_status_error:
544 error_code EC = windows_error(::GetLastError());
545 if (EC == windows_error::file_not_found ||
546 EC == windows_error::path_not_found)
547 Result = file_status(file_type::file_not_found);
548 else if (EC == windows_error::sharing_violation)
549 Result = file_status(file_type::type_unknown);
Rafael Espindola107b74c2013-07-31 00:10:25 +0000550 else
Rafael Espindola77021c92013-07-16 03:20:13 +0000551 Result = file_status(file_type::status_error);
Rafael Espindola107b74c2013-07-31 00:10:25 +0000552 return EC;
Rafael Espindola77021c92013-07-16 03:20:13 +0000553}
554
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000555error_code status(const Twine &path, file_status &result) {
556 SmallString<128> path_storage;
557 SmallVector<wchar_t, 128> path_utf16;
558
NAKAMURA Takumia3d47492011-03-16 02:53:32 +0000559 StringRef path8 = path.toStringRef(path_storage);
Benjamin Kramer58994ec2011-08-20 21:36:38 +0000560 if (isReservedName(path8)) {
NAKAMURA Takumia3d47492011-03-16 02:53:32 +0000561 result = file_status(file_type::character_file);
David Blaikie18544b92012-02-09 19:24:12 +0000562 return error_code::success();
NAKAMURA Takumia3d47492011-03-16 02:53:32 +0000563 }
564
Michael J. Spencer203d7802011-12-12 06:04:28 +0000565 if (error_code ec = UTF8ToUTF16(path8, path_utf16))
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000566 return ec;
567
568 DWORD attr = ::GetFileAttributesW(path_utf16.begin());
569 if (attr == INVALID_FILE_ATTRIBUTES)
Rafael Espindola77021c92013-07-16 03:20:13 +0000570 return getStatus(INVALID_HANDLE_VALUE, result);
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000571
572 // Handle reparse points.
573 if (attr & FILE_ATTRIBUTE_REPARSE_POINT) {
Michael J. Spencer513f1b62011-12-12 06:03:33 +0000574 ScopedFileHandle h(
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000575 ::CreateFileW(path_utf16.begin(),
576 0, // Attributes only.
577 FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
578 NULL,
579 OPEN_EXISTING,
580 FILE_FLAG_BACKUP_SEMANTICS,
581 0));
Michael J. Spencer513f1b62011-12-12 06:03:33 +0000582 if (!h)
Rafael Espindola77021c92013-07-16 03:20:13 +0000583 return getStatus(INVALID_HANDLE_VALUE, result);
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000584 }
585
Rafael Espindolaa5932af2013-07-30 20:25:53 +0000586 ScopedFileHandle h(
587 ::CreateFileW(path_utf16.begin(), 0, // Attributes only.
Michael J. Spencer203d7802011-12-12 06:04:28 +0000588 FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
Rafael Espindolaa5932af2013-07-30 20:25:53 +0000589 NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, 0));
Michael J. Spencer203d7802011-12-12 06:04:28 +0000590 if (!h)
Rafael Espindola77021c92013-07-16 03:20:13 +0000591 return getStatus(INVALID_HANDLE_VALUE, result);
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000592
Rafael Espindola77021c92013-07-16 03:20:13 +0000593 return getStatus(h, result);
Rafael Espindola77021c92013-07-16 03:20:13 +0000594}
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000595
Rafael Espindola77021c92013-07-16 03:20:13 +0000596error_code status(int FD, file_status &Result) {
597 HANDLE FileHandle = reinterpret_cast<HANDLE>(_get_osfhandle(FD));
598 return getStatus(FileHandle, Result);
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000599}
600
Rafael Espindola4a3365c2013-06-20 20:56:14 +0000601error_code setLastModificationAndAccessTime(int FD, TimeValue Time) {
602 ULARGE_INTEGER UI;
603 UI.QuadPart = Time.toWin32Time();
604 FILETIME FT;
605 FT.dwLowDateTime = UI.LowPart;
606 FT.dwHighDateTime = UI.HighPart;
607 HANDLE FileHandle = reinterpret_cast<HANDLE>(_get_osfhandle(FD));
608 if (!SetFileTime(FileHandle, NULL, &FT, &FT))
609 return windows_error(::GetLastError());
610 return error_code::success();
611}
Nick Kledzik18497e92012-06-20 00:28:54 +0000612
Michael J. Spenceree1699c2011-01-15 18:52:33 +0000613error_code get_magic(const Twine &path, uint32_t len,
614 SmallVectorImpl<char> &result) {
615 SmallString<128> path_storage;
616 SmallVector<wchar_t, 128> path_utf16;
617 result.set_size(0);
618
619 // Convert path to UTF-16.
620 if (error_code ec = UTF8ToUTF16(path.toStringRef(path_storage),
621 path_utf16))
622 return ec;
623
624 // Open file.
625 HANDLE file = ::CreateFileW(c_str(path_utf16),
626 GENERIC_READ,
627 FILE_SHARE_READ,
628 NULL,
629 OPEN_EXISTING,
630 FILE_ATTRIBUTE_READONLY,
631 NULL);
632 if (file == INVALID_HANDLE_VALUE)
633 return windows_error(::GetLastError());
634
635 // Allocate buffer.
636 result.reserve(len);
637
638 // Get magic!
639 DWORD bytes_read = 0;
640 BOOL read_success = ::ReadFile(file, result.data(), len, &bytes_read, NULL);
641 error_code ec = windows_error(::GetLastError());
642 ::CloseHandle(file);
643 if (!read_success || (bytes_read != len)) {
644 // Set result size to the number of bytes read if it's valid.
NAKAMURA Takumi3e76b502011-03-07 00:12:53 +0000645 if (bytes_read <= len)
Michael J. Spenceree1699c2011-01-15 18:52:33 +0000646 result.set_size(bytes_read);
647 // ERROR_HANDLE_EOF is mapped to errc::value_too_large.
648 return ec;
649 }
650
651 result.set_size(len);
David Blaikie18544b92012-02-09 19:24:12 +0000652 return error_code::success();
Michael J. Spenceree1699c2011-01-15 18:52:33 +0000653}
654
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000655error_code mapped_file_region::init(int FD, bool CloseFD, uint64_t Offset) {
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000656 FileDescriptor = FD;
657 // Make sure that the requested size fits within SIZE_T.
658 if (Size > std::numeric_limits<SIZE_T>::max()) {
Michael J. Spencerd932d412013-03-15 19:25:47 +0000659 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000660 if (CloseFD)
661 _close(FileDescriptor);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000662 } else
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000663 ::CloseHandle(FileHandle);
664 return make_error_code(errc::invalid_argument);
665 }
666
667 DWORD flprotect;
668 switch (Mode) {
669 case readonly: flprotect = PAGE_READONLY; break;
670 case readwrite: flprotect = PAGE_READWRITE; break;
671 case priv: flprotect = PAGE_WRITECOPY; break;
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000672 }
673
674 FileMappingHandle = ::CreateFileMapping(FileHandle,
675 0,
676 flprotect,
NAKAMURA Takumi7b5d4f92013-08-22 15:14:53 +0000677 (Offset + Size) >> 32,
678 (Offset + Size) & 0xffffffff,
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000679 0);
680 if (FileMappingHandle == NULL) {
681 error_code ec = windows_error(GetLastError());
Michael J. Spencerd932d412013-03-15 19:25:47 +0000682 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000683 if (CloseFD)
684 _close(FileDescriptor);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000685 } else
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000686 ::CloseHandle(FileHandle);
687 return ec;
688 }
689
690 DWORD dwDesiredAccess;
691 switch (Mode) {
692 case readonly: dwDesiredAccess = FILE_MAP_READ; break;
693 case readwrite: dwDesiredAccess = FILE_MAP_WRITE; break;
694 case priv: dwDesiredAccess = FILE_MAP_COPY; break;
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000695 }
696 Mapping = ::MapViewOfFile(FileMappingHandle,
697 dwDesiredAccess,
698 Offset >> 32,
699 Offset & 0xffffffff,
700 Size);
701 if (Mapping == NULL) {
702 error_code ec = windows_error(GetLastError());
703 ::CloseHandle(FileMappingHandle);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000704 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000705 if (CloseFD)
706 _close(FileDescriptor);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000707 } else
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000708 ::CloseHandle(FileHandle);
709 return ec;
710 }
711
712 if (Size == 0) {
713 MEMORY_BASIC_INFORMATION mbi;
714 SIZE_T Result = VirtualQuery(Mapping, &mbi, sizeof(mbi));
715 if (Result == 0) {
716 error_code ec = windows_error(GetLastError());
717 ::UnmapViewOfFile(Mapping);
718 ::CloseHandle(FileMappingHandle);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000719 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000720 if (CloseFD)
721 _close(FileDescriptor);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000722 } else
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000723 ::CloseHandle(FileHandle);
724 return ec;
725 }
726 Size = mbi.RegionSize;
727 }
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000728
729 // Close all the handles except for the view. It will keep the other handles
730 // alive.
731 ::CloseHandle(FileMappingHandle);
Michael J. Spencerd932d412013-03-15 19:25:47 +0000732 if (FileDescriptor) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000733 if (CloseFD)
734 _close(FileDescriptor); // Also closes FileHandle.
Michael J. Spencerd932d412013-03-15 19:25:47 +0000735 } else
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000736 ::CloseHandle(FileHandle);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000737 return error_code::success();
738}
739
740mapped_file_region::mapped_file_region(const Twine &path,
741 mapmode mode,
742 uint64_t length,
743 uint64_t offset,
NAKAMURA Takumiedf76152013-08-22 15:14:45 +0000744 error_code &ec)
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000745 : Mode(mode)
746 , Size(length)
747 , Mapping()
748 , FileDescriptor()
749 , FileHandle(INVALID_HANDLE_VALUE)
750 , FileMappingHandle() {
751 SmallString<128> path_storage;
752 SmallVector<wchar_t, 128> path_utf16;
753
754 // Convert path to UTF-16.
Nico Webereada11b2012-09-25 05:24:16 +0000755 if ((ec = UTF8ToUTF16(path.toStringRef(path_storage), path_utf16)))
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000756 return;
757
758 // Get file handle for creating a file mapping.
759 FileHandle = ::CreateFileW(c_str(path_utf16),
760 Mode == readonly ? GENERIC_READ
761 : GENERIC_READ | GENERIC_WRITE,
762 Mode == readonly ? FILE_SHARE_READ
763 : 0,
764 0,
765 Mode == readonly ? OPEN_EXISTING
766 : OPEN_ALWAYS,
767 Mode == readonly ? FILE_ATTRIBUTE_READONLY
768 : FILE_ATTRIBUTE_NORMAL,
769 0);
770 if (FileHandle == INVALID_HANDLE_VALUE) {
771 ec = windows_error(::GetLastError());
772 return;
773 }
774
775 FileDescriptor = 0;
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000776 ec = init(FileDescriptor, true, offset);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000777 if (ec) {
778 Mapping = FileMappingHandle = 0;
779 FileHandle = INVALID_HANDLE_VALUE;
780 FileDescriptor = 0;
781 }
782}
783
784mapped_file_region::mapped_file_region(int fd,
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000785 bool closefd,
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000786 mapmode mode,
787 uint64_t length,
788 uint64_t offset,
789 error_code &ec)
790 : Mode(mode)
791 , Size(length)
792 , Mapping()
793 , FileDescriptor(fd)
794 , FileHandle(INVALID_HANDLE_VALUE)
795 , FileMappingHandle() {
796 FileHandle = reinterpret_cast<HANDLE>(_get_osfhandle(fd));
797 if (FileHandle == INVALID_HANDLE_VALUE) {
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000798 if (closefd)
799 _close(FileDescriptor);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000800 FileDescriptor = 0;
801 ec = make_error_code(errc::bad_file_descriptor);
802 return;
803 }
804
Michael J. Spencer42ad29f2013-03-14 00:20:10 +0000805 ec = init(FileDescriptor, closefd, offset);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000806 if (ec) {
807 Mapping = FileMappingHandle = 0;
808 FileHandle = INVALID_HANDLE_VALUE;
809 FileDescriptor = 0;
810 }
811}
812
813mapped_file_region::~mapped_file_region() {
814 if (Mapping)
815 ::UnmapViewOfFile(Mapping);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000816}
817
Chandler Carruthf12e3a62012-11-30 11:45:22 +0000818#if LLVM_HAS_RVALUE_REFERENCES
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000819mapped_file_region::mapped_file_region(mapped_file_region &&other)
820 : Mode(other.Mode)
821 , Size(other.Size)
822 , Mapping(other.Mapping)
823 , FileDescriptor(other.FileDescriptor)
824 , FileHandle(other.FileHandle)
825 , FileMappingHandle(other.FileMappingHandle) {
826 other.Mapping = other.FileMappingHandle = 0;
827 other.FileHandle = INVALID_HANDLE_VALUE;
828 other.FileDescriptor = 0;
829}
830#endif
831
832mapped_file_region::mapmode mapped_file_region::flags() const {
833 assert(Mapping && "Mapping failed but used anyway!");
834 return Mode;
835}
836
837uint64_t mapped_file_region::size() const {
838 assert(Mapping && "Mapping failed but used anyway!");
839 return Size;
840}
841
842char *mapped_file_region::data() const {
843 assert(Mode != readonly && "Cannot get non const data for readonly mapping!");
844 assert(Mapping && "Mapping failed but used anyway!");
845 return reinterpret_cast<char*>(Mapping);
846}
847
848const char *mapped_file_region::const_data() const {
849 assert(Mapping && "Mapping failed but used anyway!");
850 return reinterpret_cast<const char*>(Mapping);
851}
852
853int mapped_file_region::alignment() {
854 SYSTEM_INFO SysInfo;
855 ::GetSystemInfo(&SysInfo);
856 return SysInfo.dwAllocationGranularity;
857}
858
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000859error_code detail::directory_iterator_construct(detail::DirIterState &it,
860 StringRef path){
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000861 SmallVector<wchar_t, 128> path_utf16;
862
863 if (error_code ec = UTF8ToUTF16(path,
864 path_utf16))
865 return ec;
866
867 // Convert path to the format that Windows is happy with.
868 if (path_utf16.size() > 0 &&
869 !is_separator(path_utf16[path.size() - 1]) &&
870 path_utf16[path.size() - 1] != L':') {
871 path_utf16.push_back(L'\\');
872 path_utf16.push_back(L'*');
873 } else {
874 path_utf16.push_back(L'*');
875 }
876
877 // Get the first directory entry.
878 WIN32_FIND_DATAW FirstFind;
Michael J. Spencer751e9aa2010-12-09 17:37:18 +0000879 ScopedFindHandle FindHandle(::FindFirstFileW(c_str(path_utf16), &FirstFind));
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000880 if (!FindHandle)
881 return windows_error(::GetLastError());
882
Michael J. Spencer98879d72011-01-05 16:39:30 +0000883 size_t FilenameLen = ::wcslen(FirstFind.cFileName);
884 while ((FilenameLen == 1 && FirstFind.cFileName[0] == L'.') ||
885 (FilenameLen == 2 && FirstFind.cFileName[0] == L'.' &&
886 FirstFind.cFileName[1] == L'.'))
887 if (!::FindNextFileW(FindHandle, &FirstFind)) {
888 error_code ec = windows_error(::GetLastError());
889 // Check for end.
890 if (ec == windows_error::no_more_files)
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000891 return detail::directory_iterator_destruct(it);
Michael J. Spencer98879d72011-01-05 16:39:30 +0000892 return ec;
893 } else
894 FilenameLen = ::wcslen(FirstFind.cFileName);
895
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000896 // Construct the current directory entry.
Michael J. Spencer98879d72011-01-05 16:39:30 +0000897 SmallString<128> directory_entry_name_utf8;
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000898 if (error_code ec = UTF16ToUTF8(FirstFind.cFileName,
899 ::wcslen(FirstFind.cFileName),
Michael J. Spencer98879d72011-01-05 16:39:30 +0000900 directory_entry_name_utf8))
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000901 return ec;
902
903 it.IterationHandle = intptr_t(FindHandle.take());
Michael J. Spencer98879d72011-01-05 16:39:30 +0000904 SmallString<128> directory_entry_path(path);
905 path::append(directory_entry_path, directory_entry_name_utf8.str());
906 it.CurrentEntry = directory_entry(directory_entry_path.str());
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000907
David Blaikie18544b92012-02-09 19:24:12 +0000908 return error_code::success();
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000909}
910
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000911error_code detail::directory_iterator_destruct(detail::DirIterState &it) {
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000912 if (it.IterationHandle != 0)
913 // Closes the handle if it's valid.
914 ScopedFindHandle close(HANDLE(it.IterationHandle));
915 it.IterationHandle = 0;
916 it.CurrentEntry = directory_entry();
David Blaikie18544b92012-02-09 19:24:12 +0000917 return error_code::success();
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000918}
919
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000920error_code detail::directory_iterator_increment(detail::DirIterState &it) {
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000921 WIN32_FIND_DATAW FindData;
922 if (!::FindNextFileW(HANDLE(it.IterationHandle), &FindData)) {
923 error_code ec = windows_error(::GetLastError());
924 // Check for end.
925 if (ec == windows_error::no_more_files)
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000926 return detail::directory_iterator_destruct(it);
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000927 return ec;
928 }
929
Michael J. Spencer98879d72011-01-05 16:39:30 +0000930 size_t FilenameLen = ::wcslen(FindData.cFileName);
931 if ((FilenameLen == 1 && FindData.cFileName[0] == L'.') ||
932 (FilenameLen == 2 && FindData.cFileName[0] == L'.' &&
933 FindData.cFileName[1] == L'.'))
934 return directory_iterator_increment(it);
935
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000936 SmallString<128> directory_entry_path_utf8;
937 if (error_code ec = UTF16ToUTF8(FindData.cFileName,
938 ::wcslen(FindData.cFileName),
939 directory_entry_path_utf8))
940 return ec;
941
942 it.CurrentEntry.replace_filename(Twine(directory_entry_path_utf8));
David Blaikie18544b92012-02-09 19:24:12 +0000943 return error_code::success();
Michael J. Spencer7ecd94c2010-12-06 04:28:42 +0000944}
945
NAKAMURA Takumiedf76152013-08-22 15:14:45 +0000946error_code map_file_pages(const Twine &path, off_t file_offset, size_t size,
Nick Kledzik18497e92012-06-20 00:28:54 +0000947 bool map_writable, void *&result) {
948 assert(0 && "NOT IMPLEMENTED");
Francois Pichet5dc987a2012-06-20 04:08:49 +0000949 return windows_error::invalid_function;
Nick Kledzik18497e92012-06-20 00:28:54 +0000950}
951
952error_code unmap_file_pages(void *base, size_t size) {
953 assert(0 && "NOT IMPLEMENTED");
Francois Pichet5dc987a2012-06-20 04:08:49 +0000954 return windows_error::invalid_function;
Nick Kledzik18497e92012-06-20 00:28:54 +0000955}
956
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000957error_code openFileForRead(const Twine &Name, int &ResultFD) {
958 SmallString<128> PathStorage;
959 SmallVector<wchar_t, 128> PathUTF16;
Nick Kledzik18497e92012-06-20 00:28:54 +0000960
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000961 if (error_code EC = UTF8ToUTF16(Name.toStringRef(PathStorage),
962 PathUTF16))
963 return EC;
964
965 HANDLE H = ::CreateFileW(PathUTF16.begin(), GENERIC_READ,
Rafael Espindola16431fe2013-07-17 19:44:07 +0000966 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL,
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000967 OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
968 if (H == INVALID_HANDLE_VALUE) {
969 error_code EC = windows_error(::GetLastError());
Rafael Espindola331aeba2013-07-17 19:58:28 +0000970 // Provide a better error message when trying to open directories.
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000971 // This only runs if we failed to open the file, so there is probably
972 // no performances issues.
973 if (EC != windows_error::access_denied)
974 return EC;
975 if (is_directory(Name))
976 return error_code(errc::is_a_directory, posix_category());
977 return EC;
978 }
979
980 int FD = ::_open_osfhandle(intptr_t(H), 0);
981 if (FD == -1) {
982 ::CloseHandle(H);
983 return windows_error::invalid_handle;
984 }
985
986 ResultFD = FD;
987 return error_code::success();
988}
Nick Kledzik18497e92012-06-20 00:28:54 +0000989
Rafael Espindola67080ce2013-07-19 15:02:03 +0000990error_code openFileForWrite(const Twine &Name, int &ResultFD,
991 sys::fs::OpenFlags Flags, unsigned Mode) {
992 // Verify that we don't have both "append" and "excl".
993 assert((!(Flags & sys::fs::F_Excl) || !(Flags & sys::fs::F_Append)) &&
994 "Cannot specify both 'excl' and 'append' file creation flags!");
995
996 SmallString<128> PathStorage;
997 SmallVector<wchar_t, 128> PathUTF16;
998
999 if (error_code EC = UTF8ToUTF16(Name.toStringRef(PathStorage),
1000 PathUTF16))
1001 return EC;
1002
1003 DWORD CreationDisposition;
1004 if (Flags & F_Excl)
1005 CreationDisposition = CREATE_NEW;
NAKAMURA Takumiedf76152013-08-22 15:14:45 +00001006 else if (Flags & F_Append)
Rafael Espindola67080ce2013-07-19 15:02:03 +00001007 CreationDisposition = OPEN_ALWAYS;
1008 else
1009 CreationDisposition = CREATE_ALWAYS;
1010
1011 HANDLE H = ::CreateFileW(PathUTF16.begin(), GENERIC_WRITE,
1012 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL,
1013 CreationDisposition, FILE_ATTRIBUTE_NORMAL, NULL);
1014
1015 if (H == INVALID_HANDLE_VALUE) {
1016 error_code EC = windows_error(::GetLastError());
1017 // Provide a better error message when trying to open directories.
1018 // This only runs if we failed to open the file, so there is probably
1019 // no performances issues.
1020 if (EC != windows_error::access_denied)
1021 return EC;
1022 if (is_directory(Name))
1023 return error_code(errc::is_a_directory, posix_category());
1024 return EC;
1025 }
1026
1027 int OpenFlags = 0;
1028 if (Flags & F_Append)
1029 OpenFlags |= _O_APPEND;
1030
1031 if (!(Flags & F_Binary))
1032 OpenFlags |= _O_TEXT;
1033
1034 int FD = ::_open_osfhandle(intptr_t(H), OpenFlags);
1035 if (FD == -1) {
1036 ::CloseHandle(H);
1037 return windows_error::invalid_handle;
1038 }
1039
1040 ResultFD = FD;
1041 return error_code::success();
1042}
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +00001043} // end namespace fs
Rui Ueyama471d0c52013-09-10 19:45:51 +00001044
1045namespace windows {
1046llvm::error_code UTF8ToUTF16(llvm::StringRef utf8,
1047 llvm::SmallVectorImpl<wchar_t> &utf16) {
1048 int len = ::MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS,
1049 utf8.begin(), utf8.size(),
1050 utf16.begin(), 0);
1051
1052 if (len == 0)
1053 return llvm::windows_error(::GetLastError());
1054
1055 utf16.reserve(len + 1);
1056 utf16.set_size(len);
1057
1058 len = ::MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS,
1059 utf8.begin(), utf8.size(),
1060 utf16.begin(), utf16.size());
1061
1062 if (len == 0)
1063 return llvm::windows_error(::GetLastError());
1064
1065 // Make utf16 null terminated.
1066 utf16.push_back(0);
1067 utf16.pop_back();
1068
1069 return llvm::error_code::success();
1070}
1071
1072llvm::error_code UTF16ToUTF8(const wchar_t *utf16, size_t utf16_len,
1073 llvm::SmallVectorImpl<char> &utf8) {
1074 // Get length.
1075 int len = ::WideCharToMultiByte(CP_UTF8, 0,
1076 utf16, utf16_len,
1077 utf8.begin(), 0,
1078 NULL, NULL);
1079
1080 if (len == 0)
1081 return llvm::windows_error(::GetLastError());
1082
1083 utf8.reserve(len);
1084 utf8.set_size(len);
1085
1086 // Now do the actual conversion.
1087 len = ::WideCharToMultiByte(CP_UTF8, 0,
1088 utf16, utf16_len,
1089 utf8.data(), utf8.size(),
1090 NULL, NULL);
1091
1092 if (len == 0)
1093 return llvm::windows_error(::GetLastError());
1094
1095 // Make utf8 null terminated.
1096 utf8.push_back(0);
1097 utf8.pop_back();
1098
1099 return llvm::error_code::success();
1100}
1101} // end namespace windows
Michael J. Spencerebad2f92010-11-29 22:28:51 +00001102} // end namespace sys
1103} // end namespace llvm