Fuse MD5 and SHA1 benchmark into one

2 files changed, 40 insertions, 87 deletions

diff --git a/hardinfo2/arch/common/md5.h b/hardinfo2/arch/common/cryptohash.h
index d839778a..9897bb6b 100644
--- a/hardinfo2/arch/common/md5.h
+++ b/hardinfo2/arch/common/cryptohash.h
@@ -17,21 +17,48 @@
  */
 
 #include <md5.h>
+#include <sha1.h>
 
-static void
-benchmark_md5(void)
+static void inline md5_step(char *data, glong srclen)
 {
     struct MD5Context ctx;
     guchar checksum[16];
-    int i;
-    GTimer *timer = g_timer_new();
-    gdouble elapsed = 0;
-    gchar src[65536], *tmpsrc;
-    glong srclen = 65536;
+    
+    MD5Init(&ctx);
+    MD5Update(&ctx, (guchar *)data, srclen);
+    MD5Final(checksum, &ctx);
+}
 
-    tmpsrc = src;
+static void inline sha1_step(char *data, glong srclen)
+{
+    SHA1_CTX ctx;
+    guchar checksum[20];
+    
+    SHA1Init(&ctx);
+    SHA1Update(&ctx, (guchar*)data, srclen);
+    SHA1Final(checksum, &ctx);
+}
 
-    gchar *bdata_path;
+static gpointer cryptohash_for(unsigned int start, unsigned int end, void *data, GTimer *timer)
+{
+    unsigned int i;
+    
+    for (i = start; i <= end; i++) { 
+        if (i % 2 == 0) {
+            md5_step(data, 65536);
+        } else {
+            sha1_step(data, 65536);
+        }
+    }
+    
+    return NULL;
+}
+
+static void
+benchmark_cryptohash(void)
+{
+    gdouble elapsed = 0;
+    gchar *tmpsrc, *bdata_path;
     
     bdata_path = g_build_filename(params.path_data, "benchmark.data", NULL);
     if (!g_file_get_contents(bdata_path, &tmpsrc, NULL, NULL)) {
@@ -40,24 +67,12 @@ benchmark_md5(void)
     }     
     
     shell_view_set_enabled(FALSE);
-    shell_status_update("Generating MD5 sum for 312MiB of data...");
-    
-    for (i = 0; i <= 5000; i++) { 
-        g_timer_start(timer);
-
-        MD5Init(&ctx);
-        MD5Update(&ctx, (guchar*)tmpsrc, srclen);
-        MD5Final(checksum, &ctx);
-        
-        g_timer_stop(timer);
-        elapsed += g_timer_elapsed(timer, NULL);
+    shell_status_update("Running CryptoHash benchmark...");
         
-        shell_status_set_percentage(i/50);
-    }
+    elapsed = benchmark_parallel_for(0, 5000, cryptohash_for, tmpsrc);
     
-    g_timer_destroy(timer);
     g_free(bdata_path);
+    g_free(tmpsrc);
     
-    bench_results[BENCHMARK_MD5] = 312.0 / elapsed;
+    bench_results[BENCHMARK_CRYPTOHASH] = 312.0 / elapsed;
 }
-
diff --git a/hardinfo2/arch/common/sha1.h b/hardinfo2/arch/common/sha1.h
deleted file mode 100644
index c506375b..00000000
--- a/hardinfo2/arch/common/sha1.h
+++ /dev/null
@@ -1,62 +0,0 @@
-/*
- *    HardInfo - Displays System Information
- *    Copyright (C) 2003-2007 Leandro A. F. Pereira <leandro@hardinfo.org>
- *
- *    This program is free software; you can redistribute it and/or modify
- *    it under the terms of the GNU General Public License as published by
- *    the Free Software Foundation, version 2.
- *
- *    This program is distributed in the hope that it will be useful,
- *    but WITHOUT ANY WARRANTY; without even the implied warranty of
- *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- *    GNU General Public License for more details.
- *
- *    You should have received a copy of the GNU General Public License
- *    along with this program; if not, write to the Free Software
- *    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
- */
-#include <sha1.h>
-
-static void
-benchmark_sha1(void)
-{
-    SHA1_CTX ctx;
-    guchar checksum[20];
-    int i;
-    GTimer *timer = g_timer_new();
-    gdouble elapsed = 0;
-    gchar src[65536], *tmpsrc;
-    glong srclen = 65536;
-    
-    tmpsrc = src;
-
-    gchar *bdata_path;
-    
-    bdata_path = g_build_filename(params.path_data, "benchmark.data", NULL);
-    if (!g_file_get_contents(bdata_path, &tmpsrc, NULL, NULL)) {
-        g_free(bdata_path);
-        return;
-    }     
-    
-    shell_view_set_enabled(FALSE);
-    shell_status_update("Generating SHA1 sum for 312MiB of data...");
-    
-    for (i = 0; i <= 5000; i++) { 
-        g_timer_start(timer);
-
-        SHA1Init(&ctx);
-        SHA1Update(&ctx, (guchar*)tmpsrc, srclen);
-        SHA1Final(checksum, &ctx);
-        
-        g_timer_stop(timer);
-        elapsed += g_timer_elapsed(timer, NULL);
-        
-        shell_status_set_percentage(i/50);
-    }
-    
-    g_timer_destroy(timer);
-    g_free(bdata_path);
-    
-    bench_results[BENCHMARK_SHA1] = 312.0 / elapsed;
-}
-