fix WebSocket.cs

Author: sta

websocket-sharp.userprefs

@@ -2,7 +2,7 @@
   <MonoDevelop.Ide.Workspace ActiveConfiguration="Release_Ubuntu" ctype="Workspace" />
   <MonoDevelop.Ide.Workbench ActiveDocument="websocket-sharp/WebSocket.cs" ctype="Workbench">
     <Files>
-      <File FileName="websocket-sharp/WebSocket.cs" Line="5" Column="54" />
+      <File FileName="websocket-sharp/WebSocket.cs" Line="231" Column="1" />
     </Files>
   </MonoDevelop.Ide.Workbench>
   <MonoDevelop.Ide.DebuggingService.Breakpoints>

websocket-sharp/WebSocket.cs

@@ -219,17 +219,15 @@ private void close(WsState state)
         }
         catch (Exception e)
         {
+          if (OnError != null)
+          {
+            OnError(this, e.Message);
+          }
 #if DEBUG
           Console.WriteLine("WS: Error @close: {0}", e.Message);
 #endif
         }
       }
-
-      if (!(Thread.CurrentThread.IsBackground) &&
-          msgThread != null && msgThread.IsAlive)
-      {
-        msgThread.Join();
-      }
 
       if (wsStream != null)
       {
@@ -363,14 +361,11 @@ private void message()
 #endif
       while (readyState == WsState.OPEN)
       {
-        while (readyState == WsState.OPEN && netStream.DataAvailable)
-        {
-          data = receive();
+        data = receive();
 
-          if (OnMessage != null && data != null)
-          {
-            OnMessage(this, data);
-          }
+        if (OnMessage != null && data != null)
+        {
+          OnMessage(this, data);
         }
       }
 #if DEBUG
@@ -433,15 +428,18 @@ private string receive()
       }
       catch (Exception e)
       {
-        if (OnError != null)
+        if (readyState == WsState.OPEN)
         {
-          OnError(this, e.Message);
-        }
+          if (OnError != null)
+          {
+            OnError(this, e.Message);
+          }
 
-        ReadyState = WsState.CLOSED;
+          ReadyState = WsState.CLOSING;
 #if DEBUG
-        Console.WriteLine("WS: Error @receive: {0}", e.Message);
+          Console.WriteLine("WS: Error @receive: {0}", e.Message);
 #endif
+        }
       }
 
       return null;