Battle.net: Implemented cache stream items

3 files changed, 121 insertions, 3 deletions

diff --git a/src/server/bnetserver/Packets/CachePackets.cpp b/src/server/bnetserver/Packets/CachePackets.cpp
new file mode 100644
index 00000000000..cd5aa05fed6
--- /dev/null
+++ b/src/server/bnetserver/Packets/CachePackets.cpp
@@ -0,0 +1,83 @@
+/*
+ * Copyright (C) 2008-2014 TrinityCore <http://www.trinitycore.org/>
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the
+ * Free Software Foundation; either version 2 of the License, or (at your
+ * option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "CachePackets.h"
+
+void Battlenet::Cache::GetStreamItemsRequest::Read()
+{
+    if (_stream.Read<bool>(1))
+    {
+        _stream.Read<uint16>(11);   // padding
+        ItemName = _stream.ReadFourCC();
+        Channel = _stream.ReadFourCC();
+    }
+    else
+        _stream.Read<uint16>(16);
+
+    _stream.Read<bool>(1);  // StreamDirection
+    ReferenceTime = _stream.Read<int32>(32) - std::numeric_limits<int32>::min();
+    Locale = _stream.ReadFourCC();
+    Index = _stream.Read<uint32>(32);
+    _stream.Read<uint8>(6); // Module count, always 0
+}
+
+std::string Battlenet::Cache::GetStreamItemsRequest::ToString() const
+{
+    std::ostringstream stream;
+    stream << "Battlenet::Cache::GetStreamItemsRequest Channel: " << Channel << ", ItemName: " << ItemName
+        << ", Locale: " << Locale << ", Index: " << Index;
+    return stream.str();
+}
+
+void Battlenet::Cache::GetStreamItemsRequest::CallHandler(Session* session)
+{
+    session->HandleGetStreamItemsRequest(*this);
+}
+
+Battlenet::Cache::GetStreamItemsResponse::~GetStreamItemsResponse()
+{
+    for (size_t i = 0; i < Modules.size(); ++i)
+        delete Modules[i];
+}
+
+void Battlenet::Cache::GetStreamItemsResponse::Write()
+{
+    _stream.Write(0, 16);
+    _stream.Write(Modules.size(), 6);
+    for (ModuleInfo const* info : Modules)
+    {
+        _stream.WriteBytes(info->Type.c_str(), 4);
+        _stream.WriteFourCC(info->Region);
+        _stream.WriteBytes(info->ModuleId, 32);
+        _stream.WriteBytes(info->Data, 4);
+    }
+
+    _stream.Write(Index, 32);
+    _stream.Write(0, 17);   // padding
+    _stream.Write(1, 16);
+    _stream.Write(0, 2);
+}
+
+std::string Battlenet::Cache::GetStreamItemsResponse::ToString() const
+{
+    std::ostringstream stream;
+    stream << "Battlenet::Cache::GetStreamItemsResponse modules " << Modules.size();
+    for (ModuleInfo const* module : Modules)
+        stream << std::endl << "Battlenet::ModuleInfo Locale " << module->Region.c_str() << ", ModuleId " << ByteArrayToHexStr(module->ModuleId, 32) << ", Data " << ByteArrayToHexStr(module->Data, module->DataSize);
+
+    return stream.str();
+}
diff --git a/src/server/bnetserver/Packets/CachePackets.h b/src/server/bnetserver/Packets/CachePackets.h
index 2e473df4e72..ba4153fe623 100644
--- a/src/server/bnetserver/Packets/CachePackets.h
+++ b/src/server/bnetserver/Packets/CachePackets.h
@@ -29,7 +29,7 @@ namespace Battlenet
             CMSG_GATEWAY_LOOKUP_REQUEST     = 0x2,  // Not implemented
             CMSG_CONNECT_REQUEST            = 0x4,  // Not implemented
             CMSG_DATA_CHUNK                 = 0x7,  // Not implemented
-            SMSG_GET_STREAM_ITEMS_REQUEST   = 0x9,  // Not implemented
+            CMSG_GET_STREAM_ITEMS_REQUEST   = 0x9,  // Not implemented
 
             SMSG_GATEWAY_LOOKUP_RESPONSE    = 0x3,  // Not implemented
             SMSG_CONNECT_RESPONSE           = 0x4,  // Not implemented
@@ -37,6 +37,41 @@ namespace Battlenet
             SMSG_RESULT                     = 0x8,  // Not implemented
             SMSG_GET_STREAM_ITEMS_RESPONSE  = 0x9   // Not implemented
         };
+
+        class GetStreamItemsRequest final : public ClientPacket
+        {
+        public:
+            GetStreamItemsRequest(PacketHeader const& header, BitStream& stream) : ClientPacket(header, stream)
+            {
+                ASSERT(header == PacketHeader(CMSG_GET_STREAM_ITEMS_REQUEST, CACHE) && "Invalid packet header for GetStreamItemsRequest");
+            }
+
+            void Read() override;
+            std::string ToString() const override;
+            void CallHandler(Session* session);
+
+            std::string Channel;
+            std::string ItemName;
+            std::string Locale;
+            uint32 Index;
+            int32 ReferenceTime;
+        };
+
+        class GetStreamItemsResponse final : public ServerPacket
+        {
+        public:
+            GetStreamItemsResponse() : ServerPacket(PacketHeader(SMSG_GET_STREAM_ITEMS_RESPONSE, CACHE))
+            {
+            }
+
+            ~GetStreamItemsResponse();
+
+            void Write() override;
+            std::string ToString() const override;
+
+            uint32 Index;
+            std::vector<ModuleInfo*> Modules;
+        };
     }
 }
 
diff --git a/src/server/bnetserver/Packets/PacketManager.cpp b/src/server/bnetserver/Packets/PacketManager.cpp
index af31b596192..019cf48ac30 100644
--- a/src/server/bnetserver/Packets/PacketManager.cpp
+++ b/src/server/bnetserver/Packets/PacketManager.cpp
@@ -217,13 +217,13 @@ void Battlenet::PacketManager::RegisterCachePackets()
     REGISTER_CLIENT_PACKET_NAME(PacketHeader(Cache::CMSG_GATEWAY_LOOKUP_REQUEST, CACHE), "Cache::GatewayLookupRequest");
     REGISTER_CLIENT_PACKET_NAME(PacketHeader(Cache::CMSG_CONNECT_REQUEST, CACHE), "Cache::ConnectRequest");
     REGISTER_CLIENT_PACKET_NAME(PacketHeader(Cache::CMSG_DATA_CHUNK, CACHE), "Cache::DataChunk");
-    REGISTER_CLIENT_PACKET_NAME(PacketHeader(Cache::SMSG_GET_STREAM_ITEMS_REQUEST, CACHE), "Cache::GetStreamItemsRequest");
+    REGISTER_CLIENT_PACKET(PacketHeader(Cache::CMSG_GET_STREAM_ITEMS_REQUEST, CACHE), Cache::GetStreamItemsRequest);
 
     REGISTER_SERVER_PACKET_NAME(PacketHeader(Cache::SMSG_GATEWAY_LOOKUP_RESPONSE, CACHE), "Cache::GatewayLookupResponse");
     REGISTER_SERVER_PACKET_NAME(PacketHeader(Cache::SMSG_CONNECT_RESPONSE, CACHE), "Cache::ConnectResponse");
     REGISTER_SERVER_PACKET_NAME(PacketHeader(Cache::SMSG_PUBLISH_LIST_RESPONSE, CACHE), "Cache::PublishListResponse");
     REGISTER_SERVER_PACKET_NAME(PacketHeader(Cache::SMSG_RESULT, CACHE), "Cache::Result");
-    REGISTER_SERVER_PACKET_NAME(PacketHeader(Cache::SMSG_GET_STREAM_ITEMS_RESPONSE, CACHE), "Cache::GetStreamItemsResponse");
+    REGISTER_SERVER_PACKET(PacketHeader(Cache::SMSG_GET_STREAM_ITEMS_RESPONSE, CACHE), Cache::GetStreamItemsResponse);
 }
 
 void Battlenet::PacketManager::RegisterProfilePackets()