src/script/cpp_api/s_security.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_security.h"
  21
  22 #include "filesys.h"
  23 #include "porting.h"
  24 #include "server.h"
  25 #include "client/client.h"
  26 #include "settings.h"
  27
  28 #include <cerrno>
  29 #include <string>
  30 #include <iostream>
  31
  32
  33 #define SECURE_API(lib, name) \
  34         lua_pushcfunction(L, sl_##lib##_##name); \
  35         lua_setfield(L, -2, #name);
  36
  37
  38 static inline void copy_safe(lua_State *L, const char *list[], unsigned len, int from=-2, int to=-1)
  39 {
  40         if (from < 0) from = lua_gettop(L) + from + 1;
  41         if (to   < 0) to   = lua_gettop(L) + to   + 1;
  42         for (unsigned i = 0; i < (len / sizeof(list[0])); i++) {
  43                 lua_getfield(L, from, list[i]);
  44                 lua_setfield(L, to,   list[i]);
  45         }
  46 }
  47
  48 // Pushes the original version of a library function on the stack, from the old version
  49 static inline void push_original(lua_State *L, const char *lib, const char *func)
  50 {
  51         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_GLOBALS_BACKUP);
  52         lua_getfield(L, -1, lib);
  53         lua_remove(L, -2);  // Remove globals_backup
  54         lua_getfield(L, -1, func);
  55         lua_remove(L, -2);  // Remove lib
  56 }
  57
  58
  59 void ScriptApiSecurity::initializeSecurity()
  60 {
  61         static const char *whitelist[] = {
  62                 "assert",
  63                 "core",
  64                 "collectgarbage",
  65                 "DIR_DELIM",
  66                 "error",
  67                 "getfenv",
  68                 "getmetatable",
  69                 "ipairs",
  70                 "next",
  71                 "pairs",
  72                 "pcall",
  73                 "print",
  74                 "rawequal",
  75                 "rawget",
  76                 "rawset",
  77                 "select",
  78                 "setfenv",
  79                 "setmetatable",
  80                 "tonumber",
  81                 "tostring",
  82                 "type",
  83                 "unpack",
  84                 "_VERSION",
  85                 "xpcall",
  86                 // Completely safe libraries
  87                 "coroutine",
  88                 "string",
  89                 "table",
  90                 "math",
  91         };
  92         static const char *io_whitelist[] = {
  93                 "open",
  94                 "close",
  95                 "flush",
  96                 "read",
  97                 "type",
  98                 "write",
  99         };
 100         static const char *os_whitelist[] = {
 101                 "clock",
 102                 "date",
 103                 "difftime",
 104                 "getenv",
 105                 "setlocale",
 106                 "time",
 107                 "tmpname",
 108         };
 109         static const char *debug_whitelist[] = {
 110                 "gethook",
 111                 "traceback",
 112                 "getinfo",
 113                 "getmetatable",
 114                 "setupvalue",
 115                 "setmetatable",
 116                 "upvalueid",
 117                 "sethook",
 118                 "debug",
 119                 "setlocal",
 120         };
 121         static const char *package_whitelist[] = {
 122                 "config",
 123                 "cpath",
 124                 "path",
 125                 "searchpath",
 126         };
 127 #if USE_LUAJIT
 128         static const char *jit_whitelist[] = {
 129                 "arch",
 130                 "flush",
 131                 "off",
 132                 "on",
 133                 "opt",
 134                 "os",
 135                 "status",
 136                 "version",
 137                 "version_num",
 138         };
 139 #endif
 140         m_secure = true;
 141
 142         lua_State *L = getStack();
 143
 144         // Backup globals to the registry
 145         lua_getglobal(L, "_G");
 146         lua_rawseti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_GLOBALS_BACKUP);
 147
 148         // Replace the global environment with an empty one
 149         int thread = getThread(L);
 150         createEmptyEnv(L);
 151         setLuaEnv(L, thread);
 152
 153         // Get old globals
 154         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_GLOBALS_BACKUP);
 155         int old_globals = lua_gettop(L);
 156
 157
 158         // Copy safe base functions
 159         lua_getglobal(L, "_G");
 160         copy_safe(L, whitelist, sizeof(whitelist));
 161
 162         // And replace unsafe ones
 163         SECURE_API(g, dofile);
 164         SECURE_API(g, load);
 165         SECURE_API(g, loadfile);
 166         SECURE_API(g, loadstring);
 167         SECURE_API(g, require);
 168         lua_pop(L, 1);
 169
 170
 171         // Copy safe IO functions
 172         lua_getfield(L, old_globals, "io");
 173         lua_newtable(L);
 174         copy_safe(L, io_whitelist, sizeof(io_whitelist));
 175
 176         // And replace unsafe ones
 177         //SECURE_API(io, open);
 178         SECURE_API(io, input);
 179         SECURE_API(io, output);
 180         SECURE_API(io, lines);
 181
 182         lua_setglobal(L, "io");
 183         lua_pop(L, 1);  // Pop old IO
 184
 185
 186         // Copy safe OS functions
 187         lua_getfield(L, old_globals, "os");
 188         lua_newtable(L);
 189         copy_safe(L, os_whitelist, sizeof(os_whitelist));
 190
 191         // And replace unsafe ones
 192         SECURE_API(os, remove);
 193         SECURE_API(os, rename);
 194
 195         lua_setglobal(L, "os");
 196         lua_pop(L, 1);  // Pop old OS
 197
 198
 199         // Copy safe debug functions
 200         lua_getfield(L, old_globals, "debug");
 201         lua_newtable(L);
 202         copy_safe(L, debug_whitelist, sizeof(debug_whitelist));
 203         lua_setglobal(L, "debug");
 204         lua_pop(L, 1);  // Pop old debug
 205
 206
 207         // Copy safe package fields
 208         lua_getfield(L, old_globals, "package");
 209         lua_newtable(L);
 210         copy_safe(L, package_whitelist, sizeof(package_whitelist));
 211         lua_setglobal(L, "package");
 212         lua_pop(L, 1);  // Pop old package
 213
 214 #if USE_LUAJIT
 215         // Copy safe jit functions, if they exist
 216         lua_getfield(L, -1, "jit");
 217         if (!lua_isnil(L, -1)) {
 218                 lua_newtable(L);
 219                 copy_safe(L, jit_whitelist, sizeof(jit_whitelist));
 220                 lua_setglobal(L, "jit");
 221         }
 222         lua_pop(L, 1);  // Pop old jit
 223 #endif
 224
 225         lua_pop(L, 1); // Pop globals_backup
 226 }
 227
 228 void ScriptApiSecurity::initializeSecurityClient()
 229 {
 230         static const char *whitelist[] = {
 231                 "assert",
 232                 "core",
 233                 "collectgarbage",
 234                 "DIR_DELIM",
 235                 "error",
 236                 "getfenv",
 237                 "ipairs",
 238                 "next",
 239                 "pairs",
 240                 "pcall",
 241                 "print",
 242                 "rawequal",
 243                 "rawget",
 244                 "rawset",
 245                 "select",
 246                 "setfenv",
 247                 // getmetatable can be used to escape the sandbox
 248                 "setmetatable",
 249                 "tonumber",
 250                 "tostring",
 251                 "type",
 252                 "unpack",
 253                 "_VERSION",
 254                 "xpcall",
 255                 // Completely safe libraries
 256                 "coroutine",
 257                 "string",
 258                 "table",
 259                 "math",
 260         };
 261         static const char *os_whitelist[] = {
 262                 "clock",
 263                 "date",
 264                 "difftime",
 265                 "time"
 266         };
 267         static const char *debug_whitelist[] = {
 268                 "getinfo",
 269                 "traceback"
 270         };
 271 #if USE_LUAJIT
 272         static const char *jit_whitelist[] = {
 273                 "arch",
 274                 "flush",
 275                 "off",
 276                 "on",
 277                 "opt",
 278                 "os",
 279                 "status",
 280                 "version",
 281                 "version_num",
 282         };
 283 #endif
 284
 285         m_secure = true;
 286
 287         lua_State *L = getStack();
 288         int thread = getThread(L);
 289
 290         // Backup globals to the registry
 291         lua_getglobal(L, "_G");
 292         lua_rawseti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_GLOBALS_BACKUP);
 293
 294         // create an empty environment
 295         createEmptyEnv(L);
 296
 297         // Copy safe base functions
 298         lua_getglobal(L, "_G");
 299         lua_getfield(L, -2, "_G");
 300         copy_safe(L, whitelist, sizeof(whitelist));
 301
 302         // And replace unsafe ones
 303         SECURE_API(g, dofile);
 304         SECURE_API(g, load);
 305         SECURE_API(g, loadfile);
 306         SECURE_API(g, loadstring);
 307         SECURE_API(g, require);
 308         lua_pop(L, 2);
 309
 310         // Copy safe OS functions
 311         lua_getglobal(L, "os");
 312         lua_newtable(L);
 313         copy_safe(L, os_whitelist, sizeof(os_whitelist));
 314         lua_setfield(L, -3, "os");
 315         lua_pop(L, 1);  // Pop old OS
 316
 317
 318         // Copy safe debug functions
 319         lua_getglobal(L, "debug");
 320         lua_newtable(L);
 321         copy_safe(L, debug_whitelist, sizeof(debug_whitelist));
 322         lua_setfield(L, -3, "debug");
 323         lua_pop(L, 1);  // Pop old debug
 324
 325
 326 #if USE_LUAJIT
 327         // Copy safe jit functions, if they exist
 328         lua_getglobal(L, "jit");
 329         lua_newtable(L);
 330         copy_safe(L, jit_whitelist, sizeof(jit_whitelist));
 331         lua_setfield(L, -3, "jit");
 332         lua_pop(L, 1);  // Pop old jit
 333 #endif
 334
 335         // Set the environment to the one we created earlier
 336         setLuaEnv(L, thread);
 337 }
 338
 339 int ScriptApiSecurity::getThread(lua_State *L)
 340 {
 341 #if LUA_VERSION_NUM <= 501
 342         int is_main = lua_pushthread(L);  // Push the main thread
 343         FATAL_ERROR_IF(!is_main, "Security: ScriptApi's Lua state "
 344                 "isn't the main Lua thread!");
 345         return lua_gettop(L);
 346 #endif
 347         return 0;
 348 }
 349
 350 void ScriptApiSecurity::createEmptyEnv(lua_State *L)
 351 {
 352         lua_newtable(L);  // Create new environment
 353         lua_pushvalue(L, -1);
 354         lua_setfield(L, -2, "_G");  // Create the _G loop
 355 }
 356
 357 void ScriptApiSecurity::setLuaEnv(lua_State *L, int thread)
 358 {
 359 #if LUA_VERSION_NUM >= 502  // Lua >= 5.2
 360         // Set the global environment
 361         lua_rawseti(L, LUA_REGISTRYINDEX, LUA_RIDX_GLOBALS);
 362 #else  // Lua <= 5.1
 363         // Set the environment of the main thread
 364         FATAL_ERROR_IF(!lua_setfenv(L, thread), "Security: Unable to set "
 365                 "environment of the main Lua thread!");
 366         lua_pop(L, 1);  // Pop thread
 367 #endif
 368 }
 369
 370 bool ScriptApiSecurity::isSecure(lua_State *L)
 371 {
 372         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_GLOBALS_BACKUP);
 373         bool secure = !lua_isnil(L, -1);
 374         lua_pop(L, 1);
 375         return secure;
 376 }
 377
 378 bool ScriptApiSecurity::safeLoadString(lua_State *L, const std::string &code, const char *chunk_name)
 379 {
 380         if (code.size() > 0 && code[0] == LUA_SIGNATURE[0]) {
 381                 lua_pushliteral(L, "Bytecode prohibited when mod security is enabled.");
 382                 return false;
 383         }
 384         if (luaL_loadbuffer(L, code.data(), code.size(), chunk_name))
 385                 return false;
 386         return true;
 387 }
 388
 389 bool ScriptApiSecurity::safeLoadFile(lua_State *L, const char *path, const char *display_name)
 390 {
 391         FILE *fp;
 392         char *chunk_name;
 393         if (!display_name)
 394                 display_name = path;
 395         if (!path) {
 396                 fp = stdin;
 397                 chunk_name = const_cast<char *>("=stdin");
 398         } else {
 399                 fp = fopen(path, "rb");
 400                 if (!fp) {
 401                         lua_pushfstring(L, "%s: %s", path, strerror(errno));
 402                         return false;
 403                 }
 404                 size_t len = strlen(display_name) + 2;
 405                 chunk_name = new char[len];
 406                 snprintf(chunk_name, len, "@%s", display_name);
 407         }
 408
 409         size_t start = 0;
 410         int c = std::getc(fp);
 411         if (c == '#') {
 412                 // Skip the first line
 413                 while ((c = std::getc(fp)) != EOF && c != '\n') {}
 414                 if (c == '\n')
 415                         std::getc(fp);
 416                 start = std::ftell(fp);
 417         }
 418
 419         // Read the file
 420         int ret = std::fseek(fp, 0, SEEK_END);
 421         if (ret) {
 422                 lua_pushfstring(L, "%s: %s", path, strerror(errno));
 423                 if (path) {
 424                         std::fclose(fp);
 425                         delete [] chunk_name;
 426                 }
 427                 return false;
 428         }
 429
 430         size_t size = std::ftell(fp) - start;
 431         std::string code(size, '\0');
 432         ret = std::fseek(fp, start, SEEK_SET);
 433         if (ret) {
 434                 lua_pushfstring(L, "%s: %s", path, strerror(errno));
 435                 if (path) {
 436                         std::fclose(fp);
 437                         delete [] chunk_name;
 438                 }
 439                 return false;
 440         }
 441
 442         size_t num_read = std::fread(&code[0], 1, size, fp);
 443         if (path)
 444                 std::fclose(fp);
 445         if (num_read != size) {
 446                 lua_pushliteral(L, "Error reading file to load.");
 447                 if (path)
 448                         delete [] chunk_name;
 449                 return false;
 450         }
 451
 452         bool result = safeLoadString(L, code, chunk_name);
 453         if (path)
 454                 delete [] chunk_name;
 455         return result;
 456 }
 457
 458
 459 bool ScriptApiSecurity::checkPath(lua_State *L, const char *path,
 460                 bool write_required, bool *write_allowed)
 461 {
 462         if (write_allowed)
 463                 *write_allowed = false;
 464
 465         std::string str;  // Transient
 466
 467         std::string abs_path = fs::AbsolutePath(path);
 468
 469         if (!abs_path.empty()) {
 470                 // Don't allow accessing the settings file
 471                 str = fs::AbsolutePath(g_settings_path);
 472                 if (str == abs_path) return false;
 473         }
 474
 475         // If we couldn't find the absolute path (path doesn't exist) then
 476         // try removing the last components until it works (to allow
 477         // non-existent files/folders for mkdir).
 478         std::string cur_path = path;
 479         std::string removed;
 480         while (abs_path.empty() && !cur_path.empty()) {
 481                 std::string component;
 482                 cur_path = fs::RemoveLastPathComponent(cur_path, &component);
 483                 if (component == "..") {
 484                         // Parent components can't be allowed or we could allow something like
 485                         // /home/user/minetest/worlds/foo/noexist/../../../../../../etc/passwd.
 486                         // If we have previous non-relative elements in the path we might be
 487                         // able to remove them so that things like worlds/foo/noexist/../auth.txt
 488                         // could be allowed, but those paths will be interpreted as nonexistent
 489                         // by the operating system anyways.
 490                         return false;
 491                 }
 492                 removed.append(component).append(removed.empty() ? "" : DIR_DELIM + removed);
 493                 abs_path = fs::AbsolutePath(cur_path);
 494         }
 495         if (abs_path.empty())
 496                 return false;
 497         // Add the removed parts back so that you can't, eg, create a
 498         // directory in worldmods if worldmods doesn't exist.
 499         if (!removed.empty())
 500                 abs_path += DIR_DELIM + removed;
 501
 502         // Get server from registry
 503         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_SCRIPTAPI);
 504         ScriptApiBase *script;
 505 #if INDIRECT_SCRIPTAPI_RIDX
 506         script = (ScriptApiBase *) *(void**)(lua_touserdata(L, -1));
 507 #else
 508         script = (ScriptApiBase *) lua_touserdata(L, -1);
 509 #endif
 510         lua_pop(L, 1);
 511         const IGameDef *gamedef = script->getGameDef();
 512         if (!gamedef)
 513                 return false;
 514
 515         // Get mod name
 516         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_CURRENT_MOD_NAME);
 517         if (lua_isstring(L, -1)) {
 518                 std::string mod_name = readParam<std::string>(L, -1);
 519
 520                 // Builtin can access anything
 521                 if (mod_name == BUILTIN_MOD_NAME) {
 522                         if (write_allowed) *write_allowed = true;
 523                         return true;
 524                 }
 525
 526                 // Allow paths in mod path
 527                 // Don't bother if write access isn't important, since it will be handled later
 528                 if (write_required || write_allowed != NULL) {
 529                         const ModSpec *mod = gamedef->getModSpec(mod_name);
 530                         if (mod) {
 531                                 str = fs::AbsolutePath(mod->path);
 532                                 if (!str.empty() && fs::PathStartsWith(abs_path, str)) {
 533                                         if (write_allowed) *write_allowed = true;
 534                                         return true;
 535                                 }
 536                         }
 537                 }
 538         }
 539         lua_pop(L, 1);  // Pop mod name
 540
 541         // Allow read-only access to all mod directories
 542         if (!write_required) {
 543                 const std::vector<ModSpec> &mods = gamedef->getMods();
 544                 for (const ModSpec &mod : mods) {
 545                         str = fs::AbsolutePath(mod.path);
 546                         if (!str.empty() && fs::PathStartsWith(abs_path, str)) {
 547                                 return true;
 548                         }
 549                 }
 550         }
 551
 552         str = fs::AbsolutePath(gamedef->getWorldPath());
 553         if (!str.empty()) {
 554                 // Don't allow access to other paths in the world mod/game path.
 555                 // These have to be blocked so you can't override a trusted mod
 556                 // by creating a mod with the same name in a world mod directory.
 557                 // We add to the absolute path of the world instead of getting
 558                 // the absolute paths directly because that won't work if they
 559                 // don't exist.
 560                 if (fs::PathStartsWith(abs_path, str + DIR_DELIM + "worldmods") ||
 561                                 fs::PathStartsWith(abs_path, str + DIR_DELIM + "game")) {
 562                         return false;
 563                 }
 564                 // Allow all other paths in world path
 565                 if (fs::PathStartsWith(abs_path, str)) {
 566                         if (write_allowed) *write_allowed = true;
 567                         return true;
 568                 }
 569         }
 570
 571         // Default to disallowing
 572         return false;
 573 }
 574
 575
 576 int ScriptApiSecurity::sl_g_dofile(lua_State *L)
 577 {
 578         int nret = sl_g_loadfile(L);
 579         if (nret != 1) {
 580                 lua_error(L);
 581                 // code after this function isn't executed
 582         }
 583         int top_precall = lua_gettop(L);
 584         lua_call(L, 0, LUA_MULTRET);
 585         // Return number of arguments returned by the function,
 586         // adjusting for the function being poped.
 587         return lua_gettop(L) - (top_precall - 1);
 588 }
 589
 590
 591 int ScriptApiSecurity::sl_g_load(lua_State *L)
 592 {
 593         size_t len;
 594         const char *buf;
 595         std::string code;
 596         const char *chunk_name = "=(load)";
 597
 598         luaL_checktype(L, 1, LUA_TFUNCTION);
 599         if (!lua_isnone(L, 2)) {
 600                 luaL_checktype(L, 2, LUA_TSTRING);
 601                 chunk_name = lua_tostring(L, 2);
 602         }
 603
 604         while (true) {
 605                 lua_pushvalue(L, 1);
 606                 lua_call(L, 0, 1);
 607                 int t = lua_type(L, -1);
 608                 if (t == LUA_TNIL) {
 609                         break;
 610                 }
 611
 612                 if (t != LUA_TSTRING) {
 613                         lua_pushnil(L);
 614                         lua_pushliteral(L, "Loader didn't return a string");
 615                         return 2;
 616                 }
 617                 buf = lua_tolstring(L, -1, &len);
 618                 code += std::string(buf, len);
 619                 lua_pop(L, 1); // Pop return value
 620         }
 621         if (!safeLoadString(L, code, chunk_name)) {
 622                 lua_pushnil(L);
 623                 lua_insert(L, -2);
 624                 return 2;
 625         }
 626         return 1;
 627 }
 628
 629
 630 int ScriptApiSecurity::sl_g_loadfile(lua_State *L)
 631 {
 632 #ifndef SERVER
 633         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_SCRIPTAPI);
 634 #if INDIRECT_SCRIPTAPI_RIDX
 635         ScriptApiBase *script = (ScriptApiBase *) *(void**)(lua_touserdata(L, -1));
 636 #else
 637         ScriptApiBase *script = (ScriptApiBase *) lua_touserdata(L, -1);
 638 #endif
 639         lua_pop(L, 1);
 640
 641         // Client implementation
 642         if (script->getType() == ScriptingType::Client) {
 643                 std::string path = readParam<std::string>(L, 1);
 644                 const std::string *contents = script->getClient()->getModFile(path);
 645                 if (!contents) {
 646                         std::string error_msg = "Coudln't find script called: " + path;
 647                         lua_pushnil(L);
 648                         lua_pushstring(L, error_msg.c_str());
 649                         return 2;
 650                 }
 651
 652                 std::string chunk_name = "@" + path;
 653                 if (!safeLoadString(L, *contents, chunk_name.c_str())) {
 654                         lua_pushnil(L);
 655                         lua_insert(L, -2);
 656                         return 2;
 657                 }
 658                 return 1;
 659         }
 660 #endif
 661
 662         // Server implementation
 663         const char *path = NULL;
 664         if (lua_isstring(L, 1)) {
 665                 path = lua_tostring(L, 1);
 666                 CHECK_SECURE_PATH_INTERNAL(L, path, false, NULL);
 667         }
 668
 669         if (!safeLoadFile(L, path)) {
 670                 lua_pushnil(L);
 671                 lua_insert(L, -2);
 672                 return 2;
 673         }
 674
 675         return 1;
 676 }
 677
 678
 679 int ScriptApiSecurity::sl_g_loadstring(lua_State *L)
 680 {
 681         const char *chunk_name = "=(load)";
 682
 683         luaL_checktype(L, 1, LUA_TSTRING);
 684         if (!lua_isnone(L, 2)) {
 685                 luaL_checktype(L, 2, LUA_TSTRING);
 686                 chunk_name = lua_tostring(L, 2);
 687         }
 688
 689         size_t size;
 690         const char *code = lua_tolstring(L, 1, &size);
 691         std::string code_s(code, size);
 692
 693         if (!safeLoadString(L, code_s, chunk_name)) {
 694                 lua_pushnil(L);
 695                 lua_insert(L, -2);
 696                 return 2;
 697         }
 698         return 1;
 699 }
 700
 701
 702 int ScriptApiSecurity::sl_g_require(lua_State *L)
 703 {
 704         lua_pushliteral(L, "require() is disabled when mod security is on.");
 705         return lua_error(L);
 706 }
 707
 708
 709 int ScriptApiSecurity::sl_io_open(lua_State *L)
 710 {
 711         bool with_mode = lua_gettop(L) > 1;
 712
 713         luaL_checktype(L, 1, LUA_TSTRING);
 714         const char *path = lua_tostring(L, 1);
 715
 716         bool write_requested = false;
 717         if (with_mode) {
 718                 luaL_checktype(L, 2, LUA_TSTRING);
 719                 const char *mode = lua_tostring(L, 2);
 720                 write_requested = strchr(mode, 'w') != NULL ||
 721                         strchr(mode, '+') != NULL ||
 722                         strchr(mode, 'a') != NULL;
 723         }
 724         CHECK_SECURE_PATH_INTERNAL(L, path, write_requested, NULL);
 725
 726         push_original(L, "io", "open");
 727         lua_pushvalue(L, 1);
 728         if (with_mode) {
 729                 lua_pushvalue(L, 2);
 730         }
 731
 732         lua_call(L, with_mode ? 2 : 1, 2);
 733         return 2;
 734 }
 735
 736
 737 int ScriptApiSecurity::sl_io_input(lua_State *L)
 738 {
 739         if (lua_isstring(L, 1)) {
 740                 const char *path = lua_tostring(L, 1);
 741                 CHECK_SECURE_PATH_INTERNAL(L, path, false, NULL);
 742         }
 743
 744         push_original(L, "io", "input");
 745         lua_pushvalue(L, 1);
 746         lua_call(L, 1, 1);
 747         return 1;
 748 }
 749
 750
 751 int ScriptApiSecurity::sl_io_output(lua_State *L)
 752 {
 753         if (lua_isstring(L, 1)) {
 754                 const char *path = lua_tostring(L, 1);
 755                 CHECK_SECURE_PATH_INTERNAL(L, path, true, NULL);
 756         }
 757
 758         push_original(L, "io", "output");
 759         lua_pushvalue(L, 1);
 760         lua_call(L, 1, 1);
 761         return 1;
 762 }
 763
 764
 765 int ScriptApiSecurity::sl_io_lines(lua_State *L)
 766 {
 767         if (lua_isstring(L, 1)) {
 768                 const char *path = lua_tostring(L, 1);
 769                 CHECK_SECURE_PATH_INTERNAL(L, path, false, NULL);
 770         }
 771
 772         int top_precall = lua_gettop(L);
 773         push_original(L, "io", "lines");
 774         lua_pushvalue(L, 1);
 775         lua_call(L, 1, LUA_MULTRET);
 776         // Return number of arguments returned by the function,
 777         // adjusting for the function being poped.
 778         return lua_gettop(L) - top_precall;
 779 }
 780
 781
 782 int ScriptApiSecurity::sl_os_rename(lua_State *L)
 783 {
 784         luaL_checktype(L, 1, LUA_TSTRING);
 785         const char *path1 = lua_tostring(L, 1);
 786         CHECK_SECURE_PATH_INTERNAL(L, path1, true, NULL);
 787
 788         luaL_checktype(L, 2, LUA_TSTRING);
 789         const char *path2 = lua_tostring(L, 2);
 790         CHECK_SECURE_PATH_INTERNAL(L, path2, true, NULL);
 791
 792         push_original(L, "os", "rename");
 793         lua_pushvalue(L, 1);
 794         lua_pushvalue(L, 2);
 795         lua_call(L, 2, 2);
 796         return 2;
 797 }
 798
 799
 800 int ScriptApiSecurity::sl_os_remove(lua_State *L)
 801 {
 802         luaL_checktype(L, 1, LUA_TSTRING);
 803         const char *path = lua_tostring(L, 1);
 804         CHECK_SECURE_PATH_INTERNAL(L, path, true, NULL);
 805
 806         push_original(L, "os", "remove");
 807         lua_pushvalue(L, 1);
 808         lua_call(L, 1, 2);
 809         return 2;
 810 }
 811