[libFuzzer] don't crash when reporting a leak in test_single_input mode

[oota-llvm.git] / lib / Fuzzer / FuzzerDriver.cpp

diff --git a/lib/Fuzzer/FuzzerDriver.cpp b/lib/Fuzzer/FuzzerDriver.cpp
index c4cfc0651562a2aa5cf1852cf0fcc7b77f80de6e..0897f9cbde72a425b558a155cef91143a96d0527 100644 (file)
--- a/lib/Fuzzer/FuzzerDriver.cpp
+++ b/lib/Fuzzer/FuzzerDriver.cpp
@@ -273,8 +273,10 @@ int FuzzerDriver(const std::vector<std::string> &Args,
   if (Flags.timeout > 0)
     SetTimer(Flags.timeout / 2 + 1);
 
-  if (Flags.test_single_input)
-    return RunOneTest(&F, Flags.test_single_input);
+  if (Flags.test_single_input) {
+    RunOneTest(&F, Flags.test_single_input);
+    exit(0);
+  }
 
   if (Flags.merge) {
     F.Merge(*Inputs);