memorymodel.py

from ceptions import MemoryException

ARRAY_SIZE_MAGIC = 1024   # enough for most cases
ZERO_WORD = "0000000000000000000000000000000000000000000000000000000000000000"


class MemoryModel:
	def __init__(self, chunk):
		# print(chunk)
		self.__array = bytearray(chunk + ZERO_WORD * ARRAY_SIZE_MAGIC)

	def load_as_str(self, offset, size):
		return str(self.load(offset, size))

	def load_as_int(self, offset):
		return int(str(self.load(offset)), 16)

	def load(self, offset, size=32):
		begin, end = offset * 2, (offset + size) * 2
		if end > len(self.__array):
			# print(end, len(self.__array))
			raise MemoryException("out of bound access")
		return self.__array[begin:end]

	def store(self, offset, size, value):
		begin, end = offset * 2, (offset + size) * 2
		self.__array[begin:end] = value

	def debug_memory(self):
		arr = str(self.__array)
		arr = [arr[i:i + 64] for i in range(0, len(arr), 64)]
		for i in arr:
			print(i)