diff options
Diffstat (limited to 'examples/python/disasm-stress-test.py')
-rwxr-xr-x | examples/python/disasm-stress-test.py | 120 |
1 files changed, 91 insertions, 29 deletions
diff --git a/examples/python/disasm-stress-test.py b/examples/python/disasm-stress-test.py index 5aa354dc24cb..2c20ffb92369 100755 --- a/examples/python/disasm-stress-test.py +++ b/examples/python/disasm-stress-test.py @@ -1,40 +1,82 @@ #!/usr/bin/python -import argparse, datetime, re, subprocess, sys, time - -parser = argparse.ArgumentParser(description="Run an exhaustive test of the LLDB disassembler for a specific architecture.") - -parser.add_argument('--arch', required=True, action='store', help='The architecture whose disassembler is to be tested') -parser.add_argument('--bytes', required=True, action='store', type=int, help='The byte width of instructions for that architecture') -parser.add_argument('--random', required=False, action='store_true', help='Enables non-sequential testing') -parser.add_argument('--start', required=False, action='store', type=int, help='The first instruction value to test') -parser.add_argument('--skip', required=False, action='store', type=int, help='The interval between instructions to test') -parser.add_argument('--log', required=False, action='store', help='A log file to write the most recent instruction being tested') -parser.add_argument('--time', required=False, action='store_true', help='Every 100,000 instructions, print an ETA to standard out') -parser.add_argument('--lldb', required=False, action='store', help='The path to LLDB.framework, if LLDB should be overridden') +import argparse +import datetime +import re +import subprocess +import sys +import time + +parser = argparse.ArgumentParser( + description="Run an exhaustive test of the LLDB disassembler for a specific architecture.") + +parser.add_argument( + '--arch', + required=True, + action='store', + help='The architecture whose disassembler is to be tested') +parser.add_argument( + '--bytes', + required=True, + action='store', + type=int, + help='The byte width of instructions for that architecture') +parser.add_argument( + '--random', + required=False, + action='store_true', + help='Enables non-sequential testing') +parser.add_argument( + '--start', + required=False, + action='store', + type=int, + help='The first instruction value to test') +parser.add_argument( + '--skip', + required=False, + action='store', + type=int, + help='The interval between instructions to test') +parser.add_argument( + '--log', + required=False, + action='store', + help='A log file to write the most recent instruction being tested') +parser.add_argument( + '--time', + required=False, + action='store_true', + help='Every 100,000 instructions, print an ETA to standard out') +parser.add_argument( + '--lldb', + required=False, + action='store', + help='The path to LLDB.framework, if LLDB should be overridden') arguments = sys.argv[1:] arg_ns = parser.parse_args(arguments) + def AddLLDBToSysPathOnMacOSX(): def GetLLDBFrameworkPath(): lldb_path = subprocess.check_output(["xcrun", "-find", "lldb"]) re_result = re.match("(.*)/Developer/usr/bin/lldb", lldb_path) - if re_result == None: + if re_result is None: return None xcode_contents_path = re_result.group(1) return xcode_contents_path + "/SharedFrameworks/LLDB.framework" - + lldb_framework_path = GetLLDBFrameworkPath() - - if lldb_framework_path == None: + + if lldb_framework_path is None: print "Couldn't find LLDB.framework" sys.exit(-1) - + sys.path.append(lldb_framework_path + "/Resources/Python") -if arg_ns.lldb == None: +if arg_ns.lldb is None: AddLLDBToSysPathOnMacOSX() else: sys.path.append(arg_ns.lldb + "/Resources/Python") @@ -53,16 +95,20 @@ if target.IsValid() == False: print "Couldn't create an SBTarget for architecture " + arg_ns.arch sys.exit(-1) + def ResetLogFile(log_file): if log_file != sys.stdout: log_file.seek(0) + def PrintByteArray(log_file, byte_array): for byte in byte_array: print >>log_file, hex(byte) + " ", print >>log_file - + + class SequentialInstructionProvider: + def __init__(self, byte_width, log_file, start=0, skip=1): self.m_byte_width = byte_width self.m_log_file = log_file @@ -70,70 +116,83 @@ class SequentialInstructionProvider: self.m_skip = skip self.m_value = start self.m_last = (1 << (byte_width * 8)) - 1 + def PrintCurrentState(self, ret): ResetLogFile(self.m_log_file) print >>self.m_log_file, self.m_value PrintByteArray(self.m_log_file, ret) + def GetNextInstruction(self): if self.m_value > self.m_last: return None ret = bytearray(self.m_byte_width) for i in range(self.m_byte_width): - ret[self.m_byte_width - (i + 1)] = (self.m_value >> (i * 8)) & 255 + ret[self.m_byte_width - (i + 1)] = (self.m_value >> (i * 8)) & 255 self.PrintCurrentState(ret) self.m_value += self.m_skip return ret + def GetNumInstructions(self): return (self.m_last - self.m_start) / self.m_skip + def __iter__(self): return self + def next(self): ret = self.GetNextInstruction() - if ret == None: + if ret is None: raise StopIteration return ret + class RandomInstructionProvider: + def __init__(self, byte_width, log_file): self.m_byte_width = byte_width self.m_log_file = log_file self.m_random_file = open("/dev/random", 'r') + def PrintCurrentState(self, ret): ResetLogFile(self.m_log_file) PrintByteArray(self.m_log_file, ret) + def GetNextInstruction(self): ret = bytearray(self.m_byte_width) for i in range(self.m_byte_width): ret[i] = self.m_random_file.read(1) self.PrintCurrentState(ret) return ret + def __iter__(self): return self + def next(self): ret = self.GetNextInstruction() - if ret == None: + if ret is None: raise StopIteration return ret log_file = None + def GetProviderWithArguments(args): global log_file - if args.log != None: + if args.log is not None: log_file = open(args.log, 'w') else: log_file = sys.stdout instruction_provider = None - if args.random == True: + if args.random: instruction_provider = RandomInstructionProvider(args.bytes, log_file) else: start = 0 skip = 1 - if args.start != None: + if args.start is not None: start = args.start - if args.skip != None: + if args.skip is not None: skip = args.skip - instruction_provider = SequentialInstructionProvider(args.bytes, log_file, start, skip) + instruction_provider = SequentialInstructionProvider( + args.bytes, log_file, start, skip) return instruction_provider instruction_provider = GetProviderWithArguments(arg_ns) @@ -149,10 +208,13 @@ if actually_time: for inst_bytes in instruction_provider: if actually_time: - if (num_instructions_logged != 0) and (num_instructions_logged % 100000 == 0): + if (num_instructions_logged != 0) and ( + num_instructions_logged % 100000 == 0): curr_time = time.time() elapsed_time = curr_time - start_time - remaining_time = float(total_num_instructions - num_instructions_logged) * (float(elapsed_time) / float(num_instructions_logged)) + remaining_time = float( + total_num_instructions - num_instructions_logged) * ( + float(elapsed_time) / float(num_instructions_logged)) print str(datetime.timedelta(seconds=remaining_time)) num_instructions_logged = num_instructions_logged + 1 inst_list = target.GetInstructions(fake_address, inst_bytes) |