[SystemZ] Add testcase missing from r186073

[oota-llvm.git] / utils / not / not.cpp

diff --git a/utils/not/not.cpp b/utils/not/not.cpp
index 13803dc8674dba4ffc915b14ad5dc738cd908570..ebd16189c9bde9a3ee620ff54ed3ffb1b2353b94 100644 (file)
--- a/utils/not/not.cpp
+++ b/utils/not/not.cpp
@@ -13,14 +13,33 @@
 using namespace llvm;
 
 int main(int argc, const char **argv) {
-  std::string Program = sys::FindProgramByName(argv[1]);
+  bool ExpectCrash = false;
+
+  ++argv;
+  --argc;
+
+  if (argc > 0 && StringRef(argv[0]) == "--crash") {
+    ++argv;
+    --argc;
+    ExpectCrash = true;
+  }
+
+  if (argc == 0)
+    return 1;
+
+  std::string Program = sys::FindProgramByName(argv[0]);
 
   std::string ErrMsg;
-  int Result = sys::ExecuteAndWait(Program, argv + 1, 0, 0, 0, 0, &ErrMsg);
+  int Result = sys::ExecuteAndWait(Program, argv, 0, 0, 0, 0, &ErrMsg);
   if (Result < 0) {
     errs() << "Error: " << ErrMsg << "\n";
+    if (ExpectCrash)
+      return 0;
     return 1;
   }
 
+  if (ExpectCrash)
+    return 1;
+
   return Result == 0;
 }