lib/System/Win32/Path.inc

   1 //===- llvm/System/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 // 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>
  24 #include <cstdio>
  25
  26 // We need to undo a macro defined in Windows.h, otherwise we won't compile:
  27 #undef CopyFile
  28 #undef GetCurrentDirectory
  29
  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
  40 static void FlipBackSlashes(std::string& s) {
  41   for (size_t i = 0; i < s.size(); i++)
  42     if (s[i] == '\\')
  43       s[i] = '/';
  44 }
  45
  46 namespace llvm {
  47 namespace sys {
  48 const char PathSeparator = ';';
  49
  50 Path::Path(llvm::StringRef p)
  51   : path(p) {
  52   FlipBackSlashes(path);
  53 }
  54
  55 Path::Path(const char *StrStart, unsigned StrLen)
  56   : path(StrStart, StrLen) {
  57   FlipBackSlashes(path);
  58 }
  59
  60 Path&
  61 Path::operator=(StringRef that) {
  62   path.assign(that.data(), that.size());
  63   FlipBackSlashes(path);
  64   return *this;
  65 }
  66
  67 bool
  68 Path::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
 128 void 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()
 138     assert(0 && "Unable to make absolute path!");
 139   } else if (RetLength > MAX_PATH) {
 140     // FIXME: Report too small buffer (needed RetLength bytes).
 141     assert(0 && "Unable to make absolute path!");
 142   } else {
 143     path = FullPath;
 144   }
 145 }
 146
 147 bool
 148 Path::isAbsolute(const char *NameStart, unsigned NameLen) {
 149   assert(NameStart);
 150   // FIXME: This does not handle correctly an absolute path starting from
 151   // a drive letter or in UNC format.
 152   switch (NameLen) {
 153   case 0:
 154     return false;
 155   case 1:
 156   case 2:
 157     return NameStart[0] == '/';
 158   default:
 159     return (NameStart[0] == '/' || (NameStart[1] == ':' && NameStart[2] == '/')) ||
 160            (NameStart[0] == '\\' || (NameStart[1] == ':' && NameStart[2] == '\\'));
 161   }
 162 }
 163
 164 bool
 165 Path::isAbsolute() const {
 166   // FIXME: This does not handle correctly an absolute path starting from
 167   // a drive letter or in UNC format.
 168   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
 179 static Path *TempDirectory;
 180
 181 Path
 182 Path::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.
 217 Path
 218 Path::GetRootDirectory() {
 219   Path result;
 220   result.set("C:/");
 221   return result;
 222 }
 223
 224 void
 225 Path::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
 230 void
 231 Path::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
 247 Path
 248 Path::GetLLVMDefaultConfigDir() {
 249   // TODO: this isn't going to fly on Windows
 250   return Path("/etc/llvm");
 251 }
 252
 253 Path
 254 Path::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 }
 264
 265 Path
 266 Path::GetCurrentDirectory() {
 267   char pathname[MAX_PATH];
 268   ::GetCurrentDirectoryA(MAX_PATH,pathname);
 269   return Path(pathname);
 270 }
 271
 272 /// GetMainExecutable - Return the path to the main executable, given the
 273 /// value of argv[0] from program startup.
 274 Path Path::GetMainExecutable(const char *argv0, void *MainAddr) {
 275   char pathname[MAX_PATH];
 276   DWORD ret = ::GetModuleFileNameA(NULL, pathname, MAX_PATH);
 277   return ret != MAX_PATH ? Path(pathname) : Path();
 278 }
 279
 280
 281 // FIXME: the above set of functions don't map to Windows very well.
 282
 283
 284 StringRef Path::getDirname() const {
 285   return getDirnameCharSep(path, "/");
 286 }
 287
 288 StringRef
 289 Path::getBasename() const {
 290   // Find the last slash
 291   size_t slash = path.rfind('/');
 292   if (slash == std::string::npos)
 293     slash = 0;
 294   else
 295     slash++;
 296
 297   size_t dot = path.rfind('.');
 298   if (dot == std::string::npos || dot < slash)
 299     return StringRef(path).substr(slash);
 300   else
 301     return StringRef(path).substr(slash, dot - slash);
 302 }
 303
 304 StringRef
 305 Path::getSuffix() const {
 306   // Find the last slash
 307   size_t slash = path.rfind('/');
 308   if (slash == std::string::npos)
 309     slash = 0;
 310   else
 311     slash++;
 312
 313   size_t dot = path.rfind('.');
 314   if (dot == std::string::npos || dot < slash)
 315     return StringRef("");
 316   else
 317     return StringRef(path).substr(dot + 1);
 318 }
 319
 320 bool
 321 Path::exists() const {
 322   DWORD attr = GetFileAttributes(path.c_str());
 323   return attr != INVALID_FILE_ATTRIBUTES;
 324 }
 325
 326 bool
 327 Path::isDirectory() const {
 328   DWORD attr = GetFileAttributes(path.c_str());
 329   return (attr != INVALID_FILE_ATTRIBUTES) &&
 330          (attr & FILE_ATTRIBUTE_DIRECTORY);
 331 }
 332
 333 bool
 334 Path::canRead() const {
 335   // FIXME: take security attributes into account.
 336   DWORD attr = GetFileAttributes(path.c_str());
 337   return attr != INVALID_FILE_ATTRIBUTES;
 338 }
 339
 340 bool
 341 Path::canWrite() const {
 342   // FIXME: take security attributes into account.
 343   DWORD attr = GetFileAttributes(path.c_str());
 344   return (attr != INVALID_FILE_ATTRIBUTES) && !(attr & FILE_ATTRIBUTE_READONLY);
 345 }
 346
 347 bool
 348 Path::canExecute() const {
 349   // FIXME: take security attributes into account.
 350   DWORD attr = GetFileAttributes(path.c_str());
 351   return attr != INVALID_FILE_ATTRIBUTES;
 352 }
 353
 354 bool
 355 Path::isRegularFile() const {
 356   if (isDirectory())
 357     return false;
 358   return true;
 359 }
 360
 361 StringRef
 362 Path::getLast() const {
 363   // Find the last slash
 364   size_t pos = path.rfind('/');
 365
 366   // Handle the corner cases
 367   if (pos == std::string::npos)
 368     return path;
 369
 370   // If the last character is a slash, we have a root directory
 371   if (pos == path.length()-1)
 372     return path;
 373
 374   // Return everything after the last slash
 375   return StringRef(path).substr(pos+1);
 376 }
 377
 378 const FileStatus *
 379 PathWithStatus::getFileStatus(bool update, std::string *ErrStr) const {
 380   if (!fsIsValid || update) {
 381     WIN32_FILE_ATTRIBUTE_DATA fi;
 382     if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
 383       MakeErrMsg(ErrStr, "getStatusInfo():" + std::string(path) +
 384                       ": Can't get status: ");
 385       return 0;
 386     }
 387
 388     status.fileSize = fi.nFileSizeHigh;
 389     status.fileSize <<= sizeof(fi.nFileSizeHigh)*8;
 390     status.fileSize += fi.nFileSizeLow;
 391
 392     status.mode = fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY ? 0555 : 0777;
 393     status.user = 9999;    // Not applicable to Windows, so...
 394     status.group = 9999;   // Not applicable to Windows, so...
 395
 396     // FIXME: this is only unique if the file is accessed by the same file path.
 397     // How do we do this for C:\dir\file and ..\dir\file ? Unix has inode
 398     // numbers, but the concept doesn't exist in Windows.
 399     status.uniqueID = 0;
 400     for (unsigned i = 0; i < path.length(); ++i)
 401       status.uniqueID += path[i];
 402
 403     __int64 ft = *reinterpret_cast<__int64*>(&fi.ftLastWriteTime);
 404     status.modTime.fromWin32Time(ft);
 405
 406     status.isDir = fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
 407     fsIsValid = true;
 408   }
 409   return &status;
 410 }
 411
 412 bool Path::makeReadableOnDisk(std::string* ErrMsg) {
 413   // All files are readable on Windows (ignoring security attributes).
 414   return false;
 415 }
 416
 417 bool Path::makeWriteableOnDisk(std::string* ErrMsg) {
 418   DWORD attr = GetFileAttributes(path.c_str());
 419
 420   // If it doesn't exist, we're done.
 421   if (attr == INVALID_FILE_ATTRIBUTES)
 422     return false;
 423
 424   if (attr & FILE_ATTRIBUTE_READONLY) {
 425     if (!SetFileAttributes(path.c_str(), attr & ~FILE_ATTRIBUTE_READONLY)) {
 426       MakeErrMsg(ErrMsg, std::string(path) + ": Can't make file writable: ");
 427       return true;
 428     }
 429   }
 430   return false;
 431 }
 432
 433 bool Path::makeExecutableOnDisk(std::string* ErrMsg) {
 434   // All files are executable on Windows (ignoring security attributes).
 435   return false;
 436 }
 437
 438 bool
 439 Path::getDirectoryContents(std::set<Path>& result, std::string* ErrMsg) const {
 440   WIN32_FILE_ATTRIBUTE_DATA fi;
 441   if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi)) {
 442     MakeErrMsg(ErrMsg, path + ": can't get status of file");
 443     return true;
 444   }
 445
 446   if (!(fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
 447     if (ErrMsg)
 448       *ErrMsg = path + ": not a directory";
 449     return true;
 450   }
 451
 452   result.clear();
 453   WIN32_FIND_DATA fd;
 454   std::string searchpath = path;
 455   if (path.size() == 0 || searchpath[path.size()-1] == '/')
 456     searchpath += "*";
 457   else
 458     searchpath += "/*";
 459
 460   HANDLE h = FindFirstFile(searchpath.c_str(), &fd);
 461   if (h == INVALID_HANDLE_VALUE) {
 462     if (GetLastError() == ERROR_FILE_NOT_FOUND)
 463       return true; // not really an error, now is it?
 464     MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
 465     return true;
 466   }
 467
 468   do {
 469     if (fd.cFileName[0] == '.')
 470       continue;
 471     Path aPath(path);
 472     aPath.appendComponent(&fd.cFileName[0]);
 473     result.insert(aPath);
 474   } while (FindNextFile(h, &fd));
 475
 476   DWORD err = GetLastError();
 477   FindClose(h);
 478   if (err != ERROR_NO_MORE_FILES) {
 479     SetLastError(err);
 480     MakeErrMsg(ErrMsg, path + ": Can't read directory: ");
 481     return true;
 482   }
 483   return false;
 484 }
 485
 486 bool
 487 Path::set(StringRef a_path) {
 488   if (a_path.empty())
 489     return false;
 490   std::string save(path);
 491   path = a_path;
 492   FlipBackSlashes(path);
 493   if (!isValid()) {
 494     path = save;
 495     return false;
 496   }
 497   return true;
 498 }
 499
 500 bool
 501 Path::appendComponent(StringRef name) {
 502   if (name.empty())
 503     return false;
 504   std::string save(path);
 505   if (!path.empty()) {
 506     size_t last = path.size() - 1;
 507     if (path[last] != '/')
 508       path += '/';
 509   }
 510   path += name;
 511   if (!isValid()) {
 512     path = save;
 513     return false;
 514   }
 515   return true;
 516 }
 517
 518 bool
 519 Path::eraseComponent() {
 520   size_t slashpos = path.rfind('/',path.size());
 521   if (slashpos == path.size() - 1 || slashpos == std::string::npos)
 522     return false;
 523   std::string save(path);
 524   path.erase(slashpos);
 525   if (!isValid()) {
 526     path = save;
 527     return false;
 528   }
 529   return true;
 530 }
 531
 532 bool
 533 Path::appendSuffix(StringRef suffix) {
 534   std::string save(path);
 535   path.append(".");
 536   path.append(suffix);
 537   if (!isValid()) {
 538     path = save;
 539     return false;
 540   }
 541   return true;
 542 }
 543
 544 bool
 545 Path::eraseSuffix() {
 546   size_t dotpos = path.rfind('.',path.size());
 547   size_t slashpos = path.rfind('/',path.size());
 548   if (dotpos != std::string::npos) {
 549     if (slashpos == std::string::npos || dotpos > slashpos+1) {
 550       std::string save(path);
 551       path.erase(dotpos, path.size()-dotpos);
 552       if (!isValid()) {
 553         path = save;
 554         return false;
 555       }
 556       return true;
 557     }
 558   }
 559   return false;
 560 }
 561
 562 inline bool PathMsg(std::string* ErrMsg, const char* pathname, const char*msg) {
 563   if (ErrMsg)
 564     *ErrMsg = std::string(pathname) + ": " + std::string(msg);
 565   return true;
 566 }
 567
 568 bool
 569 Path::createDirectoryOnDisk(bool create_parents, std::string* ErrMsg) {
 570   // Get a writeable copy of the path name
 571   size_t len = path.length();
 572   char *pathname = reinterpret_cast<char *>(_alloca(len+2));
 573   path.copy(pathname, len);
 574   pathname[len] = 0;
 575
 576   // Make sure it ends with a slash.
 577   if (len == 0 || pathname[len - 1] != '/') {
 578     pathname[len] = '/';
 579     pathname[++len] = 0;
 580   }
 581
 582   // Determine starting point for initial / search.
 583   char *next = pathname;
 584   if (pathname[0] == '/' && pathname[1] == '/') {
 585     // Skip host name.
 586     next = strchr(pathname+2, '/');
 587     if (next == NULL)
 588       return PathMsg(ErrMsg, pathname, "badly formed remote directory");
 589
 590     // Skip share name.
 591     next = strchr(next+1, '/');
 592     if (next == NULL)
 593       return PathMsg(ErrMsg, pathname,"badly formed remote directory");
 594
 595     next++;
 596     if (*next == 0)
 597       return PathMsg(ErrMsg, pathname, "badly formed remote directory");
 598
 599   } else {
 600     if (pathname[1] == ':')
 601       next += 2;    // skip drive letter
 602     if (*next == '/')
 603       next++;       // skip root directory
 604   }
 605
 606   // If we're supposed to create intermediate directories
 607   if (create_parents) {
 608     // Loop through the directory components until we're done
 609     while (*next) {
 610       next = strchr(next, '/');
 611       *next = 0;
 612       if (!CreateDirectory(pathname, NULL) &&
 613           GetLastError() != ERROR_ALREADY_EXISTS)
 614           return MakeErrMsg(ErrMsg,
 615             std::string(pathname) + ": Can't create directory: ");
 616       *next++ = '/';
 617     }
 618   } else {
 619     // Drop trailing slash.
 620     pathname[len-1] = 0;
 621     if (!CreateDirectory(pathname, NULL) &&
 622         GetLastError() != ERROR_ALREADY_EXISTS) {
 623       return MakeErrMsg(ErrMsg, std::string(pathname) + ": Can't create directory: ");
 624     }
 625   }
 626   return false;
 627 }
 628
 629 bool
 630 Path::createFileOnDisk(std::string* ErrMsg) {
 631   // Create the file
 632   HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
 633                         FILE_ATTRIBUTE_NORMAL, NULL);
 634   if (h == INVALID_HANDLE_VALUE)
 635     return MakeErrMsg(ErrMsg, path + ": Can't create file: ");
 636
 637   CloseHandle(h);
 638   return false;
 639 }
 640
 641 bool
 642 Path::eraseFromDisk(bool remove_contents, std::string *ErrStr) const {
 643   WIN32_FILE_ATTRIBUTE_DATA fi;
 644   if (!GetFileAttributesEx(path.c_str(), GetFileExInfoStandard, &fi))
 645     return true;
 646
 647   if (fi.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
 648     // If it doesn't exist, we're done.
 649     if (!exists())
 650       return false;
 651
 652     char *pathname = reinterpret_cast<char *>(_alloca(path.length()+3));
 653     int lastchar = path.length() - 1 ;
 654     path.copy(pathname, lastchar+1);
 655
 656     // Make path end with '/*'.
 657     if (pathname[lastchar] != '/')
 658       pathname[++lastchar] = '/';
 659     pathname[lastchar+1] = '*';
 660     pathname[lastchar+2] = 0;
 661
 662     if (remove_contents) {
 663       WIN32_FIND_DATA fd;
 664       HANDLE h = FindFirstFile(pathname, &fd);
 665
 666       // It's a bad idea to alter the contents of a directory while enumerating
 667       // its contents. So build a list of its contents first, then destroy them.
 668
 669       if (h != INVALID_HANDLE_VALUE) {
 670         std::vector<Path> list;
 671
 672         do {
 673           if (strcmp(fd.cFileName, ".") == 0)
 674             continue;
 675           if (strcmp(fd.cFileName, "..") == 0)
 676             continue;
 677
 678           Path aPath(path);
 679           aPath.appendComponent(&fd.cFileName[0]);
 680           list.push_back(aPath);
 681         } while (FindNextFile(h, &fd));
 682
 683         DWORD err = GetLastError();
 684         FindClose(h);
 685         if (err != ERROR_NO_MORE_FILES) {
 686           SetLastError(err);
 687           return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
 688         }
 689
 690         for (std::vector<Path>::iterator I = list.begin(); I != list.end();
 691              ++I) {
 692           Path &aPath = *I;
 693           aPath.eraseFromDisk(true);
 694         }
 695       } else {
 696         if (GetLastError() != ERROR_FILE_NOT_FOUND)
 697           return MakeErrMsg(ErrStr, path + ": Can't read directory: ");
 698       }
 699     }
 700
 701     pathname[lastchar] = 0;
 702     if (!RemoveDirectory(pathname))
 703       return MakeErrMsg(ErrStr,
 704         std::string(pathname) + ": Can't destroy directory: ");
 705     return false;
 706   } else {
 707     // Read-only files cannot be deleted on Windows.  Must remove the read-only
 708     // attribute first.
 709     if (fi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
 710       if (!SetFileAttributes(path.c_str(),
 711                              fi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
 712         return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
 713     }
 714
 715     if (!DeleteFile(path.c_str()))
 716       return MakeErrMsg(ErrStr, path + ": Can't destroy file: ");
 717     return false;
 718   }
 719 }
 720
 721 bool Path::getMagicNumber(std::string& Magic, unsigned len) const {
 722   assert(len < 1024 && "Request for magic string too long");
 723   char* buf = (char*) alloca(1 + len);
 724
 725   HANDLE h = CreateFile(path.c_str(),
 726                         GENERIC_READ,
 727                         FILE_SHARE_READ,
 728                         NULL,
 729                         OPEN_EXISTING,
 730                         FILE_ATTRIBUTE_NORMAL,
 731                         NULL);
 732   if (h == INVALID_HANDLE_VALUE)
 733     return false;
 734
 735   DWORD nRead = 0;
 736   BOOL ret = ReadFile(h, buf, len, &nRead, NULL);
 737   CloseHandle(h);
 738
 739   if (!ret || nRead != len)
 740     return false;
 741
 742   buf[len] = '\0';
 743   Magic = buf;
 744   return true;
 745 }
 746
 747 bool
 748 Path::renamePathOnDisk(const Path& newName, std::string* ErrMsg) {
 749   if (!MoveFileEx(path.c_str(), newName.c_str(), MOVEFILE_REPLACE_EXISTING))
 750     return MakeErrMsg(ErrMsg, "Can't move '" + path + "' to '" + newName.path
 751         + "': ");
 752   return false;
 753 }
 754
 755 bool
 756 Path::setStatusInfoOnDisk(const FileStatus &si, std::string *ErrMsg) const {
 757   // FIXME: should work on directories also.
 758   if (!si.isFile) {
 759     return true;
 760   }
 761
 762   HANDLE h = CreateFile(path.c_str(),
 763                         FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES,
 764                         FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
 765                         NULL,
 766                         OPEN_EXISTING,
 767                         FILE_ATTRIBUTE_NORMAL,
 768                         NULL);
 769   if (h == INVALID_HANDLE_VALUE)
 770     return true;
 771
 772   BY_HANDLE_FILE_INFORMATION bhfi;
 773   if (!GetFileInformationByHandle(h, &bhfi)) {
 774     DWORD err = GetLastError();
 775     CloseHandle(h);
 776     SetLastError(err);
 777     return MakeErrMsg(ErrMsg, path + ": GetFileInformationByHandle: ");
 778   }
 779
 780   FILETIME ft;
 781   (uint64_t&)ft = si.modTime.toWin32Time();
 782   BOOL ret = SetFileTime(h, NULL, &ft, &ft);
 783   DWORD err = GetLastError();
 784   CloseHandle(h);
 785   if (!ret) {
 786     SetLastError(err);
 787     return MakeErrMsg(ErrMsg, path + ": SetFileTime: ");
 788   }
 789
 790   // Best we can do with Unix permission bits is to interpret the owner
 791   // writable bit.
 792   if (si.mode & 0200) {
 793     if (bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY) {
 794       if (!SetFileAttributes(path.c_str(),
 795               bhfi.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY))
 796         return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
 797     }
 798   } else {
 799     if (!(bhfi.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) {
 800       if (!SetFileAttributes(path.c_str(),
 801               bhfi.dwFileAttributes | FILE_ATTRIBUTE_READONLY))
 802         return MakeErrMsg(ErrMsg, path + ": SetFileAttributes: ");
 803     }
 804   }
 805
 806   return false;
 807 }
 808
 809 bool
 810 CopyFile(const sys::Path &Dest, const sys::Path &Src, std::string* ErrMsg) {
 811   // Can't use CopyFile macro defined in Windows.h because it would mess up the
 812   // above line.  We use the expansion it would have in a non-UNICODE build.
 813   if (!::CopyFileA(Src.c_str(), Dest.c_str(), false))
 814     return MakeErrMsg(ErrMsg, "Can't copy '" + Src.str() +
 815                "' to '" + Dest.str() + "': ");
 816   return false;
 817 }
 818
 819 bool
 820 Path::makeUnique(bool reuse_current, std::string* ErrMsg) {
 821   if (reuse_current && !exists())
 822     return false; // File doesn't exist already, just use it!
 823
 824   // Reserve space for -XXXXXX at the end.
 825   char *FNBuffer = (char*) alloca(path.size()+8);
 826   unsigned offset = path.size();
 827   path.copy(FNBuffer, offset);
 828
 829   // Find a numeric suffix that isn't used by an existing file.  Assume there
 830   // won't be more than 1 million files with the same prefix.  Probably a safe
 831   // bet.
 832   static unsigned FCounter = 0;
 833   do {
 834     sprintf(FNBuffer+offset, "-%06u", FCounter);
 835     if (++FCounter > 999999)
 836       FCounter = 0;
 837     path = FNBuffer;
 838   } while (exists());
 839   return false;
 840 }
 841
 842 bool
 843 Path::createTemporaryFileOnDisk(bool reuse_current, std::string* ErrMsg) {
 844   // Make this into a unique file name
 845   makeUnique(reuse_current, ErrMsg);
 846
 847   // Now go and create it
 848   HANDLE h = CreateFile(path.c_str(), GENERIC_WRITE, 0, NULL, CREATE_NEW,
 849                         FILE_ATTRIBUTE_NORMAL, NULL);
 850   if (h == INVALID_HANDLE_VALUE)
 851     return MakeErrMsg(ErrMsg, path + ": can't create file");
 852
 853   CloseHandle(h);
 854   return false;
 855 }
 856
 857 /// MapInFilePages - Not yet implemented on win32.
 858 const char *Path::MapInFilePages(int FD, uint64_t FileSize) {
 859   return 0;
 860 }
 861
 862 /// MapInFilePages - Not yet implemented on win32.
 863 void Path::UnMapFilePages(const char *Base, uint64_t FileSize) {
 864   assert(0 && "NOT IMPLEMENTED");
 865 }
 866
 867 }
 868 }