// 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_USER_SECRET_STASH_H_
#define CRYPTOHOME_USER_SECRET_STASH_H_

#include <base/optional.h>
#include <brillo/secure_blob.h>
#include <stdint.h>

#include <memory>

#include "cryptohome/flatbuffer_secure_allocator_bridge.h"
#include "cryptohome/user_secret_stash_generated.h"

namespace cryptohome {

// This wraps the UserSecretStash flatbuffer message, and is the only way that
// the UserSecretStash is accessed. Don't pass the raw flatbuffer around.
class UserSecretStash {
 public:
  UserSecretStash() = default;
  virtual ~UserSecretStash() = default;

  // Because this class contains raw secrets, it should never be copy-able.
  UserSecretStash(const UserSecretStash&) = delete;
  UserSecretStash& operator=(const UserSecretStash&) = delete;

  const brillo::SecureBlob& GetFileSystemKey() const;
  void SetFileSystemKey(const brillo::SecureBlob& key);

  const brillo::SecureBlob& GetResetSecret() const;
  void SetResetSecret(const brillo::SecureBlob& secret);

  // Sets up a UserSecretStash with a random file system key, and a random reset
  // secret.
  void InitializeRandom();

  // This uses the |main_key|, which should be 256-bit as of right now, to
  // encrypted this UserSecretStash class. The object is converted to a
  // UserSecretStashProto, serialized, and then encrypted with AES-GCM-256.
  base::Optional<brillo::SecureBlob> GetAesGcmEncrypted(
      const brillo::SecureBlob& main_key);

  // This deserializes the |flatbuffer| into a AesGcmEncryptedUSS table.
  // That table is contains a ciphertext, which is decrypted with the |main_key|
  // using AES-GMC-256. It doesn't return the plaintext, it populates the fields
  // of the class with the encrypted message.
  bool FromAesGcmEncrypted(const brillo::SecureBlob& flatbuffer,
                           const brillo::SecureBlob& main_key);

 private:
  // A key registered with the kernel to decrypt files.
  base::Optional<brillo::SecureBlob> file_system_key_;
  // The reset secret used for any PinWeaver backed credentials.
  base::Optional<brillo::SecureBlob> reset_secret_;
};

}  // namespace cryptohome

#endif  // CRYPTOHOME_USER_SECRET_STASH_H_