741 lines
19 KiB
C
741 lines
19 KiB
C
/*
|
|
* $Id$
|
|
*
|
|
Copyright (c) 2014-2016 Chung, Hyung-Hwan. All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions
|
|
are met:
|
|
1. Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
2. Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in the
|
|
documentation and/or other materials provided with the distribution.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR
|
|
IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
|
OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
|
IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
|
NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
|
THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "moo-prv.h"
|
|
|
|
moo_t* moo_open (moo_mmgr_t* mmgr, moo_oow_t xtnsize, moo_oow_t heapsize, const moo_vmprim_t* vmprim, moo_errnum_t* errnum)
|
|
{
|
|
moo_t* moo;
|
|
|
|
/* if this assertion fails, correct the type definition in moo.h */
|
|
MOO_ASSERT (moo, MOO_SIZEOF(moo_oow_t) == MOO_SIZEOF(moo_oop_t));
|
|
|
|
moo = MOO_MMGR_ALLOC (mmgr, MOO_SIZEOF(*moo) + xtnsize);
|
|
if (moo)
|
|
{
|
|
if (moo_init(moo, mmgr, heapsize, vmprim) <= -1)
|
|
{
|
|
if (errnum) *errnum = moo->errnum;
|
|
MOO_MMGR_FREE (mmgr, moo);
|
|
moo = MOO_NULL;
|
|
}
|
|
else MOO_MEMSET (moo + 1, 0, xtnsize);
|
|
}
|
|
else if (errnum) *errnum = MOO_ESYSMEM;
|
|
|
|
return moo;
|
|
}
|
|
|
|
void moo_close (moo_t* moo)
|
|
{
|
|
moo_fini (moo);
|
|
MOO_MMGR_FREE (moo->mmgr, moo);
|
|
}
|
|
|
|
static void fill_bigint_tables (moo_t* moo)
|
|
{
|
|
int radix, safe_ndigits;
|
|
moo_oow_t ub, w;
|
|
moo_oow_t multiplier;
|
|
|
|
for (radix = 2; radix <= 36; radix++)
|
|
{
|
|
w = 0;
|
|
ub = (moo_oow_t)MOO_TYPE_MAX(moo_liw_t) / radix - (radix - 1);
|
|
multiplier = 1;
|
|
safe_ndigits = 0;
|
|
|
|
while (w <= ub)
|
|
{
|
|
w = w * radix + (radix - 1);
|
|
multiplier *= radix;
|
|
safe_ndigits++;
|
|
}
|
|
|
|
/* safe_ndigits contains the number of digits that never
|
|
* cause overflow when computed normally with a native type. */
|
|
moo->bigint[radix].safe_ndigits = safe_ndigits;
|
|
moo->bigint[radix].multiplier = multiplier;
|
|
}
|
|
}
|
|
|
|
int moo_init (moo_t* moo, moo_mmgr_t* mmgr, moo_oow_t heapsz, const moo_vmprim_t* vmprim)
|
|
{
|
|
int modtab_inited = 0;
|
|
|
|
MOO_MEMSET (moo, 0, MOO_SIZEOF(*moo));
|
|
moo->mmgr = mmgr;
|
|
moo->cmgr = moo_getutf8cmgr ();
|
|
moo->vmprim = *vmprim;
|
|
|
|
moo->option.log_mask = ~0u;
|
|
moo->option.dfl_symtab_size = MOO_DFL_SYMTAB_SIZE;
|
|
moo->option.dfl_sysdic_size = MOO_DFL_SYSDIC_SIZE;
|
|
moo->option.dfl_procstk_size = MOO_DFL_PROCSTK_SIZE;
|
|
|
|
/* TODO: intoduct a permanent heap */
|
|
/*moo->permheap = moo_makeheap (moo, what is the best size???);
|
|
if (!moo->permheap) goto oops; */
|
|
moo->curheap = moo_makeheap (moo, heapsz);
|
|
if (!moo->curheap) goto oops;
|
|
moo->newheap = moo_makeheap (moo, heapsz);
|
|
if (!moo->newheap) goto oops;
|
|
|
|
if (moo_rbt_init (&moo->modtab, moo, MOO_SIZEOF(moo_ooch_t), 1) <= -1) goto oops;
|
|
modtab_inited = 1;
|
|
moo_rbt_setstyle (&moo->modtab, moo_getrbtstyle(MOO_RBT_STYLE_INLINE_COPIERS));
|
|
|
|
fill_bigint_tables (moo);
|
|
|
|
moo->tagged_classes[MOO_OOP_TAG_SMINT] = &moo->_small_integer;
|
|
moo->tagged_classes[MOO_OOP_TAG_CHAR] = &moo->_character;
|
|
moo->tagged_classes[MOO_OOP_TAG_ERROR] = &moo->_error_class;
|
|
|
|
return 0;
|
|
|
|
oops:
|
|
if (modtab_inited) moo_rbt_fini (&moo->modtab);
|
|
if (moo->newheap) moo_killheap (moo, moo->newheap);
|
|
if (moo->curheap) moo_killheap (moo, moo->curheap);
|
|
if (moo->permheap) moo_killheap (moo, moo->permheap);
|
|
return -1;
|
|
}
|
|
|
|
static moo_rbt_walk_t unload_module (moo_rbt_t* rbt, moo_rbt_pair_t* pair, void* ctx)
|
|
{
|
|
moo_t* moo = (moo_t*)ctx;
|
|
moo_mod_data_t* mdp;
|
|
|
|
mdp = MOO_RBT_VPTR(pair);
|
|
MOO_ASSERT (moo, mdp != MOO_NULL);
|
|
|
|
mdp->pair = MOO_NULL; /* to prevent moo_closemod() from calling moo_rbt_delete() */
|
|
moo_closemod (moo, mdp);
|
|
|
|
return MOO_RBT_WALK_FORWARD;
|
|
}
|
|
|
|
void moo_fini (moo_t* moo)
|
|
{
|
|
moo_cb_t* cb;
|
|
moo_oow_t i;
|
|
|
|
if (moo->sem_list)
|
|
{
|
|
moo_freemem (moo, moo->sem_list);
|
|
moo->sem_list_capa = 0;
|
|
moo->sem_list_count = 0;
|
|
}
|
|
|
|
if (moo->sem_heap)
|
|
{
|
|
moo_freemem (moo, moo->sem_heap);
|
|
moo->sem_heap_capa = 0;
|
|
moo->sem_heap_count = 0;
|
|
}
|
|
|
|
for (cb = moo->cblist; cb; cb = cb->next)
|
|
{
|
|
if (cb->fini) cb->fini (moo);
|
|
}
|
|
|
|
moo_rbt_walk (&moo->modtab, unload_module, moo); /* unload all modules */
|
|
moo_rbt_fini (&moo->modtab);
|
|
|
|
/* TOOD: persistency? storing objects to image file? */
|
|
moo_killheap (moo, moo->newheap);
|
|
moo_killheap (moo, moo->curheap);
|
|
moo_killheap (moo, moo->permheap);
|
|
|
|
/* deregister all callbacks */
|
|
while (moo->cblist) moo_deregcb (moo, moo->cblist);
|
|
|
|
for (i = 0; i < MOO_COUNTOF(moo->sbuf); i++)
|
|
{
|
|
if (moo->sbuf[i].ptr)
|
|
{
|
|
moo_freemem (moo, moo->sbuf[i].ptr);
|
|
moo->sbuf[i].ptr = MOO_NULL;
|
|
moo->sbuf[i].len = 0;
|
|
moo->sbuf[i].capa = 0;
|
|
}
|
|
}
|
|
|
|
if (moo->log.ptr)
|
|
{
|
|
/* make sure to flush your log message */
|
|
/* TODO: flush unwritten message */
|
|
|
|
moo_freemem (moo, moo->log.ptr);
|
|
moo->log.capa = 0;
|
|
moo->log.len = 0;
|
|
}
|
|
}
|
|
|
|
const moo_ooch_t* moo_geterrstr (moo_t* moo)
|
|
{
|
|
return moo_errnumtoerrstr (moo->errnum);
|
|
}
|
|
|
|
int moo_setoption (moo_t* moo, moo_option_t id, const void* value)
|
|
{
|
|
switch (id)
|
|
{
|
|
case MOO_TRAIT:
|
|
moo->option.trait = *(const unsigned int*)value;
|
|
return 0;
|
|
|
|
case MOO_LOG_MASK:
|
|
moo->option.log_mask = *(const unsigned int*)value;
|
|
return 0;
|
|
|
|
case MOO_SYMTAB_SIZE:
|
|
{
|
|
moo_oow_t w;
|
|
|
|
w = *(moo_oow_t*)value;
|
|
if (w <= 0 || w > MOO_SMOOI_MAX) goto einval;
|
|
|
|
moo->option.dfl_symtab_size = *(moo_oow_t*)value;
|
|
return 0;
|
|
}
|
|
|
|
case MOO_SYSDIC_SIZE:
|
|
{
|
|
moo_oow_t w;
|
|
|
|
w = *(moo_oow_t*)value;
|
|
if (w <= 0 || w > MOO_SMOOI_MAX) goto einval;
|
|
|
|
moo->option.dfl_sysdic_size = *(moo_oow_t*)value;
|
|
return 0;
|
|
}
|
|
|
|
case MOO_PROCSTK_SIZE:
|
|
{
|
|
moo_oow_t w;
|
|
|
|
w = *(moo_oow_t*)value;
|
|
if (w <= 0 || w > MOO_SMOOI_MAX) goto einval;
|
|
|
|
moo->option.dfl_procstk_size = *(moo_oow_t*)value;
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
einval:
|
|
moo->errnum = MOO_EINVAL;
|
|
return -1;
|
|
}
|
|
|
|
int moo_getoption (moo_t* moo, moo_option_t id, void* value)
|
|
{
|
|
switch (id)
|
|
{
|
|
case MOO_TRAIT:
|
|
*(unsigned int*)value = moo->option.trait;
|
|
return 0;
|
|
|
|
case MOO_LOG_MASK:
|
|
*(unsigned int*)value = moo->option.log_mask;
|
|
return 0;
|
|
|
|
case MOO_SYMTAB_SIZE:
|
|
*(moo_oow_t*)value = moo->option.dfl_symtab_size;
|
|
return 0;
|
|
|
|
case MOO_SYSDIC_SIZE:
|
|
*(moo_oow_t*)value = moo->option.dfl_sysdic_size;
|
|
return 0;
|
|
|
|
case MOO_PROCSTK_SIZE:
|
|
*(moo_oow_t*)value = moo->option.dfl_procstk_size;
|
|
return 0;
|
|
};
|
|
|
|
moo->errnum = MOO_EINVAL;
|
|
return -1;
|
|
}
|
|
|
|
moo_cb_t* moo_regcb (moo_t* moo, moo_cb_t* tmpl)
|
|
{
|
|
moo_cb_t* actual;
|
|
|
|
actual = moo_allocmem (moo, MOO_SIZEOF(*actual));
|
|
if (!actual) return MOO_NULL;
|
|
|
|
*actual = *tmpl;
|
|
actual->next = moo->cblist;
|
|
actual->prev = MOO_NULL;
|
|
moo->cblist = actual;
|
|
|
|
return actual;
|
|
}
|
|
|
|
void moo_deregcb (moo_t* moo, moo_cb_t* cb)
|
|
{
|
|
if (cb == moo->cblist)
|
|
{
|
|
moo->cblist = moo->cblist->next;
|
|
if (moo->cblist) moo->cblist->prev = MOO_NULL;
|
|
}
|
|
else
|
|
{
|
|
if (cb->next) cb->next->prev = cb->prev;
|
|
if (cb->prev) cb->prev->next = cb->next;
|
|
}
|
|
|
|
moo_freemem (moo, cb);
|
|
}
|
|
|
|
void* moo_allocmem (moo_t* moo, moo_oow_t size)
|
|
{
|
|
void* ptr;
|
|
|
|
ptr = MOO_MMGR_ALLOC (moo->mmgr, size);
|
|
if (!ptr) moo->errnum = MOO_ESYSMEM;
|
|
return ptr;
|
|
}
|
|
|
|
void* moo_callocmem (moo_t* moo, moo_oow_t size)
|
|
{
|
|
void* ptr;
|
|
|
|
ptr = MOO_MMGR_ALLOC (moo->mmgr, size);
|
|
if (!ptr) moo->errnum = MOO_ESYSMEM;
|
|
else MOO_MEMSET (ptr, 0, size);
|
|
return ptr;
|
|
}
|
|
|
|
void* moo_reallocmem (moo_t* moo, void* ptr, moo_oow_t size)
|
|
{
|
|
ptr = MOO_MMGR_REALLOC (moo->mmgr, ptr, size);
|
|
if (!ptr) moo->errnum = MOO_ESYSMEM;
|
|
return ptr;
|
|
}
|
|
|
|
void moo_freemem (moo_t* moo, void* ptr)
|
|
{
|
|
MOO_MMGR_FREE (moo->mmgr, ptr);
|
|
}
|
|
|
|
/* -------------------------------------------------------------------------- */
|
|
|
|
#define MOD_PREFIX "moo_mod_"
|
|
#define MOD_PREFIX_LEN 9
|
|
|
|
#if defined(MOO_ENABLE_STATIC_MODULE)
|
|
|
|
#include "../mod/console.h"
|
|
#include "../mod/_stdio.h"
|
|
|
|
static struct
|
|
{
|
|
moo_bch_t* modname;
|
|
int (*modload) (moo_t* moo, moo_mod_t* mod);
|
|
}
|
|
static_modtab[] =
|
|
{
|
|
{ "console", moo_mod_console },
|
|
{ "stdio", moo_mod_stdio },
|
|
};
|
|
#endif
|
|
|
|
moo_mod_data_t* moo_openmod (moo_t* moo, const moo_ooch_t* name, moo_oow_t namelen)
|
|
{
|
|
moo_rbt_pair_t* pair;
|
|
moo_mod_data_t* mdp;
|
|
moo_mod_data_t md;
|
|
moo_mod_load_t load = MOO_NULL;
|
|
#if defined(MOO_ENABLE_STATIC_MODULE)
|
|
int n;
|
|
#endif
|
|
|
|
/* maximum module name length is MOO_MOD_NAME_LEN_MAX.
|
|
* MOD_PREFIX_LEN for MOD_PREFIX
|
|
* 1 for _ at the end when moo_mod_xxx_ is attempted.
|
|
* 1 for the terminating '\0'.
|
|
*/
|
|
moo_ooch_t buf[MOD_PREFIX_LEN + MOO_MOD_NAME_LEN_MAX + 1 + 1];
|
|
|
|
/* copy instead of encoding conversion. MOD_PREFIX must not
|
|
* include a character that requires encoding conversion.
|
|
* note the terminating null isn't needed in buf here. */
|
|
moo_copybctooochars (buf, MOD_PREFIX, MOD_PREFIX_LEN);
|
|
|
|
if (namelen > MOO_COUNTOF(buf) - (MOD_PREFIX_LEN + 1 + 1))
|
|
{
|
|
/* module name too long */
|
|
moo->errnum = MOO_EINVAL; /* TODO: change the error number to something more specific */
|
|
return MOO_NULL;
|
|
}
|
|
|
|
moo_copyoochars (&buf[MOD_PREFIX_LEN], name, namelen);
|
|
buf[MOD_PREFIX_LEN + namelen] = '\0';
|
|
|
|
#if defined(MOO_ENABLE_STATIC_MODULE)
|
|
/* attempt to find a statically linked module */
|
|
|
|
/*TODO: CHANGE THIS PART */
|
|
|
|
/* TODO: binary search ... */
|
|
for (n = 0; n < MOO_COUNTOF(static_modtab); n++)
|
|
{
|
|
if (moo_compoocharsbcstr (name, namelen, static_modtab[n].modname) == 0)
|
|
{
|
|
load = static_modtab[n].modload;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (n >= MOO_COUNTOF(static_modtab))
|
|
{
|
|
moo->errnum = MOO_ENOENT;
|
|
return MOO_NULL;
|
|
}
|
|
|
|
if (load)
|
|
{
|
|
/* found the module in the staic module table */
|
|
|
|
MOO_MEMSET (&md, 0, MOO_SIZEOF(md));
|
|
moo_copyoochars ((moo_ooch_t*)md.mod.name, name, namelen);
|
|
/* Note md.handle is MOO_NULL for a static module */
|
|
|
|
/* i copy-insert 'md' into the table before calling 'load'.
|
|
* to pass the same address to load(), query(), etc */
|
|
pair = moo_rbt_insert (&moo->modtab, (moo_ooch_t*)name, namelen, &md, MOO_SIZEOF(md));
|
|
if (pair == MOO_NULL)
|
|
{
|
|
moo->errnum = MOO_ESYSMEM;
|
|
return MOO_NULL;
|
|
}
|
|
|
|
mdp = (moo_mod_data_t*)MOO_RBT_VPTR(pair);
|
|
if (load (moo, &mdp->mod) <= -1)
|
|
{
|
|
moo_rbt_delete (&moo->modtab, (moo_ooch_t*)name, namelen);
|
|
return MOO_NULL;
|
|
}
|
|
|
|
return mdp;
|
|
}
|
|
#endif
|
|
|
|
/* attempt to find an external module */
|
|
MOO_MEMSET (&md, 0, MOO_SIZEOF(md));
|
|
moo_copyoochars ((moo_ooch_t*)md.mod.name, name, namelen);
|
|
if (moo->vmprim.dl_open && moo->vmprim.dl_getsym && moo->vmprim.dl_close)
|
|
{
|
|
md.handle = moo->vmprim.dl_open (moo, &buf[MOD_PREFIX_LEN]);
|
|
}
|
|
|
|
if (md.handle == MOO_NULL)
|
|
{
|
|
MOO_DEBUG2 (moo, "Cannot open a module [%.*js]\n", namelen, name);
|
|
moo->errnum = MOO_ENOENT; /* TODO: be more descriptive about the error */
|
|
return MOO_NULL;
|
|
}
|
|
|
|
/* attempt to get moo_mod_xxx where xxx is the module name*/
|
|
load = moo->vmprim.dl_getsym (moo, md.handle, buf);
|
|
if (!load)
|
|
{
|
|
MOO_DEBUG3 (moo, "Cannot get a module symbol [%js] in [%.*js]\n", buf, namelen, name);
|
|
moo->errnum = MOO_ENOENT; /* TODO: be more descriptive about the error */
|
|
moo->vmprim.dl_close (moo, md.handle);
|
|
return MOO_NULL;
|
|
}
|
|
|
|
/* i copy-insert 'md' into the table before calling 'load'.
|
|
* to pass the same address to load(), query(), etc */
|
|
pair = moo_rbt_insert (&moo->modtab, (void*)name, namelen, &md, MOO_SIZEOF(md));
|
|
if (pair == MOO_NULL)
|
|
{
|
|
MOO_DEBUG2 (moo, "Cannot register a module [%.*js]\n", namelen, name);
|
|
moo->errnum = MOO_ESYSMEM;
|
|
moo->vmprim.dl_close (moo, md.handle);
|
|
return MOO_NULL;
|
|
}
|
|
|
|
mdp = (moo_mod_data_t*)MOO_RBT_VPTR(pair);
|
|
if (load (moo, &mdp->mod) <= -1)
|
|
{
|
|
MOO_DEBUG3 (moo, "Module function [%js] returned failure in [%.*js]\n", buf, namelen, name);
|
|
moo->errnum = MOO_ENOENT; /* TODO: proper error code and handling */
|
|
moo_rbt_delete (&moo->modtab, name, namelen);
|
|
moo->vmprim.dl_close (moo, mdp->handle);
|
|
return MOO_NULL;
|
|
}
|
|
|
|
mdp->pair = pair;
|
|
|
|
MOO_DEBUG2 (moo, "Opened a module [%js] - %p\n", mdp->mod.name, mdp->handle);
|
|
|
|
/* the module loader must ensure to set a proper query handler */
|
|
MOO_ASSERT (moo, mdp->mod.query != MOO_NULL);
|
|
|
|
return mdp;
|
|
}
|
|
|
|
void moo_closemod (moo_t* moo, moo_mod_data_t* mdp)
|
|
{
|
|
if (mdp->mod.unload) mdp->mod.unload (moo, &mdp->mod);
|
|
|
|
if (mdp->handle)
|
|
{
|
|
moo->vmprim.dl_close (moo, mdp->handle);
|
|
MOO_DEBUG2 (moo, "Closed a module [%js] - %p\n", mdp->mod.name, mdp->handle);
|
|
mdp->handle = MOO_NULL;
|
|
}
|
|
|
|
if (mdp->pair)
|
|
{
|
|
/*mdp->pair = MOO_NULL;*/ /* this reset isn't needed as the area will get freed by moo_rbt_delete()) */
|
|
moo_rbt_delete (&moo->modtab, mdp->mod.name, moo_countoocstr(mdp->mod.name));
|
|
}
|
|
}
|
|
|
|
int moo_importmod (moo_t* moo, moo_oop_t _class, const moo_ooch_t* name, moo_oow_t len)
|
|
{
|
|
moo_rbt_pair_t* pair;
|
|
moo_mod_data_t* mdp;
|
|
int r = 0;
|
|
|
|
/* moo_openmod(), moo_closemod(), etc calls a user-defined callback.
|
|
* i need to protect _class in case the user-defined callback allocates
|
|
* a OOP memory chunk and GC occurs */
|
|
moo_pushtmp (moo, &_class);
|
|
|
|
pair = moo_rbt_search (&moo->modtab, name, len);
|
|
if (pair)
|
|
{
|
|
mdp = (moo_mod_data_t*)MOO_RBT_VPTR(pair);
|
|
MOO_ASSERT (moo, mdp != MOO_NULL);
|
|
}
|
|
else
|
|
{
|
|
mdp = moo_openmod (moo, name, len);
|
|
if (!mdp)
|
|
{
|
|
r = -1;
|
|
goto done2;
|
|
}
|
|
}
|
|
|
|
if (!mdp->mod.import)
|
|
{
|
|
MOO_DEBUG1 (moo, "Cannot import module [%js] - importing not supported by the module\n", mdp->mod.name);
|
|
moo->errnum = MOO_ENOIMPL;
|
|
r = -1;
|
|
goto done;
|
|
}
|
|
|
|
if (mdp->mod.import (moo, &mdp->mod, _class) <= -1)
|
|
{
|
|
MOO_DEBUG1 (moo, "Cannot import module [%js] - module's import() returned failure\n", mdp->mod.name);
|
|
r = -1;
|
|
goto done;
|
|
}
|
|
|
|
done:
|
|
if (!pair)
|
|
{
|
|
/* close the module if it has been opened in this function. */
|
|
moo_closemod (moo, mdp);
|
|
}
|
|
|
|
done2:
|
|
moo_poptmp (moo);
|
|
return r;
|
|
}
|
|
|
|
moo_pfimpl_t moo_querymod (moo_t* moo, const moo_ooch_t* pfid, moo_oow_t pfidlen)
|
|
{
|
|
/* primitive function identifier
|
|
* _funcname
|
|
* modname_funcname
|
|
*/
|
|
moo_rbt_pair_t* pair;
|
|
moo_mod_data_t* mdp;
|
|
const moo_ooch_t* sep;
|
|
|
|
moo_oow_t mod_name_len;
|
|
moo_pfimpl_t handler;
|
|
|
|
sep = moo_findoochar (pfid, pfidlen, '.');
|
|
if (!sep)
|
|
{
|
|
/* i'm writing a conservative code here. the compiler should
|
|
* guarantee that an underscore is included in an primitive identifer.
|
|
* what if the compiler is broken? imagine a buggy compiler rewritten
|
|
* in moo itself? */
|
|
MOO_DEBUG2 (moo, "Internal error - no period in a primitive function identifier [%.*js] - buggy compiler?\n", pfidlen, pfid);
|
|
moo->errnum = MOO_EINTERN;
|
|
return MOO_NULL;
|
|
}
|
|
|
|
mod_name_len = sep - pfid;
|
|
|
|
pair = moo_rbt_search (&moo->modtab, pfid, mod_name_len);
|
|
if (pair)
|
|
{
|
|
mdp = (moo_mod_data_t*)MOO_RBT_VPTR(pair);
|
|
MOO_ASSERT (moo, mdp != MOO_NULL);
|
|
}
|
|
else
|
|
{
|
|
mdp = moo_openmod (moo, pfid, mod_name_len);
|
|
if (!mdp) return MOO_NULL;
|
|
}
|
|
|
|
if ((handler = mdp->mod.query (moo, &mdp->mod, sep + 1)) == MOO_NULL)
|
|
{
|
|
/* the primitive function is not found. keep the module open */
|
|
MOO_DEBUG2 (moo, "Cannot find a primitive function [%js] in a module [%js]\n", sep + 1, mdp->mod.name);
|
|
moo->errnum = MOO_ENOENT; /* TODO: proper error code and handling */
|
|
return MOO_NULL;
|
|
}
|
|
|
|
MOO_DEBUG3 (moo, "Found a primitive function [%js] in a module [%js] - %p\n", sep + 1, mdp->mod.name, handler);
|
|
return handler;
|
|
}
|
|
|
|
/* -------------------------------------------------------------------------- */
|
|
|
|
/* add a new primitive method */
|
|
int moo_genpfmethod (moo_t* moo, moo_mod_t* mod, moo_oop_t _class, moo_method_type_t type, const moo_ooch_t* mthname, int variadic, const moo_ooch_t* pfname)
|
|
{
|
|
/* NOTE: this function is a subset of add_compiled_method() in comp.c */
|
|
|
|
moo_oop_char_t mnsym, pfidsym;
|
|
moo_oop_method_t mth;
|
|
moo_oop_class_t cls;
|
|
moo_oow_t tmp_count = 0, i;
|
|
moo_ooi_t arg_count = 0;
|
|
moo_oocs_t cs;
|
|
moo_ooi_t preamble_flags = 0;
|
|
static moo_ooch_t dot[] = { '.', '\0' };
|
|
|
|
MOO_ASSERT (moo, MOO_CLASSOF(moo, _class) == moo->_class);
|
|
|
|
if (!pfname) pfname = mthname;
|
|
|
|
cls = (moo_oop_class_t)_class;
|
|
moo_pushtmp (moo, (moo_oop_t*)&cls); tmp_count++;
|
|
MOO_ASSERT (moo, MOO_CLASSOF(moo, (moo_oop_t)cls->mthdic[type]) == moo->_method_dictionary);
|
|
|
|
for (i = 0; mthname[i]; i++)
|
|
{
|
|
if (mthname[i] == ':')
|
|
{
|
|
if (variadic) goto oops_inval;
|
|
arg_count++;
|
|
}
|
|
}
|
|
/* TODO: check if name is a valid method name - more checks... */
|
|
/* TOOD: if the method name is a binary selector, it can still have an argument.. so the check below is invalid... */
|
|
if (arg_count > 0 && mthname[i - 1] != ':')
|
|
{
|
|
oops_inval:
|
|
MOO_DEBUG2 (moo, "Cannot generate primitive function method [%js] in [%O] - invalid name\n", mthname, cls->name);
|
|
moo->errnum = MOO_EINVAL;
|
|
goto oops;
|
|
}
|
|
|
|
cs.ptr = (moo_ooch_t*)mthname;
|
|
cs.len = i;
|
|
if (moo_lookupdic (moo, cls->mthdic[type], &cs) != MOO_NULL)
|
|
{
|
|
MOO_DEBUG2 (moo, "Cannot generate primitive function method [%js] in [%O] - duplicate\n", mthname, cls->name);
|
|
moo->errnum = MOO_EEXIST;
|
|
goto oops;
|
|
}
|
|
|
|
mnsym = (moo_oop_char_t)moo_makesymbol (moo, mthname, i);
|
|
if (!mnsym) goto oops;
|
|
moo_pushtmp (moo, (moo_oop_t*)&mnsym); tmp_count++;
|
|
|
|
/* compose a full primitive function identifier to VM's string buffer.
|
|
* pfid => mod->name + '.' + pfname */
|
|
if (moo_copyoocstrtosbuf(moo, mod->name, 0) <= -1 ||
|
|
moo_concatoocstrtosbuf(moo, dot, 0) <= -1 ||
|
|
moo_concatoocstrtosbuf(moo, pfname, 0) <= -1)
|
|
{
|
|
MOO_DEBUG2 (moo, "Cannot generate primitive function method [%js] in [%O] - VM memory shortage\n", mthname, cls->name);
|
|
return -1;
|
|
}
|
|
|
|
pfidsym = (moo_oop_char_t)moo_makesymbol (moo, moo->sbuf[0].ptr, moo->sbuf[0].len);
|
|
if (!pfidsym)
|
|
{
|
|
MOO_DEBUG2 (moo, "Cannot generate primitive function method [%js] in [%O] - symbol instantiation failure\n", mthname, cls->name);
|
|
goto oops;
|
|
}
|
|
moo_pushtmp (moo, (moo_oop_t*)&pfidsym); tmp_count++;
|
|
|
|
#if defined(MOO_USE_OBJECT_TRAILER)
|
|
mth = (moo_oop_method_t)moo_instantiatewithtrailer (moo, moo->_method, 1, MOO_NULL, 0);
|
|
#else
|
|
mth = (moo_oop_method_t)moo_instantiate (moo, moo->_method, MOO_NULL, 1);
|
|
#endif
|
|
if (!mth)
|
|
{
|
|
MOO_DEBUG2 (moo, "Cannot generate primitive function method [%js] in [%O] - method instantiation failure\n", mthname, cls->name);
|
|
goto oops;
|
|
}
|
|
|
|
/* store the primitive function name symbol to the literal frame */
|
|
mth->slot[0] = (moo_oop_t)pfidsym;
|
|
|
|
/* premable should contain the index to the literal frame which is always 0 */
|
|
mth->owner = cls;
|
|
mth->name = mnsym;
|
|
if (variadic) preamble_flags |= MOO_METHOD_PREAMBLE_FLAG_VARIADIC;
|
|
mth->preamble = MOO_SMOOI_TO_OOP(MOO_METHOD_MAKE_PREAMBLE(MOO_METHOD_PREAMBLE_NAMED_PRIMITIVE, 0, preamble_flags));
|
|
mth->preamble_data[0] = MOO_SMOOI_TO_OOP(0);
|
|
mth->preamble_data[1] = MOO_SMOOI_TO_OOP(0);
|
|
mth->tmpr_count = MOO_SMOOI_TO_OOP(arg_count);
|
|
mth->tmpr_nargs = MOO_SMOOI_TO_OOP(arg_count);
|
|
|
|
/* TODO: emit BCODE_RETURN_NIL ? */
|
|
|
|
if (!moo_putatdic (moo, cls->mthdic[type], (moo_oop_t)mnsym, (moo_oop_t)mth))
|
|
{
|
|
MOO_DEBUG2 (moo, "Cannot generate primitive function method [%js] in [%O] - failed to add to method dictionary\n", mthname, cls->name);
|
|
goto oops;
|
|
}
|
|
|
|
MOO_DEBUG2 (moo, "Generated primitive function method [%js] in [%O]\n", mthname, cls->name);
|
|
|
|
moo_poptmps (moo, tmp_count); tmp_count = 0;
|
|
return 0;
|
|
|
|
oops:
|
|
moo_poptmps (moo, tmp_count);
|
|
return -1;
|
|
}
|