#include "pch.h"
#include <windows.h>
#include <WinSock2.h>
#include <fstream> // Required to output logs to files
#include <iomanip> // Required to display the hex properly
#include <iostream> //cout ...
// Detours: https://www.microsoft.com/en-us/research/project/detours/
#include "E:\Informatica\Detours-4.0.1\src\detours.h" // Version 3.0 use for this hook. Be sure to include the library and includes to your project in visual studio
#pragma comment(lib,"detours.lib") // Need to include this so we can use Detours
#pragma comment(lib,"ws2_32.lib") // Required to hook Send and Recv since they both reside in this library
// Pointers to the original functions
static int (WINAPI* originalSend)(SOCKET s, const char* buf, int len, int flags) = send; // https://msdn.microsoft.com/en-us/library/windows/desktop/ms740149(v=vs.85).aspx
static int (WINAPI* originalRecv)(SOCKET s, char* buf, int len, int flags) = recv; // https://msdn.microsoft.com/en-us/library/windows/desktop/ms740121(v=vs.85).aspx
typedef UINT_PTR SOCKET; // Socket redefinition
HMODULE hModule;
std::ofstream sendLog;
std::ofstream recvLog;
bool CreateConsole()
{
FILE* file = NULL;
SetConsoleTitleA("Pack Viewer [ALPHA]");
if (!AllocConsole())
return false;
if (freopen_s(&file, "CONIN$", "r", stdin))
return false;
if (freopen_s(&file, "CONOUT$", "w", stdout))
return false;
if (freopen_s(&file, "CONERR$", "w", stderr))
return false;
}
int WINAPI newSend(SOCKET s, char* buf, int len, int flags) // Dumps each buffer to a new line in the "send.txt" file in the games directory
{
//sendLog.open("send.txt", std::ios::app); // Opens a handle to the send file
printf("> SEND:\n");
for (int i = 0; i < len; i++)
{ // For each byte:
//sendLog << std::hex << std::setfill('0') << std::setw(2) << (unsigned int)(unsigned char)buf[i] << " "; // Log the hex of the byte with a width of 2 (leading 0 added if necessary) and a space after to separate bytes
std::cout << std::hex << std::setfill('0') << std::setw(2) << (unsigned int)(unsigned char)buf[i] << " ";
}
std::cout << std::endl;
// sendLog << std::endl; // Add a newline to the text file, indicating the end of this request
// sendLog.close(); // Close the text file
return originalSend(s, buf, len, flags); // Send the buffer to the original send function
//return send(s,buf,len,flags);
}
int WINAPI newRecv(SOCKET s, char* buf, int len, int flags) // Dumps each buffer to a new line in the "recv.txt" file in the games directory
{
len = originalRecv(s, buf, len, flags); // Send the request with a pointer to the buffer for recv to store the response
// recvLog.open("recv.txt", std::ios::app); // Opens a handle to the recv file
printf("> RECV:\n");
for (int i = 0; i < len; i++) // For each byte in the response:
{
//recvLog << std::hex << std::setfill('0') << std::setw(2) << (unsigned int)buf[i] << " "; // Log the hex of the byte with a width of 2 (leading 0 added if necessary) and a space after to separate bytes
std::cout << std::hex << std::setfill('0') << std::setw(2) << (unsigned int)(unsigned char)buf[i] << " ";
}
std::cout << std::endl;
// recvLog << std::endl; // Add a newline to the text file, indicating the end of this request
// recvLog.close();
// // Close the text file
//return len; why len [original]??? // Returns the output from the original recv call
return originalRecv(s, buf, len, flags);
//return recv(s,buf,len,flags);
}
DWORD WINAPI hook(LPVOID args) // Basic detours
{
CreateConsole();
DetourTransactionBegin();
DetourUpdateThread(GetCurrentThread());
DetourAttach(&(PVOID&)originalSend, newSend);
DetourAttach(&(PVOID&)originalRecv, newRecv);
DetourTransactionCommit();
return 0;
}
BOOL APIENTRY DllMain(HMODULE hModule, DWORD Reason, LPVOID reserved)
{
switch (Reason)
{
case DLL_PROCESS_ATTACH:
DisableThreadLibraryCalls(hModule);
CreateThread(NULL, 0, hook, NULL, 0, NULL);
break;
case DLL_PROCESS_DETACH:
break;
}
return TRUE;
}
