148f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar//===- not.cpp - The 'not' testing tool -----------------------------------===// 248f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar// 348f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar// The LLVM Compiler Infrastructure 448f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar// 548f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar// This file is distributed under the University of Illinois Open Source 648f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar// License. See LICENSE.TXT for details. 748f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar// 848f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar//===----------------------------------------------------------------------===// 948f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar 101f6efa3996dd1929fbc129203ce5009b620e6969Michael J. Spencer#include "llvm/Support/Path.h" 111f6efa3996dd1929fbc129203ce5009b620e6969Michael J. Spencer#include "llvm/Support/Program.h" 129dbb79aaac8dc2a94b13ddd4c8fdc659f0167716Dan Gohman#include "llvm/Support/raw_ostream.h" 1348f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbarusing namespace llvm; 1448f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar 1548f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbarint main(int argc, const char **argv) { 16a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola bool ExpectCrash = false; 17a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola 18a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola ++argv; 19a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola --argc; 20a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola 21a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola if (argc > 0 && StringRef(argv[0]) == "--crash") { 22a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola ++argv; 23a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola --argc; 24a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola ExpectCrash = true; 25a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola } 26a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola 27a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola if (argc == 0) 28a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola return 1; 29a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola 30a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola std::string Program = sys::FindProgramByName(argv[0]); 319dbb79aaac8dc2a94b13ddd4c8fdc659f0167716Dan Gohman 329dbb79aaac8dc2a94b13ddd4c8fdc659f0167716Dan Gohman std::string ErrMsg; 33cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines int Result = sys::ExecuteAndWait(Program, argv, nullptr, nullptr, 0, 0, 34cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines &ErrMsg); 35cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines#ifdef _WIN32 36cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // Handle abort() in msvcrt -- It has exit code as 3. abort(), aka 37cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // unreachable, should be recognized as a crash. However, some binaries use 38cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // exit code 3 on non-crash failure paths, so only do this if we expect a 39cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // crash. 40cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines if (ExpectCrash && Result == 3) 41cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Result = -3; 42cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines#endif 439dbb79aaac8dc2a94b13ddd4c8fdc659f0167716Dan Gohman if (Result < 0) { 449dbb79aaac8dc2a94b13ddd4c8fdc659f0167716Dan Gohman errs() << "Error: " << ErrMsg << "\n"; 45a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola if (ExpectCrash) 46a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola return 0; 479dbb79aaac8dc2a94b13ddd4c8fdc659f0167716Dan Gohman return 1; 489dbb79aaac8dc2a94b13ddd4c8fdc659f0167716Dan Gohman } 499dbb79aaac8dc2a94b13ddd4c8fdc659f0167716Dan Gohman 50a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola if (ExpectCrash) 51a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola return 1; 52a5db79d5148d3972b90390f526fd35d707729c5aRafael Espindola 539dbb79aaac8dc2a94b13ddd4c8fdc659f0167716Dan Gohman return Result == 0; 5448f7ce88a6040c12d4c7d0a5bd208cf1e0bda9d1Daniel Dunbar} 55