blob: 3a43e7380b89ba0e79873ed81853ea1b0a221d29 (
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
|
/*
* (C) 2017 Jack Lloyd
*
* Botan is released under the Simplified BSD License (see license.txt)
*/
#include <botan/psk_db_sql.h>
namespace Botan {
Encrypted_PSK_Database_SQL::Encrypted_PSK_Database_SQL(const secure_vector<uint8_t>& master_key,
std::shared_ptr<SQL_Database> db,
const std::string& table_name) :
Encrypted_PSK_Database(master_key),
m_db(db),
m_table_name(table_name)
{
m_db->create_table(
"create table if not exists " + m_table_name +
"(psk_name TEXT PRIMARY KEY, psk_value TEXT)");
}
void Encrypted_PSK_Database_SQL::kv_del(const std::string& name)
{
auto stmt = m_db->new_statement("delete from " + m_table_name + " where psk_name=?1");
stmt->bind(1, name);
stmt->spin();
}
void Encrypted_PSK_Database_SQL::kv_set(const std::string& name, const std::string& value)
{
auto stmt = m_db->new_statement("insert or replace into " + m_table_name + " values(?1, ?2)");
stmt->bind(1, name);
stmt->bind(2, value);
stmt->spin();
}
std::string Encrypted_PSK_Database_SQL::kv_get(const std::string& name) const
{
auto stmt = m_db->new_statement("select psk_value from " + m_table_name +
" where psk_name = ?1");
stmt->bind(1, name);
while(stmt->step())
{
return stmt->get_str(0);
}
return "";
}
std::set<std::string> Encrypted_PSK_Database_SQL::kv_get_all() const
{
std::set<std::string> names;
auto stmt = m_db->new_statement("select psk_name from " + m_table_name);
while(stmt->step())
{
names.insert(stmt->get_str(0));
}
return names;
}
}
|
