JezuzLizard 4135e7653a Add script errors for added builtins.
Fix generatepath() to return nodenums instead of pathnodes to workaround an obscure VM bug which is caused by sending a random origin as an argument and returning pathnodes.
2023-03-27 16:34:28 -07:00

382 lines
9.6 KiB
C++

#include <stdinc.hpp>
#include "loader/component_loader.hpp"
#include <json.hpp>
#include <utils/io.hpp>
#include <utils/hook.hpp>
#include <utils/string.hpp>
namespace gsc
{
std::unordered_map<std::string, game::BuiltinFunction> functions;
std::unordered_map<std::string, game::BuiltinMethod> methods;
namespace
{
void* original_scr_get_gsc_funcs_jump_loc;
void* original_scr_get_method_funcs_call_loc;
game::BuiltinFunction find_function(const std::string& name)
{
auto itr = functions.find(name);
if (itr == functions.end())
{
return 0;
}
return itr->second;
}
game::BuiltinMethod find_method(const std::string& name)
{
auto itr = methods.find(name);
if (itr == methods.end())
{
return 0;
}
return itr->second;
}
game::BuiltinFunction scr_get_funcs_stub(const char** pName, int* type)
{
const auto func = find_function(*pName);
if (func)
{
return func;
}
return 0;
}
game::BuiltinMethod scr_get_methods_stub(const char** pName, int* type)
{
const auto meth = find_method(*pName);
if (meth)
{
return meth;
}
return 0;
}
void __declspec(naked) original_scr_get_gsc_funcs_hook()
{
__asm
{
push eax;
pushad;
lea eax, [esp + 0x24 + 0x2C - 0x1C];
push eax;
push edx;
call scr_get_funcs_stub;
add esp, 8;
mov[esp + 0x20], eax;
popad;
pop eax;
test eax, eax;
jnz og;
// pluto
push original_scr_get_gsc_funcs_jump_loc;
retn;
og:
// retn
add esp, 4;
push 0x682DC8;
retn;
}
}
void __declspec(naked) original_scr_get_method_funcs_hook()
{
__asm
{
push eax;
pushad;
push edi;
push esi;
call scr_get_methods_stub;
add esp, 8;
mov[esp + 0x20], eax; // move answer into eax when pop happens
popad;
pop eax;
test eax, eax;
jz pluto_code;
retn;
pluto_code:
push original_scr_get_method_funcs_call_loc;
retn;
}
}
}
namespace function
{
void add(const std::string& name, const game::BuiltinFunction function)
{
functions.insert_or_assign(name, function);
}
}
namespace method
{
void add(const std::string& name, const game::BuiltinMethod method)
{
methods.insert_or_assign(name, method);
}
}
class component final : public component_interface
{
public:
void post_unpack() override
{
original_scr_get_gsc_funcs_jump_loc = utils::hook::get_displacement_addr(0x682D99);
original_scr_get_method_funcs_call_loc = utils::hook::get_displacement_addr(0x683043);
utils::hook::jump(0x682D99, original_scr_get_gsc_funcs_hook);
utils::hook::call(0x683043, original_scr_get_method_funcs_hook);
function::add("funny_func", []()
{
printf("How funny\n");
printf("You found the funny monkey funkey\n");
});
method::add("is_monkey", [](game::scr_entref_s ent)
{
printf("Monkey number %d\n", ent.entnum);
});
function::add("inttest", []()
{
int answer = game::Scr_GetInt(game::SCRIPTINSTANCE_SERVER, 0);
answer += 69;
game::Scr_AddInt(game::SCRIPTINSTANCE_SERVER, answer);
});
function::add("floattest", []()
{
float answer = game::Scr_GetFloat(game::SCRIPTINSTANCE_SERVER, 0);
answer *= 69.0f;
game::Scr_AddFloat(game::SCRIPTINSTANCE_SERVER, answer);
});
function::add("stringtest", []()
{
char* answer = game::Scr_GetString(game::SCRIPTINSTANCE_SERVER, 0);
std::string string(answer);
string += " haha funny"s;
game::Scr_AddString(game::SCRIPTINSTANCE_SERVER, string.data());
});
function::add("istringtest", []()
{
const char* answer = game::Scr_GetIString(game::SCRIPTINSTANCE_SERVER, 0);
std::string string(answer);
string += " haha funny"s;
game::Scr_AddIString(game::SCRIPTINSTANCE_SERVER, string.data());
});
function::add("vectortest", []()
{
float answer[3] = {};
game::Scr_GetVector(game::SCRIPTINSTANCE_SERVER, 0, answer);
answer[0] += 69.0f;
game::Scr_AddVector(game::SCRIPTINSTANCE_SERVER, answer);
});
function::add("undefinedtest", []()
{
game::Scr_AddUndefined(game::SCRIPTINSTANCE_SERVER);
});
function::add("pathnodetest", []()
{
game::pathnode_t* node = game::Scr_GetPathnode(game::SCRIPTINSTANCE_SERVER);
game::Scr_AddPathnode(game::SCRIPTINSTANCE_SERVER, node);
});
function::add("numberindexedarraytest", []()
{
game::Scr_MakeArray(game::SCRIPTINSTANCE_SERVER);
game::Scr_AddFloat(game::SCRIPTINSTANCE_SERVER, 69.0f);
game::Scr_AddArray(game::SCRIPTINSTANCE_SERVER);
game::Scr_AddInt(game::SCRIPTINSTANCE_SERVER, 420);
game::Scr_AddArray(game::SCRIPTINSTANCE_SERVER);
game::Scr_AddString(game::SCRIPTINSTANCE_SERVER, "ree");
game::Scr_AddArray(game::SCRIPTINSTANCE_SERVER);
});
function::add("stringindexedarraytest", []()
{
game::Scr_MakeArray(game::SCRIPTINSTANCE_SERVER);
std::string key1("key1");
game::Scr_AddFloat(game::SCRIPTINSTANCE_SERVER, 69.0f);
game::Scr_AddArrayStringIndexed(game::SCRIPTINSTANCE_SERVER, game::SL_GetStringOfSize(game::SCRIPTINSTANCE_SERVER, key1.data(), 0, key1.length() + 1));
std::string key2("key2");
game::Scr_AddInt(game::SCRIPTINSTANCE_SERVER, 420);
game::Scr_AddArrayStringIndexed(game::SCRIPTINSTANCE_SERVER, game::SL_GetStringOfSize(game::SCRIPTINSTANCE_SERVER, key2.data(), 0, key2.length() + 1));
std::string key3("key3");
game::Scr_AddString(game::SCRIPTINSTANCE_SERVER, "ree");
game::Scr_AddArrayStringIndexed(game::SCRIPTINSTANCE_SERVER, game::SL_GetStringOfSize(game::SCRIPTINSTANCE_SERVER, key3.data(), 0, key3.length() + 1));
});
method::add("entitytest", [](game::scr_entref_s ent)
{
if (ent.classnum != game::CLASS_NUM_ENTITY)
{
game::Scr_Error("Not an entity", game::SCRIPTINSTANCE_SERVER, false);
return;
}
game::gentity_s* ent2 = &game::g_entities[ent.entnum];
game::Scr_AddEntity(game::SCRIPTINSTANCE_SERVER, ent2);
});
method::add("getlinkednodes", [](game::scr_entref_s ent)
{
if (ent.classnum != game::CLASS_NUM_PATHNODE)
{
game::Scr_Error("Not a pathnode", game::SCRIPTINSTANCE_SERVER, false);
return;
}
auto primary_node = &(*game::gameWorldCurrent)->path.nodes[ent.entnum];
game::Scr_MakeArray(game::SCRIPTINSTANCE_SERVER);
for (auto i = 0; i < primary_node->constant.totalLinkCount; i++)
{
auto linked_node = &(*game::gameWorldCurrent)->path.nodes[primary_node->constant.Links[i].nodeNum];
game::Scr_AddPathnode(game::SCRIPTINSTANCE_SERVER, linked_node);
game::Scr_AddArray(game::SCRIPTINSTANCE_SERVER);
}
});
method::add("getnodenumber", [](game::scr_entref_s ent)
{
if (ent.classnum != game::CLASS_NUM_PATHNODE)
{
game::Scr_Error("Not a pathnode", game::SCRIPTINSTANCE_SERVER, false);
return;
}
auto node = &(*game::gameWorldCurrent)->path.nodes[ent.entnum];
auto entnum = node - (*game::gameWorldCurrent)->path.nodes;
game::Scr_AddInt(game::SCRIPTINSTANCE_SERVER, entnum);
});
function::add("getnodebynumber", []()
{
auto node_num = game::Scr_GetInt(game::SCRIPTINSTANCE_SERVER, 0);
if (node_num == game::g_path->actualNodeCount)
{
game::Scr_AddUndefined(game::SCRIPTINSTANCE_SERVER);
return;
}
if (node_num < 0 || node_num > game::g_path->actualNodeCount)
{
game::Scr_Error(utils::string::va("Number %d is not valid for a node", node_num), game::SCRIPTINSTANCE_SERVER, false);
return;
}
auto node = &(*game::gameWorldCurrent)->path.nodes[node_num];
game::Scr_AddPathnode(game::SCRIPTINSTANCE_SERVER, node);
});
function::add("generatepath", []()
{
auto path = std::make_unique<game::path_t>();
float start_pos[3] = {};
float goal_pos[3] = {};
auto team = "neutral"s;
auto allow_negotiation_links = false;
game::Scr_GetVector(game::SCRIPTINSTANCE_SERVER, 0, start_pos);
game::Scr_GetVector(game::SCRIPTINSTANCE_SERVER, 1, goal_pos);
if (game::Scr_GetNumParam(game::SCRIPTINSTANCE_SERVER) >= 3)
{
if (game::Scr_GetType(game::SCRIPTINSTANCE_SERVER, 2) != game::VAR_UNDEFINED)
{
team = game::Scr_GetString(game::SCRIPTINSTANCE_SERVER, 2);
}
if (game::Scr_GetNumParam(game::SCRIPTINSTANCE_SERVER) >= 4)
{
allow_negotiation_links = game::Scr_GetInt(game::SCRIPTINSTANCE_SERVER, 3);
}
}
if (!game::team_map.contains(team))
{
game::Scr_Error(utils::string::va("Team %s is not valid", team.data()), game::SCRIPTINSTANCE_SERVER, false);
return;
}
auto eTeam = game::team_map.at(team);
auto success = game::Path_FindPath(path.get(), eTeam, start_pos, goal_pos, allow_negotiation_links);
if (!success)
{
game::Scr_AddUndefined(game::SCRIPTINSTANCE_SERVER);
return;
}
game::Scr_MakeArray(game::SCRIPTINSTANCE_SERVER);
for (auto i = 0; i < path->wPathLen; i++)
{
//Return the number of the node instead of the node itself because of spooky GSC VM corruption
game::Scr_AddInt(game::SCRIPTINSTANCE_SERVER, path->pts[i].iNodeNum);
game::Scr_AddArray(game::SCRIPTINSTANCE_SERVER);
}
});
}
private:
};
}
REGISTER_COMPONENT(gsc::component)