83 lines
2.9 KiB
Objective-C
83 lines
2.9 KiB
Objective-C
//
|
|
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "OWSProvisioningMessage.h"
|
|
#import "OWSProvisioningCipher.h"
|
|
#import "OWSProvisioningProtos.pb.h"
|
|
#import <Curve25519Kit/Curve25519.h>
|
|
#import <AxolotlKit/NSData+keyVersionByte.h>
|
|
#import <HKDFKit/HKDFKit.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
@interface OWSProvisioningMessage ()
|
|
|
|
@property (nonatomic, readonly) NSData *myPublicKey;
|
|
@property (nonatomic, readonly) NSData *myPrivateKey;
|
|
@property (nonatomic, readonly) NSString *accountIdentifier;
|
|
@property (nonatomic, readonly) NSData *theirPublicKey;
|
|
@property (nonatomic, readonly) NSData *profileKey;
|
|
@property (nonatomic, readonly) BOOL areReadReceiptsEnabled;
|
|
@property (nonatomic, readonly) NSString *provisioningCode;
|
|
|
|
@end
|
|
|
|
@implementation OWSProvisioningMessage
|
|
|
|
- (instancetype)initWithMyPublicKey:(NSData *)myPublicKey
|
|
myPrivateKey:(NSData *)myPrivateKey
|
|
theirPublicKey:(NSData *)theirPublicKey
|
|
accountIdentifier:(NSString *)accountIdentifier
|
|
profileKey:(NSData *)profileKey
|
|
readReceiptsEnabled:(BOOL)areReadReceiptsEnabled
|
|
provisioningCode:(NSString *)provisioningCode
|
|
{
|
|
self = [super init];
|
|
if (!self) {
|
|
return self;
|
|
}
|
|
|
|
_myPublicKey = myPublicKey;
|
|
_myPrivateKey = myPrivateKey;
|
|
_theirPublicKey = theirPublicKey;
|
|
_accountIdentifier = accountIdentifier;
|
|
_profileKey = profileKey;
|
|
_areReadReceiptsEnabled = areReadReceiptsEnabled;
|
|
_provisioningCode = provisioningCode;
|
|
|
|
return self;
|
|
}
|
|
|
|
- (nullable NSData *)buildEncryptedMessageBody
|
|
{
|
|
OWSProvisioningProtosProvisionMessageBuilder *messageBuilder = [OWSProvisioningProtosProvisionMessageBuilder new];
|
|
messageBuilder.identityKeyPublic = self.myPublicKey;
|
|
messageBuilder.identityKeyPrivate = self.myPrivateKey;
|
|
messageBuilder.number = self.accountIdentifier;
|
|
messageBuilder.provisioningCode = self.provisioningCode;
|
|
messageBuilder.userAgent = @"OWI";
|
|
messageBuilder.readReceipts = self.areReadReceiptsEnabled;
|
|
messageBuilder.profileKey = self.profileKey;
|
|
|
|
NSData *plainTextProvisionMessage = [[messageBuilder build] data];
|
|
|
|
OWSProvisioningCipher *cipher = [[OWSProvisioningCipher alloc] initWithTheirPublicKey:self.theirPublicKey];
|
|
NSData *_Nullable encryptedProvisionMessage = [cipher encrypt:plainTextProvisionMessage];
|
|
if (encryptedProvisionMessage == nil) {
|
|
OWSFail(@"Failed to encrypt provision message");
|
|
return nil;
|
|
}
|
|
|
|
OWSProvisioningProtosProvisionEnvelopeBuilder *envelopeBuilder = [OWSProvisioningProtosProvisionEnvelopeBuilder new];
|
|
// Note that this is a one-time-use *cipher* public key, not our Signal *identity* public key
|
|
envelopeBuilder.publicKey = [cipher.ourPublicKey prependKeyType];
|
|
envelopeBuilder.body = encryptedProvisionMessage;
|
|
|
|
return [[envelopeBuilder build] data];
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|