29
|
1 import pydc
|
|
2 import sys
|
|
3 import platform
|
57
|
4 import copy
|
|
5 import types
|
29
|
6
|
|
7
|
|
8
|
|
9 def theader(title):
|
34
|
10 print("\n"+title)
|
|
11 print('%8s %20s %16s %-20s %11s %-16s -> %16s %-16s %12s %-16s # %s' % ('DC_SIG', 'C_RET_T', 'C_FSYM', 'C_PARAMLIST', 'PY_ARG_T', 'IN_ARGS', 'RET_VAL', 'PY_RET_T', 'OUT_ARGS', 'OUT_ARGS_T', 'NOTES'))
|
29
|
12
|
|
13
|
57
|
14 def t(lib, dcsig, c_rtype, c_fsym, c_paramlist, extra_msg, **kwargs):
|
|
15 args = kwargs['i'] if 'i' in kwargs else ()
|
|
16 post_args = kwargs['p'] if 'p' in kwargs else copy.deepcopy(args) # expected args after call (as some can be modified in-place)
|
|
17 exp_ret = kwargs['r'] if 'r' in kwargs else None # expected return value
|
|
18 err_sgr = ''
|
34
|
19 # before call
|
|
20 inarg_types = '('+','.join(map(lambda x: type(x).__name__, args))+')'
|
57
|
21 inargs_str = ','.join(map(str, args))
|
34
|
22 # call
|
29
|
23 try:
|
|
24 fp = pydc.find(lib, c_fsym)
|
|
25 r = pydc.call(fp, dcsig, *args)
|
|
26 rt = type(r).__name__
|
58
|
27 if(r != exp_ret or args != post_args): # @@@ type test also or type(r) is not type(exp_ret)):
|
|
28 if((type(exp_ret) is not types.LambdaType) or exp_ret(r) == False):
|
57
|
29 err_sgr = '\033[41m'
|
29
|
30 except:
|
|
31 r = '[EXCEPTION]'
|
|
32 rt = '!'
|
57
|
33 if(exp_ret != Exception):
|
|
34 err_sgr = '\033[41m'
|
34
|
35 e = str(sys.exc_info()[1])
|
|
36 extra_msg += ' "'+(e if len(e)<32 else e[0:32]+'...')+'"'
|
|
37 # after call
|
|
38 outarg_types = '('+','.join(map(lambda x: type(x).__name__, args))+')'
|
|
39 outargs = ','.join(map(str, args))
|
57
|
40 print('%s%8s %20s %16s %-20s %11s \033[33m%-16s\033[39m -> \033[32m%16.16s\033[39m %-16s %12s \033[32m%-16s\033[0m # %s' % (err_sgr, dcsig, c_rtype, c_fsym, c_paramlist, inarg_types, inargs_str, str(r), '('+rt+')', outarg_types, outargs, extra_msg))
|
|
41
|
|
42 return r
|
29
|
43
|
|
44
|
|
45
|
|
46 # some libc tests ------------------
|
|
47
|
|
48 try:
|
57
|
49 if len(sys.argv) > 1:
|
|
50 libc = pydc.load(sys.argv[1])
|
|
51 elif sys.platform == "win32":
|
29
|
52 libc = pydc.load("msvcrt")
|
|
53 elif sys.platform == "darwin":
|
|
54 libc = pydc.load("/usr/lib/libc.dylib")
|
|
55 elif "bsd" in sys.platform:
|
|
56 libc = pydc.load("/usr/lib/libc.so")
|
|
57 #libc = pydc.load("/lib/libc.so.7")
|
|
58 elif platform.architecture()[0] == "64bit":
|
|
59 libc = pydc.load("/lib64/libc.so.6")
|
|
60 else:
|
57
|
61 libc = pydc.load("/lib/libc.so")
|
29
|
62
|
|
63 theader('CLIB TESTS:')
|
|
64
|
|
65 # void()
|
|
66 t(libc, ")v", "void", "sranddev", "(void)", '')
|
|
67
|
|
68 # int()
|
58
|
69 t(libc, ")i", "int", "rand", "(void)", '', r=lambda i: type(i) is int)
|
29
|
70
|
|
71 # void(unsigned int)
|
57
|
72 t(libc, "I)v", "void", "srand", "(int)", '', i=(123,))
|
29
|
73
|
|
74 # int() (and one different helper call for test)
|
57
|
75 x = \
|
|
76 t(libc, ")i", "int", "rand", "(void)", 'with seed <------,', r=lambda i: type(i) is int)
|
|
77 t(libc, "I)v", "void", "srand", "(int)", 'set same seed |', i=(123,))
|
|
78 t(libc, ")i", "int", "rand", "(void)", 'should be same result...', r=x)
|
|
79 t(libc, ")i", "int", "rand", "(void)", '...and now different result', r=lambda i: type(i) is int and i!=x)
|
29
|
80
|
|
81 # int(int)
|
57
|
82 t(libc, "i)i", "int", "abs", "(int)", ' 10 => 10', i=( 10,), r=10)
|
|
83 t(libc, "i)i", "int", "abs", "(int)", ' 0 => 0', i=( 0,), r=0)
|
|
84 t(libc, "i)i", "int", "abs", "(int)", ' -9209 => 9209', i=(-9209,), r=9209)
|
29
|
85
|
|
86 # long(long)
|
57
|
87 t(libc, "j)j", "long", "labs", "(long)", ' 48 => 48', i=( 48,), r=48)
|
|
88 t(libc, "j)j", "long", "labs", "(long)", ' 0 => 0', i=( 0,), r=0)
|
|
89 t(libc, "j)j", "long", "labs", "(long)", '-4271477497 => 4271477497', i=(-4271477497,), r=4271477497)
|
29
|
90
|
|
91 # long long(long long)
|
57
|
92 t(libc, "l)l", "long long", "labs", "(long long)", ' 6334810198 => 6334810198', i=( 6334810198,), r=6334810198)
|
|
93 t(libc, "l)l", "long long", "labs", "(long long)", ' 1 => 1', i=( 1,), r=1)
|
|
94 t(libc, "l)l", "long long", "labs", "(long long)", ' 0 => 0', i=( 0,), r=0)
|
|
95 t(libc, "l)l", "long long", "labs", "(long long)", ' -1 => 1', i=( -1,), r=1)
|
|
96 t(libc, "l)l", "long long", "labs", "(long long)", '-7358758407 => 7358758407', i=(-7358758407,), r=7358758407)
|
29
|
97
|
|
98 pydc.free(libc)
|
|
99 except:
|
57
|
100 print("\033[33mskipping clib tests because: "+str(sys.exc_info()[1])+"\033[0m\nnote: c-lib to use can be specified as command line param")
|
29
|
101
|
|
102
|
|
103 # tests with own .so for testing all conversions ------------------
|
|
104
|
|
105 l = pydc.load(sys.path[0]+"/test.so")
|
|
106
|
34
|
107 # "long" typed test value use for Python 2
|
|
108 long_i = 11234
|
|
109 long_h = 0xdeadc0de
|
|
110 if sys.version_info < (3, 0):
|
|
111 long_i = long(11234)
|
|
112 long_h = long(0xdeadc0de)
|
|
113
|
29
|
114 # test all possible arg types and their conversions to and from python, with
|
|
115 # specific focus/tests in areas where python 2 and 3 differ
|
|
116 theader('ARG & RET TYPE CONVERSION TESTS:')
|
57
|
117 t(l, "B)B", "int", "i_plus_one", "(int)", ' False => True (using int func in C)', i=(False,), r=True)
|
34
|
118
|
57
|
119 t(l, "c)c", "char", "c_plus_one", "(char)", ' "a" (97) => 98', i=( 'a',), r=98)
|
|
120 t(l, "C)C", "unsigned char", "uc_plus_one", "(unsigned char)", ' "a" (97) => 98', i=( 'a',), r=98)
|
|
121 t(l, "c)c", "char", "c_plus_one", "(char)", ' -2 => -1', i=( -2,), r=-1)
|
|
122 t(l, "C)C", "unsigned char", "uc_plus_one", "(unsigned char)", ' 10 => 11', i=( 10,), r=11)
|
34
|
123
|
57
|
124 t(l, "s)s", "short", "s_plus_one", "(short)", ' 10 => 11', i=( 10,), r=11)
|
|
125 t(l, "S)S", "unsigned short", "us_plus_one", "(unsigned short)", ' 10 => 11', i=( 10,), r=11)
|
34
|
126
|
57
|
127 t(l, "i)i", "int", "i_plus_one", "(int)", ' 10 => 11', i=( 10,), r=11)
|
|
128 t(l, "I)I", "unsigned int", "ui_plus_one", "(unsigned int)", ' 10 => 11', i=( 10,), r=11)
|
34
|
129
|
57
|
130 t(l, "j)j", "long", "l_plus_one", "(long)", ' 10 => 11', i=( 10,), r=11)
|
|
131 t(l, "J)J", "unsigned long", "ul_plus_one", "(unsigned long)", ' 10 => 11', i=( 10,), r=11)
|
34
|
132
|
57
|
133 t(l, "l)l", "long long", "ll_plus_one", "(long long)", ' 10 => 11', i=( 10,), r=11)
|
|
134 t(l, "L)L", "unsigned long long", "ull_plus_one", "(unsigned long long)", ' 10 => 11', i=( 10,), r=11)
|
|
135 t(l, "l)l", "long long", "ll_plus_one", "(long long)", ' 11234 => 11235', i=(long_i,), r=11235)
|
|
136 t(l, "L)L", "unsigned long long", "ull_plus_one", "(unsigned long long)", ' 11234 => 11235', i=(long_i,), r=11235)
|
34
|
137
|
57
|
138 t(l, "f)f", "float", "f_plus_one", "(float)", ' -1.23 => -0.23... (w/ fp imprecision)', i=( -1.23,), r=-0.23000001907348633)
|
|
139 t(l, "d)d", "double", "d_plus_one", "(double)", ' 5.67 => 6.67', i=( 5.67,), r= 6.67)
|
34
|
140
|
57
|
141 t(l, "Z)Z", "const char*", "ccp_plus_one", "(const char*)", '"lose char" => "ose char"', i=( 'lose char',), r= 'ose char') # string object
|
|
142 t(l, "Z)Z", "const char*", "ccp_plus_one", "(const char*)", '"X_unicode" => "_unicode"', i=( u'X_unicode',), r=u'_unicode') # string object (unicode in Python 2)
|
|
143 t(l, "Z)Z", "const char*", "ccp_plus_one", "(const char*)", '"1lessbyte" => "lessbyte"', i=( b'1lessbyte',), r= 'lessbyte') # bytes object
|
|
144 t(l, "Z)Z", "const char*", "ccp_plus_one", "(const char*)", ' "xY" => "Y"', i=(bytearray(b'xY'),), r= 'Y') # bytearray object
|
34
|
145
|
57
|
146 t(l, "p)Z", "const char*", "ccp_plus_one", "(const char*)", ' "xY" => "Y"', i=(bytearray(b'xY'),), r='Y') # bytearray object
|
58
|
147 t(l, "p)p", "const char*", "ccp_plus_one", "(const char*)", ' "xY" => p+1 (~ odd addr)', i=(bytearray(b'xY'),), r=lambda i: type(i) is int and (i%2)==1) # bytearray object
|
57
|
148 t(l, "p)p", "const char*", "ccp_plus_one", "(const char*)", ' 0xdeadc0de => 0xdeadc0de+1=3735929055', i=(long_h,), r=3735929055) # handle (integer interpreted as ptr)
|
|
149 t(l, "p)p", "const char*", "ccp_plus_one", "(const char*)", ' 0xdeadc0de => 0xdeadc0de+1=3735929055', i=(long_h,), r=3735929055) # handle (integer interpreted as ptr, long in Python 2)
|
|
150 t(l, "p)p", "const char*", "ccp_plus_one", "(const char*)", ' NULL => NULL+1=1', i=( None,), r=1) # NULL, adding one will result in 0x1
|
34
|
151
|
|
152 # functions that change buffers
|
|
153 theader('TESTS OF IMMUTABLE AND MUTABLE PYTHON BUFFERS:')
|
57
|
154 t(l, "Z)v", "const char*", "cp_head_incr", "(const char*)", ' "string" => None / arg => "string" (not modified)"', i=( 'string',)) # string object
|
|
155 t(l, "Z)v", "const char*", "cp_head_incr", "(const char*)", ' "UnIcOdE" => None / arg => "UnIcOdE" (not modified)"', i=( u'UnIcOdE',)) # string object (unicode in Python 2)
|
|
156 t(l, "Z)v", "const char*", "cp_head_incr", "(const char*)", ' "BCssk#" => None / arg => "BCssk#" (not modified)"', i=( b'BCssk#',)) # bytes object
|
|
157 t(l, "Z)v", "const char*", "cp_head_incr", "(const char*)", ' "xY" => None / arg => "xY" (not modified)"', i=(bytearray(b'xY'),)) # bytearray object
|
|
158 t(l, "p)v", "const char*", "cp_head_incr", "(const char*)", ' "xY" => None / arg => "yY" (!MODIFIED!)"', i=(bytearray(b'xY'),), p=(bytearray(b'yY'),)) # bytearray object
|
29
|
159
|
|
160 # tested checked value conversions
|
|
161 theader('ARG & RET TYPE CONVERSION TESTS FOR RANGE CHECKED TYPES:')
|
57
|
162 t(l, "c)c", "char", "c_plus_one", "(char)", ' "~" => 127', i=( '~',), r=127)
|
|
163 t(l, "c)c", "char", "c_plus_one", "(char)", ' "~" => 127', i=( '~',), r=127)
|
|
164 t(l, "c)c", "char", "c_plus_one", "(char)", ' "" => input exc:', i=( '',), r=Exception)
|
|
165 t(l, "c)c", "char", "c_plus_one", "(char)", ' "" => input exc:', i=( '',), r=Exception)
|
|
166 t(l, "C)C", "unsigned char", "uc_plus_one", "(unsigned char)", ' "ab" => input exc:', i=('ab',), r=Exception)
|
|
167 t(l, "C)C", "unsigned char", "uc_plus_one", "(unsigned char)", ' "ab" => input exc:', i=('ab',), r=Exception)
|
29
|
168
|
57
|
169 t(l, "c)c", "char", "c_plus_one", "(char)", ' -128 => -127', i=(-128,), r=-127)
|
|
170 t(l, "c)c", "char", "c_plus_one", "(char)", ' 127 => -128 (wrapped)', i=( 127,), r=-128)
|
|
171 t(l, "c)c", "char", "c_plus_one", "(char)", ' -129 => input exc:', i=(-129,), r=Exception)
|
|
172 t(l, "c)c", "char", "c_plus_one", "(char)", ' 128 => input exc:', i=( 128,), r=Exception)
|
|
173
|
|
174 t(l, "C)C", "unsigned char", "uc_plus_one", "(unsigned char)", ' 0 => 1', i=( 0,), r=1)
|
|
175 t(l, "C)C", "unsigned char", "uc_plus_one", "(unsigned char)", ' 255 => 0 (wrapped)', i=( 255,), r=0)
|
|
176 t(l, "C)C", "unsigned char", "uc_plus_one", "(unsigned char)", ' -1 => input exc:', i=( -1,), r=Exception)
|
|
177 t(l, "C)C", "unsigned char", "uc_plus_one", "(unsigned char)", ' 256 => input exc:', i=( 256,), r=Exception)
|
|
178
|
|
179 t(l, "s)s", "short", "s_plus_one", "(short)", ' -32768 => -32767', i=(-32768,), r=-32767)
|
|
180 t(l, "s)s", "short", "s_plus_one", "(short)", ' 32767 => -32768 (wrapped)',i=( 32767,), r=-32768)
|
|
181 t(l, "s)s", "short", "s_plus_one", "(short)", ' -32769 => input exc:', i=(-32769,), r=Exception)
|
|
182 t(l, "s)s", "short", "s_plus_one", "(short)", ' 32768 => input exc:', i=( 32768,), r=Exception)
|
|
183
|
|
184 t(l, "S)S", "unsigned short", "us_plus_one", "(unsigned short)", ' 0 => 1', i=( 0,), r=1)
|
|
185 t(l, "S)S", "unsigned short", "us_plus_one", "(unsigned short)", ' 65535 => 0 (wrapped)', i=(65535,), r=0)
|
|
186 t(l, "S)S", "unsigned short", "us_plus_one", "(unsigned short)", ' -1 => input exc:', i=( -1,), r=Exception)
|
|
187 t(l, "S)S", "unsigned short", "us_plus_one", "(unsigned short)", ' 65536 => input exc:', i=(65536,), r=Exception)
|
29
|
188
|
57
|
189 t(l, "p)Z", "const char*", "ccp_plus_one", "(const char*)", '"w/pointer" => input exc:',i=( 'w/pointer',), r=Exception) # string object, not passable as 'p'ointer
|
|
190 t(l, "p)Z", "const char*", "ccp_plus_one", "(const char*)", '"X_unicode" => input exc:',i=(u'X_unicode',), r=Exception) # string object (unicode in Python 2), not passable as 'p'ointer
|
|
191 t(l, "p)Z", "const char*", "ccp_plus_one", "(const char*)", '"1less/ptr" => input exc:',i=(b'1less/ptr',), r=Exception) # bytes object, not passable as 'p'ointer
|
|
192 t(l, "p)p", "const char*", "ccp_plus_one", "(const char*)", ' "x" => input exc:',i=( 'x',), r=Exception) # string object, not passable as 'p'ointer
|
29
|
193
|