| //===- 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; |
| } |