src/script/cpp_api/s_base.cpp

   1 /*
   2 Minetest
   3 Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>
   4
   5 This program is free software; you can redistribute it and/or modify
   6 it under the terms of the GNU Lesser General Public License as published by
   7 the Free Software Foundation; either version 2.1 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU Lesser General Public License for more details.
  14
  15 You should have received a copy of the GNU Lesser General Public License along
  16 with this program; if not, write to the Free Software Foundation, Inc.,
  17 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 */
  19
  20 #include "cpp_api/s_base.h"
  21 #include "cpp_api/s_internal.h"
  22 #include "cpp_api/s_security.h"
  23 #include "lua_api/l_object.h"
  24 #include "common/c_converter.h"
  25 #include "serverobject.h"
  26 #include "debug.h"
  27 #include "filesys.h"
  28 #include "log.h"
  29 #include "mods.h"
  30 #include "porting.h"
  31 #include "util/string.h"
  32
  33
  34 extern "C" {
  35 #include "lualib.h"
  36 #if USE_LUAJIT
  37         #include "luajit.h"
  38 #endif
  39 }
  40
  41 #include <stdio.h>
  42 #include <cstdarg>
  43
  44
  45 class ModNameStorer
  46 {
  47 private:
  48         lua_State *L;
  49 public:
  50         ModNameStorer(lua_State *L_, const std::string &mod_name):
  51                 L(L_)
  52         {
  53                 // Store current mod name in registry
  54                 lua_pushstring(L, mod_name.c_str());
  55                 lua_rawseti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_CURRENT_MOD_NAME);
  56         }
  57         ~ModNameStorer()
  58         {
  59                 // Clear current mod name from registry
  60                 lua_pushnil(L);
  61                 lua_rawseti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_CURRENT_MOD_NAME);
  62         }
  63 };
  64
  65
  66 /*
  67         ScriptApiBase
  68 */
  69
  70 ScriptApiBase::ScriptApiBase()
  71 {
  72 #ifdef SCRIPTAPI_LOCK_DEBUG
  73         m_locked = false;
  74 #endif
  75
  76         m_luastack = luaL_newstate();
  77         FATAL_ERROR_IF(!m_luastack, "luaL_newstate() failed");
  78
  79         luaL_openlibs(m_luastack);
  80
  81         // Make the ScriptApiBase* accessible to ModApiBase
  82         lua_pushlightuserdata(m_luastack, this);
  83         lua_rawseti(m_luastack, LUA_REGISTRYINDEX, CUSTOM_RIDX_SCRIPTAPI);
  84
  85         // Add and save an error handler
  86         lua_pushcfunction(m_luastack, script_error_handler);
  87         lua_rawseti(m_luastack, LUA_REGISTRYINDEX, CUSTOM_RIDX_ERROR_HANDLER);
  88
  89         // If we are using LuaJIT add a C++ wrapper function to catch
  90         // exceptions thrown in Lua -> C++ calls
  91 #if USE_LUAJIT
  92         lua_pushlightuserdata(m_luastack, (void*) script_exception_wrapper);
  93         luaJIT_setmode(m_luastack, -1, LUAJIT_MODE_WRAPCFUNC | LUAJIT_MODE_ON);
  94         lua_pop(m_luastack, 1);
  95 #endif
  96
  97         // Add basic globals
  98         lua_newtable(m_luastack);
  99         lua_setglobal(m_luastack, "core");
 100
 101         lua_pushstring(m_luastack, DIR_DELIM);
 102         lua_setglobal(m_luastack, "DIR_DELIM");
 103
 104         lua_pushstring(m_luastack, porting::getPlatformName());
 105         lua_setglobal(m_luastack, "PLATFORM");
 106
 107         // m_secure gets set to true inside
 108         // ScriptApiSecurity::initializeSecurity(), if neccessary.
 109         // Default to false otherwise
 110         m_secure = false;
 111
 112         m_server = NULL;
 113         m_environment = NULL;
 114         m_guiengine = NULL;
 115 }
 116
 117 ScriptApiBase::~ScriptApiBase()
 118 {
 119         lua_close(m_luastack);
 120 }
 121
 122 bool ScriptApiBase::loadMod(const std::string &script_path,
 123                 const std::string &mod_name, std::string *error)
 124 {
 125         ModNameStorer mod_name_storer(getStack(), mod_name);
 126
 127         return loadScript(script_path, error);
 128 }
 129
 130 bool ScriptApiBase::loadScript(const std::string &script_path, std::string *error)
 131 {
 132         verbosestream << "Loading and running script from " << script_path << std::endl;
 133
 134         lua_State *L = getStack();
 135
 136         int error_handler = PUSH_ERROR_HANDLER(L);
 137
 138         bool ok;
 139         if (m_secure) {
 140                 ok = ScriptApiSecurity::safeLoadFile(L, script_path.c_str());
 141         } else {
 142                 ok = !luaL_loadfile(L, script_path.c_str());
 143         }
 144         ok = ok && !lua_pcall(L, 0, 0, error_handler);
 145         if (!ok) {
 146                 std::string error_msg = lua_tostring(L, -1);
 147                 if (error)
 148                         *error = error_msg;
 149                 errorstream << "========== ERROR FROM LUA ===========" << std::endl
 150                         << "Failed to load and run script from " << std::endl
 151                         << script_path << ":" << std::endl << std::endl
 152                         << error_msg << std::endl << std::endl
 153                         << "======= END OF ERROR FROM LUA ========" << std::endl;
 154                 lua_pop(L, 1); // Pop error message from stack
 155         }
 156         lua_pop(L, 1); // Pop error handler
 157         return ok;
 158 }
 159
 160 // Push the list of callbacks (a lua table).
 161 // Then push nargs arguments.
 162 // Then call this function, which
 163 // - runs the callbacks
 164 // - replaces the table and arguments with the return value,
 165 //     computed depending on mode
 166 void ScriptApiBase::runCallbacksRaw(int nargs,
 167                 RunCallbacksMode mode, const char *fxn)
 168 {
 169         lua_State *L = getStack();
 170         FATAL_ERROR_IF(lua_gettop(L) < nargs + 1, "Not enough arguments");
 171
 172         // Insert error handler
 173         PUSH_ERROR_HANDLER(L);
 174         int error_handler = lua_gettop(L) - nargs - 1;
 175         lua_insert(L, error_handler);
 176
 177         // Insert run_callbacks between error handler and table
 178         lua_getglobal(L, "core");
 179         lua_getfield(L, -1, "run_callbacks");
 180         lua_remove(L, -2);
 181         lua_insert(L, error_handler + 1);
 182
 183         // Insert mode after table
 184         lua_pushnumber(L, (int)mode);
 185         lua_insert(L, error_handler + 3);
 186
 187         // Stack now looks like this:
 188         // ... <error handler> <run_callbacks> <table> <mode> <arg#1> <arg#2> ... <arg#n>
 189
 190         int result = lua_pcall(L, nargs + 2, 1, error_handler);
 191         if (result != 0)
 192                 scriptError(result, fxn);
 193
 194         lua_remove(L, error_handler);
 195 }
 196
 197 void ScriptApiBase::realityCheck()
 198 {
 199         int top = lua_gettop(m_luastack);
 200         if (top >= 30) {
 201                 dstream << "Stack is over 30:" << std::endl;
 202                 stackDump(dstream);
 203                 std::string traceback = script_get_backtrace(m_luastack);
 204                 throw LuaError("Stack is over 30 (reality check)\n" + traceback);
 205         }
 206 }
 207
 208 void ScriptApiBase::scriptError(int result, const char *fxn)
 209 {
 210         script_error(getStack(), result, m_last_run_mod.c_str(), fxn);
 211 }
 212
 213 void ScriptApiBase::stackDump(std::ostream &o)
 214 {
 215         int top = lua_gettop(m_luastack);
 216         for (int i = 1; i <= top; i++) {  /* repeat for each level */
 217                 int t = lua_type(m_luastack, i);
 218                 switch (t) {
 219                         case LUA_TSTRING:  /* strings */
 220                                 o << "\"" << lua_tostring(m_luastack, i) << "\"";
 221                                 break;
 222                         case LUA_TBOOLEAN:  /* booleans */
 223                                 o << (lua_toboolean(m_luastack, i) ? "true" : "false");
 224                                 break;
 225                         case LUA_TNUMBER:  /* numbers */ {
 226                                 char buf[10];
 227                                 snprintf(buf, 10, "%g", lua_tonumber(m_luastack, i));
 228                                 o << buf;
 229                                 break;
 230                         }
 231                         default:  /* other values */
 232                                 o << lua_typename(m_luastack, t);
 233                                 break;
 234                 }
 235                 o << " ";
 236         }
 237         o << std::endl;
 238 }
 239
 240 void ScriptApiBase::setOriginDirect(const char *origin)
 241 {
 242         m_last_run_mod = origin ? origin : "??";
 243 }
 244
 245 void ScriptApiBase::setOriginFromTableRaw(int index, const char *fxn)
 246 {
 247 #ifdef SCRIPTAPI_DEBUG
 248         lua_State *L = getStack();
 249
 250         m_last_run_mod = lua_istable(L, index) ?
 251                 getstringfield_default(L, index, "mod_origin", "") : "";
 252         //printf(">>>> running %s for mod: %s\n", fxn, m_last_run_mod.c_str());
 253 #endif
 254 }
 255
 256 void ScriptApiBase::addObjectReference(ServerActiveObject *cobj)
 257 {
 258         SCRIPTAPI_PRECHECKHEADER
 259         //infostream<<"scriptapi_add_object_reference: id="<<cobj->getId()<<std::endl;
 260
 261         // Create object on stack
 262         ObjectRef::create(L, cobj); // Puts ObjectRef (as userdata) on stack
 263         int object = lua_gettop(L);
 264
 265         // Get core.object_refs table
 266         lua_getglobal(L, "core");
 267         lua_getfield(L, -1, "object_refs");
 268         luaL_checktype(L, -1, LUA_TTABLE);
 269         int objectstable = lua_gettop(L);
 270
 271         // object_refs[id] = object
 272         lua_pushnumber(L, cobj->getId()); // Push id
 273         lua_pushvalue(L, object); // Copy object to top of stack
 274         lua_settable(L, objectstable);
 275 }
 276
 277 void ScriptApiBase::removeObjectReference(ServerActiveObject *cobj)
 278 {
 279         SCRIPTAPI_PRECHECKHEADER
 280         //infostream<<"scriptapi_rm_object_reference: id="<<cobj->getId()<<std::endl;
 281
 282         // Get core.object_refs table
 283         lua_getglobal(L, "core");
 284         lua_getfield(L, -1, "object_refs");
 285         luaL_checktype(L, -1, LUA_TTABLE);
 286         int objectstable = lua_gettop(L);
 287
 288         // Get object_refs[id]
 289         lua_pushnumber(L, cobj->getId()); // Push id
 290         lua_gettable(L, objectstable);
 291         // Set object reference to NULL
 292         ObjectRef::set_null(L);
 293         lua_pop(L, 1); // pop object
 294
 295         // Set object_refs[id] = nil
 296         lua_pushnumber(L, cobj->getId()); // Push id
 297         lua_pushnil(L);
 298         lua_settable(L, objectstable);
 299 }
 300
 301 // Creates a new anonymous reference if cobj=NULL or id=0
 302 void ScriptApiBase::objectrefGetOrCreate(lua_State *L,
 303                 ServerActiveObject *cobj)
 304 {
 305         if (cobj == NULL || cobj->getId() == 0) {
 306                 ObjectRef::create(L, cobj);
 307         } else {
 308                 objectrefGet(L, cobj->getId());
 309         }
 310 }
 311
 312 void ScriptApiBase::objectrefGet(lua_State *L, u16 id)
 313 {
 314         // Get core.object_refs[i]
 315         lua_getglobal(L, "core");
 316         lua_getfield(L, -1, "object_refs");
 317         luaL_checktype(L, -1, LUA_TTABLE);
 318         lua_pushnumber(L, id);
 319         lua_gettable(L, -2);
 320         lua_remove(L, -2); // object_refs
 321         lua_remove(L, -2); // core
 322 }