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