mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
de3b3f53bd
* IConsole overlay now clears the oldest log entries to stay under the vector limit instead of clearing the whole vector. * IConsole code cleanup. * IBrowser code cleanup, reordered class methods. * Use ConVar utilities to get/set values for strings in IBrowser and IConsole. * New ConVar's for RUI console overlay to tweak colors.
307 lines
9.7 KiB
C++
307 lines
9.7 KiB
C++
//=============================================================================//
|
|
//
|
|
// Purpose: Squirrel VM
|
|
//
|
|
//=============================================================================//
|
|
|
|
#include "core/stdafx.h"
|
|
#include "core/logdef.h"
|
|
#include "tier0/basetypes.h"
|
|
#include "tier0/cvar.h"
|
|
#include "tier0/IConVar.h"
|
|
#include "engine/sys_utils.h"
|
|
#include "squirrel/sqvm.h"
|
|
#include "vgui/CEngineVGui.h"
|
|
#include "gameui/IConsole.h"
|
|
|
|
//---------------------------------------------------------------------------------
|
|
// Purpose: prints the output of each VM to the console
|
|
//---------------------------------------------------------------------------------
|
|
void* HSQVM_PrintFunc(void* sqvm, char* fmt, ...)
|
|
{
|
|
#ifdef GAMEDLL_S3
|
|
int vmIdx = *(int*)((std::uintptr_t)sqvm + 0x18);
|
|
#else // TODO [ AMOS ]: nothing equal to 'rdx + 18h' exist in the vm pointers for anything below S3.
|
|
int vmIdx = 3;
|
|
#endif
|
|
static bool initialized = false;
|
|
|
|
static char buf[1024];
|
|
static std::string vmType[4] = { "Script(S):", "Script(C):", "Script(U):", "Script(X):" };
|
|
|
|
static auto iconsole = spdlog::stdout_logger_mt("sqvm_print_iconsole"); // in-game console.
|
|
static auto wconsole = spdlog::stdout_logger_mt("sqvm_print_wconsole"); // windows console.
|
|
static auto sqlogger = spdlog::basic_logger_mt("sqvm_print_logger", "platform\\logs\\sqvm_print.log"); // file logger.
|
|
|
|
std::string vmStr = vmType[vmIdx].c_str();
|
|
|
|
g_spd_sqvm_p_oss.str("");
|
|
g_spd_sqvm_p_oss.clear();
|
|
|
|
if (!initialized)
|
|
{
|
|
iconsole = std::make_shared<spdlog::logger>("sqvm_print_ostream", g_spd_sqvm_p_ostream_sink);
|
|
iconsole->set_pattern("[%S.%e] %v");
|
|
iconsole->set_level(spdlog::level::debug);
|
|
wconsole->set_pattern("[%S.%e] %v");
|
|
wconsole->set_level(spdlog::level::debug);
|
|
sqlogger->set_pattern("[%S.%e] %v");
|
|
sqlogger->set_level(spdlog::level::debug);
|
|
initialized = true;
|
|
}
|
|
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
|
|
vsnprintf(buf, sizeof(buf), fmt, args);
|
|
|
|
buf[sizeof(buf) - 1] = 0;
|
|
va_end(args);
|
|
|
|
vmStr.append(buf);
|
|
|
|
if (sq_showvmoutput->GetInt() > 0)
|
|
{
|
|
sqlogger->debug(vmStr);
|
|
}
|
|
if (sq_showvmoutput->GetInt() > 1)
|
|
{
|
|
iconsole->debug(vmStr);
|
|
wconsole->debug(vmStr);
|
|
#ifndef DEDICATED
|
|
std::string s = g_spd_sqvm_p_oss.str();
|
|
const char* c = s.c_str();
|
|
g_pIConsole->m_ivConLog.push_back(Strdup((const char*)c));
|
|
#endif // !DEDICATED
|
|
}
|
|
#ifndef DEDICATED
|
|
if (sq_showvmoutput->GetInt() > 2)
|
|
{
|
|
std::string s = g_spd_sqvm_p_oss.str();
|
|
g_pLogSystem.AddLog((LogType_t)vmIdx, s);
|
|
}
|
|
#endif // !DEDICATED
|
|
return NULL;
|
|
}
|
|
|
|
//---------------------------------------------------------------------------------
|
|
// Purpose: prints the warning output of each VM to the console
|
|
//---------------------------------------------------------------------------------
|
|
void* HSQVM_WarningFunc(void* sqvm, int a2, int a3, int* nStringSize, void** ppString)
|
|
{
|
|
void* result = SQVM_WarningFunc(sqvm, a2, a3, nStringSize, ppString);
|
|
if (g_bSQVM_WarnFuncCalled) // Check if its SQVM_Warning calling.
|
|
{
|
|
return result; // If not return.
|
|
}
|
|
|
|
static bool initialized = false;
|
|
static std::string vmType[4] = { "Script(S): WARNING: ", "Script(C): WARNING: ", "Script(U): WARNING: ", "Script(X): WARNING: " };
|
|
|
|
static auto iconsole = spdlog::stdout_logger_mt("sqvm_warn_iconsole"); // in-game console.
|
|
static auto wconsole = spdlog::stdout_logger_mt("sqvm_warn_wconsole"); // windows console.
|
|
static auto sqlogger = spdlog::basic_logger_mt("sqvm_warn_logger", "platform\\logs\\sqvm_warn.log"); // file logger.
|
|
|
|
#ifdef GAMEDLL_S3
|
|
int vmIdx = *(int*)((std::uintptr_t)sqvm + 0x18);
|
|
#else // TODO [ AMOS ]: nothing equal to 'rdx + 18h' exist in the vm pointers for anything below S3.
|
|
int vmIdx = 3;
|
|
#endif
|
|
std::string vmStr = vmType[vmIdx].c_str();
|
|
|
|
g_spd_sqvm_w_oss.str("");
|
|
g_spd_sqvm_w_oss.clear();
|
|
|
|
if (!initialized)
|
|
{
|
|
iconsole = std::make_shared<spdlog::logger>("sqvm_warn_ostream", g_spd_sqvm_p_ostream_sink);
|
|
iconsole->set_pattern("[%S.%e] %v\n");
|
|
iconsole->set_level(spdlog::level::debug);
|
|
wconsole->set_pattern("[%S.%e] %v\n");
|
|
wconsole->set_level(spdlog::level::debug);
|
|
sqlogger->set_pattern("[%S.%e] %v\n");
|
|
sqlogger->set_level(spdlog::level::debug);
|
|
initialized = true;
|
|
}
|
|
|
|
std::string stringConstructor((char*)*ppString, *nStringSize); // Get string from memory via std::string constructor.
|
|
vmStr.append(stringConstructor);
|
|
|
|
std::string s = g_spd_sqvm_w_oss.str();
|
|
const char* c = s.c_str();
|
|
|
|
if (sq_showvmwarning->GetInt() > 0)
|
|
{
|
|
sqlogger->debug(vmStr); // Emit to file.
|
|
}
|
|
if (sq_showvmwarning->GetInt() > 1)
|
|
{
|
|
iconsole->debug(vmStr); // Emit to in-game console.
|
|
wconsole->debug(vmStr); // Emit to windows console.
|
|
#ifndef DEDICATED
|
|
std::string s = g_spd_sqvm_w_oss.str();
|
|
const char* c = s.c_str();
|
|
g_pIConsole->m_ivConLog.push_back(Strdup(c));
|
|
#endif // !DEDICATED
|
|
}
|
|
#ifndef DEDICATED
|
|
if (sq_showvmwarning->GetInt() > 2)
|
|
{
|
|
g_pLogSystem.AddLog((LogType_t)vmIdx, s);
|
|
const char* c = s.c_str();
|
|
g_pIConsole->m_ivConLog.push_back(Strdup(c));
|
|
}
|
|
#endif // !DEDICATED
|
|
g_bSQVM_WarnFuncCalled = false;
|
|
|
|
return result;
|
|
}
|
|
|
|
//---------------------------------------------------------------------------------
|
|
// Purpose:
|
|
//---------------------------------------------------------------------------------
|
|
void* HSQVM_WarningCmd(int a1, int a2)
|
|
{
|
|
g_bSQVM_WarnFuncCalled = true;
|
|
return SQVM_WarningCmd(a1, a2);
|
|
}
|
|
|
|
//---------------------------------------------------------------------------------
|
|
// Purpose: loads the include file from the mods directory
|
|
//---------------------------------------------------------------------------------
|
|
void* HSQVM_LoadRson(const char* szRsonName)
|
|
{
|
|
char szFilePath[MAX_PATH] = { 0 };
|
|
sprintf_s(szFilePath, MAX_PATH, "platform\\%s", szRsonName);
|
|
|
|
// Flip forward slashes in filepath to windows-style backslash
|
|
for (int i = 0; i < strlen(szFilePath); i++)
|
|
{
|
|
if (szFilePath[i] == '/')
|
|
{
|
|
szFilePath[i] = '\\';
|
|
}
|
|
}
|
|
|
|
// Returns the new path if the rson exists on the disk
|
|
if (FileExists(szFilePath) && SQVM_LoadRson(szRsonName))
|
|
{
|
|
if (sq_showrsonloading->GetBool())
|
|
{
|
|
DevMsg(eDLL_T::ENGINE, "\n");
|
|
DevMsg(eDLL_T::ENGINE, "______________________________________________________________\n");
|
|
DevMsg(eDLL_T::ENGINE, "] RSON_DISK --------------------------------------------------\n");
|
|
DevMsg(eDLL_T::ENGINE, "] PATH: '%s'\n", szFilePath);
|
|
DevMsg(eDLL_T::ENGINE, "--------------------------------------------------------------\n");
|
|
DevMsg(eDLL_T::ENGINE, "\n");
|
|
}
|
|
return SQVM_LoadRson(szFilePath);
|
|
}
|
|
else
|
|
{
|
|
if (sq_showrsonloading->GetBool())
|
|
{
|
|
DevMsg(eDLL_T::ENGINE, "\n");
|
|
DevMsg(eDLL_T::ENGINE, "______________________________________________________________\n");
|
|
DevMsg(eDLL_T::ENGINE, "] RSON_VPK ---------------------------------------------------\n");
|
|
DevMsg(eDLL_T::ENGINE, "] PATH: '%s'\n", szRsonName);
|
|
DevMsg(eDLL_T::ENGINE, "--------------------------------------------------------------\n");
|
|
DevMsg(eDLL_T::ENGINE, "\n");
|
|
}
|
|
}
|
|
return SQVM_LoadRson(szRsonName);
|
|
}
|
|
|
|
//---------------------------------------------------------------------------------
|
|
// Purpose: loads the script file from the mods directory
|
|
//---------------------------------------------------------------------------------
|
|
bool HSQVM_LoadScript(void* sqvm, const char* szScriptPath, const char* szScriptName, int nFlag)
|
|
{
|
|
char filepath[MAX_PATH] = { 0 };
|
|
sprintf_s(filepath, MAX_PATH, "platform\\%s", szScriptPath);
|
|
|
|
// Flip forward slashes in filepath to windows-style backslash
|
|
for (int i = 0; i < strlen(filepath); i++)
|
|
{
|
|
if (filepath[i] == '/')
|
|
{
|
|
filepath[i] = '\\';
|
|
}
|
|
}
|
|
|
|
if (sq_showscriptloading->GetBool())
|
|
{
|
|
DevMsg(eDLL_T::ENGINE, "Loading SQVM Script '%s'\n", filepath);
|
|
}
|
|
|
|
// Returns true if the script exists on the disk
|
|
if (FileExists(filepath) && SQVM_LoadScript(sqvm, filepath, szScriptName, nFlag))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (sq_showscriptloading->GetBool())
|
|
{
|
|
DevMsg(eDLL_T::ENGINE, "FAILED. Try SP / VPK for '%s'\n", filepath);
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
return SQVM_LoadScript(sqvm, szScriptPath, szScriptName, nFlag);
|
|
}
|
|
|
|
void HSQVM_RegisterFunction(void* sqvm, const char* szName, const char* szHelpString, const char* szRetValType, const char* szArgTypes, void* pFunction)
|
|
{
|
|
SQFuncRegistration* sqFunc = new SQFuncRegistration();
|
|
|
|
sqFunc->m_szScriptName = szName;
|
|
sqFunc->m_szNativeName = szName;
|
|
sqFunc->m_szHelpString = szHelpString;
|
|
sqFunc->m_szRetValType = szRetValType;
|
|
sqFunc->m_szArgTypes = szArgTypes;
|
|
sqFunc->m_pFunction = pFunction;
|
|
|
|
SQVM_RegisterFunc(sqvm, sqFunc, 1);
|
|
}
|
|
|
|
int HSQVM_NativeTest(void* sqvm)
|
|
{
|
|
// Function code goes here.
|
|
return 1;
|
|
}
|
|
|
|
void RegisterUIScriptFunctions(void* sqvm)
|
|
{
|
|
HSQVM_RegisterFunction(sqvm, "UINativeTest", "native ui function", "void", "", &HSQVM_NativeTest);
|
|
}
|
|
|
|
void RegisterClientScriptFunctions(void* sqvm)
|
|
{
|
|
HSQVM_RegisterFunction(sqvm, "ClientNativeTest", "native client function", "void", "", &HSQVM_NativeTest);
|
|
}
|
|
|
|
void RegisterServerScriptFunctions(void* sqvm)
|
|
{
|
|
HSQVM_RegisterFunction(sqvm, "ServerNativeTest", "native server function", "void", "", &HSQVM_NativeTest);
|
|
}
|
|
|
|
void SQVM_Attach()
|
|
{
|
|
DetourAttach((LPVOID*)&SQVM_PrintFunc, &HSQVM_PrintFunc);
|
|
DetourAttach((LPVOID*)&SQVM_WarningFunc, &HSQVM_WarningFunc);
|
|
DetourAttach((LPVOID*)&SQVM_WarningCmd, &HSQVM_WarningCmd);
|
|
DetourAttach((LPVOID*)&SQVM_LoadRson, &HSQVM_LoadRson);
|
|
DetourAttach((LPVOID*)&SQVM_LoadScript, &HSQVM_LoadScript);
|
|
}
|
|
|
|
void SQVM_Detach()
|
|
{
|
|
DetourDetach((LPVOID*)&SQVM_PrintFunc, &HSQVM_PrintFunc);
|
|
DetourDetach((LPVOID*)&SQVM_WarningFunc, &HSQVM_WarningFunc);
|
|
DetourDetach((LPVOID*)&SQVM_WarningCmd, &HSQVM_WarningCmd);
|
|
DetourDetach((LPVOID*)&SQVM_LoadRson, &HSQVM_LoadRson);
|
|
DetourDetach((LPVOID*)&SQVM_LoadScript, &HSQVM_LoadScript);
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
bool g_bSQVM_WarnFuncCalled = false;
|