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