mirror of
https://github.com/mkxp-z/mkxp-z.git
synced 2025-08-23 15:23:44 +02:00

The binding coroutines in libretro builds are constructed on the VM stack, so reallocating the VM memory would corrupt the memory of any currently existing coroutines. I've changed it so that the coroutines are no longer constructed on the VM stack so that they're unaffected by VM memory reallocations, and added a "slot" mechanism for storing variables on the VM stack. (Any Ruby `VALUE`s used by a coroutine have to be stored on the VM stack so that the Ruby garbage collector doesn't free them while they're being used, which is why the slot mechanism is necessary.)
159 lines
5.4 KiB
C++
159 lines
5.4 KiB
C++
/*
|
|
** sandbox.cpp
|
|
**
|
|
** This file is part of mkxp.
|
|
**
|
|
** Copyright (C) 2013 - 2021 Amaryllis Kulla <ancurio@mapleshrine.eu>
|
|
**
|
|
** mkxp is free software: you can redistribute it and/or modify
|
|
** it under the terms of the GNU General Public License as published by
|
|
** the Free Software Foundation, either version 2 of the License, or
|
|
** (at your option) any later version.
|
|
**
|
|
** mkxp is distributed in the hope that it will be useful,
|
|
** but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
** GNU General Public License for more details.
|
|
**
|
|
** You should have received a copy of the GNU General Public License
|
|
** along with mkxp. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
#include <string>
|
|
#include <wasm-rt.h>
|
|
#include "wasi.h"
|
|
#include <mkxp-sandbox-ruby.h>
|
|
#include "mkxp-polyfill.h"
|
|
#include "sandbox.h"
|
|
|
|
#define RB (ruby.get())
|
|
#define WASM_nullptr 0
|
|
#define WASM_MEM(address) ((void *)&ruby->w2c_memory.data[address])
|
|
#define AWAIT(statement) do statement; while (w2c_ruby_mkxp_sandbox_yield(RB))
|
|
|
|
using namespace mkxp_sandbox;
|
|
|
|
usize sandbox::sandbox_malloc(usize size) {
|
|
usize buf = w2c_ruby_mkxp_sandbox_malloc(RB, size);
|
|
|
|
// Verify that the returned pointer is non-null and the entire allocated buffer is in valid memory
|
|
usize buf_end;
|
|
if (buf == WASM_nullptr || (buf_end = buf + size) < buf || buf_end >= ruby->w2c_memory.size) {
|
|
throw std::bad_alloc();
|
|
}
|
|
|
|
return buf;
|
|
}
|
|
|
|
void sandbox::sandbox_free(usize ptr) {
|
|
w2c_ruby_mkxp_sandbox_free(RB, ptr);
|
|
}
|
|
|
|
sandbox::sandbox() : ruby(new struct w2c_ruby), wasi(new wasi_t(ruby)), bindings(ruby), movie(nullptr), yielding(false), trans_map(nullptr), transitioning(false) {
|
|
// Initialize the sandbox
|
|
wasm2c_ruby_instantiate(RB, wasi.get());
|
|
w2c_ruby_mkxp_sandbox_init(
|
|
RB,
|
|
0, // heap_free_slots
|
|
1.1, // growth_factor
|
|
0, // growth_max_slots
|
|
0, // heap_free_slots_min_ratio
|
|
0, // heap_free_slots_goal_ratio
|
|
0, // heap_free_slots_max_ratio
|
|
0, // uncollectible_wb_unprotected_objects_limit_ratio
|
|
0, // oldobject_limit_factor
|
|
1 * 0x100000, // malloc_limit_min
|
|
4 * 0x100000, // malloc_limit_max
|
|
1.1, // malloc_limit_growth_factor
|
|
4 * 0x100000, // oldmalloc_limit_min
|
|
8 * 0x100000, // oldmalloc_limit_max
|
|
1.1 // oldmalloc_limit_growth_factor
|
|
);
|
|
|
|
// Change the current working directory to the game directory
|
|
usize chdir_buf = sandbox_malloc(sizeof("/game"));
|
|
std::strcpy((char *)WASM_MEM(chdir_buf), "/game");
|
|
w2c_ruby_mkxp_sandbox_chdir(RB, chdir_buf);
|
|
sandbox_free(chdir_buf);
|
|
|
|
// Determine Ruby command-line arguments
|
|
std::vector<std::string> args{"mkxp-z"};
|
|
args.push_back("/dist/bin/mkxp-z");
|
|
|
|
// Copy all the command-line arguments into the sandbox (sandboxed code can't access memory that's outside the sandbox!)
|
|
usize argv_buf = sandbox_malloc(args.size() * sizeof(usize));
|
|
for (usize i = 0; i < args.size(); ++i) {
|
|
usize arg_buf = sandbox_malloc(args[i].length() + 1);
|
|
std::strcpy((char *)WASM_MEM(arg_buf), args[i].c_str());
|
|
WASM_SET(usize, argv_buf + i * sizeof(usize), arg_buf);
|
|
}
|
|
|
|
// Pass the command-line arguments to Ruby
|
|
AWAIT(w2c_ruby_ruby_init_stack(RB, w2c_ruby_rb_wasm_get_stack_pointer(RB)));
|
|
AWAIT(w2c_ruby_ruby_init(RB));
|
|
usize node;
|
|
AWAIT(node = w2c_ruby_ruby_options(RB, args.size(), argv_buf));
|
|
|
|
// Start up Ruby executable node
|
|
bool valid;
|
|
u32 state;
|
|
usize state_buf = sandbox_malloc(sizeof(usize));
|
|
AWAIT(valid = w2c_ruby_ruby_executable_node(RB, node, state_buf));
|
|
if (valid) {
|
|
AWAIT(state = w2c_ruby_ruby_exec_node(RB, node));
|
|
}
|
|
if (!valid || state) {
|
|
std::abort();
|
|
}
|
|
sandbox_free(state_buf);
|
|
|
|
// Set the default encoding to UTF-8
|
|
usize encoding;
|
|
AWAIT(encoding = w2c_ruby_rb_utf8_encoding(RB));
|
|
usize enc;
|
|
AWAIT(enc = w2c_ruby_rb_enc_from_encoding(RB, encoding));
|
|
AWAIT(w2c_ruby_rb_enc_set_default_internal(RB, enc));
|
|
AWAIT(w2c_ruby_rb_enc_set_default_external(RB, enc));
|
|
}
|
|
|
|
sandbox::~sandbox() {
|
|
set_movie(nullptr);
|
|
if (yielding) {
|
|
w2c_ruby_asyncify_stop_unwind(ruby.get());
|
|
}
|
|
bindings.reset(); // Destroy the bindings before destroying the runtime since the bindings destructor requires the runtime to be alive
|
|
wasm2c_ruby_free(RB);
|
|
}
|
|
|
|
Movie *sandbox::get_movie_from_main_thread() {
|
|
return movie.load(std::memory_order_relaxed); // No need for synchronization because we always set the movie from the main thread
|
|
}
|
|
|
|
Movie *sandbox::get_movie_from_audio_thread() {
|
|
return movie.load(std::memory_order_seq_cst);
|
|
}
|
|
|
|
void sandbox::set_movie(Movie *new_movie) {
|
|
Movie *old_movie = get_movie_from_main_thread();
|
|
if (old_movie == new_movie) {
|
|
return;
|
|
}
|
|
if (old_movie != nullptr) {
|
|
movie.store(nullptr, std::memory_order_seq_cst);
|
|
AudioMutexGuard guard(movie_mutex);
|
|
Graphics::stopMovie(old_movie);
|
|
}
|
|
if (new_movie != nullptr) {
|
|
movie.store(new_movie, std::memory_order_seq_cst);
|
|
}
|
|
}
|
|
|
|
const char *sandbox::getcwd() {
|
|
if (w2c_ruby_mkxp_sandbox_getcwd(ruby.get())) {
|
|
return (const char *)(**bindings + ruby->w2c_mkxp_sandbox_cwd);
|
|
} else {
|
|
return "/game";
|
|
}
|
|
}
|