teak-llvm/lldb/utils/lldb-repro/lldb-repro.py
Jonas Devlieghere be2bc6b1d0 [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.
2020-01-23 21:46:50 -08:00

68 lines
1.9 KiB
Python
Executable File

#!/usr/bin/env python
"""lldb-repro
lldb-repro is a utility to transparently capture and replay debugger sessions
through the command line driver. Its used to test the reproducers by running
the test suite twice.
During the first run, with 'capture' as its first argument, it captures a
reproducer for every lldb invocation and saves it to a well-know location
derived from the arguments and current working directory.
During the second run, with 'replay' as its first argument, the test suite is
run again but this time every invocation of lldb replays the previously
recorded session.
"""
import sys
import os
import tempfile
import subprocess
import hashlib
def help():
print("usage: {} capture|replay [args]".format(sys.argv[0]))
def main():
if len(sys.argv) < 2:
help()
return 1
# Compute an MD5 hash based on the input arguments and the current working
# directory.
h = hashlib.md5()
h.update(' '.join(sys.argv[2:]).encode('utf-8'))
h.update(os.getcwd().encode('utf-8'))
input_hash = h.hexdigest()
# Use the hash to "uniquely" identify a reproducer path.
reproducer_path = os.path.join(tempfile.gettempdir(), input_hash)
# 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,
'--reproducer-auto-generate'
])
new_args.extend(sys.argv[2:])
else:
help()
return 1
exit_code = subprocess.call(new_args)
if cleanup:
shutil.rmtree(reproducer_path, True)
return exit_code
if __name__ == '__main__':
exit(main())