inline bool PartialProcessPacket(Battlenet::Session* session, MessageBuffer& inputBuffer) { MessageBuffer& buffer = session->*outputBuffer; // We have full read header, now check the data payload if (buffer.GetRemainingSpace() > 0) { // need more data in the payload std::size_t readDataSize = std::min(inputBuffer.GetActiveSize(), buffer.GetRemainingSpace()); buffer.Write(inputBuffer.GetReadPointer(), readDataSize); inputBuffer.ReadCompleted(readDataSize); } if (buffer.GetRemainingSpace() > 0) { // Couldn't receive the whole data this time. ASSERT(inputBuffer.GetActiveSize() == 0); return false; } // just received fresh new payload if (!(session->*processMethod)()) { session->CloseSocket(); return false; } return true; }
bool WorldSocket::Update() { EncryptablePacket* queued; MessageBuffer buffer; while (_bufferQueue.Dequeue(queued)) { ServerPktHeader header(queued->size() + 2, queued->GetOpcode()); if (queued->NeedsEncryption()) _authCrypt.EncryptSend(header.header, header.getHeaderLength()); if (buffer.GetRemainingSpace() < queued->size() + header.getHeaderLength()) { QueuePacket(std::move(buffer)); buffer.Resize(4096); } if (buffer.GetRemainingSpace() >= queued->size() + header.getHeaderLength()) { buffer.Write(header.header, header.getHeaderLength()); if (!queued->empty()) buffer.Write(queued->contents(), queued->size()); } else // single packet larger than 4096 bytes { MessageBuffer packetBuffer(queued->size() + header.getHeaderLength()); packetBuffer.Write(header.header, header.getHeaderLength()); if (!queued->empty()) packetBuffer.Write(queued->contents(), queued->size()); QueuePacket(std::move(packetBuffer)); } delete queued; } if (buffer.GetActiveSize() > 0) QueuePacket(std::move(buffer)); if (!BaseSocket::Update()) return false; if (_queryFuture.valid() && _queryFuture.wait_for(std::chrono::seconds(0)) == std::future_status::ready) { auto callback = _queryCallback; _queryCallback = nullptr; callback(_queryFuture.get()); } return true; }
void Battlenet::Session::AsyncWrite(ServerPacket* packet) { if (!IsOpen()) { delete packet; return; } TC_LOG_DEBUG("session.packets", "%s Sending %s", GetClientInfo().c_str(), PacketToStringHelper(packet).c_str()); packet->Write(); MessageBuffer buffer; buffer.Write(packet->GetData(), packet->GetSize()); delete packet; std::unique_lock<std::mutex> guard(_writeLock); _crypt.EncryptSend(buffer.GetReadPointer(), buffer.GetActiveSize()); QueuePacket(std::move(buffer), guard); }