1HEADmaster

1 files changed, 460 insertions, 0 deletions

diff --git a/utils/gamestats_reader/gamestats_reader.cpp b/utils/gamestats_reader/gamestats_reader.cpp
new file mode 100644
index 0000000..038ae58
--- /dev/null
+++ b/utils/gamestats_reader/gamestats_reader.cpp
@@ -0,0 +1,460 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//
+//===========================================================================//
+#include <stdio.h>
+#include <stdlib.h>
+#include <direct.h>
+#include "tier1/strtools.h"
+#include <conio.h>
+#include "tier1/utlbuffer.h"
+#include "tier2/tier2.h"
+#include "filesystem.h"
+#include "imysqlwrapper.h"
+#include "../../game/server/dod/dod_gamestats.h"
+#include <time.h>
+
+
+static void Pause( void )
+{
+	printf( "Hit a key to continue\n" );
+	getch();
+}
+
+static void Usage()
+{
+	fprintf( stderr, "Usage: gamestats_reader <hostname> <database> <username> <password> <table> <directory to parse> ( -verbose )\n" );
+	Pause();
+	exit( -1 );
+}
+
+#define SQL_CMD_BUFSIZE		16000
+
+char sqlCmd[SQL_CMD_BUFSIZE];
+bool g_bFirstCmd;
+bool g_bVerbose;
+
+IMySQL *mysql;
+
+char **g_argv;
+
+void StartMYSQLInsert( void )
+{
+	g_bFirstCmd = true;
+	sqlCmd[0] = '\0';
+
+	Q_snprintf( sqlCmd, SQL_CMD_BUFSIZE, "INSERT INTO %s SET ", g_argv[5] );
+}
+
+void AddField( const char *field, const char *value )
+{	
+	char buf[128];
+
+	if ( !g_bFirstCmd )
+	{
+		Q_strncat( sqlCmd, ", ", SQL_CMD_BUFSIZE, COPY_ALL_CHARACTERS );
+	}
+
+	Q_snprintf( buf, sizeof(buf), "%s=\"%s\"", field, value );
+	Q_strncat( sqlCmd, buf, SQL_CMD_BUFSIZE, COPY_ALL_CHARACTERS );
+
+	g_bFirstCmd = false;
+}
+
+void AddField( const char *field, const int value )
+{	
+	char buf[128];
+	
+	if ( !g_bFirstCmd )
+	{
+		Q_strncat( sqlCmd, ", ", SQL_CMD_BUFSIZE, COPY_ALL_CHARACTERS );
+	}
+
+	Q_snprintf( buf, sizeof(buf), "%s=%d", field, value );
+
+#ifdef _DEBUG
+	if ( Q_strlen(buf) + Q_strlen(sqlCmd) > SQL_CMD_BUFSIZE )
+	{
+		Assert( !"increase buf size\n" );
+	}
+#endif
+
+	Q_strncat( sqlCmd, buf, SQL_CMD_BUFSIZE, COPY_ALL_CHARACTERS );
+
+	g_bFirstCmd = false;
+}
+
+int CompleteMYSQLInsert( void )
+{
+	// terminate command and execute it
+	Q_strncat( sqlCmd, "\n", SQL_CMD_BUFSIZE, COPY_ALL_CHARACTERS );
+
+	if ( g_bVerbose )
+	{
+		printf( "%s", sqlCmd );
+	}
+
+	int retcode = mysql->Execute( sqlCmd );
+
+	if ( retcode != 0 )
+	{
+		const char *asdf = mysql->GetLastError();
+		printf( "Error: %s\n", asdf );
+	}
+
+	return retcode;
+}
+
+static const char *pszTeamNames[] = 
+{
+	"allies",
+	"axis"
+};
+
+static const char *pszClassNames[] = 
+{
+	"rifleman",
+	"assault",
+	"support",
+	"sniper",
+	"mg",
+	"rocket"
+};
+
+int iDistanceStatWeapons[DOD_NUM_DISTANCE_STAT_WEAPONS] = 
+{
+	WEAPON_COLT,
+	WEAPON_P38,
+	WEAPON_C96,
+	WEAPON_GARAND,
+	WEAPON_GARAND_ZOOMED,
+	WEAPON_M1CARBINE,
+	WEAPON_K98,
+	WEAPON_K98_ZOOMED,
+	WEAPON_SPRING,
+	WEAPON_SPRING_ZOOMED,
+	WEAPON_K98_SCOPED,
+	WEAPON_K98_SCOPED_ZOOMED,
+	WEAPON_THOMPSON,
+	WEAPON_MP40,
+	WEAPON_MP44,
+	WEAPON_MP44_SEMIAUTO,
+	WEAPON_BAR,
+	WEAPON_BAR_SEMIAUTO,
+	WEAPON_30CAL,
+	WEAPON_30CAL_UNDEPLOYED,
+	WEAPON_MG42,
+	WEAPON_MG42_UNDEPLOYED,
+};
+
+// Send hit/shots only for the following weapons
+int iNoDistStatWeapons[DOD_NUM_NODIST_STAT_WEAPONS] =
+{
+	WEAPON_AMERKNIFE,
+	WEAPON_SPADE,
+	WEAPON_BAZOOKA,
+	WEAPON_PSCHRECK,
+	WEAPON_FRAG_US,
+	WEAPON_FRAG_GER,
+	WEAPON_FRAG_US_LIVE,
+	WEAPON_FRAG_GER_LIVE,
+	WEAPON_RIFLEGREN_US,
+	WEAPON_RIFLEGREN_GER,
+	WEAPON_RIFLEGREN_US_LIVE,
+	WEAPON_RIFLEGREN_GER_LIVE,
+	WEAPON_THOMPSON_PUNCH,
+	WEAPON_MP40_PUNCH,
+};
+
+const char * s_WeaponAliasInfo[] = 
+{
+	"none",	//	WEAPON_NONE = 0,
+
+	//Melee
+	"amerknife",	//WEAPON_AMERKNIFE,
+	"spade",		//WEAPON_SPADE,
+
+	//Pistols
+	"colt",			//WEAPON_COLT,
+	"p38",			//WEAPON_P38,
+	"c96",			//WEAPON_C96
+
+	//Rifles
+	"garand",		//WEAPON_GARAND,
+	"m1carbine",	//WEAPON_M1CARBINE,
+	"k98",			//WEAPON_K98,
+
+	//Sniper Rifles
+	"spring",		//WEAPON_SPRING,
+	"k98_scoped",	//WEAPON_K98_SCOPED,
+
+	//SMG
+	"thompson",		//WEAPON_THOMPSON,
+	"mp40",			//WEAPON_MP40,
+	"mp44",			//WEAPON_MP44,
+	"bar",			//WEAPON_BAR,
+
+	//Machine guns
+	"30cal",		//WEAPON_30CAL,
+	"mg42",			//WEAPON_MG42,
+
+	//Rocket weapons
+	"bazooka",		//WEAPON_BAZOOKA,
+	"pschreck",		//WEAPON_PSCHRECK,
+
+	//Grenades
+	"frag_us",		//WEAPON_FRAG_US,
+	"frag_ger",		//WEAPON_FRAG_GER,
+
+	"frag_us_live",		//WEAPON_FRAG_US_LIVE
+	"frag_ger_live",	//WEAPON_FRAG_GER_LIVE
+
+	"smoke_us",			//WEAPON_SMOKE_US
+	"smoke_ger",		//WEAPON_SMOKE_GER
+
+	"riflegren_us",	//WEAPON_RIFLEGREN_US
+	"riflegren_ger",	//WEAPON_RIFLEGREN_GER
+
+	"riflegren_us_live",	//WEAPON_RIFLEGREN_US_LIVE
+	"riflegren_ger_live",		//WEAPON_RIFLEGREN_GER_LIVE
+
+	// not actually separate weapons, but defines used in stats recording
+	"thompson_punch",		//WEAPON_THOMPSON_PUNCH
+	"mp40_punch",			//WEAPON_MP40_PUNCH
+	"garand_zoomed",		//WEAPON_GARAND_ZOOMED,	
+
+	"k98_zoomed",			//WEAPON_K98_ZOOMED
+	"spring_zoomed",		//WEAPON_SPRING_ZOOMED
+	"k98_scoped_zoomed",	//WEAPON_K98_SCOPED_ZOOMED	
+
+	"30cal_undeployed",		//WEAPON_30CAL_UNDEPLOYED,
+	"mg42_undeployed",		//WEAPON_MG42_UNDEPLOYED,
+
+	"bar_semiauto",			//WEAPON_BAR_SEMIAUTO,
+	"mp44_semiauto",		//WEAPON_MP44_SEMIAUTO,
+
+	0,		// end of list marker
+};
+
+void ParseFile( const char *fileName )
+{
+	FileHandle_t file = g_pFullFileSystem->Open( fileName, "rb" );
+
+	if ( !file )
+	{
+		return;
+	}
+
+	dod_gamestats_t stats;
+	g_pFullFileSystem->Read( &stats, sizeof( dod_gamestats_t ), file );
+
+	if ( stats.header.iVersion != DOD_STATS_BLOB_VERSION || Q_stricmp( stats.header.szGameName, "dod" ) )
+	{
+		printf( "Error parsing file, bad header info: %s\n", fileName );
+		return;
+	}
+
+	StartMYSQLInsert();
+
+	AddField( "map", stats.header.szMapName );
+
+	const time_t mapfiletime = g_pFullFileSystem->GetFileTime( fileName );
+
+	struct tm *t = localtime( &mapfiletime );
+
+	// YYYY-MM-DD HH:MM::SS 
+
+	char filetimebuf[64];
+
+	Q_snprintf( filetimebuf, sizeof(filetimebuf), "%04d-%02d-%02d %02d:%02d:%02d", 
+		t->tm_year + 1900,
+		t->tm_mon + 1,
+		t->tm_mday,
+		t->tm_hour,
+		t->tm_min,
+		t->tm_sec );
+
+	AddField( "time", filetimebuf );
+
+	AddField( "version", stats.header.iVersion );
+
+	AddField( "ipaddr_0", stats.header.ipAddr[0] );
+	AddField( "ipaddr_1", stats.header.ipAddr[1] );
+	AddField( "ipaddr_2", stats.header.ipAddr[2] );
+	AddField( "ipaddr_3", stats.header.ipAddr[3] );
+	AddField( "port", stats.header.port );
+
+	AddField( "minutes_map", stats.iMinutesPlayed );
+	AddField( "wins_allies", stats.iNumAlliesWins );
+	AddField( "wins_axis", stats.iNumAxisWins);
+	AddField( "tickpoints_allies", stats.iAlliesTickPoints );
+	AddField( "tickpoints_axis",  stats.iAxisTickPoints );
+
+	char buf[128];
+
+	for ( int cls=0;cls<6;cls++ )
+	{
+		Q_snprintf( buf, sizeof(buf), "minutes_allies_%s", pszClassNames[cls] );
+		AddField( buf, stats.iMinutesPlayedPerClass_Allies[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "kills_allies_%s", pszClassNames[cls] );
+		AddField( buf, stats.iKillsPerClass_Allies[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "defenses_allies_%s", pszClassNames[cls] );
+		AddField( buf, stats.iDefensesPerClass_Allies[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "caps_allies_%s", pszClassNames[cls] );
+		AddField( buf, stats.iCapsPerClass_Allies[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "spawns_allies_%s", pszClassNames[cls] );
+		AddField( buf, stats.iSpawnsPerClass_Allies[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "classlimit_allies_%s", pszClassNames[cls] );
+		AddField( buf, stats.iClassLimits_Allies[cls] );
+
+
+		Q_snprintf( buf, sizeof(buf), "minutes_axis_%s", pszClassNames[cls] );
+		AddField( buf, stats.iMinutesPlayedPerClass_Axis[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "kills_axis_%s", pszClassNames[cls] );
+		AddField( buf, stats.iKillsPerClass_Axis[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "defenses_axis_%s", pszClassNames[cls] );
+		AddField( buf, stats.iDefensesPerClass_Axis[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "caps_axis_%s", pszClassNames[cls] );
+		AddField( buf, stats.iCapsPerClass_Axis[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "spawns_axis_%s", pszClassNames[cls] );
+		AddField( buf, stats.iSpawnsPerClass_Axis[cls] );
+
+		Q_snprintf( buf, sizeof(buf), "classlimit_axis_%s", pszClassNames[cls] );
+		AddField( buf, stats.iClassLimits_Axis[cls] );
+	}
+
+	int i;
+
+	for ( i=0;i<DOD_NUM_DISTANCE_STAT_WEAPONS;i++ )
+	{
+		int iWeapon = iDistanceStatWeapons[i];
+		const char *pszWeapon = s_WeaponAliasInfo[iWeapon];
+
+		Q_snprintf( buf, sizeof(buf), "weapon_shots_%s", pszWeapon );
+		AddField( buf, stats.weaponStatsDistance[i].iNumAttacks );
+
+		Q_snprintf( buf, sizeof(buf), "weapon_hits_%s", pszWeapon );
+		AddField( buf, stats.weaponStatsDistance[i].iNumHits );
+
+		for ( int j=0;j<DOD_NUM_WEAPON_DISTANCE_BUCKETS;j++ )
+		{
+			Q_snprintf( buf, sizeof(buf), "weapon_distance_%s_%d", pszWeapon, j );
+			AddField( buf, stats.weaponStatsDistance[i].iDistanceBuckets[j] );
+		}
+	}
+
+	for ( i=0;i<DOD_NUM_NODIST_STAT_WEAPONS;i++ )
+	{
+		int iWeapon = iNoDistStatWeapons[i];
+		const char *pszWeapon = s_WeaponAliasInfo[iWeapon];
+
+		Q_snprintf( buf, sizeof(buf), "weapon_shots_%s", pszWeapon );
+		AddField( buf, stats.weaponStats[i].iNumAttacks );
+
+		Q_snprintf( buf, sizeof(buf), "weapon_hits_%s", pszWeapon );
+		AddField( buf, stats.weaponStats[i].iNumHits );
+	}
+
+	CompleteMYSQLInsert();
+
+	g_pFullFileSystem->Close( file );
+}
+
+int main( int argc, char **argv )
+{
+	g_argv = argv;
+
+	if( argc < 6 )
+	{
+		Usage();
+	}
+
+	if ( argc == 7 && !Q_stricmp( argv[6], "-verbose" ) )
+	{
+		g_bVerbose = true;
+	}
+
+	InitDefaultFileSystem();
+
+	// Init MYSQL connection
+	CSysModule *sql = Sys_LoadModule( "mysql_wrapper" );
+	if ( sql )
+	{
+		CreateInterfaceFn factory = Sys_GetFactory( sql );
+		if ( factory )
+		{
+			mysql = ( IMySQL * )factory( MYSQL_WRAPPER_VERSION_NAME, NULL );
+			if ( mysql )
+			{
+				if ( mysql->InitMySQL( argv[ 2 ], argv[ 1 ], argv[ 3 ], argv[ 4 ] ) )
+				{
+					// insert rows
+
+					const char *dir = argv[6];
+
+					char searchString[MAX_PATH*2];
+					Q_strncpy( searchString, dir, sizeof( searchString ) );
+					Q_AppendSlash( searchString, sizeof( searchString ) );
+					Q_strncat( searchString, "*.dat", sizeof( searchString ), COPY_ALL_CHARACTERS );
+
+					int iNumFiles = 0;
+					FileFindHandle_t findHandle = NULL;
+					const char *filename = g_pFullFileSystem->FindFirst( searchString, &findHandle );
+					while ( filename )
+					{
+						char fullFileName[MAX_PATH*2];
+
+						Q_strncpy( fullFileName, dir, sizeof( fullFileName ) );
+						Q_AppendSlash( fullFileName, sizeof( fullFileName ) );
+						Q_strncat( fullFileName, filename, sizeof( fullFileName ), COPY_ALL_CHARACTERS );
+
+						ParseFile( fullFileName );
+
+						printf( "processing file: %s\n", fullFileName );
+						iNumFiles++;
+
+						filename = g_pFullFileSystem->FindNext(findHandle);
+					}
+					g_pFullFileSystem->FindClose(findHandle);	
+
+					printf( "Completed: %d files processed from directory \"%s\"\n", iNumFiles, dir );
+
+				}
+				else
+				{
+					printf( "InitMySQL failed ( %s )\n", mysql->GetLastError() );
+				}
+
+				mysql->Release();
+			}
+			else
+			{
+				printf( "Unable to connect via mysql_wrapper\n");
+			}
+		}
+		else
+		{
+			printf( "Unable to get factory from mysql_wrapper.dll, not updating access mysql table!!!" );
+		}
+
+		Sys_UnloadModule( sql );
+	}
+	else
+	{
+		printf( "Unable to load mysql_wrapper.dll, not updating access mysql table!!!" );
+	}
+
+	return 0;
+}