72 lines
2.2 KiB
Swift
72 lines
2.2 KiB
Swift
//
|
|
// Copyright 2023 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Foundation
|
|
|
|
#if TESTABLE_BUILD
|
|
|
|
public class SecureValueRecoveryMock: SecureValueRecovery {
|
|
|
|
public init() {}
|
|
|
|
public func refreshBackupIfNecessary() async throws {
|
|
}
|
|
|
|
public func refreshCredentialsIfNecessary() async throws {
|
|
}
|
|
|
|
public var reglockToken: String?
|
|
|
|
public var backupMasterKeyMock: ((_ pin: String, _ masterKey: MasterKey, _ force: Bool, _ authMethod: SVR.AuthMethod) -> Promise<Void>)?
|
|
|
|
public func backupMasterKey(pin: String, masterKey: MasterKey, force: Bool, authMethod: SVR.AuthMethod) async throws {
|
|
try await backupMasterKeyMock!(pin, masterKey, force, authMethod).awaitable()
|
|
}
|
|
|
|
public var restoreKeysMock: ((_ pin: String, _ authMethod: SVR.AuthMethod) -> Guarantee<SVR.RestoreKeysResult>)?
|
|
|
|
public func restoreKeys(pin: String, authMethod: SVR.AuthMethod) async -> SVR.RestoreKeysResult {
|
|
return await restoreKeysMock!(pin, authMethod).awaitable()
|
|
}
|
|
|
|
public var syncedMasterKey: MasterKey?
|
|
|
|
public func storeKeys(
|
|
fromKeysSyncMessage syncMessage: SSKProtoSyncMessageKeys,
|
|
authedDevice: AuthedDevice,
|
|
tx: DBWriteTransaction,
|
|
) throws(SVR.KeysError) {
|
|
syncedMasterKey = syncMessage.master.map { try! MasterKey(data: $0) }
|
|
}
|
|
|
|
public func storeKeys(
|
|
fromProvisioningMessage provisioningMessage: LinkingProvisioningMessage,
|
|
authedDevice: AuthedDevice,
|
|
tx: DBWriteTransaction,
|
|
) {
|
|
let masterKey = switch provisioningMessage.rootKey {
|
|
case .accountEntropyPool(let aep): aep.getMasterKey()
|
|
case .masterKey(let masterKey): masterKey
|
|
}
|
|
syncedMasterKey = masterKey
|
|
}
|
|
|
|
public var doesHavePendingRestoration = false
|
|
|
|
public func hasPendingRestoration(transaction: DBReadTransaction) -> Bool {
|
|
return doesHavePendingRestoration
|
|
}
|
|
|
|
public func recordPendingRestoration(transaction: DBWriteTransaction) {
|
|
doesHavePendingRestoration = true
|
|
}
|
|
|
|
public func clearPendingRestoration(transaction: DBWriteTransaction) {
|
|
doesHavePendingRestoration = false
|
|
}
|
|
}
|
|
|
|
#endif
|