blob: f4f762aeb750a4d2866900d61de4e5a760ee45ff (
plain)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
/*
* XMSS Tools
* (C) 2017 Matthias Gierlings
*
* Botan is released under the Simplified BSD License (see license.txt)
**/
#include <botan/xmss_tools.h>
namespace Botan {
#if defined(BOTAN_TARGET_OS_HAS_THREADS)
size_t XMSS_Tools::max_threads()
{
static const size_t threads { bench_threads() };
return threads;
}
size_t XMSS_Tools::bench_threads()
{
if(std::thread::hardware_concurrency() <= 1)
{
return 1;
}
const size_t BENCH_ITERATIONS = 1000;
std::vector<std::thread> threads;
threads.reserve(std::thread::hardware_concurrency());
std::vector<std::chrono::nanoseconds> durations;
std::vector<size_t> concurrency { std::thread::hardware_concurrency(),
std::thread::hardware_concurrency() / 2 };
for(const auto& cc : concurrency)
{
std::vector<XMSS_Hash> hash(std::thread::hardware_concurrency(),
XMSS_Hash("SHA-256"));
const std::vector<uint8_t> buffer(hash[0].output_length());
std::vector<secure_vector<uint8_t>> data(
std::thread::hardware_concurrency(),
secure_vector<uint8_t>(hash[0].output_length()));
auto start = std::chrono::high_resolution_clock::now();
for(size_t i = 0; i < cc; ++i)
{
auto& hs = hash[i];
auto& d = data[i];
const size_t n_iters = BENCH_ITERATIONS * (std::thread::hardware_concurrency() / cc);
threads.emplace_back(std::thread([n_iters, &hs, &d]()
{
for(size_t n = 0; n < n_iters; n++)
{
hs.h(d, d, d);
}
}
));
}
durations.emplace_back(std::chrono::duration_cast<std::chrono::nanoseconds>(std::chrono::high_resolution_clock::now() - start));
for(auto& t : threads)
{
t.join();
}
threads.clear();
}
if(durations[0].count() < durations[1].count())
{
return concurrency[0];
}
else
{
return concurrency[1];
}
}
#endif
}
|