From 7b0fabdbc4ff0d3d88faa4f13e3e6d4417104a94 Mon Sep 17 00:00:00 2001
From: Dirk <dirk@testssl.sh>
Date: Tue, 8 Mar 2016 10:38:21 +0100
Subject: [PATCH] - making the read buffer for server hello bigger+variable

---
 utils/ccs-injection.bash | 5 +++--
 utils/heartbleed.bash    | 5 +++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/utils/ccs-injection.bash b/utils/ccs-injection.bash
index 7687d93..4d7b53f 100755
--- a/utils/ccs-injection.bash
+++ b/utils/ccs-injection.bash
@@ -25,6 +25,7 @@ MAXSLEEP=10
 OCKREPLY=""
 COL_WIDTH=32
 DEBUG=${DEBUG:-0}
+HELLO_READBYTES=${HELLO_READBYTES:-65535}
 
 TLSV=${2:-01}
 # TLS 1.0=x01  1.1=0x02, 1.2=0x3
@@ -305,8 +306,8 @@ echo "##### sending standard client hello with TLS version 03,$TLSV:"
 socksend "$msg" $TLSV
 sleep 1
 
-sockread 16384
-echo "##### reading server hello:"
+sockread $HELLO_READBYTES
+echo "##### reading server hello ($HELLO_READBYTES bytes):"
 if test $DEBUG ; then
      echo "$SOCKREPLY" | xxd -c$COL_WIDTH | head -10
      echo "[...]"
diff --git a/utils/heartbleed.bash b/utils/heartbleed.bash
index cebf85f..c2bd122 100755
--- a/utils/heartbleed.bash
+++ b/utils/heartbleed.bash
@@ -23,6 +23,7 @@ MAXSLEEP=10
 SOCKREPLY=""
 COL_WIDTH=32
 DEBUG=${DEBUG:-0}
+HELLO_READBYTES=${HELLO_READBYTES:-65535}
 
 # TLS 1.0=x01  1.1=0x02, 1.2=0x3
 TLSV=${2:-01}
@@ -272,8 +273,8 @@ fd_socket $PORT
 echo "##### sending standard client hello with TLS version 03,$TLSV:"
 socksend "$msg" $TLSV
 
-sockread 16384 
-echo "##### reading server hello:"
+sockread $HELLO_READBYTES
+echo "##### reading server hello ($HELLO_READBYTES bytes):"
 echo "$SOCKREPLY" | xxd -c$COL_WIDTH | head -10
 echo "[...]"
 echo