84 lines
2.2 KiB
C++
84 lines
2.2 KiB
C++
|
#include "STDInclude.hpp"
|
||
|
|
|
||
|
|
namespace Components
|
||
|
|
{
|
||
|
|
std::unordered_map<std::string, IPCHandler::Callback> IPCHandler::WorkerCallbacks;
|
||
|
|
std::unordered_map<std::string, IPCHandler::Callback> IPCHandler::ClientCallbacks;
|
||
|
|
|
||
|
|
std::unique_ptr<Utils::IPC::BidirectionalChannel> IPCHandler::WorkerChannel;
|
||
|
|
std::unique_ptr<Utils::IPC::BidirectionalChannel> IPCHandler::ClientChannel;
|
||
|
|
|
||
|
|
void IPCHandler::SendWorker(std::string message, std::string data)
|
||
|
|
{
|
||
|
|
IPCHandler::InitChannels();
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
void IPCHandler::SendClient(std::string message, std::string data)
|
||
|
|
{
|
||
|
|
IPCHandler::InitChannels();
|
||
|
|
//IPCHandler::ClientChannel->send()
|
||
|
|
}
|
||
|
|
|
||
|
|
void IPCHandler::OnWorker(std::string message, IPCHandler::Callback callback)
|
||
|
|
{
|
||
|
|
IPCHandler::WorkerCallbacks[message] = callback;
|
||
|
|
}
|
||
|
|
|
||
|
|
void IPCHandler::OnClient(std::string message, IPCHandler::Callback callback)
|
||
|
|
{
|
||
|
|
IPCHandler::ClientCallbacks[message] = callback;
|
||
|
|
}
|
||
|
|
|
||
|
|
void IPCHandler::InitChannels()
|
||
|
|
{
|
||
|
|
if (!IPCHandler::WorkerChannel)
|
||
|
|
{
|
||
|
|
IPCHandler::WorkerChannel.reset(new Utils::IPC::BidirectionalChannel("IW4x-Worker-Channel", !Worker::IsWorker()));
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!IPCHandler::ClientChannel)
|
||
|
|
{
|
||
|
|
IPCHandler::ClientChannel.reset(new Utils::IPC::BidirectionalChannel("IW4x-Client-Channel", Singleton::IsFirstInstance()));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void IPCHandler::StartWorker()
|
||
|
|
{
|
||
|
|
STARTUPINFOA sInfo;
|
||
|
|
PROCESS_INFORMATION pInfo;
|
||
|
|
|
||
|
|
ZeroMemory(&sInfo, sizeof(sInfo));
|
||
|
|
ZeroMemory(&pInfo, sizeof(pInfo));
|
||
|
|
sInfo.cb = sizeof(sInfo);
|
||
|
|
|
||
|
|
CreateProcessA("iw4x.exe", const_cast<char*>(Utils::String::VA("-parent %d", GetCurrentProcessId())), nullptr, nullptr, false, NULL, nullptr, nullptr, &sInfo, &pInfo);
|
||
|
|
|
||
|
|
if (pInfo.hThread && pInfo.hThread != INVALID_HANDLE_VALUE) CloseHandle(pInfo.hThread);
|
||
|
|
if (pInfo.hProcess && pInfo.hProcess != INVALID_HANDLE_VALUE) CloseHandle(pInfo.hProcess);
|
||
|
|
}
|
||
|
|
|
||
|
|
IPCHandler::IPCHandler()
|
||
|
|
{
|
||
|
|
if (Dedicated::IsEnabled()) return;
|
||
|
|
|
||
|
|
IPCHandler::InitChannels();
|
||
|
|
IPCHandler::StartWorker();
|
||
|
|
|
||
|
|
QuickPatch::OnFrame([]()
|
||
|
|
{
|
||
|
|
std::string buffer;
|
||
|
|
if(IPCHandler::WorkerChannel->receive(&buffer))
|
||
|
|
{
|
||
|
|
Logger::Print("Data received: %s\n", buffer.data());
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
IPCHandler::~IPCHandler()
|
||
|
|
{
|
||
|
|
IPCHandler::WorkerCallbacks.clear();
|
||
|
|
IPCHandler::ClientCallbacks.clear();
|
||
|
|
}
|
||
|
|
}
|