Path.inc revision 542c063f9614a6d11cffb3b8f1802e62f3a46136
1//===- llvm/Support/Win32/Path.cpp - Win32 Path Implementation ---*- C++ -*-===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file provides the Win32 specific implementation of the Path class.
11//
12//===----------------------------------------------------------------------===//
13
14//===----------------------------------------------------------------------===//
15//=== WARNING: Implementation here must contain only generic Win32 code that
16//===          is guaranteed to work on *all* Win32 variants.
17//===----------------------------------------------------------------------===//
18
19#include "Windows.h"
20#include <malloc.h>
21#include <cstdio>
22
23// We need to undo a macro defined in Windows.h, otherwise we won't compile:
24#undef CopyFile
25#undef GetCurrentDirectory
26
27// 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
37static void FlipBackSlashes(std::string& s) {
38  for (size_t i = 0; i < s.size(); i++)
39    if (s[i] == '\\')
40      s[i] = '/';
41}
42
43namespace llvm {
44namespace sys {
45
46const char PathSeparator = ';';
47
48StringRef Path::GetEXESuffix() {
49  return "exe";
50}
51
52Path::Path(llvm::StringRef p)
53  : path(p) {
54  FlipBackSlashes(path);
55}
56
57Path::Path(const char *StrStart, unsigned StrLen)
58  : path(StrStart, StrLen) {
59  FlipBackSlashes(path);
60}
61
62Path&
63Path::operator=(StringRef that) {
64  path.assign(that.data(), that.size());
65  FlipBackSlashes(path);
66  return *this;
67}
68
69bool
70Path::isValid() const {
71  if (path.empty())
72    return false;
73
74  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
80  // If there is a colon, it must be the second character, preceded by a letter
81  // and followed by something.
82  pos = path.rfind(':',len);
83  size_t rootslash = 0;
84  if (pos != std::string::npos) {
85    if (pos != 1 || !isalpha(path[0]) || len < 3)
86      return false;
87      rootslash = 2;
88  }
89
90  // 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;
95  }
96
97  // 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;
103
104  // Remove trailing slash, unless it's a root slash.
105  if (len > rootslash+1 && path[len-1] == '/')
106    path.erase(--len);
107
108  // 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] == ' ') {
112      if (pos+1 == len || path[pos+1] == '/' || path[pos+1] == '\0')
113        return false;
114    }
115
116    // A component may not end in a period.
117    if (path[pos] == '.') {
118      if (pos+1 == len || path[pos+1] == '/') {
119        // 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  }
131
132  return true;
133}
134
135void Path::makeAbsolute() {
136  TCHAR  FullPath[MAX_PATH + 1] = {0};
137  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()
145    assert(0 && "Unable to make absolute path!");
146  } else if (RetLength > MAX_PATH) {
147    // FIXME: Report too small buffer (needed RetLength bytes).
148    assert(0 && "Unable to make absolute path!");
149  } else {
150    path = FullPath;
151  }
152}
153
154bool
155Path::isAbsolute(const char *NameStart, unsigned NameLen) {
156  assert(NameStart);
157  // FIXME: This does not handle correctly an absolute path starting from
158  // a drive letter or in UNC format.
159  switch (NameLen) {
160  case 0:
161    return false;
162  case 1:
163  case 2:
164    return NameStart[0] == '/';
165  default:
166    return
167      (NameStart[0] == '/' || (NameStart[1] == ':' && NameStart[2] == '/')) ||
168      (NameStart[0] == '\\' || (NameStart[1] == ':' && NameStart[2] == '\\'));
169  }
170}
171
172bool
173Path::isAbsolute() const {
174  // FIXME: This does not handle correctly an absolute path starting from
175  // a drive letter or in UNC format.
176  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  }
185}
186
187static Path *TempDirectory;
188
189Path
190Path::GetTemporaryDirectory(std::string* ErrMsg) {
191  if (TempDirectory) {
192    assert(TempDirectory->exists() && "Who has removed TempDirectory?");
193    return *TempDirectory;
194  }
195
196  char pathname[MAX_PATH];
197  if (!GetTempPath(MAX_PATH, pathname)) {
198    if (ErrMsg)
199      *ErrMsg = "Can't determine temporary directory";
200    return Path();
201  }
202
203  Path result;
204  result.set(pathname);
205
206  // Append a subdirectory passed on our process id so multiple LLVMs don't
207  // step on each other's toes.
208#ifdef __MINGW32__
209  // Mingw's Win32 header files are broken.
210  sprintf(pathname, "LLVM_%u", unsigned(GetCurrentProcessId()));
211#else
212  sprintf(pathname, "LLVM_%u", GetCurrentProcessId());
213#endif
214  result.appendComponent(pathname);
215
216  // If there's a directory left over from a previous LLVM execution that
217  // happened to have the same process id, get rid of it.
218  result.eraseFromDisk(true);
219
220  // And finally (re-)create the empty directory.
221  result.createDirectoryOnDisk(false);
222  TempDirectory = new Path(result);
223  return *TempDirectory;
224}
225
226// FIXME: the following set of functions don't map to Windows very well.
227Path
228Path::GetRootDirectory() {
229  // This is the only notion that that Windows has of a root directory. Nothing
230  // is here except for drives.
231  return Path("file:///");
232}
233
234void
235Path::GetSystemLibraryPaths(std::vector<sys::Path>& Paths) {
236  char buff[MAX_PATH];
237  // Generic form of C:\Windows\System32
238  HRESULT res =  SHGetFolderPathA(NULL,
239                                  CSIDL_FLAG_CREATE | CSIDL_SYSTEM,
240                                  NULL,
241                                  SHGFP_TYPE_CURRENT,
242                                  buff);
243  if (res != S_OK) {
244    assert(0 && "Failed to get system directory");
245    return;
246  }
247  Paths.push_back(sys::Path(buff));
248
249  // Reset buff.
250  buff[0] = 0;
251  // Generic form of C:\Windows
252  res =  SHGetFolderPathA(NULL,
253                          CSIDL_FLAG_CREATE | CSIDL_WINDOWS,
254                          NULL,
255                          SHGFP_TYPE_CURRENT,
256                          buff);
257  if (res != S_OK) {
258    assert(0 && "Failed to get windows directory");
259    return;
260  }
261  Paths.push_back(sys::Path(buff));
262}
263
264void
265Path::GetBitcodeLibraryPaths(std::vector<sys::Path>& Paths) {
266  char * env_var = getenv("LLVM_LIB_SEARCH_PATH");
267  if (env_var != 0) {
268    getPathList(env_var,Paths);
269  }
270#ifdef LLVM_LIBDIR
271  {
272    Path tmpPath;
273    if (tmpPath.set(LLVM_LIBDIR))
274      if (tmpPath.canRead())
275        Paths.push_back(tmpPath);
276  }
277#endif
278  GetSystemLibraryPaths(Paths);
279}
280
281Path
282Path::GetUserHomeDirectory() {
283  char buff[MAX_PATH];
284  HRESULT res = SHGetFolderPathA(NULL,
285                                 CSIDL_FLAG_CREATE | CSIDL_APPDATA,
286                                 NULL,
287                                 SHGFP_TYPE_CURRENT,
288                                 buff);
289  if (res != S_OK)
290    assert(0 && "Failed to get user home directory");
291  return Path(buff);
292}
293
294Path
295Path::GetCurrentDirectory() {
296  char pathname[MAX_PATH];
297  ::GetCurrentDirectoryA(MAX_PATH,pathname);
298  return Path(pathname);
299}
300
301/// GetMainExecutable - Return the path to the main executable, given the
302/// value of argv[0] from program startup.
303Path Path::GetMainExecutable(const char *argv0, void *MainAddr) {
304  char pathname[MAX_PATH];
305  DWORD ret = ::GetModuleFileNameA(NULL, pathname, MAX_PATH);
306  return ret != MAX_PATH ? Path(pathname) : Path();
307}
308
309
310// FIXME: the above set of functions don't map to Windows very well.
311
312
313StringRef Path::getDirname() const {
314  return getDirnameCharSep(path, "/");
315}
316
317StringRef
318Path::getBasename() const {
319  // Find the last slash
320  size_t slash = path.rfind('/');
321  if (slash == std::string::npos)
322    slash = 0;
323  else
324    slash++;
325
326  size_t dot = path.rfind('.');
327  if (dot == std::string::npos || dot < slash)
328    return StringRef(path).substr(slash);
329  else
330    return StringRef(path).substr(slash, dot - slash);
331}
332
333StringRef
334Path::getSuffix() const {
335  // Find the last slash
336  size_t slash = path.rfind('/');
337  if (slash == std::string::npos)
338    slash = 0;
339  else
340    slash++;
341
342  size_t dot = path.rfind('.');
343  if (dot == std::string::npos || dot < slash)
344    return StringRef("");
345  else
346    return StringRef(path).substr(dot + 1);
347}
348
349bool
350Path::exists() const {
351  DWORD attr = GetFileAttributes(path.c_str());
352  return attr != INVALID_FILE_ATTRIBUTES;
353}
354
355bool
356Path::isDirectory() const {
357  DWORD attr = GetFileAttributes(path.c_str());
358  return (attr != INVALID_FILE_ATTRIBUTES) &&
359         (attr & FILE_ATTRIBUTE_DIRECTORY);
360}
361
362bool
363Path::isSymLink() const {
364  DWORD attributes = GetFileAttributes(path.c_str());
365
366  if (attributes == INVALID_FILE_ATTRIBUTES)
367    // There's no sane way to report this :(.
368    assert(0 && "GetFileAttributes returned INVALID_FILE_ATTRIBUTES");
369
370  // This isn't exactly what defines a NTFS symlink, but it is only true for
371  // paths that act like a symlink.
372  return attributes & FILE_ATTRIBUTE_REPARSE_POINT;
373}
374
375bool
376Path::canRead() const {
377  // FIXME: take security attributes into account.
378  DWORD attr = GetFileAttributes(path.c_str());
379  return attr != INVALID_FILE_ATTRIBUTES;
380}
381
382bool
383Path::canWrite() const {
384  // FIXME: take security attributes into account.
385  DWORD attr = GetFileAttributes(path.c_str());
386  return (attr != INVALID_FILE_ATTRIBUTES) && !(attr & FILE_ATTRIBUTE_READONLY);
387}
388
389bool
390Path::canExecute() const {
391  // FIXME: take security attributes into account.
392  DWORD attr = GetFileAttributes(path.c_str());
393  return attr != INVALID_FILE_ATTRIBUTES;
394}
395
396bool
397Path::isRegularFile() const {
398  bool res;
399  if (fs::is_regular_file(path, res))
400    return false;
401  return res;
402}
403
404StringRef
405Path::getLast() const {
406  // Find the last slash
407  size_t pos = path.rfind('/');
408
409  // Handle the corner cases
410  if (pos == std::string::npos)
411    return path;
412
413  // If the last character is a slash, we have a root directory
414  if (pos == path.length()-1)
415    return path;
416
417  // Return everything after the last slash
418  return StringRef(path).substr(pos+1);
419}
420
421const FileStatus *
422PathWithStatus::getFileStatus(bool update, std::string *ErrStr) const {
423  if (!fsIsValid || update) {
424    WIN32_FILE_ATTRIBUTE_DATA fi;
425    if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
426      MakeErrMsg(ErrStr, "getStatusInfo():" + std::string(path) +
427                      ": Can't get status: ");
428      return 0;
429    }
430
431    status.fileSize = fi.nFileSizeHigh;
432    status.fileSize <<= sizeof(fi.nFileSizeHigh)*8;
433    status.fileSize += fi.nFileSizeLow;
434
435    status.mode = fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY ? 0555 : 0777;
436    status.user = 9999;    // Not applicable to Windows, so...
437    status.group = 9999;   // Not applicable to Windows, so...
438
439    // FIXME: this is only unique if the file is accessed by the same file path.
440    // How do we do this for C:\dir\file and ..\dir\file ? Unix has inode
441    // numbers, but the concept doesn't exist in Windows.
442    status.uniqueID = 0;
443    for (unsigned i = 0; i < path.length(); ++i)
444      status.uniqueID += path[i];
445
446    ULARGE_INTEGER ui;
447    ui.LowPart = fi.ftLastWriteTime.dwLowDateTime;
448    ui.HighPart = fi.ftLastWriteTime.dwHighDateTime;
449    status.modTime.fromWin32Time(ui.QuadPart);
450
451    status.isDir = fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
452    fsIsValid = true;
453  }
454  return &status;
455}
456
457bool Path::makeReadableOnDisk(std::string* ErrMsg) {
458  // All files are readable on Windows (ignoring security attributes).
459  return false;
460}
461
462bool Path::makeWriteableOnDisk(std::string* ErrMsg) {
463  DWORD attr = GetFileAttributes(path.c_str());
464
465  // If it doesn't exist, we're done.
466  if (attr == INVALID_FILE_ATTRIBUTES)
467    return false;
468
469  if (attr & FILE_ATTRIBUTE_READONLY) {
470    if (!SetFileAttributes(path.c_str(), attr & ~FILE_ATTRIBUTE_READONLY)) {
471      MakeErrMsg(ErrMsg, std::string(path) + ": Can't make file writable: ");
472      return true;
473    }
474  }
475  return false;
476}
477
478bool Path::makeExecutableOnDisk(std::string* ErrMsg) {
479  // All files are executable on Windows (ignoring security attributes).
480  return false;
481}
482
483bool
484Path::getDirectoryContents(std::set& result, std::string* ErrMsg) const {
485  WIN32_FILE_ATTRIBUTE_DATA fi;
486  if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
487    MakeErrMsg(ErrMsg, path + ": can't get status of file");
488    return true;
489  }
490
491  if (!(fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
492    if (ErrMsg)
493      *ErrMsg = path + ": not a directory";
494    return true;
495  }
496
497  result.clear();
498  WIN32_FIND_DATA fd;
499  std::string searchpath = path;
500  if (path.size() == 0 || searchpath[path.size()-1] == '/')
501    searchpath += "*";
502  else
503    searchpath += "/*";
504
505  HANDLE h = FindFirstFile(searchpath.c_str(), &fd);
506  if (h == INVALID_HANDLE_VALUE) {
507    if (GetLastError() == ERROR_FILE_NOT_FOUND)
508      return true; // not really an error, now is it?
509    MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
510    return true;
511  }
512
513  do {
514    if (fd.cFileName[0] == '.')
515      continue;
516    Path aPath(path);
517    aPath.appendComponent(&fd.cFileName[0]);
518    result.insert(aPath);
519  } while (FindNextFile(h, &fd));
520
521  DWORD err = GetLastError();
522  FindClose(h);
523  if (err != ERROR_NO_MORE_FILES) {
524    SetLastError(err);
525    MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
526    return true;
527  }
528  return false;
529}
530
531bool
532Path::set(StringRef a_path) {
533  if (a_path.empty())
534    return false;
535  std::string save(path);
536  path = a_path;
537  FlipBackSlashes(path);
538  if (!isValid()) {
539    path = save;
540    return false;
541  }
542  return true;
543}
544
545bool
546Path::appendComponent(StringRef name) {
547  if (name.empty())
548    return false;
549  std::string save(path);
550  if (!path.empty()) {
551    size_t last = path.size() - 1;
552    if (path[last] != '/')
553      path += '/';
554  }
555  path += name;
556  if (!isValid()) {
557    path = save;
558    return false;
559  }
560  return true;
561}
562
563bool
564Path::eraseComponent() {
565  size_t slashpos = path.rfind('/',path.size());
566  if (slashpos == path.size() - 1 || slashpos == std::string::npos)
567    return false;
568  std::string save(path);
569  path.erase(slashpos);
570  if (!isValid()) {
571    path = save;
572    return false;
573  }
574  return true;
575}
576
577bool
578Path::eraseSuffix() {
579  size_t dotpos = path.rfind('.',path.size());
580  size_t slashpos = path.rfind('/',path.size());
581  if (dotpos != std::string::npos) {
582    if (slashpos == std::string::npos || dotpos > slashpos+1) {
583      std::string save(path);
584      path.erase(dotpos, path.size()-dotpos);
585      if (!isValid()) {
586        path = save;
587        return false;
588      }
589      return true;
590    }
591  }
592  return false;
593}
594
595inline bool PathMsg(std::string* ErrMsg, const char* pathname, const char*msg) {
596  if (ErrMsg)
597    *ErrMsg = std::string(pathname) + ": " + std::string(msg);
598  return true;
599}
600
601bool
602Path::createDirectoryOnDisk(bool create_parents, std::string* ErrMsg) {
603  // Get a writeable copy of the path name
604  size_t len = path.length();
605  char *pathname = reinterpret_cast<char *>(_alloca(len+2));
606  path.copy(pathname, len);
607  pathname[len] = 0;
608
609  // Make sure it ends with a slash.
610  if (len == 0 || pathname[len - 1] != '/') {
611    pathname[len] = '/';
612    pathname[++len] = 0;
613  }
614
615  // Determine starting point for initial / search.
616  char *next = pathname;
617  if (pathname[0] == '/' && pathname[1] == '/') {
618    // Skip host name.
619    next = strchr(pathname+2, '/');
620    if (next == NULL)
621      return PathMsg(ErrMsg, pathname, "badly formed remote directory");
622
623    // Skip share name.
624    next = strchr(next+1, '/');
625    if (next == NULL)
626      return PathMsg(ErrMsg, pathname,"badly formed remote directory");
627
628    next++;
629    if (*next == 0)
630      return PathMsg(ErrMsg, pathname, "badly formed remote directory");
631
632  } else {
633    if (pathname[1] == ':')
634      next += 2;    // skip drive letter
635    if (*next == '/')
636      next++;       // skip root directory
637  }
638
639  // If we're supposed to create intermediate directories
640  if (create_parents) {
641    // Loop through the directory components until we're done
642    while (*next) {
643      next = strchr(next, '/');
644      *next = 0;
645      if (!CreateDirectory(pathname, NULL) &&
646          GetLastError() != ERROR_ALREADY_EXISTS)
647          return MakeErrMsg(ErrMsg,
648            std::string(pathname) + ": Can't create directory: ");
649      *next++ = '/';
650    }
651  } else {
652    // Drop trailing slash.
653    pathname[len-1] = 0;
654    if (!CreateDirectory(pathname, NULL) &&
655        GetLastError() != ERROR_ALREADY_EXISTS) {
656      return MakeErrMsg(ErrMsg, std::string(pathname) +
657                        ": Can't create directory: ");
658    }
659  }
660  return false;
661}
662
663bool
664Path::createFileOnDisk(std::string* ErrMsg) {
665  // Create the file
666  HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
667                        FILE_ATTRIBUTE_NORMAL, NULL);
668  if (h == INVALID_HANDLE_VALUE)
669    return MakeErrMsg(ErrMsg, path + ": Can't create file: ");
670
671  CloseHandle(h);
672  return false;
673}
674
675bool
676Path::eraseFromDisk(bool remove_contents, std::string *ErrStr) const {
677  WIN32_FILE_ATTRIBUTE_DATA fi;
678  if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi))
679    return true;
680
681  if (fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
682    // If it doesn't exist, we're done.
683    bool Exists;
684    if (fs::exists(path, Exists) || !Exists)
685      return false;
686
687    char *pathname = reinterpret_cast<char *>(_alloca(path.length()+3));
688    int lastchar = path.length() - 1 ;
689    path.copy(pathname, lastchar+1);
690
691    // Make path end with '/*'.
692    if (pathname[lastchar] != '/')
693      pathname[++lastchar] = '/';
694    pathname[lastchar+1] = '*';
695    pathname[lastchar+2] = 0;
696
697    if (remove_contents) {
698      WIN32_FIND_DATA fd;
699      HANDLE h = FindFirstFile(pathname, &fd);
700
701      // It's a bad idea to alter the contents of a directory while enumerating
702      // its contents. So build a list of its contents first, then destroy them.
703
704      if (h != INVALID_HANDLE_VALUE) {
705        std::vector<Path> list;
706
707        do {
708          if (strcmp(fd.cFileName, ".") == 0)
709            continue;
710          if (strcmp(fd.cFileName, "..") == 0)
711            continue;
712
713          Path aPath(path);
714          aPath.appendComponent(&fd.cFileName[0]);
715          list.push_back(aPath);
716        } while (FindNextFile(h, &fd));
717
718        DWORD err = GetLastError();
719        FindClose(h);
720        if (err != ERROR_NO_MORE_FILES) {
721          SetLastError(err);
722          return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
723        }
724
725        for (std::vector<Path>::iterator I = list.begin(); I != list.end();
726             ++I) {
727          Path &aPath = *I;
728          aPath.eraseFromDisk(true);
729        }
730      } else {
731        if (GetLastError() != ERROR_FILE_NOT_FOUND)
732          return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
733      }
734    }
735
736    pathname[lastchar] = 0;
737    if (!RemoveDirectory(pathname))
738      return MakeErrMsg(ErrStr,
739        std::string(pathname) + ": Can't destroy directory: ");
740    return false;
741  } else {
742    // Read-only files cannot be deleted on Windows.  Must remove the read-only
743    // attribute first.
744    if (fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
745      if (!SetFileAttributes(path.c_str(),
746                             fi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
747        return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
748    }
749
750    if (!DeleteFile(path.c_str()))
751      return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
752    return false;
753  }
754}
755
756bool Path::getMagicNumber(std::string& Magic, unsigned len) const {
757  assert(len < 1024 && "Request for magic string too long");
758  char* buf = reinterpret_cast<char*>(alloca(len));
759
760  HANDLE h = CreateFile(path.c_str(),
761                        GENERIC_READ,
762                        FILE_SHARE_READ,
763                        NULL,
764                        OPEN_EXISTING,
765                        FILE_ATTRIBUTE_NORMAL,
766                        NULL);
767  if (h == INVALID_HANDLE_VALUE)
768    return false;
769
770  DWORD nRead = 0;
771  BOOL ret = ReadFile(h, buf, len, &nRead, NULL);
772  CloseHandle(h);
773
774  if (!ret || nRead != len)
775    return false;
776
777  Magic = std::string(buf, len);
778  return true;
779}
780
781bool
782Path::renamePathOnDisk(const Path& newName, std::string* ErrMsg) {
783  if (!MoveFileEx(path.c_str(), newName.c_str(), MOVEFILE_REPLACE_EXISTING))
784    return MakeErrMsg(ErrMsg, "Can't move '" + path + "' to '" + newName.path
785        + "': ");
786  return false;
787}
788
789bool
790Path::setStatusInfoOnDisk(const FileStatus &si, std::string *ErrMsg) const {
791  // FIXME: should work on directories also.
792  if (!si.isFile) {
793    return true;
794  }
795
796  HANDLE h = CreateFile(path.c_str(),
797                        FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES,
798                        FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
799                        NULL,
800                        OPEN_EXISTING,
801                        FILE_ATTRIBUTE_NORMAL,
802                        NULL);
803  if (h == INVALID_HANDLE_VALUE)
804    return true;
805
806  BY_HANDLE_FILE_INFORMATION bhfi;
807  if (!GetFileInformationByHandle(h, &bhfi)) {
808    DWORD err = GetLastError();
809    CloseHandle(h);
810    SetLastError(err);
811    return MakeErrMsg(ErrMsg, path + ": GetFileInformationByHandle: ");
812  }
813
814  ULARGE_INTEGER ui;
815  ui.QuadPart = si.modTime.toWin32Time();
816  FILETIME ft;
817  ft.dwLowDateTime = ui.LowPart;
818  ft.dwHighDateTime = ui.HighPart;
819  BOOL ret = SetFileTime(h, NULL, &ft, &ft);
820  DWORD err = GetLastError();
821  CloseHandle(h);
822  if (!ret) {
823    SetLastError(err);
824    return MakeErrMsg(ErrMsg, path + ": SetFileTime: ");
825  }
826
827  // Best we can do with Unix permission bits is to interpret the owner
828  // writable bit.
829  if (si.mode & 0200) {
830    if (bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
831      if (!SetFileAttributes(path.c_str(),
832              bhfi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
833        return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
834    }
835  } else {
836    if (!(bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) {
837      if (!SetFileAttributes(path.c_str(),
838              bhfi.dwFileAttributes | FILE_ATTRIBUTE_READONLY))
839        return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
840    }
841  }
842
843  return false;
844}
845
846bool
847CopyFile(const sys::Path &Dest, const sys::Path &Src, std::string* ErrMsg) {
848  // Can't use CopyFile macro defined in Windows.h because it would mess up the
849  // above line.  We use the expansion it would have in a non-UNICODE build.
850  if (!::CopyFileA(Src.c_str(), Dest.c_str(), false))
851    return MakeErrMsg(ErrMsg, "Can't copy '" + Src.str() +
852               "' to '" + Dest.str() + "': ");
853  return false;
854}
855
856bool
857Path::makeUnique(bool reuse_current, std::string* ErrMsg) {
858  bool Exists;
859  if (reuse_current && (fs::exists(path, Exists) || !Exists))
860    return false; // File doesn't exist already, just use it!
861
862  // Reserve space for -XXXXXX at the end.
863  char *FNBuffer = (char*) alloca(path.size()+8);
864  unsigned offset = path.size();
865  path.copy(FNBuffer, offset);
866
867  // Find a numeric suffix that isn't used by an existing file.  Assume there
868  // won't be more than 1 million files with the same prefix.  Probably a safe
869  // bet.
870  static int FCounter = -1;
871  if (FCounter < 0) {
872    // Give arbitrary initial seed.
873    // FIXME: We should use sys::fs::unique_file() in future.
874    LARGE_INTEGER cnt64;
875    DWORD x = GetCurrentProcessId();
876    x = (x << 16) | (x >> 16);
877    if (QueryPerformanceCounter(&cnt64))    // RDTSC
878      x ^= cnt64.HighPart ^ cnt64.LowPart;
879    FCounter = x % 1000000;
880  }
881  do {
882    sprintf(FNBuffer+offset, "-%06u", FCounter);
883    if (++FCounter > 999999)
884      FCounter = 0;
885    path = FNBuffer;
886  } while (!fs::exists(path, Exists) && Exists);
887  return false;
888}
889
890bool
891Path::createTemporaryFileOnDisk(bool reuse_current, std::string* ErrMsg) {
892  // Make this into a unique file name
893  makeUnique(reuse_current, ErrMsg);
894
895  // Now go and create it
896  HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
897                        FILE_ATTRIBUTE_NORMAL, NULL);
898  if (h == INVALID_HANDLE_VALUE)
899    return MakeErrMsg(ErrMsg, path + ": can't create file");
900
901  CloseHandle(h);
902  return false;
903}
904
905/// MapInFilePages - Not yet implemented on win32.
906const char *Path::MapInFilePages(int FD, size_t FileSize, off_t Offset) {
907  return 0;
908}
909
910/// MapInFilePages - Not yet implemented on win32.
911void Path::UnMapFilePages(const char *Base, size_t FileSize) {
912  assert(0 && "NOT IMPLEMENTED");
913}
914
915}
916}
917