86 lines
3.0 KiB
Objective-C
86 lines
3.0 KiB
Objective-C
//
|
|
// TSInvalidIdentityKeySendingErrorMessage.m
|
|
// Signal
|
|
//
|
|
// Created by Frederic Jacobs on 15/02/15.
|
|
// Copyright (c) 2015 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "TSOutgoingMessage.h"
|
|
|
|
#import <AxolotlKit/NSData+keyVersionByte.h>
|
|
|
|
#import "PreKeyBundle+jsonDict.h"
|
|
#import "TSInvalidIdentityKeySendingErrorMessage.h"
|
|
#import "TSFingerprintGenerator.h"
|
|
#import "TSErrorMessage_privateConstructor.h"
|
|
#import "TSStorageManager+IdentityKeyStore.h"
|
|
#import "TSMessagesManager+sendMessages.h"
|
|
|
|
@interface TSInvalidIdentityKeySendingErrorMessage()
|
|
|
|
@property (nonatomic, readonly) PreKeyBundle *preKeyBundle;
|
|
@property (nonatomic, readonly) NSString *recipientId;
|
|
@property (nonatomic, readonly) NSString *messageId;
|
|
|
|
@end
|
|
|
|
@implementation TSInvalidIdentityKeySendingErrorMessage
|
|
|
|
- (instancetype)initWithOutgoingMessage:(TSOutgoingMessage*)message
|
|
inThread:(TSThread*)thread
|
|
forRecipient:(NSString*)recipientId
|
|
preKeyBundle:(PreKeyBundle*)preKeyBundle
|
|
transaction:(YapDatabaseReadWriteTransaction*)transaction
|
|
{
|
|
self = [super initWithTimestamp:message.timestamp inThread:thread failedMessageType:TSErrorMessageWrongTrustedIdentityKey];
|
|
|
|
if (self) {
|
|
_messageId = message.uniqueId;
|
|
_preKeyBundle = preKeyBundle;
|
|
_recipientId = recipientId;
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
+ (instancetype)untrustedKeyWithOutgoingMessage:(TSOutgoingMessage*)outgoingMessage
|
|
inThread:(TSThread*)thread
|
|
forRecipient:(NSString*)recipientId
|
|
preKeyBundle:(PreKeyBundle*)preKeyBundle
|
|
withTransaction:(YapDatabaseReadWriteTransaction*)transaction
|
|
{
|
|
TSInvalidIdentityKeySendingErrorMessage *message = [[self alloc] initWithOutgoingMessage:outgoingMessage inThread:thread forRecipient:recipientId preKeyBundle:preKeyBundle transaction:transaction];
|
|
return message;
|
|
}
|
|
|
|
- (void)acceptNewIdentityKey {
|
|
[[TSStorageManager sharedManager] saveRemoteIdentity:[self newKey] recipientId:_recipientId];
|
|
|
|
__block TSOutgoingMessage *message;
|
|
__block TSThread *thread;
|
|
|
|
[[TSStorageManager sharedManager].newDatabaseConnection readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction) {
|
|
thread = [TSContactThread fetchObjectWithUniqueID:self.uniqueThreadId transaction:transaction];
|
|
message = [TSOutgoingMessage fetchObjectWithUniqueID:_messageId transaction:transaction];
|
|
|
|
[self removeWithTransaction:transaction];
|
|
}];
|
|
|
|
if (message) {
|
|
[[TSMessagesManager sharedManager] sendMessage:message inThread:thread success:nil failure:nil];
|
|
}
|
|
}
|
|
|
|
- (NSString*)newIdentityKey {
|
|
NSData *identityKey = [self newKey];
|
|
|
|
return [TSFingerprintGenerator getFingerprintForDisplay:identityKey];
|
|
}
|
|
|
|
- (NSData*)newKey {
|
|
return [self.preKeyBundle.identityKey removeKeyType];
|
|
}
|
|
|
|
@end
|