[lldb/Util] Remove reproducers after replay

Except for debugging purposes there's no point in leaving the reproducer
behind on disk after replay. This patch adds a cleanup in the replay
case.

1 files changed, 6 insertions, 1 deletions

diff --git a/lldb/utils/lldb-repro/lldb-repro.py b/lldb/utils/lldb-repro/lldb-repro.py
index 313d3855a025..f7bcd3baa7a4 100755
--- a/lldb/utils/lldb-repro/lldb-repro.py
+++ b/lldb/utils/lldb-repro/lldb-repro.py
@@ -43,8 +43,10 @@ def main():
     # Create a new lldb invocation with capture or replay enabled.
     lldb = os.path.join(os.path.dirname(sys.argv[0]), 'lldb')
     new_args = [lldb]
+    cleanup = False
     if sys.argv[1] == "replay":
         new_args.extend(['--replay', reproducer_path])
+        cleanup = True
     elif sys.argv[1] == "capture":
         new_args.extend([
             '--capture', '--capture-path', reproducer_path,
@@ -55,7 +57,10 @@ def main():
         help()
         return 1
 
-    return subprocess.call(new_args)
+    exit_code = subprocess.call(new_args)
+    if cleanup:
+        shutil.rmtree(reproducer_path, True)
+    return exit_code
 
 
 if __name__ == '__main__':