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 "settings.h"
  26
  27 #include <cerrno>
  28 #include <string>
  29 #include <iostream>
  30
  31
  32 #define SECURE_API(lib, name) \
  33         lua_pushcfunction(L, sl_##lib##_##name); \
  34         lua_setfield(L, -2, #name);
  35
  36
  37 static inline void copy_safe(lua_State *L, const char *list[], unsigned len, int from=-2, int to=-1)
  38 {
  39         if (from < 0) from = lua_gettop(L) + from + 1;
  40         if (to   < 0) to   = lua_gettop(L) + to   + 1;
  41         for (unsigned i = 0; i < (len / sizeof(list[0])); i++) {
  42                 lua_getfield(L, from, list[i]);
  43                 lua_setfield(L, to,   list[i]);
  44         }
  45 }
  46
  47 // Pushes the original version of a library function on the stack, from the old version
  48 static inline void push_original(lua_State *L, const char *lib, const char *func)
  49 {
  50         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_GLOBALS_BACKUP);
  51         lua_getfield(L, -1, lib);
  52         lua_remove(L, -2);  // Remove globals_backup
  53         lua_getfield(L, -1, func);
  54         lua_remove(L, -2);  // Remove lib
  55 }
  56
  57
  58 void ScriptApiSecurity::initializeSecurity()
  59 {
  60         static const char *whitelist[] = {
  61                 "assert",
  62                 "core",
  63                 "collectgarbage",
  64                 "DIR_DELIM",
  65                 "error",
  66                 "getfenv",
  67                 "getmetatable",
  68                 "ipairs",
  69                 "next",
  70                 "pairs",
  71                 "pcall",
  72                 "print",
  73                 "rawequal",
  74                 "rawget",
  75                 "rawset",
  76                 "select",
  77                 "setfenv",
  78                 "setmetatable",
  79                 "tonumber",
  80                 "tostring",
  81                 "type",
  82                 "unpack",
  83                 "_VERSION",
  84                 "xpcall",
  85                 // Completely safe libraries
  86                 "coroutine",
  87                 "string",
  88                 "table",
  89                 "math",
  90         };
  91         static const char *io_whitelist[] = {
  92                 "close",
  93                 "flush",
  94                 "read",
  95                 "type",
  96                 "write",
  97         };
  98         static const char *os_whitelist[] = {
  99                 "clock",
 100                 "date",
 101                 "difftime",
 102                 "exit",
 103                 "getenv",
 104                 "setlocale",
 105                 "time",
 106                 "tmpname",
 107         };
 108         static const char *debug_whitelist[] = {
 109                 "gethook",
 110                 "traceback",
 111                 "getinfo",
 112                 "getmetatable",
 113                 "setupvalue",
 114                 "setmetatable",
 115                 "upvalueid",
 116                 "upvaluejoin",
 117                 "sethook",
 118                 "debug",
 119                 "setlocal",
 120         };
 121         static const char *package_whitelist[] = {
 122                 "config",
 123                 "cpath",
 124                 "path",
 125                 "searchpath",
 126         };
 127         static const char *jit_whitelist[] = {
 128                 "arch",
 129                 "flush",
 130                 "off",
 131                 "on",
 132                 "opt",
 133                 "os",
 134                 "status",
 135                 "version",
 136                 "version_num",
 137         };
 138
 139         m_secure = true;
 140
 141         lua_State *L = getStack();
 142
 143         // Backup globals to the registry
 144         lua_getglobal(L, "_G");
 145         lua_rawseti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_GLOBALS_BACKUP);
 146
 147         // Replace the global environment with an empty one
 148 #if LUA_VERSION_NUM <= 501
 149         int is_main = lua_pushthread(L);  // Push the main thread
 150         FATAL_ERROR_IF(!is_main, "Security: ScriptApi's Lua state "
 151                         "isn't the main Lua thread!");
 152 #endif
 153         lua_newtable(L);  // Create new environment
 154         lua_pushvalue(L, -1);
 155         lua_setfield(L, -2, "_G");  // Set _G of new environment
 156 #if LUA_VERSION_NUM >= 502  // Lua >= 5.2
 157         // Set the global environment
 158         lua_rawseti(L, LUA_REGISTRYINDEX, LUA_RIDX_GLOBALS);
 159 #else  // Lua <= 5.1
 160         // Set the environment of the main thread
 161         FATAL_ERROR_IF(!lua_setfenv(L, -2), "Security: Unable to set "
 162                         "environment of the main Lua thread!");
 163         lua_pop(L, 1);  // Pop thread
 164 #endif
 165
 166         // Get old globals
 167         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_GLOBALS_BACKUP);
 168         int old_globals = lua_gettop(L);
 169
 170
 171         // Copy safe base functions
 172         lua_getglobal(L, "_G");
 173         copy_safe(L, whitelist, sizeof(whitelist));
 174
 175         // And replace unsafe ones
 176         SECURE_API(g, dofile);
 177         SECURE_API(g, load);
 178         SECURE_API(g, loadfile);
 179         SECURE_API(g, loadstring);
 180         SECURE_API(g, require);
 181         lua_pop(L, 1);
 182
 183
 184         // Copy safe IO functions
 185         lua_getfield(L, old_globals, "io");
 186         lua_newtable(L);
 187         copy_safe(L, io_whitelist, sizeof(io_whitelist));
 188
 189         // And replace unsafe ones
 190         SECURE_API(io, open);
 191         SECURE_API(io, input);
 192         SECURE_API(io, output);
 193         SECURE_API(io, lines);
 194
 195         lua_setglobal(L, "io");
 196         lua_pop(L, 1);  // Pop old IO
 197
 198
 199         // Copy safe OS functions
 200         lua_getfield(L, old_globals, "os");
 201         lua_newtable(L);
 202         copy_safe(L, os_whitelist, sizeof(os_whitelist));
 203
 204         // And replace unsafe ones
 205         SECURE_API(os, remove);
 206         SECURE_API(os, rename);
 207
 208         lua_setglobal(L, "os");
 209         lua_pop(L, 1);  // Pop old OS
 210
 211
 212         // Copy safe debug functions
 213         lua_getfield(L, old_globals, "debug");
 214         lua_newtable(L);
 215         copy_safe(L, debug_whitelist, sizeof(debug_whitelist));
 216         lua_setglobal(L, "debug");
 217         lua_pop(L, 1);  // Pop old debug
 218
 219
 220         // Copy safe package fields
 221         lua_getfield(L, old_globals, "package");
 222         lua_newtable(L);
 223         copy_safe(L, package_whitelist, sizeof(package_whitelist));
 224         lua_setglobal(L, "package");
 225         lua_pop(L, 1);  // Pop old package
 226
 227
 228         // Copy safe jit functions, if they exist
 229         lua_getfield(L, -1, "jit");
 230         if (!lua_isnil(L, -1)) {
 231                 lua_newtable(L);
 232                 copy_safe(L, jit_whitelist, sizeof(jit_whitelist));
 233                 lua_setglobal(L, "jit");
 234         }
 235         lua_pop(L, 1);  // Pop old jit
 236
 237         lua_pop(L, 1); // Pop globals_backup
 238 }
 239
 240
 241 bool ScriptApiSecurity::isSecure(lua_State *L)
 242 {
 243         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_GLOBALS_BACKUP);
 244         bool secure = !lua_isnil(L, -1);
 245         lua_pop(L, 1);
 246         return secure;
 247 }
 248
 249
 250 #define CHECK_FILE_ERR(ret, fp) \
 251         if (ret) { \
 252                 lua_pushfstring(L, "%s: %s", path, strerror(errno)); \
 253                 if (fp) std::fclose(fp); \
 254                 return false; \
 255         }
 256
 257
 258 bool ScriptApiSecurity::safeLoadFile(lua_State *L, const char *path)
 259 {
 260         FILE *fp;
 261         char *chunk_name;
 262         if (path == NULL) {
 263                 fp = stdin;
 264                 chunk_name = const_cast<char *>("=stdin");
 265         } else {
 266                 fp = fopen(path, "rb");
 267                 if (!fp) {
 268                         lua_pushfstring(L, "%s: %s", path, strerror(errno));
 269                         return false;
 270                 }
 271                 chunk_name = new char[strlen(path) + 2];
 272                 chunk_name[0] = '@';
 273                 chunk_name[1] = '\0';
 274                 strcat(chunk_name, path);
 275         }
 276
 277         size_t start = 0;
 278         int c = std::getc(fp);
 279         if (c == '#') {
 280                 // Skip the first line
 281                 while ((c = std::getc(fp)) != EOF && c != '\n');
 282                 if (c == '\n') c = std::getc(fp);
 283                 start = std::ftell(fp);
 284         }
 285
 286         if (c == LUA_SIGNATURE[0]) {
 287                 lua_pushliteral(L, "Bytecode prohibited when mod security is enabled.");
 288                 std::fclose(fp);
 289                 if (path) {
 290                         delete [] chunk_name;
 291                 }
 292                 return false;
 293         }
 294
 295         // Read the file
 296         int ret = std::fseek(fp, 0, SEEK_END);
 297         CHECK_FILE_ERR(ret, fp);
 298
 299         size_t size = std::ftell(fp) - start;
 300         char *code = new char[size];
 301         ret = std::fseek(fp, start, SEEK_SET);
 302         if (ret) {
 303                 lua_pushfstring(L, "%s: %s", path, strerror(errno));
 304                 std::fclose(fp);
 305                 delete [] code;
 306                 if (path) {
 307                         delete [] chunk_name;
 308                 }
 309                 return false;
 310         }
 311
 312         size_t num_read = std::fread(code, 1, size, fp);
 313         if (path) {
 314                 std::fclose(fp);
 315         }
 316         if (num_read != size) {
 317                 lua_pushliteral(L, "Error reading file to load.");
 318                 delete [] code;
 319                 if (path) {
 320                         delete [] chunk_name;
 321                 }
 322                 return false;
 323         }
 324
 325         if (luaL_loadbuffer(L, code, size, chunk_name)) {
 326                 return false;
 327         }
 328
 329         if (path) {
 330                 delete [] chunk_name;
 331         }
 332         return true;
 333 }
 334
 335
 336 bool ScriptApiSecurity::checkPath(lua_State *L, const char *path)
 337 {
 338         std::string str;  // Transient
 339
 340         std::string norel_path = fs::RemoveRelativePathComponents(path);
 341         std::string abs_path = fs::AbsolutePath(norel_path);
 342
 343         if (!abs_path.empty()) {
 344                 // Don't allow accessing the settings file
 345                 str = fs::AbsolutePath(g_settings_path);
 346                 if (str == abs_path) return false;
 347         }
 348
 349         // If we couldn't find the absolute path (path doesn't exist) then
 350         // try removing the last components until it works (to allow
 351         // non-existent files/folders for mkdir).
 352         std::string cur_path = norel_path;
 353         std::string removed;
 354         while (abs_path.empty() && !cur_path.empty()) {
 355                 std::string tmp_rmed;
 356                 cur_path = fs::RemoveLastPathComponent(cur_path, &tmp_rmed);
 357                 removed = tmp_rmed + (removed.empty() ? "" : DIR_DELIM + removed);
 358                 abs_path = fs::AbsolutePath(cur_path);
 359         }
 360         if (abs_path.empty()) return false;
 361         // Add the removed parts back so that you can't, eg, create a
 362         // directory in worldmods if worldmods doesn't exist.
 363         if (!removed.empty()) abs_path += DIR_DELIM + removed;
 364
 365         // Get server from registry
 366         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_SCRIPTAPI);
 367         ScriptApiBase *script = (ScriptApiBase *) lua_touserdata(L, -1);
 368         lua_pop(L, 1);
 369         const Server *server = script->getServer();
 370
 371         if (!server) return false;
 372
 373         // Get mod name
 374         lua_rawgeti(L, LUA_REGISTRYINDEX, CUSTOM_RIDX_CURRENT_MOD_NAME);
 375         if (lua_isstring(L, -1)) {
 376                 std::string mod_name = lua_tostring(L, -1);
 377
 378                 // Builtin can access anything
 379                 if (mod_name == BUILTIN_MOD_NAME) {
 380                         return true;
 381                 }
 382
 383                 // Allow paths in mod path
 384                 const ModSpec *mod = server->getModSpec(mod_name);
 385                 if (mod) {
 386                         str = fs::AbsolutePath(mod->path);
 387                         if (!str.empty() && fs::PathStartsWith(abs_path, str)) {
 388                                 return true;
 389                         }
 390                 }
 391         }
 392         lua_pop(L, 1);  // Pop mod name
 393
 394         str = fs::AbsolutePath(server->getWorldPath());
 395         if (str.empty()) return false;
 396         // Don't allow access to world mods.  We add to the absolute path
 397         // of the world instead of getting the absolute paths directly
 398         // because that won't work if they don't exist.
 399         if (fs::PathStartsWith(abs_path, str + DIR_DELIM + "worldmods") ||
 400                         fs::PathStartsWith(abs_path, str + DIR_DELIM + "game")) {
 401                 return false;
 402         }
 403         // Allow all other paths in world path
 404         if (fs::PathStartsWith(abs_path, str)) {
 405                 return true;
 406         }
 407
 408         // Default to disallowing
 409         return false;
 410 }
 411
 412
 413 int ScriptApiSecurity::sl_g_dofile(lua_State *L)
 414 {
 415         int nret = sl_g_loadfile(L);
 416         if (nret != 1) {
 417                 lua_error(L);
 418                 // code after this function isn't executed
 419         }
 420         int top_precall = lua_gettop(L);
 421         lua_call(L, 0, LUA_MULTRET);
 422         // Return number of arguments returned by the function,
 423         // adjusting for the function being poped.
 424         return lua_gettop(L) - (top_precall - 1);
 425 }
 426
 427
 428 int ScriptApiSecurity::sl_g_load(lua_State *L)
 429 {
 430         size_t len;
 431         const char *buf;
 432         std::string code;
 433         const char *chunk_name = "=(load)";
 434
 435         luaL_checktype(L, 1, LUA_TFUNCTION);
 436         if (!lua_isnone(L, 2)) {
 437                 luaL_checktype(L, 2, LUA_TSTRING);
 438                 chunk_name = lua_tostring(L, 2);
 439         }
 440
 441         while (true) {
 442                 lua_pushvalue(L, 1);
 443                 lua_call(L, 0, 1);
 444                 int t = lua_type(L, -1);
 445                 if (t == LUA_TNIL) {
 446                         break;
 447                 } else if (t != LUA_TSTRING) {
 448                         lua_pushnil(L);
 449                         lua_pushliteral(L, "Loader didn't return a string");
 450                         return 2;
 451                 }
 452                 buf = lua_tolstring(L, -1, &len);
 453                 code += std::string(buf, len);
 454                 lua_pop(L, 1); // Pop return value
 455         }
 456         if (code[0] == LUA_SIGNATURE[0]) {
 457                 lua_pushnil(L);
 458                 lua_pushliteral(L, "Bytecode prohibited when mod security is enabled.");
 459                 return 2;
 460         }
 461         if (luaL_loadbuffer(L, code.data(), code.size(), chunk_name)) {
 462                 lua_pushnil(L);
 463                 lua_insert(L, lua_gettop(L) - 1);
 464                 return 2;
 465         }
 466         return 1;
 467 }
 468
 469
 470 int ScriptApiSecurity::sl_g_loadfile(lua_State *L)
 471 {
 472         const char *path = NULL;
 473
 474         if (lua_isstring(L, 1)) {
 475                 path = lua_tostring(L, 1);
 476                 CHECK_SECURE_PATH(L, path);
 477         }
 478
 479         if (!safeLoadFile(L, path)) {
 480                 lua_pushnil(L);
 481                 lua_insert(L, -2);
 482                 return 2;
 483         }
 484
 485         return 1;
 486 }
 487
 488
 489 int ScriptApiSecurity::sl_g_loadstring(lua_State *L)
 490 {
 491         const char *chunk_name = "=(load)";
 492
 493         luaL_checktype(L, 1, LUA_TSTRING);
 494         if (!lua_isnone(L, 2)) {
 495                 luaL_checktype(L, 2, LUA_TSTRING);
 496                 chunk_name = lua_tostring(L, 2);
 497         }
 498
 499         size_t size;
 500         const char *code = lua_tolstring(L, 1, &size);
 501
 502         if (size > 0 && code[0] == LUA_SIGNATURE[0]) {
 503                 lua_pushnil(L);
 504                 lua_pushliteral(L, "Bytecode prohibited when mod security is enabled.");
 505                 return 2;
 506         }
 507         if (luaL_loadbuffer(L, code, size, chunk_name)) {
 508                 lua_pushnil(L);
 509                 lua_insert(L, lua_gettop(L) - 1);
 510                 return 2;
 511         }
 512         return 1;
 513 }
 514
 515
 516 int ScriptApiSecurity::sl_g_require(lua_State *L)
 517 {
 518         lua_pushliteral(L, "require() is disabled when mod security is on.");
 519         return lua_error(L);
 520 }
 521
 522
 523 int ScriptApiSecurity::sl_io_open(lua_State *L)
 524 {
 525         luaL_checktype(L, 1, LUA_TSTRING);
 526         const char *path = lua_tostring(L, 1);
 527         CHECK_SECURE_PATH(L, path);
 528
 529         push_original(L, "io", "open");
 530         lua_pushvalue(L, 1);
 531         lua_pushvalue(L, 2);
 532         lua_call(L, 2, 2);
 533         return 2;
 534 }
 535
 536
 537 int ScriptApiSecurity::sl_io_input(lua_State *L)
 538 {
 539         if (lua_isstring(L, 1)) {
 540                 const char *path = lua_tostring(L, 1);
 541                 CHECK_SECURE_PATH(L, path);
 542         }
 543
 544         push_original(L, "io", "input");
 545         lua_pushvalue(L, 1);
 546         lua_call(L, 1, 1);
 547         return 1;
 548 }
 549
 550
 551 int ScriptApiSecurity::sl_io_output(lua_State *L)
 552 {
 553         if (lua_isstring(L, 1)) {
 554                 const char *path = lua_tostring(L, 1);
 555                 CHECK_SECURE_PATH(L, path);
 556         }
 557
 558         push_original(L, "io", "output");
 559         lua_pushvalue(L, 1);
 560         lua_call(L, 1, 1);
 561         return 1;
 562 }
 563
 564
 565 int ScriptApiSecurity::sl_io_lines(lua_State *L)
 566 {
 567         if (lua_isstring(L, 1)) {
 568                 const char *path = lua_tostring(L, 1);
 569                 CHECK_SECURE_PATH(L, path);
 570         }
 571
 572         push_original(L, "io", "lines");
 573         lua_pushvalue(L, 1);
 574         int top_precall = lua_gettop(L);
 575         lua_call(L, 1, LUA_MULTRET);
 576         // Return number of arguments returned by the function,
 577         // adjusting for the function being poped.
 578         return lua_gettop(L) - (top_precall - 1);
 579 }
 580
 581
 582 int ScriptApiSecurity::sl_os_rename(lua_State *L)
 583 {
 584         luaL_checktype(L, 1, LUA_TSTRING);
 585         const char *path1 = lua_tostring(L, 1);
 586         CHECK_SECURE_PATH(L, path1);
 587
 588         luaL_checktype(L, 2, LUA_TSTRING);
 589         const char *path2 = lua_tostring(L, 2);
 590         CHECK_SECURE_PATH(L, path2);
 591
 592         push_original(L, "os", "rename");
 593         lua_pushvalue(L, 1);
 594         lua_pushvalue(L, 2);
 595         lua_call(L, 2, 2);
 596         return 2;
 597 }
 598
 599
 600 int ScriptApiSecurity::sl_os_remove(lua_State *L)
 601 {
 602         luaL_checktype(L, 1, LUA_TSTRING);
 603         const char *path = lua_tostring(L, 1);
 604         CHECK_SECURE_PATH(L, path);
 605
 606         push_original(L, "os", "remove");
 607         lua_pushvalue(L, 1);
 608         lua_call(L, 1, 2);
 609         return 2;
 610 }
 611