Пример #1
0
 def test_windll(self):
     if os.name != 'nt':
         py.test.skip('Run only on windows')
     from rpython.rlib.libffi import WinDLL
     dll = WinDLL('Kernel32.dll')
     sleep = dll.getpointer('Sleep', [types.uint], types.void)
     chain = ArgChain()
     chain.arg(10)
     sleep.call(chain, lltype.Void, is_struct=False)
Пример #2
0
 def test_windll(self):
     if os.name != 'nt':
         py.test.skip('Run only on windows')
     from rpython.rlib.libffi import WinDLL
     dll = WinDLL('Kernel32.dll')
     sleep = dll.getpointer('Sleep',[types.uint], types.void)
     chain = ArgChain()
     chain.arg(10)
     sleep.call(chain, lltype.Void, is_struct=False)
Пример #3
0
 def libffi_stuff(i, j):
     lib = CDLL(libm_name)
     func = lib.getpointer("fabs", [types.double], types.double)
     res = 0.0
     x = float(j)
     while i > 0:
         jitdriver2.jit_merge_point(i=i, res=res, func=func, x=x)
         promote(func)
         argchain = ArgChain()
         argchain.arg(x)
         res = func.call(argchain, rffi.DOUBLE)
         i -= 1
     return res
Пример #4
0
 def libffi_stuff(i, j):
     lib = CDLL(libm_name)
     func = lib.getpointer('fabs', [types.double], types.double)
     res = 0.0
     x = float(j)
     v2 = Virt2(i)
     while v2.i > 0:
         jitdriver2.jit_merge_point(v2=v2, res=res, func=func, x=x)
         promote(func)
         argchain = ArgChain()
         argchain.arg(x)
         res = func.call(argchain, rffi.DOUBLE)
         v2.i -= 1
     return res
Пример #5
0
 def test_argchain(self):
     chain = ArgChain()
     assert chain.numargs == 0
     chain2 = chain.arg(42)
     assert chain2 is chain
     assert chain.numargs == 1
     intarg = chain.first
     assert chain.last is intarg
     assert intarg.intval == 42
     chain.arg(123.45)
     assert chain.numargs == 2
     assert chain.first is intarg
     assert intarg.next is chain.last
     floatarg = intarg.next
     assert floatarg.floatval == 123.45
Пример #6
0
 def test_argchain(self):
     chain = ArgChain()
     assert chain.numargs == 0
     chain2 = chain.arg(42)
     assert chain2 is chain
     assert chain.numargs == 1
     intarg = chain.first
     assert chain.last is intarg
     assert intarg.intval == 42
     chain.arg(123.45)
     assert chain.numargs == 2
     assert chain.first is intarg
     assert intarg.next is chain.last
     floatarg = intarg.next
     assert floatarg.floatval == 123.45
Пример #7
0
 def test_wrong_args(self):
     # so far the test passes but for the wrong reason :-), i.e. because
     # .arg() only supports integers and floats
     chain = ArgChain()
     x = lltype.malloc(lltype.GcStruct('xxx'))
     y = lltype.malloc(lltype.GcArray(rffi.SIGNED), 3)
     z = lltype.malloc(lltype.Array(rffi.SIGNED), 4, flavor='raw')
     py.test.raises(TypeError, "chain.arg(x)")
     py.test.raises(TypeError, "chain.arg(y)")
     py.test.raises(TypeError, "chain.arg(z)")
     lltype.free(z, flavor='raw')
Пример #8
0
    def call(self, funcspec, args, RESULT, is_struct=False, jitif=[]):
        """
        Call the specified function after constructing and ArgChain with the
        arguments in ``args``.

        The function is specified with ``funcspec``, which is a tuple of the
        form (lib, name, argtypes, restype).

        This method is overridden by metainterp/test/test_fficall.py in
        order to do the call in a loop and JIT it. The optional arguments are
        used only by that overridden method.

        """
        lib, name, argtypes, restype = funcspec
        func = lib.getpointer(name, argtypes, restype)
        chain = ArgChain()
        for arg in args:
            if isinstance(arg, tuple):
                methname, arg = arg
                meth = getattr(chain, methname)
                meth(arg)
            else:
                chain.arg(arg)
        return func.call(chain, RESULT, is_struct=is_struct)
Пример #9
0
    def call(self, funcspec, args, RESULT, is_struct=False, jitif=[]):
        """
        Call the specified function after constructing and ArgChain with the
        arguments in ``args``.

        The function is specified with ``funcspec``, which is a tuple of the
        form (lib, name, argtypes, restype).

        This method is overridden by metainterp/test/test_fficall.py in
        order to do the call in a loop and JIT it. The optional arguments are
        used only by that overridden method.

        """
        lib, name, argtypes, restype = funcspec
        func = lib.getpointer(name, argtypes, restype)
        chain = ArgChain()
        for arg in args:
            if isinstance(arg, tuple):
                methname, arg = arg
                meth = getattr(chain, methname)
                meth(arg)
            else:
                chain.arg(arg)
        return func.call(chain, RESULT, is_struct=is_struct)
Пример #10
0
 def test_by_ordinal(self):
     """
     RPY_EXPORTED
     int AAA_first_ordinal_function()
     {
         return 42;
     }
     """
     libfoo = self.get_libfoo()
     f_by_name = libfoo.getpointer('AAA_first_ordinal_function', [],
                                   types.sint)
     f_by_ordinal = libfoo.getpointer_by_ordinal(1, [], types.sint)
     print dir(f_by_name)
     assert f_by_name.funcsym == f_by_ordinal.funcsym
     chain = ArgChain()
     assert 42 == f_by_ordinal.call(chain, rffi.INT, is_struct=False)
Пример #11
0
 def test_by_ordinal2(self):
     """
     RPY_EXPORTED
     int __stdcall BBB_second_ordinal_function()
     {
         return 24;
     }
     """
     from rpython.rlib.libffi import WinDLL
     dll = WinDLL(self.libfoo_name)
     # __stdcall without a DEF file decorates the name with the number of bytes
     # that the callee will remove from the call stack
     f_by_name = dll.getpointer('_BBB_second_ordinal_function@0', [],
                                types.uint)
     f_by_ordinal = dll.getpointer_by_ordinal(2, [], types.uint)
     print dir(f_by_name)
     assert f_by_name.funcsym == f_by_ordinal.funcsym
     chain = ArgChain()
     assert 24 == f_by_ordinal.call(chain,
                                    lltype.Signed,
                                    is_struct=False)
Пример #12
0
 def test_by_ordinal2(self):
     """
     RPY_EXPORTED
     int __stdcall BBB_second_ordinal_function()
     {
         return 24;
     }
     """
     from rpython.rlib.libffi import WinDLL
     dll = WinDLL(self.libfoo_name)
     # __stdcall without a DEF file decorates the name with the number of bytes
     # that the callee will remove from the call stack
     # identical to __fastcall for amd64
     if IS_32_BIT:
         f_name = '_BBB_second_ordinal_function@0'
     else:
         f_name = 'BBB_second_ordinal_function'
     f_by_name = dll.getpointer(f_name, [], types.sint)
     f_by_ordinal = dll.getpointer_by_ordinal(2, [], types.sint)
     print dir(f_by_name)
     assert f_by_name.funcsym == f_by_ordinal.funcsym
     chain = ArgChain()
     assert 24 == f_by_ordinal.call(chain, rffi.INT, is_struct=False)