diff --git a/native/ios/Comm/TemporaryMessageStorage/EncryptedFileUtils.mm b/native/ios/Comm/TemporaryMessageStorage/EncryptedFileUtils.mm index 60b1b1238..6fb776dea 100644 --- a/native/ios/Comm/TemporaryMessageStorage/EncryptedFileUtils.mm +++ b/native/ios/Comm/TemporaryMessageStorage/EncryptedFileUtils.mm @@ -1,12 +1,88 @@ #import "EncryptedFileUtils.h" #import "CommSecureStoreIOSWrapper.h" #import "Logger.h" #import @interface EncryptedFileUtils () + (NSData *)_runCryptor:(NSData *)binary operation:(CCOperation)operation error:(NSError **)error; + (NSData *)_encryptData:(NSString *)data error:(NSError **)error; + (NSString *)_decryptData:(NSString *)data error:(NSError **)error; @end + +@implementation EncryptedFileUtils ++ (NSData *)_encryptData:(NSString *)data error:(NSError **)error { + NSData *encryptedData = [EncryptedFileUtils + _runCryptor:[data dataUsingEncoding:NSUTF8StringEncoding] + operation:kCCEncrypt + error:error]; + if (!encryptedData) { + return nil; + } + return [encryptedData base64EncodedDataWithOptions:0]; +} + ++ (NSString *)_decryptData:(NSString *)data error:(NSError **)error { + NSData *base64DecodedData = [[NSData alloc] initWithBase64EncodedString:data + options:0]; + NSString *decryptedData = [[NSString alloc] + initWithData:[EncryptedFileUtils _runCryptor:base64DecodedData + operation:kCCDecrypt + error:error] + encoding:NSUTF8StringEncoding]; + if (!decryptedData) { + return nil; + } + return decryptedData; +} + ++ (NSData *)_runCryptor:(NSData *)binary + operation:(CCOperation)operation + error:(NSError **)err { + NSString *keyString = + [[CommSecureStoreIOSWrapper sharedInstance] get:@"comm.encryptionKey"]; + if (!keyString) { + *err = [NSError + errorWithDomain:@"app.comm" + code:NSCoderValueNotFoundError + userInfo:@{ + NSLocalizedDescriptionKey : @"Encryption key not created yet" + }]; + return nil; + } + + NSUInteger AES256KeyByteCount = 32; + NSData *key = [[keyString substringToIndex:AES256KeyByteCount] + dataUsingEncoding:NSUTF8StringEncoding]; + NSMutableData *resultBinary = + [NSMutableData dataWithLength:binary.length + kCCBlockSizeAES128]; + + size_t processedBytes = 0; + CCCryptorStatus ccStatus = CCCrypt( + operation, + kCCAlgorithmAES, + kCCOptionPKCS7Padding, + key.bytes, + key.length, + nil, + binary.bytes, + binary.length, + resultBinary.mutableBytes, + resultBinary.length, + &processedBytes); + + resultBinary.length = processedBytes; + if (ccStatus != kCCSuccess) { + *err = [NSError + errorWithDomain:@"app.comm" + code:ccStatus + userInfo:@{ + NSLocalizedDescriptionKey : @"Cryptographic operation failed" + }]; + return nil; + } + return resultBinary; +} + +@end