Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
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.

461 lines
11 KiB

5 years ago
//========= 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;
}