1 //===- not.cpp - The 'not' testing tool -----------------------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // Usage: 10 // not cmd 11 // Will return true if cmd doesn't crash and returns false. 12 // not --crash cmd 13 // Will return true if cmd crashes (e.g. for testing crash reporting). 14 15 #include "llvm/Support/Program.h" 16 #include "llvm/Support/raw_ostream.h" 17 using namespace llvm; 18 19 int main(int argc, const char **argv) { 20 bool ExpectCrash = false; 21 22 ++argv; 23 --argc; 24 25 if (argc > 0 && StringRef(argv[0]) == "--crash") { 26 ++argv; 27 --argc; 28 ExpectCrash = true; 29 } 30 31 if (argc == 0) 32 return 1; 33 34 auto Program = sys::findProgramByName(argv[0]); 35 if (!Program) { 36 errs() << "Error: Unable to find `" << argv[0] 37 << "' in PATH: " << Program.getError().message() << "\n"; 38 return 1; 39 } 40 41 std::vector<StringRef> Argv; 42 Argv.reserve(argc); 43 for (int i = 0; i < argc; ++i) 44 Argv.push_back(argv[i]); 45 std::string ErrMsg; 46 int Result = sys::ExecuteAndWait(*Program, Argv, None, {}, 0, 0, &ErrMsg); 47 #ifdef _WIN32 48 // Handle abort() in msvcrt -- It has exit code as 3. abort(), aka 49 // unreachable, should be recognized as a crash. However, some binaries use 50 // exit code 3 on non-crash failure paths, so only do this if we expect a 51 // crash. 52 if (ExpectCrash && Result == 3) 53 Result = -3; 54 #endif 55 if (Result < 0) { 56 errs() << "Error: " << ErrMsg << "\n"; 57 if (ExpectCrash) 58 return 0; 59 return 1; 60 } 61 62 if (ExpectCrash) 63 return 1; 64 65 return Result == 0; 66 } 67
