summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorBjørn Christian Seime <bjorncs@oath.com>2018-11-05 16:48:30 +0100
committerBjørn Christian Seime <bjorncs@oath.com>2018-11-05 16:48:30 +0100
commitc91312a1a2e005224af729f951cb2c71b98c9ac0 (patch)
tree36beab042d4df8c9368f17709e89dca87ca46e77
parent1c3afda78a9dcfa7e5aadc4c31910002a567ab8f (diff)
Verify handshake is complete in write() + flush()
-rw-r--r--jrt/src/com/yahoo/jrt/TlsCryptoSocket.java2
1 files changed, 2 insertions, 0 deletions
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;
}