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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
/*
* OTP tests
* (C) 2017 Jack Lloyd
*
* Botan is released under the Simplified BSD License (see license.txt)
*/
#include "tests.h"
#if defined(BOTAN_HAS_HOTP)
#include <botan/hotp.h>
#endif
#if defined(BOTAN_HAS_TOTP)
#include <botan/totp.h>
#include <botan/calendar.h>
#endif
namespace Botan_Tests {
#if defined(BOTAN_HAS_HOTP)
class HOTP_KAT_Tests : public Text_Based_Test
{
public:
HOTP_KAT_Tests()
: Text_Based_Test("otp/hotp.vec", "Key,Digits,Counter,OTP")
{}
bool clear_between_callbacks() const override { return false; }
Test::Result run_one_test(const std::string& hash_algo, const VarMap& vars) override
{
Test::Result result("HOTP " + hash_algo);
const std::vector<uint8_t> key = get_req_bin(vars, "Key");
const size_t otp = get_req_sz(vars, "OTP");
const uint64_t counter = get_req_sz(vars, "Counter");
const size_t digits = get_req_sz(vars, "Digits");
Botan::HOTP hotp(key, hash_algo, digits);
result.test_eq("OTP", hotp.generate_hotp(counter), otp);
std::pair<bool, uint64_t> otp_res = hotp.verify_hotp(otp, counter, 0);
result.test_eq("OTP verify result", otp_res.first, true);
result.confirm("OTP verify next counter", otp_res.second == counter + 1);
// Test invalid OTP
otp_res = hotp.verify_hotp(otp + 1, counter, 0);
result.test_eq("OTP verify result", otp_res.first, false);
result.confirm("OTP verify next counter", otp_res.second == counter);
// Test invalid OTP with long range
otp_res = hotp.verify_hotp(otp + 1, counter, 100);
result.test_eq("OTP verify result", otp_res.first, false);
result.confirm("OTP verify next counter", otp_res.second == counter);
// Test valid OTP with long range
otp_res = hotp.verify_hotp(otp, counter - 90, 100);
result.test_eq("OTP verify result", otp_res.first, true);
result.confirm("OTP verify next counter", otp_res.second == counter + 1);
return result;
}
};
BOTAN_REGISTER_TEST("otp_hotp", HOTP_KAT_Tests);
#endif
#if defined(BOTAN_HAS_TOTP)
class TOTP_KAT_Tests : public Text_Based_Test
{
public:
TOTP_KAT_Tests()
: Text_Based_Test("otp/totp.vec", "Key,Digits,Timestep,Timestamp,OTP")
{}
bool clear_between_callbacks() const override { return false; }
Test::Result run_one_test(const std::string& hash_algo, const VarMap& vars) override
{
Test::Result result("TOTP " + hash_algo);
const std::vector<uint8_t> key = get_req_bin(vars, "Key");
const size_t otp = get_req_sz(vars, "OTP");
const size_t digits = get_req_sz(vars, "Digits");
const size_t timestep = get_req_sz(vars, "Timestep");
const std::string timestamp = get_req_str(vars, "Timestamp");
Botan::TOTP totp(key, hash_algo, digits, timestep);
std::chrono::system_clock::time_point time = from_timestring(timestamp);
std::chrono::system_clock::time_point later_time = time + std::chrono::seconds(timestep);
std::chrono::system_clock::time_point too_late = time + std::chrono::seconds(2*timestep);
result.test_eq("TOTP generate", totp.generate_totp(time), otp);
result.test_eq("TOTP verify valid", totp.verify_totp(otp, time, 0), true);
result.test_eq("TOTP verify invalid", totp.verify_totp(otp ^ 1, time, 0), false);
result.test_eq("TOTP verify time slip", totp.verify_totp(otp, later_time, 0), false);
result.test_eq("TOTP verify time slip allowed", totp.verify_totp(otp, later_time, 1), true);
result.test_eq("TOTP verify time slip out of range", totp.verify_totp(otp, too_late, 1), false);
return result;
}
private:
std::chrono::system_clock::time_point from_timestring(const std::string& time_str)
{
if(time_str.size() != 19)
throw std::invalid_argument("Invalid TOTP timestamp string " + time_str);
// YYYY-MM-DDTHH:MM:SS
// 0123456789012345678
const uint32_t year = Botan::to_u32bit(time_str.substr(0, 4));
const uint32_t month = Botan::to_u32bit(time_str.substr(5, 2));
const uint32_t day = Botan::to_u32bit(time_str.substr(8, 2));
const uint32_t hour = Botan::to_u32bit(time_str.substr(11, 2));
const uint32_t minute = Botan::to_u32bit(time_str.substr(14, 2));
const uint32_t second = Botan::to_u32bit(time_str.substr(17, 2));
return Botan::calendar_point(year, month, day, hour, minute, second).to_std_timepoint();
}
};
BOTAN_REGISTER_TEST("otp_totp", TOTP_KAT_Tests);
#endif
}
|
