vm.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715
  1. #include "pocketpy/interpreter/vm.h"
  2. #include "pocketpy/common/memorypool.h"
  3. #include "pocketpy/common/sstream.h"
  4. #include "pocketpy/common/utils.h"
  5. #include "pocketpy/interpreter/generator.h"
  6. #include "pocketpy/interpreter/modules.h"
  7. #include "pocketpy/interpreter/typeinfo.h"
  8. #include "pocketpy/objects/base.h"
  9. #include "pocketpy/common/_generated.h"
  10. #include "pocketpy/pocketpy.h"
  11. #include <stdbool.h>
  12. static char* pk_default_importfile(const char* path) {
  13. #if PK_ENABLE_OS
  14. FILE* f = fopen(path, "rb");
  15. if(f == NULL) return NULL;
  16. fseek(f, 0, SEEK_END);
  17. long size = ftell(f);
  18. fseek(f, 0, SEEK_SET);
  19. char* buffer = malloc(size + 1);
  20. size = fread(buffer, 1, size, f);
  21. buffer[size] = 0;
  22. fclose(f);
  23. return buffer;
  24. #else
  25. return NULL;
  26. #endif
  27. }
  28. static void pk_default_print(const char* data) { printf("%s", data); }
  29. static void py_TypeInfo__ctor(py_TypeInfo* self,
  30. py_Name name,
  31. py_Type index,
  32. py_Type base,
  33. py_TypeInfo* base_ti,
  34. py_TValue module) {
  35. memset(self, 0, sizeof(py_TypeInfo));
  36. self->name = name;
  37. self->base = base;
  38. self->base_ti = base_ti;
  39. // create type object with __dict__
  40. ManagedHeap* heap = &pk_current_vm->heap;
  41. PyObject* typeobj = ManagedHeap__new(heap, tp_type, -1, sizeof(py_Type));
  42. *(py_Type*)PyObject__userdata(typeobj) = index;
  43. self->self = (py_TValue){
  44. .type = typeobj->type,
  45. .is_ptr = true,
  46. ._obj = typeobj,
  47. };
  48. self->module = module;
  49. self->annotations = *py_NIL;
  50. }
  51. void VM__ctor(VM* self) {
  52. self->top_frame = NULL;
  53. ModuleDict__ctor(&self->modules, NULL, *py_NIL);
  54. TypeList__ctor(&self->types);
  55. self->builtins = *py_NIL;
  56. self->main = *py_NIL;
  57. self->callbacks.importfile = pk_default_importfile;
  58. self->callbacks.print = pk_default_print;
  59. self->last_retval = *py_NIL;
  60. self->curr_exception = *py_NIL;
  61. self->is_curr_exc_handled = false;
  62. self->ctx = NULL;
  63. self->__curr_class = NULL;
  64. self->__curr_function = NULL;
  65. ManagedHeap__ctor(&self->heap, self);
  66. ValueStack__ctor(&self->stack);
  67. /* Init Builtin Types */
  68. // 0: unused
  69. void* placeholder = TypeList__emplace(&self->types);
  70. memset(placeholder, 0, sizeof(py_TypeInfo));
  71. #define validate(t, expr) \
  72. if(t != (expr)) abort()
  73. validate(tp_object, pk_newtype("object", 0, NULL, NULL, true, false));
  74. validate(tp_type, pk_newtype("type", 1, NULL, NULL, false, true));
  75. pk_object__register();
  76. validate(tp_int, pk_newtype("int", tp_object, NULL, NULL, false, true));
  77. validate(tp_float, pk_newtype("float", tp_object, NULL, NULL, false, true));
  78. validate(tp_bool, pk_newtype("bool", tp_object, NULL, NULL, false, true));
  79. pk_number__register();
  80. validate(tp_str, pk_str__register());
  81. validate(tp_str_iterator, pk_str_iterator__register());
  82. validate(tp_list, pk_list__register());
  83. validate(tp_tuple, pk_tuple__register());
  84. validate(tp_array_iterator, pk_array_iterator__register());
  85. validate(tp_slice, pk_slice__register());
  86. validate(tp_range, pk_range__register());
  87. validate(tp_range_iterator, pk_range_iterator__register());
  88. validate(tp_module, pk_newtype("module", tp_object, NULL, NULL, false, true));
  89. validate(tp_function, pk_function__register());
  90. validate(tp_nativefunc, pk_nativefunc__register());
  91. validate(tp_boundmethod, pk_boundmethod__register());
  92. validate(tp_super, pk_super__register());
  93. validate(tp_BaseException, pk_BaseException__register());
  94. validate(tp_Exception, pk_Exception__register());
  95. validate(tp_bytes, pk_bytes__register());
  96. validate(tp_namedict, pk_namedict__register());
  97. validate(tp_locals, pk_locals__register());
  98. validate(tp_code, pk_code__register());
  99. validate(tp_dict, pk_dict__register());
  100. validate(tp_dict_items, pk_dict_items__register());
  101. validate(tp_property, pk_property__register());
  102. validate(tp_star_wrapper, pk_newtype("star_wrapper", tp_object, NULL, NULL, false, true));
  103. validate(tp_staticmethod, pk_staticmethod__register());
  104. validate(tp_classmethod, pk_classmethod__register());
  105. validate(tp_NoneType, pk_newtype("NoneType", tp_object, NULL, NULL, false, true));
  106. validate(tp_NotImplementedType,
  107. pk_newtype("NotImplementedType", tp_object, NULL, NULL, false, true));
  108. validate(tp_ellipsis, pk_newtype("ellipsis", tp_object, NULL, NULL, false, true));
  109. validate(tp_generator, pk_generator__register());
  110. self->builtins = pk_builtins__register();
  111. // inject some builtin expections
  112. #define INJECT_BUILTIN_EXC(name, TBase) \
  113. do { \
  114. py_Type type = pk_newtype(#name, TBase, &self->builtins, NULL, false, true); \
  115. py_setdict(&self->builtins, py_name(#name), py_tpobject(type)); \
  116. validate(tp_##name, type); \
  117. } while(0)
  118. INJECT_BUILTIN_EXC(SystemExit, tp_BaseException);
  119. INJECT_BUILTIN_EXC(KeyboardInterrupt, tp_BaseException);
  120. INJECT_BUILTIN_EXC(StopIteration, tp_Exception);
  121. INJECT_BUILTIN_EXC(SyntaxError, tp_Exception);
  122. INJECT_BUILTIN_EXC(StackOverflowError, tp_Exception);
  123. INJECT_BUILTIN_EXC(IOError, tp_Exception);
  124. INJECT_BUILTIN_EXC(OSError, tp_Exception);
  125. INJECT_BUILTIN_EXC(NotImplementedError, tp_Exception);
  126. INJECT_BUILTIN_EXC(TypeError, tp_Exception);
  127. INJECT_BUILTIN_EXC(IndexError, tp_Exception);
  128. INJECT_BUILTIN_EXC(ValueError, tp_Exception);
  129. INJECT_BUILTIN_EXC(RuntimeError, tp_Exception);
  130. INJECT_BUILTIN_EXC(ZeroDivisionError, tp_Exception);
  131. INJECT_BUILTIN_EXC(NameError, tp_Exception);
  132. INJECT_BUILTIN_EXC(UnboundLocalError, tp_Exception);
  133. INJECT_BUILTIN_EXC(AttributeError, tp_Exception);
  134. INJECT_BUILTIN_EXC(ImportError, tp_Exception);
  135. INJECT_BUILTIN_EXC(AssertionError, tp_Exception);
  136. INJECT_BUILTIN_EXC(KeyError, tp_Exception);
  137. #undef INJECT_BUILTIN_EXC
  138. #undef validate
  139. /* Setup Public Builtin Types */
  140. py_Type public_types[] = {
  141. tp_object,
  142. tp_type,
  143. tp_int,
  144. tp_float,
  145. tp_bool,
  146. tp_str,
  147. tp_list,
  148. tp_tuple,
  149. tp_slice,
  150. tp_range,
  151. tp_bytes,
  152. tp_dict,
  153. tp_property,
  154. tp_staticmethod,
  155. tp_classmethod,
  156. tp_super,
  157. tp_BaseException,
  158. tp_Exception,
  159. };
  160. for(int i = 0; i < c11__count_array(public_types); i++) {
  161. py_TypeInfo* ti = pk__type_info(public_types[i]);
  162. py_setdict(&self->builtins, ti->name, &ti->self);
  163. }
  164. py_newnotimplemented(py_emplacedict(&self->builtins, py_name("NotImplemented")));
  165. pk__add_module_linalg();
  166. // add modules
  167. pk__add_module_pkpy();
  168. pk__add_module_os();
  169. pk__add_module_sys();
  170. pk__add_module_math();
  171. pk__add_module_dis();
  172. pk__add_module_random();
  173. pk__add_module_json();
  174. pk__add_module_gc();
  175. pk__add_module_time();
  176. pk__add_module_easing();
  177. pk__add_module_traceback();
  178. pk__add_module_enum();
  179. // add python builtins
  180. do {
  181. bool ok;
  182. ok = py_exec(kPythonLibs_builtins, "<builtins>", EXEC_MODE, &self->builtins);
  183. if(!ok) goto __ABORT;
  184. break;
  185. __ABORT:
  186. py_printexc();
  187. c11__abort("failed to load python builtins!");
  188. } while(0);
  189. self->main = *py_newmodule("__main__");
  190. }
  191. void VM__dtor(VM* self) {
  192. // destroy all objects
  193. ManagedHeap__dtor(&self->heap);
  194. // clear frames
  195. while(self->top_frame)
  196. VM__pop_frame(self);
  197. ModuleDict__dtor(&self->modules);
  198. TypeList__dtor(&self->types);
  199. ValueStack__clear(&self->stack);
  200. }
  201. void VM__push_frame(VM* self, Frame* frame) {
  202. frame->f_back = self->top_frame;
  203. self->top_frame = frame;
  204. }
  205. void VM__pop_frame(VM* self) {
  206. assert(self->top_frame);
  207. Frame* frame = self->top_frame;
  208. // reset stack pointer
  209. self->stack.sp = frame->p0;
  210. // pop frame and delete
  211. self->top_frame = frame->f_back;
  212. Frame__delete(frame);
  213. }
  214. static void _clip_int(int* value, int min, int max) {
  215. if(*value < min) *value = min;
  216. if(*value > max) *value = max;
  217. }
  218. bool pk__parse_int_slice(py_Ref slice, int length, int* start, int* stop, int* step) {
  219. py_Ref s_start = py_getslot(slice, 0);
  220. py_Ref s_stop = py_getslot(slice, 1);
  221. py_Ref s_step = py_getslot(slice, 2);
  222. if(py_isnone(s_step))
  223. *step = 1;
  224. else {
  225. if(!py_checkint(s_step)) return false;
  226. *step = py_toint(s_step);
  227. }
  228. if(*step == 0) return ValueError("slice step cannot be zero");
  229. if(*step > 0) {
  230. if(py_isnone(s_start))
  231. *start = 0;
  232. else {
  233. if(!py_checkint(s_start)) return false;
  234. *start = py_toint(s_start);
  235. if(*start < 0) *start += length;
  236. _clip_int(start, 0, length);
  237. }
  238. if(py_isnone(s_stop))
  239. *stop = length;
  240. else {
  241. if(!py_checkint(s_stop)) return false;
  242. *stop = py_toint(s_stop);
  243. if(*stop < 0) *stop += length;
  244. _clip_int(stop, 0, length);
  245. }
  246. } else {
  247. if(py_isnone(s_start))
  248. *start = length - 1;
  249. else {
  250. if(!py_checkint(s_start)) return false;
  251. *start = py_toint(s_start);
  252. if(*start < 0) *start += length;
  253. _clip_int(start, -1, length - 1);
  254. }
  255. if(py_isnone(s_stop))
  256. *stop = -1;
  257. else {
  258. if(!py_checkint(s_stop)) return false;
  259. *stop = py_toint(s_stop);
  260. if(*stop < 0) *stop += length;
  261. _clip_int(stop, -1, length - 1);
  262. }
  263. }
  264. return true;
  265. }
  266. bool pk__normalize_index(int* index, int length) {
  267. if(*index < 0) *index += length;
  268. if(*index < 0 || *index >= length) { return IndexError("%d not in [0, %d)", *index, length); }
  269. return true;
  270. }
  271. py_Type pk_newtype(const char* name,
  272. py_Type base,
  273. const py_GlobalRef module,
  274. void (*dtor)(void*),
  275. bool is_python,
  276. bool is_sealed) {
  277. py_Type index = pk_current_vm->types.length;
  278. py_TypeInfo* ti = TypeList__emplace(&pk_current_vm->types);
  279. py_TypeInfo* base_ti = base ? pk__type_info(base) : NULL;
  280. if(base_ti && base_ti->is_sealed) {
  281. c11__abort("type '%s' is not an acceptable base type", py_name2str(base_ti->name));
  282. }
  283. py_TypeInfo__ctor(ti, py_name(name), index, base, base_ti, module ? *module : *py_NIL);
  284. if(!dtor && base) dtor = base_ti->dtor;
  285. ti->dtor = dtor;
  286. ti->is_python = is_python;
  287. ti->is_sealed = is_sealed;
  288. return index;
  289. }
  290. py_Type py_newtype(const char* name, py_Type base, const py_GlobalRef module, void (*dtor)(void*)) {
  291. py_Type type = pk_newtype(name, base, module, dtor, false, false);
  292. if(module) py_setdict(module, py_name(name), py_tpobject(type));
  293. return type;
  294. }
  295. static bool
  296. prepare_py_call(py_TValue* buffer, py_Ref argv, py_Ref p1, int kwargc, const FuncDecl* decl) {
  297. const CodeObject* co = &decl->code;
  298. int decl_argc = decl->args.length;
  299. if(p1 - argv < decl_argc) {
  300. return TypeError("%s() takes %d positional arguments but %d were given",
  301. co->name->data,
  302. decl_argc,
  303. (int)(p1 - argv));
  304. }
  305. py_TValue* t = argv;
  306. // prepare args
  307. memset(buffer, 0, co->nlocals * sizeof(py_TValue));
  308. c11__foreach(int, &decl->args, index) buffer[*index] = *t++;
  309. // prepare kwdefaults
  310. c11__foreach(FuncDeclKwArg, &decl->kwargs, kv) buffer[kv->index] = kv->value;
  311. // handle *args
  312. if(decl->starred_arg != -1) {
  313. int exceed_argc = p1 - t;
  314. py_Ref vargs = &buffer[decl->starred_arg];
  315. py_newtuple(vargs, exceed_argc);
  316. for(int j = 0; j < exceed_argc; j++) {
  317. py_tuple_setitem(vargs, j, t++);
  318. }
  319. } else {
  320. // kwdefaults override
  321. // def f(a, b, c=None)
  322. // f(1, 2, 3) -> c=3
  323. c11__foreach(FuncDeclKwArg, &decl->kwargs, kv) {
  324. if(t >= p1) break;
  325. buffer[kv->index] = *t++;
  326. }
  327. // not able to consume all args
  328. if(t < p1) return TypeError("too many arguments (%s)", co->name->data);
  329. }
  330. if(decl->starred_kwarg != -1) py_newdict(&buffer[decl->starred_kwarg]);
  331. for(int j = 0; j < kwargc; j++) {
  332. py_Name key = py_toint(&p1[2 * j]);
  333. int index = c11_smallmap_n2i__get(&decl->kw_to_index, key, -1);
  334. // if key is an explicit key, set as local variable
  335. if(index >= 0) {
  336. buffer[index] = p1[2 * j + 1];
  337. } else {
  338. // otherwise, set as **kwargs if possible
  339. if(decl->starred_kwarg == -1) {
  340. return TypeError("'%n' is an invalid keyword argument for %s()",
  341. key,
  342. co->name->data);
  343. } else {
  344. // add to **kwargs
  345. bool ok = py_dict_setitem_by_str(&buffer[decl->starred_kwarg],
  346. py_name2str(key),
  347. &p1[2 * j + 1]);
  348. if(!ok) return false;
  349. }
  350. }
  351. }
  352. return true;
  353. }
  354. FrameResult VM__vectorcall(VM* self, uint16_t argc, uint16_t kwargc, bool opcall) {
  355. pk_print_stack(self, self->top_frame, (Bytecode){0});
  356. py_Ref p1 = self->stack.sp - kwargc * 2;
  357. py_Ref p0 = p1 - argc - 2;
  358. // [callable, <self>, args..., kwargs...]
  359. // ^p0 ^p1 ^_sp
  360. // handle boundmethod, do a patch
  361. if(p0->type == tp_boundmethod) {
  362. assert(py_isnil(p0 + 1)); // self must be NULL
  363. py_TValue* slots = PyObject__slots(p0->_obj);
  364. p0[0] = slots[1]; // callable
  365. p0[1] = slots[0]; // self
  366. // [unbound, self, args..., kwargs...]
  367. }
  368. py_Ref argv = p0 + 1 + (int)py_isnil(p0 + 1);
  369. if(p0->type == tp_function) {
  370. // check stack overflow
  371. if(self->stack.sp > self->stack.end) {
  372. py_exception(tp_StackOverflowError, "");
  373. return RES_ERROR;
  374. }
  375. Function* fn = py_touserdata(p0);
  376. const CodeObject* co = &fn->decl->code;
  377. switch(fn->decl->type) {
  378. case FuncType_NORMAL: {
  379. bool ok = prepare_py_call(self->__vectorcall_buffer, argv, p1, kwargc, fn->decl);
  380. if(!ok) return RES_ERROR;
  381. // copy buffer back to stack
  382. self->stack.sp = argv + co->nlocals;
  383. memcpy(argv, self->__vectorcall_buffer, co->nlocals * sizeof(py_TValue));
  384. // submit the call
  385. if(!fn->cfunc) {
  386. // python function
  387. VM__push_frame(self, Frame__new(co, &fn->module, p0, argv, true));
  388. return opcall ? RES_CALL : VM__run_top_frame(self);
  389. } else {
  390. // decl-based binding
  391. self->__curr_function = p0;
  392. bool ok = py_callcfunc(fn->cfunc, co->nlocals, argv);
  393. self->stack.sp = p0;
  394. self->__curr_function = NULL;
  395. return ok ? RES_RETURN : RES_ERROR;
  396. }
  397. }
  398. case FuncType_SIMPLE:
  399. if(p1 - argv != fn->decl->args.length) {
  400. const char* fmt = "%s() takes %d positional arguments but %d were given";
  401. TypeError(fmt, co->name->data, fn->decl->args.length, (int)(p1 - argv));
  402. return RES_ERROR;
  403. }
  404. if(kwargc) {
  405. TypeError("%s() takes no keyword arguments", co->name->data);
  406. return RES_ERROR;
  407. }
  408. // [callable, <self>, args..., local_vars...]
  409. // ^p0 ^p1 ^_sp
  410. self->stack.sp = argv + co->nlocals;
  411. // initialize local variables to py_NIL
  412. memset(p1, 0, (char*)self->stack.sp - (char*)p1);
  413. // submit the call
  414. if(!fn->cfunc) {
  415. // python function
  416. VM__push_frame(self, Frame__new(co, &fn->module, p0, argv, true));
  417. return opcall ? RES_CALL : VM__run_top_frame(self);
  418. } else {
  419. // decl-based binding
  420. self->__curr_function = p0;
  421. bool ok = py_callcfunc(fn->cfunc, co->nlocals, argv);
  422. self->stack.sp = p0;
  423. self->__curr_function = NULL;
  424. return ok ? RES_RETURN : RES_ERROR;
  425. }
  426. case FuncType_GENERATOR: {
  427. bool ok = prepare_py_call(self->__vectorcall_buffer, argv, p1, kwargc, fn->decl);
  428. if(!ok) return RES_ERROR;
  429. Frame* frame = Frame__new(co, &fn->module, p0, argv, false);
  430. pk_newgenerator(py_retval(), frame, self->__vectorcall_buffer, co->nlocals);
  431. self->stack.sp = p0;
  432. return RES_RETURN;
  433. }
  434. default: c11__unreachedable();
  435. };
  436. c11__unreachedable();
  437. /*****************_py_call*****************/
  438. }
  439. if(p0->type == tp_nativefunc) {
  440. if(kwargc && p0->_cfunc != pk__object_new) {
  441. TypeError("nativefunc does not accept keyword arguments");
  442. return RES_ERROR;
  443. }
  444. bool ok = py_callcfunc(p0->_cfunc, p1 - argv, argv);
  445. self->stack.sp = p0;
  446. return ok ? RES_RETURN : RES_ERROR;
  447. }
  448. if(p0->type == tp_type) {
  449. // [cls, NULL, args..., kwargs...]
  450. py_Ref new_f = py_tpfindmagic(py_totype(p0), __new__);
  451. assert(new_f && py_isnil(p0 + 1));
  452. // prepare a copy of args and kwargs
  453. int span = self->stack.sp - argv;
  454. *self->stack.sp++ = *new_f; // push __new__
  455. *self->stack.sp++ = *p0; // push cls
  456. memcpy(self->stack.sp, argv, span * sizeof(py_TValue));
  457. self->stack.sp += span;
  458. // [new_f, cls, args..., kwargs...]
  459. if(VM__vectorcall(self, argc, kwargc, false) == RES_ERROR) return RES_ERROR;
  460. // by recursively using vectorcall, args and kwargs are consumed
  461. // try __init__
  462. // NOTE: previously we use `get_unbound_method` but here we just use `tpfindmagic`
  463. // >> [cls, NULL, args..., kwargs...]
  464. // >> py_retval() is the new instance
  465. py_Ref init_f = py_tpfindmagic(py_totype(p0), __init__);
  466. if(init_f) {
  467. // do an inplace patch
  468. *p0 = *init_f; // __init__
  469. p0[1] = self->last_retval; // self
  470. // [__init__, self, args..., kwargs...]
  471. if(VM__vectorcall(self, argc, kwargc, false) == RES_ERROR) return RES_ERROR;
  472. *py_retval() = p0[1]; // restore the new instance
  473. }
  474. // reset the stack
  475. self->stack.sp = p0;
  476. return RES_RETURN;
  477. }
  478. // handle `__call__` overload
  479. if(pk_loadmethod(p0, __call__)) {
  480. // [__call__, self, args..., kwargs...]
  481. return VM__vectorcall(self, argc, kwargc, opcall);
  482. }
  483. TypeError("'%t' object is not callable", p0->type);
  484. return RES_ERROR;
  485. }
  486. /****************************************/
  487. void PyObject__delete(PyObject* self) {
  488. py_TypeInfo* ti = pk__type_info(self->type);
  489. if(ti->dtor) ti->dtor(PyObject__userdata(self));
  490. if(self->slots == -1) NameDict__dtor(PyObject__dict(self));
  491. if(self->gc_is_large) {
  492. free(self);
  493. } else {
  494. PoolObject_dealloc(self);
  495. }
  496. }
  497. static void mark_object(PyObject* obj);
  498. void pk__mark_value(py_TValue* val) {
  499. if(val->is_ptr) mark_object(val->_obj);
  500. }
  501. void pk__mark_namedict(NameDict* dict) {
  502. for(int i = 0; i < dict->length; i++) {
  503. NameDict_KV* kv = c11__at(NameDict_KV, dict, i);
  504. pk__mark_value(&kv->value);
  505. }
  506. }
  507. void pk__tp_set_marker(py_Type type, void (*gc_mark)(void*)) {
  508. py_TypeInfo* ti = pk__type_info(type);
  509. assert(ti->gc_mark == NULL);
  510. ti->gc_mark = gc_mark;
  511. }
  512. static void mark_object(PyObject* obj) {
  513. if(obj->gc_marked) return;
  514. obj->gc_marked = true;
  515. if(obj->slots > 0) {
  516. py_TValue* p = PyObject__slots(obj);
  517. for(int i = 0; i < obj->slots; i++)
  518. pk__mark_value(p + i);
  519. } else if(obj->slots == -1) {
  520. NameDict* dict = PyObject__dict(obj);
  521. pk__mark_namedict(dict);
  522. }
  523. py_TypeInfo* ti = pk__type_info(obj->type);
  524. if(ti->gc_mark) ti->gc_mark(PyObject__userdata(obj));
  525. }
  526. void CodeObject__gc_mark(const CodeObject* self) {
  527. c11__foreach(py_TValue, &self->consts, i) { pk__mark_value(i); }
  528. c11__foreach(FuncDecl_, &self->func_decls, i) { CodeObject__gc_mark(&(*i)->code); }
  529. }
  530. void ManagedHeap__mark(ManagedHeap* self) {
  531. VM* vm = self->vm;
  532. // mark heap objects
  533. for(int i = 0; i < self->no_gc.length; i++) {
  534. PyObject* obj = c11__getitem(PyObject*, &self->no_gc, i);
  535. mark_object(obj);
  536. }
  537. // mark value stack
  538. for(py_TValue* p = vm->stack.begin; p != vm->stack.end; p++) {
  539. pk__mark_value(p);
  540. }
  541. // mark types
  542. int types_length = vm->types.length;
  543. // 0-th type is placeholder
  544. for(py_Type i = 1; i < types_length; i++) {
  545. py_TypeInfo* ti = TypeList__get(&vm->types, i);
  546. // mark magic slots
  547. for(int j = 0; j <= __missing__; j++) {
  548. py_TValue* slot = ti->magic + j;
  549. if(py_isnil(slot)) continue;
  550. pk__mark_value(slot);
  551. }
  552. // mark type annotations
  553. pk__mark_value(&ti->annotations);
  554. }
  555. // mark frame
  556. for(Frame* frame = vm->top_frame; frame; frame = frame->f_back) {
  557. Frame__gc_mark(frame);
  558. }
  559. // mark vm's registers
  560. pk__mark_value(&vm->last_retval);
  561. pk__mark_value(&vm->curr_exception);
  562. for(int i = 0; i < c11__count_array(vm->reg); i++) {
  563. pk__mark_value(&vm->reg[i]);
  564. }
  565. }
  566. void pk_print_stack(VM* self, Frame* frame, Bytecode byte) {
  567. return;
  568. if(frame == NULL || py_isnil(&self->main)) return;
  569. py_TValue* sp = self->stack.sp;
  570. c11_sbuf buf;
  571. c11_sbuf__ctor(&buf);
  572. for(py_Ref p = self->stack.begin; p != sp; p++) {
  573. switch(p->type) {
  574. case tp_nil: c11_sbuf__write_cstr(&buf, "nil"); break;
  575. case tp_int: c11_sbuf__write_i64(&buf, p->_i64); break;
  576. case tp_float: c11_sbuf__write_f64(&buf, p->_f64, -1); break;
  577. case tp_bool: c11_sbuf__write_cstr(&buf, p->_bool ? "True" : "False"); break;
  578. case tp_NoneType: c11_sbuf__write_cstr(&buf, "None"); break;
  579. case tp_list: {
  580. pk_sprintf(&buf, "list(%d)", py_list_len(p));
  581. break;
  582. }
  583. case tp_tuple: {
  584. pk_sprintf(&buf, "tuple(%d)", py_tuple_len(p));
  585. break;
  586. }
  587. case tp_function: {
  588. Function* ud = py_touserdata(p);
  589. c11_sbuf__write_cstr(&buf, ud->decl->code.name->data);
  590. c11_sbuf__write_cstr(&buf, "()");
  591. break;
  592. }
  593. case tp_type: {
  594. pk_sprintf(&buf, "<class '%t'>", py_totype(p));
  595. break;
  596. }
  597. case tp_str: {
  598. pk_sprintf(&buf, "%q", py_tosv(p));
  599. break;
  600. }
  601. case tp_module: {
  602. py_Ref path = py_getdict(p, __path__);
  603. pk_sprintf(&buf, "<module '%v'>", py_tosv(path));
  604. break;
  605. }
  606. default: {
  607. pk_sprintf(&buf, "(%t)", p->type);
  608. break;
  609. }
  610. }
  611. if(p != &sp[-1]) c11_sbuf__write_cstr(&buf, ", ");
  612. }
  613. c11_string* stack_str = c11_sbuf__submit(&buf);
  614. printf("%s:%-3d: %-25s %-6d [%s]\n",
  615. frame->co->src->filename->data,
  616. Frame__lineno(frame),
  617. pk_opname(byte.op),
  618. byte.arg,
  619. stack_str->data);
  620. c11_string__delete(stack_str);
  621. }
  622. bool pk_wrapper__self(int argc, py_Ref argv) {
  623. PY_CHECK_ARGC(1);
  624. py_assign(py_retval(), argv);
  625. return true;
  626. }
  627. bool pk_wrapper__NotImplementedError(int argc, py_Ref argv) {
  628. return py_exception(tp_NotImplementedError, "");
  629. }
  630. py_TypeInfo* pk__type_info(py_Type type) { return TypeList__get(&pk_current_vm->types, type); }