From: Nikos Mavrogiannopoulos <nmav@gnutls.org>
Date: Sat, 19 Jun 2010 09:20:40 +0000 (+0200)
Subject: Allow setting debug level via cmd.
X-Git-Tag: gnutls_2_11_3~168
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=4774cbdfdb94db66e5eb52e5195573622903a37e;p=thirdparty%2Fgnutls.git

Allow setting debug level via cmd.
---

diff --git a/src/benchmark.c b/src/benchmark.c
index cd2d9fcdad..e3be4c4b50 100644
--- a/src/benchmark.c
+++ b/src/benchmark.c
@@ -34,6 +34,12 @@ static unsigned char data[64 * 1024];
 
 #define TOTAL_ITER 8*1024
 
+static void
+tls_log_func (int level, const char *str)
+{
+  fprintf (stderr, "|<%d>| %s", level, str);
+}
+
 static void
 cipher_bench (int algo, int size)
 {
@@ -64,7 +70,6 @@ cipher_bench (int algo, int size)
   key.data = _key;
   key.size = keysize;
 
-  gnutls_global_init ();
 
   printf ("Checking %s (%dkb payload)... ", gnutls_cipher_get_name (algo),
 	  size);
@@ -117,8 +122,6 @@ mac_bench (int algo, int size)
     return;
   memset (_key, 0xf0, blocksize);
 
-  gnutls_global_init ();
-
   printf ("Checking %s (%dkb payload)... ", gnutls_mac_get_name (algo), size);
   fflush (stdout);
   gettime (&start);
@@ -143,8 +146,17 @@ mac_bench (int algo, int size)
 }
 
 int
-main (void)
+main (int argc, char** argv)
 {
+  int debug_level = 0;
+
+  if (argc > 1)
+    debug_level = 2;
+
+  gnutls_global_set_log_function (tls_log_func);
+  gnutls_global_set_log_level (debug_level);
+  gnutls_global_init ();
+
   mac_bench (GNUTLS_MAC_SHA1, 4);
   mac_bench (GNUTLS_MAC_SHA1, 8);
   mac_bench (GNUTLS_MAC_SHA1, 16);