mirror of
https://github.com/zekexiao/pocketlang.git
synced 2025-02-06 04:37:47 +08:00
776ea0ab87
This commit adds the builtin types to the VM's builtin classes buffer (however no methods were added to those classes) and a super class parameter introduced while creating a new class.
1681 lines
47 KiB
C
1681 lines
47 KiB
C
/*
|
|
* Copyright (c) 2020-2022 Thakee Nathees
|
|
* Copyright (c) 2021-2022 Pocketlang Contributors
|
|
* Distributed Under The MIT License
|
|
*/
|
|
|
|
#include "pk_vm.h"
|
|
|
|
#include <math.h>
|
|
#include "pk_core.h"
|
|
#include "pk_utils.h"
|
|
#include "pk_debug.h"
|
|
|
|
/*****************************************************************************/
|
|
/* VM PUBLIC API */
|
|
/*****************************************************************************/
|
|
|
|
// The default allocator that will be used to initialize the vm's configuration
|
|
// if the host doesn't provided any allocators for us.
|
|
static void* defaultRealloc(void* memory, size_t new_size, void* user_data);
|
|
|
|
// Runs the [fiber] if it's at yielded state, this will resume the execution
|
|
// till the next yield or return statement, and return result.
|
|
static PkResult runFiber(PKVM* vm, Fiber* fiber);
|
|
|
|
PkConfiguration pkNewConfiguration(void) {
|
|
PkConfiguration config;
|
|
config.realloc_fn = defaultRealloc;
|
|
|
|
config.error_fn = NULL;
|
|
config.write_fn = NULL;
|
|
config.read_fn = NULL;
|
|
|
|
config.load_script_fn = NULL;
|
|
config.resolve_path_fn = NULL;
|
|
config.user_data = NULL;
|
|
|
|
return config;
|
|
}
|
|
|
|
PkCompileOptions pkNewCompilerOptions(void) {
|
|
PkCompileOptions options;
|
|
options.debug = false;
|
|
options.repl_mode = false;
|
|
return options;
|
|
}
|
|
|
|
PKVM* pkNewVM(PkConfiguration* config) {
|
|
|
|
PkConfiguration default_config = pkNewConfiguration();
|
|
|
|
if (config == NULL) config = &default_config;
|
|
|
|
PKVM* vm = (PKVM*)config->realloc_fn(NULL, sizeof(PKVM), config->user_data);
|
|
memset(vm, 0, sizeof(PKVM));
|
|
|
|
vm->config = *config;
|
|
vm->working_set_count = 0;
|
|
vm->working_set_capacity = MIN_CAPACITY;
|
|
vm->working_set = (Object**)vm->config.realloc_fn(
|
|
NULL, sizeof(Object*) * vm->working_set_capacity, NULL);
|
|
vm->next_gc = INITIAL_GC_SIZE;
|
|
vm->min_heap_size = MIN_HEAP_SIZE;
|
|
vm->heap_fill_percent = HEAP_FILL_PERCENT;
|
|
|
|
vm->modules = newMap(vm);
|
|
vm->builtins_count = 0;
|
|
|
|
// This is necessary to prevent garbage collection skip the entry in this
|
|
// array while we're building it.
|
|
for (int i = 0; i < PK_INSTANCE; i++) {
|
|
vm->builtin_classes[i] = NULL;
|
|
}
|
|
|
|
initializeCore(vm);
|
|
return vm;
|
|
}
|
|
|
|
void pkFreeVM(PKVM* vm) {
|
|
|
|
Object* obj = vm->first;
|
|
while (obj != NULL) {
|
|
Object* next = obj->next;
|
|
freeObject(vm, obj);
|
|
obj = next;
|
|
}
|
|
|
|
vm->working_set = (Object**)vm->config.realloc_fn(
|
|
vm->working_set, 0, vm->config.user_data);
|
|
|
|
// Tell the host application that it forget to release all of it's handles
|
|
// before freeing the VM.
|
|
__ASSERT(vm->handles == NULL, "Not all handles were released.");
|
|
|
|
DEALLOCATE(vm, vm);
|
|
}
|
|
|
|
void* pkGetUserData(const PKVM* vm) {
|
|
return vm->config.user_data;
|
|
}
|
|
|
|
void pkSetUserData(PKVM* vm, void* user_data) {
|
|
vm->config.user_data = user_data;
|
|
}
|
|
|
|
PkHandle* pkNewHandle(PKVM* vm, PkVar value) {
|
|
return vmNewHandle(vm, *((Var*)value));
|
|
}
|
|
|
|
PkVar pkGetHandleValue(const PkHandle* handle) {
|
|
return (PkVar)&handle->value;
|
|
}
|
|
|
|
void pkReleaseHandle(PKVM* vm, PkHandle* handle) {
|
|
__ASSERT(handle != NULL, "Given handle was NULL.");
|
|
|
|
// If the handle is the head of the vm's handle chain set it to the next one.
|
|
if (handle == vm->handles) {
|
|
vm->handles = handle->next;
|
|
}
|
|
|
|
// Remove the handle from the chain by connecting the both ends together.
|
|
if (handle->next) handle->next->prev = handle->prev;
|
|
if (handle->prev) handle->prev->next = handle->next;
|
|
|
|
// Free the handle.
|
|
DEALLOCATE(vm, handle);
|
|
}
|
|
|
|
// This function is responsible to call on_done function if it's done with the
|
|
// provided string pointers.
|
|
PkResult pkInterpretSource(PKVM* vm, PkStringPtr source, PkStringPtr path,
|
|
const PkCompileOptions* options) {
|
|
|
|
String* path_ = newString(vm, path.string);
|
|
if (path.on_done) path.on_done(vm, path);
|
|
vmPushTempRef(vm, &path_->_super); // path_
|
|
|
|
// FIXME:
|
|
// Should I clean the module if it already exists before compiling it?
|
|
|
|
// Load a new module to the vm's modules cache.
|
|
Module* module = vmGetModule(vm, path_);
|
|
if (module == NULL) {
|
|
module = newModule(vm);
|
|
module->path = path_;
|
|
vmPushTempRef(vm, &module->_super); // module.
|
|
vmRegisterModule(vm, module, path_);
|
|
vmPopTempRef(vm); // module.
|
|
}
|
|
vmPopTempRef(vm); // path_
|
|
|
|
// Compile the source.
|
|
PkResult result = compile(vm, module, source.string, options);
|
|
if (source.on_done) source.on_done(vm, source);
|
|
if (result != PK_RESULT_SUCCESS) return result;
|
|
|
|
// Set module initialized to true before the execution ends to prevent cyclic
|
|
// inclusion cause a crash.
|
|
module->initialized = true;
|
|
|
|
return runFiber(vm, newFiber(vm, module->body));
|
|
}
|
|
|
|
PkResult pkRunFiber(PKVM* vm, PkHandle* fiber,
|
|
int argc, PkHandle** argv) {
|
|
__ASSERT(fiber != NULL, "Handle fiber was NULL.");
|
|
Var fb = fiber->value;
|
|
__ASSERT(IS_OBJ_TYPE(fb, OBJ_FIBER), "Given handle is not a fiber.");
|
|
Fiber* _fiber = (Fiber*)AS_OBJ(fb);
|
|
|
|
Var* args[MAX_ARGC];
|
|
for (int i = 0; i < argc; i++) {
|
|
args[i] = &(argv[i]->value);
|
|
}
|
|
|
|
if (!vmPrepareFiber(vm, _fiber, argc, args)) {
|
|
return PK_RESULT_RUNTIME_ERROR;
|
|
}
|
|
|
|
ASSERT(_fiber->frame_count == 1, OOPS);
|
|
return runFiber(vm, _fiber);
|
|
}
|
|
|
|
PkResult pkResumeFiber(PKVM* vm, PkHandle* fiber, PkVar value) {
|
|
__ASSERT(fiber != NULL, "Handle fiber was NULL.");
|
|
Var fb = fiber->value;
|
|
__ASSERT(IS_OBJ_TYPE(fb, OBJ_FIBER), "Given handle is not a fiber.");
|
|
Fiber* _fiber = (Fiber*)AS_OBJ(fb);
|
|
|
|
if (!vmSwitchFiber(vm, _fiber, (Var*)value)) {
|
|
return PK_RESULT_RUNTIME_ERROR;
|
|
}
|
|
|
|
return runFiber(vm, _fiber);
|
|
}
|
|
|
|
void pkSetRuntimeError(PKVM* vm, const char* message) {
|
|
__ASSERT(vm->fiber != NULL, "This function can only be called at runtime.");
|
|
VM_SET_ERROR(vm, newString(vm, message));
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* SHARED FUNCTIONS */
|
|
/*****************************************************************************/
|
|
|
|
PkHandle* vmNewHandle(PKVM* vm, Var value) {
|
|
PkHandle* handle = (PkHandle*)ALLOCATE(vm, PkHandle);
|
|
handle->value = value;
|
|
handle->prev = NULL;
|
|
handle->next = vm->handles;
|
|
if (handle->next != NULL) handle->next->prev = handle;
|
|
vm->handles = handle;
|
|
return handle;
|
|
}
|
|
|
|
void* vmRealloc(PKVM* vm, void* memory, size_t old_size, size_t new_size) {
|
|
|
|
// TODO: Debug trace allocations here.
|
|
|
|
// Track the total allocated memory of the VM to trigger the GC.
|
|
// if vmRealloc is called for freeing, the old_size would be 0 since
|
|
// deallocated bytes are traced by garbage collector.
|
|
vm->bytes_allocated += new_size - old_size;
|
|
|
|
if (new_size > 0 && vm->bytes_allocated > vm->next_gc) {
|
|
vmCollectGarbage(vm);
|
|
}
|
|
|
|
return vm->config.realloc_fn(memory, new_size, vm->config.user_data);
|
|
}
|
|
|
|
void vmPushTempRef(PKVM* vm, Object* obj) {
|
|
ASSERT(obj != NULL, "Cannot reference to NULL.");
|
|
ASSERT(vm->temp_reference_count < MAX_TEMP_REFERENCE,
|
|
"Too many temp references");
|
|
vm->temp_reference[vm->temp_reference_count++] = obj;
|
|
}
|
|
|
|
void vmPopTempRef(PKVM* vm) {
|
|
ASSERT(vm->temp_reference_count > 0,
|
|
"Temporary reference is empty to pop.");
|
|
vm->temp_reference_count--;
|
|
}
|
|
|
|
void vmRegisterModule(PKVM* vm, Module* module, String* key) {
|
|
ASSERT((((module->name != NULL) && IS_STR_EQ(module->name, key)) ||
|
|
IS_STR_EQ(module->path, key)), OOPS);
|
|
// FIXME:
|
|
// Not sure what to do, if a module the the same key already exists. Should
|
|
// I override or assert.
|
|
mapSet(vm, vm->modules, VAR_OBJ(key), VAR_OBJ(module));
|
|
}
|
|
|
|
Module* vmGetModule(PKVM* vm, String* key) {
|
|
Var module = mapGet(vm->modules, VAR_OBJ(key));
|
|
if (IS_UNDEF(module)) return NULL;
|
|
ASSERT(AS_OBJ(module)->type == OBJ_MODULE, OOPS);
|
|
return (Module*)AS_OBJ(module);
|
|
}
|
|
|
|
void vmCollectGarbage(PKVM* vm) {
|
|
|
|
// Reset VM's bytes_allocated value and count it again so that we don't
|
|
// required to know the size of each object that'll be freeing.
|
|
vm->bytes_allocated = 0;
|
|
|
|
// Mark builtin functions.
|
|
for (int i = 0; i < vm->builtins_count; i++) {
|
|
markObject(vm, &vm->builtins_funcs[i]->_super);
|
|
}
|
|
|
|
// Mark primitive types' classes.
|
|
for (int i = 0; i < PK_INSTANCE; i++) {
|
|
// It's possible that a garbage collection could be triggered while we're
|
|
// building the primitives and the class could be NULL.
|
|
if (vm->builtin_classes[i] == NULL) continue;
|
|
|
|
markObject(vm, &vm->builtin_classes[i]->_super);
|
|
}
|
|
|
|
// Mark the modules.
|
|
markObject(vm, &vm->modules->_super);
|
|
|
|
// Mark temp references.
|
|
for (int i = 0; i < vm->temp_reference_count; i++) {
|
|
markObject(vm, vm->temp_reference[i]);
|
|
}
|
|
|
|
// Mark the handles.
|
|
for (PkHandle* h = vm->handles; h != NULL; h = h->next) {
|
|
markValue(vm, h->value);
|
|
}
|
|
|
|
// Garbage collection triggered at the middle of a compilation.
|
|
if (vm->compiler != NULL) {
|
|
compilerMarkObjects(vm, vm->compiler);
|
|
}
|
|
|
|
if (vm->fiber != NULL) {
|
|
markObject(vm, &vm->fiber->_super);
|
|
}
|
|
|
|
// Pop the marked objects from the working set and push all of it's
|
|
// referenced objects. This will repeat till no more objects left in the
|
|
// working set.
|
|
popMarkedObjects(vm);
|
|
|
|
// Now sweep all the un-marked objects in then link list and remove them
|
|
// from the chain.
|
|
|
|
// [ptr] is an Object* reference that should be equal to the next
|
|
// non-garbage Object*.
|
|
Object** ptr = &vm->first;
|
|
while (*ptr != NULL) {
|
|
|
|
// If the object the pointer points to wasn't marked it's unreachable.
|
|
// Clean it. And update the pointer points to the next object.
|
|
if (!(*ptr)->is_marked) {
|
|
Object* garbage = *ptr;
|
|
*ptr = garbage->next;
|
|
freeObject(vm, garbage);
|
|
|
|
} else {
|
|
// Unmark the object for the next garbage collection.
|
|
(*ptr)->is_marked = false;
|
|
ptr = &(*ptr)->next;
|
|
}
|
|
}
|
|
|
|
// Next GC heap size will be change depends on the byte we've left with now,
|
|
// and the [heap_fill_percent].
|
|
vm->next_gc = vm->bytes_allocated + (
|
|
(vm->bytes_allocated * vm->heap_fill_percent) / 100);
|
|
if (vm->next_gc < vm->min_heap_size) vm->next_gc = vm->min_heap_size;
|
|
}
|
|
|
|
#define _ERR_FAIL(msg) \
|
|
do { \
|
|
if (vm->fiber != NULL) VM_SET_ERROR(vm, msg); \
|
|
return false; \
|
|
} while (false)
|
|
|
|
bool vmPrepareFiber(PKVM* vm, Fiber* fiber, int argc, Var** argv) {
|
|
ASSERT(fiber->closure->fn->arity >= -1,
|
|
OOPS " (Forget to initialize arity.)");
|
|
|
|
if (argc != fiber->closure->fn->arity) {
|
|
char buff[STR_INT_BUFF_SIZE];
|
|
sprintf(buff, "%d", fiber->closure->fn->arity);
|
|
_ERR_FAIL(stringFormat(vm, "Expected exactly $ argument(s).", buff));
|
|
}
|
|
|
|
if (fiber->state != FIBER_NEW) {
|
|
switch (fiber->state) {
|
|
case FIBER_NEW: UNREACHABLE();
|
|
case FIBER_RUNNING:
|
|
_ERR_FAIL(newString(vm, "The fiber has already been running."));
|
|
case FIBER_YIELDED:
|
|
_ERR_FAIL(newString(vm, "Cannot run a fiber which is yielded, use "
|
|
"fiber_resume() instead."));
|
|
case FIBER_DONE:
|
|
_ERR_FAIL(newString(vm, "The fiber has done running."));
|
|
}
|
|
UNREACHABLE();
|
|
}
|
|
|
|
ASSERT(fiber->stack != NULL && fiber->sp == fiber->stack + 1, OOPS);
|
|
ASSERT(fiber->ret + 1 == fiber->sp, OOPS);
|
|
|
|
// Pass the function arguments.
|
|
|
|
// Assert we have the first frame (to push the arguments). And assert we have
|
|
// enough stack space for parameters.
|
|
ASSERT(fiber->frame_count == 1, OOPS);
|
|
ASSERT(fiber->frames[0].rbp == fiber->ret, OOPS);
|
|
ASSERT((fiber->stack + fiber->stack_size) - fiber->sp >= argc, OOPS);
|
|
|
|
// ARG1 is fiber, function arguments are ARG(2), ARG(3), ... ARG(argc).
|
|
// And ret[0] is the return value, parameters starts at ret[1], ...
|
|
for (int i = 0; i < argc; i++) {
|
|
fiber->ret[1 + i] = *argv[i]; // +1: ret[0] is return value.
|
|
}
|
|
fiber->sp += argc; // Parameters.
|
|
|
|
// Set the new fiber as the vm's fiber.
|
|
fiber->caller = vm->fiber;
|
|
vm->fiber = fiber;
|
|
|
|
// On success return true.
|
|
return true;
|
|
}
|
|
|
|
bool vmSwitchFiber(PKVM* vm, Fiber* fiber, Var* value) {
|
|
if (fiber->state != FIBER_YIELDED) {
|
|
switch (fiber->state) {
|
|
case FIBER_NEW:
|
|
_ERR_FAIL(newString(vm, "The fiber hasn't started. call fiber_run() "
|
|
"to start."));
|
|
case FIBER_RUNNING:
|
|
_ERR_FAIL(newString(vm, "The fiber has already been running."));
|
|
case FIBER_YIELDED: UNREACHABLE();
|
|
case FIBER_DONE:
|
|
_ERR_FAIL(newString(vm, "The fiber has done running."));
|
|
}
|
|
UNREACHABLE();
|
|
}
|
|
|
|
// Pass the resume argument if it has any.
|
|
|
|
// Assert if we have a call frame and the stack size enough for the return
|
|
// value and the resumed value.
|
|
ASSERT(fiber->frame_count != 0, OOPS);
|
|
ASSERT((fiber->stack + fiber->stack_size) - fiber->sp >= 2, OOPS);
|
|
|
|
// fb->ret will points to the return value of the 'yield()' call.
|
|
if (value == NULL) *fiber->ret = VAR_NULL;
|
|
else *fiber->ret = *value;
|
|
|
|
// Switch fiber.
|
|
fiber->caller = vm->fiber;
|
|
vm->fiber = fiber;
|
|
|
|
// On success return true.
|
|
return true;
|
|
}
|
|
|
|
#undef _ERR_FAIL
|
|
|
|
void vmYieldFiber(PKVM* vm, Var* value) {
|
|
|
|
Fiber* caller = vm->fiber->caller;
|
|
|
|
// Return the yield value to the caller fiber.
|
|
if (caller != NULL) {
|
|
if (value == NULL) *caller->ret = VAR_NULL;
|
|
else *caller->ret = *value;
|
|
}
|
|
|
|
// Can be resumed by another caller fiber.
|
|
vm->fiber->caller = NULL;
|
|
vm->fiber->state = FIBER_YIELDED;
|
|
vm->fiber = caller;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* VM INTERNALS */
|
|
/*****************************************************************************/
|
|
|
|
// The default allocator that will be used to initialize the vm's configuration
|
|
// if the host doesn't provided any allocators for us.
|
|
static void* defaultRealloc(void* memory, size_t new_size, void* user_data) {
|
|
if (new_size == 0) {
|
|
free(memory);
|
|
return NULL;
|
|
}
|
|
return realloc(memory, new_size);
|
|
}
|
|
|
|
// FIXME:
|
|
// We're assuming that the module should be available at the VM's modules cache
|
|
// which is added by the compilation pahse, but we cannot rely on the
|
|
// compilation phase here as it could be a seperate system from the runtime and
|
|
// should throw a runtime error if the module is not present in the modules
|
|
// cache (or try to load).
|
|
// Example: If we may support to store the compiled script as a separate file
|
|
// (like python's ".pyc" or java's ".class" the runtime cannot ensure that the
|
|
// module it import is already cached.
|
|
//
|
|
// Import and return the Module object with the [name] (if it's a scirpt
|
|
// doesn't have a module name, the name would be it's resolved path).
|
|
static inline Var importModule(PKVM* vm, String* key) {
|
|
|
|
Var entry = mapGet(vm->modules, VAR_OBJ(key));
|
|
if (!IS_UNDEF(entry)) {
|
|
ASSERT(AS_OBJ(entry)->type == OBJ_MODULE, OOPS);
|
|
return entry;
|
|
}
|
|
|
|
// FIXME: Should be a runtime error.
|
|
// Imported modules were resolved at compile time.
|
|
UNREACHABLE();
|
|
|
|
return VAR_NULL;
|
|
}
|
|
|
|
static inline void growStack(PKVM* vm, int size) {
|
|
Fiber* fiber = vm->fiber;
|
|
ASSERT(fiber->stack_size <= size, OOPS);
|
|
int new_size = utilPowerOf2Ceil(size);
|
|
|
|
Var* old_rbp = fiber->stack; //< Old stack base pointer.
|
|
fiber->stack = (Var*)vmRealloc(vm, fiber->stack,
|
|
sizeof(Var) * fiber->stack_size,
|
|
sizeof(Var) * new_size);
|
|
fiber->stack_size = new_size;
|
|
|
|
// If the old stack base pointer is the same as the current, that means the
|
|
// stack hasn't been moved by the reallocation. In that case we're done.
|
|
if (old_rbp == fiber->stack) return;
|
|
|
|
// If we reached here that means the stack is moved by the reallocation and
|
|
// we have to update all the pointers that pointing to the old stack slots.
|
|
|
|
//
|
|
// ' '
|
|
// ' ' ' '
|
|
// ' ' | | <new_rsp
|
|
// old_rsp> | | | |
|
|
// | | .----> | value | <new_ptr
|
|
// | | | | |
|
|
// old_ptr> | value | ------' |________| <new_rbp
|
|
// | | ^ new stack
|
|
// old_rbp> |________| | height
|
|
// old stack
|
|
//
|
|
// new_ptr = new_rbp + height
|
|
// = fiber->stack + ( old_ptr - old_rbp )
|
|
#define MAP_PTR(old_ptr) (fiber->stack + ((old_ptr) - old_rbp))
|
|
|
|
// Update the stack top pointer and the return pointer.
|
|
fiber->sp = MAP_PTR(fiber->sp);
|
|
fiber->ret = MAP_PTR(fiber->ret);
|
|
|
|
// Update the stack base pointer of the call frames.
|
|
for (int i = 0; i < fiber->frame_count; i++) {
|
|
CallFrame* frame = fiber->frames + i;
|
|
frame->rbp = MAP_PTR(frame->rbp);
|
|
}
|
|
}
|
|
|
|
static inline void pushCallFrame(PKVM* vm, const Closure* closure, Var* rbp) {
|
|
ASSERT(!closure->fn->is_native, OOPS);
|
|
|
|
// Grow the stack frame if needed.
|
|
if (vm->fiber->frame_count + 1 > vm->fiber->frame_capacity) {
|
|
int new_capacity = vm->fiber->frame_capacity << 1;
|
|
vm->fiber->frames = (CallFrame*)vmRealloc(vm, vm->fiber->frames,
|
|
sizeof(CallFrame) * vm->fiber->frame_capacity,
|
|
sizeof(CallFrame) * new_capacity);
|
|
vm->fiber->frame_capacity = new_capacity;
|
|
}
|
|
|
|
// Grow the stack if needed.
|
|
int needed = (closure->fn->fn->stack_size +
|
|
(int)(vm->fiber->sp - vm->fiber->stack));
|
|
if (vm->fiber->stack_size <= needed) growStack(vm, needed);
|
|
|
|
CallFrame* frame = vm->fiber->frames + vm->fiber->frame_count++;
|
|
frame->rbp = rbp;
|
|
frame->closure = closure;
|
|
frame->ip = closure->fn->fn->opcodes.data;
|
|
frame->self = VAR_UNDEFINED;
|
|
}
|
|
|
|
static inline void reuseCallFrame(PKVM* vm, const Closure* closure) {
|
|
|
|
ASSERT(!closure->fn->is_native, OOPS);
|
|
ASSERT(closure->fn->arity >= 0, OOPS);
|
|
ASSERT(vm->fiber->frame_count > 0, OOPS);
|
|
|
|
Fiber* fb = vm->fiber;
|
|
|
|
CallFrame* frame = fb->frames + fb->frame_count - 1;
|
|
frame->closure = closure;
|
|
frame->ip = closure->fn->fn->opcodes.data;
|
|
|
|
ASSERT(*frame->rbp == VAR_NULL, OOPS);
|
|
|
|
// Move all the argument(s) to the base of the current frame.
|
|
Var* arg = fb->sp - closure->fn->arity;
|
|
Var* target = frame->rbp + 1;
|
|
for (; arg < fb->sp; arg++, target++) {
|
|
*target = *arg;
|
|
}
|
|
|
|
// At this point target points to the stack pointer of the next call.
|
|
fb->sp = target;
|
|
|
|
// Grow the stack if needed (least probably).
|
|
int needed = (closure->fn->fn->stack_size +
|
|
(int)(vm->fiber->sp - vm->fiber->stack));
|
|
if (vm->fiber->stack_size <= needed) growStack(vm, needed);
|
|
}
|
|
|
|
// Capture the [local] into an upvalue and return it. If the upvalue already
|
|
// exists on the fiber, it'll return it.
|
|
static Upvalue* captureUpvalue(PKVM* vm, Fiber* fiber, Var* local) {
|
|
|
|
// If the fiber doesn't have any upvalues yet, create new one and add it.
|
|
if (fiber->open_upvalues == NULL) {
|
|
Upvalue* upvalue = newUpvalue(vm, local);
|
|
fiber->open_upvalues = upvalue;
|
|
return upvalue;
|
|
}
|
|
|
|
// In the bellow diagram 'u0' is the head of the open upvalues of the fiber.
|
|
// We'll walk through the upvalues to see if any of it's value is similar
|
|
// to the [local] we want to capture.
|
|
//
|
|
// This can be optimized with binary search since the upvalues are sorted
|
|
// but it's not a frequent task neither the number of upvalues would be very
|
|
// few and the local mostly located at the stack top.
|
|
//
|
|
// 1. If say 'l3' is what we want to capture, that local already has an
|
|
// upavlue 'u1' return it.
|
|
// 2. If say 'l4' is what we want to capture, It doesn't have an upvalue yet.
|
|
// Create a new upvalue and insert to the link list (ie. u1.next = u3,
|
|
// u3.next = u2) and return it.
|
|
//
|
|
// | |
|
|
// | l1 | <-- u0 (u1.value = l3)
|
|
// | l2 | |
|
|
// | l3 | <-- u1 (u1.value = l3)
|
|
// | l4 | |
|
|
// | l5 | <-- u2 (u2.value = l5)
|
|
// '------' |
|
|
// stack NULL
|
|
|
|
// Edge case: if the local is located higher than all the open upvalues, we
|
|
// cannot walk the chain, it's going to be the new head of the open upvalues.
|
|
if (fiber->open_upvalues->ptr < local) {
|
|
Upvalue* head = newUpvalue(vm, local);
|
|
head->next = fiber->open_upvalues;
|
|
fiber->open_upvalues = head;
|
|
return head;
|
|
}
|
|
|
|
// Now we walk the chain of open upvalues and if we find an upvalue for the
|
|
// local return it, otherwise insert it in the chain.
|
|
Upvalue* last = NULL;
|
|
Upvalue* current = fiber->open_upvalues;
|
|
|
|
while (current->ptr > local) {
|
|
last = current;
|
|
current = current->next;
|
|
|
|
// If the current is NULL, we've walked all the way to the end of the open
|
|
// upvalues, and there isn't one upvalue for the local.
|
|
if (current == NULL) {
|
|
last->next = newUpvalue(vm, local);
|
|
return last->next;
|
|
}
|
|
}
|
|
|
|
// If [current] is the upvalue that captured [local] then return it.
|
|
if (current->ptr == local) return current;
|
|
|
|
ASSERT(last != NULL, OOPS);
|
|
|
|
// If we've reached here, the upvalue isn't found, create a new one and
|
|
// insert it to the chain.
|
|
Upvalue* upvalue = newUpvalue(vm, local);
|
|
last->next = upvalue;
|
|
upvalue->next = current;
|
|
return upvalue;
|
|
}
|
|
|
|
// Close all the upvalues for the locals including [top] and higher in the
|
|
// stack.
|
|
static void closeUpvalues(Fiber* fiber, Var* top) {
|
|
|
|
while (fiber->open_upvalues != NULL && fiber->open_upvalues->ptr >= top) {
|
|
Upvalue* upvalue = fiber->open_upvalues;
|
|
upvalue->closed = *upvalue->ptr;
|
|
upvalue->ptr = &upvalue->closed;
|
|
|
|
fiber->open_upvalues = upvalue->next;
|
|
}
|
|
|
|
}
|
|
|
|
static void reportError(PKVM* vm) {
|
|
ASSERT(VM_HAS_ERROR(vm), "runtimeError() should be called after an error.");
|
|
// TODO: pass the error to the caller of the fiber.
|
|
|
|
// Print the Error message and stack trace.
|
|
if (vm->config.error_fn == NULL) return;
|
|
Fiber* fiber = vm->fiber;
|
|
vm->config.error_fn(vm, PK_ERROR_RUNTIME, NULL, -1, fiber->error->data);
|
|
for (int i = fiber->frame_count - 1; i >= 0; i--) {
|
|
CallFrame* frame = &fiber->frames[i];
|
|
const Function* fn = frame->closure->fn;
|
|
ASSERT(!fn->is_native, OOPS);
|
|
int line = fn->fn->oplines.data[frame->ip - fn->fn->opcodes.data - 1];
|
|
vm->config.error_fn(vm, PK_ERROR_STACKTRACE, fn->owner->path->data, line,
|
|
fn->name);
|
|
}
|
|
}
|
|
|
|
/******************************************************************************
|
|
* RUNTIME *
|
|
*****************************************************************************/
|
|
|
|
static PkResult runFiber(PKVM* vm, Fiber* fiber_) {
|
|
|
|
// Set the fiber as the vm's current fiber (another root object) to prevent
|
|
// it from garbage collection and get the reference from native functions.
|
|
vm->fiber = fiber_;
|
|
|
|
ASSERT(fiber_->state == FIBER_NEW || fiber_->state == FIBER_YIELDED, OOPS);
|
|
fiber_->state = FIBER_RUNNING;
|
|
|
|
// The instruction pointer.
|
|
register const uint8_t* ip;
|
|
|
|
register Var* rbp; //< Stack base pointer register.
|
|
register Var* self; //< Points to the self in the current call frame.
|
|
register CallFrame* frame; //< Current call frame.
|
|
register Module* module; //< Currently executing module.
|
|
register Fiber* fiber = fiber_;
|
|
|
|
#if DEBUG
|
|
#define PUSH(value) \
|
|
do { \
|
|
ASSERT(fiber->sp < (fiber->stack + (fiber->stack_size - 1)), \
|
|
OOPS); \
|
|
(*fiber->sp++ = (value)); \
|
|
} while (false)
|
|
#else
|
|
#define PUSH(value) (*fiber->sp++ = (value))
|
|
#endif
|
|
|
|
#define POP() (*(--fiber->sp))
|
|
#define DROP() (--fiber->sp)
|
|
#define PEEK(off) (*(fiber->sp + (off)))
|
|
#define READ_BYTE() (*ip++)
|
|
#define READ_SHORT() (ip+=2, (uint16_t)((ip[-2] << 8) | ip[-1]))
|
|
|
|
// Switch back to the caller of the current fiber, will be called when we're
|
|
// done with the fiber or aborting it for runtime errors.
|
|
#define FIBER_SWITCH_BACK() \
|
|
do { \
|
|
Fiber* caller = fiber->caller; \
|
|
ASSERT(caller == NULL || caller->state == FIBER_RUNNING, OOPS); \
|
|
fiber->state = FIBER_DONE; \
|
|
fiber->caller = NULL; \
|
|
fiber = caller; \
|
|
vm->fiber = fiber; \
|
|
} while (false)
|
|
|
|
// Check if any runtime error exists and if so returns RESULT_RUNTIME_ERROR.
|
|
#define CHECK_ERROR() \
|
|
do { \
|
|
if (VM_HAS_ERROR(vm)) { \
|
|
UPDATE_FRAME(); \
|
|
reportError(vm); \
|
|
FIBER_SWITCH_BACK(); \
|
|
return PK_RESULT_RUNTIME_ERROR; \
|
|
} \
|
|
} while (false)
|
|
|
|
// [err_msg] must be of type String.
|
|
#define RUNTIME_ERROR(err_msg) \
|
|
do { \
|
|
VM_SET_ERROR(vm, err_msg); \
|
|
UPDATE_FRAME(); \
|
|
reportError(vm); \
|
|
FIBER_SWITCH_BACK(); \
|
|
return PK_RESULT_RUNTIME_ERROR; \
|
|
} while (false)
|
|
|
|
// Load the last call frame to vm's execution variables to resume/run the
|
|
// function.
|
|
#define LOAD_FRAME() \
|
|
do { \
|
|
frame = &fiber->frames[fiber->frame_count-1]; \
|
|
ip = frame->ip; \
|
|
rbp = frame->rbp; \
|
|
self = &frame->self; \
|
|
module = frame->closure->fn->owner; \
|
|
} while (false)
|
|
|
|
// Update the frame's execution variables before pushing another call frame.
|
|
#define UPDATE_FRAME() frame->ip = ip
|
|
|
|
#ifdef OPCODE
|
|
#error "OPCODE" should not be deifined here.
|
|
#endif
|
|
|
|
#define SWITCH() Opcode instruction; switch (instruction = (Opcode)READ_BYTE())
|
|
#define OPCODE(code) case OP_##code
|
|
#define DISPATCH() goto L_vm_main_loop
|
|
|
|
// Load the fiber's top call frame to the vm's execution variables.
|
|
LOAD_FRAME();
|
|
|
|
L_vm_main_loop:
|
|
// This NO_OP is required since Labels can only be followed by statements
|
|
// and, declarations are not statements, If the macro DUMP_STACK isn't
|
|
// defined, the next line become a declaration (Opcode instruction;).
|
|
NO_OP;
|
|
|
|
#if DUMP_STACK
|
|
system("cls"); // FIXME:
|
|
dumpGlobalValues(vm);
|
|
dumpStackFrame(vm);
|
|
DEBUG_BREAK();
|
|
#endif
|
|
|
|
SWITCH() {
|
|
|
|
OPCODE(PUSH_CONSTANT):
|
|
{
|
|
uint16_t index = READ_SHORT();
|
|
ASSERT_INDEX(index, module->constants.count);
|
|
PUSH(module->constants.data[index]);
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_NULL):
|
|
PUSH(VAR_NULL);
|
|
DISPATCH();
|
|
|
|
OPCODE(PUSH_0):
|
|
PUSH(VAR_NUM(0));
|
|
DISPATCH();
|
|
|
|
OPCODE(PUSH_TRUE):
|
|
PUSH(VAR_TRUE);
|
|
DISPATCH();
|
|
|
|
OPCODE(PUSH_FALSE):
|
|
PUSH(VAR_FALSE);
|
|
DISPATCH();
|
|
|
|
OPCODE(SWAP):
|
|
{
|
|
Var tmp = *(fiber->sp - 1);
|
|
*(fiber->sp - 1) = *(fiber->sp - 2);
|
|
*(fiber->sp - 2) = tmp;
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_LIST):
|
|
{
|
|
List* list = newList(vm, (uint32_t)READ_SHORT());
|
|
PUSH(VAR_OBJ(list));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_MAP):
|
|
{
|
|
Map* map = newMap(vm);
|
|
PUSH(VAR_OBJ(map));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_SELF):
|
|
{
|
|
PUSH(*self);
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(LIST_APPEND):
|
|
{
|
|
Var elem = PEEK(-1); // Don't pop yet, we need the reference for gc.
|
|
Var list = PEEK(-2);
|
|
ASSERT(IS_OBJ_TYPE(list, OBJ_LIST), OOPS);
|
|
pkVarBufferWrite(&((List*)AS_OBJ(list))->elements, vm, elem);
|
|
DROP(); // elem
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(MAP_INSERT):
|
|
{
|
|
Var value = PEEK(-1); // Don't pop yet, we need the reference for gc.
|
|
Var key = PEEK(-2); // Don't pop yet, we need the reference for gc.
|
|
Var on = PEEK(-3);
|
|
|
|
ASSERT(IS_OBJ_TYPE(on, OBJ_MAP), OOPS);
|
|
|
|
if (IS_OBJ(key) && !isObjectHashable(AS_OBJ(key)->type)) {
|
|
RUNTIME_ERROR(stringFormat(vm, "$ type is not hashable.",
|
|
varTypeName(key)));
|
|
}
|
|
mapSet(vm, (Map*)AS_OBJ(on), key, value);
|
|
|
|
DROP(); // value
|
|
DROP(); // key
|
|
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_LOCAL_0):
|
|
OPCODE(PUSH_LOCAL_1):
|
|
OPCODE(PUSH_LOCAL_2):
|
|
OPCODE(PUSH_LOCAL_3):
|
|
OPCODE(PUSH_LOCAL_4):
|
|
OPCODE(PUSH_LOCAL_5):
|
|
OPCODE(PUSH_LOCAL_6):
|
|
OPCODE(PUSH_LOCAL_7):
|
|
OPCODE(PUSH_LOCAL_8):
|
|
{
|
|
int index = (int)(instruction - OP_PUSH_LOCAL_0);
|
|
PUSH(rbp[index + 1]); // +1: rbp[0] is return value.
|
|
DISPATCH();
|
|
}
|
|
OPCODE(PUSH_LOCAL_N):
|
|
{
|
|
uint8_t index = READ_BYTE();
|
|
PUSH(rbp[index + 1]); // +1: rbp[0] is return value.
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(STORE_LOCAL_0):
|
|
OPCODE(STORE_LOCAL_1):
|
|
OPCODE(STORE_LOCAL_2):
|
|
OPCODE(STORE_LOCAL_3):
|
|
OPCODE(STORE_LOCAL_4):
|
|
OPCODE(STORE_LOCAL_5):
|
|
OPCODE(STORE_LOCAL_6):
|
|
OPCODE(STORE_LOCAL_7):
|
|
OPCODE(STORE_LOCAL_8):
|
|
{
|
|
int index = (int)(instruction - OP_STORE_LOCAL_0);
|
|
rbp[index + 1] = PEEK(-1); // +1: rbp[0] is return value.
|
|
DISPATCH();
|
|
}
|
|
OPCODE(STORE_LOCAL_N):
|
|
{
|
|
uint8_t index = READ_BYTE();
|
|
rbp[index + 1] = PEEK(-1); // +1: rbp[0] is return value.
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_GLOBAL):
|
|
{
|
|
uint8_t index = READ_BYTE();
|
|
ASSERT_INDEX(index, module->globals.count);
|
|
PUSH(module->globals.data[index]);
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(STORE_GLOBAL):
|
|
{
|
|
uint8_t index = READ_BYTE();
|
|
ASSERT_INDEX(index, module->globals.count);
|
|
module->globals.data[index] = PEEK(-1);
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_BUILTIN_FN):
|
|
{
|
|
uint8_t index = READ_BYTE();
|
|
ASSERT_INDEX(index, vm->builtins_count);
|
|
Closure* closure = vm->builtins_funcs[index];
|
|
PUSH(VAR_OBJ(closure));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_BUILTIN_TY):
|
|
{
|
|
uint8_t index = READ_BYTE();
|
|
ASSERT_INDEX(index, PK_INSTANCE);
|
|
Class* cls = vm->builtin_classes[index];
|
|
PUSH(VAR_OBJ(cls));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_UPVALUE):
|
|
{
|
|
uint8_t index = READ_BYTE();
|
|
PUSH(*(frame->closure->upvalues[index]->ptr));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(STORE_UPVALUE):
|
|
{
|
|
uint8_t index = READ_BYTE();
|
|
*(frame->closure->upvalues[index]->ptr) = PEEK(-1);
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(PUSH_CLOSURE):
|
|
{
|
|
uint16_t index = READ_SHORT();
|
|
ASSERT_INDEX(index, module->constants.count);
|
|
ASSERT(IS_OBJ_TYPE(module->constants.data[index], OBJ_FUNC), OOPS);
|
|
Function* fn = (Function*)AS_OBJ(module->constants.data[index]);
|
|
Closure* closure = newClosure(vm, fn);
|
|
|
|
// Capture the vaupes.
|
|
for (int i = 0; i < fn->upvalue_count; i++) {
|
|
uint8_t is_immediate = READ_BYTE();
|
|
uint8_t index = READ_BYTE();
|
|
|
|
if (is_immediate) {
|
|
// rbp[0] is the return value, rbp + 1 is the first local and so on.
|
|
closure->upvalues[i] = captureUpvalue(vm, fiber, (rbp + 1 + index));
|
|
} else {
|
|
// The upvalue is already captured by the current function, reuse it.
|
|
closure->upvalues[i] = frame->closure->upvalues[index];
|
|
}
|
|
}
|
|
|
|
PUSH(VAR_OBJ(closure));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(CLOSE_UPVALUE):
|
|
{
|
|
closeUpvalues(fiber, fiber->sp - 1);
|
|
DROP();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(POP):
|
|
DROP();
|
|
DISPATCH();
|
|
|
|
OPCODE(IMPORT):
|
|
{
|
|
uint16_t index = READ_SHORT();
|
|
String* name = moduleGetStringAt(module, (int)index);
|
|
ASSERT(name != NULL, OOPS);
|
|
|
|
Var _imported = importModule(vm, name);
|
|
ASSERT(IS_OBJ_TYPE(_imported, OBJ_MODULE), OOPS);
|
|
PUSH(_imported);
|
|
|
|
// TODO: If the body doesn't have any statements (just the functions).
|
|
// This initialization call is un-necessary.
|
|
|
|
Module* imported = (Module*)AS_OBJ(_imported);
|
|
if (!imported->initialized) {
|
|
imported->initialized = true;
|
|
|
|
ASSERT(imported->body != NULL, OOPS);
|
|
|
|
// Note that we're setting the main function's return address to the
|
|
// module itself (for every other function we'll push a null at the rbp
|
|
// before calling them and it'll be returned without modified if the
|
|
// function doesn't returned anything). Also We can't return from the
|
|
// body of the module, so the main function will return what's at the
|
|
// rbp without modifying it. So at the end of the main function the
|
|
// stack top would be the module itself.
|
|
Var* module_ret = fiber->sp - 1;
|
|
|
|
UPDATE_FRAME(); //< Update the current frame's ip.
|
|
pushCallFrame(vm, imported->body, module_ret);
|
|
LOAD_FRAME(); //< Load the top frame to vm's execution variables.
|
|
}
|
|
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(CALL):
|
|
OPCODE(TAIL_CALL):
|
|
{
|
|
const uint8_t argc = READ_BYTE();
|
|
Var* callable = fiber->sp - argc - 1;
|
|
|
|
const Closure* closure = NULL;
|
|
|
|
// Raw functions cannot be on the stack, since they're not first class
|
|
// citizens.
|
|
ASSERT(!IS_OBJ_TYPE(*callable, OBJ_FUNC), OOPS);
|
|
|
|
if (IS_OBJ_TYPE(*callable, OBJ_CLOSURE)) {
|
|
closure = (const Closure*)AS_OBJ(*callable);
|
|
|
|
} else if (IS_OBJ_TYPE(*callable, OBJ_CLASS)) {
|
|
closure = (const Closure*)((Class*)AS_OBJ(*callable))->ctor;
|
|
|
|
} else {
|
|
RUNTIME_ERROR(stringFormat(vm, "$ $(@).", "Expected a callable to "
|
|
"call, instead got",
|
|
varTypeName(*callable), toString(vm, *callable)));
|
|
DISPATCH();
|
|
}
|
|
|
|
// If we reached here it's a valid callable.
|
|
|
|
// -1 argument means multiple number of args.
|
|
if (closure->fn->arity != -1 && closure->fn->arity != argc) {
|
|
char buff[STR_INT_BUFF_SIZE]; sprintf(buff, "%d", closure->fn->arity);
|
|
String* msg = stringFormat(vm, "Expected exactly $ argument(s).",
|
|
buff);
|
|
RUNTIME_ERROR(msg);
|
|
}
|
|
|
|
// Next call frame starts here. (including return value).
|
|
fiber->ret = callable;
|
|
*(fiber->ret) = VAR_NULL; //< Set the return value to null.
|
|
|
|
if (closure->fn->is_native) {
|
|
|
|
if (closure->fn->native == NULL) {
|
|
RUNTIME_ERROR(stringFormat(vm,
|
|
"Native function pointer of $ was NULL.", closure->fn->name));
|
|
}
|
|
|
|
// Update the current frame's ip.
|
|
UPDATE_FRAME();
|
|
|
|
closure->fn->native(vm); //< Call the native function.
|
|
|
|
// Calling yield() will change vm->fiber to it's caller fiber, which
|
|
// would be null if we're not running the function with a fiber.
|
|
if (vm->fiber == NULL) return PK_RESULT_SUCCESS;
|
|
|
|
// Pop function arguments except for the return value.
|
|
// Note that calling fiber_new() and yield() would change the
|
|
// vm->fiber so we're using fiber.
|
|
fiber->sp = fiber->ret + 1;
|
|
|
|
// If the fiber has changed, Load the top frame to vm's execution
|
|
// variables.
|
|
if (vm->fiber != fiber) {
|
|
fiber = vm->fiber;
|
|
LOAD_FRAME();
|
|
}
|
|
|
|
CHECK_ERROR();
|
|
|
|
} else {
|
|
|
|
if (instruction == OP_CALL) {
|
|
UPDATE_FRAME(); //< Update the current frame's ip.
|
|
pushCallFrame(vm, closure, callable);
|
|
LOAD_FRAME(); //< Load the top frame to vm's execution variables.
|
|
|
|
} else {
|
|
ASSERT(instruction == OP_TAIL_CALL, OOPS);
|
|
|
|
reuseCallFrame(vm, closure);
|
|
LOAD_FRAME(); //< Re-load the frame to vm's execution variables.
|
|
}
|
|
}
|
|
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(ITER_TEST):
|
|
{
|
|
Var seq = PEEK(-3);
|
|
|
|
// Primitive types are not iterable.
|
|
if (!IS_OBJ(seq)) {
|
|
if (IS_NULL(seq)) {
|
|
RUNTIME_ERROR(newString(vm, "Null is not iterable."));
|
|
} else if (IS_BOOL(seq)) {
|
|
RUNTIME_ERROR(newString(vm, "Boolenan is not iterable."));
|
|
} else if (IS_NUM(seq)) {
|
|
RUNTIME_ERROR(newString(vm, "Number is not iterable."));
|
|
} else {
|
|
UNREACHABLE();
|
|
}
|
|
}
|
|
|
|
DISPATCH();
|
|
}
|
|
|
|
// TODO: move this to a function in pk_core.c.
|
|
OPCODE(ITER):
|
|
{
|
|
Var* value = (fiber->sp - 1);
|
|
Var* iterator = (fiber->sp - 2);
|
|
Var seq = PEEK(-3);
|
|
uint16_t jump_offset = READ_SHORT();
|
|
|
|
#define JUMP_ITER_EXIT() \
|
|
do { \
|
|
ip += jump_offset; \
|
|
DISPATCH(); \
|
|
} while (false)
|
|
|
|
ASSERT(IS_NUM(*iterator), OOPS);
|
|
double it = AS_NUM(*iterator); //< Nth iteration.
|
|
ASSERT(AS_NUM(*iterator) == (int32_t)trunc(it), OOPS);
|
|
|
|
Object* obj = AS_OBJ(seq);
|
|
switch (obj->type) {
|
|
|
|
case OBJ_STRING: {
|
|
uint32_t iter = (int32_t)trunc(it);
|
|
|
|
// TODO: // Need to consider utf8.
|
|
String* str = ((String*)obj);
|
|
if (iter >= str->length) JUMP_ITER_EXIT();
|
|
|
|
//TODO: vm's char (and reusable) strings.
|
|
*value = VAR_OBJ(newStringLength(vm, str->data + iter, 1));
|
|
*iterator = VAR_NUM((double)iter + 1);
|
|
|
|
} DISPATCH();
|
|
|
|
case OBJ_LIST: {
|
|
uint32_t iter = (int32_t)trunc(it);
|
|
pkVarBuffer* elems = &((List*)obj)->elements;
|
|
if (iter >= elems->count) JUMP_ITER_EXIT();
|
|
*value = elems->data[iter];
|
|
*iterator = VAR_NUM((double)iter + 1);
|
|
|
|
} DISPATCH();
|
|
|
|
case OBJ_MAP: {
|
|
uint32_t iter = (int32_t)trunc(it);
|
|
|
|
Map* map = (Map*)obj;
|
|
if (map->entries == NULL) JUMP_ITER_EXIT();
|
|
MapEntry* e = map->entries + iter;
|
|
for (; iter < map->capacity; iter++, e++) {
|
|
if (!IS_UNDEF(e->key)) break;
|
|
}
|
|
if (iter >= map->capacity) JUMP_ITER_EXIT();
|
|
|
|
*value = map->entries[iter].key;
|
|
*iterator = VAR_NUM((double)iter + 1);
|
|
|
|
} DISPATCH();
|
|
|
|
case OBJ_RANGE: {
|
|
double from = ((Range*)obj)->from;
|
|
double to = ((Range*)obj)->to;
|
|
if (from == to) JUMP_ITER_EXIT();
|
|
|
|
double current;
|
|
if (from <= to) { //< Straight range.
|
|
current = from + it;
|
|
} else { //< Reversed range.
|
|
current = from - it;
|
|
}
|
|
if (current == to) JUMP_ITER_EXIT();
|
|
*value = VAR_NUM(current);
|
|
*iterator = VAR_NUM(it + 1);
|
|
|
|
} DISPATCH();
|
|
|
|
case OBJ_MODULE:
|
|
case OBJ_FUNC:
|
|
case OBJ_CLOSURE:
|
|
case OBJ_UPVALUE:
|
|
case OBJ_FIBER:
|
|
case OBJ_CLASS:
|
|
case OBJ_INST:
|
|
TODO; break;
|
|
default:
|
|
UNREACHABLE();
|
|
}
|
|
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(JUMP):
|
|
{
|
|
uint16_t offset = READ_SHORT();
|
|
ip += offset;
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(LOOP):
|
|
{
|
|
uint16_t offset = READ_SHORT();
|
|
ip -= offset;
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(JUMP_IF):
|
|
{
|
|
Var cond = POP();
|
|
uint16_t offset = READ_SHORT();
|
|
if (toBool(cond)) {
|
|
ip += offset;
|
|
}
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(JUMP_IF_NOT):
|
|
{
|
|
Var cond = POP();
|
|
uint16_t offset = READ_SHORT();
|
|
if (!toBool(cond)) {
|
|
ip += offset;
|
|
}
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(OR):
|
|
{
|
|
Var cond = PEEK(-1);
|
|
uint16_t offset = READ_SHORT();
|
|
if (toBool(cond)) {
|
|
ip += offset;
|
|
} else {
|
|
DROP();
|
|
}
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(AND):
|
|
{
|
|
Var cond = PEEK(-1);
|
|
uint16_t offset = READ_SHORT();
|
|
if (!toBool(cond)) {
|
|
ip += offset;
|
|
} else {
|
|
DROP();
|
|
}
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(RETURN):
|
|
{
|
|
|
|
// Close all the locals of the current frame.
|
|
closeUpvalues(fiber, rbp + 1);
|
|
|
|
// Set the return value.
|
|
Var ret_value = POP();
|
|
|
|
// Pop the last frame, and if no more call frames, we're done with the
|
|
// current fiber.
|
|
if (--fiber->frame_count == 0) {
|
|
// TODO: if we're evaluating an expression we need to set it's
|
|
// value on the stack.
|
|
//fiber->sp = fiber->stack; ??
|
|
|
|
FIBER_SWITCH_BACK();
|
|
|
|
if (fiber == NULL) {
|
|
return PK_RESULT_SUCCESS;
|
|
|
|
} else {
|
|
*fiber->ret = ret_value;
|
|
}
|
|
|
|
} else {
|
|
*rbp = ret_value;
|
|
// Pop the params (locals should have popped at this point) and update
|
|
// stack pointer.
|
|
fiber->sp = rbp + 1; // +1: rbp is returned value.
|
|
}
|
|
|
|
LOAD_FRAME();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(GET_ATTRIB):
|
|
{
|
|
Var on = PEEK(-1); // Don't pop yet, we need the reference for gc.
|
|
String* name = moduleGetStringAt(module, READ_SHORT());
|
|
ASSERT(name != NULL, OOPS);
|
|
Var value = varGetAttrib(vm, on, name);
|
|
DROP(); // on
|
|
PUSH(value);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(GET_ATTRIB_KEEP):
|
|
{
|
|
Var on = PEEK(-1);
|
|
String* name = moduleGetStringAt(module, READ_SHORT());
|
|
ASSERT(name != NULL, OOPS);
|
|
PUSH(varGetAttrib(vm, on, name));
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(SET_ATTRIB):
|
|
{
|
|
Var value = PEEK(-1); // Don't pop yet, we need the reference for gc.
|
|
Var on = PEEK(-2); // Don't pop yet, we need the reference for gc.
|
|
String* name = moduleGetStringAt(module, READ_SHORT());
|
|
ASSERT(name != NULL, OOPS);
|
|
varSetAttrib(vm, on, name, value);
|
|
|
|
DROP(); // value
|
|
DROP(); // on
|
|
PUSH(value);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(GET_SUBSCRIPT):
|
|
{
|
|
Var key = PEEK(-1); // Don't pop yet, we need the reference for gc.
|
|
Var on = PEEK(-2); // Don't pop yet, we need the reference for gc.
|
|
Var value = varGetSubscript(vm, on, key);
|
|
DROP(); // key
|
|
DROP(); // on
|
|
PUSH(value);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(GET_SUBSCRIPT_KEEP):
|
|
{
|
|
Var key = PEEK(-1);
|
|
Var on = PEEK(-2);
|
|
PUSH(varGetSubscript(vm, on, key));
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(SET_SUBSCRIPT):
|
|
{
|
|
Var value = PEEK(-1); // Don't pop yet, we need the reference for gc.
|
|
Var key = PEEK(-2); // Don't pop yet, we need the reference for gc.
|
|
Var on = PEEK(-3); // Don't pop yet, we need the reference for gc.
|
|
varsetSubscript(vm, on, key, value);
|
|
DROP(); // value
|
|
DROP(); // key
|
|
DROP(); // on
|
|
PUSH(value);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(NEGATIVE):
|
|
{
|
|
Var num = POP();
|
|
if (!IS_NUM(num)) {
|
|
RUNTIME_ERROR(newString(vm, "Can not negate a non numeric value."));
|
|
}
|
|
PUSH(VAR_NUM(-AS_NUM(num)));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(NOT):
|
|
{
|
|
Var val = POP();
|
|
PUSH(VAR_BOOL(!toBool(val)));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(BIT_NOT):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var val = PEEK(-1);
|
|
|
|
Var result = varBitNot(vm, val);
|
|
DROP(); // val
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
// Do not ever use PUSH(binaryOp(vm, POP(), POP()));
|
|
// Function parameters are not evaluated in a defined order in C.
|
|
|
|
OPCODE(ADD):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
Var result = varAdd(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(SUBTRACT):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
Var result = varSubtract(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(MULTIPLY):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
Var result = varMultiply(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(DIVIDE):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
Var result = varDivide(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(MOD):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
Var result = varModulo(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(BIT_AND) :
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
Var result = varBitAnd(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(BIT_OR):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
|
|
Var result = varBitOr(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(BIT_XOR):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
|
|
Var result = varBitXor(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(BIT_LSHIFT):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
|
|
Var result = varBitLshift(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(BIT_RSHIFT):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var r = PEEK(-1), l = PEEK(-2);
|
|
|
|
Var result = varBitRshift(vm, l, r);
|
|
DROP(); DROP(); // r, l
|
|
PUSH(result);
|
|
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(EQEQ):
|
|
{
|
|
Var r = POP(), l = POP();
|
|
PUSH(VAR_BOOL(isValuesEqual(l, r)));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(NOTEQ):
|
|
{
|
|
Var r = POP(), l = POP();
|
|
PUSH(VAR_BOOL(!isValuesEqual(l, r)));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(LT):
|
|
{
|
|
Var r = POP(), l = POP();
|
|
PUSH(VAR_BOOL(varLesser(l, r)));
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(LTEQ):
|
|
{
|
|
Var r = POP(), l = POP();
|
|
bool lteq = varLesser(l, r);
|
|
CHECK_ERROR();
|
|
|
|
if (!lteq) {
|
|
lteq = isValuesEqual(l, r);
|
|
CHECK_ERROR();
|
|
}
|
|
|
|
PUSH(VAR_BOOL(lteq));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(GT):
|
|
{
|
|
Var r = POP(), l = POP();
|
|
PUSH(VAR_BOOL(varGreater(l, r)));
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(GTEQ):
|
|
{
|
|
Var r = POP(), l = POP();
|
|
bool gteq = varGreater(l, r);
|
|
CHECK_ERROR();
|
|
|
|
if (!gteq) {
|
|
gteq = isValuesEqual(l, r);
|
|
CHECK_ERROR();
|
|
}
|
|
|
|
PUSH(VAR_BOOL(gteq));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(RANGE):
|
|
{
|
|
Var to = PEEK(-1); // Don't pop yet, we need the reference for gc.
|
|
Var from = PEEK(-2); // Don't pop yet, we need the reference for gc.
|
|
if (!IS_NUM(from) || !IS_NUM(to)) {
|
|
RUNTIME_ERROR(newString(vm, "Range arguments must be number."));
|
|
}
|
|
DROP(); // to
|
|
DROP(); // from
|
|
PUSH(VAR_OBJ(newRange(vm, AS_NUM(from), AS_NUM(to))));
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(IN):
|
|
{
|
|
// Don't pop yet, we need the reference for gc.
|
|
Var container = PEEK(-1), elem = PEEK(-2);
|
|
bool contains = varContains(vm, elem, container);
|
|
DROP(); DROP(); // container, elem
|
|
PUSH(VAR_BOOL(contains));
|
|
CHECK_ERROR();
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(REPL_PRINT):
|
|
{
|
|
if (vm->config.write_fn != NULL) {
|
|
Var tmp = PEEK(-1);
|
|
if (!IS_NULL(tmp)) {
|
|
vm->config.write_fn(vm, toRepr(vm, tmp)->data);
|
|
vm->config.write_fn(vm, "\n");
|
|
}
|
|
}
|
|
DISPATCH();
|
|
}
|
|
|
|
OPCODE(END):
|
|
UNREACHABLE();
|
|
break;
|
|
|
|
default:
|
|
UNREACHABLE();
|
|
|
|
}
|
|
|
|
UNREACHABLE();
|
|
return PK_RESULT_RUNTIME_ERROR;
|
|
}
|