procxx
/
kepka
mirror of https://github.com/procxx/kepka.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Fix socket packet reading.

This commit is contained in:
John Preston 2019-07-10 13:16:07 +02:00
parent 33fce38f90
commit c894ce30c4
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Telegram/SourceFiles/mtproto/connection_tcp.cpp
View File

@ -307,6 +307,10 @@ void TcpConnection::socketRead() {
_leftBytes -= readCount; _leftBytes -= readCount;
if (!_leftBytes) { if (!_leftBytes) {
socketPacket(full.subspan(0, _readBytes)); socketPacket(full.subspan(0, _readBytes));
if (!_socket || !_socket->isConnected()) {
return;
}
_usingLargeBuffer = false; _usingLargeBuffer = false;
_largeBuffer.clear(); _largeBuffer.clear();
_offsetBytes = _readBytes = 0; _offsetBytes = _readBytes = 0;
@ -331,6 +335,10 @@ void TcpConnection::socketRead() {
return; return;
} else if (available.size() >= packetSize) { } else if (available.size() >= packetSize) {
socketPacket(available.subspan(0, packetSize)); socketPacket(available.subspan(0, packetSize));
if (!_socket || !_socket->isConnected()) {
return;
}
available = available.subspan(packetSize); available = available.subspan(packetSize);
_offsetBytes += packetSize; _offsetBytes += packetSize;
_readBytes -= packetSize; _readBytes -= packetSize;