From f166c543ffc65c11bd46e6d84ac283ecd2534368 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Manuel=20P=C3=A9gouri=C3=A9-Gonnard?= <mpg@elzevir.fr>
Date: Mon, 1 Dec 2014 11:30:56 +0100
Subject: [PATCH] Adapt memory.sh to config-suite-b

---
 .gitignore        |  3 ++
 scripts/memory.sh | 95 +++++++++++++++++++++++++++++++++++++++++++----
 2 files changed, 90 insertions(+), 8 deletions(-)

diff --git a/.gitignore b/.gitignore
index 916d7311c..fee2a31cd 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,6 +7,9 @@ Coverage
 *.gcno
 *.gcda
 
+# generated by scripts/memory.sh
+massif-*
+
 # MSVC files generated by CMake:
 /*.sln
 /*.vcxproj
diff --git a/scripts/memory.sh b/scripts/memory.sh
index 53b20e500..fb178eb93 100755
--- a/scripts/memory.sh
+++ b/scripts/memory.sh
@@ -1,7 +1,10 @@
 #!/bin/sh
 
 # Measure memory usage of a minimal client using a small configuration
-# Currently hardwired to the ccm-psk configuration, may be expanded later
+# Currently hardwired to ccm-psk and suite-b, may be expanded later
+#
+# Use different build options for measuring executable size and memory usage,
+# since for memory we want debug information.
 
 set -eu
 
@@ -13,12 +16,39 @@ if [ -r $CONFIG_H ]; then :; else
     exit 1
 fi
 
+if grep -i cmake Makefile >/dev/null; then
+    echo "Not compatible with CMake" >&2
+    exit 1
+fi
+
+if git status | grep -F $CONFIG_H >/dev/null 2>&1; then
+    echo "config.h not clean" >&2
+    exit 1
+fi
+
+# preparation
+
 CONFIG_BAK=${CONFIG_H}.bak
 cp $CONFIG_H $CONFIG_BAK
 
+rm -f massif.out.*
+
+printf "building server... "
+
+make clean
+make lib >/dev/null 2>&1
+(cd programs && make ssl/ssl_server2) >/dev/null
+cp programs/ssl/ssl_server2 .
+
+echo "done"
+
+# first config
+
+echo ""
+echo "config-ccm-psk-tls1_2:"
 cp configs/config-ccm-psk-tls1_2.h $CONFIG_H
 
-printf "Executable size... "
+printf "    Executable size... "
 
 make clean
 CFLAGS=-fno-asynchronous-unwind-tables make OFLAGS=-Os lib >/dev/null 2>&1
@@ -28,21 +58,19 @@ strip ssl/$CLIENT
 stat -c'%s' ssl/$CLIENT
 cd ..
 
-printf "Peak ram usage... "
+printf "    Peak ram usage... "
 
 make clean
 CFLAGS=-g3 make OFLAGS=-Os lib >/dev/null 2>&1
 cd programs
-CFLAGS=-g3 make OFLAGS=-Os ssl/$CLIENT ssl/ssl_server2 >/dev/null
+CFLAGS=-g3 make OFLAGS=-Os ssl/$CLIENT >/dev/null
 cd ..
 
-rm -f massif.out.*
-
-programs/ssl/ssl_server2 psk=000102030405060708090A0B0C0D0E0F >/dev/null &
+./ssl_server2 psk=000102030405060708090A0B0C0D0E0F >/dev/null &
 SRV_PID=$!
 sleep 1;
 
-if valgrind --tool=massif --stacks=yes programs/ssl/$CLIENT > /dev/null 2>&1
+if valgrind --tool=massif --stacks=yes programs/ssl/$CLIENT >/dev/null 2>&1
 then
     FAILED=0
 else
@@ -54,8 +82,59 @@ kill $SRV_PID
 wait $SRV_PID
 
 scripts/massif_max.pl massif.out.*
+mv massif.out.* massif-ccm-psk.$$
+
+# second config
+
+echo ""
+echo "config-suite-b:"
+cp configs/config-suite-b.h $CONFIG_H
+scripts/config.pl unset POLARSSL_BASE64_C
+scripts/config.pl unset POLARSSL_PEM_PARSE_C
+scripts/config.pl unset POLARSSL_CERTS_C
+
+printf "    Executable size... "
+
+make clean
+CFLAGS=-fno-asynchronous-unwind-tables make OFLAGS=-Os lib >/dev/null 2>&1
+cd programs
+CFLAGS=-fno-asynchronous-unwind-tables make OFLAGS=-Os ssl/$CLIENT >/dev/null
+strip ssl/$CLIENT
+stat -c'%s' ssl/$CLIENT
+cd ..
+
+printf "    Peak ram usage... "
+
+make clean
+CFLAGS=-g3 make OFLAGS=-Os lib >/dev/null 2>&1
+cd programs
+CFLAGS=-g3 make OFLAGS=-Os ssl/$CLIENT >/dev/null
+cd ..
 
 rm -f massif.out.*
+
+./ssl_server2 >/dev/null &
+SRV_PID=$!
+sleep 1;
+
+if valgrind --tool=massif --stacks=yes programs/ssl/$CLIENT >/dev/null 2>&1
+then
+    FAILED=0
+else
+    echo "client failed" >&2
+    FAILED=1
+fi
+
+kill $SRV_PID
+wait $SRV_PID
+
+scripts/massif_max.pl massif.out.*
+mv massif.out.* massif-suite-b.$$
+
+# cleanup
+
 mv $CONFIG_BAK $CONFIG_H
+make clean
+rm ssl_server2
 
 exit $FAILED