Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //===- llvm/Support/FileUtilities.h - File System Utilities -----*- C++ -*-===// | 
| 2 | // | ||
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | ||
| 4 | // See https://llvm.org/LICENSE.txt for license information. | ||
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
| 6 | // | ||
| 7 | //===----------------------------------------------------------------------===// | ||
| 8 | // | ||
| 9 | // This file defines a family of utility functions which are useful for doing | ||
| 10 | // various things with files. | ||
| 11 | // | ||
| 12 | //===----------------------------------------------------------------------===// | ||
| 13 | |||
| 14 | #ifndef LLVM_SUPPORT_FILEUTILITIES_H | ||
| 15 | #define LLVM_SUPPORT_FILEUTILITIES_H | ||
| 16 | |||
| 17 | #include "llvm/ADT/StringRef.h" | ||
| 18 | #include "llvm/Support/Error.h" | ||
| 19 | #include "llvm/Support/FileSystem.h" | ||
| 20 | |||
| 21 | #include <system_error> | ||
| 22 | |||
| 23 | namespace llvm { | ||
| 24 | |||
| 25 |   /// DiffFilesWithTolerance - Compare the two files specified, returning 0 if | ||
| 26 |   /// the files match, 1 if they are different, and 2 if there is a file error. | ||
| 27 |   /// This function allows you to specify an absolute and relative FP error that | ||
| 28 |   /// is allowed to exist.  If you specify a string to fill in for the error | ||
| 29 |   /// option, it will set the string to an error message if an error occurs, or | ||
| 30 |   /// if the files are different. | ||
| 31 |   /// | ||
| 32 | int DiffFilesWithTolerance(StringRef FileA, | ||
| 33 | StringRef FileB, | ||
| 34 | double AbsTol, double RelTol, | ||
| 35 | std::string *Error = nullptr); | ||
| 36 | |||
| 37 | |||
| 38 |   /// FileRemover - This class is a simple object meant to be stack allocated. | ||
| 39 |   /// If an exception is thrown from a region, the object removes the filename | ||
| 40 |   /// specified (if deleteIt is true). | ||
| 41 |   /// | ||
| 42 | class FileRemover { | ||
| 43 | SmallString<128> Filename; | ||
| 44 | bool DeleteIt; | ||
| 45 | public: | ||
| 46 | FileRemover() : DeleteIt(false) {} | ||
| 47 | |||
| 48 | explicit FileRemover(const Twine& filename, bool deleteIt = true) | ||
| 49 | : DeleteIt(deleteIt) { | ||
| 50 | filename.toVector(Filename); | ||
| 51 |     } | ||
| 52 | |||
| 53 | ~FileRemover() { | ||
| 54 | if (DeleteIt) { | ||
| 55 |         // Ignore problems deleting the file. | ||
| 56 | sys::fs::remove(Filename); | ||
| 57 |       } | ||
| 58 |     } | ||
| 59 | |||
| 60 |     /// setFile - Give ownership of the file to the FileRemover so it will | ||
| 61 |     /// be removed when the object is destroyed.  If the FileRemover already | ||
| 62 |     /// had ownership of a file, remove it first. | ||
| 63 | void setFile(const Twine& filename, bool deleteIt = true) { | ||
| 64 | if (DeleteIt) { | ||
| 65 |         // Ignore problems deleting the file. | ||
| 66 | sys::fs::remove(Filename); | ||
| 67 |       } | ||
| 68 | |||
| 69 | Filename.clear(); | ||
| 70 | filename.toVector(Filename); | ||
| 71 | DeleteIt = deleteIt; | ||
| 72 |     } | ||
| 73 | |||
| 74 |     /// releaseFile - Take ownership of the file away from the FileRemover so it | ||
| 75 |     /// will not be removed when the object is destroyed. | ||
| 76 | void releaseFile() { DeleteIt = false; } | ||
| 77 | }; | ||
| 78 | |||
| 79 | enum class atomic_write_error { | ||
| 80 | failed_to_create_uniq_file = 0, | ||
| 81 | output_stream_error, | ||
| 82 | failed_to_rename_temp_file | ||
| 83 | }; | ||
| 84 | |||
| 85 | class AtomicFileWriteError : public llvm::ErrorInfo<AtomicFileWriteError> { | ||
| 86 | public: | ||
| 87 | AtomicFileWriteError(atomic_write_error Error) : Error(Error) {} | ||
| 88 | |||
| 89 | void log(raw_ostream &OS) const override; | ||
| 90 | |||
| 91 | const atomic_write_error Error; | ||
| 92 | static char ID; | ||
| 93 | |||
| 94 | private: | ||
| 95 |     // Users are not expected to use error_code. | ||
| 96 | std::error_code convertToErrorCode() const override { | ||
| 97 | return llvm::inconvertibleErrorCode(); | ||
| 98 |     } | ||
| 99 | }; | ||
| 100 | |||
| 101 |   // atomic_write_error + whatever the Writer can return | ||
| 102 | |||
| 103 |   /// Creates a unique file with name according to the given \p TempPathModel, | ||
| 104 |   /// writes content of \p Buffer to the file and renames it to \p FinalPath. | ||
| 105 |   /// | ||
| 106 |   /// \returns \c AtomicFileWriteError in case of error. | ||
| 107 | llvm::Error writeFileAtomically(StringRef TempPathModel, StringRef FinalPath, | ||
| 108 | StringRef Buffer); | ||
| 109 | |||
| 110 | llvm::Error | ||
| 111 |   writeFileAtomically(StringRef TempPathModel, StringRef FinalPath, | ||
| 112 | std::function<llvm::Error(llvm::raw_ostream &)> Writer); | ||
| 113 | |||
| 114 |   /// FilePermssionsApplier helps to copy permissions from an input file to | ||
| 115 |   /// an output one. It memorizes the status of the input file and can apply | ||
| 116 |   /// permissions and dates to the output file. | ||
| 117 | class FilePermissionsApplier { | ||
| 118 | public: | ||
| 119 | static Expected<FilePermissionsApplier> create(StringRef InputFilename); | ||
| 120 | |||
| 121 |     /// Apply stored permissions to the \p OutputFilename. | ||
| 122 |     /// Copy LastAccess and ModificationTime if \p CopyDates is true. | ||
| 123 |     /// Overwrite stored permissions if \p OverwritePermissions is specified. | ||
| 124 | Error | ||
| 125 | apply(StringRef OutputFilename, bool CopyDates = false, | ||
| 126 | std::optional<sys::fs::perms> OverwritePermissions = std::nullopt); | ||
| 127 | |||
| 128 | private: | ||
| 129 | FilePermissionsApplier(StringRef InputFilename, sys::fs::file_status Status) | ||
| 130 | : InputFilename(InputFilename), InputStatus(Status) {} | ||
| 131 | |||
| 132 |     StringRef InputFilename; | ||
| 133 | sys::fs::file_status InputStatus; | ||
| 134 | }; | ||
| 135 | } // End llvm namespace | ||
| 136 | |||
| 137 | #endif |