parser_c/cpu.py

def get_value(b:bytearray):
  ret=0
  for i,item in enumerate(b):
    ret|=item<<(8*i)
  return ret

def set_value(b:bytearray,value:int):
  for i in range(len(b)):
    b[i]=(value>>(8*i))&0xff


class cpu(object):
  def __init__(self,rom_addr:int,rom_size:int,mem_addr:int,mem_size:int) -> None:
    self.rom_addr=rom_addr
    self.rom=bytearray(rom_size)
    self.mem_addr=mem_addr
    self.mem=bytearray(mem_size)
    self.pc=self.mem_addr
    self.sp=self.mem_addr
    self.rp=self.pc
    self.fp=self.sp
  def get_value(self,addr:int,size:int):
    if not (size in [1,2,4,8]):
      raise Exception(f"CPU 异常 因为地址未对齐 {size}")
    if(addr>=self.rom_addr) and (addr<self.rom_addr+len(self.rom)):
      return get_value(self.rom[addr-self.rom_addr,addr-self.rom_addr+size])
  def set_value(self,addr:int,size:int,value:int):
    if not (size in [1,2,4,8]):
      raise Exception(f"CPU 异常 因为地址未对齐 {size}")
    if(addr>=self.rom_addr) and (addr<self.rom_addr+len(self.rom)):
      set_value(self.rom[addr-self.rom_addr,addr-self.rom_addr+size],value)
  def opt_add(self,src_addr1:int,src_addr2:int,out_addr:int):
    src1=self.get_value(src_addr1,4)
    src2=self.get_value(src_addr2,4)
    self.set_value(out_addr,4,src1+src2)
  def opt_sub(self,src_addr1:int,src_addr2:int,out_addr:int):
    src1=self.get_value(src_addr1,4)
    src2=self.get_value(src_addr2,4)
    self.set_value(out_addr,4,src1-src2)
  def opt_bitor(self,src_addr1:int,src_addr2:int,out_addr:int):
    src1=self.get_value(src_addr1,4)
    src2=self.get_value(src_addr2,4)
    self.set_value(out_addr,4,src1|src2)
  def opt_bitand(self,src_addr1:int,src_addr2:int,out_addr:int):
    src1=self.get_value(src_addr1,4)
    src2=self.get_value(src_addr2,4)
    self.set_value(out_addr,4,src1&src2)
  def opt_bitxor(self,src_addr1:int,src_addr2:int,out_addr:int):
    src1=self.get_value(src_addr1,4)
    src2=self.get_value(src_addr2,4)
    self.set_value(out_addr,4,src1^src2)
  def opt_jmp(self,jmp_addr:int):
    self.pc=jmp_addr
  # 要进入子函数 先enter 再设置形式参数
  def opt_enter(self,enter_addr:int):
    self.set_value(self.sp,4,self.rp)
    self.rp=self.pc
    self.sp+=4
    self.set_value(self.sp,4,self.fp)
    self.sp+=4
    self.fp=self.sp
    self.pc=enter_addr
  # 返回的时候先回溯sp指针 然后把pc指针指定到之前的pc指针+4的位置
  def opt_return(self,return_value:int):
    self.sp=self.fp
    self.sp-=4
    self.fp=self.get_value(self.sp,4)
    self.pc=self.rp+4
    self.sp-=4
    self.rp=self.get_value(self.sp,4)
    self.set_value(self.sp,4,return_value)
  # 在栈上创建变量的存储空间
  def create_var(self,size:int):
    ret=self.sp
    self.sp+=((size+3)//4)*4
    return ret