mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-21 03:50:16 +00:00
88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
import c
|
||
|
||
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_void_1.read_struct(32)
|
||
assert my_struct2.size() == 32
|
||
|
||
data_bytes = bytes([1,2,3])
|
||
my_struct4 = c.struct(data_bytes)
|
||
|
||
try:
|
||
c.struct(True)
|
||
raise Exception('c.struct 的构造方法未能触发 TypeError("expected int or bytes")')
|
||
except TypeError:
|
||
pass
|
||
|
||
try:
|
||
c.struct(1,2,3)
|
||
raise Exception('c.struct 的构造方法未能触发 TypeError("expected 1 or 2 arguments")')
|
||
except TypeError:
|
||
pass
|
||
# ------------------------------------------------
|
||
my_struct1 = c.struct(16)
|
||
assert my_struct1.size() == 16
|
||
|
||
# 对 c.struct 的 copy 方法的测试不完全
|
||
assert my_struct1.copy().size() == 16
|
||
|
||
data_bytes = bytes([1,2,3])
|
||
my_struct4 = c.struct(data_bytes)
|
||
assert my_struct4.addr().read_bytes(
|
||
my_struct4.size()
|
||
) == data_bytes
|
||
|
||
|
||
# ------------------------------------------------
|
||
# 此处测试并不完全
|
||
c_void_1 = c.malloc(16)
|
||
my_struct1 = c.struct(16)
|
||
c_void_1.write_struct(my_struct1)
|
||
assert c_void_1.read_struct(16) == my_struct1
|
||
|
||
from c import array, int_
|
||
a = array(10, item_size=4)
|
||
assert a.item_count == 10
|
||
assert a.item_size == 4
|
||
|
||
_ = hash(a)
|
||
a[4] = int_(123)
|
||
assert a[4] == int_(123) |