1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
//========= 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;
}
|