From d594b1447fec66b970d84cb5efcc9fa42c5e96e7 Mon Sep 17 00:00:00 2001
From: Jan Eggers <jan.eggers.ferag@hotmail.com>
Date: Tue, 24 Oct 2017 12:45:26 +0200
Subject: [PATCH] fixed index out of range

---
 .../MQTTnet.NetStandard/Implementations/WebSocketStream.cs     | 1 +
 .../Implementations/WebSocketStream.cs                         | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/Frameworks/MQTTnet.NetStandard/Implementations/WebSocketStream.cs b/Frameworks/MQTTnet.NetStandard/Implementations/WebSocketStream.cs
index d38de6f..f037b38 100644
--- a/Frameworks/MQTTnet.NetStandard/Implementations/WebSocketStream.cs
+++ b/Frameworks/MQTTnet.NetStandard/Implementations/WebSocketStream.cs
@@ -41,6 +41,7 @@ namespace MQTTnet.Implementations
             {
                 var response = await _webSocket.ReceiveAsync(new ArraySegment<byte>(buffer, currentOffset, count), cancellationToken).ConfigureAwait(false);
                 currentOffset += response.Count;
+                count -= response.Count;
 
                 if (response.MessageType == WebSocketMessageType.Close)
                 {
diff --git a/Frameworks/MQTTnet.UniversalWindows/Implementations/WebSocketStream.cs b/Frameworks/MQTTnet.UniversalWindows/Implementations/WebSocketStream.cs
index 1843c6a..2a269c0 100644
--- a/Frameworks/MQTTnet.UniversalWindows/Implementations/WebSocketStream.cs
+++ b/Frameworks/MQTTnet.UniversalWindows/Implementations/WebSocketStream.cs
@@ -27,7 +27,8 @@ namespace MQTTnet.Implementations
             {
                 var response = await _webSocket.ReceiveAsync(new ArraySegment<byte>(buffer, currentOffset, count), cancellationToken).ConfigureAwait(false);
                 currentOffset += response.Count;
-                
+                count -= response.Count;
+
                 if (response.MessageType == WebSocketMessageType.Close)
                 {
                     await _webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, string.Empty, cancellationToken).ConfigureAwait(false);