def main() -> int: parser = argparse.ArgumentParser() parser.add_argument('elf') parser.add_argument('-v', '--verbose', action='store_true') parser.add_argument('--dmem-dump') args = parser.parse_args() sim = OTBNSim() load_elf(sim, args.elf) sim.state.pc = 0 sim.state.start() sim.run(verbose=args.verbose) if args.dmem_dump is not None: try: with open(args.dmem_dump, 'wb') as dmem_file: dmem_file.write(sim.dump_data()) except OSError as err: sys.stderr.write('Failed to write DMEM to {!r}: {}.' .format(args.dmem_dump, err)) return 1 return 0
def on_load_elf(sim: OTBNSim, args: List[str]) -> None: '''Load contents of ELF at path given by only argument''' if len(args) != 1: raise ValueError( 'load_elf expects exactly 1 argument. Got {}.'.format(args)) path = args[0] print('LOAD_ELF {!r}'.format(path)) load_elf(sim, path)
def main() -> int: parser = argparse.ArgumentParser() parser.add_argument('elf') parser.add_argument('-v', '--verbose', action='store_true') parser.add_argument( '--dump-dmem', metavar="FILE", type=argparse.FileType('wb'), help=("after execution, write the data memory contents to this file. " "Use '-' to write to STDOUT.") ) parser.add_argument( '--dump-regs', metavar="FILE", type=argparse.FileType('w'), help=("after execution, write the GPR and WDR contents to this file. " "Use '-' to write to STDOUT.") ) parser.add_argument( '--dump-stats', metavar="FILE", type=argparse.FileType('w'), help=("after execution, write execution statistics to this file. " "Use '-' to write to STDOUT.") ) args = parser.parse_args() collect_stats = args.dump_stats is not None sim = OTBNSim() load_elf(sim, args.elf) sim.state.ext_regs.write('START_ADDR', 0, False) sim.state.ext_regs.commit() sim.start() sim.run(verbose=args.verbose, collect_stats=collect_stats) if args.dump_dmem is not None: args.dump_dmem.write(sim.dump_data()) if args.dump_regs is not None: for idx, value in enumerate(sim.state.gprs.peek_unsigned_values()): args.dump_regs.write(' x{:<2} = 0x{:08x}\n'.format(idx, value)) for idx, value in enumerate(sim.state.wdrs.peek_unsigned_values()): args.dump_regs.write(' w{:<2} = 0x{:064x}\n'.format(idx, value)) if collect_stats: assert sim.stats is not None stat_analyzer = ExecutionStatAnalyzer(sim.stats, args.elf) args.dump_stats.write(stat_analyzer.dump()) return 0
def _run_sim_for_stats(asm_file: str, tmpdir: str) -> ExecutionStats: """ Run the OTBN simulator, collect statistics, and return them. """ assert os.path.exists(asm_file) elf_file = testutil.asm_and_link_one_file(asm_file, tmpdir) sim = OTBNSim() load_elf(sim, elf_file) sim.state.start(0) sim.run(verbose=False, collect_stats=True) return sim.stats
def prepare_sim_for_asm_file(asm_file: str, tmpdir: py.path.local, start_addr: int = 0) -> OTBNSim: '''Set up the simulation of a single assembly file. The returned simulation is ready to be run through the run() method. ''' assert os.path.exists(asm_file) elf_file = asm_and_link_one_file(asm_file, tmpdir) sim = OTBNSim() load_elf(sim, elf_file) sim.state.ext_regs.write('START_ADDR', start_addr, False) sim.state.ext_regs.commit() sim.start() return sim
def main() -> int: parser = argparse.ArgumentParser() parser.add_argument('elf') parser.add_argument('-v', '--verbose', action='store_true') parser.add_argument( '--dump-dmem', metavar="FILE", type=argparse.FileType('wb'), help="after execution, write the data memory contents to this file") parser.add_argument( '--dump-regs', metavar="FILE", type=argparse.FileType('w'), default=sys.stdout, help= "after execution, write the GPR and WDR contents to this file (default: STDOUT)" ) args = parser.parse_args() sim = OTBNSim() load_elf(sim, args.elf) sim.state.pc = 0 sim.state.start() sim.run(verbose=args.verbose) if args.dump_dmem is not None: args.dump_dmem.write(sim.dump_data()) if args.dump_regs is not None: for idx, value in enumerate(sim.state.gprs.peek_unsigned_values()): args.dump_regs.write(' x{:<2} = 0x{:08x}\n'.format(idx, value)) for idx, value in enumerate(sim.state.wdrs.peek_unsigned_values()): args.dump_regs.write(' w{:<2} = 0x{:064x}\n'.format(idx, value)) return 0