mirror of https://github.com/oxen-io/session-ios
Implement session request
parent
d407467279
commit
e8892a1f0f
@ -0,0 +1,13 @@
|
|||||||
|
|
||||||
|
public struct Configuration {
|
||||||
|
public let storage: SessionMessagingKitStorageProtocol
|
||||||
|
|
||||||
|
internal static var shared: Configuration!
|
||||||
|
}
|
||||||
|
|
||||||
|
public enum SessionMessagingKit { // Just to make the external API nice
|
||||||
|
|
||||||
|
public static func configure(with configuration: Configuration) {
|
||||||
|
Configuration.shared = configuration
|
||||||
|
}
|
||||||
|
}
|
@ -1,5 +1,70 @@
|
|||||||
|
import SessionUtilities
|
||||||
|
import SessionProtocolKit
|
||||||
|
|
||||||
@objc(SNSessionRequest)
|
@objc(SNSessionRequest)
|
||||||
public final class SessionRequest : ControlMessage {
|
public final class SessionRequest : ControlMessage {
|
||||||
|
private var preKeyBundle: PreKeyBundle?
|
||||||
|
|
||||||
|
// MARK: Initialization
|
||||||
|
init(preKeyBundle: PreKeyBundle) {
|
||||||
|
super.init()
|
||||||
|
self.preKeyBundle = preKeyBundle
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: Coding
|
||||||
|
public required init?(coder: NSCoder) {
|
||||||
|
super.init(coder: coder)
|
||||||
|
if let preKeyBundle = coder.decodeObject(forKey: "preKeyBundle") as! PreKeyBundle? { self.preKeyBundle = preKeyBundle }
|
||||||
|
}
|
||||||
|
|
||||||
|
public override func encode(with coder: NSCoder) {
|
||||||
|
super.encode(with: coder)
|
||||||
|
coder.encode(preKeyBundle, forKey: "preKeyBundle")
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARK: Proto Conversion
|
||||||
|
public override class func fromProto(_ proto: SNProtoContent) -> SessionRequest? {
|
||||||
|
guard proto.nullMessage != nil, let preKeyBundleProto = proto.prekeyBundleMessage else { return nil }
|
||||||
|
var registrationID: UInt32 = 0
|
||||||
|
Configuration.shared.storage.with { transaction in
|
||||||
|
registrationID = Configuration.shared.storage.getOrGenerateRegistrationID(using: transaction)
|
||||||
|
}
|
||||||
|
guard let preKeyBundle = PreKeyBundle(registrationId: Int32(registrationID),
|
||||||
|
deviceId: 1,
|
||||||
|
preKeyId: Int32(preKeyBundleProto.prekeyID),
|
||||||
|
preKeyPublic: preKeyBundleProto.prekey,
|
||||||
|
signedPreKeyPublic: preKeyBundleProto.signedKey,
|
||||||
|
signedPreKeyId: Int32(preKeyBundleProto.signedKeyID),
|
||||||
|
signedPreKeySignature: preKeyBundleProto.signature,
|
||||||
|
identityKey: preKeyBundleProto.identityKey) else { return nil }
|
||||||
|
return SessionRequest(preKeyBundle: preKeyBundle)
|
||||||
|
}
|
||||||
|
|
||||||
|
public override func toProto() -> SNProtoContent? {
|
||||||
|
guard let preKeyBundle = preKeyBundle else {
|
||||||
|
SNLog("Couldn't construct session request proto from: \(self).")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
let nullMessageProto = SNProtoNullMessage.builder()
|
||||||
|
let paddingSize = UInt.random(in: 0..<512) // random(in:) uses the system's default random generator, which is cryptographically secure
|
||||||
|
let padding = Data.getSecureRandomData(ofSize: paddingSize)!
|
||||||
|
nullMessageProto.setPadding(padding)
|
||||||
|
let preKeyBundleProto = SNProtoPrekeyBundleMessage.builder()
|
||||||
|
preKeyBundleProto.setIdentityKey(preKeyBundle.identityKey)
|
||||||
|
preKeyBundleProto.setDeviceID(UInt32(preKeyBundle.deviceId))
|
||||||
|
preKeyBundleProto.setPrekeyID(UInt32(preKeyBundle.preKeyId))
|
||||||
|
preKeyBundleProto.setPrekey(preKeyBundle.preKeyPublic)
|
||||||
|
preKeyBundleProto.setSignedKeyID(UInt32(preKeyBundle.signedPreKeyId))
|
||||||
|
preKeyBundleProto.setSignedKey(preKeyBundle.signedPreKeyPublic)
|
||||||
|
preKeyBundleProto.setSignature(preKeyBundle.signedPreKeySignature)
|
||||||
|
let contentProto = SNProtoContent.builder()
|
||||||
|
do {
|
||||||
|
contentProto.setNullMessage(try nullMessageProto.build())
|
||||||
|
contentProto.setPrekeyBundleMessage(try preKeyBundleProto.build())
|
||||||
|
return try contentProto.build()
|
||||||
|
} catch {
|
||||||
|
SNLog("Couldn't construct session request proto from: \(self).")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,8 @@
|
|||||||
|
import SessionProtocolKit
|
||||||
|
|
||||||
|
public protocol SessionMessagingKitStorageProtocol {
|
||||||
|
|
||||||
|
func with(_ work: (Any) -> Void)
|
||||||
|
|
||||||
|
func getOrGenerateRegistrationID(using transaction: Any) -> UInt32
|
||||||
|
}
|
Loading…
Reference in New Issue