80 lines
2.7 KiB
Swift
80 lines
2.7 KiB
Swift
//
|
|
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
import PromiseKit
|
|
|
|
@objc(SSKRotateSignedPreKeyOperation)
|
|
public class RotateSignedPreKeyOperation: OWSOperation {
|
|
private var tsAccountManager: TSAccountManager {
|
|
return TSAccountManager.sharedInstance()
|
|
}
|
|
|
|
private var accountServiceClient: AccountServiceClient {
|
|
return AccountServiceClient.shared
|
|
}
|
|
|
|
private var primaryStorage: OWSPrimaryStorage {
|
|
return OWSPrimaryStorage.shared()
|
|
}
|
|
|
|
public override func run() {
|
|
Logger.debug("")
|
|
|
|
guard tsAccountManager.isRegistered() else {
|
|
Logger.debug("skipping - not registered")
|
|
return
|
|
}
|
|
|
|
// Loki: Doing this on the global queue to match Signal
|
|
DispatchQueue.global().async {
|
|
SessionManagementProtocol.rotateSignedPreKey()
|
|
self.reportSuccess()
|
|
}
|
|
|
|
/* Loki: Original code
|
|
* ================
|
|
let signedPreKeyRecord: SignedPreKeyRecord = self.primaryStorage.generateRandomSignedRecord()
|
|
|
|
self.primaryStorage.storeSignedPreKey(signedPreKeyRecord.id, signedPreKeyRecord: signedPreKeyRecord)
|
|
firstly {
|
|
return self.accountServiceClient.setSignedPreKey(signedPreKeyRecord)
|
|
}.done(on: DispatchQueue.global()) {
|
|
Logger.info("Successfully uploaded signed PreKey")
|
|
signedPreKeyRecord.markAsAcceptedByService()
|
|
self.primaryStorage.storeSignedPreKey(signedPreKeyRecord.id, signedPreKeyRecord: signedPreKeyRecord)
|
|
self.primaryStorage.setCurrentSignedPrekeyId(signedPreKeyRecord.id)
|
|
|
|
TSPreKeyManager.clearPreKeyUpdateFailureCount()
|
|
TSPreKeyManager.clearSignedPreKeyRecords()
|
|
|
|
Logger.debug("done")
|
|
self.reportSuccess()
|
|
}.catch { error in
|
|
self.reportError(error)
|
|
}.retainUntilComplete()
|
|
* ================
|
|
*/
|
|
}
|
|
|
|
override public func didFail(error: Error) {
|
|
switch error {
|
|
case let networkManagerError as NetworkManagerError:
|
|
guard !networkManagerError.isNetworkError else {
|
|
Logger.debug("don't report SPK rotation failure w/ network error")
|
|
return
|
|
}
|
|
|
|
guard networkManagerError.statusCode >= 400 && networkManagerError.statusCode <= 599 else {
|
|
Logger.debug("don't report SPK rotation failure w/ non application error")
|
|
return
|
|
}
|
|
|
|
TSPreKeyManager.incrementPreKeyUpdateFailureCount()
|
|
default:
|
|
Logger.debug("don't report SPK rotation failure w/ non NetworkManager error: \(error)")
|
|
}
|
|
}
|
|
}
|