From b63dbfede548bb9eb0ad7a5b7a83a8a3fac1e1f1 Mon Sep 17 00:00:00 2001 From: oy Date: Wed, 16 Feb 2011 12:07:54 +0100 Subject: fixed problems with map downloading if the map changes during download. Closes #236 --- src/engine/client/client.cpp | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'src/engine/client/client.cpp') 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) -- cgit 1.4.1