Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- Job.h - Commands to Execute ------------------------------*- 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 | #ifndef LLVM_CLANG_DRIVER_JOB_H |
||
10 | #define LLVM_CLANG_DRIVER_JOB_H |
||
11 | |||
12 | #include "clang/Basic/LLVM.h" |
||
13 | #include "clang/Driver/InputInfo.h" |
||
14 | #include "llvm/ADT/ArrayRef.h" |
||
15 | #include "llvm/ADT/SmallVector.h" |
||
16 | #include "llvm/ADT/StringRef.h" |
||
17 | #include "llvm/ADT/iterator.h" |
||
18 | #include "llvm/Option/Option.h" |
||
19 | #include "llvm/Support/Program.h" |
||
20 | #include <memory> |
||
21 | #include <optional> |
||
22 | #include <string> |
||
23 | #include <utility> |
||
24 | #include <vector> |
||
25 | |||
26 | namespace clang { |
||
27 | namespace driver { |
||
28 | |||
29 | class Action; |
||
30 | class InputInfo; |
||
31 | class Tool; |
||
32 | |||
33 | struct CrashReportInfo { |
||
34 | StringRef Filename; |
||
35 | StringRef VFSPath; |
||
36 | |||
37 | CrashReportInfo(StringRef Filename, StringRef VFSPath) |
||
38 | : Filename(Filename), VFSPath(VFSPath) {} |
||
39 | }; |
||
40 | |||
41 | // Encodes the kind of response file supported for a command invocation. |
||
42 | // Response files are necessary if the command line gets too large, requiring |
||
43 | // the arguments to be transferred to a file. |
||
44 | struct ResponseFileSupport { |
||
45 | enum ResponseFileKind { |
||
46 | // Provides full support for response files, which means we can transfer |
||
47 | // all tool input arguments to a file. |
||
48 | RF_Full, |
||
49 | // Input file names can live in a file, but flags can't. This is a special |
||
50 | // case for old versions of Apple's ld64. |
||
51 | RF_FileList, |
||
52 | // Does not support response files: all arguments must be passed via |
||
53 | // command line. |
||
54 | RF_None |
||
55 | }; |
||
56 | /// The level of support for response files. |
||
57 | ResponseFileKind ResponseKind; |
||
58 | |||
59 | /// The encoding to use when writing response files on Windows. Ignored on |
||
60 | /// other host OSes. |
||
61 | /// |
||
62 | /// Windows use cases: - GCC and Binutils on mingw only accept ANSI response |
||
63 | /// files encoded with the system current code page. |
||
64 | /// - MSVC's CL.exe and LINK.exe accept UTF16 on Windows. |
||
65 | /// - Clang accepts both UTF8 and UTF16. |
||
66 | /// |
||
67 | /// FIXME: When GNU tools learn how to parse UTF16 on Windows, we should |
||
68 | /// always use UTF16 for Windows, which is the Windows official encoding for |
||
69 | /// international characters. |
||
70 | llvm::sys::WindowsEncodingMethod ResponseEncoding; |
||
71 | |||
72 | /// What prefix to use for the command-line argument when passing a response |
||
73 | /// file. |
||
74 | const char *ResponseFlag; |
||
75 | |||
76 | /// Returns a ResponseFileSupport indicating that response files are not |
||
77 | /// supported. |
||
78 | static constexpr ResponseFileSupport None() { |
||
79 | return {RF_None, llvm::sys::WEM_UTF8, nullptr}; |
||
80 | } |
||
81 | |||
82 | /// Returns a ResponseFileSupport indicating that response files are |
||
83 | /// supported, using the @file syntax. On windows, the file is written in the |
||
84 | /// UTF8 encoding. On other OSes, no re-encoding occurs. |
||
85 | static constexpr ResponseFileSupport AtFileUTF8() { |
||
86 | return {RF_Full, llvm::sys::WEM_UTF8, "@"}; |
||
87 | } |
||
88 | |||
89 | /// Returns a ResponseFileSupport indicating that response files are |
||
90 | /// supported, using the @file syntax. On windows, the file is written in the |
||
91 | /// current ANSI code-page encoding. On other OSes, no re-encoding occurs. |
||
92 | static constexpr ResponseFileSupport AtFileCurCP() { |
||
93 | return {RF_Full, llvm::sys::WEM_CurrentCodePage, "@"}; |
||
94 | } |
||
95 | |||
96 | /// Returns a ResponseFileSupport indicating that response files are |
||
97 | /// supported, using the @file syntax. On windows, the file is written in the |
||
98 | /// UTF-16 encoding. On other OSes, no re-encoding occurs. |
||
99 | static constexpr ResponseFileSupport AtFileUTF16() { |
||
100 | return {RF_Full, llvm::sys::WEM_UTF16, "@"}; |
||
101 | } |
||
102 | }; |
||
103 | |||
104 | /// Command - An executable path/name and argument vector to |
||
105 | /// execute. |
||
106 | class Command { |
||
107 | /// Source - The action which caused the creation of this job. |
||
108 | const Action &Source; |
||
109 | |||
110 | /// Tool - The tool which caused the creation of this job. |
||
111 | const Tool &Creator; |
||
112 | |||
113 | /// Whether and how to generate response files if the arguments are too long. |
||
114 | ResponseFileSupport ResponseSupport; |
||
115 | |||
116 | /// The executable to run. |
||
117 | const char *Executable; |
||
118 | |||
119 | /// The list of program arguments (not including the implicit first |
||
120 | /// argument, which will be the executable). |
||
121 | llvm::opt::ArgStringList Arguments; |
||
122 | |||
123 | /// The list of program inputs. |
||
124 | std::vector<InputInfo> InputInfoList; |
||
125 | |||
126 | /// The list of program arguments which are outputs. May be empty. |
||
127 | std::vector<std::string> OutputFilenames; |
||
128 | |||
129 | /// Response file name, if this command is set to use one, or nullptr |
||
130 | /// otherwise |
||
131 | const char *ResponseFile = nullptr; |
||
132 | |||
133 | /// The input file list in case we need to emit a file list instead of a |
||
134 | /// proper response file |
||
135 | llvm::opt::ArgStringList InputFileList; |
||
136 | |||
137 | /// String storage if we need to create a new argument to specify a response |
||
138 | /// file |
||
139 | std::string ResponseFileFlag; |
||
140 | |||
141 | /// See Command::setEnvironment |
||
142 | std::vector<const char *> Environment; |
||
143 | |||
144 | /// Optional redirection for stdin, stdout, stderr. |
||
145 | std::vector<std::optional<std::string>> RedirectFiles; |
||
146 | |||
147 | /// Information on executable run provided by OS. |
||
148 | mutable std::optional<llvm::sys::ProcessStatistics> ProcStat; |
||
149 | |||
150 | /// When a response file is needed, we try to put most arguments in an |
||
151 | /// exclusive file, while others remains as regular command line arguments. |
||
152 | /// This functions fills a vector with the regular command line arguments, |
||
153 | /// argv, excluding the ones passed in a response file. |
||
154 | void buildArgvForResponseFile(llvm::SmallVectorImpl<const char *> &Out) const; |
||
155 | |||
156 | /// Encodes an array of C strings into a single string separated by whitespace. |
||
157 | /// This function will also put in quotes arguments that have whitespaces and |
||
158 | /// will escape the regular backslashes (used in Windows paths) and quotes. |
||
159 | /// The results are the contents of a response file, written into a raw_ostream. |
||
160 | void writeResponseFile(raw_ostream &OS) const; |
||
161 | |||
162 | public: |
||
163 | /// Whether to print the input filenames when executing. |
||
164 | bool PrintInputFilenames = false; |
||
165 | |||
166 | /// Whether the command will be executed in this process or not. |
||
167 | bool InProcess = false; |
||
168 | |||
169 | Command(const Action &Source, const Tool &Creator, |
||
170 | ResponseFileSupport ResponseSupport, const char *Executable, |
||
171 | const llvm::opt::ArgStringList &Arguments, ArrayRef<InputInfo> Inputs, |
||
172 | ArrayRef<InputInfo> Outputs = std::nullopt); |
||
173 | // FIXME: This really shouldn't be copyable, but is currently copied in some |
||
174 | // error handling in Driver::generateCompilationDiagnostics. |
||
175 | Command(const Command &) = default; |
||
176 | virtual ~Command() = default; |
||
177 | |||
178 | virtual void Print(llvm::raw_ostream &OS, const char *Terminator, bool Quote, |
||
179 | CrashReportInfo *CrashInfo = nullptr) const; |
||
180 | |||
181 | virtual int Execute(ArrayRef<std::optional<StringRef>> Redirects, |
||
182 | std::string *ErrMsg, bool *ExecutionFailed) const; |
||
183 | |||
184 | /// getSource - Return the Action which caused the creation of this job. |
||
185 | const Action &getSource() const { return Source; } |
||
186 | |||
187 | /// getCreator - Return the Tool which caused the creation of this job. |
||
188 | const Tool &getCreator() const { return Creator; } |
||
189 | |||
190 | /// Returns the kind of response file supported by the current invocation. |
||
191 | const ResponseFileSupport &getResponseFileSupport() { |
||
192 | return ResponseSupport; |
||
193 | } |
||
194 | |||
195 | /// Set to pass arguments via a response file when launching the command |
||
196 | void setResponseFile(const char *FileName); |
||
197 | |||
198 | /// Set an input file list, necessary if you specified an RF_FileList response |
||
199 | /// file support. |
||
200 | void setInputFileList(llvm::opt::ArgStringList List) { |
||
201 | InputFileList = std::move(List); |
||
202 | } |
||
203 | |||
204 | /// Sets the environment to be used by the new process. |
||
205 | /// \param NewEnvironment An array of environment variables. |
||
206 | /// \remark If the environment remains unset, then the environment |
||
207 | /// from the parent process will be used. |
||
208 | virtual void setEnvironment(llvm::ArrayRef<const char *> NewEnvironment); |
||
209 | |||
210 | void |
||
211 | setRedirectFiles(const std::vector<std::optional<std::string>> &Redirects); |
||
212 | |||
213 | void replaceArguments(llvm::opt::ArgStringList List) { |
||
214 | Arguments = std::move(List); |
||
215 | } |
||
216 | |||
217 | void replaceExecutable(const char *Exe) { Executable = Exe; } |
||
218 | |||
219 | const char *getExecutable() const { return Executable; } |
||
220 | |||
221 | const llvm::opt::ArgStringList &getArguments() const { return Arguments; } |
||
222 | |||
223 | const std::vector<InputInfo> &getInputInfos() const { return InputInfoList; } |
||
224 | |||
225 | const std::vector<std::string> &getOutputFilenames() const { |
||
226 | return OutputFilenames; |
||
227 | } |
||
228 | |||
229 | std::optional<llvm::sys::ProcessStatistics> getProcessStatistics() const { |
||
230 | return ProcStat; |
||
231 | } |
||
232 | |||
233 | protected: |
||
234 | /// Optionally print the filenames to be compiled |
||
235 | void PrintFileNames() const; |
||
236 | }; |
||
237 | |||
238 | /// Use the CC1 tool callback when available, to avoid creating a new process |
||
239 | class CC1Command : public Command { |
||
240 | public: |
||
241 | CC1Command(const Action &Source, const Tool &Creator, |
||
242 | ResponseFileSupport ResponseSupport, const char *Executable, |
||
243 | const llvm::opt::ArgStringList &Arguments, |
||
244 | ArrayRef<InputInfo> Inputs, |
||
245 | ArrayRef<InputInfo> Outputs = std::nullopt); |
||
246 | |||
247 | void Print(llvm::raw_ostream &OS, const char *Terminator, bool Quote, |
||
248 | CrashReportInfo *CrashInfo = nullptr) const override; |
||
249 | |||
250 | int Execute(ArrayRef<std::optional<StringRef>> Redirects, std::string *ErrMsg, |
||
251 | bool *ExecutionFailed) const override; |
||
252 | |||
253 | void setEnvironment(llvm::ArrayRef<const char *> NewEnvironment) override; |
||
254 | }; |
||
255 | |||
256 | /// Like Command, but always pretends that the wrapped command succeeded. |
||
257 | class ForceSuccessCommand : public Command { |
||
258 | public: |
||
259 | ForceSuccessCommand(const Action &Source_, const Tool &Creator_, |
||
260 | ResponseFileSupport ResponseSupport, |
||
261 | const char *Executable_, |
||
262 | const llvm::opt::ArgStringList &Arguments_, |
||
263 | ArrayRef<InputInfo> Inputs, |
||
264 | ArrayRef<InputInfo> Outputs = std::nullopt); |
||
265 | |||
266 | void Print(llvm::raw_ostream &OS, const char *Terminator, bool Quote, |
||
267 | CrashReportInfo *CrashInfo = nullptr) const override; |
||
268 | |||
269 | int Execute(ArrayRef<std::optional<StringRef>> Redirects, std::string *ErrMsg, |
||
270 | bool *ExecutionFailed) const override; |
||
271 | }; |
||
272 | |||
273 | /// JobList - A sequence of jobs to perform. |
||
274 | class JobList { |
||
275 | public: |
||
276 | using list_type = SmallVector<std::unique_ptr<Command>, 4>; |
||
277 | using size_type = list_type::size_type; |
||
278 | using iterator = llvm::pointee_iterator<list_type::iterator>; |
||
279 | using const_iterator = llvm::pointee_iterator<list_type::const_iterator>; |
||
280 | |||
281 | private: |
||
282 | list_type Jobs; |
||
283 | |||
284 | public: |
||
285 | void Print(llvm::raw_ostream &OS, const char *Terminator, |
||
286 | bool Quote, CrashReportInfo *CrashInfo = nullptr) const; |
||
287 | |||
288 | /// Add a job to the list (taking ownership). |
||
289 | void addJob(std::unique_ptr<Command> J) { Jobs.push_back(std::move(J)); } |
||
290 | |||
291 | /// Clear the job list. |
||
292 | void clear(); |
||
293 | |||
294 | const list_type &getJobs() const { return Jobs; } |
||
295 | |||
296 | bool empty() const { return Jobs.empty(); } |
||
297 | size_type size() const { return Jobs.size(); } |
||
298 | iterator begin() { return Jobs.begin(); } |
||
299 | const_iterator begin() const { return Jobs.begin(); } |
||
300 | iterator end() { return Jobs.end(); } |
||
301 | const_iterator end() const { return Jobs.end(); } |
||
302 | }; |
||
303 | |||
304 | } // namespace driver |
||
305 | } // namespace clang |
||
306 | |||
307 | #endif // LLVM_CLANG_DRIVER_JOB_H |