std/crypto/throughput_test.zig


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

// Modify the HashFunction variable to the one wanted to test.
//
// NOTE: The throughput measurement may be slightly lower than other measurements since we run
// through our block alignment functions as well. Be aware when comparing against other tests.
//
// ```
// zig build-exe --release-fast --library c throughput_test.zig
// ./throughput_test
// ```
const HashFunction = @import("md5.zig").Md5;
const BytesToHash  = 1024 * Mb;

const std = @import("std");

const c = @cImport({
    @cInclude("time.h");
});

const Mb = 1024 * 1024;

pub fn main() !void {
    var stdout_file = try std.io.getStdOut();
    var stdout_out_stream = std.io.FileOutStream.init(&stdout_file);
    const stdout = &stdout_out_stream.stream;

    var block: [HashFunction.block_size]u8 = undefined;
    std.mem.set(u8, block[0..], 0);

    var h = HashFunction.init();
    var offset: usize = 0;

    const start = c.clock();
    while (offset < BytesToHash) : (offset += block.len) {
        h.update(block[0..]);
    }
    const end = c.clock();

    const elapsed_s = f64((end - start) * c.CLOCKS_PER_SEC) / 1000000;
    const throughput = u64(BytesToHash / elapsed_s);

    try stdout.print("{}: ", @typeName(HashFunction));
    try stdout.print("{} Mb/s\n", throughput);
}