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 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_TRUSTED_VAULT_ICLOUD_KEYCHAIN_RECOVERY_FACTOR_H_
#define COMPONENTS_TRUSTED_VAULT_ICLOUD_KEYCHAIN_RECOVERY_FACTOR_H_
#include <optional>
#include "base/memory/weak_ptr.h"
#include "components/signin/public/identity_manager/account_info.h"
#include "components/trusted_vault/local_recovery_factor.h"
#include "components/trusted_vault/standalone_trusted_vault_storage.h"
#include "components/trusted_vault/trusted_vault_histograms.h"
#include "components/trusted_vault/trusted_vault_throttling_connection.h"
#include "google_apis/gaia/gaia_id.h"
namespace trusted_vault {
class ICloudRecoveryKey;
// This class represents the iCloud Keychain as recovery factor.
// It stores required (private) keys in the iCloud Keychain.
class ICloudKeychainRecoveryFactor : public LocalRecoveryFactor {
public:
// `storage` and `connection` must not be null and must outlive this object.
// `storage` must contain a vault for `primary_account` when calling any
// method of this class.
// TODO(crbug.com/405381481): Refactor / remove the usage of
// StandaloneTrustedVaultStorage in this class.
ICloudKeychainRecoveryFactor(
const std::string& icloud_keychain_access_group_prefix,
const SecurityDomainId security_domain_id,
StandaloneTrustedVaultStorage* storage,
TrustedVaultThrottlingConnection* connection,
CoreAccountInfo primary_account);
ICloudKeychainRecoveryFactor(const ICloudKeychainRecoveryFactor&) = delete;
ICloudKeychainRecoveryFactor& operator=(ICloudKeychainRecoveryFactor&) =
delete;
~ICloudKeychainRecoveryFactor() override;
LocalRecoveryFactorType GetRecoveryFactorType() const override;
void AttemptRecovery(AttemptRecoveryCallback cb) override;
bool IsRegistered() override;
void MarkAsNotRegistered() override;
TrustedVaultRecoveryFactorRegistrationStateForUMA MaybeRegister(
RegisterCallback cb) override;
private:
trusted_vault_pb::LocalTrustedVaultPerUser* GetPrimaryAccountVault();
void OnICloudKeysRetrievedForRecovery(
AttemptRecoveryCallback cb,
std::vector<std::unique_ptr<ICloudRecoveryKey>> local_icloud_keys);
void OnRecoveryFactorStateDownloadedForRecovery(
AttemptRecoveryCallback cb,
std::vector<std::unique_ptr<ICloudRecoveryKey>> local_icloud_keys,
DownloadAuthenticationFactorsRegistrationStateResult result);
void FulfillRecoveryWithFailure(
TrustedVaultDownloadKeysStatusForUMA status_for_uma,
AttemptRecoveryCallback cb);
void MarkAsRegistered();
void OnICloudKeysRetrievedForRegistration(
RegisterCallback cb,
std::vector<std::unique_ptr<ICloudRecoveryKey>> local_icloud_keys);
void OnRecoveryFactorStateDownloadedForRegistration(
RegisterCallback cb,
std::vector<std::unique_ptr<ICloudRecoveryKey>> local_icloud_keys,
DownloadAuthenticationFactorsRegistrationStateResult result);
void OnICloudKeyCreatedForRegistration(
RegisterCallback cb,
std::unique_ptr<ICloudRecoveryKey> local_icloud_key);
void OnRegistered(RegisterCallback cb,
TrustedVaultRegistrationStatus status,
int key_version);
void FulfillRegistrationWithFailure(TrustedVaultRegistrationStatus status,
RegisterCallback cb);
const std::string icloud_keychain_access_group_;
const SecurityDomainId security_domain_id_;
const raw_ptr<StandaloneTrustedVaultStorage> storage_;
const raw_ptr<TrustedVaultThrottlingConnection> connection_;
const CoreAccountInfo primary_account_;
// Destroying this will cancel the ongoing request.
std::unique_ptr<TrustedVaultConnection::Request>
ongoing_download_registration_state_request_for_recovery_;
// Destroying this will cancel the ongoing request.
std::unique_ptr<TrustedVaultConnection::Request>
ongoing_download_registration_state_request_for_registration_;
// Destroying this will cancel the ongoing request.
std::unique_ptr<TrustedVaultConnection::Request>
ongoing_registration_request_;
base::WeakPtrFactory<ICloudKeychainRecoveryFactor> weak_ptr_factory_{this};
};
} // namespace trusted_vault
#endif // COMPONENTS_TRUSTED_VAULT_ICLOUD_KEYCHAIN_RECOVERY_FACTOR_H_
|