100 lines
2.3 KiB
Objective-C
100 lines
2.3 KiB
Objective-C
//
|
|
// AESGCMTest.m
|
|
// Monal Tests
|
|
//
|
|
// Created by Anurodh Pokharel on 1/7/20.
|
|
// Copyright © 2020 Monal.im. All rights reserved.
|
|
//
|
|
|
|
#import <XCTest/XCTest.h>
|
|
#import "AESGcm.h"
|
|
|
|
@interface AESGCMTest : XCTestCase
|
|
|
|
@end
|
|
|
|
@implementation AESGCMTest
|
|
|
|
- (void)setUp {
|
|
// Put setup code here. This method is called before the invocation of each test method in the class.
|
|
}
|
|
|
|
- (void)tearDown {
|
|
// Put teardown code here. This method is called after the invocation of each test method in the class.
|
|
}
|
|
|
|
-(void) encryptWithSize:(int) size
|
|
{
|
|
NSString* plaintext = @"ABCDEFGHIKLMOPQRSTUVWXYZ1234567890!\"§$%&/()=?*+#-.,;:_";
|
|
NSData* plaintextUTF8 = [plaintext dataUsingEncoding:NSUTF8StringEncoding];
|
|
MLEncryptedPayload* payload = [AESGcm encrypt:plaintextUTF8 keySize:size];
|
|
|
|
NSData* key = [payload.key subdataWithRange:NSMakeRange(0, size)];
|
|
NSData* auth = [payload.key subdataWithRange:NSMakeRange(size, 16)];
|
|
|
|
NSData* decryptedResult = [AESGcm decrypt:payload.body withKey:key andIv:payload.iv withAuth:auth];
|
|
NSString* decryptedResultString = [[NSString alloc] initWithData:decryptedResult encoding:NSUTF8StringEncoding];
|
|
XCTAssert([decryptedResultString isEqualToString:plaintext]);
|
|
}
|
|
|
|
-(void) testEncrypt16
|
|
{
|
|
[self encryptWithSize:16];
|
|
}
|
|
|
|
-(void) testEncrypt32
|
|
{
|
|
[self encryptWithSize:32];
|
|
}
|
|
|
|
/*
|
|
* This test doesn't check for real IV uniqueness!
|
|
*/
|
|
-(void) testGenIV
|
|
{
|
|
const UInt32 ivCnt = 40000;
|
|
NSMutableArray<NSData*>* ivArray = [[NSMutableArray<NSData*> alloc] init];
|
|
|
|
for(UInt32 i = 0; i < ivCnt; i++)
|
|
{
|
|
NSData* iv = [AESGcm genIV];
|
|
XCTAssertFalse([ivArray containsObject:iv], "IV should be unique");
|
|
[ivArray addObject:iv];
|
|
}
|
|
}
|
|
|
|
|
|
/*
|
|
* This test doesn't check for real key uniqueness!
|
|
*/
|
|
-(void) genKeyWithSize:(uint) size
|
|
{
|
|
const UInt32 ivCnt = 40000;
|
|
NSMutableArray<NSData*>* ivArray = [[NSMutableArray<NSData*> alloc] init];
|
|
|
|
for(UInt32 i = 0; i < ivCnt; i++)
|
|
{
|
|
NSData* iv = [AESGcm genKey:size];
|
|
XCTAssertFalse([ivArray containsObject:iv], "key should be unique");
|
|
[ivArray addObject:iv];
|
|
}
|
|
}
|
|
|
|
/*
|
|
* This test doesn't check for real key uniqueness!
|
|
*/
|
|
-(void) testGenKey16
|
|
{
|
|
[self genKeyWithSize:16];
|
|
}
|
|
|
|
/*
|
|
* This test doesn't check for real key uniqueness!
|
|
*/
|
|
-(void) testGenKey32
|
|
{
|
|
[self genKeyWithSize:32];
|
|
}
|
|
|
|
@end
|