ꦫꦣꦺꦤ꧀ꦄꦤ꧀ꦠꦱꦺꦤ

PATH: usr / lib64 / python2.7 / ctypes / test

from ctypes import *
from ctypes.test import need_symbol
import unittest

# IMPORTANT INFO:
#
# Consider this call:
#    func.restype = c_char_p
#    func(c_char_p("123"))
# It returns
#    "123"
#
# WHY IS THIS SO?
#
# argument tuple (c_char_p("123"), ) is destroyed after the function
# func is called, but NOT before the result is actually built.
#
# If the arglist would be destroyed BEFORE the result has been built,
# the c_char_p("123") object would already have a zero refcount,
# and the pointer passed to (and returned by) the function would
# probably point to deallocated space.
#
# In this case, there would have to be an additional reference to the argument...

import _ctypes_test
testdll = CDLL(_ctypes_test.__file__)

# Return machine address `a` as a (possibly long) non-negative integer.
# Starting with Python 2.5, id(anything) is always non-negative, and
# the ctypes addressof() inherits that via PyLong_FromVoidPtr().
def positive_address(a):
    if a >= 0:
        return a
    # View the bits in `a` as unsigned instead.
    import struct
    num_bits = struct.calcsize("P") * 8 # num bits in native machine address
    a += 1L << num_bits
    assert a >= 0
    return a

def c_wbuffer(init):
    n = len(init) + 1
    return (c_wchar * n)(*init)

class CharPointersTestCase(unittest.TestCase):

def setUp(self):
        func = testdll._testfunc_p_p
        func.restype = c_long
        func.argtypes = None

def test_paramflags(self):
        # function returns c_void_p result,
        # and has a required parameter named 'input'
        prototype = CFUNCTYPE(c_void_p, c_void_p)
        func = prototype(("_testfunc_p_p", testdll),
                         ((1, "input"),))

try:
            func()
        except TypeError, details:
            self.assertEqual(str(details), "required argument 'input' missing")
        else:
            self.fail("TypeError not raised")

self.assertEqual(func(None), None)
        self.assertEqual(func(input=None), None)

def test_int_pointer_arg(self):
        func = testdll._testfunc_p_p
        func.restype = c_long
        self.assertEqual(0, func(0))

ci = c_int(0)

func.argtypes = POINTER(c_int),
        self.assertEqual(positive_address(addressof(ci)),
                             positive_address(func(byref(ci))))

func.argtypes = c_char_p,
        self.assertRaises(ArgumentError, func, byref(ci))

func.argtypes = POINTER(c_short),
        self.assertRaises(ArgumentError, func, byref(ci))

func.argtypes = POINTER(c_double),
        self.assertRaises(ArgumentError, func, byref(ci))

def test_POINTER_c_char_arg(self):
        func = testdll._testfunc_p_p
        func.restype = c_char_p
        func.argtypes = POINTER(c_char),

self.assertEqual(None, func(None))
        self.assertEqual("123", func("123"))
        self.assertEqual(None, func(c_char_p(None)))
        self.assertEqual("123", func(c_char_p("123")))

self.assertEqual("123", func(c_buffer("123")))
        ca = c_char("a")
        self.assertEqual("a", func(pointer(ca))[0])
        self.assertEqual("a", func(byref(ca))[0])

def test_c_char_p_arg(self):
        func = testdll._testfunc_p_p
        func.restype = c_char_p
        func.argtypes = c_char_p,

self.assertEqual(None, func(None))
        self.assertEqual("123", func("123"))
        self.assertEqual(None, func(c_char_p(None)))
        self.assertEqual("123", func(c_char_p("123")))

self.assertEqual("123", func(c_buffer("123")))
        ca = c_char("a")
        self.assertEqual("a", func(pointer(ca))[0])
        self.assertEqual("a", func(byref(ca))[0])

def test_c_void_p_arg(self):
        func = testdll._testfunc_p_p
        func.restype = c_char_p
        func.argtypes = c_void_p,

self.assertEqual(None, func(None))
        self.assertEqual("123", func("123"))
        self.assertEqual("123", func(c_char_p("123")))
        self.assertEqual(None, func(c_char_p(None)))

self.assertEqual("123", func(c_buffer("123")))
        ca = c_char("a")
        self.assertEqual("a", func(pointer(ca))[0])
        self.assertEqual("a", func(byref(ca))[0])

func(byref(c_int()))
        func(pointer(c_int()))
        func((c_int * 3)())

@need_symbol('c_wchar_p')
    def test_c_void_p_arg_with_c_wchar_p(self):
        func = testdll._testfunc_p_p
        func.restype = c_wchar_p
        func.argtypes = c_void_p,

self.assertEqual(None, func(c_wchar_p(None)))
        self.assertEqual(u"123", func(c_wchar_p(u"123")))

def test_instance(self):
        func = testdll._testfunc_p_p
        func.restype = c_void_p

class X:
            _as_parameter_ = None

func.argtypes = c_void_p,
        self.assertEqual(None, func(X()))

func.argtypes = None
        self.assertEqual(None, func(X()))

@need_symbol('c_wchar')
class WCharPointersTestCase(unittest.TestCase):

def setUp(self):
        func = testdll._testfunc_p_p
        func.restype = c_int
        func.argtypes = None

def test_POINTER_c_wchar_arg(self):
        func = testdll._testfunc_p_p
        func.restype = c_wchar_p
        func.argtypes = POINTER(c_wchar),

self.assertEqual(None, func(None))
        self.assertEqual(u"123", func(u"123"))
        self.assertEqual(None, func(c_wchar_p(None)))
        self.assertEqual(u"123", func(c_wchar_p(u"123")))

self.assertEqual(u"123", func(c_wbuffer(u"123")))
        ca = c_wchar("a")
        self.assertEqual(u"a", func(pointer(ca))[0])
        self.assertEqual(u"a", func(byref(ca))[0])

def test_c_wchar_p_arg(self):
        func = testdll._testfunc_p_p
        func.restype = c_wchar_p
        func.argtypes = c_wchar_p,

c_wchar_p.from_param(u"123")

self.assertEqual(None, func(None))
        self.assertEqual("123", func(u"123"))
        self.assertEqual(None, func(c_wchar_p(None)))
        self.assertEqual("123", func(c_wchar_p("123")))

# XXX Currently, these raise TypeErrors, although they shouldn't:
        self.assertEqual("123", func(c_wbuffer("123")))
        ca = c_wchar("a")
        self.assertEqual("a", func(pointer(ca))[0])
        self.assertEqual("a", func(byref(ca))[0])

class ArrayTest(unittest.TestCase):
    def test(self):
        func = testdll._testfunc_ai8
        func.restype = POINTER(c_int)
        func.argtypes = c_int * 8,

func((c_int * 8)(1, 2, 3, 4, 5, 6, 7, 8))

# This did crash before:

def func(): pass
        CFUNCTYPE(None, c_int * 3)(func)

################################################################

if __name__ == '__main__':
    unittest.main()

[+] ..
[-] test_values.pyc [open]
[-] runtests.pyc [open]
[-] test_objects.pyc [open]
[-] test_checkretval.pyc [open]
[-] test_find.pyo [open]
[-] test_macholib.py [open]
[-] test_loading.pyo [open]
[-] test_wintypes.pyo [open]
[-] test_win32.py [open]
[-] test_loading.pyc [open]
[-] test_frombuffer.pyo [open]
[-] test_returnfuncptrs.py [open]
[-] test_sizes.pyc [open]
[-] test_libc.pyc [open]
[-] test_libc.pyo [open]
[-] test_random_things.py [open]
[-] test_as_parameter.pyo [open]
[-] test_refcounts.pyo [open]
[-] test_random_things.pyc [open]
[-] test_numbers.py [open]
[-] test_pickling.pyo [open]
[-] test_values.pyo [open]
[-] test_simplesubclasses.pyo [open]
[-] test_repr.py [open]
[-] test_cfuncs.pyc [open]
[-] test_structures.pyo [open]
[-] test_incomplete.pyo [open]
[-] test_frombuffer.pyc [open]
[-] test_byteswap.py [open]
[-] test_pickling.pyc [open]
[-] test_parameters.py [open]
[-] test_macholib.pyc [open]
[-] test_array_in_pointer.pyo [open]
[-] test_repr.pyo [open]
[-] test_delattr.pyo [open]
[-] test_callbacks.pyc [open]
[-] test_strings.pyo [open]
[-] test_funcptr.pyc [open]
[-] test_byteswap.pyo [open]
[-] test_frombuffer.py [open]
[-] test_find.py [open]
[-] test_anon.pyc [open]
[-] test_errno.pyc [open]
[-] test_pep3118.pyo [open]
[-] test_funcptr.py [open]
[-] test_strings.pyc [open]
[-] test_cfuncs.py [open]
[-] test_array_in_pointer.py [open]
[-] test_bitfields.pyc [open]
[-] test_python_api.pyc [open]
[-] test_unicode.pyo [open]
[-] test_anon.pyo [open]
[-] test_slicing.pyc [open]
[-] test_refcounts.pyc [open]
[-] runtests.pyo [open]
[-] test_pointers.pyo [open]
[-] test_delattr.pyc [open]
[-] __init__.pyc [open]
[-] test_prototypes.pyc [open]
[-] test_cast.pyo [open]
[-] test_checkretval.pyo [open]
[-] test_pep3118.py [open]
[-] test_wintypes.pyc [open]
[-] test_sizes.py [open]
[-] test_numbers.pyc [open]
[-] test_varsize_struct.pyo [open]
[-] test_internals.py [open]
[-] test_objects.py [open]
[-] test_anon.py [open]
[-] test_struct_fields.py [open]
[-] test_functions.pyo [open]
[-] test_libc.py [open]
[-] test_struct_fields.pyc [open]
[-] test_random_things.pyo [open]
[-] test_arrays.pyc [open]
[-] test_memfunctions.py [open]
[-] test_python_api.pyo [open]
[-] test_incomplete.pyc [open]
[-] test_returnfuncptrs.pyo [open]
[-] test_array_in_pointer.pyc [open]
[-] test_structures.pyc [open]
[-] test_errno.py [open]
[-] test_returnfuncptrs.pyc [open]
[-] test_as_parameter.pyc [open]
[-] test_keeprefs.pyo [open]
[-] test_memfunctions.pyc [open]
[-] test_pep3118.pyc [open]
[-] test_parameters.pyo [open]
[-] test_cast.pyc [open]
[-] test_simplesubclasses.py [open]
[-] test_internals.pyc [open]
[-] test_keeprefs.py [open]
[-] test_pointers.py [open]
[-] test_bitfields.pyo [open]
[-] test_macholib.pyo [open]
[-] test_refcounts.py [open]
[-] test_values.py [open]
[-] test_checkretval.py [open]
[-] test_arrays.py [open]
[-] test_buffers.pyc [open]
[-] test_python_api.py [open]
[-] test_strings.py [open]
[-] test_unicode.pyc [open]
[-] test_parameters.pyc [open]
[-] test_buffers.pyo [open]
[-] test_cfuncs.pyo [open]
[-] test_prototypes.pyo [open]
[-] test_keeprefs.pyc [open]
[-] test_stringptr.pyc [open]
[-] test_unicode.py [open]
[-] test_struct_fields.pyo [open]
[-] test_cast.py [open]
[-] test_slicing.pyo [open]
[-] test_buffers.py [open]
[-] test_prototypes.py [open]
[-] test_internals.pyo [open]
[-] test_unaligned_structures.py [open]
[-] test_functions.pyc [open]
[-] test_stringptr.pyo [open]
[-] test_repr.pyc [open]
[-] __init__.py [open]
[-] test_numbers.pyo [open]
[-] test_init.pyc [open]
[-] test_errno.pyo [open]
[-] test_structures.py [open]
[-] test_win32.pyo [open]
[-] test_varsize_struct.pyc [open]
[-] test_byteswap.pyc [open]
[-] test_stringptr.py [open]
[-] test_arrays.pyo [open]
[-] test_slicing.py [open]
[-] test_loading.py [open]
[-] test_delattr.py [open]
[-] test_objects.pyo [open]
[-] test_init.pyo [open]
[-] test_varsize_struct.py [open]
[-] test_init.py [open]
[-] test_memfunctions.pyo [open]
[-] test_as_parameter.py [open]
[-] test_incomplete.py [open]
[-] test_simplesubclasses.pyc [open]
[-] test_bitfields.py [open]
[-] test_callbacks.py [open]
[-] test_win32.pyc [open]
[-] test_unaligned_structures.pyo [open]
[-] test_unaligned_structures.pyc [open]
[-] test_sizes.pyo [open]
[-] test_funcptr.pyo [open]
[-] __init__.pyo [open]
[-] test_pointers.pyc [open]
[-] runtests.py [open]
[-] test_find.pyc [open]
[-] test_callbacks.pyo [open]
[-] test_wintypes.py [open]
[-] test_functions.py [open]
[-] test_pickling.py [open]