From 40674518bac3efaa2691bf0b29218940683248a1 Mon Sep 17 00:00:00 2001
From: Dnomd343 <dnomd343@gmail.com>
Date: Sun, 24 Mar 2024 17:43:17 +0800
Subject: [PATCH] update: md5 performance benchmark

---
 benchmark.cc | 43 ++++++++++++++++++++++++++-----------------
 1 file changed, 26 insertions(+), 17 deletions(-)

diff --git a/benchmark.cc b/benchmark.cc
index e2d3a6d..77a30ca 100644
--- a/benchmark.cc
+++ b/benchmark.cc
@@ -1,34 +1,43 @@
-#include <benchmark/benchmark.h>
-
 #include "md5.h"
+#include "benchmark/benchmark.h"
+
+using md5::MD5;
 
-std::string test_data() {
-    char data[64];
-    for (char i = 0; i < 64; ++i) {
-        data[i] = i;
+std::string build_test_data() {
+    std::string data(65536, 0x00);
+    for (uint32_t i = 0; i < data.size(); ++i) {
+        data[i] = static_cast<char>(i & 0xff);
     }
-    return {data, data + 64};
+    return data;
 }
 
-static void MD5_Update(benchmark::State &state) {
-    const auto data = test_data();
-
-    md5::MD5 kk;
-
+static void MD5_Digest(benchmark::State &state) {
+    MD5 md5;
     for (auto _ : state) {
-        kk.Update(data.c_str(), 64);
+        auto volatile holder = md5.Digest();
     }
 }
 
-static void MD5_Digest(benchmark::State &state) {
-    md5::MD5 kk;
+static void MD5_Update(benchmark::State &state) {
+    MD5 md5;
+    const auto data = build_test_data();
     for (auto _ : state) {
-        auto pp = kk.Digest();
+        md5.Update(data.c_str(), state.range(0));
     }
 }
 
-BENCHMARK(MD5_Update);
+static void MD5_Hash(benchmark::State &state) {
+    const auto data = build_test_data();
+    for (auto _ : state) {
+        MD5::Hash(data.c_str(), state.range(0));
+    }
+}
 
 BENCHMARK(MD5_Digest);
 
+BENCHMARK(MD5_Update)->RangeMultiplier(4)->Range(64, 4096);
+
+BENCHMARK(MD5_Hash)->Arg(0);
+BENCHMARK(MD5_Hash)->RangeMultiplier(4)->Range(64, 4096);
+
 BENCHMARK_MAIN();