[refactor] ssl handshake wrap-ing - do tasks eagerly

Author: kares

src/main/java/org/jruby/ext/openssl/SSLSocket.java

@@ -586,10 +586,8 @@ private IRubyObject doHandshake(final boolean blocking, final boolean exception)
                 if ( netData.hasRemaining() ) {
                     while ( flushData(blocking) ) { /* loop */ }
                 }
-                netData.clear();
-                SSLEngineResult result = engine.wrap(dummy, netData);
-                handshakeStatus = result.getHandshakeStatus();
-                netData.flip();
+                assert !netData.hasRemaining();
+                doWrap(blocking);
                 flushData(blocking);
                 break;
             default:
@@ -598,6 +596,18 @@ private IRubyObject doHandshake(final boolean blocking, final boolean exception)
         }
     }
 
+    private void doWrap(boolean blocking) throws IOException {
+        netData.clear();
+        SSLEngineResult result = engine.wrap(dummy, netData);
+        netData.flip();
+        handshakeStatus = result.getHandshakeStatus();
+        status = result.getStatus();
+        if (handshakeStatus == SSLEngineResult.HandshakeStatus.NEED_TASK
+                && status == SSLEngineResult.Status.OK) {
+            doTasks();
+        }
+    }
+
     private void doTasks() {
         Runnable task;
         while ((task = engine.getDelegatedTask()) != null) {