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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
|
/*
* Data Store
* (C) 1999-2007 Jack Lloyd
*
* Botan is released under the Simplified BSD License (see license.txt)
*/
#include <botan/datastor.h>
#include <botan/exceptn.h>
#include <botan/parsing.h>
#include <botan/hex.h>
#include <botan/internal/stl_util.h>
namespace Botan {
/*
* Data_Store Equality Comparison
*/
bool Data_Store::operator==(const Data_Store& other) const
{
return (m_contents == other.m_contents);
}
/*
* Check if this key has at least one value
*/
bool Data_Store::has_value(const std::string& key) const
{
return (m_contents.lower_bound(key) != m_contents.end());
}
/*
* Search based on an arbitrary predicate
*/
std::multimap<std::string, std::string> Data_Store::search_for(
std::function<bool (std::string, std::string)> predicate) const
{
std::multimap<std::string, std::string> out;
for(auto i = m_contents.begin(); i != m_contents.end(); ++i)
if(predicate(i->first, i->second))
out.insert(std::make_pair(i->first, i->second));
return out;
}
/*
* Search based on key equality
*/
std::vector<std::string> Data_Store::get(const std::string& looking_for) const
{
std::vector<std::string> out;
auto range = m_contents.equal_range(looking_for);
for(auto i = range.first; i != range.second; ++i)
out.push_back(i->second);
return out;
}
/*
* Get a single atom
*/
std::string Data_Store::get1(const std::string& key) const
{
std::vector<std::string> vals = get(key);
if(vals.empty())
throw Invalid_State("Data_Store::get1: No values set for " + key);
if(vals.size() > 1)
throw Invalid_State("Data_Store::get1: More than one value for " + key);
return vals[0];
}
std::string Data_Store::get1(const std::string& key,
const std::string& default_value) const
{
std::vector<std::string> vals = get(key);
if(vals.size() > 1)
throw Invalid_State("Data_Store::get1: More than one value for " + key);
if(vals.empty())
return default_value;
return vals[0];
}
/*
* Get a single std::vector atom
*/
std::vector<uint8_t>
Data_Store::get1_memvec(const std::string& key) const
{
std::vector<std::string> vals = get(key);
if(vals.empty())
return std::vector<uint8_t>();
if(vals.size() > 1)
throw Invalid_State("Data_Store::get1_memvec: Multiple values for " +
key);
return hex_decode(vals[0]);
}
/*
* Get a single uint32_t atom
*/
uint32_t Data_Store::get1_uint32(const std::string& key,
uint32_t default_val) const
{
std::vector<std::string> vals = get(key);
if(vals.empty())
return default_val;
else if(vals.size() > 1)
throw Invalid_State("Data_Store::get1_uint32: Multiple values for " + key);
return to_u32bit(vals[0]);
}
/*
* Insert a single key and value
*/
void Data_Store::add(const std::string& key, const std::string& val)
{
multimap_insert(m_contents, key, val);
}
/*
* Insert a single key and value
*/
void Data_Store::add(const std::string& key, uint32_t val)
{
add(key, std::to_string(val));
}
/*
* Insert a single key and value
*/
void Data_Store::add(const std::string& key, const secure_vector<uint8_t>& val)
{
add(key, hex_encode(val.data(), val.size()));
}
void Data_Store::add(const std::string& key, const std::vector<uint8_t>& val)
{
add(key, hex_encode(val.data(), val.size()));
}
/*
* Insert a mapping of key/value pairs
*/
void Data_Store::add(const std::multimap<std::string, std::string>& in)
{
std::multimap<std::string, std::string>::const_iterator i = in.begin();
while(i != in.end())
{
m_contents.insert(*i);
++i;
}
}
/*
* Create and populate a X509_DN
*/
X509_DN create_dn(const Data_Store& info)
{
auto names = info.search_for(
[](const std::string& key, const std::string&)
{
return (key.find("X520.") != std::string::npos);
});
X509_DN dn;
for(auto i = names.begin(); i != names.end(); ++i)
dn.add_attribute(i->first, i->second);
return dn;
}
/*
* Create and populate an AlternativeName
*/
AlternativeName create_alt_name(const Data_Store& info)
{
auto names = info.search_for(
[](const std::string& key, const std::string&)
{
return (key == "RFC822" ||
key == "DNS" ||
key == "URI" ||
key == "IP");
});
AlternativeName alt_name;
for(auto i = names.begin(); i != names.end(); ++i)
alt_name.add_attribute(i->first, i->second);
return alt_name;
}
}
|
