337 lines
13 KiB
Objective-C
337 lines
13 KiB
Objective-C
//
|
|
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "OWSFingerprint.h"
|
|
#import "OWSError.h"
|
|
#import <AxolotlKit/NSData+keyVersionByte.h>
|
|
#import <CommonCrypto/CommonDigest.h>
|
|
#import <SignalCoreKit/NSData+OWS.h>
|
|
#import <SignalServiceKit/SignalServiceKit-Swift.h>
|
|
#import <UIKit/UIImage.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
static uint32_t const OWSFingerprintHashingVersion = 0;
|
|
static uint32_t const OWSFingerprintScannableFormatVersion = 1;
|
|
static uint32_t const OWSFingerprintDefaultHashIterations = 5200;
|
|
|
|
@interface OWSFingerprint ()
|
|
|
|
@property (nonatomic, readonly) NSUInteger hashIterations;
|
|
@property (nonatomic, readonly) NSString *text;
|
|
@property (nonatomic, readonly) NSData *myFingerprintData;
|
|
@property (nonatomic, readonly) NSData *theirFingerprintData;
|
|
@property (nonatomic, readonly) NSString *theirName;
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation OWSFingerprint
|
|
|
|
- (instancetype)initWithMyStableId:(NSString *)myStableId
|
|
myIdentityKey:(NSData *)myIdentityKeyWithoutKeyType
|
|
theirStableId:(NSString *)theirStableId
|
|
theirIdentityKey:(NSData *)theirIdentityKeyWithoutKeyType
|
|
theirName:(NSString *)theirName
|
|
hashIterations:(uint32_t)hashIterations
|
|
{
|
|
OWSAssertDebug(theirIdentityKeyWithoutKeyType.length == 32);
|
|
OWSAssertDebug(myIdentityKeyWithoutKeyType.length == 32);
|
|
|
|
self = [super init];
|
|
if (!self) {
|
|
return self;
|
|
}
|
|
|
|
_myStableIdData = [myStableId dataUsingEncoding:NSUTF8StringEncoding];
|
|
_myIdentityKey = [myIdentityKeyWithoutKeyType prependKeyType];
|
|
_theirStableId = theirStableId;
|
|
_theirStableIdData = [theirStableId dataUsingEncoding:NSUTF8StringEncoding];
|
|
_theirIdentityKey = [theirIdentityKeyWithoutKeyType prependKeyType];
|
|
_theirName = theirName;
|
|
_hashIterations = hashIterations;
|
|
|
|
_myFingerprintData = [self dataForStableId:_myStableIdData publicKey:_myIdentityKey];
|
|
_theirFingerprintData = [self dataForStableId:_theirStableIdData publicKey:_theirIdentityKey];
|
|
|
|
return self;
|
|
}
|
|
|
|
+ (instancetype)fingerprintWithMyStableId:(NSString *)myStableId
|
|
myIdentityKey:(NSData *)myIdentityKeyWithoutKeyType
|
|
theirStableId:(NSString *)theirStableId
|
|
theirIdentityKey:(NSData *)theirIdentityKeyWithoutKeyType
|
|
theirName:(NSString *)theirName
|
|
hashIterations:(uint32_t)hashIterations
|
|
{
|
|
return [[self alloc] initWithMyStableId:myStableId
|
|
myIdentityKey:myIdentityKeyWithoutKeyType
|
|
theirStableId:theirStableId
|
|
theirIdentityKey:theirIdentityKeyWithoutKeyType
|
|
theirName:theirName
|
|
hashIterations:hashIterations];
|
|
}
|
|
|
|
+ (instancetype)fingerprintWithMyStableId:(NSString *)myStableId
|
|
myIdentityKey:(NSData *)myIdentityKeyWithoutKeyType
|
|
theirStableId:(NSString *)theirStableId
|
|
theirIdentityKey:(NSData *)theirIdentityKeyWithoutKeyType
|
|
theirName:(NSString *)theirName
|
|
{
|
|
return [[self alloc] initWithMyStableId:myStableId
|
|
myIdentityKey:myIdentityKeyWithoutKeyType
|
|
theirStableId:theirStableId
|
|
theirIdentityKey:theirIdentityKeyWithoutKeyType
|
|
theirName:theirName
|
|
hashIterations:OWSFingerprintDefaultHashIterations];
|
|
}
|
|
|
|
- (BOOL)matchesLogicalFingerprintsData:(NSData *)data error:(NSError **)error
|
|
{
|
|
OWSAssertDebug(data.length > 0);
|
|
OWSAssertDebug(error);
|
|
|
|
*error = nil;
|
|
FingerprintProtoLogicalFingerprints *_Nullable logicalFingerprints;
|
|
logicalFingerprints = [FingerprintProtoLogicalFingerprints parseData:data error:error];
|
|
if (!logicalFingerprints || *error) {
|
|
OWSFailDebug(@"fingerprint failure: %@", *error);
|
|
|
|
NSString *description = NSLocalizedString(@"PRIVACY_VERIFICATION_FAILURE_INVALID_QRCODE", @"alert body");
|
|
*error = OWSErrorWithCodeDescription(OWSErrorCodePrivacyVerificationFailure, description);
|
|
return NO;
|
|
}
|
|
|
|
if (logicalFingerprints.version < OWSFingerprintScannableFormatVersion) {
|
|
OWSLogWarn(@"Verification failed. They're running an old version.");
|
|
NSString *description
|
|
= NSLocalizedString(@"PRIVACY_VERIFICATION_FAILED_WITH_OLD_REMOTE_VERSION", @"alert body");
|
|
*error = OWSErrorWithCodeDescription(OWSErrorCodePrivacyVerificationFailure, description);
|
|
return NO;
|
|
}
|
|
|
|
if (logicalFingerprints.version > OWSFingerprintScannableFormatVersion) {
|
|
OWSLogWarn(@"Verification failed. We're running an old version.");
|
|
NSString *description = NSLocalizedString(@"PRIVACY_VERIFICATION_FAILED_WITH_OLD_LOCAL_VERSION", @"alert body");
|
|
*error = OWSErrorWithCodeDescription(OWSErrorCodePrivacyVerificationFailure, description);
|
|
return NO;
|
|
}
|
|
|
|
// Their local is *our* remote.
|
|
FingerprintProtoLogicalFingerprint *localFingerprint = logicalFingerprints.remoteFingerprint;
|
|
FingerprintProtoLogicalFingerprint *remoteFingerprint = logicalFingerprints.localFingerprint;
|
|
|
|
if (![remoteFingerprint.identityData isEqual:[self scannableData:self.theirFingerprintData]]) {
|
|
OWSLogWarn(@"Verification failed. We have the wrong fingerprint for them");
|
|
NSString *descriptionFormat = NSLocalizedString(@"PRIVACY_VERIFICATION_FAILED_I_HAVE_WRONG_KEY_FOR_THEM",
|
|
@"Alert body when verifying with {{contact name}}");
|
|
NSString *description = [NSString stringWithFormat:descriptionFormat, self.theirName];
|
|
*error = OWSErrorWithCodeDescription(OWSErrorCodePrivacyVerificationFailure, description);
|
|
return NO;
|
|
}
|
|
|
|
if (![localFingerprint.identityData isEqual:[self scannableData:self.myFingerprintData]]) {
|
|
OWSLogWarn(@"Verification failed. They have the wrong fingerprint for us");
|
|
NSString *descriptionFormat = NSLocalizedString(@"PRIVACY_VERIFICATION_FAILED_THEY_HAVE_WRONG_KEY_FOR_ME",
|
|
@"Alert body when verifying with {{contact name}}");
|
|
NSString *description = [NSString stringWithFormat:descriptionFormat, self.theirName];
|
|
*error = OWSErrorWithCodeDescription(OWSErrorCodePrivacyVerificationFailure, description);
|
|
return NO;
|
|
}
|
|
|
|
OWSLogWarn(@"Verification Succeeded.");
|
|
return YES;
|
|
}
|
|
|
|
- (NSString *)text
|
|
{
|
|
NSString *myDisplayString = [self stringForFingerprintData:self.myFingerprintData];
|
|
NSString *theirDisplayString = [self stringForFingerprintData:self.theirFingerprintData];
|
|
|
|
if ([theirDisplayString compare:myDisplayString] == NSOrderedAscending) {
|
|
return [NSString stringWithFormat:@"%@%@", theirDisplayString, myDisplayString];
|
|
} else {
|
|
return [NSString stringWithFormat:@"%@%@", myDisplayString, theirDisplayString];
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Formats numeric fingerprint, 3 lines in groups of 5 digits.
|
|
*/
|
|
- (NSString *)displayableText
|
|
{
|
|
NSString *input = self.text;
|
|
|
|
NSMutableArray<NSString *> *lines = [NSMutableArray new];
|
|
|
|
NSUInteger lineLength = self.text.length / 3;
|
|
for (uint i = 0; i < 3; i++) {
|
|
NSString *line = [input substringWithRange:NSMakeRange(i * lineLength, lineLength)];
|
|
|
|
NSMutableArray<NSString *> *chunks = [NSMutableArray new];
|
|
for (uint i = 0; i < line.length / 5; i++) {
|
|
NSString *nextChunk = [line substringWithRange:NSMakeRange(i * 5, 5)];
|
|
[chunks addObject:nextChunk];
|
|
}
|
|
[lines addObject:[chunks componentsJoinedByString:@" "]];
|
|
}
|
|
|
|
return [lines componentsJoinedByString:@"\n"];
|
|
}
|
|
|
|
|
|
- (NSData *)dataFromShort:(uint32_t)aShort
|
|
{
|
|
uint8_t bytes[] = {
|
|
((uint8_t)(aShort & 0xFF00) >> 8),
|
|
(uint8_t)(aShort & 0x00FF)
|
|
};
|
|
|
|
return [NSData dataWithBytes:bytes length:2];
|
|
}
|
|
|
|
/**
|
|
* An identifier for a mutable public key, belonging to an immutable identifier (stableId).
|
|
*
|
|
* This method is intended to be somewhat expensive to produce in order to be brute force adverse.
|
|
*
|
|
* @param stableIdData
|
|
* Immutable global identifier e.g. Signal Identifier, an e164 formatted phone number encoded as UTF-8 data
|
|
* @param publicKey
|
|
* The current public key for <stableId>
|
|
* @return
|
|
* All-number textual representation
|
|
*/
|
|
- (NSData *)dataForStableId:(NSData *)stableIdData publicKey:(NSData *)publicKey
|
|
{
|
|
OWSAssertDebug(stableIdData);
|
|
OWSAssertDebug(publicKey);
|
|
|
|
NSData *versionData = [self dataFromShort:OWSFingerprintHashingVersion];
|
|
NSMutableData *hash = [versionData mutableCopy];
|
|
[hash appendData:publicKey];
|
|
[hash appendData:stableIdData];
|
|
|
|
NSMutableData *_Nullable digestData = [[NSMutableData alloc] initWithLength:CC_SHA512_DIGEST_LENGTH];
|
|
if (!digestData) {
|
|
@throw [NSException exceptionWithName:NSGenericException reason:@"Couldn't allocate buffer." userInfo:nil];
|
|
}
|
|
for (int i = 0; i < self.hashIterations; i++) {
|
|
[hash appendData:publicKey];
|
|
|
|
if (hash.length >= UINT32_MAX) {
|
|
@throw [NSException exceptionWithName:@"Oversize Data" reason:@"Oversize hash." userInfo:nil];
|
|
}
|
|
|
|
CC_SHA512(hash.bytes, (uint32_t)hash.length, digestData.mutableBytes);
|
|
// TODO get rid of this loop-allocation
|
|
hash = [digestData mutableCopy];
|
|
}
|
|
|
|
return [hash copy];
|
|
}
|
|
|
|
|
|
- (NSString *)stringForFingerprintData:(NSData *)data
|
|
{
|
|
OWSAssertDebug(data);
|
|
|
|
return [NSString stringWithFormat:@"%@%@%@%@%@%@",
|
|
[self encodedChunkFromData:data offset:0],
|
|
[self encodedChunkFromData:data offset:5],
|
|
[self encodedChunkFromData:data offset:10],
|
|
[self encodedChunkFromData:data offset:15],
|
|
[self encodedChunkFromData:data offset:20],
|
|
[self encodedChunkFromData:data offset:25]];
|
|
}
|
|
|
|
- (NSString *)encodedChunkFromData:(NSData *)data offset:(uint)offset
|
|
{
|
|
OWSAssertDebug(data);
|
|
|
|
uint8_t fiveBytes[5];
|
|
[data getBytes:fiveBytes range:NSMakeRange(offset, 5)];
|
|
|
|
int chunk = [self uint64From5Bytes:fiveBytes] % 100000;
|
|
return [NSString stringWithFormat:@"%05d", chunk];
|
|
}
|
|
|
|
- (int64_t)uint64From5Bytes:(uint8_t[])bytes
|
|
{
|
|
int64_t result = ((bytes[0] & 0xffLL) << 32) |
|
|
((bytes[1] & 0xffLL) << 24) |
|
|
((bytes[2] & 0xffLL) << 16) |
|
|
((bytes[3] & 0xffLL) << 8) |
|
|
((bytes[4] & 0xffLL));
|
|
|
|
return result;
|
|
}
|
|
|
|
- (NSData *)scannableData:(NSData *)data
|
|
{
|
|
return [data subdataWithRange:NSMakeRange(0, 32)];
|
|
}
|
|
|
|
- (nullable UIImage *)image
|
|
{
|
|
FingerprintProtoLogicalFingerprintBuilder *remoteFingerprintBuilder =
|
|
[FingerprintProtoLogicalFingerprintBuilder new];
|
|
remoteFingerprintBuilder.identityData = [self scannableData:self.theirFingerprintData];
|
|
NSError *error;
|
|
FingerprintProtoLogicalFingerprint *_Nullable remoteFingerprint =
|
|
[remoteFingerprintBuilder buildAndReturnError:&error];
|
|
if (!remoteFingerprint || error) {
|
|
OWSFailDebug(@"could not build proto: %@", error);
|
|
return nil;
|
|
}
|
|
|
|
FingerprintProtoLogicalFingerprintBuilder *localFingerprintBuilder =
|
|
[FingerprintProtoLogicalFingerprintBuilder new];
|
|
localFingerprintBuilder.identityData = [self scannableData:self.myFingerprintData];
|
|
FingerprintProtoLogicalFingerprint *_Nullable localFingerprint =
|
|
[localFingerprintBuilder buildAndReturnError:&error];
|
|
if (!localFingerprint || error) {
|
|
OWSFailDebug(@"could not build proto: %@", error);
|
|
return nil;
|
|
}
|
|
|
|
FingerprintProtoLogicalFingerprintsBuilder *logicalFingerprintsBuilder =
|
|
[[FingerprintProtoLogicalFingerprintsBuilder alloc] initWithVersion:OWSFingerprintScannableFormatVersion
|
|
localFingerprint:localFingerprint
|
|
remoteFingerprint:remoteFingerprint];
|
|
|
|
// Build ByteMode QR (Latin-1 encodable data)
|
|
NSData *_Nullable fingerprintData = [logicalFingerprintsBuilder buildSerializedDataAndReturnError:&error];
|
|
if (!fingerprintData || error) {
|
|
OWSFailDebug(@"could not serialize proto: %@", error);
|
|
return nil;
|
|
}
|
|
|
|
OWSLogDebug(@"Building fingerprint with data: %@", fingerprintData);
|
|
|
|
CIFilter *filter = [CIFilter filterWithName:@"CIQRCodeGenerator"];
|
|
[filter setDefaults];
|
|
[filter setValue:fingerprintData forKey:@"inputMessage"];
|
|
|
|
CIImage *ciImage = [filter outputImage];
|
|
if (!ciImage) {
|
|
OWSLogError(@"Failed to create QR image from fingerprint text: %@", self.text);
|
|
return nil;
|
|
}
|
|
|
|
// UIImages backed by a CIImage won't render without antialiasing, so we convert the backign image to a CGImage,
|
|
// which can be scaled crisply.
|
|
CIContext *context = [CIContext contextWithOptions:nil];
|
|
CGImageRef cgImage = [context createCGImage:ciImage fromRect:ciImage.extent];
|
|
UIImage *qrImage = [UIImage imageWithCGImage:cgImage];
|
|
CGImageRelease(cgImage);
|
|
|
|
return qrImage;
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|