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 |