From 99cfe7ca245db97fe960bf7bd3c9a99409445fe9 Mon Sep 17 00:00:00 2001 From: Mikunj Date: Mon, 22 Jul 2019 11:07:38 +1000 Subject: [PATCH] Handle nil state on session reset. --- SignalServiceKit/src/Loki/Crypto/SessionCipher+Loki.m | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/SignalServiceKit/src/Loki/Crypto/SessionCipher+Loki.m b/SignalServiceKit/src/Loki/Crypto/SessionCipher+Loki.m index a98e3b7c2..466cba9af 100644 --- 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]; }