1

materialsystem/shader_dll_verify.cpp

@@ -0,0 +1,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;
+}