[Mobies-commit] [commit] r3870 - UDM/trunk/tests/test_staticleaks

[ksmyth at redhat1.isis.vanderbilt.edu]

Author: ksmyth
Date: Sat Aug 27 11:53:58 2011
New Revision: 3870

Log:
Test dll and proc load

Modified:
   UDM/trunk/tests/test_staticleaks/test_staticleaks.cpp

Modified: UDM/trunk/tests/test_staticleaks/test_staticleaks.cpp
==============================================================================
--- UDM/trunk/tests/test_staticleaks/test_staticleaks.cpp	Sat Aug 27 11:53:57 2011	(r3869)
+++ UDM/trunk/tests/test_staticleaks/test_staticleaks.cpp	Sat Aug 27 11:53:58 2011	(r3870)
@@ -31,11 +31,15 @@
 	_CrtMemState dllState;
 
 	HMODULE test_staticleaksdll = LoadLibrary("test_staticleaksdll.dll");
-#if defined(_M_AMD64)
+	if (!test_staticleaksdll)
+		DebugBreak();
+#if defined(_M_AMD64)
 	init = (init_t)GetProcAddress(test_staticleaksdll , "?init@@YAXPEAX at Z");
-#elif defined(_M_IX86)
+#elif defined(_M_IX86)
 	init = (init_t)GetProcAddress(test_staticleaksdll , "?init@@YAXPAX at Z");
 #endif
+	if (!init)
+		DebugBreak();
 	init(&dllState);
 
 	typedef void (*terminate_t)();