Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 11:30:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
blueloveTH 2023-05-15 15:13:47 +08:00
parent 686905a7bf
commit 649f0d54e7
2 changed files with 54 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/c.pyi
View File

@ -4,6 +4,8 @@ def malloc(size: int) -> 'void_p': ...
def free(ptr: 'void_p') -> None: ...
def sizeof(type: str) -> int: ...
def refl(name: str) -> '_refl': ...
def memset(ptr: 'void_p', value: int, size: int) -> None: ...
def memcpy(dst: 'void_p', src: 'void_p', size: int) -> None: ...
class _refl:
pass
@ -29,6 +31,7 @@ class void_p:
def read_bool(self) -> bool: ...
def read_void_p(self) -> 'void_p': ...
def read_bytes(self, size: int) -> bytes: ...
def read_struct(self, type: str) -> 'struct': ...
def write_char(self, value: int) -> None: ...
def write_uchar(self, value: int) -> None: ...
@ -45,6 +48,7 @@ class void_p:
def write_bool(self, value: bool) -> None: ...
def write_void_p(self, value: 'void_p') -> None: ...
def write_bytes(self, value: bytes) -> None: ...
def write_struct(self, value: 'struct') -> None: ...
def get_base_offset(self) -> int: ...
@overload
@ -55,4 +59,4 @@ class void_p:
class struct:
def address(self) -> 'void_p': ...
def copy(self) -> 'struct': ...
def size(self) -> int: ...
def size(self) -> int: ...

83
src/cffi.h
View File

@ -153,31 +153,33 @@ struct C99Struct{
char* p;
int size;
template<typename T>
C99Struct(const T& data){
static_assert(std::is_pod_v<T>);
this->size = sizeof(T);
if(sizeof(T) <= INLINE_SIZE){
memcpy(_inlined, &data, size);
void _init(int size){
this->size = size;
if(size <= INLINE_SIZE){
p = _inlined;
}else{
p = (char*)malloc(size);
memcpy(p, &data, size);
}
}
template<typename T>
C99Struct(const T& data){
static_assert(std::is_pod_v<T>);
static_assert(!std::is_pointer_v<T>);
_init(sizeof(T));
memcpy(p, &data, this->size);
}
C99Struct() { p = _inlined; }
C99Struct(void* p, int size){
_init(size);
memcpy(this->p, p, size);
}
~C99Struct(){ if(p!=_inlined) free(p); }
C99Struct(const C99Struct& other){
this->size = other.size;
if(other.p == other._inlined){
memcpy(_inlined, other._inlined, INLINE_SIZE);
p = _inlined;
}else{
p = (char*)malloc(other.size);
memcpy(p, other.p, other.size);
}
_init(other.size);
memcpy(p, other.p, size);
}
static void _register(VM* vm, PyObject* mod, PyObject* type){
@ -198,21 +200,22 @@ struct C99Struct{
return VAR_T(C99Struct, self);
});
// type = vm->_t(VoidP::_type(vm));
// patch VoidP
type = vm->_t(VoidP::_type(vm));
// vm->bind_method<1>(type, "read_struct", [](VM* vm, ArgsView args){
// VoidP& self = _CAST(VoidP&, args[0]);
// const Str& type = CAST(Str&, args[1]);
// int size = c99_sizeof(vm, type);
// return VAR_T(C99Struct)
// });
vm->bind_method<1>(type, "read_struct", [](VM* vm, ArgsView args){
VoidP& self = _CAST(VoidP&, args[0]);
const Str& type = CAST(Str&, args[1]);
int size = c99_sizeof(vm, type);
return VAR_T(C99Struct, self.ptr, size);
});
// vm->bind_method<1>(type, "write_struct", [](VM* vm, ArgsView args){
// VoidP& self = _CAST(VoidP&, args[0]);
// C99Struct& other = CAST(C99Struct&, args[1]);
// memcpy(self.ptr, other.p, other.size);
// return vm->None;
// });
vm->bind_method<1>(type, "write_struct", [](VM* vm, ArgsView args){
VoidP& self = _CAST(VoidP&, args[0]);
C99Struct& other = CAST(C99Struct&, args[1]);
memcpy(self.ptr, other.p, other.size);
return vm->None;
});
}
};
@ -259,14 +262,8 @@ inline static int c99_sizeof(VM* vm, const Str& type){
struct C99ReflType{
PY_CLASS(C99ReflType, c, "_refl")
Str repr;
static void _register(VM* vm, PyObject* mod, PyObject* type){
vm->bind_constructor<-1>(type, CPP_NOT_IMPLEMENTED());
vm->bind_method<0>(type, "__repr__", [](VM* vm, ArgsView args){
C99ReflType& self = _CAST(C99ReflType&, args[0]);
return VAR(self.repr);
});
}
};
@ -376,10 +373,28 @@ inline void add_module_c(VM* vm){
const ReflType& rt = it->second;
PyObject* obj = VAR_T(C99ReflType);
obj->enable_instance_dict();
obj->attr().set("__name__", VAR(rt.name));
obj->attr().set("__size__", VAR(rt.size));
for(auto [k,v]: rt.fields) obj->attr().set(StrName::get(k), VAR(v));
return obj;
});
vm->bind_func<3>(mod, "memset", [](VM* vm, ArgsView args){
void* p = CAST(void*, args[0]);
i64 c = CAST(i64, args[1]);
i64 size = CAST(i64, args[2]);
memset(p, c, size);
return vm->None;
});
vm->bind_func<3>(mod, "memcpy", [](VM* vm, ArgsView args){
void* dst = CAST(void*, args[0]);
void* src = CAST(void*, args[1]);
i64 size = CAST(i64, args[2]);
memcpy(dst, src, size);
return vm->None;
});
VoidP::register_class(vm, mod);
C99Struct::register_class(vm, mod);
C99ReflType::register_class(vm, mod);