main.h 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166
  1. #ifndef main_H
  2. #define main_H
  3. #ifdef _DEBUG
  4. #include <iostream>
  5. #include <sstream>
  6. #include <vector>
  7. #ifndef _LTMDB
  8. #define _CRTDBG_MAP_ALLOC
  9. #include <stdlib.h>
  10. #include <crtdbg.h>
  11. #define DEBUG_CLIENTBLOCK new( _CLIENT_BLOCK, __FILE__, __LINE__)
  12. #define new DEBUG_CLIENTBLOCK
  13. #define _LTMDB
  14. #endif
  15. #endif
  16. #define WIN32_LEAN_AND_MEAN
  17. #include <Windows.h>
  18. #pragma warning( push )
  19. #pragma warning( disable : 4091 )
  20. #include <DbgHelp.h>
  21. #pragma warning( pop )
  22. #define KSGStart __stdcall
  23. #include "Globals.h"
  24. namespace Framework
  25. {
  26. // Speichert die dem Programm vom Betriebssystem beim Start übergebenen Parameter
  27. struct Startparam
  28. {
  29. HINSTANCE hinst, hpinst;
  30. LPSTR cmd;
  31. int show;
  32. };
  33. // Überschreibe diese Funktion. Sie wird vom Framework automatisch beim Start des Programmes aufgerufen
  34. // p: Die Parameter, die dem Programm beim Start vom Betriebssystem übergeben wurden
  35. int KSGStart Start( Startparam p );
  36. // Initialisiert das Framework
  37. // Wird in der (WinMain) des Frameworks automatisch aufgerufen
  38. __declspec( dllexport ) void initFramework();
  39. // Gibt den duch (initFramework) benutzten Arbeitsspeicher wieder frei
  40. // Wird in der (WinMain) des Frameworks automatisch aufgerufen
  41. __declspec( dllexport ) void releaseFramework();
  42. }
  43. typedef BOOL( WINAPI *MINIDUMPWRITEDUMP )( HANDLE hProcess, DWORD dwPid, HANDLE hFile, MINIDUMP_TYPE DumpType, CONST PMINIDUMP_EXCEPTION_INFORMATION ExceptionParam, CONST PMINIDUMP_USER_STREAM_INFORMATION UserStreamParam, CONST PMINIDUMP_CALLBACK_INFORMATION CallbackParam );
  44. void createMinidump( struct _EXCEPTION_POINTERS *apExceptionInfo )
  45. {
  46. HMODULE mhLib = ::LoadLibrary( "dbghelp.dll" );
  47. MINIDUMPWRITEDUMP pDump = ( MINIDUMPWRITEDUMP )::GetProcAddress( mhLib, "MiniDumpWriteDump" );
  48. HANDLE hFile = ::CreateFile( "error_core_memory_dump.dmp", GENERIC_WRITE, FILE_SHARE_WRITE, NULL, CREATE_ALWAYS,
  49. FILE_ATTRIBUTE_NORMAL, NULL );
  50. _MINIDUMP_EXCEPTION_INFORMATION ExInfo;
  51. ExInfo.ThreadId = ::GetCurrentThreadId();
  52. ExInfo.ExceptionPointers = apExceptionInfo;
  53. ExInfo.ClientPointers = FALSE;
  54. pDump( GetCurrentProcess(), GetCurrentProcessId(), hFile, MiniDumpNormal, &ExInfo, NULL, NULL );
  55. ::CloseHandle( hFile );
  56. }
  57. LONG WINAPI unhandledHandler( struct _EXCEPTION_POINTERS *apExceptionInfo )
  58. {
  59. createMinidump( apExceptionInfo );
  60. return EXCEPTION_CONTINUE_SEARCH;
  61. }
  62. #ifdef _DEBUG
  63. template<typename TChar, typename TTraits>
  64. class OutputDebugStringBuf : public std::basic_stringbuf<TChar, TTraits>
  65. {
  66. public:
  67. explicit OutputDebugStringBuf() : _buffer( 1 )
  68. {
  69. setg( nullptr, nullptr, nullptr );
  70. setp( _buffer.data(), _buffer.data(), _buffer.data() + _buffer.size() );
  71. }
  72. ~OutputDebugStringBuf()
  73. {}
  74. static_assert( std::is_same<TChar, char>::value || std::is_same<TChar, wchar_t>::value, "OutputDebugStringBuf only supports char and wchar_t types" );
  75. int sync() try
  76. {
  77. MessageOutputer<TChar, TTraits>()( pbase(), pptr() );
  78. setp( _buffer.data(), _buffer.data(), _buffer.data() + _buffer.size() );
  79. return 0;
  80. } catch( ... )
  81. {
  82. return -1;
  83. }
  84. int overflow( int c = TTraits::eof() )
  85. {
  86. auto syncRet = sync();
  87. if( c != TTraits::eof() )
  88. {
  89. _buffer[ 0 ] = c;
  90. setp( _buffer.data(), _buffer.data() + 1, _buffer.data() + _buffer.size() );
  91. }
  92. return syncRet == -1 ? TTraits::eof() : 0;
  93. }
  94. private:
  95. std::vector<TChar> _buffer;
  96. template<typename TChar, typename TTraits>
  97. struct MessageOutputer;
  98. template<>
  99. struct MessageOutputer<char, std::char_traits<char>>
  100. {
  101. template<typename TIterator>
  102. void operator()( TIterator begin, TIterator end ) const
  103. {
  104. std::string s( begin, end );
  105. OutputDebugStringA( s.c_str() );
  106. }
  107. };
  108. template<>
  109. struct MessageOutputer<wchar_t, std::char_traits<wchar_t>>
  110. {
  111. template<typename TIterator>
  112. void operator()( TIterator begin, TIterator end ) const
  113. {
  114. std::wstring s( begin, end );
  115. OutputDebugStringW( s.c_str() );
  116. }
  117. };
  118. };
  119. #endif
  120. int WINAPI WinMain( _In_ HINSTANCE hinst, _In_opt_ HINSTANCE hpinst, _In_ LPSTR cmd, int _In_ show )
  121. {
  122. #ifdef _DEBUG
  123. _CrtSetDbgFlag( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF );
  124. OutputDebugStringBuf<char, std::char_traits<char>> charDebugOutput;
  125. std::streambuf *buf = std::cout.rdbuf();
  126. std::cout.rdbuf( &charDebugOutput );
  127. #else
  128. SetUnhandledExceptionFilter( unhandledHandler );
  129. #endif
  130. Framework::initFramework( hinst );
  131. Framework::Startparam stp;
  132. stp.hinst = hinst;
  133. stp.hpinst = hpinst;
  134. stp.cmd = cmd;
  135. stp.show = show;
  136. int ret = Framework::Start( stp );
  137. Framework::releaseFramework();
  138. #ifdef _DEBUG
  139. std::cout.rdbuf( buf );
  140. #endif
  141. return ret;
  142. }
  143. #endif