[Modify] Polish it

8 years ago · e5accd3e69
parent f3d16101f5
commit e5accd3e69
1 changed files with 6 additions and 1 deletions
--- a/websocket-sharp/Net/WebSockets/TcpListenerWebSocketContext.cs
+++ b/websocket-sharp/Net/WebSockets/TcpListenerWebSocketContext.cs
@ -481,7 +481,12 @@ namespace WebSocketSharp.Net.WebSockets

    internal void Close (HttpStatusCode code)
    {
-      _websocket.Close (HttpResponse.CreateCloseResponse (code));
+      var res = HttpResponse.CreateCloseResponse (code);
+      var bytes = res.ToByteArray ();
+      _stream.Write (bytes, 0, bytes.Length);
+
+      _stream.Close ();
+      _tcpClient.Close ();
    }

    internal void SendAuthenticationChallenge (string challenge)