From 3d2df68f19de0aee655a7eaa68b7a552e03fb698 Mon Sep 17 00:00:00 2001 From: nielsandriesse Date: Tue, 16 Jun 2020 16:55:24 +1000 Subject: [PATCH] Fix crash --- .../Session Management/SessionManagementProtocol.swift | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/SignalServiceKit/src/Loki/Protocol/Session Management/SessionManagementProtocol.swift b/SignalServiceKit/src/Loki/Protocol/Session Management/SessionManagementProtocol.swift index 3cea52686..4c82f3810 100644 --- a/SignalServiceKit/src/Loki/Protocol/Session Management/SessionManagementProtocol.swift +++ b/SignalServiceKit/src/Loki/Protocol/Session Management/SessionManagementProtocol.swift @@ -179,13 +179,15 @@ public final class SessionManagementProtocol : NSObject { } @objc(isSessionRestorationRequest:) - public static func isSessionRestorationRequest(_ dataMessage: SSKProtoDataMessage) -> Bool { + public static func isSessionRestorationRequest(_ dataMessage: SSKProtoDataMessage?) -> Bool { + guard let dataMessage = dataMessage else { return false } let sessionRestoreFlag = SSKProtoDataMessage.SSKProtoDataMessageFlags.sessionRestore return dataMessage.hasFlags && (dataMessage.flags & UInt32(sessionRestoreFlag.rawValue) != 0) } @objc(isSessionRequestMessage:) - public static func isSessionRequestMessage(_ dataMessage: SSKProtoDataMessage) -> Bool { + public static func isSessionRequestMessage(_ dataMessage: SSKProtoDataMessage?) -> Bool { + guard let dataMessage = dataMessage else { return false } let sessionRequestFlag = SSKProtoDataMessage.SSKProtoDataMessageFlags.sessionRequest return dataMessage.hasFlags && (dataMessage.flags & UInt32(sessionRequestFlag.rawValue) != 0) }