mirror of
https://github.com/nillerusr/source-engine.git
synced 2024-12-22 06:06:50 +00:00
540 lines
15 KiB
C++
540 lines
15 KiB
C++
//========= Copyright Valve Corporation, All rights reserved. ============//
|
|
#include <windows.h>
|
|
#include "mdmpRipper.h"
|
|
#include <vgui/IVGui.h>
|
|
#include "vgui_controls/MessageMap.h"
|
|
#include "vgui_controls/MenuBar.h"
|
|
#include "vgui_controls/Menu.h"
|
|
#include "vgui_controls/MessageBox.h"
|
|
#include "tier1/KeyValues.h"
|
|
#include "vgui/ISurface.h"
|
|
#include <vgui/ILocalize.h>
|
|
#include "vgui_controls/Frame.h"
|
|
#include "CMDModulePanel.h"
|
|
#include "vgui_controls/ListPanel.h"
|
|
#include <vgui_controls/RichText.h>
|
|
#include "KeyValues.h"
|
|
#include "vgui/ISystem.h"
|
|
#include "vgui_controls/FileOpenDialog.h"
|
|
#include "isqlwrapper.h"
|
|
#include "CMDRipperMain.h"
|
|
|
|
extern ISQLWrapper *g_pSqlWrapper;
|
|
|
|
using namespace vgui;
|
|
|
|
|
|
|
|
CMDModulePanel::CMDModulePanel( vgui::Panel *pParent, const char *pName ) :
|
|
BaseClass( pParent, pName, true )
|
|
{
|
|
m_pTokenList = new ListPanel(this, "ModuleList");
|
|
m_pTokenList->AddColumnHeader(0, "name", "Module Name", 600, 0);
|
|
m_pTokenList->AddColumnHeader(1, "version", "Version", 100, 0);
|
|
m_pTokenList->AddColumnHeader(2, "count", "Count", 86, 0);
|
|
m_pTokenList->AddActionSignalTarget( this );
|
|
|
|
m_pAnalyzeText = new RichText(this, "AnalyzeText");
|
|
m_pAnalyzeText->SetVerticalScrollbar(true);
|
|
LoadControlSettings( "MDModulePanel.res" );
|
|
m_pAnalyzeText->InsertString("Initializing...\n");
|
|
|
|
InitializeDebugEngine();
|
|
LoadKnownModules();
|
|
|
|
m_hThread = NULL;
|
|
|
|
// SetTitleBarVisible( false );
|
|
// SetSizeable( false );
|
|
|
|
|
|
//SETUP_PANEL( this );
|
|
}
|
|
|
|
CMDModulePanel::~CMDModulePanel( void )
|
|
{
|
|
ReleaseDebugEngine( );
|
|
}
|
|
|
|
void CMDModulePanel::OnKeyCodeTyped( KeyCode code )
|
|
{
|
|
switch ( code )
|
|
{
|
|
case KEY_G:
|
|
UpdateKnownDB( "GOOD" );
|
|
break;
|
|
case KEY_B:
|
|
UpdateKnownDB( "BAD" );
|
|
break;
|
|
case KEY_U:
|
|
UpdateKnownDB( "UNKNOWN" );
|
|
break;
|
|
case KEY_F:
|
|
ModuleLookUp();
|
|
break;
|
|
}
|
|
}
|
|
|
|
void CMDModulePanel::OnCommand( const char *pCommand )
|
|
{
|
|
if ( !Q_strcmp( pCommand, "Close" ) )
|
|
{
|
|
//we want to close
|
|
Close();
|
|
}
|
|
if ( !Q_strcmp( pCommand, "ModuleLookUp" ) )
|
|
{
|
|
ModuleLookUp();
|
|
}
|
|
if ( !Q_strcmp( pCommand, "SetGood" ) )
|
|
{
|
|
UpdateKnownDB( "GOOD" );
|
|
}
|
|
if ( !Q_strcmp( pCommand, "SetBad" ) )
|
|
{
|
|
UpdateKnownDB( "BAD" );
|
|
}
|
|
if ( !Q_strcmp( pCommand, "SetUnknown" ) )
|
|
{
|
|
UpdateKnownDB( "UNKNOWN" );
|
|
}
|
|
}
|
|
|
|
void CMDModulePanel::Close()
|
|
{
|
|
if ( this )
|
|
{
|
|
m_pTokenList->DeleteAllItems();
|
|
m_MiniDumpList.RemoveAll();
|
|
m_knownModuleList.RemoveAll();
|
|
m_pAnalyzeText->SetText("");
|
|
SetVisible( false );
|
|
KeyValues *kv = new KeyValues( "Refresh" );
|
|
this->PostActionSignal( kv );
|
|
}
|
|
}
|
|
|
|
void CMDModulePanel::Create( CUtlVector<CMiniDumpObject *> *pMiniDump )
|
|
{
|
|
LoadKnownModules();
|
|
|
|
for ( int i = 0; i < pMiniDump->Count(); i++ )
|
|
{
|
|
pMiniDump->Element(i)->PopulateListPanel( m_pTokenList, true );
|
|
}
|
|
}
|
|
|
|
void CMDModulePanel::Create( const char *filename )
|
|
{
|
|
if ( g_pFullFileSystem->FileExists( filename ) )
|
|
{
|
|
LoadKnownModules();
|
|
|
|
CMiniDumpObject *newMDObj = new CMiniDumpObject( filename, &m_knownModuleList );
|
|
m_MiniDumpList.AddToTail( newMDObj );
|
|
newMDObj->PopulateListPanel( m_pTokenList, false );
|
|
|
|
AnalyzeDumpFile( filename );
|
|
}
|
|
}
|
|
|
|
void CMDModulePanel::ModuleLookUp()
|
|
{
|
|
int selectedIndex = m_pTokenList->GetSelectedItem( 0 );
|
|
void *kv = m_pTokenList->GetItem( selectedIndex );
|
|
if ( kv )
|
|
{
|
|
const char *val = ((KeyValues *)kv)->GetString( "name", "" );
|
|
if ( val )
|
|
{
|
|
const char *moduleName = strrchr( val, '\\' ) + 1;
|
|
char google[1024] = "";
|
|
sprintf( google, "http://www.google.com/search?hl=en&q=%s", moduleName);
|
|
KeyValues *kvPost = new KeyValues( "ModuleLookUp", "url", google );
|
|
this->PostActionSignal( kvPost );
|
|
}
|
|
}
|
|
}
|
|
|
|
void SeparateVersion( const char *version, char *v1buf, char *v2buf, char *v3buf, char *v4buf )
|
|
{
|
|
const char *endV1 = strchr( version, '.' )+1;
|
|
const char *endV2 = strchr( endV1+1, '.' )+1;
|
|
const char *endV3 = strchr( endV2+1, '.' )+1;
|
|
_mbsnbcpy( (unsigned char *)v1buf, (const unsigned char*)version, endV1 - version );
|
|
v1buf[endV1 - version - 1] = 0;
|
|
_mbsnbcpy( (unsigned char *)v2buf, (const unsigned char*)endV1, endV2 - endV1 );
|
|
v2buf[endV2 - endV1 - 1] = 0;
|
|
_mbsnbcpy( (unsigned char *)v3buf, (const unsigned char*)endV2, endV3 - endV2 );
|
|
v3buf[endV3 - endV2 - 1] = 0;
|
|
strcpy( v4buf, endV3 );
|
|
}
|
|
|
|
void SetKeyValueColor( char *type, KeyValues *kv, bool knownVersion )
|
|
{
|
|
int colorValue = 255;
|
|
if( !knownVersion )
|
|
colorValue = 155;
|
|
|
|
if ( !Q_strcmp( "GOOD", type ) )
|
|
{
|
|
((KeyValues *)kv)->SetColor( "cellcolor", Color(0,colorValue,0,255));
|
|
}
|
|
else if ( !Q_strcmp( "BAD", type ) )
|
|
{
|
|
((KeyValues *)kv)->SetColor( "cellcolor", Color(colorValue,0,0,255));
|
|
}
|
|
else
|
|
{
|
|
((KeyValues *)kv)->SetColor( "cellcolor", Color(255,255,0,255));
|
|
}
|
|
}
|
|
|
|
void CMDModulePanel::UpdateKnownDB( char *type )
|
|
{
|
|
int selectedIndex = m_pTokenList->GetSelectedItem( 0 );
|
|
void *kv = m_pTokenList->GetItem( selectedIndex );
|
|
char v1buf[10];
|
|
char v2buf[10];
|
|
char v3buf[10];
|
|
char v4buf[10];
|
|
char name[65];
|
|
char keybuf[10];
|
|
if ( kv )
|
|
{
|
|
SetKeyValueColor( type, (KeyValues *)kv, true );
|
|
int key = ((KeyValues *)kv)->GetInt( "key" );
|
|
itoa( key, keybuf, 10 );
|
|
strcpy( name, strrchr(((KeyValues *)kv)->GetString( "name" ), '\\')+1);
|
|
SeparateVersion( ((KeyValues *)kv)->GetString("version"), v1buf, v2buf, v3buf, v4buf );
|
|
if ( key == 0 )
|
|
{
|
|
//as far as we know, this is a non-existant module.
|
|
if ( !Q_strcmp( type, "UNKNOWN" ) )
|
|
{
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
char query[1024];
|
|
sprintf( query, "select * from knownmodules where name = \"%s\" and version1 = %s and version2 = %s and version3 = %s and version4 = %s;",
|
|
name, v1buf, v2buf, v3buf, v4buf );
|
|
IResultSet *results = g_pSqlWrapper->PResultSetQuery( query ); // do the query
|
|
if ( !results )
|
|
{
|
|
return;
|
|
}
|
|
|
|
int numResults = results->GetCSQLRow();
|
|
if ( numResults > 0 )
|
|
{
|
|
//there is an entry... get our module list up to date with this entry
|
|
|
|
const ISQLRow *row = results->PSQLRowNextResult();
|
|
Assert( row != NULL );
|
|
int realKey = row->NData(0);
|
|
const char *realType = row->PchData(6);
|
|
g_pSqlWrapper->FreeResult();
|
|
|
|
((KeyValues *)kv)->SetInt( "key", realKey );
|
|
|
|
if ( !Q_strcmp( realType, type ) )
|
|
{
|
|
//this user was out of sync with the database. It doesn't actually need updating.
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
char update[1024];
|
|
sprintf( update, "update knownmodules set type=\"%s\" where id = %i;", type, realKey);
|
|
g_pSqlWrapper->BInsert( update );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
g_pSqlWrapper->FreeResult();
|
|
//it isn't in there. Let's add it.
|
|
char update[1024];
|
|
sprintf( update, "insert into knownmodules set name = \"%s\", version1 = %s, version2 = %s, version3 = %s, version4 = %s, type = \"%s\";",
|
|
name, v1buf, v2buf, v3buf, v4buf, type);
|
|
g_pSqlWrapper->BInsert( update );
|
|
results = g_pSqlWrapper->PResultSetQuery( query ); // do the query
|
|
int numResults = results->GetCSQLRow();
|
|
if ( numResults > 0 )
|
|
{
|
|
const ISQLRow *row = results->PSQLRowNextResult();
|
|
Assert( row != NULL );
|
|
int realKey = row->NData(0);
|
|
((KeyValues *)kv)->SetInt( "key", realKey );
|
|
}
|
|
g_pSqlWrapper->FreeResult();
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
char query[1024];
|
|
sprintf( query, "select * from knownmodules where id = %i;",
|
|
key);
|
|
IResultSet *results = g_pSqlWrapper->PResultSetQuery( query ); // do the query
|
|
|
|
int numResults = results->GetCSQLRow();
|
|
if ( numResults > 0 )
|
|
{
|
|
//there is an entry... update it with the new info...
|
|
|
|
const ISQLRow *row = results->PSQLRowNextResult();
|
|
Assert( row != NULL );
|
|
Assert( numResults == 1 );
|
|
Assert( !Q_stricmp( name, row->PchData(1) ) && atoi( v1buf ) == row->NData(2) && atoi( v2buf ) == row->NData(3) &&
|
|
atoi( v3buf ) == row->NData(4) && atoi( v4buf ) == row->NData(5) );
|
|
int realKey = row->NData(0);
|
|
const char *realType = row->PchData(6);
|
|
g_pSqlWrapper->FreeResult();
|
|
|
|
if ( !Q_strcmp( realType, type ) )
|
|
{
|
|
//we don't need to update... it is already updated already
|
|
return;
|
|
}
|
|
|
|
char update[1024];
|
|
sprintf( update, "update knownmodules set type=\"%s\" where id = %i;", type, realKey);
|
|
g_pSqlWrapper->BInsert( update );
|
|
|
|
}
|
|
else
|
|
{
|
|
//the module entry was mis-keyed. First, check for an existing entry of this module.
|
|
char query[1024];
|
|
sprintf( query, "select * from knownmodules where name = \"%s\" and version1 = %s and version2 = %s and version3 = %s and version4 = %s;",
|
|
name, v1buf, v2buf, v3buf, v4buf );
|
|
IResultSet *results = g_pSqlWrapper->PResultSetQuery( query ); // do the query
|
|
|
|
int numResults = results->GetCSQLRow();
|
|
if ( numResults > 0 )
|
|
{
|
|
//there is an existing entry. Update its type and update the key for this keyvalue;
|
|
const ISQLRow *row = results->PSQLRowNextResult();
|
|
int realKey = row->NData(0);
|
|
((KeyValues *)kv)->SetInt( "key", realKey );
|
|
g_pSqlWrapper->FreeResult();
|
|
char update[1024];
|
|
sprintf( update, "update knownmodules set type=\"%s\" where id = %i;", type, realKey);
|
|
g_pSqlWrapper->BInsert( update );
|
|
}
|
|
else
|
|
{
|
|
g_pSqlWrapper->FreeResult();
|
|
//no exisiting entry. Insert it.
|
|
char update[1024];
|
|
sprintf( update, "insert into knownmodules set name = \"%s\", version1 = %s, version2 = %s, version3 = %s, version4 = %s, type = \"%s\";",
|
|
name, v1buf, v2buf, v3buf, v4buf, type);
|
|
g_pSqlWrapper->BInsert( update );
|
|
results = g_pSqlWrapper->PResultSetQuery( query ); // do the query
|
|
int numResults = results->GetCSQLRow();
|
|
if ( numResults > 0 )
|
|
{
|
|
const ISQLRow *row = results->PSQLRowNextResult();
|
|
Assert( row != NULL );
|
|
int realKey = row->NData(0);
|
|
((KeyValues *)kv)->SetInt( "key", realKey );
|
|
}
|
|
g_pSqlWrapper->FreeResult();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void CMDModulePanel::OnCompare( KeyValues *data )
|
|
{
|
|
LoadKnownModules();
|
|
|
|
CUtlVector<HANDLE> *pMiniDumpHandles = (CUtlVector<HANDLE> *)(void *)data->GetInt( "handlePointer" );
|
|
|
|
|
|
DWORD error;
|
|
int returnValue = 0;
|
|
for( int i = 0; i < pMiniDumpHandles->Count(); i++ )
|
|
{
|
|
m_MiniDumpList.AddToTail( new CMiniDumpObject( pMiniDumpHandles->Element( i ), &m_knownModuleList ) );
|
|
returnValue = CloseHandle( pMiniDumpHandles->Element( i ) );
|
|
error = GetLastError();
|
|
}
|
|
Create( &m_MiniDumpList );
|
|
SetVisible( true );
|
|
MoveToFront();
|
|
|
|
pMiniDumpHandles->RemoveAll();
|
|
system("rmdir c:\\minidumptool /s/q");
|
|
}
|
|
|
|
|
|
void CMDModulePanel::OnDbgOutput( int iMask, const char *pszDebugText)
|
|
{
|
|
if ( m_pAnalyzeText && pszDebugText )
|
|
{
|
|
m_pAnalyzeText->InsertString( pszDebugText );
|
|
}
|
|
}
|
|
|
|
|
|
DWORD WINAPI CMDModulePanel::StaticAnalyzeThread( LPVOID lParam )
|
|
{
|
|
CMDModulePanel *pClass = (CMDModulePanel *)lParam;
|
|
if ( pClass )
|
|
{
|
|
pClass->AnalyzeThread( );
|
|
}
|
|
|
|
return ( 0 );
|
|
}
|
|
|
|
void CMDModulePanel::LoadKnownModules()
|
|
{
|
|
if ( m_knownModuleList.Count() > 0 )
|
|
return;
|
|
|
|
char rgchQueryBuf[ 1024 ] = "SELECT * from knownmodules;";
|
|
IResultSet *results = g_pSqlWrapper->PResultSetQuery( rgchQueryBuf );
|
|
|
|
if ( !results )
|
|
{
|
|
ivgui()->DPrintf( "LoadKnownModules() results are NULL" );
|
|
VGUIMessageBox( GetParent(), "Error", "Unable to retrieve known modules from database" );
|
|
return;
|
|
}
|
|
|
|
for ( int i = 0; i < results->GetCSQLRow(); i++ )
|
|
{
|
|
module newModule;
|
|
const ISQLRow *row = results->PSQLRowNextResult();
|
|
Assert( row != NULL );
|
|
newModule.key = row->NData(0);
|
|
strcpy( newModule.name, row->PchData(1));
|
|
newModule.versionInfo.v1 = row->NData(2);
|
|
newModule.versionInfo.v2 = row->NData(3);
|
|
newModule.versionInfo.v3 = row->NData(4);
|
|
newModule.versionInfo.v4 = row->NData(5);
|
|
if ( !Q_strcmp( row->PchData(6), "GOOD" ) )
|
|
{
|
|
newModule.myType = GOOD;
|
|
}
|
|
else if ( !Q_strcmp( row->PchData(6), "BAD" ) )
|
|
{
|
|
newModule.myType = BAD;
|
|
}
|
|
else
|
|
{
|
|
newModule.myType = UNKNOWN;
|
|
}
|
|
m_knownModuleList.AddToTail( newModule );
|
|
}
|
|
|
|
g_pSqlWrapper->FreeResult();
|
|
}
|
|
|
|
|
|
void CMDModulePanel::InitializeDebugEngine( void )
|
|
{
|
|
// Start things off by getting an initial interface from
|
|
// the engine. This can be any engine interface but is
|
|
// generally IDebugClient as the client interface is
|
|
// where sessions are started.
|
|
if ( S_OK == DebugCreate( __uuidof ( IDebugClient ),
|
|
(void**)&m_pDbgClient ) )
|
|
{
|
|
m_pDbgClient->QueryInterface( __uuidof ( IDebugControl ),
|
|
( void** )&m_pDbgControl );
|
|
m_pDbgClient->QueryInterface( __uuidof ( IDebugSymbols2 ),
|
|
( void** )&m_pDbgSymbols );
|
|
|
|
// Set out Panel to receive the debug outputs from the engine
|
|
m_cDbgOutput.SetOutputPanel( GetVPanel() );
|
|
|
|
// Install output callbacks so we get any output that the
|
|
// later calls produce.
|
|
m_pDbgClient->SetOutputCallbacks(&m_cDbgOutput);
|
|
|
|
if ( m_pDbgSymbols )
|
|
{
|
|
// Make sure we have a symbol path to use
|
|
char szSymbolSrv[ 512 ] = { 0 };
|
|
ExpandEnvironmentStrings( "%_NT_SYMBOL_PATH%", szSymbolSrv, sizeof (szSymbolSrv) );
|
|
if ( !Q_stricmp( "%_NT_SYMBOL_PATH%", szSymbolSrv ) )
|
|
{
|
|
ivgui()->DPrintf( "Setting symbol server" );
|
|
Q_strcpy( szSymbolSrv, "SRV*c:\\localsymbols*\\\\perforce\\symbols*http://msdl.microsoft.com/download/symbols" );
|
|
m_pDbgSymbols->SetSymbolPath( szSymbolSrv );
|
|
}
|
|
m_pDbgSymbols->AddSymbolOptions(SYMOPT_LOAD_LINES);
|
|
}
|
|
}
|
|
}
|
|
|
|
void CMDModulePanel::ReleaseDebugEngine( void )
|
|
{
|
|
// Clean up any resources.
|
|
if ( m_pDbgSymbols != NULL )
|
|
{
|
|
m_pDbgSymbols->Release( );
|
|
}
|
|
|
|
if ( m_pDbgControl != NULL )
|
|
{
|
|
m_pDbgControl->Release( );
|
|
}
|
|
|
|
if ( m_pDbgClient != NULL )
|
|
{
|
|
// We don't want to see any output from the shutdown.
|
|
m_pDbgClient->SetOutputCallbacks( NULL );
|
|
|
|
m_pDbgClient->EndSession( DEBUG_END_PASSIVE );
|
|
|
|
m_pDbgClient->Release( );
|
|
}
|
|
}
|
|
|
|
|
|
void CMDModulePanel::AnalyzeDumpFile( const char *pszDumpFile )
|
|
{
|
|
if (m_pDbgClient && m_pDbgControl)
|
|
{
|
|
char szBuf[ 256 ] = { 0 };
|
|
Q_snprintf( szBuf, sizeof ( szBuf ), "About to open [%s].\n", pszDumpFile );
|
|
m_pAnalyzeText->InsertString( szBuf );
|
|
|
|
// Everything's set up so open the dump file.
|
|
m_pDbgClient->OpenDumpFile(pszDumpFile);
|
|
|
|
// Finish initialization by waiting for the event that
|
|
// caused the dump. This will return immediately as the
|
|
// dump file is considered to be at its event.
|
|
m_pDbgControl->WaitForEvent(DEBUG_WAIT_DEFAULT, INFINITE);
|
|
|
|
DWORD dwThreadId = 0;
|
|
m_hThread = CreateThread( NULL, 0, StaticAnalyzeThread, (LPVOID)this, 0, &dwThreadId );
|
|
}
|
|
}
|
|
|
|
|
|
DWORD CMDModulePanel::AnalyzeThread( void )
|
|
{
|
|
if ( m_pDbgControl )
|
|
{
|
|
// Tell the debug engine to analyze the current dump file
|
|
m_pDbgControl->Execute( DEBUG_OUTCTL_THIS_CLIENT,
|
|
"!analyze -v",
|
|
DEBUG_EXECUTE_DEFAULT);
|
|
}
|
|
|
|
CloseHandle( m_hThread );
|
|
m_pAnalyzeText->InsertString( "Finished analyzing minidump file.\n" );
|
|
|
|
return ( 0 );
|
|
}
|
|
|