def wrap(funcname, retcode, signature): argcount = len(signature) ourcode = MyPPCAssembler() ourcode.mflr(r0) ourcode.stmw(r31, r1, -4) ourcode.stw(r0, r1, 8) ourcode.stwu(r1, r1, -80) ourcode.lwz(r3, r4, 8) ourcode.cmpwi(r3, argcount) ourcode.bne("argserror") assert argcount < 9 if argcount > 0: load_arg(ourcode, 0, signature[0]) for i in range(2, argcount): load_arg(ourcode, i, signature[i]) if argcount > 1: load_arg(ourcode, 1, signature[1]) ourcode.load_word(r0, lookup(funcname)) ourcode.mtctr(r0) ourcode.bctrl() if retcode == 'i': s = lookup("PyInt_FromLong") ourcode.load_word(r0, s) ourcode.mtctr(r0) ourcode.bctrl() elif retcode == 'f': s = lookup("PyFloat_FromDouble") ourcode.load_word(r0, s) ourcode.mtctr(r0) ourcode.bctrl() ourcode.label("epilogue") ourcode.lwz(r0, r1, 88) ourcode.addi(r1, r1, 80) ourcode.mtlr(r0) ourcode.lmw(r31, r1, -4) ourcode.blr() err_set = lookup("PyErr_SetObject") exc = lookup("PyExc_TypeError") ourcode.label("argserror") ourcode.load_word(r5, err_set) ourcode.mtctr(r5) ourcode.load_from(r3, exc) ourcode.mr(r4, r3) ourcode.bctrl() ourcode.li(r3, 0) ourcode.b("epilogue") return ourcode.assemble()
def test_numberadd(self): a = MyPPCAssembler() a.lwz(r5, r4, pystructs.PyVarObject.ob_size) a.cmpwi(r5, 2) a.bne("err_out") a.lwz(r3, r4, 12) a.lwz(r4, r4, 16) a.load_word(r5, lookup("PyNumber_Add")) a.mtctr(r5) a.bctr() a.label("err_out") a.mflr(r0) a.stw(r0, r1, 8) a.stwu(r1, r1, -80) err_set = lookup("PyErr_SetObject") exc = lookup("PyExc_TypeError") a.load_word(r5, err_set) a.mtctr(r5) a.load_from(r3, exc) a.mr(r4, r3) a.bctrl() a.li(r3, 0) a.lwz(r0, r1, 88) a.addi(r1, r1, 80) a.mtlr(r0) a.blr() f = a.assemble() raises(TypeError, f) raises(TypeError, f, '', 1) raises(TypeError, f, 1) raises(TypeError, f, 1, 2, 3) assert f(1, 2) == 3 assert f('a', 'b') == 'ab'
def test_tuplelength2(self): a = MyPPCAssembler() a.mflr(0) a.stw(0, 1, 8) a.stwu(1, 1, -80) a.mr(3, 4) a.load_word(5, lookup("PyTuple_Size")) a.mtctr(5) a.bctrl() a.load_word(5, lookup("PyInt_FromLong")) a.mtctr(5) a.bctrl() a.lwz(0, 1, 88) a.addi(1, 1, 80) a.mtlr(0) a.blr() f = a.assemble() assert f() == 0 assert f(1) == 1 assert f('') == 1 assert f('', 3) == 2
def test_raise(self): a = MyPPCAssembler() a.mflr(0) a.stw(0, 1, 8) a.stwu(1, 1, -80) err_set = lookup("PyErr_SetObject") exc = lookup("PyExc_ValueError") a.load_word(5, err_set) a.mtctr(5) a.load_from(3, exc) a.mr(4, 3) a.bctrl() a.li(3, 0) a.lwz(0, 1, 88) a.addi(1, 1, 80) a.mtlr(0) a.blr() raises(ValueError, a.assemble())
def make_func(code, retcode, signature, localwords=0): """code shouldn't contain prologue/epilogue (or touch r31)""" stacksize = 80 + 4*localwords argcount = len(signature) ourcode = MyPPCAssembler() ourcode.mflr(r0) ourcode.stmw(r31, r1, -4) ourcode.stw(r0, r1, 8) ourcode.stwu(r1, r1, -stacksize) ourcode.lwz(r3, r4, 8) ourcode.cmpwi(r3, argcount) ourcode.bne("argserror") assert argcount < 9 if argcount > 0: load_arg(ourcode, 0, signature[0]) for i in range(2, argcount): load_arg(ourcode, i, signature[i]) if argcount > 1: load_arg(ourcode, 1, signature[1]) ourcode.bl(FAST_ENTRY_LABEL) if retcode == 'i': s = lookup("PyInt_FromLong") ourcode.load_word(r0, s) ourcode.mtctr(r0) ourcode.bctrl() elif retcode == 'f': s = lookup("PyFloat_FromDouble") ourcode.load_word(r0, s) ourcode.mtctr(r0) ourcode.bctrl() ourcode.label("epilogue") ourcode.lwz(r0, r1, stacksize + 8) ourcode.addi(r1, r1, stacksize) ourcode.mtlr(r0) ourcode.lmw(r31, r1, -4) ourcode.blr() err_set = lookup("PyErr_SetObject") exc = lookup("PyExc_TypeError") ourcode.label("argserror") ourcode.load_word(r5, err_set) ourcode.mtctr(r5) ourcode.load_from(r3, exc) ourcode.mr(r4, r3) ourcode.bctrl() ourcode.li(r3, 0) ourcode.b("epilogue") ourcode.label(FAST_ENTRY_LABEL) # err, should be an Assembler method: l = {} for k in code.labels: l[k] = code.labels[k] + 4*len(ourcode.insts) r = code.rlabels.copy() for k in code.rlabels: r[k + 4*len(ourcode.insts)] = code.rlabels[k] ourcode.insts.extend(code.insts) ourcode.labels.update(l) ourcode.rlabels.update(r) r = ourcode.assemble() r.FAST_ENTRY_LABEL = ourcode.labels[FAST_ENTRY_LABEL] return r