hydrar/mistserver
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added socket lib strict mode spool, used in WebSocket::readFrame

Browse source
This commit is contained in:
Thulinma 2021-06-17 14:44:24 +02:00
parent 829c8b7948
commit fe6a0777a5
3 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
lib/websocket.cpp
View file

@ -130,7 +130,7 @@ namespace HTTP{
while (true){
// Check if we can receive the minimum frame size (2 header bytes, 0 payload)
if (!C.Received().available(2)){
if (C.spool()){continue;}
if (C.spool(true)){continue;}
return false;
}
std::string head = C.Received().copy(2);
@ -141,7 +141,7 @@ namespace HTTP{
if (headSize > 2){
// Check if we can receive the whole header
if (!C.Received().available(headSize)){
if (C.spool()){continue;}
if (C.spool(true)){continue;}
return false;
}
// Read entire header, re-read real payload length
@ -154,7 +154,7 @@ namespace HTTP{
}
// Check if we can receive the whole frame (header + payload)
if (!C.Received().available(headSize + payLen)){
if (C.spool()){continue;}
if (C.spool(true)){continue;}
return false;
}
C.Received().remove(headSize); // delete the header