diff options
Diffstat (limited to 'utils/not/not.cpp')
-rw-r--r-- | utils/not/not.cpp | 24 |
1 files changed, 21 insertions, 3 deletions
diff --git a/utils/not/not.cpp b/utils/not/not.cpp index 9a924b5..ebd1618 100644 --- a/utils/not/not.cpp +++ b/utils/not/not.cpp @@ -13,15 +13,33 @@ using namespace llvm; int main(int argc, const char **argv) { - sys::Path Program = sys::Program::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::Program::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; } |