cryptohome/cryptohome_keys_manager.h

// Copyright 2021 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CRYPTOHOME_CRYPTOHOME_KEYS_MANAGER_H_
#define CRYPTOHOME_CRYPTOHOME_KEYS_MANAGER_H_

#include <map>
#include <memory>
#include <utility>
#include <vector>

#include <libhwsec/frontend/cryptohome/frontend.h>
#include <libhwsec/status.h>

#include "cryptohome/cryptohome_key_loader.h"

namespace cryptohome {

enum class CryptohomeKeyType {
  kRSA,
  kECC,
};

class CryptohomeKeysManager {
 public:
  CryptohomeKeysManager(hwsec::CryptohomeFrontend* hwsec, Platform* platform);
  CryptohomeKeysManager(const CryptohomeKeysManager&) = delete;
  CryptohomeKeysManager& operator=(const CryptohomeKeysManager&) = delete;

  // constructor for testing purpose.
  CryptohomeKeysManager(
      hwsec::CryptohomeFrontend* hwsec,
      std::vector<std::pair<CryptohomeKeyType,
                            std::unique_ptr<CryptohomeKeyLoader>>> init_list)
      : hwsec_(hwsec) {
    for (auto& pair : init_list) {
      key_loaders_.emplace(pair.first, std::move(pair.second));
    }
  }

  virtual ~CryptohomeKeysManager() = default;

  // Init all key loaders.
  virtual void Init();

  // Return the specific key loader.
  virtual CryptohomeKeyLoader* GetKeyLoader(CryptohomeKeyType key_type);

  // Whether the key manager has any cryptohome key or not.
  virtual bool HasAnyCryptohomeKey();

  // Whether the specific key loader has cryptohome key or not.
  virtual bool HasCryptohomeKey(CryptohomeKeyType key_type);

 protected:
  // constructor for mock testing purpose.
  CryptohomeKeysManager() : hwsec_(nullptr) {}

 private:
  hwsec::CryptohomeFrontend* const hwsec_;
  std::map<CryptohomeKeyType, std::unique_ptr<CryptohomeKeyLoader>>
      key_loaders_;
};

}  // namespace cryptohome

#endif  // CRYPTOHOME_CRYPTOHOME_KEYS_MANAGER_H_