X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/e515cb46d8c3c86b35a918cbb8e8f7860bf31d94..f45b6eaa01ced9382f2ec977555b66b55bc72fbf:/src/bindings/lua/simgrid_lua.c diff --git a/src/bindings/lua/simgrid_lua.c b/src/bindings/lua/simgrid_lua.c index fc62dcb78a..f3367cb597 100644 --- a/src/bindings/lua/simgrid_lua.c +++ b/src/bindings/lua/simgrid_lua.c @@ -21,43 +21,19 @@ XBT_LOG_NEW_DEFAULT_SUBCATEGORY(lua, bindings, "Lua Bindings"); static lua_State* sglua_maestro_state; +int luaopen_simgrid(lua_State *L); static void register_c_functions(lua_State *L); +static int run_lua_code(int argc, char **argv); -/** - * \brief Like luaL_checkudata, with additional debug logs. - * \param L a lua state - * \param ud index of the userdata to check in the stack - * \param tname key of the metatable of this userdata in the registry - */ -static void* my_checkudata(lua_State* L, int ud, const char* tname) { - - XBT_DEBUG("Checking the userdata: ud = %d", ud); - sglua_stack_dump("my_checkudata: ", L); - void* p = lua_touserdata(L, ud); - lua_getfield(L, LUA_REGISTRYINDEX, tname); - const void* correct_mt = lua_topointer(L, -1); - - int has_mt = lua_getmetatable(L, ud); - XBT_DEBUG("Checking the userdata: has metatable ? %d", has_mt); - const void* actual_mt = NULL; - if (has_mt) { - actual_mt = lua_topointer(L, -1); - lua_pop(L, 1); - } - XBT_DEBUG("Checking the task's metatable: expected %p, found %p", correct_mt, actual_mt); - sglua_stack_dump("my_checkudata: ", L); - - if (p == NULL || !lua_getmetatable(L, ud) || !lua_rawequal(L, -1, -2)) - luaL_typerror(L, ud, tname); - lua_pop(L, 2); - return p; -} +/* ********************************************************************************* */ +/* simgrid.task API */ +/* ********************************************************************************* */ /** - * @brief Ensures that a value is a valid task and returns it. - * @param L a Lua state - * @param index an index in the Lua stack - * @return the task at this index + * \brief Ensures that a value in the stack is a valid task and returns it. + * \param L a Lua state + * \param index an index in the Lua stack + * \return the C task corresponding to this Lua task */ static m_task_t sglua_checktask(lua_State* L, int index) { @@ -77,17 +53,20 @@ static m_task_t sglua_checktask(lua_State* L, int index) return task; } -/* ********************************************************************************* */ -/* wrapper functions */ -/* ********************************************************************************* */ - /** * \brief Creates a new task and leaves it onto the stack. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (string): name of the task + * - Argument 2 (number): computation size + * - Argument 3 (number): communication size + * - Return value (task): the task created * - * Lua arguments: - * - task name (string) - * - computation size (integer) - * - communication size (integer) + * A Lua task is a regular table with a full userdata inside, and both share + * the same metatable. For the regular table, the metatable allows OO-style + * writing such as your_task:send(someone). + * For the userdata, the metatable is used to check its type. */ static int l_task_new(lua_State* L) { @@ -102,6 +81,10 @@ static int l_task_new(lua_State* L) lua_newtable(L); /* task */ + luaL_getmetatable(L, TASK_MODULE_NAME); + /* task mt */ + lua_setmetatable(L, -2); + /* task */ m_task_t* lua_task = (m_task_t*) lua_newuserdata(L, sizeof(m_task_t)); /* task ctask */ *lua_task = msg_task; @@ -114,6 +97,14 @@ static int l_task_new(lua_State* L) return 1; } +/** + * \brief Returns the name of a task. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (task): a task + * - Return value (string): name of the task + */ static int l_task_get_name(lua_State* L) { m_task_t task = sglua_checktask(L, 1); @@ -121,21 +112,65 @@ static int l_task_get_name(lua_State* L) return 1; } -static int l_task_computation_duration(lua_State* L) +/** + * \brief Returns the computation duration of a task. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (task): a task + * - Return value (number): computation duration of this task + */ +static int l_task_get_computation_duration(lua_State* L) { m_task_t task = sglua_checktask(L, 1); lua_pushnumber(L, MSG_task_get_compute_duration(task)); return 1; } +/** + * \brief Executes a task. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (task): the task to execute + * - Return value (nil or error): none if the task was successfully executed, or an error + * string in case of failure, which may be "task canceled" or "host failure" + */ static int l_task_execute(lua_State* L) { m_task_t task = sglua_checktask(L, 1); - int res = MSG_task_execute(task); - lua_pushnumber(L, res); - return 1; + MSG_error_t res = MSG_task_execute(task); + + switch (res) { + + case MSG_OK: + return 0; + + case MSG_TASK_CANCELED: + lua_pushliteral(L, "task canceled"); + return 1; + + case MSG_HOST_FAILURE: + lua_pushliteral(L, "host failure"); + return 1; + + default: + xbt_die("Unexpected result of MSG_task_execute(): %d, please report this bug", res); + } } +/** + * \brief Sends a task to a mailbox and waits for its completion. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (task): the task to send + * - Argument 2 (string or compatible): mailbox name, as a real string or any + * type convertible to string (numbers always are) + * - Return value (nil or error): none if the communication was successful, + * or an error string in case of failure, which may be "timeout", + * "host failure" or "transfer failure" + */ static int l_task_send(lua_State* L) { m_task_t task = sglua_checktask(L, 1); @@ -151,92 +186,122 @@ static int l_task_send(lua_State* L) MSG_process_sleep(0); } - if (res == MSG_OK) { + switch (res) { + + case MSG_OK: /* the receiver is the owner of the task and may destroy it: * remove the C task on my side so that I don't garbage collect it */ lua_getfield(L, 1, "__simgrid_task"); /* task ctask */ m_task_t* udata = (m_task_t*) luaL_checkudata(L, -1, TASK_MODULE_NAME); *udata = NULL; - lua_pop(L, 1); - /* task */ + return 0; + + case MSG_TIMEOUT: + XBT_DEBUG("MSG_task_send failed: timeout"); + lua_settop(L, 0); + lua_pushliteral(L, "timeout"); + return 1; + + case MSG_TRANSFER_FAILURE: + XBT_DEBUG("MSG_task_send failed: transfer failure"); + lua_settop(L, 0); + lua_pushliteral(L, "transfer failure"); + return 1; + + case MSG_HOST_FAILURE: + XBT_DEBUG("MSG_task_send failed: host failure"); + lua_settop(L, 0); + lua_pushliteral(L, "host failure"); + return 1; + + default: + xbt_die("Unexpected result of MSG_task_send: %d, please report this bug", res); } - else { - switch (res) { - case MSG_TIMEOUT: - XBT_DEBUG("MSG_task_send failed : Timeout"); - break; - case MSG_TRANSFER_FAILURE: - XBT_DEBUG("MSG_task_send failed : Transfer Failure"); - break; - case MSG_HOST_FAILURE: - XBT_DEBUG("MSG_task_send failed : Host Failure "); - break; - default: - XBT_ERROR - ("MSG_task_send failed : Unexpected error , please report this bug"); - break; - } - } - return 0; } -static int l_task_recv_with_timeout(lua_State *L) +/** + * \brief Receives a task. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (string or compatible): mailbox name, as a real string or any + * type convertible to string (numbers always are) + * - Argument 2 (number, optional): timeout (default is no timeout) + * - Return value (task or nil+err): the task received, or nil plus an error message if + * the communication has failed + */ +static int l_task_recv(lua_State* L) { m_task_t task = NULL; const char* mailbox = luaL_checkstring(L, 1); - int timeout = luaL_checknumber(L, 2); + int timeout; + if (lua_gettop(L) >= 2) { /* mailbox timeout */ + timeout = luaL_checknumber(L, 2); + } + else { + /* mailbox */ + timeout = -1; + /* no timeout by default */ + } lua_settop(L, 0); /* -- */ MSG_error_t res = MSG_task_receive_with_timeout(&task, mailbox, timeout); - if (res == MSG_OK) { + switch (res) { + + case MSG_OK: /* copy the data directly from sender's stack */ + { lua_State* sender_stack = MSG_task_get_data(task); sglua_copy_value(sender_stack, L); /* task */ MSG_task_set_data(task, NULL); + return 1; } - else { - switch (res) { - case MSG_TIMEOUT: - XBT_DEBUG("MSG_task_receive failed : Timeout"); - break; - case MSG_TRANSFER_FAILURE: - XBT_DEBUG("MSG_task_receive failed : Transfer Failure"); - break; - case MSG_HOST_FAILURE: - XBT_DEBUG("MSG_task_receive failed : Host Failure "); - break; - default: - XBT_ERROR("MSG_task_receive failed : Unexpected error , please report this bug"); - break; - } + + case MSG_TIMEOUT: + XBT_DEBUG("MSG_task_send failed: timeout"); lua_pushnil(L); - /* nil */ - } - /* task/nil */ - return 1; -} + lua_pushliteral(L, "timeout"); + return 2; -static int l_task_recv(lua_State * L) -{ - lua_pushnumber(L, -1.0); - return l_task_recv_with_timeout(L); + case MSG_TRANSFER_FAILURE: + XBT_DEBUG("MSG_task_send failed: transfer failure"); + lua_pushnil(L); + lua_pushliteral(L, "transfer failure"); + return 2; + + case MSG_HOST_FAILURE: + XBT_DEBUG("MSG_task_send failed: host failure"); + lua_pushnil(L); + lua_pushliteral(L, "host failure"); + return 2; + + default: + xbt_die("Unexpected result of MSG_task_recv: %d, please report this bug", res); + } } static const luaL_reg task_functions[] = { {"new", l_task_new}, - {"name", l_task_get_name}, - {"computation_duration", l_task_computation_duration}, + {"get_name", l_task_get_name}, + {"get_computation_duration", l_task_get_computation_duration}, {"execute", l_task_execute}, {"send", l_task_send}, {"recv", l_task_recv}, - {"recv_timeout", l_task_recv_with_timeout}, {NULL, NULL} }; +/** + * \brief Finalizes the userdata of a task. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (userdata): a C task, possibly NULL if it was sent to another + * Lua state + */ static int l_task_gc(lua_State* L) { /* ctask */ @@ -248,21 +313,39 @@ static int l_task_gc(lua_State* L) return 0; } +/** + * \brief Returns a string representation of a C task. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (userdata): a task + * - Return value (string): a string describing this task + */ static int l_task_tostring(lua_State* L) { - m_task_t task = sglua_checktask(L, 1); + m_task_t task = *((m_task_t*) luaL_checkudata(L, 1, TASK_MODULE_NAME)); lua_pushfstring(L, "Task: %p", task); return 1; } +/** + * \brief Metamethods of both a task table and the userdata inside it. + */ static const luaL_reg task_meta[] = { - {"__gc", l_task_gc}, + {"__gc", l_task_gc}, /* will be called only for userdata */ {"__tostring", l_task_tostring}, {NULL, NULL} }; +/* ********************************************************************************* */ +/* simgrid.host API */ +/* ********************************************************************************* */ + /** - * Host + * \brief Ensures that a value in the stack is a host and returns it. + * \param L a Lua state + * \param index an index in the Lua stack + * \return the C host corresponding to this Lua host */ static m_host_t sglua_checkhost(lua_State * L, int index) { @@ -279,6 +362,14 @@ static m_host_t sglua_checkhost(lua_State * L, int index) return ht; } +/** + * \brief Returns a host given its name. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (string): name of a host + * - Return value (host): the corresponding host + */ static int l_host_get_by_name(lua_State * L) { const char *name = luaL_checkstring(L, 1); @@ -298,6 +389,14 @@ static int l_host_get_by_name(lua_State * L) return 1; } +/** + * \brief Returns the name of a host. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (host): a host + * - Return value (string): name of this host + */ static int l_host_get_name(lua_State * L) { m_host_t ht = sglua_checkhost(L, 1); @@ -305,12 +404,27 @@ static int l_host_get_name(lua_State * L) return 1; } +/** + * \brief Returns the number of existing hosts. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Return value (number): number of hosts + */ static int l_host_number(lua_State * L) { lua_pushnumber(L, MSG_get_host_number()); return 1; } +/** + * \brief Returns the host given its index. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (number): an index (1 is the first) + * - Return value (host): the host at this index + */ static int l_host_at(lua_State * L) { int index = luaL_checkinteger(L, 1); @@ -322,9 +436,15 @@ static int l_host_at(lua_State * L) lua_setmetatable(L, -2); lua_setfield(L, -2, "__simgrid_host"); /* put the userdata as field of the table */ return 1; - } +/** + * \brief Returns the host where the current process is located. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Return value (host): the current host + */ static int l_host_self(lua_State * L) { /* -- */ @@ -343,6 +463,15 @@ static int l_host_self(lua_State * L) return 1; } +/** + * \brief Returns the value of a host property. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (host): a host + * - Argument 2 (string): name of the property to get + * - Return value (string): the value of this property + */ static int l_host_get_property_value(lua_State * L) { m_host_t ht = sglua_checkhost(L, 1); @@ -351,20 +480,68 @@ static int l_host_get_property_value(lua_State * L) return 1; } +/** + * \brief Makes the current process sleep for a while. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (number): duration of the sleep + */ static int l_host_sleep(lua_State *L) { int time = luaL_checknumber(L, 1); MSG_process_sleep(time); - return 1; + return 0; } +/** + * \brief Destroys a host. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (host): the host to destroy + */ static int l_host_destroy(lua_State *L) { m_host_t ht = sglua_checkhost(L, 1); __MSG_host_destroy(ht); + return 0; +} + +static const luaL_reg host_functions[] = { + {"get_by_name", l_host_get_by_name}, + {"name", l_host_get_name}, + {"number", l_host_number}, + {"at", l_host_at}, + {"self", l_host_self}, + {"get_prop_value", l_host_get_property_value}, + {"sleep", l_host_sleep}, + {"destroy", l_host_destroy}, + // Bypass XML Methods + {"set_function", console_set_function}, + {"set_property", console_host_set_property}, + {NULL, NULL} +}; + +/** + * \brief Returns a string representation of a host. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (userdata): a host + * - Return value (string): a string describing this host + */ +static int l_host_tostring(lua_State * L) +{ + lua_pushfstring(L, "Host :%p", lua_touserdata(L, 1)); return 1; } +static const luaL_reg host_meta[] = { + {"__tostring", l_host_tostring}, + {0, 0} +}; + /* ********************************************************************************* */ /* lua_stub_generator functions */ /* ********************************************************************************* */ @@ -424,7 +601,6 @@ static int gras_add_process_function(lua_State * L) return 0; } - static int gras_generate(lua_State * L) { const char *project_name = luaL_checkstring(L, 1); @@ -434,41 +610,10 @@ static int gras_generate(lua_State * L) return 0; } +/* ********************************************************************************* */ +/* simgrid.platf API */ +/* ********************************************************************************* */ -// *********** Register Methods ******************************************* // - -/* - * Host Methods - */ -static const luaL_reg host_functions[] = { - {"get_by_name", l_host_get_by_name}, - {"name", l_host_get_name}, - {"number", l_host_number}, - {"at", l_host_at}, - {"self", l_host_self}, - {"get_prop_value", l_host_get_property_value}, - {"sleep", l_host_sleep}, - {"destroy", l_host_destroy}, - // Bypass XML Methods - {"set_function", console_set_function}, - {"set_property", console_host_set_property}, - {NULL, NULL} -}; - -static int l_host_tostring(lua_State * L) -{ - lua_pushfstring(L, "Host :%p", lua_touserdata(L, 1)); - return 1; -} - -static const luaL_reg host_meta[] = { - {"__tostring", l_host_tostring}, - {0, 0} -}; - -/* - * AS Methods - */ static const luaL_reg platf_functions[] = { {"open", console_open}, {"close", console_close}, @@ -481,96 +626,103 @@ static const luaL_reg platf_functions[] = { {NULL, NULL} }; -/* - * Environment related - */ +/* ********************************************************************************* */ +/* simgrid API */ +/* ********************************************************************************* */ /** - * @brief Runs a Lua function as a new simulated process. - * @param argc number of arguments of the function - * @param argv name of the Lua function and array of its arguments - * @return result of the function + * \brief Deploys your application. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (string): name of the deployment file to load */ -static int run_lua_code(int argc, char **argv) -{ - XBT_DEBUG("Run lua code %s", argv[0]); - - lua_State *L = sglua_clone_maestro(); - int res = 1; - - /* start the function */ - lua_getglobal(L, argv[0]); - xbt_assert(lua_isfunction(L, -1), - "The lua function %s does not seem to exist", argv[0]); - - /* push arguments onto the stack */ - int i; - for (i = 1; i < argc; i++) - lua_pushstring(L, argv[i]); - - /* call the function */ - _XBT_GNUC_UNUSED int err; - err = lua_pcall(L, argc - 1, 1, 0); - xbt_assert(err == 0, "error running function `%s': %s", argv[0], - lua_tostring(L, -1)); +static int launch_application(lua_State* L) { - /* retrieve result */ - if (lua_isnumber(L, -1)) { - res = lua_tonumber(L, -1); - lua_pop(L, 1); /* pop returned value */ - } - - XBT_DEBUG("Execution of Lua code %s is over", (argv ? argv[0] : "(null)")); - - return res; -} - -static int launch_application(lua_State * L) -{ - const char *file = luaL_checkstring(L, 1); + const char* file = luaL_checkstring(L, 1); MSG_function_register_default(run_lua_code); MSG_launch_application(file); return 0; } -static int create_environment(lua_State * L) -{ - const char *file = luaL_checkstring(L, 1); +/** + * \brief Creates the platform. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (string): name of the platform file to load + */ +static int create_environment(lua_State* L) { + + const char* file = luaL_checkstring(L, 1); XBT_DEBUG("Loading environment file %s", file); MSG_create_environment(file); return 0; } -static int debug(lua_State * L) -{ - const char *str = luaL_checkstring(L, 1); +/** + * \brief Prints a log string with debug level. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (string): the text to print + */ +static int debug(lua_State* L) { + + const char* str = luaL_checkstring(L, 1); XBT_DEBUG("%s", str); return 0; } -static int info(lua_State * L) -{ - const char *str = luaL_checkstring(L, 1); +/** + * \brief Prints a log string with info level. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Argument 1 (string): the text to print + */ +static int info(lua_State* L) { + + const char* str = luaL_checkstring(L, 1); XBT_INFO("%s", str); return 0; } -static int run(lua_State * L) -{ +/** + * \brief Runs your application. + * \param L a Lua state + * \return number of values returned to Lua + */ +static int run(lua_State* L) { + MSG_main(); return 0; } -static int clean(lua_State * L) +/** + * \brief Returns the current simulated time. + * \param L a Lua state + * \return number of values returned to Lua + * + * - Return value (number): the simulated time + */ +static int get_clock(lua_State* L) { + + lua_pushnumber(L, MSG_get_clock()); + return 1; +} + +/** + * \brief Cleans the simulation. + * \param L a Lua state + * \return number of values returned to Lua + */ +static int simgrid_gc(lua_State * L) { MSG_clean(); return 0; } -/* - * Bypass XML Parser (lua console) - */ - /* * Register platform for MSG */ @@ -634,7 +786,7 @@ static const luaL_Reg simgrid_functions[] = { {"debug", debug}, {"info", info}, {"run", run}, - {"clean", clean}, + {"get_clock", get_clock}, /* short names */ {"platform", create_environment}, {"application", launch_application}, @@ -651,17 +803,18 @@ static const luaL_Reg simgrid_functions[] = { }; /* ********************************************************************************* */ -/* module management functions */ +/* module management functions */ /* ********************************************************************************* */ #define LUA_MAX_ARGS_COUNT 10 /* maximum amount of arguments we can get from lua on command line */ -int luaopen_simgrid(lua_State *L); // Fuck gcc: we don't need that prototype - /** - * This function is called automatically by the Lua interpreter when some Lua code requires - * the "simgrid" module. - * @param L the Lua state + * \brief Opens the simgrid Lua module. + * + * This function is called automatically by the Lua interpreter when some + * Lua code requires the "simgrid" module. + * + * \param L the Lua state */ int luaopen_simgrid(lua_State *L) { @@ -715,57 +868,173 @@ int luaopen_simgrid(lua_State *L) } /** - * @brief Returns whether a Lua state is the maestro state. - * @param L a Lua state - * @return true if this is maestro + * \brief Returns whether a Lua state is the maestro state. + * \param L a Lua state + * \return true if this is maestro */ int sglua_is_maestro(lua_State* L) { return L == sglua_maestro_state; } /** - * @brief Returns the maestro state. - * @return true the maestro Lua state + * \brief Returns the maestro state. + * \return the maestro Lua state */ lua_State* sglua_get_maestro(void) { return sglua_maestro_state; } /** - * Makes the appropriate Simgrid functions available to the Lua world. - * @param L a Lua world + * \brief Registers the task functions into the table simgrid.task. + * + * Also initialize the metatable of the task userdata type. + * + * \param L a lua state */ -void register_c_functions(lua_State *L) { - - /* register the core C functions to lua */ - luaL_register(L, "simgrid", simgrid_functions); +static void register_task_functions(lua_State* L) +{ + /* create a table simgrid.task and fill it with task functions */ + luaL_openlib(L, TASK_MODULE_NAME, task_functions, 0); + /* simgrid.task */ + + /* create the metatable for tasks, add it to the Lua registry */ + luaL_newmetatable(L, TASK_MODULE_NAME); + /* simgrid.task mt */ + /* fill the metatable */ + luaL_openlib(L, NULL, task_meta, 0); + /* simgrid.task mt */ + lua_pushvalue(L, -2); + /* simgrid.task mt simgrid.task */ + /* metatable.__index = simgrid.task + * we put the task functions inside the task itself: + * this allows to write task:method(args) for + * simgrid.task.method(task, args) */ + lua_setfield(L, -2, "__index"); + /* simgrid.task mt */ + lua_pop(L, 2); + /* -- */ +} - /* register the task methods to lua */ - luaL_openlib(L, TASK_MODULE_NAME, task_functions, 0); // create methods table, add it to the globals - luaL_newmetatable(L, TASK_MODULE_NAME); // create metatable for Task, add it to the Lua registry - luaL_openlib(L, 0, task_meta, 0); // fill metatable - lua_pushliteral(L, "__index"); - lua_pushvalue(L, -3); // dup methods table - lua_rawset(L, -3); // metatable.__index = methods - lua_pushliteral(L, "__metatable"); - lua_pushvalue(L, -3); // dup methods table - lua_rawset(L, -3); // hide metatable:metatable.__metatable = methods - lua_pop(L, 1); // drop metatable - - /* register the hosts methods to lua */ +/** + * \brief Registers the host functions into the table simgrid.host. + * + * Also initialize the metatable of the host userdata type. + * + * \param L a lua state + */ +static void register_host_functions(lua_State* L) +{ + /* create a table simgrid.host and fill it with host functions */ luaL_openlib(L, HOST_MODULE_NAME, host_functions, 0); + /* simgrid.host */ + + /* create the metatable for host, add it to the Lua registry */ luaL_newmetatable(L, HOST_MODULE_NAME); - luaL_openlib(L, 0, host_meta, 0); - lua_pushliteral(L, "__index"); - lua_pushvalue(L, -3); - lua_rawset(L, -3); - lua_pushliteral(L, "__metatable"); - lua_pushvalue(L, -3); - lua_rawset(L, -3); - lua_pop(L, 1); + /* simgrid.host mt */ + /* fill the metatable */ + luaL_openlib(L, NULL, host_meta, 0); + /* simgrid.host mt */ + lua_pushvalue(L, -2); + /* simgrid.host mt simgrid.host */ + /* metatable.__index = simgrid.host + * we put the host functions inside the host userdata itself: + * this allows to write host(args) for + * simgrid.host.method(host, args) */ + // FIXME: cannot work currently, same problem as tasks + lua_setfield(L, -2, "__index"); + /* simgrid.host mt */ + lua_pop(L, 2); + /* -- */ +} - /* register the platform methods to lua */ +/** + * \brief Registers the platform functions into the table simgrid.platf. + * \param L a lua state + */ +static void register_platf_functions(lua_State* L) +{ luaL_openlib(L, PLATF_MODULE_NAME, platf_functions, 0); - luaL_newmetatable(L, PLATF_MODULE_NAME); + /* simgrid.platf */ + lua_pop(L, 1); +} + +/** + * \brief Makes the core functions available to the Lua world. + * \param L a Lua world + */ +static void register_core_functions(lua_State *L) +{ + /* register the core C functions to lua */ + luaL_register(L, "simgrid", simgrid_functions); + /* simgrid */ + + /* set a finalizer that cleans simgrid, by adding to the simgrid module a + * dummy userdata whose __gc metamethod calls MSG_clean() */ + lua_newuserdata(L, sizeof(void*)); + /* simgrid udata */ + lua_newtable(L); + /* simgrid udata mt */ + lua_pushcfunction(L, simgrid_gc); + /* simgrid udata mt simgrid_gc */ + lua_setfield(L, -2, "__gc"); + /* simgrid udata mt */ + lua_setmetatable(L, -2); + /* simgrid udata */ + lua_setfield(L, -2, "__simgrid_loaded"); + /* simgrid */ lua_pop(L, 1); + /* -- */ +} + +/** + * \brief Creates the simgrid module and make it available to Lua. + * \param L a Lua world + */ +static void register_c_functions(lua_State *L) +{ + register_core_functions(L); + register_task_functions(L); + register_host_functions(L); + register_platf_functions(L); +} + +/** + * \brief Runs a Lua function as a new simulated process. + * \param argc number of arguments of the function + * \param argv name of the Lua function and array of its arguments + * \return result of the function + */ +static int run_lua_code(int argc, char **argv) +{ + XBT_DEBUG("Run lua code %s", argv[0]); + + /* create a new state, getting globals from maestro */ + lua_State *L = sglua_clone_maestro(); + int res = 1; + + /* start the function */ + lua_getglobal(L, argv[0]); + xbt_assert(lua_isfunction(L, -1), + "There is no Lua function with name `%s'", argv[0]); + + /* push arguments onto the stack */ + int i; + for (i = 1; i < argc; i++) + lua_pushstring(L, argv[i]); + + /* call the function */ + _XBT_GNUC_UNUSED int err; + err = lua_pcall(L, argc - 1, 1, 0); + xbt_assert(err == 0, "Error running function `%s': %s", argv[0], + lua_tostring(L, -1)); + + /* retrieve result */ + if (lua_isnumber(L, -1)) { + res = lua_tonumber(L, -1); + lua_pop(L, 1); /* pop returned value */ + } + + XBT_DEBUG("Execution of Lua code %s is over", (argv ? argv[0] : "(null)")); + + return res; }