From e91b12161aeb4066e117cfe9a1c52e6b0e25accf Mon Sep 17 00:00:00 2001
From: sta <sta.blockhead@gmail.com>
Date: Mon, 17 Jun 2019 20:10:30 +0900
Subject: [PATCH] [Modify] Polish it

---
 websocket-sharp/Ext.cs | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/websocket-sharp/Ext.cs b/websocket-sharp/Ext.cs
index 7e90f02a..07bc0579 100644
--- a/websocket-sharp/Ext.cs
+++ b/websocket-sharp/Ext.cs
@@ -752,16 +752,23 @@ namespace WebSocketSharp
         ar => {
           try {
             var nread = stream.EndRead (ar);
-            if (nread == 0 && retry < _retry) {
-              retry++;
-              stream.BeginRead (buff, offset, length, callback, null);
+            if (nread == 0) {
+              if (retry < _retry) {
+                retry++;
+                stream.BeginRead (buff, offset, length, callback, null);
+
+                return;
+              }
+
+              if (completed != null)
+                completed (buff.SubArray (0, offset));
 
               return;
             }
 
-            if (nread == 0 || nread == length) {
+            if (nread == length) {
               if (completed != null)
-                completed (buff.SubArray (0, offset + nread));
+                completed (buff);
 
               return;
             }