blob: 573369e97d492b2f2c797a1a954d7e6a820e8197 [file] [log] [blame]
Argiris Kirtzidisfb155472008-06-16 10:14:09 +00001//===- llvm/System/Win32/Path.cpp - Win32 Path Implementation ---*- C++ -*-===//
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner081ce942007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Dan Gohmanf17a25c2007-07-18 16:29:46 +00007//
8// Modified by Henrik Bach to comply with at least MinGW.
9// Ported to Win32 by Jeff Cohen.
10//
11//===----------------------------------------------------------------------===//
12//
13// This file provides the Win32 specific implementation of the Path class.
14//
15//===----------------------------------------------------------------------===//
16
17//===----------------------------------------------------------------------===//
18//=== WARNING: Implementation here must contain only generic Win32 code that
19//=== is guaranteed to work on *all* Win32 variants.
20//===----------------------------------------------------------------------===//
21
22#include "Win32.h"
23#include <malloc.h>
Chris Lattner4b1f1e42009-04-01 02:03:38 +000024#include <cstdio>
Dan Gohmanf17a25c2007-07-18 16:29:46 +000025
26// We need to undo a macro defined in Windows.h, otherwise we won't compile:
27#undef CopyFile
Anton Korobeynikove41a4bd2007-12-22 14:26:49 +000028#undef GetCurrentDirectory
Dan Gohmanf17a25c2007-07-18 16:29:46 +000029
30// Windows happily accepts either forward or backward slashes, though any path
31// returned by a Win32 API will have backward slashes. As LLVM code basically
32// assumes forward slashes are used, backward slashs are converted where they
33// can be introduced into a path.
34//
35// Another invariant is that a path ends with a slash if and only if the path
36// is a root directory. Any other use of a trailing slash is stripped. Unlike
37// in Unix, Windows has a rather complicated notion of a root path and this
38// invariant helps simply the code.
39
40static void FlipBackSlashes(std::string& s) {
41 for (size_t i = 0; i < s.size(); i++)
42 if (s[i] == '\\')
43 s[i] = '/';
44}
45
46namespace llvm {
47namespace sys {
Chris Lattner510190e2008-03-13 05:17:59 +000048const char PathSeparator = ';';
Chris Lattner4b8f1c62008-02-27 06:17:10 +000049
Nick Lewycky4fd4b462008-05-11 17:37:40 +000050Path::Path(const std::string& p)
51 : path(p) {
52 FlipBackSlashes(path);
53}
54
55Path::Path(const char *StrStart, unsigned StrLen)
56 : path(StrStart, StrLen) {
57 FlipBackSlashes(path);
58}
59
Chris Lattner43ac42b2008-08-11 23:39:47 +000060Path&
61Path::operator=(const std::string &that) {
62 path = that;
63 FlipBackSlashes(path);
64 return *this;
65}
66
Dan Gohmanf17a25c2007-07-18 16:29:46 +000067bool
68Path::isValid() const {
69 if (path.empty())
70 return false;
71
72 // If there is a colon, it must be the second character, preceded by a letter
73 // and followed by something.
74 size_t len = path.size();
75 size_t pos = path.rfind(':',len);
76 size_t rootslash = 0;
77 if (pos != std::string::npos) {
78 if (pos != 1 || !isalpha(path[0]) || len < 3)
79 return false;
80 rootslash = 2;
81 }
82
83 // Look for a UNC path, and if found adjust our notion of the root slash.
84 if (len > 3 && path[0] == '/' && path[1] == '/') {
85 rootslash = path.find('/', 2);
86 if (rootslash == std::string::npos)
87 rootslash = 0;
88 }
89
90 // Check for illegal characters.
91 if (path.find_first_of("\\<>\"|\001\002\003\004\005\006\007\010\011\012"
92 "\013\014\015\016\017\020\021\022\023\024\025\026"
93 "\027\030\031\032\033\034\035\036\037")
94 != std::string::npos)
95 return false;
96
97 // Remove trailing slash, unless it's a root slash.
98 if (len > rootslash+1 && path[len-1] == '/')
99 path.erase(--len);
100
101 // Check each component for legality.
102 for (pos = 0; pos < len; ++pos) {
103 // A component may not end in a space.
104 if (path[pos] == ' ') {
105 if (path[pos+1] == '/' || path[pos+1] == '\0')
106 return false;
107 }
108
109 // A component may not end in a period.
110 if (path[pos] == '.') {
111 if (path[pos+1] == '/' || path[pos+1] == '\0') {
112 // Unless it is the pseudo-directory "."...
113 if (pos == 0 || path[pos-1] == '/' || path[pos-1] == ':')
114 return true;
115 // or "..".
116 if (pos > 0 && path[pos-1] == '.') {
117 if (pos == 1 || path[pos-2] == '/' || path[pos-2] == ':')
118 return true;
119 }
120 return false;
121 }
122 }
123 }
124
125 return true;
126}
127
Daniel Dunbarbf8e8712009-07-12 20:23:56 +0000128void Path::makeAbsolute() {
129 TCHAR FullPath[MAX_PATH + 1] = {0};
130 LPTSTR FilePart = NULL;
131
132 DWORD RetLength = ::GetFullPathNameA(path.c_str(),
133 sizeof(FullPath)/sizeof(FullPath[0]),
134 FullPath, &FilePart);
135
136 if (0 == RetLength) {
137 // FIXME: Report the error GetLastError()
Daniel Dunbar79d92a12009-07-26 21:16:42 +0000138 assert(0 && "Unable to make absolute path!");
Daniel Dunbarbf8e8712009-07-12 20:23:56 +0000139 } else if (RetLength > MAX_PATH) {
140 // FIXME: Report too small buffer (needed RetLength bytes).
Daniel Dunbar79d92a12009-07-26 21:16:42 +0000141 assert(0 && "Unable to make absolute path!");
Daniel Dunbarbf8e8712009-07-12 20:23:56 +0000142 } else {
143 path = FullPath;
144 }
145}
146
Chris Lattnerd024d8d2009-06-15 04:17:07 +0000147bool
148Path::isAbsolute(const char *NameStart, unsigned NameLen) {
149 assert(NameStart);
Daniel Dunbarbf8e8712009-07-12 20:23:56 +0000150 // FIXME: This does not handle correctly an absolute path starting from
151 // a drive letter or in UNC format.
Chris Lattnerd024d8d2009-06-15 04:17:07 +0000152 switch (NameLen) {
153 case 0:
154 return false;
155 case 1:
156 case 2:
157 return NameStart[0] == '/';
158 default:
Chris Lattner3209a3e2009-08-12 17:47:06 +0000159 return (NameStart[0] == '/' || (NameStart[1] == ':' && NameStart[2] == '/')) ||
160 (NameStart[0] == '\\' || (NameStart[1] == ':' && NameStart[2] == '\\'));
Chris Lattnerd024d8d2009-06-15 04:17:07 +0000161 }
162}
163
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000164bool
165Path::isAbsolute() const {
Daniel Dunbarbf8e8712009-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.
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000168 switch (path.length()) {
169 case 0:
170 return false;
171 case 1:
172 case 2:
173 return path[0] == '/';
174 default:
175 return path[0] == '/' || (path[1] == ':' && path[2] == '/');
176 }
177}
178
179static Path *TempDirectory = NULL;
180
181Path
182Path::GetTemporaryDirectory(std::string* ErrMsg) {
183 if (TempDirectory)
184 return *TempDirectory;
185
186 char pathname[MAX_PATH];
187 if (!GetTempPath(MAX_PATH, pathname)) {
188 if (ErrMsg)
189 *ErrMsg = "Can't determine temporary directory";
190 return Path();
191 }
192
193 Path result;
194 result.set(pathname);
195
196 // Append a subdirectory passed on our process id so multiple LLVMs don't
197 // step on each other's toes.
198#ifdef __MINGW32__
199 // Mingw's Win32 header files are broken.
200 sprintf(pathname, "LLVM_%u", unsigned(GetCurrentProcessId()));
201#else
202 sprintf(pathname, "LLVM_%u", GetCurrentProcessId());
203#endif
204 result.appendComponent(pathname);
205
206 // If there's a directory left over from a previous LLVM execution that
207 // happened to have the same process id, get rid of it.
208 result.eraseFromDisk(true);
209
210 // And finally (re-)create the empty directory.
211 result.createDirectoryOnDisk(false);
212 TempDirectory = new Path(result);
213 return *TempDirectory;
214}
215
216// FIXME: the following set of functions don't map to Windows very well.
217Path
218Path::GetRootDirectory() {
219 Path result;
220 result.set("C:/");
221 return result;
222}
223
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000224void
225Path::GetSystemLibraryPaths(std::vector<sys::Path>& Paths) {
226 Paths.push_back(sys::Path("C:/WINDOWS/SYSTEM32"));
227 Paths.push_back(sys::Path("C:/WINDOWS"));
228}
229
230void
231Path::GetBitcodeLibraryPaths(std::vector<sys::Path>& Paths) {
232 char * env_var = getenv("LLVM_LIB_SEARCH_PATH");
233 if (env_var != 0) {
234 getPathList(env_var,Paths);
235 }
236#ifdef LLVM_LIBDIR
237 {
238 Path tmpPath;
239 if (tmpPath.set(LLVM_LIBDIR))
240 if (tmpPath.canRead())
241 Paths.push_back(tmpPath);
242 }
243#endif
244 GetSystemLibraryPaths(Paths);
245}
246
247Path
248Path::GetLLVMDefaultConfigDir() {
249 // TODO: this isn't going to fly on Windows
250 return Path("/etc/llvm");
251}
252
253Path
254Path::GetUserHomeDirectory() {
255 // TODO: Typical Windows setup doesn't define HOME.
256 const char* home = getenv("HOME");
257 if (home) {
258 Path result;
259 if (result.set(home))
260 return result;
261 }
262 return GetRootDirectory();
263}
Ted Kremenekb05c9352007-12-18 22:07:33 +0000264
265Path
266Path::GetCurrentDirectory() {
267 char pathname[MAX_PATH];
Anton Korobeynikove41a4bd2007-12-22 14:26:49 +0000268 ::GetCurrentDirectoryA(MAX_PATH,pathname);
Ted Kremenekb05c9352007-12-18 22:07:33 +0000269 return Path(pathname);
270}
271
Chris Lattner365f2202008-03-03 02:55:43 +0000272/// GetMainExecutable - Return the path to the main executable, given the
273/// value of argv[0] from program startup.
274Path Path::GetMainExecutable(const char *argv0, void *MainAddr) {
Chris Lattner36e1e2d2009-06-15 05:38:04 +0000275 char pathname[MAX_PATH];
276 DWORD ret = ::GetModuleFileNameA(NULL, pathname, MAX_PATH);
277 return ret != MAX_PATH ? Path(pathname) : Path();
Chris Lattner365f2202008-03-03 02:55:43 +0000278}
279
Ted Kremenekb05c9352007-12-18 22:07:33 +0000280
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000281// FIXME: the above set of functions don't map to Windows very well.
282
283
284bool
285Path::isRootDirectory() const {
286 size_t len = path.size();
287 return len > 0 && path[len-1] == '/';
288}
289
Ted Kremenekb169dfa2008-04-07 22:01:32 +0000290std::string Path::getDirname() const {
Nick Lewycky4fd4b462008-05-11 17:37:40 +0000291 return getDirnameCharSep(path, '/');
Ted Kremenekb169dfa2008-04-07 22:01:32 +0000292}
Ted Kremenek4a4f5ed2008-04-07 21:53:57 +0000293
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000294std::string
295Path::getBasename() const {
296 // Find the last slash
297 size_t slash = path.rfind('/');
298 if (slash == std::string::npos)
299 slash = 0;
300 else
301 slash++;
302
303 size_t dot = path.rfind('.');
304 if (dot == std::string::npos || dot < slash)
305 return path.substr(slash);
306 else
307 return path.substr(slash, dot - slash);
308}
309
Argiris Kirtzidisc8f4a3a2008-06-15 15:15:19 +0000310std::string
311Path::getSuffix() const {
312 // Find the last slash
313 size_t slash = path.rfind('/');
314 if (slash == std::string::npos)
315 slash = 0;
316 else
317 slash++;
318
319 size_t dot = path.rfind('.');
320 if (dot == std::string::npos || dot < slash)
321 return std::string();
322 else
323 return path.substr(dot + 1);
324}
325
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000326bool
327Path::exists() const {
328 DWORD attr = GetFileAttributes(path.c_str());
329 return attr != INVALID_FILE_ATTRIBUTES;
330}
331
332bool
Ted Kremenek65149be2007-12-18 19:46:22 +0000333Path::isDirectory() const {
334 DWORD attr = GetFileAttributes(path.c_str());
335 return (attr != INVALID_FILE_ATTRIBUTES) &&
336 (attr & FILE_ATTRIBUTE_DIRECTORY);
337}
338
339bool
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000340Path::canRead() const {
341 // FIXME: take security attributes into account.
342 DWORD attr = GetFileAttributes(path.c_str());
343 return attr != INVALID_FILE_ATTRIBUTES;
344}
345
346bool
347Path::canWrite() const {
348 // FIXME: take security attributes into account.
349 DWORD attr = GetFileAttributes(path.c_str());
350 return (attr != INVALID_FILE_ATTRIBUTES) && !(attr & FILE_ATTRIBUTE_READONLY);
351}
352
353bool
354Path::canExecute() const {
355 // FIXME: take security attributes into account.
356 DWORD attr = GetFileAttributes(path.c_str());
357 return attr != INVALID_FILE_ATTRIBUTES;
358}
359
360std::string
361Path::getLast() const {
362 // Find the last slash
363 size_t pos = path.rfind('/');
364
365 // Handle the corner cases
366 if (pos == std::string::npos)
367 return path;
368
369 // If the last character is a slash, we have a root directory
370 if (pos == path.length()-1)
371 return path;
372
373 // Return everything after the last slash
374 return path.substr(pos+1);
375}
376
377const FileStatus *
378PathWithStatus::getFileStatus(bool update, std::string *ErrStr) const {
379 if (!fsIsValid || update) {
380 WIN32_FILE_ATTRIBUTE_DATA fi;
381 if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
382 MakeErrMsg(ErrStr, "getStatusInfo():" + std::string(path) +
383 ": Can't get status: ");
384 return 0;
385 }
386
387 status.fileSize = fi.nFileSizeHigh;
388 status.fileSize <<= sizeof(fi.nFileSizeHigh)*8;
389 status.fileSize += fi.nFileSizeLow;
390
391 status.mode = fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY ? 0555 : 0777;
392 status.user = 9999; // Not applicable to Windows, so...
393 status.group = 9999; // Not applicable to Windows, so...
394
395 // FIXME: this is only unique if the file is accessed by the same file path.
396 // How do we do this for C:\dir\file and ..\dir\file ? Unix has inode
397 // numbers, but the concept doesn't exist in Windows.
398 status.uniqueID = 0;
399 for (unsigned i = 0; i < path.length(); ++i)
400 status.uniqueID += path[i];
401
402 __int64 ft = *reinterpret_cast<__int64*>(&fi.ftLastWriteTime);
403 status.modTime.fromWin32Time(ft);
404
405 status.isDir = fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
406 fsIsValid = true;
407 }
408 return &status;
409}
410
411bool Path::makeReadableOnDisk(std::string* ErrMsg) {
412 // All files are readable on Windows (ignoring security attributes).
413 return false;
414}
415
416bool Path::makeWriteableOnDisk(std::string* ErrMsg) {
417 DWORD attr = GetFileAttributes(path.c_str());
418
419 // If it doesn't exist, we're done.
420 if (attr == INVALID_FILE_ATTRIBUTES)
421 return false;
422
423 if (attr & FILE_ATTRIBUTE_READONLY) {
424 if (!SetFileAttributes(path.c_str(), attr & ~FILE_ATTRIBUTE_READONLY)) {
425 MakeErrMsg(ErrMsg, std::string(path) + ": Can't make file writable: ");
426 return true;
427 }
428 }
429 return false;
430}
431
432bool Path::makeExecutableOnDisk(std::string* ErrMsg) {
433 // All files are executable on Windows (ignoring security attributes).
434 return false;
435}
436
437bool
438Path::getDirectoryContents(std::set<Path>& result, std::string* ErrMsg) const {
439 WIN32_FILE_ATTRIBUTE_DATA fi;
440 if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
441 MakeErrMsg(ErrMsg, path + ": can't get status of file");
442 return true;
443 }
444
445 if (!(fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
446 if (ErrMsg)
447 *ErrMsg = path + ": not a directory";
448 return true;
449 }
450
451 result.clear();
452 WIN32_FIND_DATA fd;
453 std::string searchpath = path;
454 if (path.size() == 0 || searchpath[path.size()-1] == '/')
455 searchpath += "*";
456 else
457 searchpath += "/*";
458
459 HANDLE h = FindFirstFile(searchpath.c_str(), &fd);
460 if (h == INVALID_HANDLE_VALUE) {
461 if (GetLastError() == ERROR_FILE_NOT_FOUND)
462 return true; // not really an error, now is it?
463 MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
464 return true;
465 }
466
467 do {
468 if (fd.cFileName[0] == '.')
469 continue;
470 Path aPath(path);
471 aPath.appendComponent(&fd.cFileName[0]);
472 result.insert(aPath);
473 } while (FindNextFile(h, &fd));
474
475 DWORD err = GetLastError();
476 FindClose(h);
477 if (err != ERROR_NO_MORE_FILES) {
478 SetLastError(err);
479 MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
480 return true;
481 }
482 return false;
483}
484
485bool
486Path::set(const std::string& a_path) {
Dan Gohman301f4052008-01-29 13:02:09 +0000487 if (a_path.empty())
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000488 return false;
489 std::string save(path);
490 path = a_path;
491 FlipBackSlashes(path);
492 if (!isValid()) {
493 path = save;
494 return false;
495 }
496 return true;
497}
498
499bool
500Path::appendComponent(const std::string& name) {
501 if (name.empty())
502 return false;
503 std::string save(path);
504 if (!path.empty()) {
505 size_t last = path.size() - 1;
506 if (path[last] != '/')
507 path += '/';
508 }
509 path += name;
510 if (!isValid()) {
511 path = save;
512 return false;
513 }
514 return true;
515}
516
517bool
518Path::eraseComponent() {
519 size_t slashpos = path.rfind('/',path.size());
520 if (slashpos == path.size() - 1 || slashpos == std::string::npos)
521 return false;
522 std::string save(path);
523 path.erase(slashpos);
524 if (!isValid()) {
525 path = save;
526 return false;
527 }
528 return true;
529}
530
531bool
532Path::appendSuffix(const std::string& suffix) {
533 std::string save(path);
534 path.append(".");
535 path.append(suffix);
536 if (!isValid()) {
537 path = save;
538 return false;
539 }
540 return true;
541}
542
543bool
544Path::eraseSuffix() {
545 size_t dotpos = path.rfind('.',path.size());
546 size_t slashpos = path.rfind('/',path.size());
547 if (dotpos != std::string::npos) {
548 if (slashpos == std::string::npos || dotpos > slashpos+1) {
549 std::string save(path);
550 path.erase(dotpos, path.size()-dotpos);
551 if (!isValid()) {
552 path = save;
553 return false;
554 }
555 return true;
556 }
557 }
558 return false;
559}
560
561inline bool PathMsg(std::string* ErrMsg, const char* pathname, const char*msg) {
562 if (ErrMsg)
563 *ErrMsg = std::string(pathname) + ": " + std::string(msg);
564 return true;
565}
566
567bool
568Path::createDirectoryOnDisk(bool create_parents, std::string* ErrMsg) {
569 // Get a writeable copy of the path name
570 size_t len = path.length();
571 char *pathname = reinterpret_cast<char *>(_alloca(len+2));
572 path.copy(pathname, len);
573 pathname[len] = 0;
574
575 // Make sure it ends with a slash.
576 if (len == 0 || pathname[len - 1] != '/') {
577 pathname[len] = '/';
578 pathname[++len] = 0;
579 }
580
581 // Determine starting point for initial / search.
582 char *next = pathname;
583 if (pathname[0] == '/' && pathname[1] == '/') {
584 // Skip host name.
585 next = strchr(pathname+2, '/');
586 if (next == NULL)
587 return PathMsg(ErrMsg, pathname, "badly formed remote directory");
588
589 // Skip share name.
590 next = strchr(next+1, '/');
591 if (next == NULL)
592 return PathMsg(ErrMsg, pathname,"badly formed remote directory");
593
594 next++;
595 if (*next == 0)
596 return PathMsg(ErrMsg, pathname, "badly formed remote directory");
597
598 } else {
599 if (pathname[1] == ':')
600 next += 2; // skip drive letter
601 if (*next == '/')
602 next++; // skip root directory
603 }
604
605 // If we're supposed to create intermediate directories
606 if (create_parents) {
607 // Loop through the directory components until we're done
608 while (*next) {
609 next = strchr(next, '/');
610 *next = 0;
Benjamin Kramerbf0e40c2009-11-05 14:32:40 +0000611 if (!CreateDirectory(pathname, NULL) &&
612 GetLastError() != ERROR_ALREADY_EXISTS)
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000613 return MakeErrMsg(ErrMsg,
614 std::string(pathname) + ": Can't create directory: ");
615 *next++ = '/';
616 }
617 } else {
618 // Drop trailing slash.
619 pathname[len-1] = 0;
Benjamin Kramerbf0e40c2009-11-05 14:32:40 +0000620 if (!CreateDirectory(pathname, NULL) &&
621 GetLastError() != ERROR_ALREADY_EXISTS) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000622 return MakeErrMsg(ErrMsg, std::string(pathname) + ": Can't create directory: ");
623 }
624 }
625 return false;
626}
627
628bool
629Path::createFileOnDisk(std::string* ErrMsg) {
630 // Create the file
631 HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
632 FILE_ATTRIBUTE_NORMAL, NULL);
633 if (h == INVALID_HANDLE_VALUE)
634 return MakeErrMsg(ErrMsg, path + ": Can't create file: ");
635
636 CloseHandle(h);
637 return false;
638}
639
640bool
641Path::eraseFromDisk(bool remove_contents, std::string *ErrStr) const {
642 WIN32_FILE_ATTRIBUTE_DATA fi;
643 if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi))
644 return true;
645
646 if (fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
647 // If it doesn't exist, we're done.
648 if (!exists())
649 return false;
650
651 char *pathname = reinterpret_cast<char *>(_alloca(path.length()+3));
652 int lastchar = path.length() - 1 ;
653 path.copy(pathname, lastchar+1);
654
655 // Make path end with '/*'.
656 if (pathname[lastchar] != '/')
657 pathname[++lastchar] = '/';
658 pathname[lastchar+1] = '*';
659 pathname[lastchar+2] = 0;
660
661 if (remove_contents) {
662 WIN32_FIND_DATA fd;
663 HANDLE h = FindFirstFile(pathname, &fd);
664
665 // It's a bad idea to alter the contents of a directory while enumerating
666 // its contents. So build a list of its contents first, then destroy them.
667
668 if (h != INVALID_HANDLE_VALUE) {
669 std::vector<Path> list;
670
671 do {
672 if (strcmp(fd.cFileName, ".") == 0)
673 continue;
674 if (strcmp(fd.cFileName, "..") == 0)
675 continue;
676
677 Path aPath(path);
678 aPath.appendComponent(&fd.cFileName[0]);
679 list.push_back(aPath);
680 } while (FindNextFile(h, &fd));
681
682 DWORD err = GetLastError();
683 FindClose(h);
684 if (err != ERROR_NO_MORE_FILES) {
685 SetLastError(err);
686 return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
687 }
688
689 for (std::vector<Path>::iterator I = list.begin(); I != list.end();
690 ++I) {
691 Path &aPath = *I;
692 aPath.eraseFromDisk(true);
693 }
694 } else {
695 if (GetLastError() != ERROR_FILE_NOT_FOUND)
696 return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
697 }
698 }
699
700 pathname[lastchar] = 0;
701 if (!RemoveDirectory(pathname))
702 return MakeErrMsg(ErrStr,
703 std::string(pathname) + ": Can't destroy directory: ");
704 return false;
705 } else {
706 // Read-only files cannot be deleted on Windows. Must remove the read-only
707 // attribute first.
708 if (fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
709 if (!SetFileAttributes(path.c_str(),
710 fi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
711 return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
712 }
713
714 if (!DeleteFile(path.c_str()))
715 return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
716 return false;
717 }
718}
719
720bool Path::getMagicNumber(std::string& Magic, unsigned len) const {
721 assert(len < 1024 && "Request for magic string too long");
722 char* buf = (char*) alloca(1 + len);
723
724 HANDLE h = CreateFile(path.c_str(),
725 GENERIC_READ,
726 FILE_SHARE_READ,
727 NULL,
728 OPEN_EXISTING,
729 FILE_ATTRIBUTE_NORMAL,
730 NULL);
731 if (h == INVALID_HANDLE_VALUE)
732 return false;
733
734 DWORD nRead = 0;
735 BOOL ret = ReadFile(h, buf, len, &nRead, NULL);
736 CloseHandle(h);
737
738 if (!ret || nRead != len)
739 return false;
740
741 buf[len] = '\0';
742 Magic = buf;
743 return true;
744}
745
746bool
747Path::renamePathOnDisk(const Path& newName, std::string* ErrMsg) {
748 if (!MoveFileEx(path.c_str(), newName.c_str(), MOVEFILE_REPLACE_EXISTING))
749 return MakeErrMsg(ErrMsg, "Can't move '" + path + "' to '" + newName.path
750 + "': ");
Nick Lewyckyc74d8f92008-05-06 03:42:21 +0000751 return false;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000752}
753
754bool
755Path::setStatusInfoOnDisk(const FileStatus &si, std::string *ErrMsg) const {
756 // FIXME: should work on directories also.
757 if (!si.isFile) {
758 return true;
759 }
760
761 HANDLE h = CreateFile(path.c_str(),
762 FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES,
763 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
764 NULL,
765 OPEN_EXISTING,
766 FILE_ATTRIBUTE_NORMAL,
767 NULL);
768 if (h == INVALID_HANDLE_VALUE)
769 return true;
770
771 BY_HANDLE_FILE_INFORMATION bhfi;
772 if (!GetFileInformationByHandle(h, &bhfi)) {
773 DWORD err = GetLastError();
774 CloseHandle(h);
775 SetLastError(err);
776 return MakeErrMsg(ErrMsg, path + ": GetFileInformationByHandle: ");
777 }
778
779 FILETIME ft;
780 (uint64_t&)ft = si.modTime.toWin32Time();
781 BOOL ret = SetFileTime(h, NULL, &ft, &ft);
782 DWORD err = GetLastError();
783 CloseHandle(h);
784 if (!ret) {
785 SetLastError(err);
786 return MakeErrMsg(ErrMsg, path + ": SetFileTime: ");
787 }
788
789 // Best we can do with Unix permission bits is to interpret the owner
790 // writable bit.
791 if (si.mode & 0200) {
792 if (bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
793 if (!SetFileAttributes(path.c_str(),
794 bhfi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
795 return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
796 }
797 } else {
798 if (!(bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) {
799 if (!SetFileAttributes(path.c_str(),
800 bhfi.dwFileAttributes | FILE_ATTRIBUTE_READONLY))
801 return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
802 }
803 }
804
805 return false;
806}
807
808bool
809CopyFile(const sys::Path &Dest, const sys::Path &Src, std::string* ErrMsg) {
810 // Can't use CopyFile macro defined in Windows.h because it would mess up the
811 // above line. We use the expansion it would have in a non-UNICODE build.
812 if (!::CopyFileA(Src.c_str(), Dest.c_str(), false))
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000813 return MakeErrMsg(ErrMsg, "Can't copy '" + Src.str() +
814 "' to '" + Dest.str() + "': ");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000815 return false;
816}
817
818bool
819Path::makeUnique(bool reuse_current, std::string* ErrMsg) {
820 if (reuse_current && !exists())
821 return false; // File doesn't exist already, just use it!
822
823 // Reserve space for -XXXXXX at the end.
824 char *FNBuffer = (char*) alloca(path.size()+8);
825 unsigned offset = path.size();
826 path.copy(FNBuffer, offset);
827
828 // Find a numeric suffix that isn't used by an existing file. Assume there
829 // won't be more than 1 million files with the same prefix. Probably a safe
830 // bet.
831 static unsigned FCounter = 0;
832 do {
833 sprintf(FNBuffer+offset, "-%06u", FCounter);
834 if (++FCounter > 999999)
835 FCounter = 0;
836 path = FNBuffer;
837 } while (exists());
838 return false;
839}
840
841bool
842Path::createTemporaryFileOnDisk(bool reuse_current, std::string* ErrMsg) {
843 // Make this into a unique file name
844 makeUnique(reuse_current, ErrMsg);
845
846 // Now go and create it
847 HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
848 FILE_ATTRIBUTE_NORMAL, NULL);
849 if (h == INVALID_HANDLE_VALUE)
850 return MakeErrMsg(ErrMsg, path + ": can't create file");
851
852 CloseHandle(h);
853 return false;
854}
855
Chris Lattner157d70a2008-04-01 06:00:12 +0000856/// MapInFilePages - Not yet implemented on win32.
857const char *Path::MapInFilePages(int FD, uint64_t FileSize) {
858 return 0;
859}
860
861/// MapInFilePages - Not yet implemented on win32.
862void Path::UnMapFilePages(const char *Base, uint64_t FileSize) {
863 assert(0 && "NOT IMPLEMENTED");
864}
865
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000866}
867}