blob: 4760dfd93d6d66a349ae3aaaaad80862af9fe04b [file] [log] [blame]
Argyrios Kyrtzidis4bd32252008-06-16 10:14:09 +00001//===- llvm/System/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
Reid Spencerd0c9e0e2004-09-18 19:29:16 +000019#include "Win32.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
Michael J. Spencera9cf7b82010-10-20 15:23:58 +000069// push_back 0 on create, and pop_back on delete.
70struct ScopedNullTerminator {
71 std::string &str;
72 ScopedNullTerminator(std::string &s) : str(s) { str.push_back(0); }
Michael J. Spencer9f608b12010-10-20 16:00:45 +000073 ~ScopedNullTerminator() {
74 // str.pop_back(); But wait, C++03 doesn't have this...
75 assert(!str.empty() && str[str.size() - 1] == 0
76 && "Null char not present!");
77 str.resize(str.size() - 1);
78 }
Michael J. Spencera9cf7b82010-10-20 15:23:58 +000079};
80
Reid Spencerb016a372004-09-15 05:49:50 +000081bool
Reid Spencer07adb282004-11-05 22:15:36 +000082Path::isValid() const {
Reid Spencerb016a372004-09-15 05:49:50 +000083 if (path.empty())
84 return false;
Reid Spencerd0c9e0e2004-09-18 19:29:16 +000085
Reid Spencer6a0ec6f2004-09-29 00:01:17 +000086 // If there is a colon, it must be the second character, preceded by a letter
87 // and followed by something.
88 size_t len = path.size();
Michael J. Spencera9cf7b82010-10-20 15:23:58 +000089 // This code assumes that path is null terminated, so make sure it is.
90 ScopedNullTerminator snt(path);
Reid Spencer6a0ec6f2004-09-29 00:01:17 +000091 size_t pos = path.rfind(':',len);
Jeff Cohen8f0e8f22005-07-08 04:50:08 +000092 size_t rootslash = 0;
Reid Spencer6a0ec6f2004-09-29 00:01:17 +000093 if (pos != std::string::npos) {
94 if (pos != 1 || !isalpha(path[0]) || len < 3)
95 return false;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +000096 rootslash = 2;
97 }
Jeff Cohen85c716f2005-07-08 05:02:13 +000098
Jeff Cohen8f0e8f22005-07-08 04:50:08 +000099 // Look for a UNC path, and if found adjust our notion of the root slash.
100 if (len > 3 && path[0] == '/' && path[1] == '/') {
101 rootslash = path.find('/', 2);
102 if (rootslash == std::string::npos)
103 rootslash = 0;
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000104 }
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000105
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000106 // Check for illegal characters.
107 if (path.find_first_of("\\<>\"|\001\002\003\004\005\006\007\010\011\012"
108 "\013\014\015\016\017\020\021\022\023\024\025\026"
109 "\027\030\031\032\033\034\035\036\037")
110 != std::string::npos)
111 return false;
Jeff Cohen85c716f2005-07-08 05:02:13 +0000112
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000113 // Remove trailing slash, unless it's a root slash.
114 if (len > rootslash+1 && path[len-1] == '/')
115 path.erase(--len);
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000116
Jeff Cohen3bbbcc12005-01-14 04:09:39 +0000117 // Check each component for legality.
118 for (pos = 0; pos < len; ++pos) {
119 // A component may not end in a space.
120 if (path[pos] == ' ') {
121 if (path[pos+1] == '/' || path[pos+1] == '\0')
122 return false;
123 }
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000124
Jeff Cohen3bbbcc12005-01-14 04:09:39 +0000125 // A component may not end in a period.
126 if (path[pos] == '.') {
127 if (path[pos+1] == '/' || path[pos+1] == '\0') {
128 // Unless it is the pseudo-directory "."...
129 if (pos == 0 || path[pos-1] == '/' || path[pos-1] == ':')
130 return true;
131 // or "..".
132 if (pos > 0 && path[pos-1] == '.') {
133 if (pos == 1 || path[pos-2] == '/' || path[pos-2] == ':')
134 return true;
135 }
136 return false;
137 }
138 }
139 }
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000140
141 return true;
Reid Spencercbad7012004-09-11 04:59:30 +0000142}
143
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000144void Path::makeAbsolute() {
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000145 TCHAR FullPath[MAX_PATH + 1] = {0};
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000146 LPTSTR FilePart = NULL;
147
148 DWORD RetLength = ::GetFullPathNameA(path.c_str(),
149 sizeof(FullPath)/sizeof(FullPath[0]),
150 FullPath, &FilePart);
151
152 if (0 == RetLength) {
153 // FIXME: Report the error GetLastError()
Daniel Dunbar2749b3e2009-07-26 21:16:42 +0000154 assert(0 && "Unable to make absolute path!");
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000155 } else if (RetLength > MAX_PATH) {
156 // FIXME: Report too small buffer (needed RetLength bytes).
Daniel Dunbar2749b3e2009-07-26 21:16:42 +0000157 assert(0 && "Unable to make absolute path!");
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000158 } else {
159 path = FullPath;
160 }
161}
162
Chris Lattner88d7e402009-06-15 04:17:07 +0000163bool
164Path::isAbsolute(const char *NameStart, unsigned NameLen) {
165 assert(NameStart);
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000166 // FIXME: This does not handle correctly an absolute path starting from
167 // a drive letter or in UNC format.
Chris Lattner88d7e402009-06-15 04:17:07 +0000168 switch (NameLen) {
169 case 0:
170 return false;
171 case 1:
172 case 2:
173 return NameStart[0] == '/';
174 default:
Mikhail Glushenkov0f2ec152010-11-02 20:32:31 +0000175 return
176 (NameStart[0] == '/' || (NameStart[1] == ':' && NameStart[2] == '/')) ||
177 (NameStart[0] == '\\' || (NameStart[1] == ':' && NameStart[2] == '\\'));
Chris Lattner88d7e402009-06-15 04:17:07 +0000178 }
179}
180
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000181bool
Reid Spencer69cce812007-03-29 16:43:20 +0000182Path::isAbsolute() const {
Daniel Dunbara00e85c2009-07-12 20:23:56 +0000183 // FIXME: This does not handle correctly an absolute path starting from
184 // a drive letter or in UNC format.
Jeff Cohen84892be2007-03-29 17:27:38 +0000185 switch (path.length()) {
186 case 0:
187 return false;
188 case 1:
189 case 2:
190 return path[0] == '/';
191 default:
192 return path[0] == '/' || (path[1] == ':' && path[2] == '/');
193 }
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000194}
Reid Spencer69cce812007-03-29 16:43:20 +0000195
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000196static Path *TempDirectory;
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000197
Reid Spencerb016a372004-09-15 05:49:50 +0000198Path
Reid Spencercab0e432006-08-22 22:46:39 +0000199Path::GetTemporaryDirectory(std::string* ErrMsg) {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000200 if (TempDirectory)
201 return *TempDirectory;
202
203 char pathname[MAX_PATH];
Reid Spencercab0e432006-08-22 22:46:39 +0000204 if (!GetTempPath(MAX_PATH, pathname)) {
205 if (ErrMsg)
206 *ErrMsg = "Can't determine temporary directory";
207 return Path();
208 }
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000209
Reid Spencerb016a372004-09-15 05:49:50 +0000210 Path result;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000211 result.set(pathname);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000212
213 // Append a subdirectory passed on our process id so multiple LLVMs don't
214 // step on each other's toes.
Jeff Cohend41b30d2006-11-05 19:31:28 +0000215#ifdef __MINGW32__
216 // Mingw's Win32 header files are broken.
Reid Spencerab4d9b02006-06-08 18:08:43 +0000217 sprintf(pathname, "LLVM_%u", unsigned(GetCurrentProcessId()));
Jeff Cohend41b30d2006-11-05 19:31:28 +0000218#else
219 sprintf(pathname, "LLVM_%u", GetCurrentProcessId());
220#endif
Jeff Cohenedb9d6b2005-07-08 02:48:42 +0000221 result.appendComponent(pathname);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000222
223 // If there's a directory left over from a previous LLVM execution that
224 // happened to have the same process id, get rid of it.
Reid Spencera229c5c2005-07-08 03:08:58 +0000225 result.eraseFromDisk(true);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000226
227 // And finally (re-)create the empty directory.
Reid Spencera229c5c2005-07-08 03:08:58 +0000228 result.createDirectoryOnDisk(false);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000229 TempDirectory = new Path(result);
230 return *TempDirectory;
Reid Spencerb016a372004-09-15 05:49:50 +0000231}
232
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000233// FIXME: the following set of functions don't map to Windows very well.
Reid Spencerb016a372004-09-15 05:49:50 +0000234Path
235Path::GetRootDirectory() {
Michael J. Spencer87e06972010-11-09 15:10:45 +0000236 // This is the only notion that that Windows has of a root directory. Nothing
237 // is here except for drives.
238 return Path("file:///");
Reid Spencerb016a372004-09-15 05:49:50 +0000239}
240
Jeff Cohen85c716f2005-07-08 05:02:13 +0000241void
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000242Path::GetSystemLibraryPaths(std::vector<sys::Path>& Paths) {
Michael J. Spencereebe9702010-11-09 15:11:19 +0000243 char buff[MAX_PATH];
244 // Generic form of C:\Windows\System32
245 HRESULT res = SHGetFolderPathA(NULL,
246 CSIDL_FLAG_CREATE | CSIDL_SYSTEM,
247 NULL,
248 SHGFP_TYPE_CURRENT,
249 buff);
250 if (res != S_OK) {
251 assert(0 && "Failed to get system directory");
252 return;
253 }
254 Paths.push_back(sys::Path(buff));
255
256 // Reset buff.
257 buff[0] = 0;
258 // Generic form of C:\Windows
259 res = SHGetFolderPathA(NULL,
260 CSIDL_FLAG_CREATE | CSIDL_WINDOWS,
261 NULL,
262 SHGFP_TYPE_CURRENT,
263 buff);
264 if (res != S_OK) {
265 assert(0 && "Failed to get windows directory");
266 return;
267 }
268 Paths.push_back(sys::Path(buff));
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000269}
270
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000271void
Gabor Greifdb5565a2007-07-06 20:28:40 +0000272Path::GetBitcodeLibraryPaths(std::vector<sys::Path>& Paths) {
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000273 char * env_var = getenv("LLVM_LIB_SEARCH_PATH");
274 if (env_var != 0) {
275 getPathList(env_var,Paths);
276 }
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000277#ifdef LLVM_LIBDIR
278 {
279 Path tmpPath;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000280 if (tmpPath.set(LLVM_LIBDIR))
Reid Spencerc7f08322005-07-07 18:21:42 +0000281 if (tmpPath.canRead())
Reid Spencer6c4b7bd2004-12-13 03:03:42 +0000282 Paths.push_back(tmpPath);
283 }
284#endif
285 GetSystemLibraryPaths(Paths);
Reid Spencerb016a372004-09-15 05:49:50 +0000286}
287
288Path
289Path::GetLLVMDefaultConfigDir() {
Michael J. Spencer525ba572010-11-09 15:11:42 +0000290 Path ret = GetUserHomeDirectory();
291 if(!ret.appendComponent(".llvm"))
292 assert(0 && "Failed to append .llvm");
293 return ret;
Reid Spencerb016a372004-09-15 05:49:50 +0000294}
295
296Path
Reid Spencerb016a372004-09-15 05:49:50 +0000297Path::GetUserHomeDirectory() {
Michael J. Spencer05283c22010-11-09 15:11:31 +0000298 char buff[MAX_PATH];
299 HRESULT res = SHGetFolderPathA(NULL,
300 CSIDL_FLAG_CREATE | CSIDL_APPDATA,
301 NULL,
302 SHGFP_TYPE_CURRENT,
303 buff);
304 if (res != S_OK)
305 assert(0 && "Failed to get user home directory");
306 return Path(buff);
Reid Spencerb016a372004-09-15 05:49:50 +0000307}
Ted Kremenek79200782007-12-18 22:07:33 +0000308
309Path
310Path::GetCurrentDirectory() {
311 char pathname[MAX_PATH];
Anton Korobeynikov64ddbe42007-12-22 14:26:49 +0000312 ::GetCurrentDirectoryA(MAX_PATH,pathname);
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000313 return Path(pathname);
Ted Kremenek79200782007-12-18 22:07:33 +0000314}
315
Chris Lattner1a091442008-03-03 02:55:43 +0000316/// GetMainExecutable - Return the path to the main executable, given the
317/// value of argv[0] from program startup.
318Path Path::GetMainExecutable(const char *argv0, void *MainAddr) {
Chris Lattnerc5718d02009-06-15 05:38:04 +0000319 char pathname[MAX_PATH];
320 DWORD ret = ::GetModuleFileNameA(NULL, pathname, MAX_PATH);
321 return ret != MAX_PATH ? Path(pathname) : Path();
Chris Lattner1a091442008-03-03 02:55:43 +0000322}
323
Ted Kremenek79200782007-12-18 22:07:33 +0000324
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000325// FIXME: the above set of functions don't map to Windows very well.
326
Jeff Cohen966fa412005-07-09 18:42:49 +0000327
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000328StringRef Path::getDirname() const {
329 return getDirnameCharSep(path, "/");
Ted Kremenek9b01cc02008-04-07 22:01:32 +0000330}
Ted Kremenekcf55c8e2008-04-07 21:53:57 +0000331
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000332StringRef
Reid Spencer07adb282004-11-05 22:15:36 +0000333Path::getBasename() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000334 // Find the last slash
335 size_t slash = path.rfind('/');
336 if (slash == std::string::npos)
337 slash = 0;
338 else
339 slash++;
340
Jeff Cohen966fa412005-07-09 18:42:49 +0000341 size_t dot = path.rfind('.');
342 if (dot == std::string::npos || dot < slash)
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000343 return StringRef(path).substr(slash);
Jeff Cohen966fa412005-07-09 18:42:49 +0000344 else
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000345 return StringRef(path).substr(slash, dot - slash);
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000346}
347
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000348StringRef
Argyrios Kyrtzidisfc199882008-06-15 15:15:19 +0000349Path::getSuffix() const {
350 // Find the last slash
351 size_t slash = path.rfind('/');
352 if (slash == std::string::npos)
353 slash = 0;
354 else
355 slash++;
356
357 size_t dot = path.rfind('.');
358 if (dot == std::string::npos || dot < slash)
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000359 return StringRef("");
Argyrios Kyrtzidisfc199882008-06-15 15:15:19 +0000360 else
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000361 return StringRef(path).substr(dot + 1);
Argyrios Kyrtzidisfc199882008-06-15 15:15:19 +0000362}
363
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000364bool
Reid Spencerb016a372004-09-15 05:49:50 +0000365Path::exists() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000366 DWORD attr = GetFileAttributes(path.c_str());
367 return attr != INVALID_FILE_ATTRIBUTES;
Reid Spencerb016a372004-09-15 05:49:50 +0000368}
369
370bool
Ted Kremenekfd527112007-12-18 19:46:22 +0000371Path::isDirectory() const {
372 DWORD attr = GetFileAttributes(path.c_str());
373 return (attr != INVALID_FILE_ATTRIBUTES) &&
374 (attr & FILE_ATTRIBUTE_DIRECTORY);
375}
376
377bool
Rafael Espindola9e07a142010-11-07 04:36:50 +0000378Path::isSymLink() const {
379 return false;
380}
381
382bool
Reid Spencerc7f08322005-07-07 18:21:42 +0000383Path::canRead() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000384 // FIXME: take security attributes into account.
385 DWORD attr = GetFileAttributes(path.c_str());
386 return attr != INVALID_FILE_ATTRIBUTES;
Reid Spencerb016a372004-09-15 05:49:50 +0000387}
388
389bool
Reid Spencerc7f08322005-07-07 18:21:42 +0000390Path::canWrite() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000391 // FIXME: take security attributes into account.
392 DWORD attr = GetFileAttributes(path.c_str());
393 return (attr != INVALID_FILE_ATTRIBUTES) && !(attr & FILE_ATTRIBUTE_READONLY);
Reid Spencerb016a372004-09-15 05:49:50 +0000394}
395
396bool
Reid Spencerc7f08322005-07-07 18:21:42 +0000397Path::canExecute() const {
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000398 // FIXME: take security attributes into account.
399 DWORD attr = GetFileAttributes(path.c_str());
400 return attr != INVALID_FILE_ATTRIBUTES;
Reid Spencerb016a372004-09-15 05:49:50 +0000401}
402
Edward O'Callaghand41e9442009-11-24 15:19:10 +0000403bool
Edward O'Callaghane49a8e42009-11-25 06:32:19 +0000404Path::isRegularFile() const {
405 if (isDirectory())
406 return false;
407 return true;
Edward O'Callaghand41e9442009-11-24 15:19:10 +0000408}
409
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000410StringRef
Reid Spencerb016a372004-09-15 05:49:50 +0000411Path::getLast() const {
412 // Find the last slash
413 size_t pos = path.rfind('/');
414
415 // Handle the corner cases
416 if (pos == std::string::npos)
417 return path;
418
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000419 // If the last character is a slash, we have a root directory
420 if (pos == path.length()-1)
421 return path;
422
Reid Spencerb016a372004-09-15 05:49:50 +0000423 // Return everything after the last slash
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000424 return StringRef(path).substr(pos+1);
Reid Spencerb016a372004-09-15 05:49:50 +0000425}
426
Reid Spencer8475ec02007-03-29 19:05:44 +0000427const FileStatus *
Reid Spencer2ae9d112007-04-07 18:52:17 +0000428PathWithStatus::getFileStatus(bool update, std::string *ErrStr) const {
429 if (!fsIsValid || update) {
Reid Spencer69cce812007-03-29 16:43:20 +0000430 WIN32_FILE_ATTRIBUTE_DATA fi;
Reid Spencer8475ec02007-03-29 19:05:44 +0000431 if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
432 MakeErrMsg(ErrStr, "getStatusInfo():" + std::string(path) +
Reid Spencer69cce812007-03-29 16:43:20 +0000433 ": Can't get status: ");
Reid Spencer8475ec02007-03-29 19:05:44 +0000434 return 0;
435 }
Jeff Cohen626e38e2004-12-14 05:26:43 +0000436
Reid Spencer2ae9d112007-04-07 18:52:17 +0000437 status.fileSize = fi.nFileSizeHigh;
438 status.fileSize <<= sizeof(fi.nFileSizeHigh)*8;
439 status.fileSize += fi.nFileSizeLow;
Jeff Cohen626e38e2004-12-14 05:26:43 +0000440
Reid Spencer2ae9d112007-04-07 18:52:17 +0000441 status.mode = fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY ? 0555 : 0777;
442 status.user = 9999; // Not applicable to Windows, so...
443 status.group = 9999; // Not applicable to Windows, so...
Jeff Cohen626e38e2004-12-14 05:26:43 +0000444
Reid Spencer4031bef2007-03-29 17:00:31 +0000445 // FIXME: this is only unique if the file is accessed by the same file path.
446 // How do we do this for C:\dir\file and ..\dir\file ? Unix has inode
447 // numbers, but the concept doesn't exist in Windows.
Reid Spencer2ae9d112007-04-07 18:52:17 +0000448 status.uniqueID = 0;
Reid Spencer4031bef2007-03-29 17:00:31 +0000449 for (unsigned i = 0; i < path.length(); ++i)
Reid Spencer2ae9d112007-04-07 18:52:17 +0000450 status.uniqueID += path[i];
Reid Spencer4031bef2007-03-29 17:00:31 +0000451
Michael J. Spencer44edb0b2010-08-28 16:39:32 +0000452 ULARGE_INTEGER ui;
453 ui.LowPart = fi.ftLastWriteTime.dwLowDateTime;
454 ui.HighPart = fi.ftLastWriteTime.dwHighDateTime;
455 status.modTime.fromWin32Time(ui.QuadPart);
Reid Spencer69cce812007-03-29 16:43:20 +0000456
Reid Spencer2ae9d112007-04-07 18:52:17 +0000457 status.isDir = fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
458 fsIsValid = true;
Reid Spencer69cce812007-03-29 16:43:20 +0000459 }
Reid Spencer2ae9d112007-04-07 18:52:17 +0000460 return &status;
Jeff Cohen626e38e2004-12-14 05:26:43 +0000461}
462
Reid Spencera34a1572006-08-22 23:54:35 +0000463bool Path::makeReadableOnDisk(std::string* ErrMsg) {
Jeff Cohen626e38e2004-12-14 05:26:43 +0000464 // All files are readable on Windows (ignoring security attributes).
Reid Spencera34a1572006-08-22 23:54:35 +0000465 return false;
Reid Spencer77cc91d2004-12-13 19:59:50 +0000466}
467
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000468bool Path::makeWriteableOnDisk(std::string* ErrMsg) {
Jeff Cohen626e38e2004-12-14 05:26:43 +0000469 DWORD attr = GetFileAttributes(path.c_str());
470
471 // If it doesn't exist, we're done.
472 if (attr == INVALID_FILE_ATTRIBUTES)
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000473 return false;
Jeff Cohen626e38e2004-12-14 05:26:43 +0000474
475 if (attr & FILE_ATTRIBUTE_READONLY) {
Reid Spencera34a1572006-08-22 23:54:35 +0000476 if (!SetFileAttributes(path.c_str(), attr & ~FILE_ATTRIBUTE_READONLY)) {
477 MakeErrMsg(ErrMsg, std::string(path) + ": Can't make file writable: ");
478 return true;
479 }
Jeff Cohen626e38e2004-12-14 05:26:43 +0000480 }
Reid Spencera34a1572006-08-22 23:54:35 +0000481 return false;
Reid Spencer77cc91d2004-12-13 19:59:50 +0000482}
483
Reid Spencera34a1572006-08-22 23:54:35 +0000484bool Path::makeExecutableOnDisk(std::string* ErrMsg) {
Jeff Cohen626e38e2004-12-14 05:26:43 +0000485 // All files are executable on Windows (ignoring security attributes).
Reid Spencera34a1572006-08-22 23:54:35 +0000486 return false;
Reid Spencer77cc91d2004-12-13 19:59:50 +0000487}
488
Reid Spencerb016a372004-09-15 05:49:50 +0000489bool
Reid Spencer142ca8e2006-08-23 06:56:27 +0000490Path::getDirectoryContents(std::set<Path>& result, std::string* ErrMsg) const {
Jeff Cohen31102892007-04-07 20:47:27 +0000491 WIN32_FILE_ATTRIBUTE_DATA fi;
492 if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
493 MakeErrMsg(ErrMsg, path + ": can't get status of file");
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000494 return true;
Jeff Cohen31102892007-04-07 20:47:27 +0000495 }
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000496
Jeff Cohen31102892007-04-07 20:47:27 +0000497 if (!(fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
498 if (ErrMsg)
499 *ErrMsg = path + ": not a directory";
Reid Spencer142ca8e2006-08-23 06:56:27 +0000500 return true;
501 }
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000502
503 result.clear();
504 WIN32_FIND_DATA fd;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000505 std::string searchpath = path;
506 if (path.size() == 0 || searchpath[path.size()-1] == '/')
Jeff Cohen85c716f2005-07-08 05:02:13 +0000507 searchpath += "*";
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000508 else
509 searchpath += "/*";
Jeff Cohen85c716f2005-07-08 05:02:13 +0000510
Jeff Cohen9437bb62005-01-27 03:49:03 +0000511 HANDLE h = FindFirstFile(searchpath.c_str(), &fd);
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000512 if (h == INVALID_HANDLE_VALUE) {
Jeff Cohen9437bb62005-01-27 03:49:03 +0000513 if (GetLastError() == ERROR_FILE_NOT_FOUND)
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000514 return true; // not really an error, now is it?
Reid Spencer142ca8e2006-08-23 06:56:27 +0000515 MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
516 return true;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000517 }
518
519 do {
Jeff Cohend40a7de2004-12-31 05:07:26 +0000520 if (fd.cFileName[0] == '.')
521 continue;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000522 Path aPath(path);
523 aPath.appendComponent(&fd.cFileName[0]);
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000524 result.insert(aPath);
525 } while (FindNextFile(h, &fd));
526
Jeff Cohen51b8d212004-12-31 19:01:08 +0000527 DWORD err = GetLastError();
528 FindClose(h);
529 if (err != ERROR_NO_MORE_FILES) {
530 SetLastError(err);
Reid Spencer142ca8e2006-08-23 06:56:27 +0000531 MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
532 return true;
Jeff Cohen51b8d212004-12-31 19:01:08 +0000533 }
Reid Spencer142ca8e2006-08-23 06:56:27 +0000534 return false;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000535}
536
537bool
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000538Path::set(StringRef a_path) {
Dan Gohman30359592008-01-29 13:02:09 +0000539 if (a_path.empty())
Reid Spencerb016a372004-09-15 05:49:50 +0000540 return false;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000541 std::string save(path);
Reid Spencerb016a372004-09-15 05:49:50 +0000542 path = a_path;
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000543 FlipBackSlashes(path);
Reid Spencer07adb282004-11-05 22:15:36 +0000544 if (!isValid()) {
Reid Spencer1cf2d042005-07-07 23:35:23 +0000545 path = save;
Reid Spencerb016a372004-09-15 05:49:50 +0000546 return false;
547 }
548 return true;
549}
550
551bool
Jeffrey Yasskin88cd3582009-12-17 21:02:39 +0000552Path::appendComponent(StringRef name) {
Reid Spencer1cf2d042005-07-07 23:35:23 +0000553 if (name.empty())
Reid Spencerb016a372004-09-15 05:49:50 +0000554 return false;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000555 std::string save(path);
556 if (!path.empty()) {
557 size_t last = path.size() - 1;
Jeff Cohen85c716f2005-07-08 05:02:13 +0000558 if (path[last] != '/')
Reid Spencer1cf2d042005-07-07 23:35:23 +0000559 path += '/';
560 }
561 path += name;
Reid Spencer07adb282004-11-05 22:15:36 +0000562 if (!isValid()) {
Reid Spencer1cf2d042005-07-07 23:35:23 +0000563 path = save;
Reid Spencerb016a372004-09-15 05:49:50 +0000564 return false;
565 }
566 return true;
567}
568
569bool
Reid Spencer1cf2d042005-07-07 23:35:23 +0000570Path::eraseComponent() {
Reid Spencerb016a372004-09-15 05:49:50 +0000571 size_t slashpos = path.rfind('/',path.size());
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000572 if (slashpos == path.size() - 1 || slashpos == std::string::npos)
Reid Spencerb016a372004-09-15 05:49:50 +0000573 return false;
Jeff Cohen966fa412005-07-09 18:42:49 +0000574 std::string save(path);
Reid Spencerb016a372004-09-15 05:49:50 +0000575 path.erase(slashpos);
Jeff Cohen966fa412005-07-09 18:42:49 +0000576 if (!isValid()) {
577 path = save;
578 return false;
579 }
Reid Spencerb016a372004-09-15 05:49:50 +0000580 return true;
581}
582
583bool
Reid Spencer1cf2d042005-07-07 23:35:23 +0000584Path::eraseSuffix() {
Reid Spencerb016a372004-09-15 05:49:50 +0000585 size_t dotpos = path.rfind('.',path.size());
586 size_t slashpos = path.rfind('/',path.size());
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000587 if (dotpos != std::string::npos) {
Jeff Cohen966fa412005-07-09 18:42:49 +0000588 if (slashpos == std::string::npos || dotpos > slashpos+1) {
589 std::string save(path);
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000590 path.erase(dotpos, path.size()-dotpos);
Jeff Cohen966fa412005-07-09 18:42:49 +0000591 if (!isValid()) {
592 path = save;
593 return false;
594 }
Jeff Cohen85c716f2005-07-08 05:02:13 +0000595 return true;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000596 }
Reid Spencerb016a372004-09-15 05:49:50 +0000597 }
598 return false;
599}
600
Reid Spencer30300992006-08-24 18:58:37 +0000601inline bool PathMsg(std::string* ErrMsg, const char* pathname, const char*msg) {
602 if (ErrMsg)
603 *ErrMsg = std::string(pathname) + ": " + std::string(msg);
604 return true;
605}
606
Reid Spencerb016a372004-09-15 05:49:50 +0000607bool
Reid Spencer30300992006-08-24 18:58:37 +0000608Path::createDirectoryOnDisk(bool create_parents, std::string* ErrMsg) {
Reid Spencerb016a372004-09-15 05:49:50 +0000609 // Get a writeable copy of the path name
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000610 size_t len = path.length();
611 char *pathname = reinterpret_cast<char *>(_alloca(len+2));
612 path.copy(pathname, len);
613 pathname[len] = 0;
Jeff Cohen85c716f2005-07-08 05:02:13 +0000614
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000615 // Make sure it ends with a slash.
616 if (len == 0 || pathname[len - 1] != '/') {
617 pathname[len] = '/';
618 pathname[++len] = 0;
619 }
Reid Spencerb016a372004-09-15 05:49:50 +0000620
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000621 // Determine starting point for initial / search.
622 char *next = pathname;
623 if (pathname[0] == '/' && pathname[1] == '/') {
624 // Skip host name.
625 next = strchr(pathname+2, '/');
626 if (next == NULL)
Reid Spencer30300992006-08-24 18:58:37 +0000627 return PathMsg(ErrMsg, pathname, "badly formed remote directory");
628
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000629 // Skip share name.
630 next = strchr(next+1, '/');
631 if (next == NULL)
Reid Spencer30300992006-08-24 18:58:37 +0000632 return PathMsg(ErrMsg, pathname,"badly formed remote directory");
633
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000634 next++;
635 if (*next == 0)
Reid Spencer30300992006-08-24 18:58:37 +0000636 return PathMsg(ErrMsg, pathname, "badly formed remote directory");
637
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000638 } else {
639 if (pathname[1] == ':')
640 next += 2; // skip drive letter
641 if (*next == '/')
642 next++; // skip root directory
643 }
Reid Spencerb016a372004-09-15 05:49:50 +0000644
645 // If we're supposed to create intermediate directories
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000646 if (create_parents) {
Reid Spencerb016a372004-09-15 05:49:50 +0000647 // Loop through the directory components until we're done
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000648 while (*next) {
649 next = strchr(next, '/');
Reid Spencerb016a372004-09-15 05:49:50 +0000650 *next = 0;
Benjamin Kramere9684c62009-11-05 14:32:40 +0000651 if (!CreateDirectory(pathname, NULL) &&
652 GetLastError() != ERROR_ALREADY_EXISTS)
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000653 return MakeErrMsg(ErrMsg,
Reid Spencer30300992006-08-24 18:58:37 +0000654 std::string(pathname) + ": Can't create directory: ");
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000655 *next++ = '/';
Reid Spencerb016a372004-09-15 05:49:50 +0000656 }
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000657 } else {
658 // Drop trailing slash.
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000659 pathname[len-1] = 0;
Benjamin Kramere9684c62009-11-05 14:32:40 +0000660 if (!CreateDirectory(pathname, NULL) &&
661 GetLastError() != ERROR_ALREADY_EXISTS) {
Mikhail Glushenkov0f2ec152010-11-02 20:32:31 +0000662 return MakeErrMsg(ErrMsg, std::string(pathname) +
663 ": Can't create directory: ");
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000664 }
Reid Spencerb016a372004-09-15 05:49:50 +0000665 }
Reid Spencer30300992006-08-24 18:58:37 +0000666 return false;
Reid Spencerb016a372004-09-15 05:49:50 +0000667}
668
669bool
Reid Spencer30300992006-08-24 18:58:37 +0000670Path::createFileOnDisk(std::string* ErrMsg) {
Reid Spencerb016a372004-09-15 05:49:50 +0000671 // Create the file
Reid Spencer6a0ec6f2004-09-29 00:01:17 +0000672 HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000673 FILE_ATTRIBUTE_NORMAL, NULL);
674 if (h == INVALID_HANDLE_VALUE)
Reid Spencer30300992006-08-24 18:58:37 +0000675 return MakeErrMsg(ErrMsg, path + ": Can't create file: ");
Reid Spencerb016a372004-09-15 05:49:50 +0000676
Reid Spencerd0c9e0e2004-09-18 19:29:16 +0000677 CloseHandle(h);
Reid Spencer30300992006-08-24 18:58:37 +0000678 return false;
Reid Spencerb016a372004-09-15 05:49:50 +0000679}
680
681bool
Chris Lattner0c332312006-07-28 22:29:50 +0000682Path::eraseFromDisk(bool remove_contents, std::string *ErrStr) const {
Jeff Cohen31102892007-04-07 20:47:27 +0000683 WIN32_FILE_ATTRIBUTE_DATA fi;
684 if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi))
685 return true;
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000686
Jeff Cohen31102892007-04-07 20:47:27 +0000687 if (fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
Reid Spencer1cf2d042005-07-07 23:35:23 +0000688 // If it doesn't exist, we're done.
Jeff Cohen85c716f2005-07-08 05:02:13 +0000689 if (!exists())
Chris Lattner0c332312006-07-28 22:29:50 +0000690 return false;
Reid Spencer1cf2d042005-07-07 23:35:23 +0000691
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000692 char *pathname = reinterpret_cast<char *>(_alloca(path.length()+3));
Reid Spencer1cf2d042005-07-07 23:35:23 +0000693 int lastchar = path.length() - 1 ;
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000694 path.copy(pathname, lastchar+1);
Reid Spencer1cf2d042005-07-07 23:35:23 +0000695
696 // Make path end with '/*'.
Jeff Cohen8f0e8f22005-07-08 04:50:08 +0000697 if (pathname[lastchar] != '/')
698 pathname[++lastchar] = '/';
Reid Spencer1cf2d042005-07-07 23:35:23 +0000699 pathname[lastchar+1] = '*';
700 pathname[lastchar+2] = 0;
701
702 if (remove_contents) {
703 WIN32_FIND_DATA fd;
704 HANDLE h = FindFirstFile(pathname, &fd);
705
706 // It's a bad idea to alter the contents of a directory while enumerating
707 // its contents. So build a list of its contents first, then destroy them.
708
709 if (h != INVALID_HANDLE_VALUE) {
710 std::vector<Path> list;
711
712 do {
713 if (strcmp(fd.cFileName, ".") == 0)
714 continue;
715 if (strcmp(fd.cFileName, "..") == 0)
716 continue;
717
Jeff Cohen85c716f2005-07-08 05:02:13 +0000718 Path aPath(path);
719 aPath.appendComponent(&fd.cFileName[0]);
Reid Spencer1cf2d042005-07-07 23:35:23 +0000720 list.push_back(aPath);
721 } while (FindNextFile(h, &fd));
722
723 DWORD err = GetLastError();
724 FindClose(h);
725 if (err != ERROR_NO_MORE_FILES) {
726 SetLastError(err);
Reid Spencer05545752006-08-25 21:37:17 +0000727 return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
Reid Spencer1cf2d042005-07-07 23:35:23 +0000728 }
729
Jeff Cohen85c716f2005-07-08 05:02:13 +0000730 for (std::vector<Path>::iterator I = list.begin(); I != list.end();
Reid Spencer1cf2d042005-07-07 23:35:23 +0000731 ++I) {
732 Path &aPath = *I;
Reid Spencera229c5c2005-07-08 03:08:58 +0000733 aPath.eraseFromDisk(true);
Reid Spencer1cf2d042005-07-07 23:35:23 +0000734 }
735 } else {
736 if (GetLastError() != ERROR_FILE_NOT_FOUND)
Reid Spencer05545752006-08-25 21:37:17 +0000737 return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
Reid Spencer1cf2d042005-07-07 23:35:23 +0000738 }
739 }
740
741 pathname[lastchar] = 0;
742 if (!RemoveDirectory(pathname))
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000743 return MakeErrMsg(ErrStr,
Reid Spencer05545752006-08-25 21:37:17 +0000744 std::string(pathname) + ": Can't destroy directory: ");
Chris Lattner0c332312006-07-28 22:29:50 +0000745 return false;
Jeff Cohen31102892007-04-07 20:47:27 +0000746 } else {
747 // Read-only files cannot be deleted on Windows. Must remove the read-only
748 // attribute first.
749 if (fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
750 if (!SetFileAttributes(path.c_str(),
751 fi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
752 return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
753 }
754
755 if (!DeleteFile(path.c_str()))
756 return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
757 return false;
758 }
Reid Spencerb016a372004-09-15 05:49:50 +0000759}
760
Reid Spencer3b0cc782004-12-14 18:42:13 +0000761bool Path::getMagicNumber(std::string& Magic, unsigned len) const {
Reid Spencer3b0cc782004-12-14 18:42:13 +0000762 assert(len < 1024 && "Request for magic string too long");
Michael J. Spencer1211d432010-08-31 06:36:33 +0000763 char* buf = reinterpret_cast<char*>(alloca(len));
Jeff Cohen51b8d212004-12-31 19:01:08 +0000764
765 HANDLE h = CreateFile(path.c_str(),
766 GENERIC_READ,
767 FILE_SHARE_READ,
768 NULL,
769 OPEN_EXISTING,
770 FILE_ATTRIBUTE_NORMAL,
771 NULL);
772 if (h == INVALID_HANDLE_VALUE)
Reid Spencer3b0cc782004-12-14 18:42:13 +0000773 return false;
Jeff Cohen51b8d212004-12-31 19:01:08 +0000774
775 DWORD nRead = 0;
776 BOOL ret = ReadFile(h, buf, len, &nRead, NULL);
777 CloseHandle(h);
778
779 if (!ret || nRead != len)
Reid Spencer3b0cc782004-12-14 18:42:13 +0000780 return false;
Jeff Cohen51b8d212004-12-31 19:01:08 +0000781
Michael J. Spencer1211d432010-08-31 06:36:33 +0000782 Magic = std::string(buf, len);
Reid Spencer3b0cc782004-12-14 18:42:13 +0000783 return true;
784}
785
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000786bool
Reid Spencer5a060772006-08-23 07:30:48 +0000787Path::renamePathOnDisk(const Path& newName, std::string* ErrMsg) {
Francois Pichet3eddd982010-09-30 00:44:58 +0000788 if (!MoveFileEx(path.c_str(), newName.c_str(), MOVEFILE_REPLACE_EXISTING))
789 return MakeErrMsg(ErrMsg, "Can't move '" + path + "' to '" + newName.path
790 + "': ");
791 return false;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000792}
793
794bool
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000795Path::setStatusInfoOnDisk(const FileStatus &si, std::string *ErrMsg) const {
Jeff Cohen966fa412005-07-09 18:42:49 +0000796 // FIXME: should work on directories also.
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000797 if (!si.isFile) {
798 return true;
799 }
Kovarththanan Rajaratnam16ceb3a2010-03-12 14:17:24 +0000800
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000801 HANDLE h = CreateFile(path.c_str(),
802 FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES,
803 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
Jeff Cohend40a7de2004-12-31 05:07:26 +0000804 NULL,
805 OPEN_EXISTING,
806 FILE_ATTRIBUTE_NORMAL,
807 NULL);
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000808 if (h == INVALID_HANDLE_VALUE)
Chris Lattner1bebfb52006-07-28 22:36:17 +0000809 return true;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000810
811 BY_HANDLE_FILE_INFORMATION bhfi;
812 if (!GetFileInformationByHandle(h, &bhfi)) {
Jeff Cohen51b8d212004-12-31 19:01:08 +0000813 DWORD err = GetLastError();
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000814 CloseHandle(h);
Jeff Cohen51b8d212004-12-31 19:01:08 +0000815 SetLastError(err);
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000816 return MakeErrMsg(ErrMsg, path + ": GetFileInformationByHandle: ");
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000817 }
818
Michael J. Spencer44edb0b2010-08-28 16:39:32 +0000819 ULARGE_INTEGER ui;
820 ui.QuadPart = si.modTime.toWin32Time();
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000821 FILETIME ft;
Michael J. Spencer44edb0b2010-08-28 16:39:32 +0000822 ft.dwLowDateTime = ui.LowPart;
823 ft.dwHighDateTime = ui.HighPart;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000824 BOOL ret = SetFileTime(h, NULL, &ft, &ft);
Jeff Cohen51b8d212004-12-31 19:01:08 +0000825 DWORD err = GetLastError();
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000826 CloseHandle(h);
Jeff Cohen51b8d212004-12-31 19:01:08 +0000827 if (!ret) {
828 SetLastError(err);
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000829 return MakeErrMsg(ErrMsg, path + ": SetFileTime: ");
Jeff Cohen51b8d212004-12-31 19:01:08 +0000830 }
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000831
832 // Best we can do with Unix permission bits is to interpret the owner
833 // writable bit.
834 if (si.mode & 0200) {
835 if (bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
836 if (!SetFileAttributes(path.c_str(),
Jeff Cohend40a7de2004-12-31 05:07:26 +0000837 bhfi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000838 return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000839 }
840 } else {
841 if (!(bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) {
842 if (!SetFileAttributes(path.c_str(),
Jeff Cohend40a7de2004-12-31 05:07:26 +0000843 bhfi.dwFileAttributes | FILE_ATTRIBUTE_READONLY))
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000844 return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000845 }
846 }
847
Chris Lattner1bebfb52006-07-28 22:36:17 +0000848 return false;
Jeff Cohenebcb9b32004-12-31 04:39:07 +0000849}
850
Reid Spencer30300992006-08-24 18:58:37 +0000851bool
852CopyFile(const sys::Path &Dest, const sys::Path &Src, std::string* ErrMsg) {
Jeff Cohencb652552004-12-24 02:38:34 +0000853 // Can't use CopyFile macro defined in Windows.h because it would mess up the
854 // above line. We use the expansion it would have in a non-UNICODE build.
855 if (!::CopyFileA(Src.c_str(), Dest.c_str(), false))
Chris Lattner74382b72009-08-23 22:45:37 +0000856 return MakeErrMsg(ErrMsg, "Can't copy '" + Src.str() +
857 "' to '" + Dest.str() + "': ");
Reid Spencer30300992006-08-24 18:58:37 +0000858 return false;
Reid Spencerc29befb2004-12-15 01:50:13 +0000859}
860
Reid Spencer30300992006-08-24 18:58:37 +0000861bool
862Path::makeUnique(bool reuse_current, std::string* ErrMsg) {
Reid Spencer07f9f4e2004-12-15 08:32:45 +0000863 if (reuse_current && !exists())
Reid Spencer30300992006-08-24 18:58:37 +0000864 return false; // File doesn't exist already, just use it!
Reid Spencerc29befb2004-12-15 01:50:13 +0000865
Jeff Cohen9437bb62005-01-27 03:49:03 +0000866 // Reserve space for -XXXXXX at the end.
867 char *FNBuffer = (char*) alloca(path.size()+8);
868 unsigned offset = path.size();
869 path.copy(FNBuffer, offset);
Reid Spencerc29befb2004-12-15 01:50:13 +0000870
Jeff Cohen966fa412005-07-09 18:42:49 +0000871 // Find a numeric suffix that isn't used by an existing file. Assume there
872 // won't be more than 1 million files with the same prefix. Probably a safe
873 // bet.
Jeff Cohen9437bb62005-01-27 03:49:03 +0000874 static unsigned FCounter = 0;
875 do {
876 sprintf(FNBuffer+offset, "-%06u", FCounter);
877 if (++FCounter > 999999)
878 FCounter = 0;
879 path = FNBuffer;
880 } while (exists());
Reid Spencer30300992006-08-24 18:58:37 +0000881 return false;
Reid Spencerc29befb2004-12-15 01:50:13 +0000882}
883
Reid Spencer07f9f4e2004-12-15 08:32:45 +0000884bool
Reid Spencer30300992006-08-24 18:58:37 +0000885Path::createTemporaryFileOnDisk(bool reuse_current, std::string* ErrMsg) {
Reid Spencer07f9f4e2004-12-15 08:32:45 +0000886 // Make this into a unique file name
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000887 makeUnique(reuse_current, ErrMsg);
Jeff Cohen9437bb62005-01-27 03:49:03 +0000888
889 // Now go and create it
890 HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
891 FILE_ATTRIBUTE_NORMAL, NULL);
892 if (h == INVALID_HANDLE_VALUE)
Anton Korobeynikov7d515442006-09-01 20:35:17 +0000893 return MakeErrMsg(ErrMsg, path + ": can't create file");
Jeff Cohen9437bb62005-01-27 03:49:03 +0000894
895 CloseHandle(h);
Reid Spencer30300992006-08-24 18:58:37 +0000896 return false;
Reid Spencer07f9f4e2004-12-15 08:32:45 +0000897}
898
Chris Lattner799ed102008-04-01 06:00:12 +0000899/// MapInFilePages - Not yet implemented on win32.
900const char *Path::MapInFilePages(int FD, uint64_t FileSize) {
901 return 0;
902}
903
904/// MapInFilePages - Not yet implemented on win32.
905void Path::UnMapFilePages(const char *Base, uint64_t FileSize) {
906 assert(0 && "NOT IMPLEMENTED");
907}
908
Reid Spencerb016a372004-09-15 05:49:50 +0000909}
Reid Spencercbad7012004-09-11 04:59:30 +0000910}