sco/pokerth
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Detect avatar file type for download. Not fully working yet.

Browse Source
This commit is contained in:
lotodore
2009-03-22 00:39:25 +00:00
parent 84cbb90e92
commit f2ec0b1ef4
7 changed files with 77 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/core/avatarmanager.h
+2
View File
@@ -53,6 +53,8 @@ public:
static unsigned ChunkReadAvatarFile(boost::shared_ptr<AvatarFileState> fileState, unsigned char *data, unsigned chunkSize);
static int AvatarFileToNetPackets(const std::string &fileName, unsigned requestId, NetPacketList &packets);
static AvatarFileType GetAvatarFileType(const std::string &fileName);
static std::string GetAvatarFileExtension(AvatarFileType fileType);
bool GetHashForAvatar(const std::string &fileName, MD5Buf &md5buf) const;
bool GetAvatarFileName(const MD5Buf &md5buf, std::string &fileName) const;
src/core/common/avatarmanager.cpp
+42 -24
View File
@@ -129,18 +129,10 @@ boost::shared_ptr<AvatarFileState>
AvatarManager::OpenAvatarFileForChunkRead(const std::string &fileName, unsigned &outFileSize, AvatarFileType &outFileType)
{
outFileSize = 0;
outFileType = AVATAR_FILE_TYPE_UNKNOWN;
boost::shared_ptr<AvatarFileState> retVal;
try
{
path filePath(fileName);
string ext(extension(filePath));
if (boost::algorithm::iequals(ext, ".png"))
outFileType = AVATAR_FILE_TYPE_PNG;
else if (boost::algorithm::iequals(ext, ".jpg") || boost::algorithm::iequals(ext, ".jpeg"))
outFileType = AVATAR_FILE_TYPE_JPG;
else if (boost::algorithm::iequals(ext, ".gif"))
outFileType = AVATAR_FILE_TYPE_GIF;
outFileType = GetAvatarFileType(fileName);
boost::shared_ptr<AvatarFileState> fileState(new AvatarFileState);
fileState->inputStream.open(fileName.c_str(), ios_base::in | ios_base::binary);
if (!fileState->inputStream.fail())
@@ -243,6 +235,46 @@ AvatarManager::AvatarFileToNetPackets(const string &fileName, unsigned requestId
return retVal;
}
AvatarFileType
AvatarManager::GetAvatarFileType(const string &fileName)
{
AvatarFileType fileType;
path filePath(fileName);
string ext(extension(filePath));
if (boost::algorithm::iequals(ext, ".png"))
fileType = AVATAR_FILE_TYPE_PNG;
else if (boost::algorithm::iequals(ext, ".jpg") || boost::algorithm::iequals(ext, ".jpeg"))
fileType = AVATAR_FILE_TYPE_JPG;
else if (boost::algorithm::iequals(ext, ".gif"))
fileType = AVATAR_FILE_TYPE_GIF;
else
fileType = AVATAR_FILE_TYPE_UNKNOWN;
return fileType;
}
string
AvatarManager::GetAvatarFileExtension(AvatarFileType fileType)
{
string ext;
switch (fileType)
{
case AVATAR_FILE_TYPE_PNG:
ext = ".png";
break;
case AVATAR_FILE_TYPE_JPG:
ext = ".jpg";
break;
case AVATAR_FILE_TYPE_GIF:
ext = ".gif";
break;
case AVATAR_FILE_TYPE_UNKNOWN:
break;
}
return ext;
}
bool
AvatarManager::GetHashForAvatar(const std::string &fileName, MD5Buf &md5buf) const
{
@@ -337,21 +369,7 @@ AvatarManager::StoreAvatarInCache(const MD5Buf &md5buf, AvatarFileType avatarFil
}
try
{
string ext;
switch (avatarFileType)
{
case AVATAR_FILE_TYPE_PNG:
ext = ".png";
break;
case AVATAR_FILE_TYPE_JPG:
ext = ".jpg";
break;
case AVATAR_FILE_TYPE_GIF:
ext = ".gif";
break;
case AVATAR_FILE_TYPE_UNKNOWN:
break;
}
string ext(GetAvatarFileExtension(avatarFileType));
if (!ext.empty() && !cacheDir.empty())
{
// Check header before storing file.
src/net/common/clientthread.cpp
+2 -2
View File
@@ -557,7 +557,7 @@ ClientThread::RetrieveAvatarIfNeeded(unsigned id, const PlayerInfo &info)
string avatarServerAddress(GetContext().GetAvatarServerAddr());
if (!avatarServerAddress.empty() && m_avatarDownloader)
{
string serverFileName(info.avatar.ToString() + ".jpg"); // TODO!
string serverFileName(info.avatar.ToString() + AvatarManager::GetAvatarFileExtension(info.avatarType));
m_avatarDownloader->QueueDownload(
id, avatarServerAddress + serverFileName, GetContext().GetCacheDir() + TEMP_AVATAR_FILENAME);
}
@@ -621,7 +621,7 @@ ClientThread::PassAvatarDataToManager(unsigned playerId, boost::shared_ptr<Avata
else
{
if (avatarData->fileType == AVATAR_FILE_TYPE_UNKNOWN)
avatarData->fileType = AVATAR_FILE_TYPE_JPG; // TODO!
avatarData->fileType = tmpPlayerInfo.avatarType;
if (!GetAvatarManager().StoreAvatarInCache(tmpPlayerInfo.avatar, avatarData->fileType, &avatarData->fileData[0], avatarData->reportedSize, false))
LOG_ERROR("Failed to store avatar in cache directory.");
src/net/common/netpacket.cpp
+17
View File
@@ -95,6 +95,7 @@ using namespace std;
#define NET_PLAYER_FLAG_HUMAN 0x01
#define NET_PLAYER_FLAG_HAS_AVATAR 0x02
#define NET_PLAYER_FLAG_AVATAR_TYPE_MASK 0x0c
#define NET_START_FLAG_FILL_WITH_CPU_PLAYERS 0x01
@@ -2454,6 +2455,7 @@ NetPacketPlayerInfo::SetData(const NetPacketPlayerInfo::Data &inData)
if (inData.playerInfo.hasAvatar)
{
tmpPlayerFlags |= NET_PLAYER_FLAG_HAS_AVATAR;
tmpPlayerFlags |= ((inData.playerInfo.avatarType) << 2);
char *avatarPtr = (char *)tmpData + sizeof(NetPacketPlayerInfoData);
memcpy(avatarPtr, inData.playerInfo.avatar.data, MD5_DATA_SIZE);
}
@@ -2479,6 +2481,21 @@ NetPacketPlayerInfo::GetData(NetPacketPlayerInfo::Data &outData) const
if (outData.playerInfo.hasAvatar)
{
switch ((tmpPlayerFlags & NET_PLAYER_FLAG_AVATAR_TYPE_MASK) >> 2)
{
case AVATAR_FILE_TYPE_PNG :
outData.playerInfo.avatarType = AVATAR_FILE_TYPE_PNG;
break;
case AVATAR_FILE_TYPE_JPG :
outData.playerInfo.avatarType = AVATAR_FILE_TYPE_JPG;
break;
case AVATAR_FILE_TYPE_GIF :
outData.playerInfo.avatarType = AVATAR_FILE_TYPE_GIF;
break;
default:
outData.playerInfo.avatarType = AVATAR_FILE_TYPE_UNKNOWN;
break;
}
char *avatarPtr = (char *)tmpData + sizeof(NetPacketPlayerInfoData);
memcpy(outData.playerInfo.avatar.data, avatarPtr, MD5_DATA_SIZE);
}
src/net/common/serverlobbythread.cpp
+11 -1
View File
@@ -523,9 +523,10 @@ ServerLobbyThread::HandleNetPacketInit(SessionWrapper session, const NetPacketIn
m_sessionManager.SetSessionPlayerData(session.sessionData->GetId(), tmpPlayerData);
session.playerData = tmpPlayerData;
string avatarFileName;
if (initData.showAvatar
&& !initData.avatar.IsZero()
&& !GetAvatarManager().HasAvatar(initData.avatar))
&& !GetAvatarManager().GetAvatarFileName(initData.avatar, avatarFileName))
{
bool avatarRecentlyRequested = false;
{
@@ -540,6 +541,8 @@ ServerLobbyThread::HandleNetPacketInit(SessionWrapper session, const NetPacketIn
}
else
{
if (!avatarFileName.empty())
session.playerData->SetAvatarFile(avatarFileName);
EstablishSession(session);
}
}
@@ -619,6 +622,10 @@ ServerLobbyThread::HandleNetPacketAvatarEnd(SessionWrapper session, const NetPac
// Free memory.
session.playerData->SetNetAvatarData(boost::shared_ptr<AvatarData>());
// Set avatar file name.
string avatarFileName;
if (GetAvatarManager().GetAvatarFileName(avatarMD5, avatarFileName))
session.playerData->SetAvatarFile(avatarFileName);
// Init finished - start session.
EstablishSession(session);
}
@@ -656,7 +663,10 @@ ServerLobbyThread::HandleNetPacketRetrievePlayerInfo(SessionWrapper session, con
infoData.playerInfo.playerName = tmpPlayer->GetName();
infoData.playerInfo.hasAvatar = !tmpPlayer->GetAvatarMD5().IsZero();
if (infoData.playerInfo.hasAvatar)
{
infoData.playerInfo.avatar = tmpPlayer->GetAvatarMD5();
infoData.playerInfo.avatarType = AvatarManager::GetAvatarFileType(tmpPlayer->GetAvatarFile());
}
static_cast<NetPacketPlayerInfo *>(info.get())->SetData(infoData);
session.sessionData->GetSender().Send(session.sessionData, info);
}
src/playerdata.h
+2 -1
View File
@@ -61,11 +61,12 @@ struct AvatarData
struct PlayerInfo
{
PlayerInfo() : ptype(PLAYER_TYPE_HUMAN) {}
PlayerInfo() : ptype(PLAYER_TYPE_HUMAN), hasAvatar(false), avatarType(AVATAR_FILE_TYPE_UNKNOWN) {}
std::string playerName;
PlayerType ptype;
bool hasAvatar;
MD5Buf avatar;
AvatarFileType avatarType;
};
class PlayerData