Esempio n. 1
0
 def test_call_onCFunc_returnsNewCFunc(self, CPROXY_CLASS, cint_type,
                                       addrspace):
     cfunc_type = cdm.CFuncType(None, addrspace=addrspace)
     cfunc_type2 = cdm.CFuncType(cint_type, [], addrspace)
     cfunc_obj = cdm.CFunc(cfunc_type, 123)
     cfunc_type2(cfunc_obj)
     CPROXY_CLASS.assert_called_once_with(cfunc_type2, 123)
Esempio n. 2
0
 def test_call_onCCode_runsAddrSpaceInvokeCCode(self, cint_type, addrspace):
     addrspace.simulate_c_code('funcname',
                               'cint f(void)',
                               retval=b'\x44\x33\x22\x11')
     func_type = cdm.CFuncType(cint_type, [], addrspace)
     func_obj = func_type('funcname')
     assert func_obj() == 0x11223344
Esempio n. 3
0
 def test_bind_bindsReturnTypeAndParameterTypes(self, unbound_cint_type,
                                                unbound_cint16_type,
                                                addrspace):
     cfunc_type = cdm.CFuncType(unbound_cint_type, [unbound_cint16_type])
     bound_cfunc_type = cfunc_type.bind(addrspace)
     assert bound_cfunc_type.returns.__addrspace__ is addrspace
     assert all(arg.__addrspace__ == addrspace
                for arg in bound_cfunc_type.args)
Esempio n. 4
0
 def test_call_onCallable_bridgesCallable(self, CPROXY_CLASS, cint_type,
                                          addrspace):
     cfunc_type = cdm.CFuncType(cint_type, [cint_type, cint_type],
                                addrspace)
     callback = Mock(return_value=0xAABBCCDD)
     cfunc_obj = cfunc_type(callback)
     _, bridge_adr = CPROXY_CLASS.call_args[0]
     result_adr = addrspace.alloc_memory(4)
     param_adr = addrspace.alloc_memory(8)
     addrspace.write_memory(param_adr, b'\x44\x33\x22\x11\x99\x88\x77\x66')
     c_sig = 'cint f(cint p0, cint p1)'
     addrspace.invoke_c_func(bridge_adr, c_sig, param_adr, result_adr)
     assert addrspace.read_memory(result_adr, 4) == b'\xDD\xCC\xBB\xAA'
     callback.assert_called_once_with(0x11223344, 0x66778899)
     assert isinstance(callback.call_args[0][0], cdm.CInt)
     assert cfunc_obj is CPROXY_CLASS.return_value
Esempio n. 5
0
 def test_eq_onBothReturnNone_returnsTrue(self):
     assert cdm.CFuncType(None, []) == cdm.CFuncType(None, [])
Esempio n. 6
0
 def test_eq_onSameFunc_returnsTrue(self, unbound_cint_type):
     assert cdm.CFuncType(unbound_cint_type, [unbound_cint_type]) \
            == cdm.CFuncType(unbound_cint_type, [unbound_cint_type])
Esempio n. 7
0
 def test_getPtr_createsNewCFuncPtrType(self, CFuncPointerType, addrspace):
     cfunc_type = cdm.CFuncType(addrspace=addrspace)
     cfuncptr_type = cfunc_type.ptr
     assert cfuncptr_type == CFuncPointerType.return_value
     CFuncPointerType.assert_called_with(cfunc_type, MACHINE_WORDSIZE,
                                         ENDIANESS, addrspace)
Esempio n. 8
0
 def test_shallowIterSubTypes_onReturnValue_yieldsReturnType(
         self, unbound_cint_type):
     test_func = cdm.CFuncType(unbound_cint_type)
     assert list(test_func.shallow_iter_subtypes()) == [unbound_cint_type]
Esempio n. 9
0
 def test_init_setsAttributes(self, unbound_cint_type, unbound_cint16_type):
     cfunc_type = cdm.CFuncType(unbound_cint_type,
                                [unbound_cint_type, unbound_cint16_type])
     assert cfunc_type.returns == unbound_cint_type
     assert cfunc_type.args == [unbound_cint_type, unbound_cint16_type]
Esempio n. 10
0
 def test_cDefinition_onParamsAndReturnVal_ok(self, unbound_cint_type,
                                              unbound_cint16_type):
     cfunc_type = cdm.CFuncType(unbound_cint_type,
                                [unbound_cint_type, unbound_cint16_type])
     assert cfunc_type.c_definition('func_name') == \
            'cint func_name(cint p0, cint16 p1)'
Esempio n. 11
0
 def test_call_withoutAddressSpace_raisesInvalidAddressSpaceError(self):
     cfunc_type = cdm.CFuncType()
     with pytest.raises(cdm.InvalidAddressSpaceError):
         cfunc_type(0)
Esempio n. 12
0
 def test_sizeof_raisesTypeError(self):
     cfunc_type = cdm.CFuncType()
     with pytest.raises(TypeError):
         _ = cfunc_type.sizeof()
Esempio n. 13
0
 def test_init_onParameterTypeHasDifferentAddrSpace_raiseInvalidAddrSpaceError(
         self, cint_type):
     with pytest.raises(cdm.InvalidAddressSpaceError):
         _ = cdm.CFuncType(None, [cint_type])
Esempio n. 14
0
 def test_repr_ok(self, unbound_cint_type, unbound_cint16_type):
     cfunc_type = cdm.CFuncType(unbound_cint_type,
                                [unbound_cint_type, unbound_cint16_type])
     assert repr(cfunc_type) == 'CFuncType(ts.cint, [ts.cint, ts.cint16])'
Esempio n. 15
0
 def test_init_onReturnTypeHasDifferentAddrSpace_raiseInvalidAddrSpaceError(
         self, cint_type):
     with pytest.raises(cdm.InvalidAddressSpaceError):
         _ = cdm.CFuncType(cint_type)
Esempio n. 16
0
 def test_call_onInt_callsConstructorWithFuncAdrOnly(
         self, CPROXY_CLASS, addrspace):
     cfunc_type = cdm.CFuncType(addrspace=addrspace)
     assert cfunc_type(123) is CPROXY_CLASS.return_value
     CPROXY_CLASS.assert_called_once_with(cfunc_type, 123)
Esempio n. 17
0
 def test_eq_onNoneRetVal_returnsFalse(self, unbound_cint_type):
     retnone_cfunc_type = cdm.CFuncType(None, [unbound_cint_type])
     retint_cfunc_type = cdm.CFuncType(unbound_cint_type,
                                       [unbound_cint_type])
     assert retnone_cfunc_type != retint_cfunc_type
     assert retint_cfunc_type != retnone_cfunc_type
Esempio n. 18
0
 def test_cDefinition_onNoneRetValAndEmptyParams_returnsVoidFunc(self):
     void_cfunc_type = cdm.CFuncType(None, [])
     assert void_cfunc_type.c_definition(
         'voidfunc') == 'void voidfunc(void)'
Esempio n. 19
0
 def test_eq_onDifferentArgs_returnsFalse(self, unbound_cint_type,
                                          unbound_cint16_type):
     assert cdm.CFuncType(unbound_cint_type, [unbound_cint_type]) \
            != cdm.CFuncType(unbound_cint16_type,
                             [unbound_cint_type, unbound_cint_type])
Esempio n. 20
0
 def test_repr_withAttr_addsAttrCall(self):
     cfunc_type = cdm.CFuncType().with_attr('attrB').with_attr('attrA')
     assert repr(cfunc_type) \
            == "CFuncType(None, []).with_attr('attrA').with_attr('attrB')"
Esempio n. 21
0
 def test_nullValue_raisesTypeError(self):
     cfunc_type = cdm.CFuncType()
     with pytest.raises(TypeError):
         _ = cfunc_type.null_val
Esempio n. 22
0
 def test_call_onArgs_passesArgs(self, cint_type, cint16_type, addrspace):
     addrspace.simulate_c_code('func_with_params',
                               exp_params=b'\x34\x12\x00\x00\x56\x00')
     cfunc_type = cdm.CFuncType(None, [cint_type, cint16_type], addrspace)
     cfunc_obj = cfunc_type('func_with_params')
     cfunc_obj(cint_type(0x1234), 0x56)
Esempio n. 23
0
 def test_cDefinition_onNoReferringDefParam_raiseTypeError(self):
     cfunc_type = cdm.CFuncType(None, [])
     assert cfunc_type.c_definition() == 'void f(void)'
Esempio n. 24
0
 def test_call_onSimpleResult_returnsCProxy(self, cint_type, addrspace):
     cfunc_type = cdm.CFuncType(cint_type, addrspace=addrspace)
     cfunc_obj = cfunc_type(Mock(return_value=123))
     result = cfunc_obj()
     assert isinstance(result, cdm.CInt)
     assert result == 123
Esempio n. 25
0
 def test_cDefintition_onAttr_ok(self):
     cdecl_cfunc_type = cdm.CFuncType().with_attr('__cdecl')
     assert cdecl_cfunc_type.c_definition('func') \
            == 'void __cdecl func(void)'
Esempio n. 26
0
 def test_call_onInvalidReturnValueType_raisesValueError(
         self, cint_type, addrspace):
     cfunc_type = cdm.CFuncType(cint_type, addrspace=addrspace)
     cfunc_obj = cfunc_type(Mock(return_value=4.4))
     with pytest.raises(TypeError):
         cfunc_obj()
Esempio n. 27
0
 def test_shallowIterSubTypes_onBasicFunc_returnsNothing(self):
     test_func = cdm.CFuncType()
     assert list(test_func.shallow_iter_subtypes()) == []
Esempio n. 28
0
 def test_call_onReturnTypeVoid_returnsNone(self, addrspace):
     void_cfunc_type = cdm.CFuncType(None, addrspace=addrspace)
     void_cfunc_obj = void_cfunc_type(Mock())
     assert void_cfunc_obj() is None
Esempio n. 29
0
 def test_shallowIterSubTypes_onNonParams_returnsParamTypes(
         self, unbound_cint_type, unbound_cint16_type):
     test_func = cdm.CFuncType(None,
                               [unbound_cint_type, unbound_cint16_type])
     assert list(test_func.shallow_iter_subtypes()) \
            == [unbound_cint_type, unbound_cint16_type]
Esempio n. 30
0
def cfunc_type(addrspace):
    return cdm.CFuncType(addrspace=addrspace)