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
|
/*
* (C) 2009 Jack Lloyd
*
* Distributed under the terms of the Botan license
*/
#include <botan/botan.h>
#include <botan/x931_rng.h>
#include <botan/hex.h>
#include <botan/lookup.h>
#include <iostream>
#include <fstream>
#include <deque>
#include <stdexcept>
using namespace Botan;
namespace {
class Fixed_Output_RNG : public RandomNumberGenerator
{
public:
bool is_seeded() const { return !buf.empty(); }
byte random()
{
if(buf.empty())
throw std::runtime_error("Out of bytes");
byte out = buf.front();
buf.pop_front();
return out;
}
void randomize(byte out[], size_t len) throw()
{
for(size_t j = 0; j != len; j++)
out[j] = random();
}
std::string name() const { return "Fixed_Output_RNG"; }
void reseed(size_t) {}
void clear() throw() {}
void add_entropy(const byte in[], size_t len)
{
buf.insert(buf.end(), in, in + len);
}
void add_entropy_source(EntropySource* es) { delete es; }
Fixed_Output_RNG() {}
private:
std::deque<byte> buf;
};
void x931_tests(std::vector<std::pair<std::string, std::string> > vecs,
const std::string& cipher)
{
for(size_t j = 0; j != vecs.size(); ++j)
{
const std::string result = vecs[j].first;
const std::string input = vecs[j].second;
ANSI_X931_RNG prng(get_block_cipher(cipher),
new Fixed_Output_RNG);
SecureVector<byte> x = hex_decode(input);
prng.add_entropy(x.begin(), x.size());
SecureVector<byte> output(result.size() / 2);
prng.randomize(output, output.size());
if(hex_decode(result) != output)
std::cout << "FAIL";
else
std::cout << "PASS";
std::cout << " Seed " << input << " "
<< "Got " << hex_encode(output) << " "
<< "Exp " << result << "\n";
}
}
std::vector<std::pair<std::string, std::string> >
read_file(const std::string& fsname)
{
std::ifstream in(fsname.c_str());
std::vector<std::pair<std::string, std::string> > out;
while(in.good())
{
std::string line;
std::getline(in, line);
if(line == "")
break;
std::vector<std::string> l = split_on(line, ':');
if(l.size() != 2)
throw std::runtime_error("Bad line " + line);
out.push_back(std::make_pair(l[0], l[1]));
}
return out;
}
}
int main()
{
Botan::LibraryInitializer init;
x931_tests(read_file("ANSI931_AES128VST.txt.vst"), "AES-128");
x931_tests(read_file("ANSI931_AES192VST.txt.vst"), "AES-192");
x931_tests(read_file("ANSI931_AES256VST.txt.vst"), "AES-256");
x931_tests(read_file("ANSI931_TDES2VST.txt.vst"), "TripleDES");
x931_tests(read_file("ANSI931_TDES3VST.txt.vst"), "TripleDES");
}
|
