From 79e83d81ac5538d5053cbb84706bbb6cefff9ed5 Mon Sep 17 00:00:00 2001
From: nielsandriesse <andriesseniels@gmail.com>
Date: Tue, 16 Jun 2020 16:53:53 +1000
Subject: [PATCH] Fix crash

---
 .../Session Management/SessionManagementProtocol.swift        | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/SignalServiceKit/src/Loki/Protocol/Session Management/SessionManagementProtocol.swift b/SignalServiceKit/src/Loki/Protocol/Session Management/SessionManagementProtocol.swift
index 2f3979c85..3cea52686 100644
--- a/SignalServiceKit/src/Loki/Protocol/Session Management/SessionManagementProtocol.swift	
+++ b/SignalServiceKit/src/Loki/Protocol/Session Management/SessionManagementProtocol.swift	
@@ -181,13 +181,13 @@ public final class SessionManagementProtocol : NSObject {
     @objc(isSessionRestorationRequest:)
     public static func isSessionRestorationRequest(_ dataMessage: SSKProtoDataMessage) -> Bool {
         let sessionRestoreFlag = SSKProtoDataMessage.SSKProtoDataMessageFlags.sessionRestore
-        return dataMessage.flags & UInt32(sessionRestoreFlag.rawValue) != 0
+        return dataMessage.hasFlags && (dataMessage.flags & UInt32(sessionRestoreFlag.rawValue) != 0)
     }
 
     @objc(isSessionRequestMessage:)
     public static func isSessionRequestMessage(_ dataMessage: SSKProtoDataMessage) -> Bool {
         let sessionRequestFlag = SSKProtoDataMessage.SSKProtoDataMessageFlags.sessionRequest
-        return dataMessage.flags & UInt32(sessionRequestFlag.rawValue) != 0
+        return dataMessage.hasFlags && (dataMessage.flags & UInt32(sessionRequestFlag.rawValue) != 0)
     }
 
     @objc(handleSessionRequestMessage:wrappedIn:using:)