blob: d8dc5226cceea3031e0911b859e10010dfa72ef6 [file] [log] [blame]
Charles Davis53ca1f32010-11-29 19:44:50 +00001//===- llvm/Support/Win32/Path.cpp - Win32 Path Implementation ---*- C++ -*-===//
Reid Spencerb016a372004-09-15 05:49:50 +00002//
Reid Spencercbad7012004-09-11 04:59:30 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Reid Spencerb016a372004-09-15 05:49:50 +00007//
Reid Spencercbad7012004-09-11 04:59:30 +00008//===----------------------------------------------------------------------===//
9//
10// This file provides the Win32 specific implementation of the Path class.
11//
12//===----------------------------------------------------------------------===//
13
14//===----------------------------------------------------------------------===//
Reid Spencerb016a372004-09-15 05:49:50 +000015//=== WARNING: Implementation here must contain only generic Win32 code that
16//=== is guaranteed to work on *all* Win32 variants.
Reid Spencercbad7012004-09-11 04:59:30 +000017//===----------------------------------------------------------------------===//
18
Michael J. Spencer1f6efa32010-11-29 18:16:10 +000019#include "Windows.h"
Reid Spencerd0c9e0e2004-09-18 19:29:16 +000020#include <malloc.h>
Chris Lattnerc23c1fc2009-04-01 02:03:38 +000021#include <cstdio>
Reid Spencercbad7012004-09-11 04:59:30 +000022
Jeff Cohencb652552004-12-24 02:38:34 +000023// We need to undo a macro defined in Windows.h, otherwise we won't compile:
24#undef CopyFile
Anton Korobeynikov64ddbe42007-12-22 14:26:49 +000025#undef GetCurrentDirectory
Jeff Cohencb652552004-12-24 02:38:34 +000026
Jeff Cohen966fa412005-07-09 18:42:49 +000027// Windows happily accepts either forward or backward slashes, though any path
28// returned by a Win32 API will have backward slashes. As LLVM code basically
29// assumes forward slashes are used, backward slashs are converted where they
30// can be introduced into a path.
31//
32// Another invariant is that a path ends with a slash if and only if the path
33// is a root directory. Any other use of a trailing slash is stripped. Unlike
34// in Unix, Windows has a rather complicated notion of a root path and this
35// invariant helps simply the code.
36
Reid Spencer6a0ec6f2004-09-29 00:01:17 +000037static void FlipBackSlashes(std::string& s) {
38 for (size_t i = 0; i < s.size(); i++)
39 if (s[i] == '\\')
40 s[i] = '/';
41}
42
Reid Spencercbad7012004-09-11 04:59:30 +000043namespace llvm {
Reid Spencerb016a372004-09-15 05:49:50 +000044namespace sys {
Mikhail Glushenkovf5a95ce2010-11-02 20:32:26 +000045
Chris Lattnera17fa282008-03-13 05:17:59 +000046const char PathSeparator = ';';
Chris Lattnere1b332a2008-02-27 06:17:10 +000047
Mikhail Glushenkovf5a95ce2010-11-02 20:32:26 +000048StringRef Path::GetEXESuffix() {
49 return "exe";
50}
51
Daniel Dunbar1edcafe2009-12-18 19:59:48 +000052Path::Path(llvm::StringRef p)
Nick Lewyckyfff116f2008-05-11 17:37:40 +000053 : path(p) {
54 FlipBackSlashes(path);
55}
56
57Path::Path(const char *StrStart, unsigned StrLen)
58 : path(StrStart, StrLen) {
59 FlipBackSlashes(path);
60}
61
Chris Lattner0eab5e22008-08-11 23:39:47 +000062Path&
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +000063Path::operator=(StringRef that) {
64 path.assign(that.data(), that.size());
Chris Lattner0eab5e22008-08-11 23:39:47 +000065 FlipBackSlashes(path);
66 return *this;
67}
68
Reid Spencerb016a372004-09-15 05:49:50 +000069bool
Reid Spencer07adb282004-11-05 22:15:36 +000070Path::isValid() const {
Reid Spencerb016a372004-09-15 05:49:50 +000071 if (path.empty())
72 return false;
Reid Spencerd0c9e0e2004-09-18 19:29:16 +000073
NAKAMURA Takumi5d070072011-10-24 03:27:19 +000074 size_t len = path.size();
75 // If there is a null character, it and all its successors are ignored.
76 size_t pos = path.find_first_of('\0');
77 if (pos != std::string::npos)
78 len = pos;
79
Reid Spencer6a0ec6f2004-09-29 00:01:17 +000080 // If there is a colon, it must be the second character, preceded by a letter
81 // and followed by something.
NAKAMURA Takumi5d070072011-10-24 03:27:19 +000082 pos = path.rfind(':',len);
Jeff Cohen8f0e8f22005-07-08 04:50:08 +000083 size_t rootslash = 0;
Reid Spencer6a0ec6f2004-09-29 00:01:17 +000084 if (pos != std::string::npos) {
85 if (pos != 1 || !isalpha(path[0]) || len < 3)
86 return false;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +000087 rootslash = 2;
88 }
Jeff Cohen85c716f2005-07-08 05:02:13 +000089
Jeff Cohen8f0e8f22005-07-08 04:50:08 +000090 // Look for a UNC path, and if found adjust our notion of the root slash.
91 if (len > 3 && path[0] == '/' && path[1] == '/') {
92 rootslash = path.find('/', 2);
93 if (rootslash == std::string::npos)
94 rootslash = 0;
Reid Spencer6a0ec6f2004-09-29 00:01:17 +000095 }
Reid Spencerd0c9e0e2004-09-18 19:29:16 +000096
Reid Spencer6a0ec6f2004-09-29 00:01:17 +000097 // Check for illegal characters.
98 if (path.find_first_of("\\<>\"|\001\002\003\004\005\006\007\010\011\012"
99 "\013\014\015\016\017\020\021\022\023\024\025\026"
100 "\027\030\031\032\033\034\035\036\037")
101 != std::string::npos)
102 return false;
Jeff Cohen85c716f2005-07-08 05:02:13 +0000103
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000104 // Remove trailing slash, unless it's a root slash.
105 if (len > rootslash+1 && path[len-1] == '/')
106 path.erase(--len);
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000107
Jeff Cohen3bbbcc12005-01-14 04:09:39 +0000108 // Check each component for legality.
109 for (pos = 0; pos < len; ++pos) {
110 // A component may not end in a space.
111 if (path[pos] == ' ') {
NAKAMURA Takumi5d070072011-10-24 03:27:19 +0000112 if (pos+1 == len || path[pos+1] == '/' || path[pos+1] == '\0')
Jeff Cohen3bbbcc12005-01-14 04:09:39 +0000113 return false;
114 }
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000115
Jeff Cohen3bbbcc12005-01-14 04:09:39 +0000116 // A component may not end in a period.
117 if (path[pos] == '.') {
NAKAMURA Takumi5d070072011-10-24 03:27:19 +0000118 if (pos+1 == len || path[pos+1] == '/') {
Jeff Cohen3bbbcc12005-01-14 04:09:39 +0000119 // Unless it is the pseudo-directory "."...
120 if (pos == 0 || path[pos-1] == '/' || path[pos-1] == ':')
121 return true;
122 // or "..".
123 if (pos > 0 && path[pos-1] == '.') {
124 if (pos == 1 || path[pos-2] == '/' || path[pos-2] == ':')
125 return true;
126 }
127 return false;
128 }
129 }
130 }
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000131
132 return true;
Reid Spencercbad7012004-09-11 04:59:30 +0000133}
134
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000135void Path::makeAbsolute() {
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000136 TCHAR FullPath[MAX_PATH + 1] = {0};
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000137 LPTSTR FilePart = NULL;
138
139 DWORD RetLength = ::GetFullPathNameA(path.c_str(),
140 sizeof(FullPath)/sizeof(FullPath[0]),
141 FullPath, &FilePart);
142
143 if (0 == RetLength) {
144 // FIXME: Report the error GetLastError()
Daniel Dunbar2749b3e2009-07-26 21:16:42 +0000145 assert(0 && "Unable to make absolute path!");
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000146 } else if (RetLength > MAX_PATH) {
147 // FIXME: Report too small buffer (needed RetLength bytes).
Daniel Dunbar2749b3e2009-07-26 21:16:42 +0000148 assert(0 && "Unable to make absolute path!");
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000149 } else {
150 path = FullPath;
151 }
152}
153
Chris Lattner88d7e402009-06-15 04:17:07 +0000154bool
155Path::isAbsolute(const char *NameStart, unsigned NameLen) {
156 assert(NameStart);
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000157 // FIXME: This does not handle correctly an absolute path starting from
158 // a drive letter or in UNC format.
Chris Lattner88d7e402009-06-15 04:17:07 +0000159 switch (NameLen) {
160 case 0:
161 return false;
162 case 1:
163 case 2:
164 return NameStart[0] == '/';
165 default:
Mikhail Glushenkov0f2ec152010-11-02 20:32:31 +0000166 return
167 (NameStart[0] == '/' || (NameStart[1] == ':' && NameStart[2] == '/')) ||
168 (NameStart[0] == '\\' || (NameStart[1] == ':' && NameStart[2] == '\\'));
Chris Lattner88d7e402009-06-15 04:17:07 +0000169 }
170}
171
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000172bool
Reid Spencer69cce812007-03-29 16:43:20 +0000173Path::isAbsolute() const {
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000174 // FIXME: This does not handle correctly an absolute path starting from
175 // a drive letter or in UNC format.
Jeff Cohen84892be2007-03-29 17:27:38 +0000176 switch (path.length()) {
177 case 0:
178 return false;
179 case 1:
180 case 2:
181 return path[0] == '/';
182 default:
183 return path[0] == '/' || (path[1] == ':' && path[2] == '/');
184 }
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000185}
Reid Spencer69cce812007-03-29 16:43:20 +0000186
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000187static Path *TempDirectory;
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000188
Reid Spencerb016a372004-09-15 05:49:50 +0000189Path
Reid Spencercab0e432006-08-22 22:46:39 +0000190Path::GetTemporaryDirectory(std::string* ErrMsg) {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000191 if (TempDirectory)
192 return *TempDirectory;
193
194 char pathname[MAX_PATH];
Reid Spencercab0e432006-08-22 22:46:39 +0000195 if (!GetTempPath(MAX_PATH, pathname)) {
196 if (ErrMsg)
197 *ErrMsg = "Can't determine temporary directory";
198 return Path();
199 }
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000200
Reid Spencerb016a372004-09-15 05:49:50 +0000201 Path result;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000202 result.set(pathname);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000203
204 // Append a subdirectory passed on our process id so multiple LLVMs don't
205 // step on each other's toes.
Jeff Cohend41b30d2006-11-05 19:31:28 +0000206#ifdef __MINGW32__
207 // Mingw's Win32 header files are broken.
Reid Spencerab4d9b02006-06-08 18:08:43 +0000208 sprintf(pathname, "LLVM_%u", unsigned(GetCurrentProcessId()));
Jeff Cohend41b30d2006-11-05 19:31:28 +0000209#else
210 sprintf(pathname, "LLVM_%u", GetCurrentProcessId());
211#endif
Jeff Cohenedb9d6b2005-07-08 02:48:42 +0000212 result.appendComponent(pathname);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000213
214 // If there's a directory left over from a previous LLVM execution that
215 // happened to have the same process id, get rid of it.
Reid Spencera229c5c2005-07-08 03:08:58 +0000216 result.eraseFromDisk(true);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000217
218 // And finally (re-)create the empty directory.
Reid Spencera229c5c2005-07-08 03:08:58 +0000219 result.createDirectoryOnDisk(false);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000220 TempDirectory = new Path(result);
221 return *TempDirectory;
Reid Spencerb016a372004-09-15 05:49:50 +0000222}
223
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000224// FIXME: the following set of functions don't map to Windows very well.
Reid Spencerb016a372004-09-15 05:49:50 +0000225Path
226Path::GetRootDirectory() {
Michael J. Spencer87e06972010-11-09 15:10:45 +0000227 // This is the only notion that that Windows has of a root directory. Nothing
228 // is here except for drives.
229 return Path("file:///");
Reid Spencerb016a372004-09-15 05:49:50 +0000230}
231
Jeff Cohen85c716f2005-07-08 05:02:13 +0000232void
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000233Path::GetSystemLibraryPaths(std::vector<sys::Path>& Paths) {
Michael J. Spencereebe9702010-11-09 15:11:19 +0000234 char buff[MAX_PATH];
235 // Generic form of C:\Windows\System32
236 HRESULT res = SHGetFolderPathA(NULL,
237 CSIDL_FLAG_CREATE | CSIDL_SYSTEM,
238 NULL,
239 SHGFP_TYPE_CURRENT,
240 buff);
241 if (res != S_OK) {
242 assert(0 && "Failed to get system directory");
243 return;
244 }
245 Paths.push_back(sys::Path(buff));
246
247 // Reset buff.
248 buff[0] = 0;
249 // Generic form of C:\Windows
250 res = SHGetFolderPathA(NULL,
251 CSIDL_FLAG_CREATE | CSIDL_WINDOWS,
252 NULL,
253 SHGFP_TYPE_CURRENT,
254 buff);
255 if (res != S_OK) {
256 assert(0 && "Failed to get windows directory");
257 return;
258 }
259 Paths.push_back(sys::Path(buff));
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000260}
261
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000262void
Gabor Greifdb5565a2007-07-06 20:28:40 +0000263Path::GetBitcodeLibraryPaths(std::vector<sys::Path>& Paths) {
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000264 char * env_var = getenv("LLVM_LIB_SEARCH_PATH");
265 if (env_var != 0) {
266 getPathList(env_var,Paths);
267 }
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000268#ifdef LLVM_LIBDIR
269 {
270 Path tmpPath;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000271 if (tmpPath.set(LLVM_LIBDIR))
Reid Spencerc7f08322005-07-07 18:21:42 +0000272 if (tmpPath.canRead())
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000273 Paths.push_back(tmpPath);
274 }
275#endif
276 GetSystemLibraryPaths(Paths);
Reid Spencerb016a372004-09-15 05:49:50 +0000277}
278
279Path
Reid Spencerb016a372004-09-15 05:49:50 +0000280Path::GetUserHomeDirectory() {
Michael J. Spencer05283c22010-11-09 15:11:31 +0000281 char buff[MAX_PATH];
Michael J. Spencer6c43de42010-11-10 15:06:00 +0000282 HRESULT res = SHGetFolderPathA(NULL,
283 CSIDL_FLAG_CREATE | CSIDL_APPDATA,
284 NULL,
285 SHGFP_TYPE_CURRENT,
286 buff);
Michael J. Spencer05283c22010-11-09 15:11:31 +0000287 if (res != S_OK)
288 assert(0 && "Failed to get user home directory");
289 return Path(buff);
Reid Spencerb016a372004-09-15 05:49:50 +0000290}
Ted Kremenek79200782007-12-18 22:07:33 +0000291
292Path
293Path::GetCurrentDirectory() {
294 char pathname[MAX_PATH];
Anton Korobeynikov64ddbe42007-12-22 14:26:49 +0000295 ::GetCurrentDirectoryA(MAX_PATH,pathname);
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000296 return Path(pathname);
Ted Kremenek79200782007-12-18 22:07:33 +0000297}
298
Chris Lattner1a091442008-03-03 02:55:43 +0000299/// GetMainExecutable - Return the path to the main executable, given the
300/// value of argv[0] from program startup.
301Path Path::GetMainExecutable(const char *argv0, void *MainAddr) {
Chris Lattnerc5718d02009-06-15 05:38:04 +0000302 char pathname[MAX_PATH];
303 DWORD ret = ::GetModuleFileNameA(NULL, pathname, MAX_PATH);
304 return ret != MAX_PATH ? Path(pathname) : Path();
Chris Lattner1a091442008-03-03 02:55:43 +0000305}
306
Ted Kremenek79200782007-12-18 22:07:33 +0000307
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000308// FIXME: the above set of functions don't map to Windows very well.
309
Jeff Cohen966fa412005-07-09 18:42:49 +0000310
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000311StringRef Path::getDirname() const {
312 return getDirnameCharSep(path, "/");
Ted Kremenek9b01cc02008-04-07 22:01:32 +0000313}
Ted Kremenekcf55c8e2008-04-07 21:53:57 +0000314
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000315StringRef
Reid Spencer07adb282004-11-05 22:15:36 +0000316Path::getBasename() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000317 // Find the last slash
318 size_t slash = path.rfind('/');
319 if (slash == std::string::npos)
320 slash = 0;
321 else
322 slash++;
323
Jeff Cohen966fa412005-07-09 18:42:49 +0000324 size_t dot = path.rfind('.');
325 if (dot == std::string::npos || dot < slash)
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000326 return StringRef(path).substr(slash);
Jeff Cohen966fa412005-07-09 18:42:49 +0000327 else
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000328 return StringRef(path).substr(slash, dot - slash);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000329}
330
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000331StringRef
Argyrios Kyrtzidisfc199882008-06-15 15:15:19 +0000332Path::getSuffix() const {
333 // Find the last slash
334 size_t slash = path.rfind('/');
335 if (slash == std::string::npos)
336 slash = 0;
337 else
338 slash++;
339
340 size_t dot = path.rfind('.');
341 if (dot == std::string::npos || dot < slash)
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000342 return StringRef("");
Argyrios Kyrtzidisfc199882008-06-15 15:15:19 +0000343 else
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000344 return StringRef(path).substr(dot + 1);
Argyrios Kyrtzidisfc199882008-06-15 15:15:19 +0000345}
346
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000347bool
Reid Spencerb016a372004-09-15 05:49:50 +0000348Path::exists() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000349 DWORD attr = GetFileAttributes(path.c_str());
350 return attr != INVALID_FILE_ATTRIBUTES;
Reid Spencerb016a372004-09-15 05:49:50 +0000351}
352
353bool
Ted Kremenekfd527112007-12-18 19:46:22 +0000354Path::isDirectory() const {
355 DWORD attr = GetFileAttributes(path.c_str());
356 return (attr != INVALID_FILE_ATTRIBUTES) &&
357 (attr & FILE_ATTRIBUTE_DIRECTORY);
358}
359
360bool
Rafael Espindola9e07a142010-11-07 04:36:50 +0000361Path::isSymLink() const {
Michael J. Spencer339b9122010-11-10 15:05:39 +0000362 DWORD attributes = GetFileAttributes(path.c_str());
363
364 if (attributes == INVALID_FILE_ATTRIBUTES)
365 // There's no sane way to report this :(.
366 assert(0 && "GetFileAttributes returned INVALID_FILE_ATTRIBUTES");
367
368 // This isn't exactly what defines a NTFS symlink, but it is only true for
369 // paths that act like a symlink.
370 return attributes & FILE_ATTRIBUTE_REPARSE_POINT;
Rafael Espindola9e07a142010-11-07 04:36:50 +0000371}
372
373bool
Reid Spencerc7f08322005-07-07 18:21:42 +0000374Path::canRead() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000375 // FIXME: take security attributes into account.
376 DWORD attr = GetFileAttributes(path.c_str());
377 return attr != INVALID_FILE_ATTRIBUTES;
Reid Spencerb016a372004-09-15 05:49:50 +0000378}
379
380bool
Reid Spencerc7f08322005-07-07 18:21:42 +0000381Path::canWrite() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000382 // FIXME: take security attributes into account.
383 DWORD attr = GetFileAttributes(path.c_str());
384 return (attr != INVALID_FILE_ATTRIBUTES) && !(attr & FILE_ATTRIBUTE_READONLY);
Reid Spencerb016a372004-09-15 05:49:50 +0000385}
386
387bool
Reid Spencerc7f08322005-07-07 18:21:42 +0000388Path::canExecute() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000389 // FIXME: take security attributes into account.
390 DWORD attr = GetFileAttributes(path.c_str());
391 return attr != INVALID_FILE_ATTRIBUTES;
Reid Spencerb016a372004-09-15 05:49:50 +0000392}
393
Edward O'Callaghand41e9442009-11-24 15:19:10 +0000394bool
Edward O'Callaghane49a8e42009-11-25 06:32:19 +0000395Path::isRegularFile() const {
Michael J. Spencer218dc3e2011-01-11 01:21:55 +0000396 bool res;
397 if (fs::is_regular_file(path, res))
Edward O'Callaghane49a8e42009-11-25 06:32:19 +0000398 return false;
Michael J. Spencer218dc3e2011-01-11 01:21:55 +0000399 return res;
Edward O'Callaghand41e9442009-11-24 15:19:10 +0000400}
401
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000402StringRef
Reid Spencerb016a372004-09-15 05:49:50 +0000403Path::getLast() const {
404 // Find the last slash
405 size_t pos = path.rfind('/');
406
407 // Handle the corner cases
408 if (pos == std::string::npos)
409 return path;
410
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000411 // If the last character is a slash, we have a root directory
412 if (pos == path.length()-1)
413 return path;
414
Reid Spencerb016a372004-09-15 05:49:50 +0000415 // Return everything after the last slash
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000416 return StringRef(path).substr(pos+1);
Reid Spencerb016a372004-09-15 05:49:50 +0000417}
418
Reid Spencer8475ec02007-03-29 19:05:44 +0000419const FileStatus *
Reid Spencer2ae9d112007-04-07 18:52:17 +0000420PathWithStatus::getFileStatus(bool update, std::string *ErrStr) const {
421 if (!fsIsValid || update) {
Reid Spencer69cce812007-03-29 16:43:20 +0000422 WIN32_FILE_ATTRIBUTE_DATA fi;
Reid Spencer8475ec02007-03-29 19:05:44 +0000423 if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
424 MakeErrMsg(ErrStr, "getStatusInfo():" + std::string(path) +
Reid Spencer69cce812007-03-29 16:43:20 +0000425 ": Can't get status: ");
Reid Spencer8475ec02007-03-29 19:05:44 +0000426 return 0;
427 }
Jeff Cohen626e38e2004-12-14 05:26:43 +0000428
Reid Spencer2ae9d112007-04-07 18:52:17 +0000429 status.fileSize = fi.nFileSizeHigh;
430 status.fileSize <<= sizeof(fi.nFileSizeHigh)*8;
431 status.fileSize += fi.nFileSizeLow;
Jeff Cohen626e38e2004-12-14 05:26:43 +0000432
Reid Spencer2ae9d112007-04-07 18:52:17 +0000433 status.mode = fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY ? 0555 : 0777;
434 status.user = 9999; // Not applicable to Windows, so...
435 status.group = 9999; // Not applicable to Windows, so...
Jeff Cohen626e38e2004-12-14 05:26:43 +0000436
Reid Spencer4031bef2007-03-29 17:00:31 +0000437 // FIXME: this is only unique if the file is accessed by the same file path.
438 // How do we do this for C:\dir\file and ..\dir\file ? Unix has inode
439 // numbers, but the concept doesn't exist in Windows.
Reid Spencer2ae9d112007-04-07 18:52:17 +0000440 status.uniqueID = 0;
Reid Spencer4031bef2007-03-29 17:00:31 +0000441 for (unsigned i = 0; i < path.length(); ++i)
Reid Spencer2ae9d112007-04-07 18:52:17 +0000442 status.uniqueID += path[i];
Reid Spencer4031bef2007-03-29 17:00:31 +0000443
Michael J. Spencer44edb0b2010-08-28 16:39:32 +0000444 ULARGE_INTEGER ui;
445 ui.LowPart = fi.ftLastWriteTime.dwLowDateTime;
446 ui.HighPart = fi.ftLastWriteTime.dwHighDateTime;
447 status.modTime.fromWin32Time(ui.QuadPart);
Reid Spencer69cce812007-03-29 16:43:20 +0000448
Reid Spencer2ae9d112007-04-07 18:52:17 +0000449 status.isDir = fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
450 fsIsValid = true;
Reid Spencer69cce812007-03-29 16:43:20 +0000451 }
Reid Spencer2ae9d112007-04-07 18:52:17 +0000452 return &status;
Jeff Cohen626e38e2004-12-14 05:26:43 +0000453}
454
Reid Spencera34a1572006-08-22 23:54:35 +0000455bool Path::makeReadableOnDisk(std::string* ErrMsg) {
Jeff Cohen626e38e2004-12-14 05:26:43 +0000456 // All files are readable on Windows (ignoring security attributes).
Reid Spencera34a1572006-08-22 23:54:35 +0000457 return false;
Reid Spencer77cc91d2004-12-13 19:59:50 +0000458}
459
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000460bool Path::makeWriteableOnDisk(std::string* ErrMsg) {
Jeff Cohen626e38e2004-12-14 05:26:43 +0000461 DWORD attr = GetFileAttributes(path.c_str());
462
463 // If it doesn't exist, we're done.
464 if (attr == INVALID_FILE_ATTRIBUTES)
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000465 return false;
Jeff Cohen626e38e2004-12-14 05:26:43 +0000466
467 if (attr & FILE_ATTRIBUTE_READONLY) {
Reid Spencera34a1572006-08-22 23:54:35 +0000468 if (!SetFileAttributes(path.c_str(), attr & ~FILE_ATTRIBUTE_READONLY)) {
469 MakeErrMsg(ErrMsg, std::string(path) + ": Can't make file writable: ");
470 return true;
471 }
Jeff Cohen626e38e2004-12-14 05:26:43 +0000472 }
Reid Spencera34a1572006-08-22 23:54:35 +0000473 return false;
Reid Spencer77cc91d2004-12-13 19:59:50 +0000474}
475
Reid Spencera34a1572006-08-22 23:54:35 +0000476bool Path::makeExecutableOnDisk(std::string* ErrMsg) {
Jeff Cohen626e38e2004-12-14 05:26:43 +0000477 // All files are executable on Windows (ignoring security attributes).
Reid Spencera34a1572006-08-22 23:54:35 +0000478 return false;
Reid Spencer77cc91d2004-12-13 19:59:50 +0000479}
480
Reid Spencerb016a372004-09-15 05:49:50 +0000481bool
Reid Spencer142ca8e2006-08-23 06:56:27 +0000482Path::getDirectoryContents(std::set<Path>& result, std::string* ErrMsg) const {
Jeff Cohen31102892007-04-07 20:47:27 +0000483 WIN32_FILE_ATTRIBUTE_DATA fi;
484 if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
485 MakeErrMsg(ErrMsg, path + ": can't get status of file");
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000486 return true;
Jeff Cohen31102892007-04-07 20:47:27 +0000487 }
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000488
Jeff Cohen31102892007-04-07 20:47:27 +0000489 if (!(fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
490 if (ErrMsg)
491 *ErrMsg = path + ": not a directory";
Reid Spencer142ca8e2006-08-23 06:56:27 +0000492 return true;
493 }
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000494
495 result.clear();
496 WIN32_FIND_DATA fd;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000497 std::string searchpath = path;
498 if (path.size() == 0 || searchpath[path.size()-1] == '/')
Jeff Cohen85c716f2005-07-08 05:02:13 +0000499 searchpath += "*";
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000500 else
501 searchpath += "/*";
Jeff Cohen85c716f2005-07-08 05:02:13 +0000502
Jeff Cohen9437bb62005-01-27 03:49:03 +0000503 HANDLE h = FindFirstFile(searchpath.c_str(), &fd);
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000504 if (h == INVALID_HANDLE_VALUE) {
Jeff Cohen9437bb62005-01-27 03:49:03 +0000505 if (GetLastError() == ERROR_FILE_NOT_FOUND)
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000506 return true; // not really an error, now is it?
Reid Spencer142ca8e2006-08-23 06:56:27 +0000507 MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
508 return true;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000509 }
510
511 do {
Jeff Cohend40a7de2004-12-31 05:07:26 +0000512 if (fd.cFileName[0] == '.')
513 continue;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000514 Path aPath(path);
515 aPath.appendComponent(&fd.cFileName[0]);
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000516 result.insert(aPath);
517 } while (FindNextFile(h, &fd));
518
Jeff Cohen51b8d212004-12-31 19:01:08 +0000519 DWORD err = GetLastError();
520 FindClose(h);
521 if (err != ERROR_NO_MORE_FILES) {
522 SetLastError(err);
Reid Spencer142ca8e2006-08-23 06:56:27 +0000523 MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
524 return true;
Jeff Cohen51b8d212004-12-31 19:01:08 +0000525 }
Reid Spencer142ca8e2006-08-23 06:56:27 +0000526 return false;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000527}
528
529bool
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000530Path::set(StringRef a_path) {
Dan Gohman30359592008-01-29 13:02:09 +0000531 if (a_path.empty())
Reid Spencerb016a372004-09-15 05:49:50 +0000532 return false;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000533 std::string save(path);
Reid Spencerb016a372004-09-15 05:49:50 +0000534 path = a_path;
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000535 FlipBackSlashes(path);
Reid Spencer07adb282004-11-05 22:15:36 +0000536 if (!isValid()) {
Reid Spencer1cf2d042005-07-07 23:35:23 +0000537 path = save;
Reid Spencerb016a372004-09-15 05:49:50 +0000538 return false;
539 }
540 return true;
541}
542
543bool
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000544Path::appendComponent(StringRef name) {
Reid Spencer1cf2d042005-07-07 23:35:23 +0000545 if (name.empty())
Reid Spencerb016a372004-09-15 05:49:50 +0000546 return false;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000547 std::string save(path);
548 if (!path.empty()) {
549 size_t last = path.size() - 1;
Jeff Cohen85c716f2005-07-08 05:02:13 +0000550 if (path[last] != '/')
Reid Spencer1cf2d042005-07-07 23:35:23 +0000551 path += '/';
552 }
553 path += name;
Reid Spencer07adb282004-11-05 22:15:36 +0000554 if (!isValid()) {
Reid Spencer1cf2d042005-07-07 23:35:23 +0000555 path = save;
Reid Spencerb016a372004-09-15 05:49:50 +0000556 return false;
557 }
558 return true;
559}
560
561bool
Reid Spencer1cf2d042005-07-07 23:35:23 +0000562Path::eraseComponent() {
Reid Spencerb016a372004-09-15 05:49:50 +0000563 size_t slashpos = path.rfind('/',path.size());
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000564 if (slashpos == path.size() - 1 || slashpos == std::string::npos)
Reid Spencerb016a372004-09-15 05:49:50 +0000565 return false;
Jeff Cohen966fa412005-07-09 18:42:49 +0000566 std::string save(path);
Reid Spencerb016a372004-09-15 05:49:50 +0000567 path.erase(slashpos);
Jeff Cohen966fa412005-07-09 18:42:49 +0000568 if (!isValid()) {
569 path = save;
570 return false;
571 }
Reid Spencerb016a372004-09-15 05:49:50 +0000572 return true;
573}
574
575bool
Reid Spencer1cf2d042005-07-07 23:35:23 +0000576Path::eraseSuffix() {
Reid Spencerb016a372004-09-15 05:49:50 +0000577 size_t dotpos = path.rfind('.',path.size());
578 size_t slashpos = path.rfind('/',path.size());
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000579 if (dotpos != std::string::npos) {
Jeff Cohen966fa412005-07-09 18:42:49 +0000580 if (slashpos == std::string::npos || dotpos > slashpos+1) {
581 std::string save(path);
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000582 path.erase(dotpos, path.size()-dotpos);
Jeff Cohen966fa412005-07-09 18:42:49 +0000583 if (!isValid()) {
584 path = save;
585 return false;
586 }
Jeff Cohen85c716f2005-07-08 05:02:13 +0000587 return true;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000588 }
Reid Spencerb016a372004-09-15 05:49:50 +0000589 }
590 return false;
591}
592
Reid Spencer30300992006-08-24 18:58:37 +0000593inline bool PathMsg(std::string* ErrMsg, const char* pathname, const char*msg) {
594 if (ErrMsg)
595 *ErrMsg = std::string(pathname) + ": " + std::string(msg);
596 return true;
597}
598
Reid Spencerb016a372004-09-15 05:49:50 +0000599bool
Reid Spencer30300992006-08-24 18:58:37 +0000600Path::createDirectoryOnDisk(bool create_parents, std::string* ErrMsg) {
Reid Spencerb016a372004-09-15 05:49:50 +0000601 // Get a writeable copy of the path name
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000602 size_t len = path.length();
603 char *pathname = reinterpret_cast<char *>(_alloca(len+2));
604 path.copy(pathname, len);
605 pathname[len] = 0;
Jeff Cohen85c716f2005-07-08 05:02:13 +0000606
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000607 // Make sure it ends with a slash.
608 if (len == 0 || pathname[len - 1] != '/') {
609 pathname[len] = '/';
610 pathname[++len] = 0;
611 }
Reid Spencerb016a372004-09-15 05:49:50 +0000612
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000613 // Determine starting point for initial / search.
614 char *next = pathname;
615 if (pathname[0] == '/' && pathname[1] == '/') {
616 // Skip host name.
617 next = strchr(pathname+2, '/');
618 if (next == NULL)
Reid Spencer30300992006-08-24 18:58:37 +0000619 return PathMsg(ErrMsg, pathname, "badly formed remote directory");
620
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000621 // Skip share name.
622 next = strchr(next+1, '/');
623 if (next == NULL)
Reid Spencer30300992006-08-24 18:58:37 +0000624 return PathMsg(ErrMsg, pathname,"badly formed remote directory");
625
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000626 next++;
627 if (*next == 0)
Reid Spencer30300992006-08-24 18:58:37 +0000628 return PathMsg(ErrMsg, pathname, "badly formed remote directory");
629
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000630 } else {
631 if (pathname[1] == ':')
632 next += 2; // skip drive letter
633 if (*next == '/')
634 next++; // skip root directory
635 }
Reid Spencerb016a372004-09-15 05:49:50 +0000636
637 // If we're supposed to create intermediate directories
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000638 if (create_parents) {
Reid Spencerb016a372004-09-15 05:49:50 +0000639 // Loop through the directory components until we're done
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000640 while (*next) {
641 next = strchr(next, '/');
Reid Spencerb016a372004-09-15 05:49:50 +0000642 *next = 0;
Benjamin Kramere9684c62009-11-05 14:32:40 +0000643 if (!CreateDirectory(pathname, NULL) &&
644 GetLastError() != ERROR_ALREADY_EXISTS)
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000645 return MakeErrMsg(ErrMsg,
Reid Spencer30300992006-08-24 18:58:37 +0000646 std::string(pathname) + ": Can't create directory: ");
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000647 *next++ = '/';
Reid Spencerb016a372004-09-15 05:49:50 +0000648 }
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000649 } else {
650 // Drop trailing slash.
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000651 pathname[len-1] = 0;
Benjamin Kramere9684c62009-11-05 14:32:40 +0000652 if (!CreateDirectory(pathname, NULL) &&
653 GetLastError() != ERROR_ALREADY_EXISTS) {
Mikhail Glushenkov0f2ec152010-11-02 20:32:31 +0000654 return MakeErrMsg(ErrMsg, std::string(pathname) +
655 ": Can't create directory: ");
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000656 }
Reid Spencerb016a372004-09-15 05:49:50 +0000657 }
Reid Spencer30300992006-08-24 18:58:37 +0000658 return false;
Reid Spencerb016a372004-09-15 05:49:50 +0000659}
660
661bool
Reid Spencer30300992006-08-24 18:58:37 +0000662Path::createFileOnDisk(std::string* ErrMsg) {
Reid Spencerb016a372004-09-15 05:49:50 +0000663 // Create the file
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000664 HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000665 FILE_ATTRIBUTE_NORMAL, NULL);
666 if (h == INVALID_HANDLE_VALUE)
Reid Spencer30300992006-08-24 18:58:37 +0000667 return MakeErrMsg(ErrMsg, path + ": Can't create file: ");
Reid Spencerb016a372004-09-15 05:49:50 +0000668
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000669 CloseHandle(h);
Reid Spencer30300992006-08-24 18:58:37 +0000670 return false;
Reid Spencerb016a372004-09-15 05:49:50 +0000671}
672
673bool
Chris Lattner0c332312006-07-28 22:29:50 +0000674Path::eraseFromDisk(bool remove_contents, std::string *ErrStr) const {
Jeff Cohen31102892007-04-07 20:47:27 +0000675 WIN32_FILE_ATTRIBUTE_DATA fi;
676 if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi))
677 return true;
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000678
Jeff Cohen31102892007-04-07 20:47:27 +0000679 if (fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
Reid Spencer1cf2d042005-07-07 23:35:23 +0000680 // If it doesn't exist, we're done.
Michael J. Spencer54453f22011-01-10 02:34:23 +0000681 bool Exists;
682 if (fs::exists(path, Exists) || !Exists)
Chris Lattner0c332312006-07-28 22:29:50 +0000683 return false;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000684
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000685 char *pathname = reinterpret_cast<char *>(_alloca(path.length()+3));
Reid Spencer1cf2d042005-07-07 23:35:23 +0000686 int lastchar = path.length() - 1 ;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000687 path.copy(pathname, lastchar+1);
Reid Spencer1cf2d042005-07-07 23:35:23 +0000688
689 // Make path end with '/*'.
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000690 if (pathname[lastchar] != '/')
691 pathname[++lastchar] = '/';
Reid Spencer1cf2d042005-07-07 23:35:23 +0000692 pathname[lastchar+1] = '*';
693 pathname[lastchar+2] = 0;
694
695 if (remove_contents) {
696 WIN32_FIND_DATA fd;
697 HANDLE h = FindFirstFile(pathname, &fd);
698
699 // It's a bad idea to alter the contents of a directory while enumerating
700 // its contents. So build a list of its contents first, then destroy them.
701
702 if (h != INVALID_HANDLE_VALUE) {
703 std::vector<Path> list;
704
705 do {
706 if (strcmp(fd.cFileName, ".") == 0)
707 continue;
708 if (strcmp(fd.cFileName, "..") == 0)
709 continue;
710
Jeff Cohen85c716f2005-07-08 05:02:13 +0000711 Path aPath(path);
712 aPath.appendComponent(&fd.cFileName[0]);
Reid Spencer1cf2d042005-07-07 23:35:23 +0000713 list.push_back(aPath);
714 } while (FindNextFile(h, &fd));
715
716 DWORD err = GetLastError();
717 FindClose(h);
718 if (err != ERROR_NO_MORE_FILES) {
719 SetLastError(err);
Reid Spencer05545752006-08-25 21:37:17 +0000720 return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
Reid Spencer1cf2d042005-07-07 23:35:23 +0000721 }
722
Jeff Cohen85c716f2005-07-08 05:02:13 +0000723 for (std::vector<Path>::iterator I = list.begin(); I != list.end();
Reid Spencer1cf2d042005-07-07 23:35:23 +0000724 ++I) {
725 Path &aPath = *I;
Reid Spencera229c5c2005-07-08 03:08:58 +0000726 aPath.eraseFromDisk(true);
Reid Spencer1cf2d042005-07-07 23:35:23 +0000727 }
728 } else {
729 if (GetLastError() != ERROR_FILE_NOT_FOUND)
Reid Spencer05545752006-08-25 21:37:17 +0000730 return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
Reid Spencer1cf2d042005-07-07 23:35:23 +0000731 }
732 }
733
734 pathname[lastchar] = 0;
735 if (!RemoveDirectory(pathname))
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000736 return MakeErrMsg(ErrStr,
Reid Spencer05545752006-08-25 21:37:17 +0000737 std::string(pathname) + ": Can't destroy directory: ");
Chris Lattner0c332312006-07-28 22:29:50 +0000738 return false;
Jeff Cohen31102892007-04-07 20:47:27 +0000739 } else {
740 // Read-only files cannot be deleted on Windows. Must remove the read-only
741 // attribute first.
742 if (fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
743 if (!SetFileAttributes(path.c_str(),
744 fi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
745 return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
746 }
747
748 if (!DeleteFile(path.c_str()))
749 return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
750 return false;
751 }
Reid Spencerb016a372004-09-15 05:49:50 +0000752}
753
Reid Spencer3b0cc782004-12-14 18:42:13 +0000754bool Path::getMagicNumber(std::string& Magic, unsigned len) const {
Reid Spencer3b0cc782004-12-14 18:42:13 +0000755 assert(len < 1024 && "Request for magic string too long");
Michael J. Spencer1211d432010-08-31 06:36:33 +0000756 char* buf = reinterpret_cast<char*>(alloca(len));
Jeff Cohen51b8d212004-12-31 19:01:08 +0000757
758 HANDLE h = CreateFile(path.c_str(),
759 GENERIC_READ,
760 FILE_SHARE_READ,
761 NULL,
762 OPEN_EXISTING,
763 FILE_ATTRIBUTE_NORMAL,
764 NULL);
765 if (h == INVALID_HANDLE_VALUE)
Reid Spencer3b0cc782004-12-14 18:42:13 +0000766 return false;
Jeff Cohen51b8d212004-12-31 19:01:08 +0000767
768 DWORD nRead = 0;
769 BOOL ret = ReadFile(h, buf, len, &nRead, NULL);
770 CloseHandle(h);
771
772 if (!ret || nRead != len)
Reid Spencer3b0cc782004-12-14 18:42:13 +0000773 return false;
Jeff Cohen51b8d212004-12-31 19:01:08 +0000774
Michael J. Spencer1211d432010-08-31 06:36:33 +0000775 Magic = std::string(buf, len);
Reid Spencer3b0cc782004-12-14 18:42:13 +0000776 return true;
777}
778
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000779bool
Reid Spencer5a060772006-08-23 07:30:48 +0000780Path::renamePathOnDisk(const Path& newName, std::string* ErrMsg) {
Francois Pichet3eddd982010-09-30 00:44:58 +0000781 if (!MoveFileEx(path.c_str(), newName.c_str(), MOVEFILE_REPLACE_EXISTING))
782 return MakeErrMsg(ErrMsg, "Can't move '" + path + "' to '" + newName.path
783 + "': ");
784 return false;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000785}
786
787bool
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000788Path::setStatusInfoOnDisk(const FileStatus &si, std::string *ErrMsg) const {
Jeff Cohen966fa412005-07-09 18:42:49 +0000789 // FIXME: should work on directories also.
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000790 if (!si.isFile) {
791 return true;
792 }
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000793
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000794 HANDLE h = CreateFile(path.c_str(),
795 FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES,
796 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
Jeff Cohend40a7de2004-12-31 05:07:26 +0000797 NULL,
798 OPEN_EXISTING,
799 FILE_ATTRIBUTE_NORMAL,
800 NULL);
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000801 if (h == INVALID_HANDLE_VALUE)
Chris Lattner1bebfb52006-07-28 22:36:17 +0000802 return true;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000803
804 BY_HANDLE_FILE_INFORMATION bhfi;
805 if (!GetFileInformationByHandle(h, &bhfi)) {
Jeff Cohen51b8d212004-12-31 19:01:08 +0000806 DWORD err = GetLastError();
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000807 CloseHandle(h);
Jeff Cohen51b8d212004-12-31 19:01:08 +0000808 SetLastError(err);
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000809 return MakeErrMsg(ErrMsg, path + ": GetFileInformationByHandle: ");
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000810 }
811
Michael J. Spencer44edb0b2010-08-28 16:39:32 +0000812 ULARGE_INTEGER ui;
813 ui.QuadPart = si.modTime.toWin32Time();
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000814 FILETIME ft;
Michael J. Spencer44edb0b2010-08-28 16:39:32 +0000815 ft.dwLowDateTime = ui.LowPart;
816 ft.dwHighDateTime = ui.HighPart;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000817 BOOL ret = SetFileTime(h, NULL, &ft, &ft);
Jeff Cohen51b8d212004-12-31 19:01:08 +0000818 DWORD err = GetLastError();
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000819 CloseHandle(h);
Jeff Cohen51b8d212004-12-31 19:01:08 +0000820 if (!ret) {
821 SetLastError(err);
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000822 return MakeErrMsg(ErrMsg, path + ": SetFileTime: ");
Jeff Cohen51b8d212004-12-31 19:01:08 +0000823 }
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000824
825 // Best we can do with Unix permission bits is to interpret the owner
826 // writable bit.
827 if (si.mode & 0200) {
828 if (bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
829 if (!SetFileAttributes(path.c_str(),
Jeff Cohend40a7de2004-12-31 05:07:26 +0000830 bhfi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000831 return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000832 }
833 } else {
834 if (!(bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) {
835 if (!SetFileAttributes(path.c_str(),
Jeff Cohend40a7de2004-12-31 05:07:26 +0000836 bhfi.dwFileAttributes | FILE_ATTRIBUTE_READONLY))
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000837 return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000838 }
839 }
840
Chris Lattner1bebfb52006-07-28 22:36:17 +0000841 return false;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000842}
843
Reid Spencer30300992006-08-24 18:58:37 +0000844bool
845CopyFile(const sys::Path &Dest, const sys::Path &Src, std::string* ErrMsg) {
Jeff Cohencb652552004-12-24 02:38:34 +0000846 // Can't use CopyFile macro defined in Windows.h because it would mess up the
847 // above line. We use the expansion it would have in a non-UNICODE build.
848 if (!::CopyFileA(Src.c_str(), Dest.c_str(), false))
Chris Lattner74382b72009-08-23 22:45:37 +0000849 return MakeErrMsg(ErrMsg, "Can't copy '" + Src.str() +
850 "' to '" + Dest.str() + "': ");
Reid Spencer30300992006-08-24 18:58:37 +0000851 return false;
Reid Spencerc29befb2004-12-15 01:50:13 +0000852}
853
Reid Spencer30300992006-08-24 18:58:37 +0000854bool
855Path::makeUnique(bool reuse_current, std::string* ErrMsg) {
Michael J. Spencer54453f22011-01-10 02:34:23 +0000856 bool Exists;
857 if (reuse_current && (fs::exists(path, Exists) || !Exists))
Reid Spencer30300992006-08-24 18:58:37 +0000858 return false; // File doesn't exist already, just use it!
Reid Spencerc29befb2004-12-15 01:50:13 +0000859
Jeff Cohen9437bb62005-01-27 03:49:03 +0000860 // Reserve space for -XXXXXX at the end.
861 char *FNBuffer = (char*) alloca(path.size()+8);
862 unsigned offset = path.size();
863 path.copy(FNBuffer, offset);
Reid Spencerc29befb2004-12-15 01:50:13 +0000864
Jeff Cohen966fa412005-07-09 18:42:49 +0000865 // Find a numeric suffix that isn't used by an existing file. Assume there
866 // won't be more than 1 million files with the same prefix. Probably a safe
867 // bet.
NAKAMURA Takumibfb25cd2011-03-16 02:53:24 +0000868 static int FCounter = -1;
869 if (FCounter < 0) {
870 // Give arbitrary initial seed.
871 // FIXME: We should use sys::fs::unique_file() in future.
872 LARGE_INTEGER cnt64;
873 DWORD x = GetCurrentProcessId();
874 x = (x << 16) | (x >> 16);
875 if (QueryPerformanceCounter(&cnt64)) // RDTSC
876 x ^= cnt64.HighPart ^ cnt64.LowPart;
877 FCounter = x % 1000000;
878 }
Jeff Cohen9437bb62005-01-27 03:49:03 +0000879 do {
880 sprintf(FNBuffer+offset, "-%06u", FCounter);
881 if (++FCounter > 999999)
882 FCounter = 0;
883 path = FNBuffer;
Michael J. Spencer54453f22011-01-10 02:34:23 +0000884 } while (!fs::exists(path, Exists) && Exists);
Reid Spencer30300992006-08-24 18:58:37 +0000885 return false;
Reid Spencerc29befb2004-12-15 01:50:13 +0000886}
887
Reid Spencer07f9f4e2004-12-15 08:32:45 +0000888bool
Reid Spencer30300992006-08-24 18:58:37 +0000889Path::createTemporaryFileOnDisk(bool reuse_current, std::string* ErrMsg) {
Reid Spencer07f9f4e2004-12-15 08:32:45 +0000890 // Make this into a unique file name
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000891 makeUnique(reuse_current, ErrMsg);
Jeff Cohen9437bb62005-01-27 03:49:03 +0000892
893 // Now go and create it
894 HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
895 FILE_ATTRIBUTE_NORMAL, NULL);
896 if (h == INVALID_HANDLE_VALUE)
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000897 return MakeErrMsg(ErrMsg, path + ": can't create file");
Jeff Cohen9437bb62005-01-27 03:49:03 +0000898
899 CloseHandle(h);
Reid Spencer30300992006-08-24 18:58:37 +0000900 return false;
Reid Spencer07f9f4e2004-12-15 08:32:45 +0000901}
902
Chris Lattner799ed102008-04-01 06:00:12 +0000903/// MapInFilePages - Not yet implemented on win32.
Rafael Espindolab78e2ae2011-03-10 18:30:48 +0000904const char *Path::MapInFilePages(int FD, size_t FileSize, off_t Offset) {
Chris Lattner799ed102008-04-01 06:00:12 +0000905 return 0;
906}
907
908/// MapInFilePages - Not yet implemented on win32.
Rafael Espindolab78e2ae2011-03-10 18:30:48 +0000909void Path::UnMapFilePages(const char *Base, size_t FileSize) {
Chris Lattner799ed102008-04-01 06:00:12 +0000910 assert(0 && "NOT IMPLEMENTED");
911}
912
Reid Spencerb016a372004-09-15 05:49:50 +0000913}
Reid Spencercbad7012004-09-11 04:59:30 +0000914}