BugTrapSort.cpp

// BugTrapSort.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"

class ATL_NO_VTABLE CZipHandler
{
public:
	CZipHandler(LPCTSTR szFilename = NULL) throw()
		: hArchive( NULL )
	{
		if ( szFilename )
			Open( szFilename );
	}

	bool Open(LPCTSTR szFilename) throw()
	{
		if ( hArchive )
			unzClose( hArchive );
		hArchive = unzOpen( CT2CA( szFilename ) );
		if ( ! hArchive )
		{
			TCHAR szFileShort[ MAX_PATH ];
			if ( GetShortPathName( szFilename, szFileShort, MAX_PATH ) )
				hArchive = unzOpen( CT2CA( szFileShort ) );
		}
		return ( hArchive != NULL );
	}

	~CZipHandler() throw()
	{
		if ( hArchive )
		{
			unzClose( hArchive );
			hArchive = NULL;
		}
	}

	operator unzFile() const throw()
	{
		return hArchive;
	}

	bool Extract(LPCTSTR szSrc, LPCTSTR szDst) throw()
	{
		bool ret = false;
		char* pBuf = NULL;
		DWORD dwSize = 0;
		if ( Extract( szSrc, &pBuf, &dwSize ) )
		{
			HANDLE hFile = CreateFile( szDst, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL );
			if ( hFile != INVALID_HANDLE_VALUE )
			{
				DWORD dwWritten = 0;
				if ( WriteFile( hFile, pBuf, dwSize, &dwWritten, NULL ) )
				{
					ret = ( dwWritten == dwSize );
				}
			}
			CloseHandle( hFile );
		}
		delete [] pBuf;
		return ret;
	}

	bool Extract(LPCTSTR szSrc, char** ppBuf, DWORD* pdwSize = NULL) throw()
	{
		if ( pdwSize )
			*pdwSize = 0;
		*ppBuf = NULL;

		bool ret = false;
		if ( unzLocateFile( hArchive, CT2CA( szSrc ), 2 ) == UNZ_OK )
		{
			unz_file_info fi = {};
			if ( unzGetCurrentFileInfo( hArchive, &fi, NULL, NULL, NULL, NULL, NULL, NULL ) == UNZ_OK )
			{
				if ( unzOpenCurrentFile( hArchive ) == UNZ_OK ) 
				{
					if ( pdwSize )
						*pdwSize = fi.uncompressed_size;
					*ppBuf = new char[ fi.uncompressed_size + 2 ];
					if ( *ppBuf )
					{
						ZeroMemory( *ppBuf, fi.uncompressed_size + 2 );
						ret = ( unzReadCurrentFile( hArchive, *ppBuf, fi.uncompressed_size ) == (int)fi.uncompressed_size );
					}
					unzCloseCurrentFile( hArchive );
				}
			}
		}
		return ret;
	}

protected:
	unzFile hArchive;
};

typedef struct 
{
	CString sIf;
	CString sIs;
	CString sThen;
} CRule;

typedef CAtlList< CRule > CRuleList;
typedef CAtlMap< CString, bool > CStringMap;

CString		g_sConfig;
CRuleList	g_oRules;
CString		g_sOutput;
CString		g_sInput;
CStringMap	g_oModules;

#define REPLACE(x) \
	x.Replace( _T("{VER_FULL}"), sVersion ); \
	x.Replace( _T("{VER_NUMBER}"), sVersionNumber ); \
	x.Replace( _T("{VER_CONFIG}"), sVersionConfig ); \
	x.Replace( _T("{VER_PLATFORM}"), sVersionPlatform ); \
	x.Replace( _T("{VER_REVISION}"), sVersionRevision ); \
	x.Replace( _T("{VER_DATE}"), sVersionDate ); \
	x.Replace( _T("{TIMESTAMP}"), sTimestamp ); \
	x.Replace( _T("{USER}"), sUser ); \
	x.Replace( _T("{COMPUTER}"), sComputer ); \
	x.Replace( _T("{CPU}"), sCPU ); \
	x.Replace( _T("{WHAT}"), sWhat ); \
	x.Replace( _T("{MODULE}"), sModule ); \
	x.Replace( _T("{ADDRESS}"), sAddress );

CString GetValue(IXMLDOMElement* pRoot, LPCTSTR szXPath)
{
	CComPtr< IXMLDOMNode > pNode;
	HRESULT hr = pRoot->selectSingleNode( CComBSTR( szXPath ), &pNode );
	if ( hr != S_OK )
		return CString();

	CComBSTR bstrVersion;
	hr = pNode->get_text( &bstrVersion );
	if ( hr != S_OK )
		return CString();

	return (LPCWSTR)bstrVersion;
}

CString GetTime(IXMLDOMElement* pRoot, LPCTSTR szNode, LPCTSTR szFormat)
{
	try 
    {
		FILETIME fTimestamp;
		(__int64&)fTimestamp = _tstoi64( GetValue( pRoot, szNode ) );
		CTime tTimeStamp( fTimestamp );
		return tTimeStamp.FormatGmt( szFormat );
	} 
    catch( ... )
    {
    }
	return CString();
}

bool ProcessReport(const CString& sInput)
{
	_tprintf( _T("Processing %s ...\n"), PathFindFileName( sInput ) );

	bool bZip = ( sInput.Find( _T(".zip") ) != -1 );

	CComPtr< IXMLDOMDocument > pFile;
	HRESULT hr = pFile.CoCreateInstance( CLSID_DOMDocument );
	if ( FAILED( hr ) )
		return false;

	hr = pFile->put_async( VARIANT_FALSE );
	if ( hr != S_OK )
		return false;

	CZipHandler pZip;
	VARIANT_BOOL ret = VARIANT_FALSE;
	if ( bZip )
	{
		if ( ! pZip.Open( sInput ) )
			return false;
		char* pBuf = NULL;
		bool res = pZip.Extract( _T("errorlog.xml"), &pBuf );
		if ( res )
			hr = pFile->loadXML( CComBSTR( pBuf ), &ret );
		delete [] pBuf;
		if ( ! res )
			return false;
	}
	else
	{
		hr = pFile->load( CComVariant( sInput + _T("errorlog.xml") ), &ret );
	}
	if ( hr != S_OK || ! ret )
		return false;

	CComPtr< IXMLDOMElement > pRoot;
	hr = pFile->get_documentElement( &pRoot );
	if ( hr != S_OK )
		return false;

	CString sTimestamp = GetTime( pRoot, _T("/report/timestamp"), _T("%y%m%d-%H%M%S") );
	if ( sTimestamp.IsEmpty() ) sTimestamp = _T("000000-000000");

	CString sVersion = GetValue( pRoot, _T("/report/version") );

	CString sComputer = GetValue( pRoot, _T("/report/computer") );
	if ( sComputer.IsEmpty() ) sComputer = _T("UNKNOWN");

	CString sUser = GetValue( pRoot, _T("/report/user") );
	if ( sUser.IsEmpty() ) sUser = _T("UNKNOWN");

	CString sCPU = GetValue( pRoot, _T("/report/cpus/cpu/description") );
	if ( sCPU.IsEmpty() )
	{
		sCPU = GetValue( pRoot, _T("/report/cpus/cpu/id") );
		if ( sCPU.IsEmpty() )
			sCPU = _T("UNKNOWN");
	}

	CString sWhat = GetValue( pRoot, _T("/report/error/what") );
	if ( sWhat.IsEmpty() ) sWhat = _T("UNKNOWN_ERROR");

	CString sAddress = GetValue( pRoot, _T("/report/error/address") );
	sAddress = sAddress.Right( 4 );
	sAddress.MakeLower();

	CString sModule = PathFindFileName( GetValue( pRoot, _T("/report/error/module") ) );
	sModule.MakeLower();
	bool bGood;
	if ( sModule.IsEmpty() ||
		( g_oModules.Lookup( sModule, bGood ) && ! bGood ) ||
		( g_oModules.Lookup( sModule + _T(":") + sAddress, bGood ) && ! bGood ) )
	{
		CString sOrigAddress = sAddress;
		CString sOrigModule = sModule;

		CComPtr< IXMLDOMNode > pStack;
		hr = pRoot->selectSingleNode( CComBSTR( _T("/report/threads/thread/stack") ), &pStack );
		if ( hr == S_OK )
		{
			CComPtr< IXMLDOMNodeList > pFrames;
			hr = pStack->get_childNodes( &pFrames );
			if ( hr == S_OK )
			{
				long n = 0;
				hr = pFrames->get_length( &n );
				for ( long i = 0; hr == S_OK && i < n; ++i )
				{
					CComPtr< IXMLDOMNode > pFrame;
					hr = pFrames->get_item( i, &pFrame );
					if ( hr == S_OK )
					{
						CComQIPtr< IXMLDOMElement > pFrameE( pFrame );

						sAddress = GetValue( pFrameE, _T("address") );
						sAddress = sAddress.Right( 4 );
						sAddress.MakeLower();

						sModule = PathFindFileName( GetValue( pFrameE, _T("module") ) );
						if ( ! sModule.IsEmpty() )
						{
							sModule.MakeLower();
							if ( ( ! g_oModules.Lookup( sModule, bGood ) || bGood ) &&
								 ( ! g_oModules.Lookup( sModule + _T(":") + sAddress, bGood ) || bGood ) )
								// Íåèçâåñòíûé ìîäóëü èëè ïîäõîäÿùèé ìîäóëü
								break;
						}
						sAddress.Empty();
						sModule.Empty();
					}
				}
			}
		}
		if ( sModule.IsEmpty() )
		{
			sAddress = sOrigAddress;
			sModule = sOrigModule;
		}
	}
	if ( sAddress.IsEmpty() ) sAddress = _T("0000");
	if ( sModule.IsEmpty() ) sModule = _T("UNKNOWN");

	// Ðàçáîð âåðñèè
	int i = 0;
	CString sVersionNumber = sVersion.Tokenize( _T(" "), i );
	if ( i == -1 ) sVersionNumber = _T("0.0.0.0");
	
	CString sVersionConfig = ( i == -1 ) ? _T("") : sVersion.Tokenize( _T(" "), i );
	if ( i == -1 ) sVersionConfig = _T("u");
	if ( sVersionConfig.CompareNoCase( _T("release") ) == 0 ) sVersionConfig = _T("r");
	else if ( sVersionConfig.CompareNoCase( _T("debug") ) == 0 ) sVersionConfig = _T("d");
	
	CString sVersionPlatform = ( i == -1 ) ? _T("") : sVersion.Tokenize( _T(" "), i );
	if ( i == -1 ) sVersionPlatform = _T("xx");
	if ( sVersionPlatform.Find( _T("32") ) != -1 ) sVersionPlatform = _T("32");
	else if ( sVersionPlatform.Find( _T("64") ) != -1 ) sVersionPlatform = _T("64");
	
	CString sVersionRevision = ( i == -1 ) ? _T("") : sVersion.Tokenize( _T(" "), i );
	if ( i == -1 ) sVersionRevision = _T("UNKNOWN");
	sVersionRevision.Trim( _T(" ()") );
	sVersionRevision.MakeLower();
	
	CString sVersionDate = ( i == -1 ) ? _T("") : sVersion.Tokenize( _T(" "), i );
	if ( i == -1 ) sVersionDate = _T("00000000");
	sVersionDate.Trim( _T(" ()") );

	if ( sVersion.IsEmpty() )
		sVersion = _T("0.0.0.0");

	_tprintf( _T("Version: %ls\n"), sVersion );

	CString sOutput = g_sOutput;
	for ( POSITION pos = g_oRules.GetHeadPosition(); pos ; )
	{
		CRule r = g_oRules.GetNext( pos );
		REPLACE( r.sIf );
		if ( StrStrI( r.sIf, r.sIs ) )
		{
			sOutput = r.sThen;
			break;
		}
	}

	if ( sOutput.IsEmpty() )
		// Íåò äåéñòâèÿ
		return true;

	REPLACE( sOutput );

	if ( sOutput.IsEmpty() )
		// Íåò äåéñòâèÿ
		return true;

	sOutput.TrimRight( _T("\\") ) += _T("\\");

	int res = SHCreateDirectory( GetDesktopWindow(), sOutput );
	if ( res != ERROR_SUCCESS && res != ERROR_FILE_EXISTS && res != ERROR_ALREADY_EXISTS )
		return false;

	if ( bZip )
	{
		if ( ! pZip.Extract( _T("errorlog.xml"), sOutput + _T("errorlog.xml") ) )
			return false;
		pZip.Extract( _T("crashdump.dmp"), sOutput + _T("crashdump.dmp") );
		pZip.Extract( _T("settings.reg"), sOutput + _T("settings.reg") );
	}
	else
	{
		if ( ! CopyFile( sInput + _T("errorlog.xml"), sOutput + _T("errorlog.xml"), FALSE ) )
			return false;
		CopyFile( sInput + _T("crashdump.dmp"), sOutput + _T("crashdump.dmp"), FALSE );
		CopyFile( sInput + _T("settings.reg"), sOutput + _T("settings.reg"), FALSE );
	}

	return true;
}

bool Enum(LPCTSTR szInput, CAtlList< CString >& oDirs)
{
	bool ret = false;
	CString sDir = szInput;
	sDir = sDir.Left( (DWORD)( PathFindFileName( szInput ) - szInput ) );

	WIN32_FIND_DATA wfa = {};
	HANDLE hFF = FindFirstFile( szInput, &wfa );
	if ( hFF != INVALID_HANDLE_VALUE )
	{
		do 
		{
			if ( *wfa.cFileName != _T('.') )
			{
				if ( ( wfa.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ) != 0 )
				{
					ret = true;

					if ( ! Enum( sDir + wfa.cFileName + _T("\\*.*"), oDirs ) )
						// Add folder without subfolders
						oDirs.AddTail( sDir + wfa.cFileName + _T("\\") );
				}
				else if ( _tcsstr( wfa.cFileName, _T(".zip") ) )
				{
					// Add .zip-file
					oDirs.AddTail( sDir + wfa.cFileName );					
				}
			}
		}
		while ( FindNextFile( hFF, &wfa ) );

		FindClose( hFF );
	}

	return ret;
}

int _tmain(int argc, _TCHAR* argv[])
{
	if ( argc != 2 )
		return 1;

	CoInitializeEx( NULL, COINIT_MULTITHREADED );

	g_sConfig = argv[ 1 ];

	GetPrivateProfileString( _T("options"), _T("output"), NULL,
		g_sOutput.GetBuffer( 4096 ), 4096, g_sConfig );
	g_sOutput.ReleaseBuffer();
	if ( ! g_sOutput.IsEmpty() )
		g_sOutput.TrimRight( _T("\\") ) += _T("\\");

	GetPrivateProfileString( _T("options"), _T("input"), NULL,
		g_sInput.GetBuffer( 4096 ), 4096, g_sConfig );
	g_sInput.ReleaseBuffer();
	if ( g_sInput.IsEmpty() )
		g_sInput = _T(".\\*.*");

	// Ïðèíÿòü
	CString sModules;
	GetPrivateProfileString( _T("options"), _T("accept"), NULL,
		sModules.GetBuffer( 4096 ), 4096, g_sConfig );
	sModules.ReleaseBuffer();
	for ( int i = 0; ; )
	{
		CString sModule = sModules.Tokenize( _T("|"), i );
		if ( sModule.IsEmpty() || i == -1 )
			break;
		sModule.MakeLower();
		bool foo;
		if ( g_oModules.Lookup( sModule, foo ) )
		{
			_tprintf( _T("Duplicate in [options] \"accept\" parameter: %s\n"), sModule );
			return 1;
		}
		g_oModules.SetAt( sModule, true );
	}

	// Íå ïðèíèìàòü
	GetPrivateProfileString( _T("options"), _T("ignore"), NULL,
		sModules.GetBuffer( 4096 ), 4096, g_sConfig );
	sModules.ReleaseBuffer();
	for ( int i = 0; ; )
	{
		CString sModule = sModules.Tokenize( _T("|"), i );
		if ( sModule.IsEmpty() || i == -1 )
			break;
		sModule.MakeLower();

		bool foo;
		if ( g_oModules.Lookup( sModule, foo ) )
		{
			_tprintf( _T("Duplicate in [options] \"ignore\" parameter: %s\n"), sModule );
			return 1;
		}
		g_oModules.SetAt( sModule, false );
	}

	// Çàãðóçêà ñïèñêà ñåêöèé
	CAutoVectorPtr< TCHAR > pSections( new TCHAR[ 16384 ] );
	GetPrivateProfileString( NULL, NULL, NULL, pSections, 16384, g_sConfig );

	// Çàãðóçêà ïðàâèë èç âñåõ ñåêöèé
	for ( LPCTSTR szSect = pSections ; szSect && *szSect ; szSect += lstrlen( szSect ) + 1 )
	{
		if  ( lstrcmpi( szSect, _T("options") ) == 0 )
			// Ïðîïóñê ñëóæåáíîé ñåêöèè
			continue;

		CRule r;

		GetPrivateProfileString( szSect, _T("if"), NULL,
			r.sIf.GetBuffer( 4096 ), 4096, g_sConfig );
		r.sIf.ReleaseBuffer();

		GetPrivateProfileString( szSect, _T("output"), NULL,
			r.sThen.GetBuffer( 4096 ), 4096, g_sConfig );
		r.sThen.ReleaseBuffer();

		CString sIses;
		GetPrivateProfileString( szSect, _T("is"), NULL,
			sIses.GetBuffer( 4096 ), 4096, g_sConfig );
		sIses.ReleaseBuffer();
		for ( int i = 0; ; )
		{
			r.sIs = sIses.Tokenize( _T("|"), i );
			if ( r.sIs.IsEmpty() || i == -1 )
				break;
			g_oRules.AddTail( r );
		}
	}

	pSections.Free();

	CAtlList< CString > oDirs;
	Enum( g_sInput, oDirs );
	_tprintf( _T("Processing %d reports from folder %s ...\n"),
		oDirs.GetCount(), g_sInput );

	for ( POSITION pos = oDirs.GetHeadPosition(); pos; )
	{
		if ( ! ProcessReport( oDirs.GetNext( pos ) ) )
		{
			_tprintf( _T("Report load error!\n") );
		}
	}

	_tprintf( _T("Done.\n") );

	CoUninitialize();

	return 0;
}