86 lines
3.5 KiB
Swift
86 lines
3.5 KiB
Swift
//
|
|
// Copyright 2023 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Foundation
|
|
import LibSignalClient
|
|
|
|
#if TESTABLE_BUILD
|
|
|
|
internal class MockPreKeyManager: PreKeyManager {
|
|
func isAppLockedDueToPreKeyUpdateFailures(tx: SignalServiceKit.DBReadTransaction) -> Bool { false }
|
|
func refreshOneTimePreKeysCheckDidSucceed() { }
|
|
func checkPreKeysIfNecessary(tx: SignalServiceKit.DBReadTransaction) { }
|
|
func rotatePreKeysOnUpgradeIfNecessary(for identity: OWSIdentity) async throws { }
|
|
|
|
func createPreKeysForRegistration() -> Task<RegistrationPreKeyUploadBundles, Error> {
|
|
let identityKeyPair = ECKeyPair.generateKeyPair()
|
|
return Task {
|
|
.init(
|
|
aci: .init(
|
|
identity: .aci,
|
|
identityKeyPair: identityKeyPair,
|
|
signedPreKey: SignedPreKeyStoreImpl.generateSignedPreKey(keyId: PreKeyId.randomSigned(), signedBy: identityKeyPair.keyPair.privateKey),
|
|
lastResortPreKey: generateLastResortKyberPreKey(signedBy: identityKeyPair.keyPair.privateKey),
|
|
),
|
|
pni: .init(
|
|
identity: .pni,
|
|
identityKeyPair: identityKeyPair,
|
|
signedPreKey: SignedPreKeyStoreImpl.generateSignedPreKey(keyId: PreKeyId.randomSigned(), signedBy: identityKeyPair.keyPair.privateKey),
|
|
lastResortPreKey: generateLastResortKyberPreKey(signedBy: identityKeyPair.keyPair.privateKey),
|
|
)
|
|
)
|
|
}
|
|
}
|
|
|
|
func createPreKeysForProvisioning(
|
|
aciIdentityKeyPair: ECKeyPair,
|
|
pniIdentityKeyPair: ECKeyPair
|
|
) -> Task<RegistrationPreKeyUploadBundles, Error> {
|
|
let identityKeyPair = ECKeyPair.generateKeyPair()
|
|
return Task {
|
|
.init(
|
|
aci: .init(
|
|
identity: .aci,
|
|
identityKeyPair: identityKeyPair,
|
|
signedPreKey: SignedPreKeyStoreImpl.generateSignedPreKey(keyId: PreKeyId.randomSigned(), signedBy: identityKeyPair.keyPair.privateKey),
|
|
lastResortPreKey: generateLastResortKyberPreKey(signedBy: identityKeyPair.keyPair.privateKey),
|
|
),
|
|
pni: .init(
|
|
identity: .pni,
|
|
identityKeyPair: identityKeyPair,
|
|
signedPreKey: SignedPreKeyStoreImpl.generateSignedPreKey(keyId: PreKeyId.randomSigned(), signedBy: identityKeyPair.keyPair.privateKey),
|
|
lastResortPreKey: generateLastResortKyberPreKey(signedBy: identityKeyPair.keyPair.privateKey),
|
|
)
|
|
)
|
|
}
|
|
}
|
|
|
|
public var didFinalizeRegistrationPrekeys = false
|
|
|
|
func finalizeRegistrationPreKeys(
|
|
_ bundles: RegistrationPreKeyUploadBundles,
|
|
uploadDidSucceed: Bool
|
|
) -> Task<Void, Error> {
|
|
didFinalizeRegistrationPrekeys = true
|
|
return Task {}
|
|
}
|
|
|
|
func rotateOneTimePreKeysForRegistration(auth: ChatServiceAuth) -> Task<Void, Error> {
|
|
return Task {}
|
|
}
|
|
|
|
func rotateSignedPreKeysIfNeeded() -> Task<Void, Error> { Task {} }
|
|
func refreshOneTimePreKeys(forIdentity identity: OWSIdentity, alsoRefreshSignedPreKey shouldRefreshSignedPreKey: Bool) { }
|
|
|
|
func generateLastResortKyberPreKey(signedBy identityKey: PrivateKey) -> LibSignalClient.KyberPreKeyRecord {
|
|
return KyberPreKeyStoreImpl.generatePreKeyRecord(keyId: PreKeyId.random(), now: Date(), signedBy: identityKey)
|
|
}
|
|
|
|
func setIsChangingNumber(_ isChangingNumber: Bool) {
|
|
}
|
|
}
|
|
|
|
#endif
|