aboutsummaryrefslogtreecommitdiff
path: root/xcode/OLMKit/OLMUtility.m
blob: f9d47c042d764f72d6c5d692a5ea25d0e1e4c8b6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
//
//  OLMUtility.m
//  olm
//
//  Created by Chris Ballinger on 4/8/16.
//
//

#import "OLMUtility.h"

#include "olm/olm.h"

NSString *const OLMErrorDomain = @"org.matrix.olm";

@interface OLMUtility()

@property (nonatomic) OlmUtility *utility;

@end

@implementation OLMUtility

- (void) dealloc {
    olm_clear_utility(_utility);
    free(_utility);
}

- (BOOL) initializeUtilityMemory {
    size_t utilitySize = olm_utility_size();
    _utility = malloc(utilitySize);
    NSParameterAssert(_utility != nil);
    if (!_utility) {
        return NO;
    }
    _utility = olm_utility(_utility);
    NSParameterAssert(_utility != nil);
    if (!_utility) {
        return NO;
    }
    return YES;
}

- (instancetype) init {
    self = [super init];
    if (!self) {
        return nil;
    }
    BOOL success = [self initializeUtilityMemory];
    if (!success) {
        return nil;
    }
    return self;
}

- (NSString *)sha256:(NSData *)message {
    size_t length = olm_sha256_length(_utility);

    NSMutableData *shaData = [NSMutableData dataWithLength:length];
    if (!shaData) {
        return nil;
    }

    size_t result = olm_sha256(_utility, message.bytes, message.length, shaData.mutableBytes, shaData.length);
    if (result == olm_error()) {
        const char *error = olm_utility_last_error(_utility);
        NSLog(@"olm_sha256 error: %s", error);
        return nil;
    }
    
    NSString *sha = [[NSString alloc] initWithData:shaData encoding:NSUTF8StringEncoding];
    return sha;
}

- (BOOL)verifyEd25519Signature:(NSString*)signature key:(NSString*)key message:(NSData*)message error:(NSError**)error {

    NSData *keyData = [key dataUsingEncoding:NSUTF8StringEncoding];
    NSData *signatureData = [signature dataUsingEncoding:NSUTF8StringEncoding];

    size_t result = olm_ed25519_verify(_utility,
                                       keyData.bytes, keyData.length,
                                       message.bytes, message.length,
                                       (void*)signatureData.bytes, signatureData.length
                                       );

    if (result == olm_error()) {
        if (error) {
            NSDictionary *userInfo = @{NSLocalizedFailureReasonErrorKey: [NSString stringWithUTF8String:olm_utility_last_error(_utility)]};

            // @TODO
            *error = [[NSError alloc] initWithDomain:@"OLMKitErrorDomain" code:0 userInfo:userInfo];
        }
        return NO;
    }
    else {
        return YES;
    }
}

+ (NSMutableData*) randomBytesOfLength:(NSUInteger)length {
    NSMutableData *randomData = [NSMutableData dataWithLength:length];
    if (!randomData) {
        return nil;
    }
    int result = SecRandomCopyBytes(kSecRandomDefault, randomData.length, randomData.mutableBytes);
    if (result != 0) {
        return nil;
    }
    return randomData;
}

@end