mirror of
https://github.com/SteamAutoCracks/DepotDownloaderMod.git
synced 2026-02-13 21:30:58 +01:00
e7387f031d
Removed packet-handling code Fixed up logger Hook CCrypto functions directly, add callbacks Added DataDumper to help with protobuf reversing --HG-- extra : convert_revision : svn%3A946a0da7-ebce-4904-9acb-2f1e67aed693%4069
114 lines
2.9 KiB
C++
114 lines
2.9 KiB
C++
#include "DataDumper.h"
|
|
#include "logger.h"
|
|
#include "utils.h"
|
|
|
|
#include "zip.h"
|
|
#include "bitbuf.h"
|
|
|
|
#include "steam/clientmsgs.h"
|
|
|
|
CDataDumper::CDataDumper() :
|
|
m_uiMsgNum(0)
|
|
{
|
|
time_t tCurrentTime;
|
|
time(&tCurrentTime);
|
|
|
|
sprintf_s(m_szSessionDir, sizeof( m_szSessionDir ), "%d\\", tCurrentTime);
|
|
|
|
g_Logger->CreateDir(m_szSessionDir);
|
|
}
|
|
|
|
void CDataDumper::DataEncrypted(const uint8* pubPlaintextData, uint32 cubPlaintextData)
|
|
{
|
|
this->HandleNetMsg(k_eNetOutgoing, (EMsg) *(short *) pubPlaintextData, pubPlaintextData, cubPlaintextData);
|
|
}
|
|
|
|
void CDataDumper::DataDecrypted(const uint8* pubPlaintextData, uint32 cubPlaintextData)
|
|
{
|
|
this->HandleNetMsg(k_eNetIncoming, (EMsg) *(short *) pubPlaintextData, pubPlaintextData, cubPlaintextData);
|
|
}
|
|
|
|
bool CDataDumper::HandleNetMsg( ENetDirection eDirection, EMsg eMsg, const uint8 *pData, uint32 cubData )
|
|
{
|
|
//if ( eMsg == k_EMsgMulti )
|
|
// return this->MultiplexMsgMulti(eDirection, pData, cubData);
|
|
|
|
const char* szFile = this->GetFileName(eDirection, eMsg);
|
|
g_Logger->LogFileData(szFile, pData, cubData);
|
|
|
|
g_Logger->LogConsole("Wrote %d bytes to %s\n", cubData, szFile);
|
|
|
|
return true;
|
|
}
|
|
|
|
const char* CDataDumper::GetFileName(ENetDirection eDirection, EMsg eMsg)
|
|
{
|
|
static char szFileName[MAX_PATH];
|
|
|
|
sprintf_s(szFileName, sizeof( szFileName ), "%s%d_%s_%d_%s.bin", m_szSessionDir,
|
|
++m_uiMsgNum, (eDirection == k_eNetIncoming ? "in" : "out"), eMsg,
|
|
PchNameFromEMsg(eMsg));
|
|
|
|
return szFileName;
|
|
}
|
|
|
|
bool CDataDumper::MultiplexMsgMulti( ENetDirection eDirection, const uint8 *pData, uint32 cubData )
|
|
{
|
|
MsgHdr_t *pMsgHdr = (MsgHdr_t *)pData;
|
|
MsgMulti_t *pMsgBody = (MsgMulti_t *)( pData + sizeof( MsgHdr_t ) );
|
|
|
|
size_t hdrSize = sizeof( MsgHdr_t ) + sizeof( MsgMulti_t );
|
|
|
|
uint8 *pMsgData = NULL;
|
|
uint32 cubMsgData = 0;
|
|
bool bDecomp = false;
|
|
|
|
if ( pMsgBody->m_cubUnzipped != 0 )
|
|
{
|
|
// decompress our data
|
|
|
|
uint8 *pDecompressed = new uint8[ pMsgBody->m_cubUnzipped ];
|
|
uint8 *pCompressed = (uint8 *)( pData + hdrSize );
|
|
uint32 cubCompressed = cubData - hdrSize;
|
|
|
|
bool bZip = CZip::Inflate( pCompressed, cubCompressed, pDecompressed, pMsgBody->m_cubUnzipped );
|
|
|
|
if ( !bZip )
|
|
{
|
|
delete [] pDecompressed;
|
|
|
|
g_Logger->AppendFile( "EMsgLog.txt", "Decompression failed!!\r\n" );
|
|
|
|
return true;
|
|
}
|
|
|
|
pMsgData = pDecompressed;
|
|
cubMsgData = pMsgBody->m_cubUnzipped;
|
|
bDecomp = bZip;
|
|
}
|
|
else
|
|
{
|
|
pMsgData = (uint8 *)( pData + hdrSize );
|
|
cubMsgData = cubData - hdrSize;
|
|
}
|
|
|
|
bf_read reader( pMsgData, cubMsgData );
|
|
|
|
while ( reader.GetNumBytesLeft() > 0 )
|
|
{
|
|
uint32 cubPayload = (uint32)reader.ReadLong();
|
|
int off = reader.GetNumBitsRead() >> 3;
|
|
|
|
uint8 *pPayload = (uint8 *)( pMsgData + off );
|
|
EMsg *pEMsg = (EMsg *)pPayload;
|
|
|
|
reader.SeekRelative( cubPayload << 3 );
|
|
|
|
this->HandleNetMsg( eDirection, *pEMsg, pPayload, cubPayload );
|
|
}
|
|
|
|
if ( bDecomp )
|
|
delete [] pMsgData;
|
|
|
|
return true;
|
|
} |