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

fix a bug

Browse Source
This commit is contained in:
blueloveTH 2024-09-02 23:17:49 +08:00
parent 8277895502
commit 62c12981df
3 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
include/pocketpy/objects/codeobject.h
View File

@ -120,15 +120,16 @@ void FuncDecl__add_arg(FuncDecl* self, py_Name name);
void FuncDecl__add_kwarg(FuncDecl* self, py_Name name, const py_TValue* value);
void FuncDecl__add_starred_arg(FuncDecl* self, py_Name name);
void FuncDecl__add_starred_kwarg(FuncDecl* self, py_Name name);
void FuncDecl__gc_mark(const FuncDecl* self);
// runtime function
typedef struct Function {
FuncDecl_ decl;
py_TValue module; // weak ref
PyObject* clazz; // weak ref
NameDict* closure; // strong ref
NameDict* closure; // strong ref
py_CFunction cfunc; // wrapped C function
} Function;
void Function__ctor(Function* self, FuncDecl_ decl, py_TValue* module);
void Function__dtor(Function* self);
void Function__dtor(Function* self);

15
src/interpreter/vm.c
View File

@ -602,6 +602,14 @@ static void mark_object(PyObject* obj) {
if(ti->gc_mark) ti->gc_mark(PyObject__userdata(obj));
}
void FuncDecl__gc_mark(const FuncDecl* self) {
CodeObject__gc_mark(&self->code);
for(int j = 0; j < self->kwargs.length; j++) {
FuncDeclKwArg* kw = c11__at(FuncDeclKwArg, &self->kwargs, j);
pk__mark_value(&kw->value);
}
}
void CodeObject__gc_mark(const CodeObject* self) {
for(int i = 0; i < self->consts.length; i++) {
py_TValue* p = c11__at(py_TValue, &self->consts, i);
@ -609,12 +617,7 @@ void CodeObject__gc_mark(const CodeObject* self) {
}
for(int i = 0; i < self->func_decls.length; i++) {
FuncDecl_ decl = c11__getitem(FuncDecl_, &self->func_decls, i);
CodeObject__gc_mark(&decl->code);
for(int j = 0; j < decl->kwargs.length; j++) {
FuncDeclKwArg* kw = c11__at(FuncDeclKwArg, &decl->kwargs, j);
pk__mark_value(&kw->value);
}
FuncDecl__gc_mark(decl);
}
}

2
src/public/modules.c
View File

@ -685,7 +685,7 @@ py_TValue pk_builtins__register() {
static void function__gc_mark(void* ud) {
Function* func = ud;
if(func->closure) pk__mark_namedict(func->closure);
CodeObject__gc_mark(&func->decl->code);
FuncDecl__gc_mark(func->decl);
}
static bool function__doc__(int argc, py_Ref argv) {