psk_db_sql.cpp

Go to the documentation of this file.

/*
* (C) 2017 Jack Lloyd
*
* Botan is released under the Simplified BSD License (see license.txt)
*/
 
#include <botan/psk_db.h>
#include <botan/database.h>
 
namespace Botan {
 
Encrypted_PSK_Database_SQL::Encrypted_PSK_Database_SQL(const secure_vector<uint8_t>& master_key,
                                                       std::shared_ptr<SQL_Database> db,
                                                       std::string_view table_name) :
   Encrypted_PSK_Database(master_key),
   m_db(std::move(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)");
   }
 
Encrypted_PSK_Database_SQL::~Encrypted_PSK_Database_SQL() = default;
 
void Encrypted_PSK_Database_SQL::kv_del(std::string_view 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(std::string_view name, std::string_view 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(std::string_view 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;
   }
 
}