void remove_filter(Filter_Base* filter)
{
if (!filter)
return;
std::lock_guard<std::recursive_mutex> lock(mutex_);
Logger_Settings settings(thread_log_settings_table_[std::hash<std::thread::id>()(std::this_thread::get_id())]);
std::string filter_id(filter->get_id());
if (!filter_id.empty())
{
settings.filter_id_ptr_map.erase(settings.filter_id_ptr_map.find(filter_id));
thread_log_settings_table_[std::hash<std::thread::id>()(std::this_thread::get_id())] = settings;
}
}
void add_filter(Filter_Base* filter)
{
if (!filter)
return;
std::string filter_id(filter->get_id());
generic_util::trim(filter_id);
if (filter_id.empty())
return;
std::lock_guard<std::recursive_mutex> lock(mutex_);
Logger_Settings settings = Logger_Settings(thread_log_settings_table_[std::hash<std::thread::id>()(std::this_thread::get_id())]);
settings.filter_id_ptr_map[filter_id] = std::shared_ptr<Filter_Base>(filter);
thread_log_settings_table_[std::hash<std::thread::id>()(std::this_thread::get_id())] = settings;
}
void OutputReceivedMessage(const Hekate::Protobuf::Proto::Server::HekateMessage &msg)
{
if (!msg.has_hook_info())
{
std::cerr << "Missing Hekate -> client message.\n";
exit(-1);
}
if (msg.hook_info().has_send())
{
auto hookmsg = msg.hook_info().send();
std::cout << "Socket: 0x" << std::hex << hookmsg.socket() << std::endl;
std::cout << "Buffer: ";
PrintHexBuffer(hookmsg.buffer().c_str(), hookmsg.length());
std::cout << "Length: 0x" << std::hex << hookmsg.length() << std::endl;
}
else if (msg.hook_info().has_sendto())
{
auto hookmsg = msg.hook_info().sendto();
std::cout << "Socket: 0x" << std::hex << hookmsg.socket() << std::endl;
std::cout << "Buffer: ";
PrintHexBuffer(hookmsg.buffer().c_str(), hookmsg.length());
std::cout << "Length: 0x" << std::hex << hookmsg.length() << std::endl;
std::cout << "to struct address: 0x" << std::hex << hookmsg.to_address() << std::endl;
std::cout << "to address length: 0x" << std::hex << hookmsg.to_length() << std::endl;
}
else if (msg.hook_info().has_wsasend())
{
auto hookmsg = msg.hook_info().wsasend();
std::cout << "Socket: 0x" << std::hex << hookmsg.socket() << std::endl;
for (int i = 0; i < hookmsg.count(); ++i)
{
std::cout << "Buffer #" << i << std::endl;
PrintHexBuffer(hookmsg.buffers(i).c_str(), hookmsg.buffer_size(i));
}
std::cout << "Count: 0x" << std::hex << hookmsg.count() << std::endl;
std::cout << "Bytes sent address: 0x" << std::hex << hookmsg.bytes_sent_address() << std::endl;
std::cout << "Flags: 0x" << std::hex << hookmsg.flags() << std::endl;
std::cout << "Overlapped address: 0x" << std::hex << hookmsg.overlapped_address() << std::endl;
std::cout << "Overlapped routine address: 0x" << std::hex << hookmsg.overlapped_routine_address() << std::endl;
}
else if (msg.hook_info().has_wsasendto())
{
auto hookmsg = msg.hook_info().wsasendto();
std::cout << "Socket: 0x" << std::hex << hookmsg.socket() << std::endl;
for (int i = 0; i < hookmsg.count(); ++i)
{
std::cout << "Buffer #" << i << std::endl;
PrintHexBuffer(hookmsg.buffers(i).c_str(), hookmsg.buffer_size(i));
}
std::cout << "Count: 0x" << std::hex << hookmsg.count() << std::endl;
std::cout << "Bytes sent address: 0x" << std::hex << hookmsg.bytes_sent_address() << std::endl;
std::cout << "Flags: 0x" << std::hex << hookmsg.flags() << std::endl;
std::cout << "to struct address: 0x" << std::hex << hookmsg.to_address() << std::endl;
std::cout << "to struct length: 0x" << std::hex << hookmsg.to_length() << std::endl;
std::cout << "Overlapped address: 0x" << std::hex << hookmsg.overlapped_address() << std::endl;
std::cout << "Overlapped routine address: 0x" << std::hex << hookmsg.overlapped_routine_address() << std::endl;
}
else if (msg.hook_info().has_recv())
{
auto hookmsg = msg.hook_info().recv();
std::cout << "Socket: 0x" << std::hex << hookmsg.socket() << std::endl;
std::cout << "Buffer: ";
std::cout << "Buffer address: " << hookmsg.buffer() << std::endl;
std::cout << "Length: 0x" << std::hex << hookmsg.length() << std::endl;
}
else if (msg.hook_info().has_recvfrom())
{
auto hookmsg = msg.hook_info().recvfrom();
std::cout << "Socket: 0x" << std::hex << hookmsg.socket() << std::endl;
std::cout << "Buffer: ";
std::cout << "Buffer address: " << hookmsg.buffer() << std::endl;
std::cout << "Length: 0x" << std::hex << hookmsg.length() << std::endl;
std::cout << "from struct address: 0x" << std::hex << hookmsg.from_address() << std::endl;
std::cout << "from length address: 0x" << std::hex << hookmsg.from_length_address() << std::endl;
}
else if (msg.hook_info().has_wsarecv())
{
auto hookmsg = msg.hook_info().wsarecv();
std::cout << "Socket: 0x" << std::hex << hookmsg.socket() << std::endl;
for (int i = 0; i < hookmsg.count(); ++i)
{
std::cout << "Buffer #" << i << std::endl;
std::cout << "Buffer address: " << hookmsg.buffers(i) << std::endl;
}
std::cout << "Count: 0x" << std::hex << hookmsg.count() << std::endl;
std::cout << "Bytes received address: 0x" << std::hex << hookmsg.bytes_received_address() << std::endl;
std::cout << "Flags address: 0x" << std::hex << hookmsg.flags_address() << std::endl;
std::cout << "Overlapped address: 0x" << std::hex << hookmsg.overlapped_address() << std::endl;
std::cout << "Overlapped routine address: 0x" << std::hex << hookmsg.overlapped_routine_address() << std::endl;
}
else if (msg.hook_info().has_wsarecvfrom())
{
auto hookmsg = msg.hook_info().wsarecvfrom();
std::cout << "Socket: 0x" << std::hex << hookmsg.socket() << std::endl;
for (int i = 0; i < hookmsg.count(); ++i)
{
std::cout << "Buffer #" << i << std::endl;
std::cout << "Buffer address: " << hookmsg.buffers(i) << std::endl;
}
std::cout << "Count: 0x" << std::hex << hookmsg.count() << std::endl;
std::cout << "Bytes received address: 0x" << std::hex << hookmsg.bytes_received_address() << std::endl;
std::cout << "Flags address: 0x" << std::hex << hookmsg.flags_address() << std::endl;
std::cout << "from struct address: 0x" << std::hex << hookmsg.from_address() << std::endl;
std::cout << "from struct length: 0x" << std::hex << hookmsg.from_length_address() << std::endl;
std::cout << "Overlapped address: 0x" << std::hex << hookmsg.overlapped_address() << std::endl;
std::cout << "Overlapped routine address: 0x" << std::hex << hookmsg.overlapped_routine_address() << std::endl;
}
else if(msg.has_acknowledge())
{
std::cout << "Received acknowledge message from server." << std::endl;
std::cout << "Code: " << msg.acknowledge().debug_response_id() << std::endl;
}
else if (msg.hook_info().has_filter_message())
{
auto filtermsg = msg.hook_info().filter_message();
std::cout << "Received filter hit message." << std::endl;
std::cout << "Filter hit id: 0x" << std::hex << filtermsg.filter_id() << std::endl;
if (filtermsg.broken())
{
std::cout << "Application is currently in a paused state." << std::endl;
}
}
else
{
std::cerr << "Received unknown/empty message from server." << std::endl;
}
}