mirror of https://github.com/oxen-io/session-ios
				
				
				
			
			You cannot select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
	
	
		
			117 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Objective-C
		
	
			
		
		
	
	
			117 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Objective-C
		
	
| //
 | |
| //  Copyright (c) 2018 Open Whisper Systems. All rights reserved.
 | |
| //
 | |
| 
 | |
| #import "OWSRecipientIdentity.h"
 | |
| #import "OWSIdentityManager.h"
 | |
| #import "OWSPrimaryStorage.h"
 | |
| #import <SignalCoreKit/Cryptography.h>
 | |
| #import <SessionMessagingKit/SessionMessagingKit-Swift.h>
 | |
| #import <YapDatabase/YapDatabase.h>
 | |
| 
 | |
| NS_ASSUME_NONNULL_BEGIN
 | |
| 
 | |
| NSString *OWSVerificationStateToString(OWSVerificationState verificationState)
 | |
| {
 | |
|     switch (verificationState) {
 | |
|         case OWSVerificationStateDefault:
 | |
|             return @"OWSVerificationStateDefault";
 | |
|         case OWSVerificationStateVerified:
 | |
|             return @"OWSVerificationStateVerified";
 | |
|         case OWSVerificationStateNoLongerVerified:
 | |
|             return @"OWSVerificationStateNoLongerVerified";
 | |
|     }
 | |
| }
 | |
| 
 | |
| @interface OWSRecipientIdentity ()
 | |
| 
 | |
| @property (atomic) OWSVerificationState verificationState;
 | |
| 
 | |
| @end
 | |
| 
 | |
| /**
 | |
|  * Record for a recipients identity key and some meta data around it used to make trust decisions.
 | |
|  *
 | |
|  * NOTE: Instances of this class MUST only be retrieved/persisted via it's internal `dbConnection`,
 | |
|  *       which makes some special accomodations to enforce consistency.
 | |
|  */
 | |
| @implementation OWSRecipientIdentity
 | |
| 
 | |
| - (instancetype)initWithCoder:(NSCoder *)coder
 | |
| {
 | |
|     self = [super initWithCoder:coder];
 | |
| 
 | |
|     if (self) {
 | |
|         if (![coder decodeObjectForKey:@"verificationState"]) {
 | |
|             _verificationState = OWSVerificationStateDefault;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     return self;
 | |
| }
 | |
| 
 | |
| - (instancetype)initWithRecipientId:(NSString *)recipientId
 | |
|                         identityKey:(NSData *)identityKey
 | |
|                     isFirstKnownKey:(BOOL)isFirstKnownKey
 | |
|                           createdAt:(NSDate *)createdAt
 | |
|                   verificationState:(OWSVerificationState)verificationState
 | |
| {
 | |
|     self = [super initWithUniqueId:recipientId];
 | |
|     if (!self) {
 | |
|         return self;
 | |
|     }
 | |
|     
 | |
|     _recipientId = recipientId;
 | |
|     _identityKey = identityKey;
 | |
|     _isFirstKnownKey = isFirstKnownKey;
 | |
|     _createdAt = createdAt;
 | |
|     _verificationState = verificationState;
 | |
| 
 | |
|     return self;
 | |
| }
 | |
| 
 | |
| - (void)updateWithVerificationState:(OWSVerificationState)verificationState
 | |
|                         transaction:(YapDatabaseReadWriteTransaction *)transaction
 | |
| {
 | |
|     // Ensure changes are persisted without clobbering any work done on another thread or instance.
 | |
|     [self updateWithChangeBlock:^(OWSRecipientIdentity *_Nonnull obj) {
 | |
|         obj.verificationState = verificationState;
 | |
|     }
 | |
|                     transaction:transaction];
 | |
| }
 | |
| 
 | |
| - (void)updateWithChangeBlock:(void (^)(OWSRecipientIdentity *obj))changeBlock
 | |
|                   transaction:(YapDatabaseReadWriteTransaction *)transaction
 | |
| {
 | |
|     changeBlock(self);
 | |
| 
 | |
|     OWSRecipientIdentity *latest = [[self class] fetchObjectWithUniqueID:self.uniqueId transaction:transaction];
 | |
|     if (latest == nil) {
 | |
|         [self saveWithTransaction:transaction];
 | |
|         return;
 | |
|     }
 | |
| 
 | |
|     changeBlock(latest);
 | |
|     [latest saveWithTransaction:transaction];
 | |
| }
 | |
| 
 | |
| - (void)updateWithChangeBlock:(void (^)(OWSRecipientIdentity *obj))changeBlock
 | |
| {
 | |
|     changeBlock(self);
 | |
| 
 | |
|     [LKStorage writeSyncWithBlock:^(YapDatabaseReadWriteTransaction *_Nonnull transaction) {
 | |
|         OWSRecipientIdentity *latest = [[self class] fetchObjectWithUniqueID:self.uniqueId transaction:transaction];
 | |
|         if (latest == nil) {
 | |
|             [self saveWithTransaction:transaction];
 | |
|             return;
 | |
|         }
 | |
|         
 | |
|         changeBlock(latest);
 | |
|         [latest saveWithTransaction:transaction];
 | |
|     }];
 | |
| }
 | |
| 
 | |
| @end
 | |
| 
 | |
| NS_ASSUME_NONNULL_END
 |