arek/arstomp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

fix: Check for 13/10 EOL in heartbeat frame

Browse Source
This commit is contained in:
Arek
2021-06-27 16:23:56 +02:00
parent 8b7d743dad
commit 6884d5715f
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
arstomp/src/Helpers.cs
View File

@@ -86,14 +86,21 @@ namespace ArStomp
var inputstream = new MemoryStream(msgBuffer); var inputstream = new MemoryStream(msgBuffer);
var bodyoutput = new MemoryStream(); var bodyoutput = new MemoryStream();
if (inputstream.ReadByte() == 10) var firstByte = inputstream.ReadByte();
if (firstByte == 10)
{ {
return HeartbeatFrame; return HeartbeatFrame;
} }
else else if (firstByte == 13) {
{ var secondByte = inputstream.ReadByte();
inputstream.Seek(0, SeekOrigin.Begin); if (secondByte == 10) {
return HeartbeatFrame;
} else {
throw new Exception("Invalid frame");
} }
}
// start from beginning
inputstream.Seek(0, SeekOrigin.Begin);
StreamReader reader = findBody(inputstream); StreamReader reader = findBody(inputstream);