You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
460 lines
11 KiB
460 lines
11 KiB
//========= 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;
|
|
}
|