Source code
Revision control
Copy as Markdown
Other Tools
/*
* 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;
}
}