Counter Strike : Global Offensive Source Code
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.
 
 
 
 
 
 

795 lines
24 KiB

//========= Copyright © 1996-2003, Valve LLC, All rights reserved. ============
//
// Purpose:
//
//=============================================================================
#include "cbase.h"
#include "tier1/keyvalues.h"
#include "econ_gcmessages.h"
#include "econ_item_system.h"
#include "econ_item_inventory.h"
#include "game_item_schema.h"
#include "utldict.h"
#include "filesystem.h"
#include "steam/isteamhttp.h"
#if defined(CLIENT_DLL) || defined(GAME_DLL)
#include "ihasattributes.h"
#include "tier0/icommandline.h"
#endif
#if defined(CLIENT_DLL)
#include "igameevents.h"
#endif
// FIXME FIXME FIXME
#if defined(TF_DLL) || defined(TF_CLIENT_DLL)
#include "tf_item_system.h"
#endif // defined(TF_DLL) || defined(TF_CLIENT_DLL)
#if defined (DOTA_CLIENT_DLL) || defined (DOTA_DLL)
#include "econ/dota_item_system.h"
#endif // defined (DOTA_CLIENT_DLL) || defined (DOTA_DLL)
#if defined( CSTRIKE_CLIENT_DLL ) || defined( CSTRIKE15 )
#include "cstrike15_item_system.h"
#endif
// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"
#if (defined(GAME_DLL) || defined(CLIENT_DLL)) && defined(_DEBUG)
ConVar item_debug( "item_debug", "0", FCVAR_REPLICATED | FCVAR_DEVELOPMENTONLY );
ConVar item_debug_validation( "item_debug_validation", "1", FCVAR_REPLICATED | FCVAR_ARCHIVE, "If set, CEconEntity::ValidateEntityAttachedToPlayer behaves as it would in release builds and also allows bot players to take the same code path as real players." );
static ConVar item_quality_chance_unique( "item_quality_chance_unique", "0.1", FCVAR_REPLICATED | FCVAR_DEVELOPMENTONLY, "Percentage chance that a random item is unique." );
static ConVar item_quality_chance_rare( "item_quality_chance_rare", "0.5", FCVAR_REPLICATED | FCVAR_DEVELOPMENTONLY, "Percentage chance that a random item is a rare." );
static ConVar item_quality_chance_common( "item_quality_chance_common", "1.0", FCVAR_REPLICATED | FCVAR_DEVELOPMENTONLY, "Percentage chance that a random item is common." );
#endif
//-----------------------------------------------------------------------------
// Purpose: Get at the global item system
//-----------------------------------------------------------------------------
CEconItemSystem *ItemSystem( void )
{
static GameItemSystem_t *pSystem = NULL;
if ( !pSystem )
{
pSystem = new GameItemSystem_t();
}
return pSystem;
}
//-----------------------------------------------------------------------------
// Purpose: Global schema access, declared in game_item_schema.h
//-----------------------------------------------------------------------------
GameItemSchema_t *GetItemSchema()
{
return ItemSystem()->GetItemSchema();
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
CEconItemSystem::CEconItemSystem( void )
{
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
CEconItemSystem::~CEconItemSystem( void )
{
Shutdown();
}
//-----------------------------------------------------------------------------
// Purpose: Parse in our data files.
//-----------------------------------------------------------------------------
void CEconItemSystem::Init( void )
{
#ifdef CLIENT_DLL
IGameEvent *event = gameeventmanager->CreateEvent( "item_schema_initialized" );
if ( event )
{
gameeventmanager->FireEventClientSide( event );
}
#endif
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
void CEconItemSystem::Shutdown( void )
{
}
extern ConVar mp_tournament;
#ifdef GAME_DLL
ConVar mp_tournament_whitelist( "mp_tournament_whitelist", "item_whitelist.txt", FCVAR_NONE, "Specifies the item whitelist file to use." );
#endif
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
void CEconItemSystem::ReloadWhitelist( void )
{
// Default state of items depends on whether we're in tourney mode, and whether there's a whitelist
bool bDefault = true;
bool bFoundWhitelist = false;
KeyValues::AutoDelete pWhitelistKV( "item_whitelist" );
#ifdef GAME_DLL
if ( mp_tournament.GetBool() && mp_tournament_whitelist.GetString() )
{
const char *pszWhitelistFile = mp_tournament_whitelist.GetString();
if ( pWhitelistKV->LoadFromFile( filesystem, pszWhitelistFile ) )
{
// Allow the whitelist to override the default, so they can turn it into a blacklist if they want to
bDefault = pWhitelistKV->GetBool( "unlisted_items_default_to" );
bFoundWhitelist = true;
}
else if ( pszWhitelistFile && pszWhitelistFile[0] )
{
Msg("Item Whitelist file '%s' could not be found. All items will be allowed.\n", pszWhitelistFile );
}
}
#endif
const CEconItemSchema::ItemDefinitionMap_t& mapItemDefs = m_itemSchema.GetItemDefinitionMap();
FOR_EACH_MAP_FAST( mapItemDefs, i )
{
mapItemDefs[i]->SetAllowedInMatch( bDefault );
}
// If we didn't find a file, we're done.
if ( !bFoundWhitelist )
return;
// Otherwise, go through the KVs and turn on the matching items.
Msg("Parsing item whitelist (default: %s)\n", bDefault ? "allowed" : "disallowed" );
for ( KeyValues *pKey = pWhitelistKV->GetFirstSubKey(); pKey != NULL; pKey = pKey->GetNextKey() )
{
bool bAllow = pKey->GetBool();
const char *pszItemName = pKey->GetName();
if ( pszItemName && pszItemName[0] && !FStrEq("unlisted_items_default_to", pszItemName) )
{
CEconItemDefinition *pItemDef = m_itemSchema.GetItemDefinitionByName( pszItemName );
if ( pItemDef )
{
pItemDef->SetAllowedInMatch( bAllow );
Msg(" -> %s '%s'\n", bAllow ? "Allowing" : "Removing", pszItemName );
}
else
{
Warning(" -> Could not find an item definition named '%s'\n", pszItemName );
}
}
}
Msg("Finished.\n");
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
void CEconItemSystem::ResetAttribStringCache( void )
{
CEconItemSchema::EconAttrDefsContainer_t &mapDefs = m_itemSchema.GetAttributeDefinitionContainer();
FOR_EACH_VEC( mapDefs, i )
{
if ( mapDefs[i] )
mapDefs[i]->ClearStringCache();
}
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
bool CEconItemSystem::DecryptItemFiles( KeyValues *pKV, const char *pName )
{
char szFullName[512];
Q_snprintf(szFullName,sizeof(szFullName), "%s.ctx", pName );
FileHandle_t f = filesystem->Open( szFullName, "rb", "MOD" );
if (!f)
{
#if !defined(CSTRIKE_DLL)
Warning("No %s file found. May be unable to create items.\n", pName );
#endif // CSTRIKE_DLL
return false;
}
int fileSize = filesystem->Size(f);
char *buffer = (char*)MemAllocScratch(fileSize + 1);
Assert(buffer);
filesystem->Read(buffer, fileSize, f); // read into local buffer
buffer[fileSize] = 0; // null terminate file as EOF
filesystem->Close( f ); // close file after reading
UTIL_DecodeICE( (unsigned char*)buffer, fileSize, GetEncryptionKey() );
bool retOK = pKV->LoadFromBuffer( szFullName, buffer, filesystem );
MemFreeScratch();
if ( !retOK )
return false;
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Read the specified item schema file. Init the item schema with the contents
//-----------------------------------------------------------------------------
void CEconItemSystem::ParseItemSchemaFile( const char *pFilename )
{
CUtlVector< CUtlString > vecErrors;
bool bSuccess = m_itemSchema.BInit( pFilename, "MOD", &vecErrors );
if( !bSuccess )
{
FOR_EACH_VEC( vecErrors, nError )
{
Msg( "%s", vecErrors[nError].String() );
}
}
}
//-----------------------------------------------------------------------------
// Purpose: Generate a random item matching the specified criteria
//-----------------------------------------------------------------------------
int CEconItemSystem::GenerateRandomItem( CItemSelectionCriteria *pCriteria, entityquality_t *outEntityQuality )
{
// First, pick a random item quality (use the one passed in first)
if ( !pCriteria->BQualitySet() )
{
pCriteria->SetQuality( GetRandomQualityForItem() );
}
pCriteria->SetIgnoreEnabledFlag( true );
// Determine which item templates match the criteria
CUtlVector<int> vecMatches;
const CEconItemSchema::ItemDefinitionMap_t &mapDefs = m_itemSchema.GetItemDefinitionMap();
HackMakeValidList:
FOR_EACH_MAP_FAST( mapDefs, i )
{
if ( pCriteria->BEvaluate( mapDefs[i], m_itemSchema ) )
{
vecMatches.AddToTail( mapDefs.Key( i ) );
}
}
// No valid items?
int iValidItems = vecMatches.Count();
if ( !iValidItems )
{
// If we were searching for a unique item, drop back to a non-unique
if ( pCriteria->GetQuality() == AE_UNIQUE )
{
pCriteria->SetQuality( GetRandomQualityForItem( true ) );
goto HackMakeValidList;
}
return INVALID_ITEM_INDEX;
}
// Choose a random match
int iChosenIdx = RandomInt( 0, (iValidItems-1) );
int iChosenItem = vecMatches[iChosenIdx];
const CEconItemDefinition *pItemDef = m_itemSchema.GetItemDefinition( iChosenItem );
if ( !pItemDef )
return INVALID_ITEM_INDEX;
// If we haven't specified an entity quality, we want to use the item's specified one
if ( pCriteria->GetQuality() == AE_USE_SCRIPT_VALUE )
{
int32 iScriptQuality = pItemDef->GetQuality();
pCriteria->SetQuality( iScriptQuality == AE_UNDEFINED ? GetRandomQualityForItem( true ) : iScriptQuality );
}
// If we haven't specified an item level, we want to use the item's specified one.
if ( !pCriteria->BItemLevelSet() )
{
pCriteria->SetItemLevel( RandomInt( pItemDef->GetMinLevel(), pItemDef->GetMaxLevel() ) );
}
if ( outEntityQuality )
{
*outEntityQuality = pCriteria->GetQuality();
}
return iChosenItem;
}
//-----------------------------------------------------------------------------
// Purpose: Return a random quality for the item specified
//-----------------------------------------------------------------------------
entityquality_t CEconItemSystem::GetRandomQualityForItem( bool bPreventUnique )
{
/*
// Start on the rarest, and work backwards
if ( !bPreventUnique )
{
if ( RandomFloat(0,1) < item_quality_chance_unique.GetFloat() )
return AE_UNIQUE;
}
if ( RandomFloat(0,1) < item_quality_chance_rare.GetFloat() )
return AE_RARITY2;
if ( RandomFloat(0,1) < item_quality_chance_common.GetFloat() )
return AE_RARITY1;
*/
return AE_NORMAL;
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
CEconItemAttribute *CEconItemSystem::GenerateAttribute( attrib_definition_index_t iAttributeDefinition, float flValue )
{
// Create a new instance of the chosen attribute
CEconItemAttribute *pAttribute = new CEconItemAttribute( iAttributeDefinition, flValue );
return pAttribute;
}
//-----------------------------------------------------------------------------
// Purpose: Generate an attribute by name. Used for debugging.
//-----------------------------------------------------------------------------
CEconItemAttribute *CEconItemSystem::GenerateAttribute( const char *pszName, float flValue )
{
// Find the attribute index matching the class
CEconItemSchema::EconAttrDefsContainer_t &mapDefs = m_itemSchema.GetAttributeDefinitionContainer();
FOR_EACH_VEC( mapDefs, i )
{
if ( !mapDefs[i] )
continue;
if ( Q_stricmp(pszName, mapDefs[i]->GetDefinitionName()) )
continue;
return GenerateAttribute( mapDefs[i]->GetDefinitionIndex(), flValue );
}
return NULL;
}
static ISteamHTTP *GetISteamHTTP()
{
if ( steamapicontext != NULL && steamapicontext->SteamHTTP() )
{
return steamapicontext->SteamHTTP();
}
#ifndef CLIENT_DLL
if ( steamgameserverapicontext != NULL )
{
return steamgameserverapicontext->SteamHTTP();
}
#endif
return NULL;
}
//-----------------------------------------------------------------------------
// Purpose: Common functionality for using our raw buffer data to initialize
// the schema when safe.
//-----------------------------------------------------------------------------
bool IDelayedSchemaData::InitializeSchemaInternal( CEconItemSchema *pItemSchema, CUtlBuffer& bufRawData, bool bInitAsBinary, uint32 nExpectedVersion )
{
DevMsg( 2, "Applying new item schema, version %08X\n", nExpectedVersion );
CUtlVector<CUtlString> vecErrors;
bool bSuccess = bInitAsBinary
? pItemSchema->BInitBinaryBuffer( bufRawData, &vecErrors )
: pItemSchema->BInitTextBuffer( bufRawData, &vecErrors );
if( bSuccess )
{
// Sanity-check that we received the version that they sent us
uint32 nOurVersion = pItemSchema->GetVersion();
if ( nExpectedVersion != 0 && nOurVersion != nExpectedVersion )
{
Warning( "**WARNING** Item schema mismatch after update!\n" );
Warning( "GC told us to expect %08X, we got %08X\n", nExpectedVersion, nOurVersion );
}
}
else
{
Warning( "**WARNING** Failed to apply item schema!\n" );
FOR_EACH_VEC( vecErrors, nError )
{
Warning( "%s\n", vecErrors[nError].Get() );
}
}
return bSuccess;
}
//-----------------------------------------------------------------------------
// Purpose: The GC sent us a single block of binary data.
//-----------------------------------------------------------------------------
class DelayedSchemaData_GCDirectData : public IDelayedSchemaData
{
public:
DelayedSchemaData_GCDirectData( const std::string& strBuffer )
: m_bufRawData( strBuffer.data(), strBuffer.size(), CUtlBuffer::READ_ONLY )
{
//
}
virtual bool InitializeSchema( CEconItemSchema *pItemSchema )
{
return InitializeSchemaInternal( pItemSchema, m_bufRawData, true, 0 );
}
private:
CUtlBuffer m_bufRawData;
};
//-----------------------------------------------------------------------------
// Purpose: We received a text file from an HTML request.
//-----------------------------------------------------------------------------
class DelayedSchemaData_HTTPResponseData : public IDelayedSchemaData
{
public:
DelayedSchemaData_HTTPResponseData( ISteamHTTP *pHTTP, HTTPRequestHandle handleHTTPRequest, uint32 unBodySize, uint32 nExpectedVersion )
: m_nExpectedVersion( nExpectedVersion )
{
Assert( pHTTP );
m_bufRawData.SetBufferType( true, true );
m_bufRawData.SeekPut( CUtlBuffer::SEEK_HEAD, unBodySize );
m_bValid = pHTTP->GetHTTPResponseBodyData( handleHTTPRequest, (uint8*)m_bufRawData.Base(), m_bufRawData.TellPut() );
}
virtual bool InitializeSchema( CEconItemSchema *pItemSchema )
{
if ( !m_bValid )
return false;
return InitializeSchemaInternal( pItemSchema, m_bufRawData, false, m_nExpectedVersion );
}
private:
bool m_bValid;
CUtlBuffer m_bufRawData;
uint32 m_nExpectedVersion;
};
#if GC_UPDATE_ITEM_SCHEMA_SUPPORTED
//-----------------------------------------------------------------------------
// Purpose: Update the item schema from the GC
//-----------------------------------------------------------------------------
class CGCUpdateItemSchema : public GCSDK::CGCClientJob
{
public:
CGCUpdateItemSchema( GCSDK::CGCClient *pClient ) : GCSDK::CGCClientJob( pClient ) {
m_szUrl[0] = '\0';
m_nExpectedVersion = 0;
bHTTPCompleted = false;
}
char m_szUrl[512];
uint32 m_nExpectedVersion;
bool bHTTPCompleted;
CCallResult< CGCUpdateItemSchema, HTTPRequestCompleted_t > callback;
virtual bool BYieldingRunGCJob( GCSDK::IMsgNetPacket *pNetPacket )
{
GCSDK::CProtoBufMsg< CMsgUpdateItemSchema > msg( pNetPacket );
#if (defined(GAME_DLL) || defined(CLIENT_DLL))
bool bUseGCCopy = true;
if ( steamapicontext && steamapicontext->SteamUtils() && steamapicontext->SteamUtils()->GetConnectedUniverse() != k_EUniversePublic )
{
bUseGCCopy = CommandLine()->FindParm( "-use_local_item_data" ) == 0;
}
#else
bool bUseGCCopy = true;
#endif
if ( bUseGCCopy == false )
{
Msg( "Loading item schema from local file.\n" );
KeyValuesAD pItemsGameKV( "ItemsGameFile" );
if ( pItemsGameKV->LoadFromFile( g_pFullFileSystem, "scripts/items/items_game.txt", "GAME" ) )
{
CUtlBuffer buffer;
pItemsGameKV->WriteAsBinary( buffer );
CUtlVector< CUtlString > vecErrors;
bool bSuccess = ItemSystem()->GetItemSchema()->BInitBinaryBuffer( buffer, &vecErrors );
if( !bSuccess )
{
FOR_EACH_VEC( vecErrors, nError )
{
Warning( "%s\n", vecErrors[nError].Get() );
}
}
}
return true;
}
// Check if we're already up-to-date
m_nExpectedVersion = msg.Body().item_schema_version();
uint32 nCurrentSchemaVersion = ItemSystem()->GetItemSchema()->GetVersion();
if ( m_nExpectedVersion != 0 && m_nExpectedVersion == nCurrentSchemaVersion )
{
Msg( "Current item schema is up-to-date with version %08X.\n", nCurrentSchemaVersion );
return true;
}
// !TEST!
//const char *szURL = "http://cdntest.steampowered.com/apps/440/scripts/items/items_game.b8b7a85b4dd98b139957004b86ec0bc070a59d18.txt";
if ( msg.Body().has_items_game() )
{
DevMsg( 2, "Received %d bytes item schema version %08X direct data; update is queued.\n", (int)msg.Body().items_game().size(), m_nExpectedVersion );
ItemSystem()->GetItemSchema()->MaybeInitFromBuffer( new DelayedSchemaData_GCDirectData( msg.Body().items_game() ) );
}
else
{
// Remember URL
const char *szURL = msg.Body().items_game_url().c_str();
if ( !szURL || !szURL[0] )
{
Warning( "GC sent malformed CGCUpdateItemSchema message: No schema data, no URL\n" );
}
else
{
Q_strncpy( m_szUrl, szURL, sizeof( m_szUrl ) );
//Msg( "Fetching %s to update item schema\n", m_szUrl );
// Send an HTTP request for the file
ISteamHTTP *pHTTP = GetISteamHTTP();
if ( !pHTTP )
{
// We must be a game server. Request the direct data.
Msg( "ISteamHTTP not available to update item schema, requesting direct data for version %08X\n", m_nExpectedVersion );
return true;
}
HTTPRequestHandle hReq = pHTTP->CreateHTTPRequest( k_EHTTPMethodGET, m_szUrl );
pHTTP->SetHTTPRequestNetworkActivityTimeout( hReq, 10 );
SteamAPICall_t hCall;
if ( !pHTTP->SendHTTPRequest( hReq, &hCall ) )
{
Warning( "Failed to update item schema: couldn't fetch %s\n", m_szUrl );
return true;
}
//
// *Wait* for completion.
//
// This is important. The GC needs to be able to safely assume that
// we will not process the next message until we have finished
// dealing with this one.
//
bHTTPCompleted = false;
#ifndef CLIENT_DLL
if ( steamgameserverapicontext != NULL && pHTTP == steamgameserverapicontext->SteamHTTP() )
{
callback.SetGameserverFlag();
}
#endif
callback.Set( hCall, this, &CGCUpdateItemSchema::OnHTTPCompleted );
// Wait for it to finish.
while ( !bHTTPCompleted )
{
BYieldingWaitOneFrame();
}
}
}
return true;
}
void OnHTTPCompleted( HTTPRequestCompleted_t *arg, bool bFailed )
{
// Clear flag, no matter what else, so we can stop yielding
bHTTPCompleted = true;
ISteamHTTP *pHTTP = GetISteamHTTP();
Assert( pHTTP );
if ( !pHTTP ) return;
if ( arg->m_eStatusCode != k_EHTTPStatusCode200OK )
{
Warning( "Failed to update item schema: HTTP status %d fetching %s\n", arg->m_eStatusCode, m_szUrl );
}
else
{
if ( !arg->m_bRequestSuccessful )
{
bFailed = true;
}
if ( !bFailed )
{
uint32 unBodySize;
if ( !pHTTP->GetHTTPResponseBodySize( arg->m_hRequest, &unBodySize ) )
{
Assert( false );
bFailed = true;
}
else
{
DevMsg( 2, "Fetched %d bytes item schema version %08X via HTTP; update is queued.\n", unBodySize, m_nExpectedVersion );
ItemSystem()->GetItemSchema()->MaybeInitFromBuffer( new DelayedSchemaData_HTTPResponseData( pHTTP, arg->m_hRequest, unBodySize, m_nExpectedVersion ) );
}
}
if ( bFailed )
{
Warning( "Failed to update item schema from %s\n", m_szUrl );
}
}
pHTTP->ReleaseHTTPRequest( arg->m_hRequest );
}
};
GC_REG_CLIENT_JOB( CGCUpdateItemSchema, k_EMsgGCUpdateItemSchema );
#endif
#ifdef CLIENT_DLL
//-----------------------------------------------------------------------------
// Purpose: Update the item schema from the GC
//-----------------------------------------------------------------------------
CON_COMMAND_F( econ_show_items_with_tag, "Lists the item definitions that have a specified tag.", FCVAR_CLIENTDLL )
{
if ( args.ArgC() != 2 )
return;
econ_tag_handle_t tagHandle = GetItemSchema()->GetHandleForTag( args.Arg( 1 ) );
FOR_EACH_MAP( GetItemSchema()->GetSortedItemDefinitionMap(), i )
{
const CEconItemDefinition *pItemDef = GetItemSchema()->GetItemDefinitionMap()[i];
if ( pItemDef->HasEconTag( tagHandle ) )
{
Msg(" '%s'\n", pItemDef->GetDefinitionName() );
}
}
}
#endif // CLIENT_DLL
#ifdef CLIENT_DLL
//-----------------------------------------------------------------------------
// Purpose: Get a file from a given URL.
// Generalized version of the item schema update.
//-----------------------------------------------------------------------------
CGCFetchWebResource::CGCFetchWebResource( GCSDK::CGCClient *pClient, CUtlString strName, CUtlString strURL, bool bForceSkipCache/*=false*/, KeyValues *pkvGETParams/*=NULL*/ ) : GCSDK::CGCClientJob( pClient )
{
m_strName = strName;
m_strURL = strURL;
m_bHTTPCompleted = false;
m_bForceSkipCache = bForceSkipCache;
m_pkvGETParams = pkvGETParams->MakeCopy();
}
CGCFetchWebResource::~CGCFetchWebResource()
{
if ( m_pkvGETParams )
{
m_pkvGETParams->deleteThis();
}
}
bool CGCFetchWebResource::BYieldingRunGCJob()
{
ISteamHTTP *pHTTP = GetISteamHTTP();
if ( !pHTTP )
{
// We must be a game server.
Msg( "ISteamHTTP not available to update web resource.\n" );
return true;
}
// Send an HTTP request for the file.
HTTPRequestHandle hReq = pHTTP->CreateHTTPRequest( k_EHTTPMethodGET, m_strURL.Get() );
pHTTP->SetHTTPRequestNetworkActivityTimeout( hReq, 10 );
// Skip any data that may be cached in the Steam client or elsewhere
if ( m_bForceSkipCache )
{
pHTTP->SetHTTPRequestHeaderValue( hReq, "Cache-Control", "no-cache, no-store" );
}
// Add any GET params
if ( m_pkvGETParams )
{
FOR_EACH_VALUE( m_pkvGETParams, kvSubKey )
{
pHTTP->SetHTTPRequestGetOrPostParameter( hReq, kvSubKey->GetName(), kvSubKey->GetString() );
}
}
SteamAPICall_t hCall;
if ( !pHTTP->SendHTTPRequest( hReq, &hCall ) )
{
Warning( "Failed to update web resource: couldn't fetch %s\n", m_strURL.Get() );
return true;
}
// Wait for completion.
m_bHTTPCompleted = false;
callback.Set( hCall, this, &CGCFetchWebResource::OnHTTPCompleted );
// Wait for it to finish.
while ( !m_bHTTPCompleted )
{
BYieldingWaitOneFrame();
}
return true;
}
void CGCFetchWebResource::OnHTTPCompleted( HTTPRequestCompleted_t *arg, bool bFailed )
{
// Clear flag so we can stop yielding.
m_bHTTPCompleted = true;
ISteamHTTP *pHTTP = GetISteamHTTP();
Assert( pHTTP );
if ( !pHTTP )
return;
if ( arg->m_eStatusCode != k_EHTTPStatusCode200OK )
{
Warning( "Failed to update web resource: HTTP status %d fetching %s\n", arg->m_eStatusCode, m_strURL.Get() );
}
else
{
if ( !arg->m_bRequestSuccessful )
{
bFailed = true;
}
if ( !bFailed )
{
uint32 unBodySize;
if ( !pHTTP->GetHTTPResponseBodySize( arg->m_hRequest, &unBodySize ) )
{
Assert( false );
bFailed = true;
}
else
{
DevMsg( 2, "Fetched %d bytes for web resource via HTTP.\n", unBodySize );
CUtlBuffer bufRawData;
bufRawData.SetBufferType( true, true );
bufRawData.SeekPut( CUtlBuffer::SEEK_HEAD, unBodySize );
bool bValid = pHTTP->GetHTTPResponseBodyData( arg->m_hRequest, (uint8*) bufRawData.Base(), bufRawData.TellPut() );
if ( bValid )
{
KeyValues* pResourceKV = new KeyValues( "" );
pResourceKV->LoadFromBuffer( "", bufRawData );
ItemSystem()->GetItemSchema()->SetWebResource( m_strName, pResourceKV );
}
}
}
if ( bFailed )
{
Warning( "Failed to update web resource from %s\n", m_strURL.Get() );
}
}
pHTTP->ReleaseHTTPRequest( arg->m_hRequest );
}
#endif