1HEADmaster

1 files changed, 473 insertions, 0 deletions

diff --git a/utils/vmpi/vmpi_job_search/JobSearchDlg.cpp b/utils/vmpi/vmpi_job_search/JobSearchDlg.cpp
new file mode 100644
index 0000000..e98b726
--- /dev/null
+++ b/utils/vmpi/vmpi_job_search/JobSearchDlg.cpp
@@ -0,0 +1,473 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//
+//=============================================================================//
+// JobSearchDlg.cpp : implementation file
+//
+
+#include "stdafx.h"
+#include "JobSearchDlg.h"
+#include "imysqlwrapper.h"
+#include "tier1/strtools.h"
+#include "utllinkedlist.h"
+#include "vmpi_browser_helpers.h"
+#include "vmpi_defs.h"
+#include "net_view_thread.h"
+
+
+#ifdef _DEBUG
+#define new DEBUG_NEW
+#undef THIS_FILE
+static char THIS_FILE[] = __FILE__;
+#endif
+
+
+// These are stored with jobs to help with sorting and to remember the job ID.
+class CJobInfo
+{
+public:
+	unsigned long	m_JobID;
+	CString			m_StartTimeUnformatted;
+	CString			m_MachineName;
+	CString			m_BSPFilename;
+	DWORD			m_RunningTimeMS;
+};
+
+
+
+/////////////////////////////////////////////////////////////////////////////
+// CJobSearchDlg dialog
+
+
+CJobSearchDlg::CJobSearchDlg(CWnd* pParent /*=NULL*/)
+	: CDialog(CJobSearchDlg::IDD, pParent)
+{
+	//{{AFX_DATA_INIT(CJobSearchDlg)
+	//}}AFX_DATA_INIT
+	m_pSQL = NULL;
+	m_hMySQLDLL = NULL;
+}
+
+
+CJobSearchDlg::~CJobSearchDlg()
+{
+	if ( m_pSQL )
+	{
+		m_pSQL->Release();
+	}
+
+	if ( m_hMySQLDLL )
+	{
+		Sys_UnloadModule( m_hMySQLDLL );
+	}
+}
+
+
+void CJobSearchDlg::DoDataExchange(CDataExchange* pDX)
+{
+	CDialog::DoDataExchange(pDX);
+	//{{AFX_DATA_MAP(CJobSearchDlg)
+	DDX_Control(pDX, IDC_WORKER_LIST, m_WorkerList);
+	DDX_Control(pDX, IDC_USER_LIST, m_UserList);
+	DDX_Control(pDX, IDC_JOBS_LIST, m_JobsList);
+	//}}AFX_DATA_MAP
+}
+
+
+BEGIN_MESSAGE_MAP(CJobSearchDlg, CDialog)
+	//{{AFX_MSG_MAP(CJobSearchDlg)
+	ON_NOTIFY(NM_DBLCLK, IDC_JOBS_LIST, OnDblclkJobsList)
+	ON_LBN_DBLCLK(IDC_USER_LIST, OnDblclkUserList)
+	ON_LBN_DBLCLK(IDC_WORKER_LIST, OnDblclkWorkerList)
+	ON_BN_CLICKED(IDC_QUIT, OnQuit)
+	ON_WM_SIZE()
+	//}}AFX_MSG_MAP
+END_MESSAGE_MAP()
+
+
+int CJobSearchDlg::GetSelectedJobIndex()
+{
+	POSITION pos = m_JobsList.GetFirstSelectedItemPosition();
+	if ( pos )
+		return m_JobsList.GetNextSelectedItem( pos );
+	else
+		return -1;
+}
+
+
+/////////////////////////////////////////////////////////////////////////////
+// CJobSearchDlg message handlers
+
+void CJobSearchDlg::OnDblclkJobsList(NMHDR* pNMHDR, LRESULT* pResult) 
+{
+	int iItem = GetSelectedJobIndex();
+	if ( iItem != -1 )
+	{
+		CJobInfo *pInfo = (CJobInfo*)m_JobsList.GetItemData( iItem );
+
+		CString cmdLine;
+		cmdLine.Format( "vmpi_job_watch -JobID %d -dbname \"%s\" -hostname \"%s\" -username \"%s\"", 
+			pInfo->m_JobID, (const char*)m_DBName, (const char*)m_HostName, (const char*)m_UserName );
+
+		STARTUPINFO si;
+		memset( &si, 0, sizeof( si ) );
+		si.cb = sizeof( si );
+
+		PROCESS_INFORMATION pi;
+		memset( &pi, 0, sizeof( pi ) );
+
+		if ( !CreateProcess( 
+			NULL, 
+			(char*)(const char*)cmdLine, 
+			NULL,							// security
+			NULL,
+			TRUE,
+			0,			// flags
+			NULL,							// environment
+			NULL,							// current directory
+			&si,
+			&pi ) )
+		{
+			CString errStr;
+			errStr.Format( "Error launching '%s'", cmdLine.GetBuffer() );
+			MessageBox( errStr, "Error", MB_OK );
+		}
+	}
+	
+	*pResult = 0;
+}
+
+
+static int CALLBACK JobsSortFn( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam )
+{
+	CJobInfo *pInfo1 = (CJobInfo*)iItem1;
+	CJobInfo *pInfo2 = (CJobInfo*)iItem2;
+
+	return strcmp( pInfo2->m_StartTimeUnformatted, pInfo1->m_StartTimeUnformatted );
+}
+
+
+void CJobSearchDlg::ClearJobsList()
+{
+	// First, delete all the JobInfo structures we have in it.
+	int nItems = m_JobsList.GetItemCount();
+	for ( int i=0; i < nItems; i++ )
+	{
+		CJobInfo *pInfo = (CJobInfo*)m_JobsList.GetItemData( i );
+		delete pInfo;
+	}
+
+	m_JobsList.DeleteAllItems();
+}		   
+
+
+void CJobSearchDlg::RepopulateJobsList()
+{
+	// It's assumed coming into this routine that the caller just executed a query that we can iterate over.
+	ClearJobsList();
+
+	CUtlLinkedList<CJobInfo*, int> jobInfos;
+	while ( GetMySQL()->NextRow() )
+	{
+		CJobInfo *pInfo = new CJobInfo;
+		pInfo->m_StartTimeUnformatted = GetMySQL()->GetColumnValue( "StartTime" ).String();
+		pInfo->m_JobID = GetMySQL()->GetColumnValue( "JobID" ).Int32();
+		pInfo->m_MachineName = GetMySQL()->GetColumnValue( "MachineName" ).String();
+		pInfo->m_BSPFilename = GetMySQL()->GetColumnValue( "BSPFilename" ).String();
+		pInfo->m_RunningTimeMS = GetMySQL()->GetColumnValue( "RunningTimeMS" ).Int32();
+
+		jobInfos.AddToTail( pInfo );
+	}
+
+
+	FOR_EACH_LL( jobInfos, j )
+	{
+		CJobInfo *pInfo = jobInfos[j];
+
+		// Add the item.
+		int iItem = m_JobsList.InsertItem( 0, "", NULL );
+		
+		// Associate it with the job structure.
+		m_JobsList.SetItemData( iItem, (DWORD)pInfo );
+		
+		char dateStr[128];
+		const char *pDate = pInfo->m_StartTimeUnformatted;
+		if ( strlen( pDate ) == 14 ) // yyyymmddhhmmss
+		{
+			Q_snprintf( dateStr, sizeof( dateStr ), "%c%c/%c%c %c%c:%c%c:00", 
+				pDate[4], pDate[5],
+				pDate[6], pDate[7],
+				pDate[8], pDate[9],
+				pDate[10], pDate[11] );
+		}
+
+		m_JobsList.SetItemText( iItem, 0, dateStr );
+		m_JobsList.SetItemText( iItem, 1, pInfo->m_MachineName );
+		m_JobsList.SetItemText( iItem, 2, pInfo->m_BSPFilename );
+
+		char timeStr[512];
+		if ( pInfo->m_RunningTimeMS == RUNNINGTIME_MS_SENTINEL )
+		{
+			Q_strncpy( timeStr, "?", sizeof( timeStr ) );
+		}
+		else
+		{
+			FormatTimeString( pInfo->m_RunningTimeMS / 1000, timeStr, sizeof( timeStr ) );
+		}
+		m_JobsList.SetItemText( iItem, 3, timeStr );
+
+		char jobIDStr[512];
+		Q_snprintf( jobIDStr, sizeof( jobIDStr ), "%d", pInfo->m_JobID );
+		m_JobsList.SetItemText( iItem, 4, jobIDStr );
+	}
+	
+	m_JobsList.SortItems( JobsSortFn, (LPARAM)&m_JobsList );
+}
+
+
+void CJobSearchDlg::OnDblclkUserList() 
+{
+	int sel = m_UserList.GetCurSel();
+	if ( sel != LB_ERR )
+	{
+		CString computerName;
+		m_UserList.GetText( sel, computerName );
+
+		// Look for jobs that this user initiated.
+		char query[4096];
+		Q_snprintf( query, sizeof( query ), "select RunningTimeMS, JobID, BSPFilename, StartTime, MachineName from job_master_start where MachineName=\"%s\"", (const char*)computerName );
+		GetMySQL()->Execute( query );
+		
+		RepopulateJobsList();
+	}
+}
+
+void CJobSearchDlg::OnDblclkWorkerList() 
+{
+	int sel = m_WorkerList.GetCurSel();
+	if ( sel != LB_ERR )
+	{
+		CString computerName;
+		m_WorkerList.GetText( sel, computerName );
+
+		// This query does:
+		// 1. Take the workers with the specified MachineName.
+		// 2. Only use IsMaster = 0.
+		// 3. Now get all the job_master_start records with the same JobID.
+		char query[4096];
+		Q_snprintf( query, sizeof( query ), "select job_master_start.RunningTimeMS, job_master_start.JobID, job_master_start.BSPFilename, job_master_start.StartTime, job_master_start.MachineName "
+			"from job_master_start, job_worker_start "
+			"where job_worker_start.MachineName = \"%s\" and "
+			"IsMaster = 0 and "
+			"job_master_start.JobID = job_worker_start.JobID",
+			(const char*)computerName );
+		GetMySQL()->Execute( query );
+		
+		RepopulateJobsList();
+	}
+}
+
+bool ReadStringFromFile( FILE *fp, char *pStr, int strSize )
+{
+	int i=0;
+	for ( i; i < strSize-2; i++ )
+	{
+		if ( fread( &pStr[i], 1, 1, fp ) != 1 ||
+			pStr[i] == '\n' )
+		{
+			break;
+		}
+	}
+
+	pStr[i] = 0;
+	return i != 0;
+}
+
+const char* FindArg( const char *pArgName, const char *pDefault="" )
+{
+	for ( int i=1; i < __argc; i++ )
+	{
+		if ( Q_stricmp( pArgName, __argv[i] ) == 0 )
+		{
+			if ( (i+1) < __argc )
+				return __argv[i+1];
+			else
+				return pDefault;
+		}
+	}
+	return NULL;
+}
+
+BOOL CJobSearchDlg::OnInitDialog() 
+{
+	CDialog::OnInitDialog();
+
+	m_JobsList.SetExtendedStyle( LVS_EX_FULLROWSELECT );
+
+	char str[512];
+
+	// Init the mysql database.
+	const char *pDBName = FindArg( "-dbname", NULL );
+	const char *pHostName = FindArg( "-hostname", NULL );
+	const char *pUserName = FindArg( "-username", NULL );
+
+	if ( pDBName && pHostName && pUserName )
+	{
+		m_DBName = pDBName;
+		m_HostName = pHostName;
+		m_UserName = pUserName;
+	}
+	else
+	{
+		// Load the dbinfo_browser.txt file to get the database information.
+		const char *pFilename = FindArg( "-dbinfo", NULL );
+		if ( !pFilename )
+			pFilename = "dbinfo_job_search.txt";
+
+		FILE *fp = fopen( pFilename, "rt" );
+		if ( !fp )
+		{
+			Q_snprintf( str, sizeof( str ), "Can't open '%s' for database info.", pFilename );
+			MessageBox( str, "Error", MB_OK );
+			EndDialog( 0 );
+			return FALSE;
+		}
+
+		char hostName[512], dbName[512], userName[512];
+		if ( !ReadStringFromFile( fp, hostName, sizeof( hostName ) ) ||
+			 !ReadStringFromFile( fp, dbName, sizeof( dbName ) ) || 
+			 !ReadStringFromFile( fp, userName, sizeof( userName ) ) 
+			 )
+		{
+			fclose( fp );
+			Q_snprintf( str, sizeof( str ), "'%s' has invalid format.", pFilename );
+			MessageBox( str, "Error", MB_OK );
+			EndDialog( 0 );
+			return FALSE;
+		}
+
+		m_DBName = dbName;
+		m_HostName = hostName;
+		m_UserName = userName;
+
+		fclose( fp );		
+	}
+
+	// Get the mysql interface.
+	if ( !Sys_LoadInterface( "mysql_wrapper", MYSQL_WRAPPER_VERSION_NAME, &m_hMySQLDLL, (void**)&m_pSQL ) )
+		return false;
+	
+	if ( !m_pSQL->InitMySQL( m_DBName, m_HostName, m_UserName ) )
+	{
+		Q_snprintf( str, sizeof( str ), "Can't init MYSQL db (db = '%s', host = '%s', user = '%s')", (const char*)m_DBName, (const char*)m_HostName, (const char*)m_UserName );
+		MessageBox( str, "Error", MB_OK );
+		EndDialog( 0 );
+		return FALSE;
+	}
+
+	// Setup the headers for the job info list.
+	struct
+	{
+		char *pText;
+		int width;
+	} titles[] =
+	{
+		{"Date", 100},
+		{"User", 100},
+		{"BSP Filename", 100},
+		{"Running Time", 100},
+		{"Job ID", 100}
+	};
+	for ( int i=0; i < ARRAYSIZE( titles ); i++ )
+	{
+		m_JobsList.InsertColumn( i, titles[i].pText, LVCFMT_LEFT, titles[i].width, i );
+	}
+	
+	
+	CUtlVector<char*> computerNames;
+	CNetViewThread netView;
+	netView.Init();
+	DWORD startTime = GetTickCount();
+	while ( 1 )
+	{
+		netView.GetComputerNames( computerNames );
+		if ( computerNames.Count() > 0 )
+			break;
+
+		Sleep( 30 );
+		if ( GetTickCount() - startTime > 5000 )
+		{
+			Q_snprintf( str, sizeof( str ), "Unable to get computer names Can't init MYSQL db (db = '%s', host = '%s', user = '%s')", (const char*)m_DBName, (const char*)m_HostName, (const char*)m_UserName );
+			MessageBox( str, "Error", MB_OK );
+			EndDialog( 0 );
+			return FALSE;
+		}
+	}
+
+	PopulateWorkerList( computerNames );
+	PopulateUserList( computerNames );	
+
+
+	// Auto-select a worker?
+	const char *pSelectWorker = FindArg( "-SelectWorker", NULL );
+	if ( pSelectWorker )
+	{
+		int index = m_WorkerList.FindString( -1, pSelectWorker );
+		if ( index != LB_ERR )
+		{
+			m_WorkerList.SetCurSel( index );
+			OnDblclkWorkerList();
+		}
+	}
+
+
+	// Setup our anchors.
+	m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_SEARCH_BY_USER_PANEL ), ANCHOR_LEFT, ANCHOR_TOP, ANCHOR_WIDTH_PERCENT, ANCHOR_HEIGHT_PERCENT );
+	m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_USER_LIST ), ANCHOR_LEFT, ANCHOR_TOP, ANCHOR_WIDTH_PERCENT, ANCHOR_HEIGHT_PERCENT );
+
+	m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_WORKER_PANEL ), ANCHOR_WIDTH_PERCENT, ANCHOR_TOP, ANCHOR_RIGHT, ANCHOR_HEIGHT_PERCENT );
+	m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_WORKER_LIST ), ANCHOR_WIDTH_PERCENT, ANCHOR_TOP, ANCHOR_RIGHT, ANCHOR_HEIGHT_PERCENT );
+
+	m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_JOBS_PANEL ), ANCHOR_LEFT, ANCHOR_HEIGHT_PERCENT, ANCHOR_RIGHT, ANCHOR_BOTTOM );
+	m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_JOBS_LIST ), ANCHOR_LEFT, ANCHOR_HEIGHT_PERCENT, ANCHOR_RIGHT, ANCHOR_BOTTOM );
+
+	m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_QUIT ), ANCHOR_WIDTH_PERCENT, ANCHOR_BOTTOM, ANCHOR_WIDTH_PERCENT, ANCHOR_BOTTOM );
+
+	return TRUE;  // return TRUE unless you set the focus to a control
+	              // EXCEPTION: OCX Property Pages should return FALSE
+}
+
+void CJobSearchDlg::PopulateWorkerList( CUtlVector<char*> &computerNames )
+{
+	m_WorkerList.ResetContent();
+	for ( int i=0; i < computerNames.Count(); i++ )
+	{
+		m_WorkerList.AddString( computerNames[i] );
+	}
+}
+
+void CJobSearchDlg::PopulateUserList( CUtlVector<char*> &computerNames )
+{
+	m_UserList.ResetContent();
+	for ( int i=0; i < computerNames.Count(); i++ )
+	{
+		m_UserList.AddString( computerNames[i] );
+	}
+}
+
+
+
+void CJobSearchDlg::OnQuit() 
+{
+	EndDialog( 0 );	
+}
+
+void CJobSearchDlg::OnSize(UINT nType, int cx, int cy) 
+{
+	CDialog::OnSize(nType, cx, cy);
+	
+	m_AnchorMgr.UpdateAnchors( this );	
+}