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