mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-20 03:20:18 +00:00
98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
import c
|
||
|
||
c_int = c.refl("int")
|
||
assert c_int.name() == "int"
|
||
assert c_int.__repr__() == '<ctype \'int\'>'
|
||
# ------------------------------------------------
|
||
c_int_1 = c.refl("int")
|
||
c_struct_1 = c_int_1()
|
||
assert (c_int_1() == c_int_1())
|
||
assert (c_struct_1 == c_struct_1) == True
|
||
# ------------------------------------------------
|
||
assert c.void_p.from_hex('0x2568b60').hex() == '0x2568b60'
|
||
|
||
# ------------------------------------------------
|
||
class HexAddress:
|
||
def __init__(self, address):
|
||
if not address.startswith("0x"): # 确保地址以0x开头
|
||
raise ValueError("Address should start with '0x'.")
|
||
self.address = address[2:] # 去除0x前缀，并保存十六进制字符串
|
||
|
||
def __str__(self):
|
||
return "0x" + self.address
|
||
|
||
def __add__(self, other):
|
||
if isinstance(other, int):
|
||
return HexAddress(hex(int(self.address, 16) + other)) # 将字符串地址转为整数进行运算
|
||
elif isinstance(other, HexAddress):
|
||
return HexAddress(hex(int(self.address, 16) + int(other.address, 16))) # 将字符串地址转为整数进行运算
|
||
else:
|
||
raise TypeError("Unsupported operand type for +: HexAddress and {}".format(type(other)))
|
||
|
||
def __sub__(self, other):
|
||
if isinstance(other, int):
|
||
return HexAddress(hex(int(self.address, 16) - other)) # 将字符串地址转为整数进行运算
|
||
elif isinstance(other, HexAddress):
|
||
return HexAddress(hex(int(self.address, 16) - int(other.address, 16))) # 将字符串地址转为整数进行运算
|
||
else:
|
||
raise TypeError("Unsupported operand type for -: HexAddress and {}".format(type(other)))
|
||
|
||
c_void_1 = c.malloc(8)
|
||
|
||
assert (c_void_1 + 8).hex() == c.void_p.from_hex(str(HexAddress(c_void_1.hex()) + 8)).hex()
|
||
assert (c_void_1 - 8).hex() == c.void_p.from_hex(str(HexAddress(c_void_1.hex()) - 8)).hex()
|
||
|
||
# ------------------------------------------------
|
||
# 此处测试并不完全
|
||
c_void_1 = c.malloc(8)
|
||
c_void_1.read_bytes(5)
|
||
c_void_1.write_bytes(c_void_1.read_bytes(5))
|
||
# ------------------------------------------------
|
||
c_void_1 = c.malloc(32)
|
||
my_struct2 = c.struct(c_void_1, 32)
|
||
|
||
data_str = "Hello, World!"
|
||
my_struct3 = c.struct(data_str)
|
||
|
||
data_bytes = bytes([1,2,3])
|
||
my_struct4 = c.struct(data_bytes)
|
||
|
||
try:
|
||
c.struct(True)
|
||
raise Exception('c.struct 的构造方法未能触发 TypeError("expected int, str or bytes")')
|
||
except TypeError:
|
||
pass
|
||
|
||
try:
|
||
c.struct(1,2,3)
|
||
raise Exception('c.struct 的构造方法未能触发 TypeError("expected 1 or 2 arguments")')
|
||
except TypeError:
|
||
pass
|
||
|
||
try:
|
||
x = c.refl("int")['a']
|
||
except KeyError:
|
||
pass
|
||
# ------------------------------------------------
|
||
my_struct1 = c.struct(16)
|
||
assert my_struct1.size() == 16
|
||
|
||
# 对 c.struct 的 copy 方法的测试不完全
|
||
assert my_struct1.copy().size() == 16
|
||
|
||
data_str = "Hello, World!"
|
||
my_struct3 = c.struct(data_str)
|
||
assert my_struct3.to_string() == data_str
|
||
|
||
data_bytes = bytes([1,2,3])
|
||
my_struct4 = c.struct(data_bytes)
|
||
assert my_struct4.to_bytes() == data_bytes
|
||
|
||
|
||
# ------------------------------------------------
|
||
# 此处测试并不完全
|
||
c_void_1 = c.malloc(16)
|
||
my_struct1 = c.struct(16)
|
||
c_void_1.read_struct('long')
|
||
c_void_1.write_struct(my_struct1)
|