mirror of
https://github.com/TrinityCore/TrinityCore.git
synced 2026-01-22 18:15:31 +01:00
*fix Character freeze after map change or login. Closes #19 patch by trazom
--HG-- branch : trunk
This commit is contained in:
@@ -188,7 +188,8 @@ int WorldSocket::SendPacket (const WorldPacket& pct)
|
||||
sWorldLog.outLog ("\n");
|
||||
}
|
||||
|
||||
if (iSendPacket (pct) == -1)
|
||||
// don't try to send the packet if there are packets on the queue
|
||||
if (!m_PacketQueue.is_empty() || iSendPacket(pct) == -1)
|
||||
{
|
||||
WorldPacket* npct;
|
||||
|
||||
@@ -202,6 +203,12 @@ int WorldSocket::SendPacket (const WorldPacket& pct)
|
||||
sLog.outError ("WorldSocket::SendPacket: m_PacketQueue.enqueue_tail failed");
|
||||
return -1;
|
||||
}
|
||||
|
||||
// iSendPacket may fail if the packet is larger than the buffer (even in the buffer is empty)
|
||||
// So we must try to flush so the packet may be sent partially
|
||||
// don't cancel_wakeup_output here
|
||||
if (iFlushPacketQueue())
|
||||
return schedule_wakeup_output(Guard);
|
||||
}
|
||||
|
||||
return 0;
|
||||
@@ -321,10 +328,15 @@ int WorldSocket::handle_output (ACE_HANDLE)
|
||||
if (closing_)
|
||||
return -1;
|
||||
|
||||
const size_t send_len = m_OutBuffer->length ();
|
||||
size_t send_len = m_OutBuffer->length();
|
||||
|
||||
if (send_len == 0)
|
||||
return cancel_wakeup_output (Guard);
|
||||
{
|
||||
if (!iFlushPacketQueue())
|
||||
return cancel_wakeup_output(Guard);
|
||||
|
||||
send_len = m_OutBuffer->length ();
|
||||
}
|
||||
|
||||
#ifdef MSG_NOSIGNAL
|
||||
ssize_t n = peer ().send (m_OutBuffer->rd_ptr (), send_len, MSG_NOSIGNAL);
|
||||
@@ -960,7 +972,7 @@ int WorldSocket::HandlePing (WorldPacket& recvPacket)
|
||||
int WorldSocket::iSendPacket (const WorldPacket& pct)
|
||||
{
|
||||
ServerPktHeader header(pct.size()+2, pct.GetOpcode());
|
||||
if (m_OutBuffer->space () < pct.size () + header.getHeaderLength())
|
||||
if (m_OutBuffer->space() < pct.size () + header.getHeaderLength())
|
||||
{
|
||||
errno = ENOBUFS;
|
||||
return -1;
|
||||
@@ -978,16 +990,65 @@ int WorldSocket::iSendPacket (const WorldPacket& pct)
|
||||
return 0;
|
||||
}
|
||||
|
||||
bool WorldSocket::iFlushPacketQueue ()
|
||||
/// Return 1 if some bytes written and packet completed
|
||||
/// Return 0 if no byte written, but still remaining data
|
||||
/// Return -1 if some bytes written and still remaining data
|
||||
int WorldSocket::iSendPartialPacket(WorldPacket& pct)
|
||||
{
|
||||
size_t remainingLen = pct.size() - pct.rpos();
|
||||
|
||||
if (pct.rpos() == 0) // nothing sent yet => send header
|
||||
{
|
||||
ServerPktHeader header(pct.size()+2, pct.GetOpcode());
|
||||
|
||||
// check if there is at least enough space for the header
|
||||
if (m_OutBuffer->space () < header.getHeaderLength())
|
||||
return 0; // nothing written, still remaining data => 0
|
||||
|
||||
m_Crypt.EncryptSend(header.header, header.getHeaderLength());
|
||||
|
||||
if (m_OutBuffer->copy((char*) header.header, header.getHeaderLength()) == -1)
|
||||
return 0; // no space left. This should not happend. Just return 0 as nothing written and still remaining data
|
||||
|
||||
if (remainingLen == 0)
|
||||
return 1; // header written, nothing else => 1
|
||||
}
|
||||
|
||||
if ((m_OutBuffer->space()) < remainingLen) {
|
||||
size_t len = m_OutBuffer->space();
|
||||
|
||||
if (m_OutBuffer->copy((char*) (pct.contents() + pct.rpos()), len) == -1)
|
||||
ACE_ASSERT (false);
|
||||
|
||||
pct.read_skip(len);
|
||||
return -1; // packet will be pushed back on the queue
|
||||
}
|
||||
|
||||
if (m_OutBuffer->copy((char*) (pct.contents() + pct.rpos()), remainingLen) == -1)
|
||||
ACE_ASSERT (false);
|
||||
|
||||
return 1; // some byte written and packet completed
|
||||
}
|
||||
|
||||
|
||||
bool WorldSocket::iFlushPacketQueue()
|
||||
{
|
||||
WorldPacket *pct;
|
||||
bool haveone = false;
|
||||
|
||||
while (m_PacketQueue.dequeue_head (pct) == 0)
|
||||
while (m_PacketQueue.dequeue_head(pct) == 0)
|
||||
{
|
||||
if (iSendPacket (*pct) == -1)
|
||||
int result = iSendPartialPacket(*pct);
|
||||
|
||||
if (result != 0)
|
||||
{
|
||||
if (m_PacketQueue.enqueue_head (pct) == -1)
|
||||
// some bytes were written
|
||||
haveone = true;
|
||||
}
|
||||
|
||||
if (result <= 0)
|
||||
{
|
||||
if (m_PacketQueue.enqueue_head(pct) == -1)
|
||||
{
|
||||
delete pct;
|
||||
sLog.outError ("WorldSocket::iFlushPacketQueue m_PacketQueue->enqueue_head");
|
||||
@@ -998,7 +1059,7 @@ bool WorldSocket::iFlushPacketQueue ()
|
||||
}
|
||||
else
|
||||
{
|
||||
haveone = true;
|
||||
// packet completed
|
||||
delete pct;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user