From a3c85f3b0b6499a07828d9cdeccfe8e7db3fe9b3 Mon Sep 17 00:00:00 2001
From: Rashed Sourov <sourov0805045@gmail.com>
Date: Tue, 28 Jul 2020 14:36:38 +0600
Subject: [PATCH] Wait and try to read 2 bytes header

---
 websocket-sharp.clone/WebSocketStreamReader.cs | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/websocket-sharp.clone/WebSocketStreamReader.cs b/websocket-sharp.clone/WebSocketStreamReader.cs
index 5613272f3..c4aaf362f 100644
--- a/websocket-sharp.clone/WebSocketStreamReader.cs
+++ b/websocket-sharp.clone/WebSocketStreamReader.cs
@@ -119,24 +119,23 @@ private async Task<StreamReadInfo> GetStreamReadInfo(WebSocketFrameHeader header
         private async Task<WebSocketFrameHeader> ReadHeader(CancellationToken cancellationToken)
         {
             cancellationToken.ThrowIfCancellationRequested();
-            var header = new byte[2];
+            byte[] header;
 
-            var headerLength = 0;
             try
             {
-                headerLength = await _innerStream.ReadAsync(header, 0, 2, cancellationToken).ConfigureAwait(false);
+                header = await _innerStream.ReadBytes(2).ConfigureAwait(false);
             }
             catch (IOException)
             {
                 return null;
             }
 
-            if (headerLength == 0)
+            if (header.Length == 0)
             {
                 return null;
             }
 
-            if (headerLength != 2)
+            if (header.Length != 2)
             {
                 throw new WebSocketException("The header part of a frame cannot be read from the data source.");
             }