kdf_registry.hh

Go to the documentation of this file.

/*
 * libkeepass - KeePass key database importer/exporter
 * Copyright (C) 2014 Christian Kindahl
 * Copyright (C) 2024 Dominik Krümpelmann
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
#pragma once
 
#include <functional>
#include <map>
#include <mutex>
 
#include "database.hh"
#include "libkeepass/export.hh"
#include "libkeepass/key.hh"
#include "libkeepass/secure.hh"
 
namespace keepass {
 
class LIBKEEPASS_API KdfRegistry final {
public:
  using Factory =
      std::function<SecureBuffer<32>(const Key&, const Database&, Key::SubKeyResolution)>;
 
  static KdfRegistry& Instance();
 
  void Register(Database::Kdf kdf, Factory factory);
 
  SecureBuffer<32> Transform(Database::Kdf kdf, const Key& key, const Database& db,
                             Key::SubKeyResolution resolution) const;
 
  bool Contains(Database::Kdf kdf) const;
 
private:
  KdfRegistry();
 
  mutable std::mutex mutex_;
  std::map<Database::Kdf, Factory> factories_;
};
 
} // namespace keepass