public void TestParse_16bPayloadLength_Masked()
{
// 0xC2 = 1100 0010
// FRRR opco
// ISSS de
// NVVV
// 123
var buffer = new byte[]
{
0xC2, 128 + 126,
0x08, 0, // 16b length in Big-Endian = 0x0800 = 2048
5, 6, 7, 8 // 32b mask
};
var payload = new DummyDataStream(2048);
var header = new ByteArray(buffer);
var parser = new WebSocketMessage.Parser();
parser.ParseHeader(header);
var binMsg = new StreamList();
binMsg.Add(header);
binMsg.Add(payload);
var msg = parser.Close(binMsg);
Assert.True(msg.FIN);
Assert.True(msg.RSV1);
Assert.False(msg.RSV2);
Assert.False(msg.RSV3);
Assert.Equal(WebSocketMessage.OpcodeType.Binary, msg.Opcode);
Assert.Equal(2048, msg.PayloadLength);
Assert.True(msg.MASK);
Assert.Equal(4, msg.MaskingKey.Length);
Assert.Equal(6, msg.MaskingKey[1]);
}
public void TestParse_64bPayloadLength_NoMask()
{
// 0xC1 = 1100 0001
// FRRR opco
// ISSS de
// NVVV
// 123
var buffer = new byte[]
{
0xC1, 127,
0, 0, 0, 0, 0, 1, 0, 0, // 64b length in Big-Endian = 0x10000 = 65 536
};
var payload = new DummyDataStream(2048);
var header = new ByteArray(buffer);
var parser = new WebSocketMessage.Parser();
parser.ParseHeader(header);
var binMsg = new StreamList();
binMsg.Add(header);
binMsg.Add(payload);
Assert.Throws(typeof(ArgumentOutOfRangeException), () => parser.Close(binMsg));
binMsg.Add(new DummyDataStream(65536 - 2048));
var msg = parser.Close(binMsg);
Assert.True(msg.FIN);
Assert.True(msg.RSV1);
Assert.False(msg.RSV2);
Assert.False(msg.RSV3);
Assert.Equal(WebSocketMessage.OpcodeType.Text, msg.Opcode);
Assert.Equal(65536, msg.PayloadLength);
Assert.False(msg.MASK);
}
public void TestParse_7bPayloadLength_NoMask()
{
// 0xC2 = 1100 0010
// FRRR opco
// ISSS de
// NVVV
// 123
var buffer = new byte[] { 0xC2, 10 };
var payload = new DummyDataStream(10);
var header = new ByteArray(buffer);
var parser = new WebSocketMessage.Parser();
parser.ParseHeader(header);
var binMsg = new StreamList();
binMsg.Add(header);
binMsg.Add(payload);
var msg = parser.Close(binMsg);
Assert.True(msg.FIN);
Assert.True(msg.RSV1);
Assert.False(msg.RSV2);
Assert.False(msg.RSV3);
Assert.Equal(WebSocketMessage.OpcodeType.Binary, msg.Opcode);
Assert.Equal(10, msg.PayloadLength);
Assert.False(msg.MASK);
}
/// <summary>
/// Push new data into the parser
/// </summary>
/// <param name="s"></param>
public void Receive(IDataStream s)
{
lock (contentLock)
{
while (s.Length > 0)
{
while (!readingPayload && s.Length > 0)
{
missing = parser.ParseHeader(s);
if (missing <= 0)
{
var inner = s;
if (missing < 0)
{
var split = s.Length + missing;
inner = new StreamSegment(s, 0, split);
s = new StreamSegment(s, split);
}
else
{
s = EmptyData.Instance;
}
var ret = parser.Close(inner);
parser = new WebSocketMessage.Parser();
onMessageParsed(ret);
}
else
{
readingPayload = true;
dataBuilder.Append(s);
return;
}
}
if (readingPayload)
{
if (s.Length < missing)
{
dataBuilder.Append(s);
missing -= s.Length;
return;
}
else
{
dataBuilder.Append(new StreamSegment(s, 0, missing));
s = new StreamSegment(s, missing);
missing = 0;
readingPayload = false;
var bin = dataBuilder.Close();
dataBuilder = new ThresholdedDataBuilder(logger);
var msg = parser.Close(bin);
parser = new WebSocketMessage.Parser();
onMessageParsed(msg);
}
}
}
}
}
