utils/not/not.cpp

   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
  10 #include "llvm/Support/Path.h"
  11 #include "llvm/Support/Program.h"
  12 #include "llvm/Support/raw_ostream.h"
  13 using namespace llvm;
  14
  15 int main(int argc, const char **argv) {
  16   bool ExpectCrash = false;
  17
  18   ++argv;
  19   --argc;
  20
  21   if (argc > 0 && StringRef(argv[0]) == "--crash") {
  22     ++argv;
  23     --argc;
  24     ExpectCrash = true;
  25   }
  26
  27   if (argc == 0)
  28     return 1;
  29
  30   auto Program = sys::findProgramByName(argv[0]);
  31   if (!Program) {
  32     errs() << "Error: Unable to find `" << argv[0]
  33            << "' in PATH: " << Program.getError().message() << "\n";
  34     return 1;
  35   }
  36
  37   std::string ErrMsg;
  38   int Result = sys::ExecuteAndWait(*Program, argv, nullptr, nullptr, 0, 0,
  39                                    &ErrMsg);
  40 #ifdef _WIN32
  41   // Handle abort() in msvcrt -- It has exit code as 3.  abort(), aka
  42   // unreachable, should be recognized as a crash.  However, some binaries use
  43   // exit code 3 on non-crash failure paths, so only do this if we expect a
  44   // crash.
  45   if (ExpectCrash && Result == 3)
  46     Result = -3;
  47 #endif
  48   if (Result < 0) {
  49     errs() << "Error: " << ErrMsg << "\n";
  50     if (ExpectCrash)
  51       return 0;
  52     return 1;
  53   }
  54
  55   if (ExpectCrash)
  56     return 1;
  57
  58   return Result == 0;
  59 }