Files
carbon-lang/common/check_internal.h
T
Jon Meow 20728dbd3a CARBON_ header guards (#1261)
This modifies scripts/check_header_guards.py to add the CARBON_ prefix; everything else is pre-commit.
2022-05-12 17:25:43 -07:00

82 lines
2.6 KiB
C++

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#ifndef CARBON_COMMON_CHECK_INTERNAL_H_
#define CARBON_COMMON_CHECK_INTERNAL_H_
#include <unistd.h>
#include "llvm/Support/ErrorHandling.h"
#include "llvm/Support/Signals.h"
#include "llvm/Support/raw_ostream.h"
namespace Carbon::Internal {
// Wraps a stream and exiting for fatal errors. Should only be used by check.h
// macros.
class ExitingStream {
public:
// A tag type that renders as ": " in an ExitingStream, but only if it is
// followed by additional output. Otherwise, it renders as "". Primarily used
// when building macros around these streams.
struct AddSeparator {};
// Internal type used in macros to dispatch to the `operator|` overload.
struct Helper {};
ExitingStream() {
// Start all messages with a stack trace.
llvm::errs() << "Stack trace:\n";
llvm::sys::PrintStackTrace(llvm::errs());
}
[[noreturn]] ~ExitingStream() {
llvm_unreachable(
"Exiting streams should only be constructed by check.h macros that "
"ensure the special operator| exits the program prior to their "
"destruction!");
}
// If the bool cast occurs, it's because the condition is false. This supports
// && short-circuiting the creation of ExitingStream.
explicit operator bool() const { return true; }
// Forward output to llvm::errs.
template <typename T>
auto operator<<(const T& message) -> ExitingStream& {
if (separator_) {
llvm::errs() << ": ";
separator_ = false;
}
llvm::errs() << message;
return *this;
}
auto operator<<(AddSeparator /*add_separator*/) -> ExitingStream& {
separator_ = true;
return *this;
}
// Low-precedence binary operator overload used in check.h macros to flush the
// output and exit the program. We do this in a binary operator rather than
// the destructor to ensure good debug info and backtraces for errors.
[[noreturn]] friend auto operator|(Helper /*helper*/,
ExitingStream& /*rhs*/) {
// Finish with a newline.
llvm::errs() << "\n";
// We assume LLVM's exit handling is installed, which will stack trace on
// std::abort(). We print a stack trace on construction, so this avoids that
// stack trace on exit.
_exit(1);
}
private:
// Whether a separator should be printed if << is used again.
bool separator_ = false;
};
} // namespace Carbon::Internal
#endif // CARBON_COMMON_CHECK_INTERNAL_H_