Fixing #285

Author: darkl

src/net45/Extensions/WampSharp.RawSocket/RawSocket/NetworkStreamExtensions.cs

@@ -1,6 +1,7 @@
 using System;
 using System.IO;
 using System.Threading.Tasks;
+using Microsoft.IO;
 
 namespace WampSharp.RawSocket
 {
@@ -42,5 +43,18 @@ public static byte[] GetBuffer(this MemoryStream stream)
         }
 
 #endif
+        public static byte[] GetBufferWorkaround(this MemoryStream stream)
+        {
+            RecyclableMemoryStream memoryStream = stream as RecyclableMemoryStream;
+
+            if (memoryStream != null)
+            {
+                return memoryStream.GetBuffer();
+            }
+
+            return stream.GetBuffer();
+        }
+
+
     }
 }

src/net45/Extensions/WampSharp.RawSocket/RawSocket/TcpClientConnection.cs

@@ -65,7 +65,7 @@ protected async override Task SendAsync(WampMessage<object> message)
 
                 byte[] buffer;
 
-                buffer = memoryStream.GetBuffer();
+                buffer = memoryStream.GetBufferWorkaround();
 
                 // Write a message header
                 mFrameHeaderParser.WriteHeader(FrameType.WampMessage, messageLength, buffer);
@@ -151,7 +151,7 @@ private async Task<MemoryStream> ReadStream(int messageLength, int position = 0)
 
             MemoryStream stream = mByteArrayPool.GetStream(Tag, length, true);
 
-            byte[] buffer = stream.GetBuffer();
+            byte[] buffer = stream.GetBufferWorkaround();
 
             await Stream
                 .ReadExactAsync(buffer, position, messageLength)
@@ -169,7 +169,7 @@ private async Task HandlePong(int messageLength)
             using (MemoryStream buffer = await ReadStream(messageLength).ConfigureAwait(false))
             {
                 ArraySegment<byte> arraySegment =
-                    new ArraySegment<byte>(buffer.GetBuffer(), 0, messageLength);
+                    new ArraySegment<byte>(buffer.GetBufferWorkaround(), 0, messageLength);
 
                 mPinger.RaiseOnPong(arraySegment);
             }
@@ -179,7 +179,7 @@ private async Task HandlePing(int messageLength)
         {
             using (MemoryStream memoryStream = await ReadStream(messageLength, FrameHeaderSize).ConfigureAwait(false))
             {
-                byte[] buffer = memoryStream.GetBuffer();
+                byte[] buffer = memoryStream.GetBufferWorkaround();
 
                 mFrameHeaderParser.WriteHeader(FrameType.Pong, messageLength, buffer);
 
@@ -197,7 +197,7 @@ private async Task SendPing(byte[] message)
 
             using (MemoryStream memoryStream = mByteArrayPool.GetStream(Tag, frameSize, true))
             {
-                byte[] buffer = memoryStream.GetBuffer();
+                byte[] buffer = memoryStream.GetBufferWorkaround();
 
                 mFrameHeaderParser.WriteHeader(FrameType.Ping, message.Length, buffer);
                 memoryStream.SetLength(FrameHeaderSize);