From c91312a1a2e005224af729f951cb2c71b98c9ac0 Mon Sep 17 00:00:00 2001
From: Bjørn Christian Seime <bjorncs@oath.com>
Date: Mon, 5 Nov 2018 16:48:30 +0100
Subject: Verify handshake is complete in write() + flush()

---
 jrt/src/com/yahoo/jrt/TlsCryptoSocket.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/jrt/src/com/yahoo/jrt/TlsCryptoSocket.java b/jrt/src/com/yahoo/jrt/TlsCryptoSocket.java
index 93403f188df..39cd9f6c4f2 100644
--- a/jrt/src/com/yahoo/jrt/TlsCryptoSocket.java
+++ b/jrt/src/com/yahoo/jrt/TlsCryptoSocket.java
@@ -154,6 +154,7 @@ public class TlsCryptoSocket implements CryptoSocket {
 
     @Override
     public int write(ByteBuffer src) throws IOException {
+        verifyHandshakeCompleted();
         if (flush() == FlushResult.NEED_WRITE) return 0;
         int totalBytesWrapped = 0;
         int bytesWrapped;
@@ -166,6 +167,7 @@ public class TlsCryptoSocket implements CryptoSocket {
 
     @Override
     public FlushResult flush() throws IOException {
+        verifyHandshakeCompleted();
         channelWrite();
         return wrapBuffer.bytes() > 0 ? FlushResult.NEED_WRITE : FlushResult.DONE;
     }
-- 
cgit v1.2.3