From 42431690e04a8614e759e0a3a3bcd76e30e6207e Mon Sep 17 00:00:00 2001
From: Martin Willi <martin@revosec.ch>
Date: Mon, 23 Mar 2015 11:37:27 +0100
Subject: diffie-hellman: Add a bool return value to get_my_public_value()

---
 scripts/dh_speed.c | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'scripts/dh_speed.c')

diff --git a/scripts/dh_speed.c b/scripts/dh_speed.c
index 8a782d80b..21652e787 100644
--- a/scripts/dh_speed.c
+++ b/scripts/dh_speed.c
@@ -15,6 +15,7 @@
 
 #include <stdio.h>
 #include <time.h>
+#include <assert.h>
 #include <library.h>
 #include <utils/debug.h>
 #include <crypto/diffie_hellman.h>
@@ -88,12 +89,12 @@ static void run_test(diffie_hellman_group_t group, int rounds)
 
 	for (round = 0; round < rounds; round++)
 	{
-		l[round]->get_my_public_value(l[round], &chunk);
+		assert(l[round]->get_my_public_value(l[round], &chunk));
 		r->set_other_public_value(r, chunk);
 		chunk_free(&chunk);
 	}
 
-	r->get_my_public_value(r, &chunk);
+	assert(r->get_my_public_value(r, &chunk));
 	start_timing(&timing);
 	for (round = 0; round < rounds; round++)
 	{
-- 
cgit v1.2.3