source-engine/materialsystem/shader_dll_verify.cpp
FluorescentCIAAfricanAmerican 3bf9df6b27 1
2020-04-22 12:56:21 -04:00

97 lines
2.1 KiB
C++

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
//=============================================================================
#include <windows.h>
#include "shader_dll_verify.h"
static unsigned char *g_pLastInputData = 0;
static HANDLE g_hDLLInst = 0;
extern "C"
{
void __declspec( dllexport ) _ftol3( char *pData );
BOOL WINAPI DllMain (HANDLE hInst, ULONG ulInit, LPVOID lpReserved);
};
BOOL WINAPI DllMain (HANDLE hInst, ULONG ulInit, LPVOID lpReserved)
{
lpReserved = lpReserved;
ulInit = ulInit;
g_hDLLInst = hInst;
return true;
}
class CShaderDLLVerification : public IShaderDLLVerification
{
public:
virtual CRC32_t Function1( unsigned char *pData );
virtual void Function2( int a, int b, int c );
virtual void Function3( int a, int b, int c );
virtual void Function4( int a, int b, int c );
virtual CRC32_t Function5();
};
static CShaderDLLVerification g_Blah;
// The main exported function.. return a pointer to g_Blah.
void __declspec( dllexport ) _ftol3( char *pData )
{
pData += SHADER_DLL_VERIFY_DATA_PTR_OFFSET;
char *pToFillIn = (char*)&g_Blah;
memcpy( pData, &pToFillIn, 4 );
}
CRC32_t CShaderDLLVerification::Function1( unsigned char *pData )
{
pData += SHADER_DLL_VERIFY_DATA_PTR_OFFSET;
g_pLastInputData = (unsigned char*)pData;
void *pVerifyPtr1 = &g_Blah;
CRC32_t testCRC;
CRC32_Init( &testCRC );
CRC32_ProcessBuffer( &testCRC, pData, SHADER_DLL_VERIFY_DATA_LEN1 );
CRC32_ProcessBuffer( &testCRC, &g_hDLLInst, 4 );
CRC32_ProcessBuffer( &testCRC, &pVerifyPtr1, 4 );
CRC32_Final( &testCRC );
return testCRC;
}
void CShaderDLLVerification::Function2( int a, int b, int c )
{
a=b=c;
MD5Context_t md5Context;
MD5Init( &md5Context );
MD5Update( &md5Context, g_pLastInputData + SHADER_DLL_VERIFY_DATA_PTR_OFFSET, SHADER_DLL_VERIFY_DATA_LEN1 - SHADER_DLL_VERIFY_DATA_PTR_OFFSET );
MD5Final( g_pLastInputData, &md5Context );
}
void CShaderDLLVerification::Function3( int a, int b, int c )
{
a=b=c;
}
void CShaderDLLVerification::Function4( int a, int b, int c )
{
a=b=c;
}
CRC32_t CShaderDLLVerification::Function5()
{
return 32423;
}