|
|
@@ -1,5 +1,4 @@
|
|
|
#include "pocketpy/cffi.h"
|
|
|
-#include "pocketpy/str.h"
|
|
|
|
|
|
namespace pkpy{
|
|
|
|
|
|
@@ -10,27 +9,20 @@ namespace pkpy{
|
|
|
return vm->heap.gcnew<VoidP>(cls, reinterpret_cast<void*>(addr));
|
|
|
});
|
|
|
|
|
|
- vm->bind_method<0>(type, "hex", [](VM* vm, ArgsView args){
|
|
|
- VoidP& self = _CAST(VoidP&, args[0]);
|
|
|
- return VAR(self.hex());
|
|
|
+ vm->bind__hash__(PK_OBJ_GET(Type, type), [](VM* vm, PyObject* obj){
|
|
|
+ VoidP& self = _CAST(VoidP&, obj);
|
|
|
+ return reinterpret_cast<i64>(self.ptr);
|
|
|
});
|
|
|
|
|
|
vm->bind__repr__(PK_OBJ_GET(Type, type), [](VM* vm, PyObject* obj){
|
|
|
VoidP& self = _CAST(VoidP&, obj);
|
|
|
- std::stringstream ss;
|
|
|
- ss << "<void* at " << self.hex() << ">";
|
|
|
- return VAR(ss.str());
|
|
|
- });
|
|
|
-
|
|
|
- vm->bind_property(type, "_value: int", [](VM* vm, ArgsView args){
|
|
|
- VoidP& self = _CAST(VoidP&, args[0]);
|
|
|
- return VAR(reinterpret_cast<i64>(self.ptr));
|
|
|
+ return VAR(fmt("<void* at ", self.hex(), ">"));
|
|
|
});
|
|
|
|
|
|
#define BIND_CMP(name, op) \
|
|
|
- vm->bind##name(PK_OBJ_GET(Type, type), [](VM* vm, PyObject* lhs, PyObject* rhs){ \
|
|
|
- if(!is_non_tagged_type(rhs, VoidP::_type(vm))) return vm->NotImplemented; \
|
|
|
- return VAR(_CAST(VoidP&, lhs) op _CAST(VoidP&, rhs)); \
|
|
|
+ vm->bind##name(PK_OBJ_GET(Type, type), [](VM* vm, PyObject* lhs, PyObject* rhs){ \
|
|
|
+ if(!vm->isinstance(rhs, VoidP::_type(vm))) return vm->NotImplemented; \
|
|
|
+ return VAR(_CAST(VoidP&, lhs) op _CAST(VoidP&, rhs)); \
|
|
|
});
|
|
|
|
|
|
BIND_CMP(__eq__, ==)
|
|
|
@@ -40,81 +32,14 @@ namespace pkpy{
|
|
|
BIND_CMP(__ge__, >=)
|
|
|
|
|
|
#undef BIND_CMP
|
|
|
-
|
|
|
- vm->bind__hash__(PK_OBJ_GET(Type, type), [](VM* vm, PyObject* obj){
|
|
|
- VoidP& self = _CAST(VoidP&, obj);
|
|
|
- return reinterpret_cast<i64>(self.ptr);
|
|
|
- });
|
|
|
-
|
|
|
- vm->bind__add__(PK_OBJ_GET(Type, type), [](VM* vm, PyObject* lhs, PyObject* rhs){
|
|
|
- VoidP& self = _CAST(VoidP&, lhs);
|
|
|
- i64 offset = CAST(i64, rhs);
|
|
|
- return VAR_T(VoidP, (char*)self.ptr + offset);
|
|
|
- });
|
|
|
-
|
|
|
- vm->bind__sub__(PK_OBJ_GET(Type, type), [](VM* vm, PyObject* lhs, PyObject* rhs){
|
|
|
- VoidP& self = _CAST(VoidP&, lhs);
|
|
|
- i64 offset = CAST(i64, rhs);
|
|
|
- return VAR_T(VoidP, (char*)self.ptr - offset);
|
|
|
- });
|
|
|
-
|
|
|
-#define BIND_SETGET(T, name) \
|
|
|
- vm->bind_method<0>(type, "read_" name, [](VM* vm, ArgsView args){ \
|
|
|
- VoidP& self = _CAST(VoidP&, args[0]); \
|
|
|
- return VAR(*(T*)self.ptr); \
|
|
|
- }); \
|
|
|
- vm->bind_method<1>(type, "write_" name, [](VM* vm, ArgsView args){ \
|
|
|
- VoidP& self = _CAST(VoidP&, args[0]); \
|
|
|
- *(T*)self.ptr = CAST(T, args[1]); \
|
|
|
- return vm->None; \
|
|
|
- });
|
|
|
-
|
|
|
- BIND_SETGET(char, "char")
|
|
|
- BIND_SETGET(unsigned char, "uchar")
|
|
|
- BIND_SETGET(short, "short")
|
|
|
- BIND_SETGET(unsigned short, "ushort")
|
|
|
- BIND_SETGET(int, "int")
|
|
|
- BIND_SETGET(unsigned int, "uint")
|
|
|
- BIND_SETGET(long, "long")
|
|
|
- BIND_SETGET(unsigned long, "ulong")
|
|
|
- BIND_SETGET(long long, "longlong")
|
|
|
- BIND_SETGET(unsigned long long, "ulonglong")
|
|
|
- BIND_SETGET(float, "float")
|
|
|
- BIND_SETGET(double, "double")
|
|
|
- BIND_SETGET(bool, "bool")
|
|
|
- BIND_SETGET(void*, "void_p")
|
|
|
-
|
|
|
- vm->bind_method<1>(type, "read_bytes", [](VM* vm, ArgsView args){
|
|
|
- VoidP& self = _CAST(VoidP&, args[0]);
|
|
|
- i64 size = CAST(i64, args[1]);
|
|
|
- std::vector<char> buffer(size);
|
|
|
- memcpy(buffer.data(), self.ptr, size);
|
|
|
- return VAR(Bytes(std::move(buffer)));
|
|
|
- });
|
|
|
-
|
|
|
- vm->bind_method<1>(type, "write_bytes", [](VM* vm, ArgsView args){
|
|
|
- VoidP& self = _CAST(VoidP&, args[0]);
|
|
|
- Bytes& bytes = CAST(Bytes&, args[1]);
|
|
|
- memcpy(self.ptr, bytes.data(), bytes.size());
|
|
|
- return vm->None;
|
|
|
- });
|
|
|
-
|
|
|
-#undef BIND_SETGET
|
|
|
}
|
|
|
|
|
|
+
|
|
|
void C99Struct::_register(VM* vm, PyObject* mod, PyObject* type){
|
|
|
vm->bind_constructor<2>(type, [](VM* vm, ArgsView args){
|
|
|
Type cls = PK_OBJ_GET(Type, args[0]);
|
|
|
- if(is_int(args[1])){
|
|
|
- int size = _CAST(int, args[1]);
|
|
|
- return vm->heap.gcnew<C99Struct>(cls, size);
|
|
|
- }
|
|
|
- if(is_non_tagged_type(args[1], vm->tp_bytes)){
|
|
|
- const Bytes& b = _CAST(Bytes&, args[1]);
|
|
|
- return vm->heap.gcnew<C99Struct>(cls, (void*)b.data(), b.size());
|
|
|
- }
|
|
|
- vm->TypeError("expected int or bytes");
|
|
|
- return vm->None;
|
|
|
+ int size = CAST(int, args[1]);
|
|
|
+ return vm->heap.gcnew<C99Struct>(cls, size);
|
|
|
});
|
|
|
|
|
|
vm->bind__repr__(PK_OBJ_GET(Type, type), [](VM* vm, PyObject* obj){
|
|
|
@@ -161,7 +86,6 @@ namespace pkpy{
|
|
|
*(T*)ptr = CAST(T, args[1]); \
|
|
|
return vm->None; \
|
|
|
});
|
|
|
-
|
|
|
BIND_SETGET(char, "char")
|
|
|
BIND_SETGET(unsigned char, "uchar")
|
|
|
BIND_SETGET(short, "short")
|
|
|
@@ -177,22 +101,6 @@ namespace pkpy{
|
|
|
BIND_SETGET(bool, "bool")
|
|
|
BIND_SETGET(void*, "void_p")
|
|
|
#undef BIND_SETGET
|
|
|
-
|
|
|
- // 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]);
|
|
|
- int size = CAST(int, args[1]);
|
|
|
- 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;
|
|
|
- });
|
|
|
}
|
|
|
|
|
|
void add_module_c(VM* vm){
|
|
|
@@ -225,28 +133,81 @@ void add_module_c(VM* vm){
|
|
|
});
|
|
|
#endif
|
|
|
|
|
|
- vm->bind_func<1>(mod, "sizeof", [](VM* vm, ArgsView args){
|
|
|
- const Str& type = CAST(Str&, args[0]);
|
|
|
- auto it = _refl_types.find(type.sv());
|
|
|
- if(it != _refl_types.end()) return VAR(it->second.size);
|
|
|
- vm->ValueError("not a valid c99 type");
|
|
|
- return vm->None;
|
|
|
- });
|
|
|
-
|
|
|
VoidP::register_class(vm, mod);
|
|
|
C99Struct::register_class(vm, mod);
|
|
|
mod->attr().set("NULL", VAR_T(VoidP, nullptr));
|
|
|
|
|
|
- add_refl_type("void_p", sizeof(void*));
|
|
|
- PyObject* void_p_t = mod->attr("void_p");
|
|
|
+ vm->bind(mod, "p_cast(ptr: 'void_p', cls: type[T]) -> T", [](VM* vm, ArgsView args){
|
|
|
+ VoidP& ptr = CAST(VoidP&, args[0]);
|
|
|
+ vm->check_non_tagged_type(args[1], vm->tp_type);
|
|
|
+ Type cls = PK_OBJ_GET(Type, args[1]);
|
|
|
+ if(!vm->issubclass(cls, VoidP::_type(vm))){
|
|
|
+ vm->ValueError("expected a subclass of void_p");
|
|
|
+ }
|
|
|
+ return vm->heap.gcnew<VoidP>(cls, ptr.ptr);
|
|
|
+ });
|
|
|
+
|
|
|
+ vm->bind(mod, "p_value(ptr: 'void_p') -> int", [](VM* vm, ArgsView args){
|
|
|
+ VoidP& ptr = CAST(VoidP&, args[0]);
|
|
|
+ return VAR(reinterpret_cast<i64>(ptr.ptr));
|
|
|
+ });
|
|
|
+
|
|
|
+ vm->bind(mod, "pp_deref(ptr: Tp) -> Tp", [](VM* vm, ArgsView args){
|
|
|
+ VoidP& ptr = CAST(VoidP&, args[0]);
|
|
|
+ void* value = *reinterpret_cast<void**>(ptr.ptr);
|
|
|
+ return vm->heap.gcnew<VoidP>(args[0]->type, value);
|
|
|
+ });
|
|
|
|
|
|
-#define BIND_PRIMITIVE(T, name) \
|
|
|
- vm->bind_func<1>(mod, name "_", [](VM* vm, ArgsView args){ \
|
|
|
- T val = CAST(T, args[0]); \
|
|
|
- return VAR_T(C99Struct, &val, sizeof(T)); \
|
|
|
- }); \
|
|
|
- add_refl_type(name, sizeof(T)); \
|
|
|
- mod->attr().set(name "_p", void_p_t); \
|
|
|
+ PyObject* type;
|
|
|
+ Type type_t;
|
|
|
+
|
|
|
+#define BIND_PRIMITIVE(T, CNAME) \
|
|
|
+ vm->bind_func<1>(mod, CNAME "_", [](VM* vm, ArgsView args){ \
|
|
|
+ T val = CAST(T, args[0]); \
|
|
|
+ return VAR_T(C99Struct, &val, sizeof(T)); \
|
|
|
+ }); \
|
|
|
+ type = vm->new_type_object(mod, CNAME "_p", VoidP::_type(vm)); \
|
|
|
+ type_t = PK_OBJ_GET(Type, type); \
|
|
|
+ vm->bind_method<0>(type, "read", [](VM* vm, ArgsView args){ \
|
|
|
+ VoidP& voidp = PK_OBJ_GET(VoidP, args[0]); \
|
|
|
+ T* target = (T*)voidp.ptr; \
|
|
|
+ return VAR(*target); \
|
|
|
+ }); \
|
|
|
+ vm->bind_method<1>(type, "write", [](VM* vm, ArgsView args){ \
|
|
|
+ VoidP& voidp = PK_OBJ_GET(VoidP, args[0]); \
|
|
|
+ T val = CAST(T, args[1]); \
|
|
|
+ T* target = (T*)voidp.ptr; \
|
|
|
+ *target = val; \
|
|
|
+ return vm->None; \
|
|
|
+ }); \
|
|
|
+ vm->bind__getitem__(type_t, [](VM* vm, PyObject* obj, PyObject* index){ \
|
|
|
+ VoidP& voidp = PK_OBJ_GET(VoidP, obj); \
|
|
|
+ i64 offset = CAST(i64, index); \
|
|
|
+ T* target = (T*)voidp.ptr; \
|
|
|
+ return VAR(target[offset]); \
|
|
|
+ }); \
|
|
|
+ vm->bind__setitem__(type_t, [](VM* vm, PyObject* obj, PyObject* index, PyObject* value){ \
|
|
|
+ VoidP& voidp = PK_OBJ_GET(VoidP, obj); \
|
|
|
+ i64 offset = CAST(i64, index); \
|
|
|
+ T* target = (T*)voidp.ptr; \
|
|
|
+ target[offset] = CAST(T, value); \
|
|
|
+ }); \
|
|
|
+ vm->bind__add__(type_t, [](VM* vm, PyObject* lhs, PyObject* rhs){ \
|
|
|
+ VoidP& voidp = PK_OBJ_GET(VoidP, lhs); \
|
|
|
+ i64 offset = CAST(i64, rhs); \
|
|
|
+ T* target = (T*)voidp.ptr; \
|
|
|
+ return vm->heap.gcnew<VoidP>(lhs->type, target + offset); \
|
|
|
+ }); \
|
|
|
+ vm->bind__sub__(type_t, [](VM* vm, PyObject* lhs, PyObject* rhs){ \
|
|
|
+ VoidP& voidp = PK_OBJ_GET(VoidP, lhs); \
|
|
|
+ i64 offset = CAST(i64, rhs); \
|
|
|
+ T* target = (T*)voidp.ptr; \
|
|
|
+ return vm->heap.gcnew<VoidP>(lhs->type, target - offset); \
|
|
|
+ }); \
|
|
|
+ vm->bind__repr__(type_t, [](VM* vm, PyObject* obj){ \
|
|
|
+ VoidP& self = _CAST(VoidP&, obj); \
|
|
|
+ return VAR(fmt("<", CNAME, "* at ", self.hex(), ">")); \
|
|
|
+ }); \
|
|
|
|
|
|
BIND_PRIMITIVE(char, "char")
|
|
|
BIND_PRIMITIVE(unsigned char, "uchar")
|
|
|
@@ -262,9 +223,7 @@ void add_module_c(VM* vm){
|
|
|
BIND_PRIMITIVE(double, "double")
|
|
|
BIND_PRIMITIVE(bool, "bool")
|
|
|
|
|
|
- // add array type
|
|
|
- CodeObject_ code = vm->compile(kPythonLibs["c"], "c.py", EXEC_MODE);
|
|
|
- vm->_exec(code, mod);
|
|
|
+#undef BIND_PRIMITIVE
|
|
|
}
|
|
|
|
|
|
} // namespace pkpy
|