gsc-tool/src/arc/context.cpp
2023-05-14 17:41:06 +02:00

498 lines
18 KiB
C++

// Copyright 2023 xensik. All rights reserved.
//
// Use of this source code is governed by a GNU GPLv3 license
// that can be found in the LICENSE file.
#include "xsk/stdinc.hpp"
#include "xsk/arc/context.hpp"
namespace xsk::arc
{
extern std::array<std::pair<opcode, std::string_view>, opcode_count> const opcode_list;
context::context(arc::props props, arc::engine engine, arc::endian endian, arc::system system, u64 magic)
: props_{ props }, engine_{ engine }, endian_{ endian }, system_{ system }, instance_{ arc::instance::server }, magic_{ magic },
source_{ this }, assembler_{ this }, disassembler_{ this }, compiler_{ this }, decompiler_{ this }
{
opcode_map_.reserve(opcode_list.size());
opcode_map_rev_.reserve(opcode_list.size());
for (auto const& entry : opcode_list)
{
opcode_map_.insert({ entry.first, entry.second });
opcode_map_rev_.insert({ entry.second, entry.first });
}
}
auto context::init(arc::build build, fs_callback callback) -> void
{
build_ = build;
fs_callback_ = callback;
}
auto context::cleanup() -> void
{
}
auto context::engine_name() const -> std::string_view
{
switch (engine_)
{
case engine::t6: return "T6";
case engine::t7: return "T7";
case engine::t8: return "T8";
case engine::t9: return "T9";
default: return "";
}
}
auto context::opcode_size(opcode op) const -> u32
{
switch (op)
{
case opcode::OP_End:
case opcode::OP_Return:
case opcode::OP_GetUndefined:
case opcode::OP_GetZero:
case opcode::OP_GetLevelObject:
case opcode::OP_GetAnimObject:
case opcode::OP_GetSelf:
case opcode::OP_GetLevel:
case opcode::OP_GetGame:
case opcode::OP_GetAnim:
case opcode::OP_GetGameRef:
case opcode::OP_CreateLocalVariable:
case opcode::OP_EvalArray:
case opcode::OP_EvalArrayRef:
case opcode::OP_ClearArray:
case opcode::OP_EmptyArray:
case opcode::OP_GetSelfObject:
case opcode::OP_ClearParams:
case opcode::OP_CheckClearParams:
case opcode::OP_SetVariableField:
case opcode::OP_Wait:
case opcode::OP_WaitTillFrameEnd:
case opcode::OP_PreScriptCall:
case opcode::OP_DecTop:
case opcode::OP_CastFieldObject:
case opcode::OP_CastBool:
case opcode::OP_BoolNot:
case opcode::OP_BoolComplement:
case opcode::OP_Inc:
case opcode::OP_Dec:
case opcode::OP_Bit_Or:
case opcode::OP_Bit_Xor:
case opcode::OP_Bit_And:
case opcode::OP_Equal:
case opcode::OP_NotEqual:
case opcode::OP_LessThan:
case opcode::OP_GreaterThan:
case opcode::OP_LessThanOrEqualTo:
case opcode::OP_GreaterThanOrEqualTo:
case opcode::OP_ShiftLeft:
case opcode::OP_ShiftRight:
case opcode::OP_Plus:
case opcode::OP_Minus:
case opcode::OP_Multiply:
case opcode::OP_Divide:
case opcode::OP_Modulus:
case opcode::OP_SizeOf:
case opcode::OP_WaitTill:
case opcode::OP_Notify:
case opcode::OP_EndOn:
case opcode::OP_VoidCodePos:
case opcode::OP_Vector:
case opcode::OP_RealWait:
case opcode::OP_IsDefined:
case opcode::OP_VectorScale:
case opcode::OP_AnglesToUp:
case opcode::OP_AnglesToRight:
case opcode::OP_AnglesToForward:
case opcode::OP_AngleClamp180:
case opcode::OP_VectorToAngles:
case opcode::OP_Abs:
case opcode::OP_GetTime:
case opcode::OP_GetDvar:
case opcode::OP_GetDvarInt:
case opcode::OP_GetDvarFloat:
case opcode::OP_GetDvarVector:
case opcode::OP_GetDvarColorRed:
case opcode::OP_GetDvarColorGreen:
case opcode::OP_GetDvarColorBlue:
case opcode::OP_GetDvarColorAlpha:
case opcode::OP_FirstArrayKey:
case opcode::OP_NextArrayKey:
case opcode::OP_ProfileStop:
case opcode::OP_SafeDecTop:
case opcode::OP_Nop:
case opcode::OP_Abort:
case opcode::OP_Object:
case opcode::OP_ThreadObject:
case opcode::OP_EvalLocalVariable:
case opcode::OP_EvalLocalVariableRef:
case opcode::OP_GetWorldObject:
case opcode::OP_GetWorld:
case opcode::OP_GetClassesObject:
case opcode::OP_GetClasses:
case opcode::OP_SuperEqual:
case opcode::OP_SuperNotEqual:
return (props_ & props::size64) ? 2 : 1;
case opcode::OP_SafeSetVariableFieldCached:
return (props_ & props::size64) ? 3 : 1;
case opcode::OP_GetByte:
case opcode::OP_GetNegByte:
case opcode::OP_SafeCreateLocalVariables:
case opcode::OP_RemoveLocalVariables:
case opcode::OP_EvalLocalVariableCached:
case opcode::OP_EvalLocalArrayRefCached:
case opcode::OP_SafeSetWaittillVariableFieldCached:
case opcode::OP_EvalLocalVariableRefCached:
case opcode::OP_ScriptFunctionCallPointer:
case opcode::OP_ScriptMethodCallPointer:
case opcode::OP_ScriptThreadCallPointer:
case opcode::OP_ScriptMethodThreadCallPointer:
case opcode::OP_WaitTillMatch:
case opcode::OP_VectorConstant:
return (props_ & props::size64) ? 3 : 2;
case opcode::OP_GetUnsignedShort:
case opcode::OP_GetNegUnsignedShort:
case opcode::OP_JumpOnFalse:
case opcode::OP_JumpOnTrue:
case opcode::OP_JumpOnFalseExpr:
case opcode::OP_JumpOnTrueExpr:
case opcode::OP_Jump:
case opcode::OP_JumpBack:
case opcode::OP_DevblockBegin:
case opcode::OP_DevblockEnd:
return (props_ & props::size64) ? 4 : 3;
case opcode::OP_GetString:
case opcode::OP_GetIString:
case opcode::OP_EvalFieldVariable:
case opcode::OP_EvalFieldVariableRef:
case opcode::OP_ClearFieldVariable:
return (props_ & props::size64) ? 6 : 3;
case opcode::OP_EvalLocalVariableCachedDebug:
case opcode::OP_EvalLocalVariableRefCachedDebug:
case opcode::OP_LevelEvalFieldVariableRef:
case opcode::OP_LevelEvalFieldVariable:
case opcode::OP_New:
case opcode::OP_SelfEvalFieldVariableRef:
case opcode::OP_SelfEvalFieldVariable:
return 6;
case opcode::OP_GetInteger:
case opcode::OP_GetFloat:
case opcode::OP_Switch:
case opcode::OP_EndSwitch:
case opcode::OP_GetHash:
return (props_ & props::size64) ? 6 : 5;
case opcode::OP_ScriptFunctionCallClass:
case opcode::OP_ScriptThreadCallClass:
return 7;
case opcode::OP_GetAPIFunction:
return 10;
case opcode::OP_ProfileStart:
return (props_ & props::size64) ? 10 : 1;
case opcode::OP_GetAnimation:
case opcode::OP_GetFunction:
return (props_ & props::size64) ? 10 : 5;
case opcode::OP_CallBuiltin:
case opcode::OP_CallBuiltinMethod:
case opcode::OP_ScriptFunctionCall:
case opcode::OP_ScriptMethodCall:
case opcode::OP_ScriptThreadCall:
case opcode::OP_ScriptMethodThreadCall:
return (props_ & props::size64) ? 11 : 6;
case opcode::OP_GetVector:
return (props_ & props::size64) ? 14 : 13;
default:
throw error(fmt::format("couldn't resolve instruction size for '{}'", opcode_name(op)));
}
}
auto context::opcode_id(opcode op) const -> u16
{
auto const itr = code_map_rev_.find(op);
if (itr != code_map_rev_.end())
{
return itr->second;
}
throw error(fmt::format("couldn't resolve opcode id for '{}'", opcode_name(op)));
}
auto context::opcode_name(opcode op) const -> std::string
{
auto const itr = opcode_map_.find(op);
if (itr != opcode_map_.end())
{
return std::string{ itr->second };
}
throw std::runtime_error(fmt::format("couldn't resolve opcode string for enum '{}'", static_cast<std::underlying_type_t<opcode>>(op)));
}
auto context::opcode_enum(std::string const& name) const -> opcode
{
auto const itr = opcode_map_rev_.find(name);
if (itr != opcode_map_rev_.end())
{
return itr->second;
}
throw std::runtime_error(fmt::format("couldn't resolve opcode enum for name '{}'", name));
}
auto context::opcode_enum(u16 id) const -> opcode
{
auto const itr = code_map_.find(id);
if (itr != code_map_.end())
{
return itr->second;
}
return opcode::OP_Invalid;
}
auto context::hash_id(std::string const& name) const -> u32
{
if (name.starts_with("_id_"))
{
return static_cast<u32>(std::stoul(name.substr(4), nullptr, 16));
}
if (props_ & props::hashids)
{
auto* str = name.data();
auto hash = 16777619u * (std::tolower(static_cast<u8>(*str)) ^ 1268436527u);
while (*str++)
{
hash = 16777619u * (std::tolower(static_cast<u8>(*str)) ^ hash);
}
return hash;
}
else
{
if (name.empty())
return 0;
auto hash = 5381u;
for (auto i = 0u; i < name.size(); i++)
{
hash = std::tolower(static_cast<u8>(name[i])) + 33 * hash;
}
return hash;
}
}
auto context::hash_name(u32 id) const -> std::string
{
auto const itr = hash_map_.find(id);
if (itr != hash_map_.end())
{
return std::string(itr->second);
}
return fmt::format("_id_{:08X}", id);
}
auto context::make_token(std::string_view str) const -> std::string
{
if (str.starts_with("_id_") || str.starts_with("_func_") || str.starts_with("_meth_"))
{
return std::string{ str };
}
auto data = std::string{ str.begin(), str.end() };
for (auto i = 0u; i < data.size(); i++)
{
data[i] = static_cast<char>(std::tolower(static_cast<unsigned char>(str[i])));
if (data[i] == '\\') data[i] = '/';
}
return data;
}
auto context::load_header(std::string const& name) -> std::tuple<std::string const*, char const*, usize>
{
auto const itr = header_files_.find(name);
if (itr != header_files_.end())
{
return { &itr->first, reinterpret_cast<char const*>(itr->second.data()), itr->second.size() };
}
auto data = fs_callback_(name);
if (!data.empty())
{
auto const res = header_files_.insert({ name, std::move(data) });
if (res.second)
{
return { &res.first->first, reinterpret_cast<char const*>(res.first->second.data()), res.first->second.size() };
}
}
throw error(fmt::format("couldn't open gsh file '{}'", name));
}
extern std::array<std::pair<opcode, std::string_view>, opcode_count> const opcode_list
{{
{ opcode::OP_Invalid, "OP_Invalid" },
{ opcode::OP_End, "OP_End" },
{ opcode::OP_Return, "OP_Return" },
{ opcode::OP_GetUndefined, "OP_GetUndefined" },
{ opcode::OP_GetZero, "OP_GetZero" },
{ opcode::OP_GetByte, "OP_GetByte" },
{ opcode::OP_GetNegByte, "OP_GetNegByte" },
{ opcode::OP_GetUnsignedShort, "OP_GetUnsignedShort" },
{ opcode::OP_GetNegUnsignedShort, "OP_GetNegUnsignedShort" },
{ opcode::OP_GetInteger, "OP_GetInteger" },
{ opcode::OP_GetFloat, "OP_GetFloat" },
{ opcode::OP_GetString, "OP_GetString" },
{ opcode::OP_GetIString, "OP_GetIString" },
{ opcode::OP_GetVector, "OP_GetVector" },
{ opcode::OP_GetLevelObject, "OP_GetLevelObject" },
{ opcode::OP_GetAnimObject, "OP_GetAnimObject" },
{ opcode::OP_GetSelf, "OP_GetSelf" },
{ opcode::OP_GetLevel, "OP_GetLevel" },
{ opcode::OP_GetGame, "OP_GetGame" },
{ opcode::OP_GetAnim, "OP_GetAnim" },
{ opcode::OP_GetAnimation, "OP_GetAnimation" },
{ opcode::OP_GetGameRef, "OP_GetGameRef" },
{ opcode::OP_GetFunction, "OP_GetFunction" },
{ opcode::OP_CreateLocalVariable, "OP_CreateLocalVariable" },
{ opcode::OP_SafeCreateLocalVariables, "OP_SafeCreateLocalVariables" },
{ opcode::OP_RemoveLocalVariables, "OP_RemoveLocalVariables" },
{ opcode::OP_EvalLocalVariableCached, "OP_EvalLocalVariableCached" },
{ opcode::OP_EvalArray, "OP_EvalArray" },
{ opcode::OP_EvalLocalArrayRefCached, "OP_EvalLocalArrayRefCached" },
{ opcode::OP_EvalArrayRef, "OP_EvalArrayRef" },
{ opcode::OP_ClearArray, "OP_ClearArray" },
{ opcode::OP_EmptyArray, "OP_EmptyArray" },
{ opcode::OP_GetSelfObject, "OP_GetSelfObject" },
{ opcode::OP_EvalFieldVariable, "OP_EvalFieldVariable" },
{ opcode::OP_EvalFieldVariableRef, "OP_EvalFieldVariableRef" },
{ opcode::OP_ClearFieldVariable, "OP_ClearFieldVariable" },
{ opcode::OP_SafeSetVariableFieldCached, "OP_SafeSetVariableFieldCached" },
{ opcode::OP_SafeSetWaittillVariableFieldCached, "OP_SafeSetWaittillVariableFieldCached" },
{ opcode::OP_ClearParams, "OP_ClearParams" },
{ opcode::OP_CheckClearParams, "OP_CheckClearParams" },
{ opcode::OP_EvalLocalVariableRefCached, "OP_EvalLocalVariableRefCached" },
{ opcode::OP_SetVariableField, "OP_SetVariableField" },
{ opcode::OP_CallBuiltin, "OP_CallBuiltin" },
{ opcode::OP_CallBuiltinMethod, "OP_CallBuiltinMethod" },
{ opcode::OP_Wait, "OP_Wait" },
{ opcode::OP_WaitTillFrameEnd, "OP_WaitTillFrameEnd" },
{ opcode::OP_PreScriptCall, "OP_PreScriptCall" },
{ opcode::OP_ScriptFunctionCall, "OP_ScriptFunctionCall" },
{ opcode::OP_ScriptFunctionCallPointer, "OP_ScriptFunctionCallPointer" },
{ opcode::OP_ScriptMethodCall, "OP_ScriptMethodCall" },
{ opcode::OP_ScriptMethodCallPointer, "OP_ScriptMethodCallPointer" },
{ opcode::OP_ScriptThreadCall, "OP_ScriptThreadCall" },
{ opcode::OP_ScriptThreadCallPointer, "OP_ScriptThreadCallPointer" },
{ opcode::OP_ScriptMethodThreadCall, "OP_ScriptMethodThreadCall" },
{ opcode::OP_ScriptMethodThreadCallPointer, "OP_ScriptMethodThreadCallPointer" },
{ opcode::OP_DecTop, "OP_DecTop" },
{ opcode::OP_CastFieldObject, "OP_CastFieldObject" },
{ opcode::OP_CastBool, "OP_CastBool" },
{ opcode::OP_BoolNot, "OP_BoolNot" },
{ opcode::OP_BoolComplement, "OP_BoolComplement" },
{ opcode::OP_JumpOnFalse, "OP_JumpOnFalse" },
{ opcode::OP_JumpOnTrue, "OP_JumpOnTrue" },
{ opcode::OP_JumpOnFalseExpr, "OP_JumpOnFalseExpr" },
{ opcode::OP_JumpOnTrueExpr, "OP_JumpOnTrueExpr" },
{ opcode::OP_Jump, "OP_Jump" },
{ opcode::OP_JumpBack, "OP_JumpBack" },
{ opcode::OP_Inc, "OP_Inc" },
{ opcode::OP_Dec, "OP_Dec" },
{ opcode::OP_Bit_Or, "OP_Bit_Or" },
{ opcode::OP_Bit_Xor, "OP_Bit_Xor" },
{ opcode::OP_Bit_And, "OP_Bit_And" },
{ opcode::OP_Equal, "OP_Equal" },
{ opcode::OP_NotEqual, "OP_NotEqual" },
{ opcode::OP_LessThan, "OP_LessThan" },
{ opcode::OP_GreaterThan, "OP_GreaterThan" },
{ opcode::OP_LessThanOrEqualTo, "OP_LessThanOrEqualTo" },
{ opcode::OP_GreaterThanOrEqualTo, "OP_GreaterThanOrEqualTo" },
{ opcode::OP_ShiftLeft, "OP_ShiftLeft" },
{ opcode::OP_ShiftRight, "OP_ShiftRight" },
{ opcode::OP_Plus, "OP_Plus" },
{ opcode::OP_Minus, "OP_Minus" },
{ opcode::OP_Multiply, "OP_Multiply" },
{ opcode::OP_Divide, "OP_Divide" },
{ opcode::OP_Modulus, "OP_Modulus" },
{ opcode::OP_SizeOf, "OP_SizeOf" },
{ opcode::OP_WaitTillMatch, "OP_WaitTillMatch" },
{ opcode::OP_WaitTill, "OP_WaitTill" },
{ opcode::OP_Notify, "OP_Notify" },
{ opcode::OP_EndOn, "OP_EndOn" },
{ opcode::OP_VoidCodePos, "OP_VoidCodePos" },
{ opcode::OP_Switch, "OP_Switch" },
{ opcode::OP_EndSwitch, "OP_EndSwitch" },
{ opcode::OP_Vector, "OP_Vector" },
{ opcode::OP_GetHash, "OP_GetHash" },
{ opcode::OP_RealWait, "OP_RealWait" },
{ opcode::OP_VectorConstant, "OP_VectorConstant" },
{ opcode::OP_IsDefined, "OP_IsDefined" },
{ opcode::OP_VectorScale, "OP_VectorScale" },
{ opcode::OP_AnglesToUp, "OP_AnglesToUp" },
{ opcode::OP_AnglesToRight, "OP_AnglesToRight" },
{ opcode::OP_AnglesToForward, "OP_AnglesToForward" },
{ opcode::OP_AngleClamp180, "OP_AngleClamp180" },
{ opcode::OP_VectorToAngles, "OP_VectorToAngles" },
{ opcode::OP_Abs, "OP_Abs" },
{ opcode::OP_GetTime, "OP_GetTime" },
{ opcode::OP_GetDvar, "OP_GetDvar" },
{ opcode::OP_GetDvarInt, "OP_GetDvarInt" },
{ opcode::OP_GetDvarFloat, "OP_GetDvarFloat" },
{ opcode::OP_GetDvarVector, "OP_GetDvarVector" },
{ opcode::OP_GetDvarColorRed, "OP_GetDvarColorRed" },
{ opcode::OP_GetDvarColorGreen, "OP_GetDvarColorGreen" },
{ opcode::OP_GetDvarColorBlue, "OP_GetDvarColorBlue" },
{ opcode::OP_GetDvarColorAlpha, "OP_GetDvarColorAlpha" },
{ opcode::OP_FirstArrayKey, "OP_FirstArrayKey" },
{ opcode::OP_NextArrayKey, "OP_NextArrayKey" },
{ opcode::OP_ProfileStart, "OP_ProfileStart" },
{ opcode::OP_ProfileStop, "OP_ProfileStop" },
{ opcode::OP_SafeDecTop, "OP_SafeDecTop" },
{ opcode::OP_Nop, "OP_Nop" },
{ opcode::OP_Abort, "OP_Abort" },
{ opcode::OP_Object, "OP_Object" },
{ opcode::OP_ThreadObject, "OP_ThreadObject" },
{ opcode::OP_EvalLocalVariable, "OP_EvalLocalVariable" },
{ opcode::OP_EvalLocalVariableRef, "OP_EvalLocalVariableRef" },
{ opcode::OP_DevblockBegin, "OP_DevblockBegin" },
{ opcode::OP_DevblockEnd, "OP_DevblockEnd" },
{ opcode::OP_EvalLocalVariableCachedDebug, "OP_EvalLocalVariableCachedDebug" },
{ opcode::OP_EvalLocalVariableRefCachedDebug, "OP_EvalLocalVariableRefCachedDebug" },
{ opcode::OP_LevelEvalFieldVariable, "OP_LevelEvalFieldVariable" },
{ opcode::OP_LevelEvalFieldVariableRef, "OP_LevelEvalFieldVariableRef" },
{ opcode::OP_SelfEvalFieldVariable, "OP_SelfEvalFieldVariable" },
{ opcode::OP_SelfEvalFieldVariableRef, "OP_SelfEvalFieldVariableRef" },
{ opcode::OP_GetWorld, "OP_GetWorld" },
{ opcode::OP_GetWorldObject, "OP_GetWorldObject" },
{ opcode::OP_GetClasses, "OP_GetClasses" },
{ opcode::OP_GetClassesObject, "OP_GetClassesObject" },
{ opcode::OP_New, "OP_New" },
{ opcode::OP_ScriptFunctionCallClass, "OP_ScriptFunctionCallClass" },
{ opcode::OP_ScriptThreadCallClass, "OP_ScriptThreadCallClass" },
{ opcode::OP_GetUintptr, "OP_GetUintptr" },
{ opcode::OP_SuperEqual, "OP_SuperEqual" },
{ opcode::OP_SuperNotEqual, "OP_SuperNotEqual" },
{ opcode::OP_GetAPIFunction, "OP_GetAPIFunction" },
}};
} // namespace xsk::arc