TPDummySigningKey.m   [plain text]

/*
 * Copyright (c) 2017 Apple Inc. All Rights Reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 *
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this
 * file.
 *
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 *
 * @APPLE_LICENSE_HEADER_END@
 */

#import "TPDummySigningKey.h"

@interface TPDummySigningKey ()
@property (nonatomic, strong) NSData *publicKey;
@end


@implementation TPDummySigningKey

- (instancetype)initWithPublicKeyData:(NSData *)publicKey
{
    self = [super init];
    if (self) {
        _publicKey = publicKey;
        _privateKeyIsAvailable = YES;
    }
    return self;
}

- (NSData *)signatureForData:(NSData *)data withError:(NSError **)error
{
    if (self.privateKeyIsAvailable) {
        return [self signatureForData:data];
    } else {
        if (error) {
            *error = [NSError errorWithDomain:@"TPDummySigningKey" code:1 userInfo:nil];
        }
        return nil;
    }
}

- (NSData *)signatureForData:(NSData *)data
{
    // A really dumb hash that is just good enough for unit tests.
    NSUInteger hash = [self.publicKey hash] ^ [data hash];
    return [NSData dataWithBytes:&hash length:sizeof(hash)];
}

- (BOOL)checkSignature:(NSData *)sig matchesData:(NSData *)data
{
    return [sig isEqualToData:[self signatureForData:data]];
}

@end


@implementation TPDummySigningKeyFactory

- (id <TPSigningKey>)keyWithPublicKeyData:(NSData *)publicKey
{
    if (0 == publicKey.length) {
        return nil;
    }
    return [[TPDummySigningKey alloc] initWithPublicKeyData:publicKey];
}

+ (instancetype) dummySigningKeyFactory
{
    return [[TPDummySigningKeyFactory alloc] init];
}

@end