92 lines
3.2 KiB
Swift
92 lines
3.2 KiB
Swift
//
|
|
// Copyright (c) 2021 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
public class CallMessagePushPayload: CustomStringConvertible {
|
|
private static let identifierKey = "CallMessageRelayPayload"
|
|
public let identifier: String
|
|
|
|
fileprivate init() {
|
|
identifier = UUID().uuidString
|
|
}
|
|
|
|
public init?(_ payloadDict: [AnyHashable: Any]) {
|
|
guard let payloadId = payloadDict[Self.identifierKey] as? String else { return nil }
|
|
identifier = payloadId
|
|
}
|
|
|
|
public var payloadDict: [String: String] {
|
|
[Self.identifierKey: identifier]
|
|
}
|
|
|
|
public var description: String {
|
|
"\(type(of: self)): \(identifier.suffix(6))"
|
|
}
|
|
}
|
|
|
|
@objc
|
|
public class CallMessageRelay: NSObject {
|
|
private static let pendingCallMessageStore = SDSKeyValueStore(collection: "PendingCallMessageStore")
|
|
|
|
public static func handleVoipPayload(_ payload: CallMessagePushPayload) {
|
|
Logger.info("Handling incoming VoIP payload: \(payload)")
|
|
defer { Logger.info("Finished handling incoming VoIP payload: \(payload)") }
|
|
// Process all the pending call messages from the NSE in 1 batch.
|
|
// This should almost always be a batch of one.
|
|
databaseStorage.write { transaction in
|
|
defer { pendingCallMessageStore.removeAll(transaction: transaction) }
|
|
let pendingPayloads: [Payload]
|
|
|
|
do {
|
|
pendingPayloads = try pendingCallMessageStore.allCodableValues(transaction: transaction).sorted {
|
|
$0.envelope.timestamp < $1.envelope.timestamp
|
|
}
|
|
} catch {
|
|
owsFailDebug("Failed to read pending call messages \(error)")
|
|
return
|
|
}
|
|
|
|
Logger.info("Processing \(pendingPayloads.count) call messages relayed from the NSE.")
|
|
owsAssertDebug(pendingPayloads.count == 1, "Unexpectedly processing multiple messages from the NSE at once")
|
|
|
|
for payload in pendingPayloads {
|
|
messageManager.processEnvelope(
|
|
payload.envelope,
|
|
plaintextData: payload.plaintextData,
|
|
wasReceivedByUD: payload.wasReceivedByUD,
|
|
serverDeliveryTimestamp: payload.serverDeliveryTimestamp,
|
|
shouldDiscardVisibleMessages: false,
|
|
transaction: transaction
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
public static func enqueueCallMessageForMainApp(
|
|
envelope: SSKProtoEnvelope,
|
|
plaintextData: Data,
|
|
wasReceivedByUD: Bool,
|
|
serverDeliveryTimestamp: UInt64,
|
|
transaction: SDSAnyWriteTransaction
|
|
) throws -> CallMessagePushPayload {
|
|
let payload = Payload(
|
|
envelope: envelope,
|
|
plaintextData: plaintextData,
|
|
wasReceivedByUD: wasReceivedByUD,
|
|
serverDeliveryTimestamp: serverDeliveryTimestamp
|
|
)
|
|
|
|
try pendingCallMessageStore.setCodable(payload, key: "\(envelope.timestamp)", transaction: transaction)
|
|
return CallMessagePushPayload()
|
|
}
|
|
|
|
private struct Payload: Codable {
|
|
let envelope: SSKProtoEnvelope
|
|
let plaintextData: Data
|
|
let wasReceivedByUD: Bool
|
|
let serverDeliveryTimestamp: UInt64
|
|
}
|
|
}
|