forked from OSchip/llvm-project
82 lines
2.2 KiB
C++
82 lines
2.2 KiB
C++
//===- not.cpp - The 'not' testing tool -----------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
// Usage:
|
|
// not cmd
|
|
// Will return true if cmd doesn't crash and returns false.
|
|
// not --crash cmd
|
|
// Will return true if cmd crashes (e.g. for testing crash reporting).
|
|
|
|
#include "llvm/Support/Program.h"
|
|
#include "llvm/Support/WithColor.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
using namespace llvm;
|
|
|
|
int main(int argc, const char **argv) {
|
|
bool ExpectCrash = false;
|
|
|
|
++argv;
|
|
--argc;
|
|
|
|
if (argc > 0 && StringRef(argv[0]) == "--crash") {
|
|
++argv;
|
|
--argc;
|
|
ExpectCrash = true;
|
|
|
|
// Crash is expected, so disable crash report and symbolization to reduce
|
|
// output and avoid potentially slow symbolization.
|
|
#ifdef _WIN32
|
|
SetEnvironmentVariableA("LLVM_DISABLE_CRASH_REPORT", "1");
|
|
SetEnvironmentVariableA("LLVM_DISABLE_SYMBOLIZATION", "1");
|
|
#else
|
|
setenv("LLVM_DISABLE_CRASH_REPORT", "1", 0);
|
|
setenv("LLVM_DISABLE_SYMBOLIZATION", "1", 0);
|
|
#endif
|
|
}
|
|
|
|
if (argc == 0)
|
|
return 1;
|
|
|
|
auto Program = sys::findProgramByName(argv[0]);
|
|
if (!Program) {
|
|
WithColor::error() << "unable to find `" << argv[0]
|
|
<< "' in PATH: " << Program.getError().message() << "\n";
|
|
return 1;
|
|
}
|
|
|
|
std::vector<StringRef> Argv;
|
|
Argv.reserve(argc);
|
|
for (int i = 0; i < argc; ++i)
|
|
Argv.push_back(argv[i]);
|
|
std::string ErrMsg;
|
|
int Result = sys::ExecuteAndWait(*Program, Argv, None, {}, 0, 0, &ErrMsg);
|
|
#ifdef _WIN32
|
|
// Handle abort() in msvcrt -- It has exit code as 3. abort(), aka
|
|
// unreachable, should be recognized as a crash. However, some binaries use
|
|
// exit code 3 on non-crash failure paths, so only do this if we expect a
|
|
// crash.
|
|
if (ExpectCrash && Result == 3)
|
|
Result = -3;
|
|
#endif
|
|
if (Result < 0) {
|
|
WithColor::error() << ErrMsg << "\n";
|
|
if (ExpectCrash)
|
|
return 0;
|
|
return 1;
|
|
}
|
|
|
|
if (ExpectCrash)
|
|
return 1;
|
|
|
|
return Result == 0;
|
|
}
|