Handle nil state on session reset.

6 years ago · 99cfe7ca24
parent 28bc6b3714
commit 99cfe7ca24
1 changed files with 3 additions and 1 deletions
--- a/SignalServiceKit/src/Loki/Crypto/SessionCipher+Loki.m
+++ b/SignalServiceKit/src/Loki/Crypto/SessionCipher+Loki.m
@ -105,7 +105,9 @@ NSString *const kNSNotificationKey_ContactPubKey = @"kNSNotificationKey_ContactP
 {
    SessionRecord *record = [self.sessionStore loadSession:self.recipientId deviceId:self.deviceId protocolContext:protocolContext];
    [record removePreviousSessionStates];
-    [record setState:state];
+    
+    SessionState *newState = state == nil ? [SessionState new] : state;
+    [record setState:newState];
    
    [self.sessionStore storeSession:self.recipientId deviceId:self.deviceId session:record protocolContext:protocolContext];
 }