Chris Lattner | 7a6ff2b | 2003-08-01 21:16:14 +0000 | [diff] [blame] | 1 | //===- Support/FileUtilities.cpp - File System Utilities ------------------===// |
| 2 | // |
| 3 | // This file implements a family of utility functions which are useful for doing |
| 4 | // various things with files. |
| 5 | // |
| 6 | //===----------------------------------------------------------------------===// |
| 7 | |
| 8 | #include "Support/FileUtilities.h" |
| 9 | #include <fstream> |
| 10 | #include <iostream> |
| 11 | #include <cstdio> |
| 12 | |
| 13 | /// DiffFiles - Compare the two files specified, returning true if they are |
| 14 | /// different or if there is a file error. If you specify a string to fill in |
| 15 | /// for the error option, it will set the string to an error message if an error |
| 16 | /// occurs, allowing the caller to distinguish between a failed diff and a file |
| 17 | /// system error. |
| 18 | /// |
| 19 | bool DiffFiles(const std::string &FileA, const std::string &FileB, |
| 20 | std::string *Error) { |
| 21 | std::ifstream FileAStream(FileA.c_str()); |
| 22 | if (!FileAStream) { |
| 23 | if (Error) *Error = "Couldn't open file '" + FileA + "'"; |
| 24 | return true; |
| 25 | } |
| 26 | |
| 27 | std::ifstream FileBStream(FileB.c_str()); |
| 28 | if (!FileBStream) { |
| 29 | if (Error) *Error = "Couldn't open file '" + FileB + "'"; |
| 30 | return true; |
| 31 | } |
| 32 | |
| 33 | // Compare the two files... |
| 34 | int C1, C2; |
| 35 | do { |
| 36 | C1 = FileAStream.get(); |
| 37 | C2 = FileBStream.get(); |
| 38 | if (C1 != C2) return true; |
| 39 | } while (C1 != EOF); |
| 40 | |
| 41 | return false; |
| 42 | } |
| 43 | |
| 44 | |
| 45 | /// MoveFileOverIfUpdated - If the file specified by New is different than Old, |
| 46 | /// or if Old does not exist, move the New file over the Old file. Otherwise, |
| 47 | /// remove the New file. |
| 48 | /// |
| 49 | void MoveFileOverIfUpdated(const std::string &New, const std::string &Old) { |
| 50 | if (DiffFiles(New, Old)) { |
| 51 | if (std::rename(New.c_str(), Old.c_str())) |
| 52 | std::cerr << "Error renaming '" << New << "' to '" << Old << "'!\n"; |
| 53 | } else { |
| 54 | std::remove(New.c_str()); |
| 55 | } |
| 56 | } |