diff options
Diffstat (limited to 'src/engine/client/client.cpp')
| -rw-r--r-- | src/engine/client/client.cpp | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/src/engine/client/client.cpp b/src/engine/client/client.cpp index a8e446ac..13e59ac5 100644 --- a/src/engine/client/client.cpp +++ b/src/engine/client/client.cpp @@ -1137,6 +1137,7 @@ void CClient::ProcessPacket(CNetChunk *pPacket) { const char *pMap = Unpacker.GetString(CUnpacker::SANITIZE_CC|CUnpacker::SKIP_START_WHITESPACES); int MapCrc = Unpacker.GetInt(); + int MapSize = Unpacker.GetInt(); const char *pError = 0; if(Unpacker.Error()) @@ -1148,6 +1149,9 @@ void CClient::ProcessPacket(CNetChunk *pPacket) pError = "strange character in map name"; } + if(MapSize < 0) + pError = "invalid map size"; + if(pError) DisconnectWithReason(pError); else @@ -1174,7 +1178,7 @@ void CClient::ProcessPacket(CNetChunk *pPacket) io_close(m_MapdownloadFile); m_MapdownloadFile = Storage()->OpenFile(m_aMapdownloadFilename, IOFLAG_WRITE, IStorage::TYPE_SAVE); m_MapdownloadCrc = MapCrc; - m_MapdownloadTotalsize = -1; + m_MapdownloadTotalsize = MapSize; m_MapdownloadAmount = 0; CMsgPacker Msg(NETMSG_REQUEST_MAP_DATA); @@ -1192,17 +1196,17 @@ void CClient::ProcessPacket(CNetChunk *pPacket) else if(Msg == NETMSG_MAP_DATA) { int Last = Unpacker.GetInt(); - int TotalSize = Unpacker.GetInt(); + int MapCRC = Unpacker.GetInt(); + int Chunk = Unpacker.GetInt(); int Size = Unpacker.GetInt(); const unsigned char *pData = Unpacker.GetRaw(Size); // check fior errors - if(Unpacker.Error() || Size <= 0 || TotalSize <= 0 || !m_MapdownloadFile) + if(Unpacker.Error() || Size <= 0 || MapCRC != m_MapdownloadCrc || Chunk != m_MapdownloadChunk || !m_MapdownloadFile) return; io_write(m_MapdownloadFile, pData, Size); - m_MapdownloadTotalsize = TotalSize; m_MapdownloadAmount += Size; if(Last) |