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 |