[Modify] Polish it

This commit is contained in:
sta
2021-12-08 19:55:29 +09:00
parent a06e8ea4c2
commit a0104e901b

View File

@@ -699,32 +699,39 @@ Extended Payload Length: {7}
) )
{ {
var exactLen = frame.ExactPayloadLength; var exactLen = frame.ExactPayloadLength;
if (exactLen > PayloadData.MaxLength) { if (exactLen > PayloadData.MaxLength) {
var msg = "A frame has too long payload length."; var msg = "A frame has too long payload length.";
throw new WebSocketException (CloseStatusCode.TooBig, msg); throw new WebSocketException (CloseStatusCode.TooBig, msg);
} }
if (exactLen == 0) { if (exactLen == 0) {
frame._payloadData = PayloadData.Empty; frame._payloadData = PayloadData.Empty;
completed (frame); completed (frame);
return; return;
} }
var len = (long) exactLen; var len = (long) exactLen;
Action<byte[]> comp = Action<byte[]> comp =
bytes => { bytes => {
if (bytes.LongLength != len) { if (bytes.LongLength != len) {
var msg = "The payload data of a frame could not be read."; var msg = "The payload data of a frame could not be read.";
throw new WebSocketException (msg); throw new WebSocketException (msg);
} }
frame._payloadData = new PayloadData (bytes, len); frame._payloadData = new PayloadData (bytes, len);
completed (frame); completed (frame);
}; };
if (frame._payloadLength < 127) { if (frame._payloadLength < 127) {
stream.ReadBytesAsync ((int) exactLen, comp, error); stream.ReadBytesAsync ((int) exactLen, comp, error);
return; return;
} }