Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===- ErrorHandler.h -------------------------------------------*- 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 | // We designed lld's error handlers with the following goals in mind: |
||
| 10 | // |
||
| 11 | // - Errors can occur at any place where we handle user input, but we don't |
||
| 12 | // want them to affect the normal execution path too much. Ideally, |
||
| 13 | // handling errors should be as simple as reporting them and exit (but |
||
| 14 | // without actually doing exit). |
||
| 15 | // |
||
| 16 | // In particular, the design to wrap all functions that could fail with |
||
| 17 | // ErrorOr<T> is rejected because otherwise we would have to wrap a large |
||
| 18 | // number of functions in lld with ErrorOr. With that approach, if some |
||
| 19 | // function F can fail, not only F but all functions that transitively call |
||
| 20 | // F have to be wrapped with ErrorOr. That seemed too much. |
||
| 21 | // |
||
| 22 | // - Finding only one error at a time is not sufficient. We want to find as |
||
| 23 | // many errors as possible with one execution of the linker. That means the |
||
| 24 | // linker needs to keep running after a first error and give up at some |
||
| 25 | // checkpoint (beyond which it would find cascading, false errors caused by |
||
| 26 | // the previous errors). |
||
| 27 | // |
||
| 28 | // - We want a simple interface to report errors. Unlike Clang, the data we |
||
| 29 | // handle is compiled binary, so we don't need an error reporting mechanism |
||
| 30 | // that's as sophisticated as the one that Clang has. |
||
| 31 | // |
||
| 32 | // The current lld's error handling mechanism is simple: |
||
| 33 | // |
||
| 34 | // - When you find an error, report it using error() and continue as far as |
||
| 35 | // you can. An internal error counter is incremented by one every time you |
||
| 36 | // call error(). |
||
| 37 | // |
||
| 38 | // A common idiom to handle an error is calling error() and then returning |
||
| 39 | // a reasonable default value. For example, if your function handles a |
||
| 40 | // user-supplied alignment value, and if you find an invalid alignment |
||
| 41 | // (e.g. 17 which is not 2^n), you may report it using error() and continue |
||
| 42 | // as if it were alignment 1 (which is the simplest reasonable value). |
||
| 43 | // |
||
| 44 | // Note that you should not continue with an invalid value; that breaks the |
||
| 45 | // internal consistency. You need to maintain all variables have some sane |
||
| 46 | // value even after an error occurred. So, when you have to continue with |
||
| 47 | // some value, always use a dummy value. |
||
| 48 | // |
||
| 49 | // - Find a reasonable checkpoint at where you want to stop the linker, and |
||
| 50 | // add code to return from the function if errorCount() > 0. In most cases, |
||
| 51 | // a checkpoint already exists, so you don't need to do anything for this. |
||
| 52 | // |
||
| 53 | // This interface satisfies all the goals that we mentioned above. |
||
| 54 | // |
||
| 55 | // You should never call fatal() except for reporting a corrupted input file. |
||
| 56 | // fatal() immediately terminates the linker, so the function is not desirable |
||
| 57 | // if you are using lld as a subroutine in other program, and with that you |
||
| 58 | // can find only one error at a time. |
||
| 59 | // |
||
| 60 | // warn() doesn't do anything but printing out a given message. |
||
| 61 | // |
||
| 62 | // It is not recommended to use llvm::outs() or lld::errs() directly in lld |
||
| 63 | // because they are not thread-safe. The functions declared in this file are |
||
| 64 | // thread-safe. |
||
| 65 | // |
||
| 66 | //===----------------------------------------------------------------------===// |
||
| 67 | |||
| 68 | #ifndef LLD_COMMON_ERRORHANDLER_H |
||
| 69 | #define LLD_COMMON_ERRORHANDLER_H |
||
| 70 | |||
| 71 | #include "lld/Common/LLVM.h" |
||
| 72 | |||
| 73 | #include "llvm/ADT/STLExtras.h" |
||
| 74 | #include "llvm/Support/Error.h" |
||
| 75 | #include "llvm/Support/FileOutputBuffer.h" |
||
| 76 | #include <mutex> |
||
| 77 | |||
| 78 | namespace llvm { |
||
| 79 | class DiagnosticInfo; |
||
| 80 | class raw_ostream; |
||
| 81 | } |
||
| 82 | |||
| 83 | namespace lld { |
||
| 84 | |||
| 85 | llvm::raw_ostream &outs(); |
||
| 86 | llvm::raw_ostream &errs(); |
||
| 87 | |||
| 88 | enum class ErrorTag { LibNotFound, SymbolNotFound }; |
||
| 89 | |||
| 90 | class ErrorHandler { |
||
| 91 | public: |
||
| 92 | ~ErrorHandler(); |
||
| 93 | |||
| 94 | void initialize(llvm::raw_ostream &stdoutOS, llvm::raw_ostream &stderrOS, |
||
| 95 | bool exitEarly, bool disableOutput); |
||
| 96 | |||
| 97 | uint64_t errorCount = 0; |
||
| 98 | uint64_t errorLimit = 20; |
||
| 99 | StringRef errorLimitExceededMsg = "too many errors emitted, stopping now"; |
||
| 100 | StringRef errorHandlingScript; |
||
| 101 | StringRef logName = "lld"; |
||
| 102 | bool exitEarly = true; |
||
| 103 | bool fatalWarnings = false; |
||
| 104 | bool suppressWarnings = false; |
||
| 105 | bool verbose = false; |
||
| 106 | bool vsDiagnostics = false; |
||
| 107 | bool disableOutput = false; |
||
| 108 | std::function<void()> cleanupCallback; |
||
| 109 | |||
| 110 | void error(const Twine &msg); |
||
| 111 | void error(const Twine &msg, ErrorTag tag, ArrayRef<StringRef> args); |
||
| 112 | [[noreturn]] void fatal(const Twine &msg); |
||
| 113 | void log(const Twine &msg); |
||
| 114 | void message(const Twine &msg, llvm::raw_ostream &s); |
||
| 115 | void warn(const Twine &msg); |
||
| 116 | |||
| 117 | raw_ostream &outs(); |
||
| 118 | raw_ostream &errs(); |
||
| 119 | void flushStreams(); |
||
| 120 | |||
| 121 | std::unique_ptr<llvm::FileOutputBuffer> outputBuffer; |
||
| 122 | |||
| 123 | private: |
||
| 124 | using Colors = raw_ostream::Colors; |
||
| 125 | |||
| 126 | std::string getLocation(const Twine &msg); |
||
| 127 | void reportDiagnostic(StringRef location, Colors c, StringRef diagKind, |
||
| 128 | const Twine &msg); |
||
| 129 | |||
| 130 | // We want to separate multi-line messages with a newline. `sep` is "\n" |
||
| 131 | // if the last messages was multi-line. Otherwise "". |
||
| 132 | llvm::StringRef sep; |
||
| 133 | |||
| 134 | // We wrap stdout and stderr so that you can pass alternative stdout/stderr as |
||
| 135 | // arguments to lld::*::link() functions. Since lld::outs() or lld::errs() can |
||
| 136 | // be indirectly called from multiple threads, we protect them using a mutex. |
||
| 137 | // In the future, we plan on supporting several concurrent linker contexts, |
||
| 138 | // which explains why the mutex is not a global but part of this context. |
||
| 139 | std::mutex mu; |
||
| 140 | llvm::raw_ostream *stdoutOS{}; |
||
| 141 | llvm::raw_ostream *stderrOS{}; |
||
| 142 | }; |
||
| 143 | |||
| 144 | /// Returns the default error handler. |
||
| 145 | ErrorHandler &errorHandler(); |
||
| 146 | |||
| 147 | void error(const Twine &msg); |
||
| 148 | void error(const Twine &msg, ErrorTag tag, ArrayRef<StringRef> args); |
||
| 149 | [[noreturn]] void fatal(const Twine &msg); |
||
| 150 | void log(const Twine &msg); |
||
| 151 | void message(const Twine &msg, llvm::raw_ostream &s = outs()); |
||
| 152 | void warn(const Twine &msg); |
||
| 153 | uint64_t errorCount(); |
||
| 154 | |||
| 155 | [[noreturn]] void exitLld(int val); |
||
| 156 | |||
| 157 | void diagnosticHandler(const llvm::DiagnosticInfo &di); |
||
| 158 | void checkError(Error e); |
||
| 159 | |||
| 160 | // check functions are convenient functions to strip errors |
||
| 161 | // from error-or-value objects. |
||
| 162 | template <class T> T check(ErrorOr<T> e) { |
||
| 163 | if (auto ec = e.getError()) |
||
| 164 | fatal(ec.message()); |
||
| 165 | return std::move(*e); |
||
| 166 | } |
||
| 167 | |||
| 168 | template <class T> T check(Expected<T> e) { |
||
| 169 | if (!e) |
||
| 170 | fatal(llvm::toString(e.takeError())); |
||
| 171 | return std::move(*e); |
||
| 172 | } |
||
| 173 | |||
| 174 | // Don't move from Expected wrappers around references. |
||
| 175 | template <class T> T &check(Expected<T &> e) { |
||
| 176 | if (!e) |
||
| 177 | fatal(llvm::toString(e.takeError())); |
||
| 178 | return *e; |
||
| 179 | } |
||
| 180 | |||
| 181 | template <class T> |
||
| 182 | T check2(ErrorOr<T> e, llvm::function_ref<std::string()> prefix) { |
||
| 183 | if (auto ec = e.getError()) |
||
| 184 | fatal(prefix() + ": " + ec.message()); |
||
| 185 | return std::move(*e); |
||
| 186 | } |
||
| 187 | |||
| 188 | template <class T> |
||
| 189 | T check2(Expected<T> e, llvm::function_ref<std::string()> prefix) { |
||
| 190 | if (!e) |
||
| 191 | fatal(prefix() + ": " + toString(e.takeError())); |
||
| 192 | return std::move(*e); |
||
| 193 | } |
||
| 194 | |||
| 195 | inline std::string toString(const Twine &s) { return s.str(); } |
||
| 196 | |||
| 197 | // To evaluate the second argument lazily, we use C macro. |
||
| 198 | #define CHECK(E, S) check2((E), [&] { return toString(S); }) |
||
| 199 | |||
| 200 | } // namespace lld |
||
| 201 | |||
| 202 | #endif |