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 2024-12-14 16:06:58 +08:00
parent 4583b57d12
commit d0546c16da
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/modules/pickle.c
View File

@ -33,7 +33,7 @@ typedef struct {
} PickleObject; } PickleObject;
typedef struct { typedef struct {
uint32_t memo_length; uint16_t memo_length;
} PickleObjectHeader; } PickleObjectHeader;
static void PickleObject__ctor(PickleObject* self) { static void PickleObject__ctor(PickleObject* self) {
@ -46,13 +46,15 @@ static void PickleObject__dtor(PickleObject* self) {
c11_vector__dtor(&self->codes); c11_vector__dtor(&self->codes);
} }
static void PickleObject__py_submit(PickleObject* self, py_OutRef out) { static bool PickleObject__py_submit(PickleObject* self, py_OutRef out) {
unsigned char* data = self->codes.data; unsigned char* data = self->codes.data;
PickleObjectHeader* p = PickleObjectHeader* p =
(PickleObjectHeader*)py_newbytes(out, sizeof(PickleObjectHeader) + self->codes.length); (PickleObjectHeader*)py_newbytes(out, sizeof(PickleObjectHeader) + self->codes.length);
p->memo_length = self->memo.length; if(self->memo.length >= UINT16_MAX) c11__abort("PickleObject__py_submit(): memo overflow");
p->memo_length = (uint16_t)self->memo.length;
memcpy(p + 1, data, self->codes.length); memcpy(p + 1, data, self->codes.length);
PickleObject__dtor(self); PickleObject__dtor(self);
return true;
} }
static void PickleObject__write_bytes(PickleObject* buf, const void* data, int size) { static void PickleObject__write_bytes(PickleObject* buf, const void* data, int size) {
@ -318,8 +320,7 @@ bool py_pickle_dumps(py_Ref val) {
return false; return false;
} }
pkl__emit_op(&buf, PKL_EOF); pkl__emit_op(&buf, PKL_EOF);
PickleObject__py_submit(&buf, py_retval()); return PickleObject__py_submit(&buf, py_retval());
return true;
} }
bool py_pickle_loads(const unsigned char* data, int size) { bool py_pickle_loads(const unsigned char* data, int size) {