def setupResetRegion(self):
self.setGlobalState("ResetPC", PcConfig.get_reset_pc())
self.virtualMemoryRequest(
"PhysicalRegion", {
"RegionType": "ResetRegion",
"Size": PcConfig.get_reset_region_size(),
"Type": 'I',
"Bank": 0
})
def _getHandlerMemoryConstraint(self):
# Need to ensure the handler memory doesn't intersect the boot region
# or initial PC of any thread
handler_memory_constr = ConstraintSet(0, 0xFFFFFFFFFFFFFFFF)
for thread_id in range(self.getThreadNumber()):
handler_memory_constr.subRange(
PcConfig.get_boot_pc(thread_id),
(PcConfig.get_boot_pc(thread_id) +
PcConfig.get_boot_region_size() - 1),
)
handler_memory_constr.subValue(PcConfig.get_initial_pc(thread_id))
return handler_memory_constr
def generate(self, **kwargs):
with ThreadSplitterContextManager(self):
# TODO add MP support
pc = PcConfig.get_base_boot_pc()
(skip_boot, skip_boot_valid) = self.getOption("SkipBootCode")
if skip_boot_valid and skip_boot == 1:
pc = PcConfig.get_base_initial_pc()
self.genInstruction('JAL##RISCV', {
'rd': 0,
'NoBnt': 1,
'BRTarget': pc
}) # Brach to calculated address
def setupResetRegion(self):
self.setGlobalState("ResetPC", PcConfig.get_reset_pc())
self.virtualMemoryRequest(
"PhysicalRegion",
{
"RegionType": "ResetRegion",
"Size": PcConfig.get_reset_region_size(),
"Type": "I",
"Bank": 0,
},
)
(skip_boot, skip_boot_valid) = self.genThread.getOption("SkipBootCode")
if skip_boot_valid and skip_boot == 1:
self.setGlobalState("ResetPC", PcConfig.get_base_initial_pc())
def setupBootRegion(self):
self.virtualMemoryRequest(
"PhysicalRegion", {
"RegionType": "BootRegion",
"Size": PcConfig.get_boot_region_size(),
"Type": 'I',
"Bank": 0
})
def generate(self, **kwargs):
with ThreadSplitterContextManager(self):
pc = PcConfig.get_base_boot_pc()
(skip_boot, skip_boot_valid) = self.getOption("SkipBootCode")
if skip_boot_valid and skip_boot == 1:
pc = PcConfig.get_base_initial_pc()
(
boot_pc_reg_index,
thread_id_reg_index,
pc_offset_reg_index,
) = self.getRandomGPRs(3, exclude="0")
assembly_helper = AssemblyHelperRISCV(self)
assembly_helper.genReadSystemRegister(
thread_id_reg_index, "mhartid"
) # Get the thread ID
load_gpr64_seq = LoadGPR64(self.genThread)
load_gpr64_seq.load(
pc_offset_reg_index, PcConfig.get_boot_pc_offset()
)
self.genInstruction(
"MUL##RISCV",
{
"rd": pc_offset_reg_index,
"rs1": thread_id_reg_index,
"rs2": pc_offset_reg_index,
},
) # Multiply the base PC offset by the thread ID
load_gpr64_seq.load(boot_pc_reg_index, PcConfig.get_base_boot_pc())
assembly_helper.genAddRegister(
boot_pc_reg_index, pc_offset_reg_index
) # Add the thread PC offset to the base initial PC
self.genInstruction(
"JALR##RISCV",
{
"rd": 0,
"rs1": boot_pc_reg_index,
"simm12": 0,
"NoBnt": 1,
"NoRestriction": 1,
},
) # Branch to calculated address
def generate(self, **kwargs):
with ThreadSplitterContextManager(self):
pc = PcConfig.get_base_boot_pc()
(skip_boot, skip_boot_valid) = self.getOption("SkipBootCode") #TODO allow for granular control of skip boot code/skip thread splitter code
if skip_boot_valid and skip_boot == 1:
pc = PcConfig.get_base_initial_pc()
(boot_pc_reg_index, thread_id_reg_index, pc_offset_reg_index) = self.getRandomGPRs(3, exclude='0')
assembly_helper = AssemblyHelperRISCV(self)
assembly_helper.genReadSystemRegister(thread_id_reg_index, 'mhartid') # Get the thread ID
load_gpr64_seq = LoadGPR64(self.genThread)
load_gpr64_seq.load(pc_offset_reg_index, PcConfig.get_boot_pc_offset())
self.genInstruction('MUL##RISCV', {'rd': pc_offset_reg_index, 'rs1': thread_id_reg_index, 'rs2': pc_offset_reg_index}) # Multiply the base PC offset by the thread ID
load_gpr64_seq.load(boot_pc_reg_index, PcConfig.get_base_boot_pc())
assembly_helper.genAddRegister(boot_pc_reg_index, pc_offset_reg_index) # Add the thread PC offset to the base initial PC
self.genInstruction('JALR##RISCV', {'rd': 0, 'rs1': boot_pc_reg_index, 'simm12': 0, 'NoBnt': 1, 'NoRestriction': 1}) # Branch to calculated address
def setupBootRegion(self):
"""Setup memory region for boot instructions"""
self.virtualMemoryRequest(
"PhysicalRegion",
{
"RegionType": "BootRegion",
"Size": PcConfig.get_boot_region_size(),
"Type": "I",
"Bank": 0,
},
)
def __enter__(self):
self.sequence.genThread.modifyGenMode('NoEscape,SimOff')
self.origPc = self.sequence.getPEstate('PC')
self.sequence.setPEstate('PC', PcConfig.get_reset_pc())
return self
def getBootPC(self):
return PcConfig.get_boot_pc(self.genThread.genThreadID)
def getInitialPC(self):
"""Return initial main test PC for the generator thread"""
return PcConfig.get_initial_pc(self.genThread.genThreadID)
def getInitialPC(self):
return PcConfig.get_initial_pc(self.genThread.genThreadID)