Mirror/TrinityCore
1
0
Fork 0
mirror of https://github.com/TrinityCore/TrinityCore.git synced 2026-01-16 07:30:42 +01:00
Code Issues Packages Projects Releases Wiki Activity

Core/Networking: Updated packet headers for 6.2.4

Browse Source
This commit is contained in:
Shauren
2016-03-24 21:15:02 +01:00
parent 6de7a0b879
commit 45c857a6bf
3 changed files with 31 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/server/game/Server/Protocol/ServerPktHeader.h
View File

@@ -26,13 +26,13 @@ union ServerPktHeader
struct
{
uint16 Size;
uint32 Command;
uint16 Command;
} Setup;
struct
{
uint32 Command : 13;
uint32 Size : 19;
uint32 Size;
uint16 Command;
} Normal;
};

42
src/server/game/Server/WorldSocket.cpp
View File

@@ -53,16 +53,13 @@ private:
using boost::asio::ip::tcp;
uint32 const WorldSocket::ConnectionInitializeMagic = 0xF5EB1CE;
std::string const WorldSocket::ServerConnectionInitialize("WORLD OF WARCRAFT CONNECTION - SERVER TO CLIENT");
std::string const WorldSocket::ClientConnectionInitialize("WORLD OF WARCRAFT CONNECTION - CLIENT TO SERVER");
uint32 const WorldSocket::MinSizeForCompression = 0x400;
uint32 const SizeOfClientHeader[2] =
{
6, 4
};
uint32 const SizeOfServerHeader[2] = { sizeof(uint16) + sizeof(uint32), sizeof(uint32) };
uint32 const SizeOfClientHeader[2] = { sizeof(uint16) + sizeof(uint16), sizeof(uint32) + sizeof(uint16) };
uint32 const SizeOfServerHeader[2] = { sizeof(uint16) + sizeof(uint16), sizeof(uint32) + sizeof(uint16) };
WorldSocket::WorldSocket(tcp::socket&& socket) : Socket(std::move(socket)),
_type(CONNECTION_TYPE_REALM), _authSeed(rand32()), _OverSpeedPings(0),
@@ -116,13 +113,14 @@ void WorldSocket::CheckIpCallback(PreparedQueryResult result)
}
}
_packetBuffer.Resize(2 + ClientConnectionInitialize.length() + 1);
_packetBuffer.Resize(4 + 2 + ClientConnectionInitialize.length() + 1);
AsyncReadWithCallback(&WorldSocket::InitializeHandler);
MessageBuffer initializer;
ServerPktHeader header;
header.Setup.Size = ServerConnectionInitialize.size();
initializer.Write(&ConnectionInitializeMagic, sizeof(ConnectionInitializeMagic));
initializer.Write(&header, sizeof(header.Setup.Size));
initializer.Write(ServerConnectionInitialize.c_str(), ServerConnectionInitialize.length());
@@ -158,8 +156,14 @@ void WorldSocket::InitializeHandler(boost::system::error_code error, std::size_t
return;
}
std::string initializer(reinterpret_cast<char const*>(_packetBuffer.GetReadPointer() + 2), std::min(_packetBuffer.GetActiveSize() - 2, ClientConnectionInitialize.length()));
if (initializer != ClientConnectionInitialize)
uint32 magic;
uint16 length;
ByteBuffer buffer(std::move(_packetBuffer));
buffer >> magic;
buffer >> length;
std::string initializer = buffer.ReadString(length);
if (magic != ConnectionInitializeMagic || initializer != ClientConnectionInitialize)
{
CloseSocket();
return;
@@ -322,12 +326,12 @@ void WorldSocket::ExtractOpcodeAndSize(ClientPktHeader const* header, uint32& op
if (_authCrypt.IsInitialized())
{
opcode = header->Normal.Command;
size = header->Normal.Size;
size = header->Normal.Size - 2;
}
else
{
opcode = header->Setup.Command;
size = header->Setup.Size - 4;
size = header->Setup.Size - 2;
}
}
@@ -342,7 +346,7 @@ bool WorldSocket::ReadHeaderHandler()
{
ASSERT(_headerBuffer.GetActiveSize() == SizeOfClientHeader[_authCrypt.IsInitialized()], "Header size " SZFMTD " different than expected %u", _headerBuffer.GetActiveSize(), SizeOfClientHeader[_authCrypt.IsInitialized()]);
_authCrypt.DecryptRecv(_headerBuffer.GetReadPointer(), _headerBuffer.GetActiveSize());
_authCrypt.DecryptRecv(_headerBuffer.GetReadPointer(), 4);
ClientPktHeader* header = reinterpret_cast<ClientPktHeader*>(_headerBuffer.GetReadPointer());
uint32 opcode;
@@ -514,8 +518,8 @@ void WorldSocket::WritePacketToBuffer(EncryptablePacket const& packet, MessageBu
if (packetSize > MinSizeForCompression && packet.NeedsEncryption())
{
CompressedWorldPacket cmp;
cmp.UncompressedSize = packetSize + 4;
cmp.UncompressedAdler = adler32(adler32(0x9827D8F1, (Bytef*)&opcode, 4), packet.contents(), packetSize);
cmp.UncompressedSize = packetSize + 2;
cmp.UncompressedAdler = adler32(adler32(0x9827D8F1, (Bytef*)&opcode, 2), packet.contents(), packetSize);
// Reserve space for compression info - uncompressed size and checksums
uint8* compressionInfo = buffer.GetWritePointer();
@@ -534,15 +538,17 @@ void WorldSocket::WritePacketToBuffer(EncryptablePacket const& packet, MessageBu
else if (!packet.empty())
buffer.Write(packet.contents(), packet.size());
packetSize += 2 /*opcode*/;
if (packet.NeedsEncryption())
{
header.Normal.Size = packetSize;
header.Normal.Command = opcode;
_authCrypt.EncryptSend((uint8*)&header, sizeOfHeader);
_authCrypt.EncryptSend((uint8*)&header, 4);
}
else
{
header.Setup.Size = packetSize + 4;
header.Setup.Size = packetSize;
header.Setup.Command = opcode;
}
@@ -552,12 +558,12 @@ void WorldSocket::WritePacketToBuffer(EncryptablePacket const& packet, MessageBu
uint32 WorldSocket::CompressPacket(uint8* buffer, WorldPacket const& packet)
{
uint32 opcode = packet.GetOpcode();
uint32 bufferSize = deflateBound(_compressionStream, packet.size() + sizeof(opcode));
uint32 bufferSize = deflateBound(_compressionStream, packet.size() + sizeof(uint16));
_compressionStream->next_out = buffer;
_compressionStream->avail_out = bufferSize;
_compressionStream->next_in = (Bytef*)&opcode;
_compressionStream->avail_in = sizeof(uint32);
_compressionStream->avail_in = sizeof(uint16);
int32 z_res = deflate(_compressionStream, Z_NO_FLUSH);
if (z_res != Z_OK)

7
src/server/game/Server/WorldSocket.h
View File

@@ -52,13 +52,13 @@ union ClientPktHeader
struct
{
uint16 Size;
uint32 Command;
uint16 Command;
} Setup;
struct
{
uint32 Command : 13;
uint32 Size : 19;
uint32 Size;
uint16 Command;
} Normal;
static bool IsValidSize(uint32 size) { return size < 10240; }
@@ -69,6 +69,7 @@ union ClientPktHeader
class TC_GAME_API WorldSocket : public Socket<WorldSocket>
{
static uint32 const ConnectionInitializeMagic;
static std::string const ServerConnectionInitialize;
static std::string const ClientConnectionInitialize;
static uint32 const MinSizeForCompression;