Files
swift-mirror/lib/Basic/Assertions.cpp
2024-05-16 15:04:49 -07:00

94 lines
2.7 KiB
C++

//===--- Assertions.cpp - Swift Version Number -------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2023 - 2023 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
//
// This file defines custom assertion support functions
//
//===----------------------------------------------------------------------===//
#include "llvm/Support/CommandLine.h"
#include "swift/Basic/Assertions.h"
#undef NDEBUG
#include <cassert>
#include <iostream>
llvm::cl::opt<bool> AssertContinue(
"assert-continue", llvm::cl::init(false),
llvm::cl::desc("Do not stop on an assertion failure"));
llvm::cl::opt<bool> AssertHelp(
"assert-help", llvm::cl::init(false),
llvm::cl::desc("Print help for managing assertions"));
int CONDITIONAL_ASSERT_Global_enable_flag =
#ifdef NDEBUG
0; // Default to `off` in release builds
#else
0; // TODO: Default to `on` in debug builds
#endif
void ASSERT_failure(const char *expr, const char *file, int line, const char *func) {
// Only print the last component of `file`
const char *f = file;
for (const char *p = file; *p != '\0'; p++) {
if ((p[0] == '/' || p[0] == '\\')
&& p[1] != '/' && p[1] != '\\' && p[1] != '\0') {
f = p + 1;
}
}
if (AssertHelp) {
ASSERT_help();
} else {
std::cerr << "Assertion help: -Xllvm -assert-help" << std::endl;
}
// Format here matches that used by `assert` on macOS:
std::cerr
<< "Assertion failed: "
<< "(" << expr << "), "
<< "function " << func << ", "
<< "file " << f << ", "
<< "line " << line << "."
<< std::endl;
if (AssertContinue) {
std::cerr << "Continuing after failed assertion (-Xllvm -assert-continue)" << std::endl;
return;
}
abort();
}
void ASSERT_help() {
static int ASSERT_help_shown = 0;
if (ASSERT_help_shown) {
return;
}
ASSERT_help_shown = 1;
std::cerr << std::endl;
std::cerr << "Control assertion behavior with one or more of the following options:" << std::endl;
std::cerr << std::endl;
std::cerr << " -Xllvm -assert-continue" << std::endl;
std::cerr << " Continue after any failed assertion" << std::endl;
std::cerr << std::endl;
}
// This has to be callable in the same way as the macro version,
// so we can't put it inside a namespace.
#undef CONDITIONAL_ASSERT_enabled
int CONDITIONAL_ASSERT_enabled() {
return (CONDITIONAL_ASSERT_Global_enable_flag != 0);
}