TooTallNate#171

fix 100% CPU issue

Author: Eugene-Kudelevsky

src/main/java/org/java_websocket/SSLSocketChannel2.java

@@ -5,6 +5,13 @@
  */
 package org.java_websocket;
 
+import javax.net.ssl.SSLEngine;
+import javax.net.ssl.SSLEngineResult;
+import javax.net.ssl.SSLEngineResult.HandshakeStatus;
+import javax.net.ssl.SSLEngineResult.Status;
+import javax.net.ssl.SSLException;
+import javax.net.ssl.SSLSession;
+import java.io.EOFException;
 import java.io.IOException;
 import java.net.Socket;
 import java.net.SocketAddress;
@@ -20,13 +27,6 @@
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Future;
 
-import javax.net.ssl.SSLEngine;
-import javax.net.ssl.SSLEngineResult;
-import javax.net.ssl.SSLEngineResult.HandshakeStatus;
-import javax.net.ssl.SSLEngineResult.Status;
-import javax.net.ssl.SSLException;
-import javax.net.ssl.SSLSession;
-
 /**
  * Implements the relevant portions of the SocketChannel interface with the SSLEngine wrapper.
  */
@@ -213,6 +213,9 @@ public int write( ByteBuffer src ) throws IOException {
 		//	createBuffers( sslEngine.getSession() );
 		//}
 		int num = socketChannel.write( wrap( src ) );
+        if (writeEngineResult.getStatus() == SSLEngineResult.Status.CLOSED) {
+            throw new EOFException("Connection is closed");
+        }
 		return num;
 
 	}
@@ -286,6 +289,9 @@ private int readRemaining( ByteBuffer dst ) throws SSLException {
 		if( inCrypt.hasRemaining() ) {
 			unwrap();
 			int amount = transfereTo( inData, dst );
+            if (readEngineResult.getStatus() == SSLEngineResult.Status.CLOSED) {
+                return -1;
+            }
 			if( amount > 0 )
 				return amount;
 		}