def Oprm(self, i):
#print i
code = [
x.movl(x.mem(x.esp, i.b.type.offset), x.eax),
x.movl(x.eax, x.mem(x.esp, -(1+i.a)*4)),
]
return code
def Iprm(self, i):
"""Generator code for IPRM : Put an Input Param on the Stack"""
if isinstance(i.b.type, il.Func):
# this is a function param not a value
code = [x.leal(i.b.type.entry, x.eax), x.movl(x.eax, x.mem(x.esp, -(2 + i.a) * 4))]
else:
code = [x.movl(x.loc(i.b.type), x.eax), x.movl(x.eax, x.mem(x.esp, -(2 + i.a) * 4))]
return code
def Iprm(self, i):
if isinstance(i.b.type, il.Func):
# this is a function param not a value
code = [
x.leal(i.b.type.entry, x.eax),
x.movl(x.eax, x.mem(x.esp, -(2+i.a)*4)),
]
else:
code = [
x.movl(x.loc(i.b.type), x.eax),
x.movl(x.eax, x.mem(x.esp, -(2+i.a)*4)),
]
return code
def FramePop(self, param_count, scope_depth):
"""Generate a framepop.
@param param_count : the number of params the function recieves.
@param scope_depth : the scope the depth the function was declared at.
"""
p = param_count
code = [
"",
# x.movl(x.cint(4*(scope_depth-1)), x.ebx),
# x.pushl(x.static('display', x.ebx)),
# x.pushl('$pop1_msg'),
# x.call("printf"),
# x.addl(x.cint(8), x.esp),
x.movl(x.cint(4 * (scope_depth - 1)), x.ebx),
x.movl(x.mem(x.ebp, -(7 + p) * 4), x.eax),
x.movl(x.eax, x.static("display", x.ebx)),
# x.movl(x.cint(4*(scope_depth-1)), x.ebx),
# x.pushl(x.static('display', x.ebx)),
# x.pushl('$pop2_msg'),
# x.call("printf"),
# x.addl(x.cint(8), x.esp),
x.movl(x.mem(x.ebp, -(6 + p) * 4), x.edx),
x.movl(x.mem(x.ebp, -(5 + p) * 4), x.ecx),
x.movl(x.mem(x.ebp, -(4 + p) * 4), x.esi),
x.movl(x.mem(x.ebp, -(3 + p) * 4), x.edi),
x.movl(x.mem(x.ebp, -(2 + p) * 4), x.ebx),
x.movl(x.ebp, x.esp), # restore stack pointer
x.movl(x.mem(x.esp, -(1 + p) * 4), x.ebp), # restore base(frame) pointer
]
return code
def FramePush(self, param_count, scope_depth):
"""Generate a framepush.
@param param_count : the number of params the function recieves.
@param scope_depth : the scope the depth the function was declared at.
"""
p = param_count
code = [
# 0(ebp) == return address
x.movl(x.ebp, x.mem(x.esp, -(1 + p) * 4)), # 1
x.movl(x.esp, x.ebp),
x.movl(x.ebx, x.mem(x.ebp, -(2 + p) * 4)), # 2
x.movl(x.edi, x.mem(x.ebp, -(3 + p) * 4)), # 3
x.movl(x.esi, x.mem(x.ebp, -(4 + p) * 4)), # 4
x.movl(x.ecx, x.mem(x.ebp, -(5 + p) * 4)), # 5
x.movl(x.edx, x.mem(x.ebp, -(6 + p) * 4)), # 6
x.movl(x.cint(4 * (scope_depth - 1)), x.ebx),
x.movl(x.static("display", x.ebx), x.eax),
x.movl(x.eax, x.mem(x.ebp, -(7 + p) * 4)), # 7
# x.subl(x.cint(8*4), x.esp),
# x.movl(x.cint(4*(scope_depth-1)), x.ebx),
# x.pushl(x.static('display', x.ebx)),
# x.pushl('$push_msg'),
# x.call("printf"),
# x.addl(x.cint(8*4 + 8), x.esp),
x.movl(x.cint(4 * (scope_depth - 1)), x.ebx),
x.movl(x.ebp, x.static("display", x.ebx)),
"",
]
return code
def FramePop(self, param_count, scope_depth):
p = param_count
code = [
'',
#x.movl(x.cint(4*(scope_depth-1)), x.ebx),
#x.pushl(x.static('display', x.ebx)),
#x.pushl('$pop1_msg'),
#x.call("printf"),
#x.addl(x.cint(8), x.esp),
x.movl(x.cint(4*(scope_depth-1)), x.ebx),
x.movl(x.mem(x.ebp, -(7+p)*4), x.eax),
x.movl(x.eax, x.static('display', x.ebx)),
#x.movl(x.cint(4*(scope_depth-1)), x.ebx),
#x.pushl(x.static('display', x.ebx)),
#x.pushl('$pop2_msg'),
#x.call("printf"),
#x.addl(x.cint(8), x.esp),
x.movl(x.mem(x.ebp, -(6+p)*4), x.edx),
x.movl(x.mem(x.ebp, -(5+p)*4), x.ecx),
x.movl(x.mem(x.ebp, -(4+p)*4), x.esi),
x.movl(x.mem(x.ebp, -(3+p)*4), x.edi),
x.movl(x.mem(x.ebp, -(2+p)*4), x.ebx),
x.movl(x.ebp, x.esp), # restore stack pointer
x.movl(x.mem(x.esp, -(1+p)*4), x.ebp), # restore base(frame) pointer
]
return code
def FramePush(self, param_count, scope_depth):
p = param_count
code = [
# 0(ebp) == return address
x.movl(x.ebp, x.mem(x.esp, -(1+p)*4)), # 1
x.movl(x.esp, x.ebp),
x.movl(x.ebx, x.mem(x.ebp, -(2+p)*4)), # 2
x.movl(x.edi, x.mem(x.ebp, -(3+p)*4)), # 3
x.movl(x.esi, x.mem(x.ebp, -(4+p)*4)), # 4
x.movl(x.ecx, x.mem(x.ebp, -(5+p)*4)), # 5
x.movl(x.edx, x.mem(x.ebp, -(6+p)*4)), # 6
x.movl(x.cint(4*(scope_depth-1)), x.ebx),
x.movl(x.static('display', x.ebx), x.eax),
x.movl(x.eax, x.mem(x.ebp, -(7+p)*4)), #7
#x.subl(x.cint(8*4), x.esp),
#x.movl(x.cint(4*(scope_depth-1)), x.ebx),
#x.pushl(x.static('display', x.ebx)),
#x.pushl('$push_msg'),
#x.call("printf"),
#x.addl(x.cint(8*4 + 8), x.esp),
x.movl(x.cint(4*(scope_depth-1)), x.ebx),
x.movl(x.ebp, x.static('display', x.ebx)),
'',
]
return code
def Rprm(self, i):
#print i
code = [
x.movl(x.mem(x.esp, -(2+i.a)*4), x.eax),
x.movl(x.eax, x.loc(i.result.type)),
]
#self.var[i.result] = self.bp_offset
#self.bp_offset += 1
return code
def emit(self, inst, src, targ=None):
'''emit the specified instruction with src as source operand and targ
as the target operand. Handles requesting the operands from a non-local
stack frame as necessary.
using ebx or ecx is not allowed for either the src or targ.
'''
#print inst, src, targ
if isinstance(src, il.Symbol) and isinstance(targ, il.Symbol):
raise Exception, 'Cannot emit an instruction with both source and target as symbols'
elif isinstance(src, il.Symbol) and targ is None:
if src.islocal(self.cfunc):
return [ inst(x.loc(src.type)) ]
return [
x.movl(x.cint(4*(src.scope_depth-1)), x.ebx),
x.movl(x.static('display', x.ebx), x.ebx),
inst(x.mem(x.ebx, src.type.offset)),
]
elif not isinstance(src, il.Symbol) and targ is None:
return [ inst(src), ]
elif isinstance(src, il.Symbol) and targ is not None:
if src.islocal(self.cfunc):
return [ inst(x.loc(src.type), targ) ]
return [
x.movl(x.cint(4*(src.scope_depth-1)), x.ebx),
x.movl(x.static('display', x.ebx), x.ebx),
inst(x.mem(x.ebx, src.type.offset), targ),
]
elif isinstance(targ, il.Symbol):
#print inst, src, targ, targ.type.basereg
if targ.islocal(self.cfunc):
return [ inst(src, x.loc(targ.type)) ]
return [
x.movl(x.cint(4*(targ.scope_depth-1)), x.ebx),
x.movl(x.static('display', x.ebx), x.ebx),
x.movl(x.mem(x.ebx, targ.type.offset), x.ecx),
inst(src, x.ecx),
x.movl(x.ecx, x.mem(x.ebx, targ.type.offset)),
]
else:
return [ inst(src, targ) ]
def Return(self, i):
code = list()
#if len(self.code) >= 90 and len(self.code) < 110:
#code = [
#(vm.IMM, 3, len(self.code), 'DEBUG'),
#(vm.PRNT, 3, 0, 'DEBUG'),
#(vm.PRNT, 2, 0, 'DEBUG'),
#(vm.EXIT, 0, 0, 'DEBUG'),
#]
code += [
x.addl(x.cint(4), x.esp),
x.jmp(x.mem(x.esp, -4, True)),
#x.ret(),
]
return code
def emit(self, inst, src, targ=None):
"""emit the specified instruction with src as source operand and targ
as the target operand. Handles requesting the operands from a
non-local stack frame as necessary.
using ebx or ecx is not allowed for either the src or targ.
"""
## We have several choices to make in order to emit the right inst.
## (1) does the inst only refer to symbols local to the current scope?
## (a) if yes: emit the instruction with no changes
## (b) if no: detirmine what changes have to be made.
## (2) We now know the instruction is not local. One (but not both) of
## the symbols does not come from this stack frame. That means they
## have to be fetched from the appropriate stack frame. We detirmine
## the appropriate frame to fetch (or put) the sym from(to) by
## consulting the display.
## (3) The display is a static array:
##
## +------------+------------+------------+------------+
## scope depth | 0 | 1 | 2 | 3 |
## +------------+------------+------------+------------+
## frame address | 0x???????? | 0x???????? | 0x???????? | 0x???????? |
## +------------+------------+------------+------------+
##
## The 'frame address' contains the address of the the stack frame
## most recently seen function. When that function returns it will
## update the display with the previous address it contained. When
## A new function is called at that scope depth a new address will
## be placed in the display and the old will be saved.
##
## (4) Therefore to get the non-local symbols we simply need to consult
## the display to find the appropriate frame. Each symbol knows what
## scope-depth it was declared at.
##
## NB: The symbols store there scope depths starting at 1. Therefore, in
## order to index into the display one must subtract 1 from the
## stored scope depth.
##
## NB: We multiply by 4 because each address is 4 bytes wide.
if isinstance(src, il.Symbol) and isinstance(targ, il.Symbol):
raise Exception, ("Cannot emit an instruction with both source and target as " "symbols")
elif isinstance(src, il.Symbol) and targ is None:
if src.islocal(self.cfunc):
return [inst(x.loc(src.type))]
return [
x.movl(x.cint(4 * (src.scope_depth - 1)), x.ebx), # compute the
# index into the
# display and
# store in ebx
x.movl(x.static("display", x.ebx), x.ebx), # get the non-
# local stack
# pointer and
# store in ebx
inst(x.mem(x.ebx, src.type.offset)), # emit the inst.
]
elif not isinstance(src, il.Symbol) and targ is None:
return [inst(src)]
elif isinstance(src, il.Symbol) and targ is not None:
if src.islocal(self.cfunc):
return [inst(x.loc(src.type), targ)]
return [
x.movl(x.cint(4 * (src.scope_depth - 1)), x.ebx),
x.movl(x.static("display", x.ebx), x.ebx),
inst(x.mem(x.ebx, src.type.offset), targ),
]
elif isinstance(targ, il.Symbol):
# print inst, src, targ, targ.type.basereg
if targ.islocal(self.cfunc):
return [inst(src, x.loc(targ.type))]
return [
## This version is slightly tricky we have to fetch and store
## the contents of the variable stored in the non-local stack
## frame.
x.movl(x.cint(4 * (targ.scope_depth - 1)), x.ebx),
x.movl(x.static("display", x.ebx), x.ebx),
x.movl(x.mem(x.ebx, targ.type.offset), x.ecx),
inst(src, x.ecx),
x.movl(x.ecx, x.mem(x.ebx, targ.type.offset)),
]
else:
return [inst(src, targ)]
def Return(self, i):
"""Generate code to return from the current function"""
return [x.addl(x.cint(4), x.esp), x.jmp(x.mem(x.esp, -4, True))]
def Rprm(self, i):
"""Generate code for RPRM : Get a Return Param"""
code = [x.movl(x.mem(x.esp, -(2 + i.a) * 4), x.eax), x.movl(x.eax, x.loc(i.result.type))]
return code
def Oprm(self, i):
"""Generator code for OPRM : Output a param"""
# print i
code = [x.movl(x.mem(x.esp, i.b.type.offset), x.eax), x.movl(x.eax, x.mem(x.esp, -(1 + i.a) * 4))]
return code
def Gprm(self, i):
"""Generator code for GPRM : Get Input Param i"""
code = [x.movl(x.mem(x.ebp, -(1 + i.a) * 4), x.eax), x.movl(x.eax, x.loc(i.result.type))]
return code
def Gprm(self, i):
code = [
x.movl(x.mem(x.ebp, -(1+i.a)*4), x.eax),
x.movl(x.eax, x.loc(i.result.type)),
]
return code