Update from SDK 2013

2025-09-20 12:36:05 +08:00 · 2025-02-19 18:39:00 -05:00
parent 6d5c024820
commit 94b660e16e
7474 changed files with 2597282 additions and 1254065 deletions
--- a/utils/vmpi/vmpi_registry_query_thread.cpp
+++ b/utils/vmpi/vmpi_registry_query_thread.cpp
@ -0,0 +1,139 @@
+//========= Copyright <20> 1996-2005, Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+//=============================================================================//
+
+#include "stdafx.h"
+#include "vmpi_registry_query_thread.h"
+#include "tier1/utlstring.h"
+#include "curl/curl.h"
+#include "tier1/fmtstr.h"
+#include "tier1/utlbuffer.h"
+#include "vmpi_defs.h"
+#include "vmpi.h"
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include "tier0/memdbgon.h"
+
+CVMPIRegistryQueryThread::CVMPIRegistryQueryThread()
+{
+	m_hThread = NULL;
+	m_hThreadExitEvent = NULL;
+	InitializeCriticalSection( &m_ComputerNamesCS );
+}
+
+
+CVMPIRegistryQueryThread::~CVMPIRegistryQueryThread()
+{
+	Term();
+	DeleteCriticalSection( &m_ComputerNamesCS );
+}
+
+
+void CVMPIRegistryQueryThread::Init()
+{
+	Term();
+
+	m_hThreadExitEvent = CreateEvent( NULL, false, false, NULL );
+
+	DWORD dwThreadID = 0;
+	m_hThread = CreateThread(
+		NULL,
+		0,
+		&CVMPIRegistryQueryThread::StaticThreadFn,
+		this,
+		0,
+		&dwThreadID );
+}
+
+
+void CVMPIRegistryQueryThread::Term()
+{
+	if ( m_hThread )
+	{
+		SetEvent( m_hThreadExitEvent );
+		WaitForSingleObject( m_hThread, INFINITE );
+		CloseHandle( m_hThread );
+		m_hThread = NULL;
+	}
+
+	if ( m_hThreadExitEvent )
+	{
+		CloseHandle( m_hThreadExitEvent );
+		m_hThreadExitEvent = NULL;
+	}
+
+	m_RegisteredServices.Purge();
+}
+
+void CVMPIRegistryQueryThread::GetRegisteredServices( CUtlVector<CServiceInfo> &services )
+{
+	EnterCriticalSection( &m_ComputerNamesCS );
+
+	services.Purge();
+	services.AddVectorToTail( m_RegisteredServices );
+	m_RegisteredServices.RemoveAll();
+
+	LeaveCriticalSection( &m_ComputerNamesCS );
+}
+
+
+static size_t CURLDataReceivedCallback( void *buffer, size_t size, size_t nmemb, void *userp )
+{
+	CUtlBuffer *pResponseBuf = reinterpret_cast<CUtlBuffer*>( userp );
+	pResponseBuf->Put( buffer, int( size * nmemb ) );
+	return size * nmemb;
+}
+
+void CVMPIRegistryQueryThread::UpdateServicesFromRegistry()
+{
+	CUtlVector<VMPIWorkerInfo_t> registeredWorkers;
+	// MEGA-HACK: There's an inconsistency between the VMPI_ func below using dlmalloc to alloc memory for registeredWorkers, but this function will call the
+	// std library free() to free it, causing a mismatch and crash. Here I'm pre-allocating "enough" space to avoid VMPI_QueryRegistryForWorkers() doing any allocations on the vector.
+	registeredWorkers.EnsureCapacity( 10000 );
+														
+	VMPI_QueryRegistryForWorkers( registeredWorkers );
+
+	// Convert results
+	EnterCriticalSection( &m_ComputerNamesCS );
+	m_RegisteredServices.RemoveAll();
+	for ( int i = 0; i < registeredWorkers.Count(); i++ )
+	{
+		CServiceInfo si;
+		si.m_Addr = registeredWorkers[ i ].addr;
+		si.m_ComputerName.SetString( registeredWorkers[i].machineName );
+		si.m_iState = registeredWorkers[ i ].nStatus;
+		si.m_LiveTimeMS = 0;
+		si.m_WorkerAppTimeMS = 0;
+		si.m_LastPingTimeMS = Plat_MSTime();
+		si.m_pLastStatusText = "";
+		si.m_LastLiveTimeMS = Plat_MSTime();
+		si.m_CPUPercentage = -1;
+		si.m_MemUsageMB = -1;
+		si.m_LastUpdateTime = 0;
+		si.m_ProtocolVersion = registeredWorkers[ i ].nProtocolVersion;
+		si.m_bFromRegistry = true;
+		V_strcpy_safe( si.m_ServiceVersion, registeredWorkers[i].vmpiVersion );
+		m_RegisteredServices.AddToTail( si );
+	}
+	LeaveCriticalSection( &m_ComputerNamesCS );
+}
+
+
+DWORD CVMPIRegistryQueryThread::ThreadFn()
+{
+	// Update the services list every 30 seconds.
+	do
+	{
+		UpdateServicesFromRegistry();
+	} while ( WaitForSingleObject( m_hThreadExitEvent, 30000 ) != WAIT_OBJECT_0 );
+
+	return 0;
+}
+
+
+DWORD CVMPIRegistryQueryThread::StaticThreadFn( LPVOID lpParameter )
+{
+	return ((CVMPIRegistryQueryThread*)lpParameter)->ThreadFn();
+}