98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
from py65 import memory
|
|
from collections import defaultdict
|
|
|
|
class WriteProtectedException(Exception):
|
|
def __init__(self, name, addr, value):
|
|
self.name = name
|
|
self.addr = addr
|
|
self.value = value
|
|
|
|
def __str__(self):
|
|
return 'Write denied to %s ($%04X): $%02X' % (self.name, self.addr, self.value)
|
|
|
|
class UndefinedEntryPointException(Exception):
|
|
def __init__(self, region, prev_addr, addr):
|
|
self.region = region
|
|
self.prev_addr = prev_addr
|
|
self.addr = addr
|
|
|
|
def __str__(self):
|
|
return 'Entered region %s via undefined entry point: $%04X --> $%04X' % (
|
|
self.region.name, self.prev_addr, self.addr)
|
|
|
|
|
|
class MemoryManager(object):
|
|
def __init__(self, memory_map):
|
|
self.entrypoints = defaultdict(list)
|
|
|
|
default_region = MemoryRegion('default', 0x0, 0xffff)
|
|
def _default_region():
|
|
return default_region
|
|
|
|
self.regions = defaultdict(_default_region)
|
|
|
|
self.memory = memory.ObservableMemory()
|
|
|
|
for region in memory_map:
|
|
self.RegisterRegion(region)
|
|
|
|
def MaybeInterceptExecution(self, cpu, old_pc):
|
|
pc = cpu.pc
|
|
if pc in self.entrypoints:
|
|
handlers = self.entrypoints[pc]
|
|
else:
|
|
handlers = []
|
|
|
|
if self.regions[old_pc] != self.regions[pc]:
|
|
print "Entering region %s" % self.regions[pc].name
|
|
if not handlers:
|
|
raise UndefinedEntryPointException(self.regions[pc], old_pc, pc)
|
|
|
|
for handler in handlers:
|
|
handler(cpu)
|
|
return
|
|
|
|
def RegisterRegion(self, region):
|
|
addr_range = xrange(region.start, region.end + 1)
|
|
|
|
if region.read_interceptor:
|
|
self.memory.subscribe_to_read(addr_range, region.read_interceptor)
|
|
|
|
if region.write_interceptor:
|
|
self.memory.subscribe_to_write(addr_range, region.write_interceptor)
|
|
|
|
if not region.writable:
|
|
self.memory.subscribe_to_write(addr_range, self.DenyWritesToRegion(region))
|
|
|
|
for addr in addr_range:
|
|
self.regions[addr] = region
|
|
|
|
for addr, handler in region.entrypoints.iteritems():
|
|
self.entrypoints[addr].append(handler)
|
|
|
|
# TODO: should trap by default
|
|
|
|
|
|
@staticmethod
|
|
def DenyWritesToRegion(region):
|
|
def _DenyWritesInterceptor(addr, value):
|
|
raise WriteProtectedException(region.name, addr, value)
|
|
return _DenyWritesInterceptor
|
|
|
|
|
|
class MemoryRegion(object):
|
|
def __init__(
|
|
self, name, start, end, read_interceptor=None, write_interceptor=None, entrypoints=None,
|
|
writable=True):
|
|
self.name = name
|
|
self.start = start
|
|
self.end = end
|
|
|
|
self.read_interceptor = read_interceptor
|
|
self.write_interceptor = write_interceptor
|
|
|
|
self.writable = writable
|
|
|
|
# Maps PC to handler
|
|
self.entrypoints = entrypoints or {}
|