Files
integration/luprex/cpp/core/world-core.cpp

1168 lines
35 KiB
C++
Raw Normal View History

2021-09-10 17:06:07 -04:00
#include "world.hpp"
#include "idalloc.hpp"
#include "animqueue.hpp"
#include "gui.hpp"
#include "traceback.hpp"
#include "pprint.hpp"
#include "util.hpp"
#include "serializelua.hpp"
2021-09-10 17:06:07 -04:00
#include <iostream>
void World::store_global_pointer(lua_State *L, World *v) {
lua_pushstring(L, "world");
lua_pushlightuserdata(L, v);
lua_rawset(L, LUA_REGISTRYINDEX);
}
World *World::fetch_global_pointer(lua_State *L) {
lua_pushstring(L, "world");
lua_rawget(L, LUA_REGISTRYINDEX);
World *result = (World *)lua_touserdata(L, -1);
if (result == nullptr) {
luaL_error(L, "No world pointer stored.");
}
lua_pop(L, 1);
return result;
}
World::~World() {
}
World::World(WorldType wt) {
2021-09-10 17:06:07 -04:00
// Master world model by default.
world_type_ = wt;
// Initialize the ID allocator in master mode.
if (is_authoritative()) {
2021-09-10 17:06:07 -04:00
id_global_pool_.init_master();
} else {
id_global_pool_.init_synch();
}
// Prepare to manipulate the lua state.
LuaVar world, globtab;
LuaExtStack LS(state(), world, globtab);
2021-09-10 17:06:07 -04:00
// Put the world pointer into the lua registry.
World::store_global_pointer(state(), this);
// Clear the global GUI pointer.
Gui::store_global_pointer(state(), nullptr);
// Clear the lthread state.
clear_lthread_state();
2021-09-10 17:06:07 -04:00
// Set the tabletype of the registry.
LS.settabletype(LuaRegistry, LUA_TT_REGISTRY);
// Set the tabletype of the global environment.
LS.getglobaltable(globtab);
LS.settabletype(globtab, LUA_TT_GLOBALENV);
// Store the world type in the registry.
LS.rawset(LuaRegistry, "worldtype", wt);
// Create the globaldb in the registry.
LS.rawset(LuaRegistry, "globaldb", LuaNewTable);
2021-09-10 17:06:07 -04:00
2023-04-10 17:58:44 -04:00
// Initialize the SourceDB.
2021-09-10 17:06:07 -04:00
source_db_.init(state());
2023-04-10 17:58:44 -04:00
2021-11-26 15:45:36 -05:00
// Clear the clock.
clock_ = 0;
// Initialize global variable state.
assign_seqno_ = 1;
2021-09-10 17:06:07 -04:00
}
Tangible::Tangible(World *w, int64_t id) : world_(w), anim_queue_(), id_player_pool_(&w->id_global_pool_) {
2021-09-10 17:06:07 -04:00
plane_item_.set_id(id);
plane_item_.track(&w->plane_map_);
}
void Tangible::update_plane_item() {
AnimCoreState pos = anim_queue_.get_final_core_state();
plane_item_.set_pos(pos.plane, pos.xyz.x, pos.xyz.y, pos.xyz.z);
2021-09-10 17:06:07 -04:00
}
void Tangible::serialize(StreamBuffer *sb) {
anim_queue_.serialize(sb);
id_player_pool_.serialize(sb);
2021-11-14 15:57:18 -05:00
print_buffer_.serialize(sb);
2021-09-10 17:06:07 -04:00
}
void Tangible::deserialize(StreamBuffer *sb) {
anim_queue_.deserialize(sb);
id_player_pool_.deserialize(sb);
2021-11-14 15:57:18 -05:00
print_buffer_.deserialize(sb);
2021-09-10 17:06:07 -04:00
update_plane_item();
}
Tangible *World::tangible_get(int64_t id) {
auto iter = tangibles_.find(id);
if (iter == tangibles_.end()) {
return nullptr;
} else {
return iter->second.get();
}
}
const Tangible *World::tangible_get(int64_t id) const {
auto iter = tangibles_.find(id);
if (iter == tangibles_.end()) {
return nullptr;
} else {
return iter->second.get();
}
}
World::TanVector World::tangible_get_all(const IdVector &ids) const {
TanVector result(ids.size());
for (int i = 0; i < int(ids.size()); i++) {
result[i] = tangible_get(ids[i]);
}
return result;
}
Tangible *World::tangible_get(const LuaCoreStack &LS, LuaSlot tab, bool allowdel) {
2021-11-26 12:28:59 -05:00
int64_t id = LS.tanid(tab);
2021-09-10 17:06:07 -04:00
if (id == 0) {
luaL_error(LS.state(), "parameter is not a tangible");
}
Tangible *result = tangible_get(id);
if (!allowdel) {
if (result == nullptr) {
luaL_error(LS.state(), "argument is a deleted tangible, which is not allowed here");
}
2021-09-10 17:06:07 -04:00
}
return result;
}
Tangible *World::tangible_make(const LuaCoreStack &LS0, LuaSlot database, int64_t id) {
assert(LS0.validpositiveint64(id));
LuaVar metatab;
LuaExtStack LS(LS0.state(), metatab);
2021-09-10 17:06:07 -04:00
// Create the C++ part of the structure.
UniqueTangible &t = tangibles_[id];
2021-09-10 17:06:07 -04:00
assert (t == nullptr);
t.reset(new Tangible(this, id));
// AnimQueue initializes itself to a valid default state.
AnimState state;
state.add_defaults(nullptr);
t->anim_queue_.clear(state);
2021-09-10 17:06:07 -04:00
t->update_plane_item();
// Fetch the tangible's Lua database and metatable.
LS.maketan(database, id);
LS.getmetatable(metatab, database);
2021-09-10 17:06:07 -04:00
// Set up the inventory and thread table.
2021-09-10 17:06:07 -04:00
LS.rawset(database, "inventory", LuaNewTable);
LS.rawset(metatab, "threads", LuaNewTable);
2021-09-10 17:06:07 -04:00
return t.get();
}
Tangible *World::tangible_make(int64_t id) {
LuaVar database;
LuaExtStack LS(state(), database);
return tangible_make(LS, database, id);
}
2021-09-10 17:06:07 -04:00
void World::tangible_delete(int64_t id) {
lua_State *L = state();
LuaVar tangibles, database, metatab;
LuaExtStack LS(L, tangibles, database, metatab);
2021-09-10 17:06:07 -04:00
// Fetch the C++ side of the tangible.
auto iter = tangibles_.find(id);
if (iter == tangibles_.end()) {
return; // Nothing to delete.
}
// Fetch the lua side of the tangible.
LS.maketan(database, id);
2021-09-10 17:06:07 -04:00
assert(LS.istable(database));
LS.getmetatable(metatab, database);
// Clear out the database and the metatable.
LS.cleartable(database, false);
LS.cleartable(metatab, true);
2021-09-10 17:06:07 -04:00
// Now put the bare minimum info back into the metatable.
LS.rawset(metatab, "id", id);
LS.rawset(metatab, "__metatable", false);
2021-09-10 17:06:07 -04:00
// Remove the C++ portion from the tangibles table.
tangibles_.erase(iter);
}
void World::get_near(PlaneScan &scan, util::IdVector *into) const {
into->clear();
// If 'near' is set, update the plane and center.
int64_t actor_id = scan.near();
if (actor_id != 0) {
const Tangible *player = tangible_get(actor_id);
if (player == nullptr) {
return;
}
const PlaneItem &pi = player->plane_item_;
scan.set_plane(pi.plane());
scan.set_center(util::XYZ(pi.x(), pi.y(), pi.z()));
2021-09-10 17:06:07 -04:00
}
plane_map_.scan(scan, into);
}
2021-09-10 17:06:07 -04:00
void World::get_near(int64_t player_id, float radius, bool exclude_nowhere, bool omit_player, bool sorted, util::IdVector *into) const {
2022-07-11 02:32:12 -04:00
PlaneScan scan;
scan.set_radius(radius);
scan.set_shape(PlaneScan::SPHERE);
2022-07-11 02:32:12 -04:00
scan.set_sorted(sorted);
scan.set_omit_nowhere(exclude_nowhere);
scan.set_near(player_id, !omit_player);
get_near(scan, into);
2021-09-10 17:06:07 -04:00
}
void World::get_encoded_animation_queues(uint32_t count, const int64_t *ids, util::SharedStdStringVec &into) {
util::SharedStdString empty = std::make_shared<std::string>("");
into.resize(count);
2023-07-24 17:20:45 -04:00
for (int i = 0; i < int(count); i++) {
Tangible *tan = tangible_get(ids[i]);
if (tan == nullptr) {
into[i] = empty;
2023-07-24 17:20:45 -04:00
} else {
into[i] = tan->anim_queue_.get_encoded_queue();
2023-07-24 17:20:45 -04:00
}
}
}
2021-09-10 17:06:07 -04:00
World::Redirects World::fetch_redirects() {
World::Redirects result = std::move(redirects_);
redirects_.clear();
2021-11-23 14:38:08 -05:00
return result;
2021-09-10 17:06:07 -04:00
}
2024-02-27 17:32:08 -05:00
int64_t World::create_login_actor(bool initialize) {
assert(stack_is_clear());
int64_t id = id_global_pool_.get_one();
2024-02-27 17:32:08 -05:00
{
LuaVar database, classtab, mt, func;
LuaExtStack LS(state(), database, classtab, mt, func);
Tangible *tan = tangible_make(LS, database, id);
LS.makeclass(classtab, "login");
LS.getmetatable(mt, database);
LS.rawset(mt, "__index", classtab);
tan->configure_id_pool_for_actor();
tan->print_buffer_.clear();
if (initialize) {
LS.rawget(func, classtab, "initialize");
spawn(LS, id, id, func, true, 0, false);
}
}
run_scheduled_threads();
return id;
2021-09-10 17:06:07 -04:00
}
eng::string World::probe_lua(int64_t actor_id, const eng::string &lua) {
assert(stack_is_clear());
lua_State *L = state();
Tangible *actor = tangible_get(actor_id);
if (actor == nullptr) {
return "";
}
LuaVar closure;
LuaExtStack LS(L, closure);
// create the compiled closure.
int status = luaL_loadbuffer(L, lua.c_str(), lua.size(), "=probe");
lua_replace(L, closure.index());
if (status != LUA_OK) {
// The closure is actually an error message. Do nothing.
// This should normally not happen: LuaConsole should filter
// out syntax errors.
return "";
}
// Call the closure.
int top = lua_gettop(L);
lua_pushvalue(L, closure.index());
open_lthread_state(actor_id, actor_id, 0, false, true);
eng::string msg = traceback_pcall(L, 0, LUA_MULTRET);
// If there's an error message, print it.
// Otherwise, pretty-print the results.
std::ostream *ostream = lthread_print_stream();
if (msg.empty()) {
for (int i = top + 1; i <= lua_gettop(L); i++) {
LuaSpecial root(i);
pprint(LS, root, PrettyPrintOptions(), ostream);
// TODO: this endl is unnecessary if we just printed a newline.
(*ostream) << std::endl;
}
} else {
(*ostream) << msg << std::endl;
}
// Collect the lthread_prints (and also make sure they
// don't go into the printbuffer).
eng::string result = lthread_prints_->str();
lthread_prints_.reset();
close_lthread_state();
return result;
}
2021-09-10 17:06:07 -04:00
void World::update_gui(int64_t actor_id, int64_t place_id, Gui *gui) {
assert(stack_is_clear());
2021-11-16 13:14:59 -05:00
gui->clear(place_id);
2021-09-10 17:06:07 -04:00
lua_State *L = state();
LuaVar actor, place, ugui, func, tangibles, mt, index;
LuaExtStack LS(L, actor, place, ugui, func, tangibles, mt, index);
2021-09-10 17:06:07 -04:00
// Get the actor and place.
LS.rawget(tangibles, LuaRegistry, "tangibles");
LS.rawget(actor, tangibles, actor_id);
LS.rawget(place, tangibles, place_id);
if (!LS.istable(actor) || !LS.istable(place)) {
return;
}
// Get the interface closure.
LS.getmetatable(mt, place);
if (!LS.istable(mt)) {
return;
}
LS.rawget(index, mt, "__index");
if (!LS.istable(index)) {
return;
}
LS.rawget(func, index, "interface");
if (!LS.isfunction(func)) {
return;
}
// Call the interface function.
lua_pushvalue(L, func.index());
lua_pushvalue(L, actor.index());
lua_pushvalue(L, place.index());
Gui::store_global_pointer(L, gui);
open_lthread_state(actor_id, place_id, 0, false, false);
eng::string msg = traceback_pcall(L, 2, 0);
close_lthread_state();
2021-09-10 17:06:07 -04:00
Gui::store_global_pointer(L, nullptr);
if (!msg.empty()) {
2021-11-16 13:14:59 -05:00
gui->clear(0);
util::dprint(msg);
2021-09-10 17:06:07 -04:00
return;
}
}
2023-04-10 17:58:44 -04:00
// This is called from World::update_source, and also
// from World::patch_source in the difference transmitter.
//
// For the moment, errors are channeled to util::dprint,
// and 'print' statements just go to std::cerr. Neither
// of these is ideal. We need to get serious about setting
// up error handling.
//
// We also need to figure out a solution for what happens if
// some lua source file tries to modify, say, tangible state
// in top-level code.
//
void World::rebuild_sourcedb() {
for (const eng::string &mod: source_db_.modules()) {
2023-04-11 16:36:30 -04:00
open_lthread_state(0, 0, 0, false, true);
2023-04-10 17:58:44 -04:00
eng::string err = source_db_.rebuild_module(mod);
2023-04-11 16:36:30 -04:00
eng::string prints = lthread_prints_->str();
lthread_prints_.reset();
close_lthread_state();
if (!err.empty() || !prints.empty()) {
2023-04-14 15:13:59 -04:00
util::dprint("Loading Module ", mod);
2023-04-11 16:36:30 -04:00
if (!err.empty()) util::dprint(err);
if (!prints.empty()) util::dprint(prints);
2023-04-10 17:58:44 -04:00
}
}
}
2021-12-15 14:18:19 -05:00
void World::update_source(const util::LuaSourceVec &source) {
assert(stack_is_clear());
source_db_.update(source);
2023-04-10 17:58:44 -04:00
rebuild_sourcedb();
assert(stack_is_clear());
}
void World::update_source(const util::LuaSourcePtr &source) {
if (source != nullptr) {
update_source(*source);
}
}
void World::update_source(std::string_view sourcepack) {
if (!sourcepack.empty()) {
try {
StreamBuffer sb(sourcepack);
util::LuaSourceVec sv;
SourceDB::deserialize_source(&sv, &sb);
update_source(sv);
} catch (const StreamException &ex) {
return;
}
}
}
2022-05-16 17:16:42 -04:00
void World::http_response(const HttpParser &response) {
// Find the request.
2022-05-06 13:16:27 -04:00
auto iter = http_requests_.find(response.request_id());
if (iter == http_requests_.end()) {
return;
}
2022-05-06 13:16:27 -04:00
HttpClientRequest request = iter->second;
http_requests_.erase(iter);
lua_State *CO;
{
// Get the place and thread as lua objects.
LuaVar tangibles, place, mt, threads, thinfo, thread;
LuaExtStack LS(state(), tangibles, place, mt, threads, thinfo, thread);
LS.rawget(tangibles, LuaRegistry, "tangibles");
LS.rawget(place, tangibles, request.place_id());
if (!LS.istable(place)) {
return;
}
LS.getmetatable(mt, place);
if (!LS.istable(mt)) {
return;
}
LS.rawget(threads, mt, "threads");
if (!LS.istable(threads)) {
return;
}
LS.rawget(thinfo, threads, request.thread_id());
if (!LS.istable(thinfo)) {
return;
}
LS.rawget(thread, thinfo, "thread");
if (!LS.isthread(thread)) {
return;
}
CO = LS.ckthread(thread);
}
// Push the response onto the awakening thread.
lua_pushnil(CO);
LuaSpecial responsetable(lua_gettop(CO));
LuaCoreStack LSCO(CO);
response.store(LSCO, responsetable);
// Awaken the thread, with its new return value.
schedule(0, request.thread_id(), request.place_id());
run_scheduled_threads();
}
2022-05-16 17:16:42 -04:00
void World::http_responses(const HttpParserVec &responses) {
for (const HttpParser &response : responses) {
2022-05-06 13:16:27 -04:00
http_response(response);
}
}
void World::abort_all_http_requests(int status_code, std::string_view error) {
2022-05-16 17:16:42 -04:00
HttpParser abortresponse;
abortresponse.fail(status_code, error);
2022-05-06 13:16:27 -04:00
while (!http_requests_.empty()) {
abortresponse.set_request_id(http_requests_.begin()->first);
http_response(abortresponse);
}
}
2022-05-20 17:12:58 -04:00
HttpServerResponse World::http_serve(const HttpParser &request) {
assert(stack_is_clear());
HttpServerResponse response;
// We're only supposed to be passed complete requests.
assert(request.complete());
// If the request is HTTP/1.1, then the response should be HTTP/1.1
response.set_http11(request.http11());
// If the incoming request has already been detected to be
// invalid by the HTTP parser, then just send the error
// message back to the client without involving lua at all.
if (request.errstatus()) {
response.fail(request.status(), request.error());
return response;
}
lua_State *L = state();
LuaVar www, func, reqtab;
LuaExtStack LS(L, www, func, reqtab);
2022-05-20 17:12:58 -04:00
// Get the www class. If there's no such class,
// return a 503 Service Unavailable to the client.
eng::string err = LS.getclass(www, "www");
if (!err.empty()) {
response.fail(503, "class www doesn't exist");
return response;
}
// Get the name of the desired function.
std::string_view orig_fn = request.first_path_component("index");
eng::string lua_fn = HttpParser::to_lua_identifier(orig_fn);
if (lua_fn.empty()) {
response.fail(404, util::ss("cannot convert to lua function name: ", orig_fn));
return response;
}
2022-05-20 17:12:58 -04:00
// Get the closure. If there's no such closure,
// return a 404 Not Found to the client.
LS.rawget(func, www, lua_fn);
2022-05-20 17:12:58 -04:00
if (!LS.isfunction(func)) {
response.fail(404, util::ss("no such lua function: www.", lua_fn));
2022-05-20 17:12:58 -04:00
return response;
}
// Store the request into a lua table.
request.store(LS, reqtab);
// Call the function.
int oldtop = lua_gettop(L);
lua_pushvalue(L, func.index());
lua_pushvalue(L, reqtab.index());
Gui::store_global_pointer(L, nullptr);
open_lthread_state(0, 0, 0, false, false);
eng::string msg = traceback_pcall(L, 1, LUA_MULTRET);
close_lthread_state();
// If the call threw an error, return
// a 500 Internal Server Error to the client.
if (!msg.empty()) {
response.fail(500, msg);
return response;
}
// If the call didn't return a single table, return
// a 500 Internal Server Error to the client.
int newtop = lua_gettop(L);
if ((newtop != oldtop + 1) || (!lua_istable(L, newtop))) {
response.fail(500, util::ss("lua function www.", lua_fn, " didn't return a table"));
2022-05-20 17:12:58 -04:00
return response;
}
// Try to convert the table into a response.
2022-07-22 17:07:40 -04:00
LuaKeywordParser kp(LS, LuaSpecial(newtop));
response.configure(kp);
2022-05-20 17:12:58 -04:00
response.set_defaults();
2022-07-22 17:07:40 -04:00
eng::string kperr = kp.final_check();
if (!kperr.empty()) {
response.fail(500, kperr);
}
2022-05-20 17:12:58 -04:00
return response;
}
void World::run_unittests() {
assert(stack_is_clear());
source_db_.run_unittests();
assert(stack_is_clear());
2021-10-15 14:47:12 -04:00
}
2021-09-10 17:06:07 -04:00
void World::invoke(const Invocation &inv) {
switch (inv.kind()) {
case Invocation::KIND_CHOOSE:
invoke_choose(inv.actor(), inv.place(), inv.datapack());
break;
case Invocation::KIND_ENGIO:
invoke_engio(inv.actor(), inv.place(), inv.datapack());
2021-09-10 17:06:07 -04:00
break;
2021-10-15 14:47:12 -04:00
case Invocation::KIND_LUA:
invoke_lua(inv.actor(), inv.place(), inv.datapack());
2021-10-15 14:47:12 -04:00
break;
case Invocation::KIND_FLUSH_PRINTS:
invoke_flush_prints(inv.actor(), inv.place(), inv.datapack());
break;
2021-11-26 15:45:36 -05:00
case Invocation::KIND_TICK:
invoke_tick(inv.actor(), inv.place(), inv.datapack());
2021-12-15 14:18:19 -05:00
break;
case Invocation::KIND_LUA_SOURCE:
invoke_lua_source(inv.actor(), inv.place(), inv.datapack());
2021-12-15 14:18:19 -05:00
break;
2021-09-10 17:06:07 -04:00
default:
// Do nothing. Standard behavior for any invalid command is to
// simply do nothing at all. Perhaps eventually we may add a flag
// to the world model to indicate that we've detected an invalid
// command, to allow us to close the connection to a client that
// is misbehaving.
break;
}
}
bool World::spawn(LuaCoreStack &LS0, int64_t actor_id, int64_t place_id, LuaSlot func,
bool passactorplace, int nargs, bool print) {
lua_State *L = LS0.state();
LuaVar actor, place, mt, index, tangibles, thread, threads, thinfo;
LuaExtStack LS(L, actor, place, mt, index, tangibles, thread, threads, thinfo);
// Get the actor and place, C++ version. Make sure both exist.
Tangible *tactor = tangible_get(actor_id);
Tangible *tplace = tangible_get(place_id);
if ((tactor == nullptr) || (tplace == nullptr)) {
return false;
}
// Get the actor and place, lua version. Make sure both exist.
LS.rawget(tangibles, LuaRegistry, "tangibles");
LS.rawget(actor, tangibles, actor_id);
LS.rawget(place, tangibles, place_id);
if (!LS.istable(actor) || !LS.istable(place)) {
return false;
}
// Get an ID for the thread.
// We currently always use the actor pool. We may extend
// this to allow the use of the place pool.
int64_t tid = tactor->id_player_pool_.get_one();
// Get the place's metatable.
LS.getmetatable(mt, place);
if (!LS.istable(mt)) {
return false;
}
// If the function is a string, look it up in the place's class.
if (LS.isstring(func)) {
LS.rawget(index, mt, "__index");
if (!LS.istable(index)) {
return false;
}
LS.rawget(func, index, func);
}
// Make sure the function is a function.
if (!LS.isfunction(func)) {
return false;
}
// Create a new thread. Push function, and maybe actor and place.
lua_State *CO = LS.newthread(thread);
lua_pushvalue(L, func.index());
if (passactorplace) {
lua_pushvalue(L, actor.index());
lua_pushvalue(L, place.index());
}
lua_xmove(L, CO, passactorplace ? 3:1);
// Push any extra arguments. Extra arguments were pushed onto
// the lua stack before calling spawn, so they are located at oldtop.
if (nargs > 0) {
int base = LS.oldtop() - nargs + 1;
for (int i = 0; i < nargs; i++) {
lua_pushvalue(L, base + i);
}
lua_xmove(L, CO, nargs);
}
// Create the thread info table.
LS.newtable(thinfo);
LS.rawset(thinfo, "thread", thread);
LS.rawset(thinfo, "actorid", actor_id);
LS.rawset(thinfo, "isnew", true);
LS.rawset(thinfo, "useppool", true);
LS.rawset(thinfo, "print", print);
// Store the thread into place's thread table.
LS.rawget(threads, mt, "threads");
if (!LS.istable(threads)) {
return false;
}
LS.rawset(threads, tid, thinfo);
schedule(0, tid, place_id);
return true;
}
void World::invoke_flush_prints(int64_t actor_id, int64_t place_id, std::string_view datapack) {
assert(stack_is_clear());
// Check argument sanity.
if (actor_id != place_id) {
return;
}
int64_t line = sv::to_int64(datapack, -1);
if ((line < 0)||(line > INT_MAX)) {
return;
}
Tangible *tactor = tangible_get(actor_id);
if (tactor == nullptr) {
return;
}
2021-11-14 15:57:18 -05:00
tactor->print_buffer_.discard_upto(line);
assert(stack_is_clear());
}
void World::invoke_lua(int64_t actor_id, int64_t place_id, std::string_view datapack) {
2021-10-15 14:47:12 -04:00
assert(stack_is_clear());
{
lua_State *L = state();
LuaVar func;
LuaExtStack LS(L, func);
// create the compiled closure.
int status = luaL_loadbuffer(L, datapack.data(), datapack.size(), "=invoke");
lua_replace(L, func.index());
if (status != LUA_OK) {
// The closure is actually an error message. Do nothing.
// This should normally not happen: LuaConsole should filter
// out syntax errors.
return;
}
2021-10-15 14:47:12 -04:00
// Spawn the thread and run it.
int nargs = 0;
spawn(LS, actor_id, place_id, func, false, nargs, true);
}
run_scheduled_threads();
2021-10-15 14:47:12 -04:00
assert(stack_is_clear());
2021-10-05 12:54:37 -04:00
}
void World::invoke_choose(int64_t actor_id, int64_t place_id, std::string_view datapack) {
2021-09-10 17:06:07 -04:00
assert(stack_is_clear());
// Validate that the action is legal.
Gui validation_gui;
update_gui(actor_id, place_id, &validation_gui);
if (!validation_gui.has_action(datapack)) {
2021-09-10 17:06:07 -04:00
return;
}
2021-12-28 14:07:15 -05:00
// Make sure the action starts with "cb_"
if (!sv::has_prefix(datapack, "cb_")) {
2021-12-28 14:07:15 -05:00
return;
}
{
lua_State *L = state();
LuaVar func, invdata;
LuaExtStack LS(L, func, invdata);
2021-09-10 17:06:07 -04:00
LS.set(func, datapack);
2021-09-10 17:06:07 -04:00
// We're planning to add the ability to pass data in as a table.
// For now the table is always empty.
LS.newtable(invdata);
2021-09-10 17:06:07 -04:00
// Spawn the thread and run it.
int nargs = 1;
lua_pushvalue(L, invdata.index());
spawn(LS, actor_id, place_id, func, true, nargs, false);
2021-09-10 17:06:07 -04:00
}
2021-11-26 15:45:36 -05:00
run_scheduled_threads();
2021-09-10 17:06:07 -04:00
assert(stack_is_clear());
}
2023-10-31 13:31:42 -04:00
// Read a SimpleDynamic value from the streambuffer and push
// it onto the lua stack.
void push_simple_dynamic(lua_State *L, StreamBuffer *sb) {
SimpleDynamicTag type = sb->read_simple_dynamic_tag();
switch (type) {
case SimpleDynamicTag::NUMBER: {
lua_pushnumber(L, sb->read_double());
break;
}
case SimpleDynamicTag::BOOLEAN: {
lua_pushboolean(L, sb->read_bool() ? 1:0);
break;
}
case SimpleDynamicTag::STRING: {
std::string_view s = sb->read_string_view();
lua_pushlstring(L, s.data(), s.size());
break;
}
case SimpleDynamicTag::VECTOR: {
double x = sb->read_double();
double y = sb->read_double();
double z = sb->read_double();
lua_newtable(L);
lua_pushnumber(L, x);
lua_rawseti(L, -2, 1);
lua_pushnumber(L, y);
lua_rawseti(L, -2, 2);
lua_pushnumber(L, z);
lua_rawseti(L, -2, 3);
break;
}
default: throw StreamCorruption();
}
}
void World::invoke_engio(int64_t actor_id, int64_t place_id, std::string_view datapack) {
assert(stack_is_clear());
// Use a streambuffer to parse the datapack.
StreamBuffer datasb(datapack);
// Extract the function name from the datapack.
eng::string funcname;
try {
funcname = datasb.read_string_limit(100);
} catch (const StreamException &ex) {
return;
}
if (!sv::is_lua_id(funcname)) {
return;
}
{
lua_State *L = state();
2024-02-27 17:32:08 -05:00
LuaVar engio, func;
LuaExtStack LS(L, engio, func);
2024-02-27 17:32:08 -05:00
// Get the closure (engio.funcname).
eng::string err = LS.getclass(engio, "engio");
if ((!err.empty()) || (!LS.istable(engio))) {
return;
}
2024-02-27 17:32:08 -05:00
LS.rawget(func, engio, funcname);
2023-10-31 13:31:42 -04:00
2024-02-27 17:32:08 -05:00
// Spawn a thread, pushing extra arguments from the datapack.
int nargs = 0;
try {
while (!datasb.empty()) {
2023-10-31 13:31:42 -04:00
push_simple_dynamic(L, &datasb);
nargs++;
}
} catch (const StreamException &exc) {
return;
}
2024-02-27 17:32:08 -05:00
spawn(LS, actor_id, place_id, func, true, nargs, false);
}
run_scheduled_threads();
assert(stack_is_clear());
}
void World::invoke_tick(int64_t actor_id, int64_t place_id, std::string_view datapack) {
if (!is_authoritative()) {
2021-12-15 14:18:19 -05:00
return;
}
clock_ += 1;
run_scheduled_threads();
}
void World::invoke_lua_source(int64_t actor_id, int64_t place_id, std::string_view datapack) {
if (!is_authoritative()) {
2021-12-15 14:18:19 -05:00
return;
}
// We need some kind of authentication here.
update_source(datapack);
2021-11-26 15:45:36 -05:00
}
void World::guard_blockable(lua_State *L, const char *fn) {
if (lthread_thread_id_ == 0) {
// in a probe, blocking functions like http.get throw an error.
luaL_error(L, "cannot %s in a probe", fn);
assert(false);
}
if (!is_authoritative()) {
// in a nonauth model, blocking functions like http.get are converted to nopredict.
lua_yield(L, 0);
luaL_error(L, "unexplained nopredict failure in %s", fn);
assert(false);
}
}
void World::guard_nopredict(lua_State *L, const char *fn) {
// Caution: this code must be equivalent to the
// code in LuaCoreStack::guard_nopredict.
if (lthread_thread_id_ == 0) {
return;
}
if (!is_authoritative()) {
lua_yield(L, 0);
luaL_error(L, "unexplained nopredict failure in %s", fn);
}
}
void World::schedule(int64_t clk, int64_t thid, int64_t plid) {
2022-05-06 15:50:07 -04:00
if (clk > 0) {
assert(is_authoritative());
}
thread_sched_.add(clk, thid, plid);
}
2021-11-26 15:45:36 -05:00
void World::run_scheduled_threads() {
2021-09-10 17:06:07 -04:00
assert(stack_is_clear());
lua_State *L = state();
LuaVar tangibles, place, mt, threads, thinfo, actorid, isnew, useppool, thread, print;
LuaExtStack LS(L, tangibles, place, mt, threads, thinfo, actorid, isnew, useppool, thread, print);
2021-09-10 17:06:07 -04:00
LS.rawget(tangibles, LuaRegistry, "tangibles");
2021-11-26 15:45:36 -05:00
while (thread_sched_.ready(clock_)) {
2021-09-10 17:06:07 -04:00
SchedEntry sched = thread_sched_.pop();
LS.rawget(place, tangibles, sched.place_id());
if (!LS.istable(place)) {
continue;
}
LS.getmetatable(mt, place);
if (!LS.istable(mt)) {
continue;
}
LS.rawget(threads, mt, "threads");
if (!LS.istable(threads)) {
continue;
}
LS.rawget(thinfo, threads, sched.thread_id());
if (!LS.istable(thinfo)) {
continue;
}
LS.rawget(actorid, thinfo, "actorid");
if (!LS.isnumber(actorid)) {
continue;
}
LS.rawget(isnew, thinfo, "isnew");
if (!LS.isboolean(isnew)) {
continue;
}
LS.rawget(useppool, thinfo, "useppool");
if (!LS.isboolean(useppool)) {
continue;
}
LS.rawget(thread, thinfo, "thread");
2021-09-10 17:06:07 -04:00
if (!LS.isthread(thread)) {
continue;
}
// Resume the coroutine.
lua_State *CO = LS.ckthread(thread);
open_lthread_state(LS.ckinteger(actorid), sched.place_id(), sched.thread_id(), LS.ckboolean(useppool), true);
int nargs = LS.ckboolean(isnew) ? (lua_gettop(CO) - 1) : lua_gettop(CO);
int status = lua_resume(CO, nullptr, nargs);
std::ostream *ostream = lthread_print_stream();
if (status == LUA_OK) {
// Successfully ran to completion. Print any return values.
// Remove from thread table.
LS.rawget(print, thinfo, "print");
LS.rawset(threads, sched.thread_id(), LuaNil);
LuaCoreStack LSCO(CO);
if (LS.ckboolean(print)) {
for (int i = 1; i <= lua_gettop(CO); i++) {
pprint(LSCO, LuaSpecial(i), PrettyPrintOptions(), ostream);
2021-12-15 23:03:43 -05:00
(*ostream) << std::endl;
}
}
} else if (status == LUA_YIELD) {
if (is_authoritative()) {
LS.rawset(thinfo, "isnew", false);
LS.rawset(thinfo, "useppool", false);
} else {
// When a nonauthoritative model yields, for any reason,
// the thread is discarded. This is also used as a way to implement
// nopredict: the thread that wants to 'nopredict' just yields,
// knowing that this will cause it to be killed.
LS.rawset(threads, sched.thread_id(), LuaNil);
}
2021-09-10 17:06:07 -04:00
} else {
// Generated an error. Add a traceback, print, and kill the coroutine.
// Currently, the error is sent to the actor. That seems... not right in the long run.
if (is_authoritative()) {
traceback_coroutine(CO);
(*ostream) << lua_tostring(CO, -1);
}
2021-09-10 17:06:07 -04:00
LS.rawset(threads, sched.thread_id(), LuaNil);
}
close_lthread_state();
2021-09-10 17:06:07 -04:00
}
}
int64_t World::alloc_id_predictable() {
if (!lthread_use_ppool_) {
return id_global_pool_.get_one();
}
Tangible *t = tangible_get(lthread_actor_id_);
if (t == nullptr) {
return id_global_pool_.get_one();
}
return t->id_player_pool_.get_one();
}
const PrintBuffer *World::get_printbuffer(int64_t actor_id) {
Tangible *actor = tangible_get(actor_id);
if (actor != nullptr) {
2021-11-14 15:57:18 -05:00
return &actor->print_buffer_;
}
return nullptr;
}
void World::clear_lthread_state() {
lthread_prints_.reset();
lthread_actor_id_ = 0;
lthread_place_id_ = 0;
lthread_thread_id_ = 0;
lthread_use_ppool_ = false;
}
void World::open_lthread_state(int64_t actor, int64_t place, int64_t thread, bool ppool, bool prints) {
lthread_actor_id_ = actor;
lthread_place_id_ = place;
lthread_thread_id_ = thread;
lthread_use_ppool_ = ppool;
if (prints) {
lthread_prints_.reset(new eng::ostringstream);
} else {
lthread_prints_.reset();
}
}
void World::close_lthread_state() {
// Copy prints from lthread_prints_ stringstream into
// the appropriate actor's PrintBuffer. If for some reason
// there isn't an actor, or if the actor doesn't have a PrintBuffer,
// send the output to std::cerr.
if (lthread_prints_ != nullptr) {
const eng::string &output = lthread_prints_->str();
if (output.size() > 0) {
Tangible *actor = tangible_get(lthread_actor_id_);
if (actor != nullptr) {
actor->print_buffer_.add_string(output, is_authoritative());
}
}
}
// Now clean up everything.
clear_lthread_state();
}
std::ostream *World::lthread_print_stream() const {
if (lthread_prints_ != nullptr) {
return lthread_prints_.get();
} else {
return &std::cerr;
}
}
void World::set_global(LuaCoreStack &LS0, const eng::string &gvar, LuaSlot value) {
lua_State *L = LS0.state();
LuaVar globaldb, copy;
LuaExtStack LS(L, globaldb, copy);
// Serialize then deserialize the data, to produce a copy.
StreamBuffer sb;
eng::string error = serialize_lua(LS, value, &sb);
if (!error.empty()) {
luaL_error(L, "%s", error.c_str());
return;
}
eng::string serialized(sb.view());
error = deserialize_lua(LS, copy, &sb);
if (!error.empty()) {
luaL_error(L, "%s", error.c_str());
return;
}
// Store the copy in the globalDB.
LS.rawget(globaldb, LuaRegistry, "globaldb");
LS.rawset(globaldb, gvar, copy);
// Store the serialized blob.
gvname_to_serial_[gvar] = serialized;
// Implement the tracking so that we can rapidly determine which global
// variables need to be difference transmitted.
//
// In the master model, we generate a sequence number for the assignment.
// We store the mapping from global variable name to that sequence number
// and vice versa.
//
// On the client side, we just record the global variable in a list
// of recently modified globals.
//
if (is_authoritative()) {
int64_t &seqno = gvname_to_seqno_[gvar];
seqno_to_gvname_.erase(seqno);
seqno = assign_seqno_++;
seqno_to_gvname_[seqno] = gvar;
} else {
gvname_modified_.insert(gvar);
}
}
2021-09-10 17:06:07 -04:00
void World::serialize(StreamBuffer *sb) {
assert(stack_is_clear());
assert(redirects_.empty());
// int64_t wc0 = sb->total_writes();
lua_snap_.serialize(sb);
id_global_pool_.serialize(sb);
2021-11-26 15:45:36 -05:00
sb->write_int64(clock_);
2021-09-10 17:06:07 -04:00
thread_sched_.serialize(sb);
http_requests_.serialize(sb);
2021-09-10 17:06:07 -04:00
sb->write_uint32(tangibles_.size());
for (const auto &p : tangibles_) {
sb->write_int64(p.first);
p.second->serialize(sb);
}
assert(stack_is_clear());
}
void World::deserialize(StreamBuffer *sb) {
assert(stack_is_clear());
redirects_.clear();
lua_snap_.deserialize(sb);
id_global_pool_.deserialize(sb);
2021-11-26 15:45:36 -05:00
clock_ = sb->read_int64();
2021-09-10 17:06:07 -04:00
thread_sched_.deserialize(sb);
http_requests_.deserialize(sb);
2021-09-10 17:06:07 -04:00
// Mark all tangibles for deletion by setting ID to zero.
for (const auto &p : tangibles_) {
p.second->plane_item_.set_id(0);
}
// Deserialize tangibles.
size_t ntan = sb->read_uint32();
for (size_t i = 0; i < ntan; i++) {
int64_t id = sb->read_int64();
UniqueTangible &t = tangibles_[id];
2021-09-10 17:06:07 -04:00
if (t == nullptr) {
t.reset(new Tangible(this, id));
} else {
t->plane_item_.set_id(id);
}
t->deserialize(sb);
}
// Delete tangibles that didn't get deserialized.
for (auto iter = tangibles_.begin(); iter != tangibles_.end(); ) {
if (iter->second->plane_item_.id() == 0) {
tangibles_.erase(iter++);
} else {
++iter;
}
}
// After a save and load, http requests no longer should exist
abort_all_http_requests(425, "http requests aborted by loading a save game");
2021-09-10 17:06:07 -04:00
assert(stack_is_clear());
}
void World::snapshot() {
2021-11-09 16:27:39 -05:00
assert(snapshot_.empty());
2021-09-10 17:06:07 -04:00
serialize(&snapshot_);
2021-11-09 16:27:39 -05:00
assert(!snapshot_.empty());
2021-09-10 17:06:07 -04:00
}
void World::rollback() {
2021-10-07 14:58:20 -04:00
assert(!snapshot_.empty());
2021-09-10 17:06:07 -04:00
deserialize(&snapshot_);
2021-11-09 16:27:39 -05:00
assert(snapshot_.empty());
2021-09-10 17:06:07 -04:00
}
2023-02-14 13:14:18 -05:00
// This is the main routine for the DLL. We have to use a registration device
// to register this main routine with DrivenEngine. DrivenEngine will then call
// it exactly once the first time that the driver initializes an EngineWrapper.
//
void engine_initialization() {
SourceDB::register_lua_builtins();
}
static DrivenEngineInitializerReg eireg(engine_initialization);