NCEndToEndEncryption.m 47 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454
  1. //
  2. // NCEndToEndEncryption.m
  3. // Nextcloud
  4. //
  5. // Created by Marino Faggiana on 19/09/17.
  6. // Copyright © 2017 Marino Faggiana. All rights reserved.
  7. //
  8. // Author Marino Faggiana <marino.faggiana@nextcloud.com>
  9. //
  10. // This program is free software: you can redistribute it and/or modify
  11. // it under the terms of the GNU General Public License as published by
  12. // the Free Software Foundation, either version 3 of the License, or
  13. // (at your option) any later version.
  14. //
  15. // This program is distributed in the hope that it will be useful,
  16. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18. // GNU General Public License for more details.
  19. //
  20. // You should have received a copy of the GNU General Public License
  21. // along with this program. If not, see <http://www.gnu.org/licenses/>.
  22. //
  23. #import "NCEndToEndEncryption.h"
  24. #import "NCBridgeSwift.h"
  25. #import "CCUtility.h"
  26. #import <CommonCrypto/CommonDigest.h>
  27. #import <CommonCrypto/CommonKeyDerivation.h>
  28. #import <OpenSSL/OpenSSL.h>
  29. #define addName(field, value) X509_NAME_add_entry_by_txt(name, field, MBSTRING_ASC, (unsigned char *)value, -1, -1, 0); NSLog(@"%s: %s", field, value);
  30. #define IV_DELIMITER_ENCODED_OLD @"fA=="
  31. #define IV_DELIMITER_ENCODED @"|"
  32. #define PBKDF2_KEY_LENGTH 256
  33. //#define PBKDF2_SALT @"$4$YmBjm3hk$Qb74D5IUYwghUmzsMqeNFx5z0/8$"
  34. #define ASYMMETRIC_STRING_TEST @"Nextcloud a safe home for all your data"
  35. #define fileNameCertificate @"cert.pem"
  36. #define fileNameCSR @"csr.pem"
  37. #define fileNamePrivateKey @"privateKey.pem"
  38. #define fileNamePubliceKey @"publicKey.pem"
  39. #define streamBuffer 1024
  40. #define AES_KEY_128_LENGTH 16
  41. #define AES_KEY_256_LENGTH 32
  42. #define AES_IVEC_LENGTH 16
  43. #define AES_GCM_TAG_LENGTH 16
  44. #define AES_SALT_LENGTH 40
  45. @interface NCEndToEndEncryption ()
  46. {
  47. NSData *_privateKeyData;
  48. NSData *_publicKeyData;
  49. NSData *_csrData;
  50. }
  51. @end
  52. @implementation NCEndToEndEncryption
  53. //Singleton
  54. + (instancetype)sharedManager {
  55. static NCEndToEndEncryption *NCEndToEndEncryption = nil;
  56. static dispatch_once_t onceToken;
  57. dispatch_once(&onceToken, ^{
  58. NCEndToEndEncryption = [self new];
  59. });
  60. return NCEndToEndEncryption;
  61. }
  62. #
  63. #pragma mark - Generate Certificate X509 - CSR - Private Key
  64. #
  65. - (BOOL)generateCertificateX509WithUserId:(NSString *)userId directory:(NSString *)directory
  66. {
  67. OPENSSL_init();
  68. int ret;
  69. EVP_PKEY * pkey;
  70. pkey = EVP_PKEY_new();
  71. RSA * rsa;
  72. BIGNUM *bignum = BN_new();
  73. ret = BN_set_word(bignum, RSA_F4);
  74. if (ret != 1) {
  75. return NO;
  76. }
  77. rsa = RSA_new();
  78. ret = RSA_generate_key_ex(rsa, 2048, bignum, NULL);
  79. if (ret != 1) {
  80. return NO;
  81. }
  82. EVP_PKEY_assign_RSA(pkey, rsa);
  83. X509 * x509;
  84. x509 = X509_new();
  85. ASN1_INTEGER_set(X509_get_serialNumber(x509), 1);
  86. long notBefore = [[NSDate date] timeIntervalSinceDate:[NSDate date]];
  87. long notAfter = [[[NSDate date] dateByAddingTimeInterval:60*60*24*365*10] timeIntervalSinceDate:[NSDate date]]; // 10 year
  88. X509_gmtime_adj(X509_get_notBefore(x509), notBefore);
  89. X509_gmtime_adj(X509_get_notAfter(x509), notAfter);
  90. X509_set_pubkey(x509, pkey);
  91. X509_NAME * name;
  92. name = X509_get_subject_name(x509);
  93. // Now to add the subject name fields to the certificate
  94. // I use a macro here to make it cleaner.
  95. const unsigned char *cUserId = (const unsigned char *) [userId cStringUsingEncoding:NSUTF8StringEncoding];
  96. // Common Name = UserID.
  97. addName("CN", cUserId);
  98. // The organizational unit for the cert. Usually this is a department.
  99. addName("OU", "Certificate Authority");
  100. // The organization of the cert.
  101. addName("O", "Nextcloud");
  102. // The city of the organization.
  103. addName("L", "Vicenza");
  104. // The state/province of the organization.
  105. addName("S", "Italy");
  106. // The country (ISO 3166) of the organization
  107. addName("C", "IT");
  108. X509_set_issuer_name(x509, name);
  109. /*
  110. for (SANObject * san in self.options.sans) {
  111. if (!san.value || san.value.length <= 0) {
  112. continue;
  113. }
  114. NSString * prefix = san.type == SANObjectTypeIP ? @"IP:" : @"DNS:";
  115. NSString * value = [NSString stringWithFormat:@"%@%@", prefix, san.value];
  116. NSLog(@"Add subjectAltName %@", value);
  117. X509_EXTENSION * extension = NULL;
  118. ASN1_STRING * asnValue = ASN1_STRING_new();
  119. ASN1_STRING_set(asnValue, (const unsigned char *)[value UTF8String], (int)value.length);
  120. X509_EXTENSION_create_by_NID(&extension, NID_subject_alt_name, 0, asnValue);
  121. X509_add_ext(x509, extension, -1);
  122. }
  123. */
  124. // Specify the encryption algorithm of the signature.
  125. // SHA256 should suit your needs.
  126. if (X509_sign(x509, pkey, EVP_sha256()) < 0) {
  127. return NO;
  128. }
  129. X509_print_fp(stdout, x509);
  130. // Extract CSR, publicKey, privateKey
  131. int len;
  132. char *keyBytes;
  133. // CSR
  134. BIO *csrBIO = BIO_new(BIO_s_mem());
  135. X509_REQ *certReq = X509_to_X509_REQ(x509, pkey, EVP_sha256());
  136. PEM_write_bio_X509_REQ(csrBIO, certReq);
  137. len = BIO_pending(csrBIO);
  138. keyBytes = malloc(len);
  139. BIO_read(csrBIO, keyBytes, len);
  140. _csrData = [NSData dataWithBytes:keyBytes length:len];
  141. NSLog(@"[LOG] \n%@", [[NSString alloc] initWithData:_csrData encoding:NSUTF8StringEncoding]);
  142. // PublicKey
  143. BIO *publicKeyBIO = BIO_new(BIO_s_mem());
  144. PEM_write_bio_PUBKEY(publicKeyBIO, pkey);
  145. len = BIO_pending(publicKeyBIO);
  146. keyBytes = malloc(len);
  147. BIO_read(publicKeyBIO, keyBytes, len);
  148. _publicKeyData = [NSData dataWithBytes:keyBytes length:len];
  149. self.generatedPublicKey = [[NSString alloc] initWithData:_publicKeyData encoding:NSUTF8StringEncoding];
  150. NSLog(@"[LOG] \n%@", self.generatedPublicKey);
  151. // PrivateKey
  152. BIO *privateKeyBIO = BIO_new(BIO_s_mem());
  153. PEM_write_bio_PKCS8PrivateKey(privateKeyBIO, pkey, NULL, NULL, 0, NULL, NULL);
  154. len = BIO_pending(privateKeyBIO);
  155. keyBytes = malloc(len);
  156. BIO_read(privateKeyBIO, keyBytes, len);
  157. _privateKeyData = [NSData dataWithBytes:keyBytes length:len];
  158. self.generatedPrivateKey = [[NSString alloc] initWithData:_privateKeyData encoding:NSUTF8StringEncoding];
  159. NSLog(@"[LOG] \n%@", self.generatedPrivateKey);
  160. if(keyBytes)
  161. free(keyBytes);
  162. #ifdef DEBUG
  163. // Save to disk [DEBUG MODE]
  164. [self saveToDiskPEMWithCert:x509 key:pkey directory:directory];
  165. #endif
  166. return YES;
  167. }
  168. - (NSString *)extractPublicKeyFromCertificate:(NSString *)pemCertificate
  169. {
  170. const char *ptrCert = [pemCertificate cStringUsingEncoding:NSUTF8StringEncoding];
  171. BIO *certBio = BIO_new(BIO_s_mem());
  172. BIO_write(certBio, ptrCert,(unsigned int)strlen(ptrCert));
  173. X509 *certX509 = PEM_read_bio_X509(certBio, NULL, NULL, NULL);
  174. if (!certX509) {
  175. fprintf(stderr, "unable to parse certificate in memory\n");
  176. return nil;
  177. }
  178. EVP_PKEY *pkey;
  179. pkey = X509_get_pubkey(certX509);
  180. NSString *publicKey = [self pubKeyToString:pkey];
  181. EVP_PKEY_free(pkey);
  182. BIO_free(certBio);
  183. X509_free(certX509);
  184. NSLog(@"[LOG] \n%@", publicKey);
  185. return publicKey;
  186. }
  187. - (BOOL)saveToDiskPEMWithCert:(X509 *)x509 key:(EVP_PKEY *)pkey directory:(NSString *)directory
  188. {
  189. FILE *f;
  190. // Certificate
  191. NSString *certificatePath = [NSString stringWithFormat:@"%@/%@", directory, fileNameCertificate];
  192. f = fopen([certificatePath fileSystemRepresentation], "wb");
  193. if (PEM_write_X509(f, x509) < 0) {
  194. // Error writing to disk.
  195. fclose(f);
  196. return NO;
  197. }
  198. NSLog(@"[LOG] Saved cert to %@", certificatePath);
  199. fclose(f);
  200. // PublicKey
  201. NSString *publicKeyPath = [NSString stringWithFormat:@"%@/%@", directory, fileNamePubliceKey];
  202. f = fopen([publicKeyPath fileSystemRepresentation], "wb");
  203. if (PEM_write_PUBKEY(f, pkey) < 0) {
  204. // Error
  205. fclose(f);
  206. return NO;
  207. }
  208. NSLog(@"[LOG] Saved publicKey to %@", publicKeyPath);
  209. fclose(f);
  210. // Here you write the private key (pkey) to disk. OpenSSL will encrypt the
  211. // file using the password and cipher you provide.
  212. //if (PEM_write_PrivateKey(f, pkey, EVP_des_ede3_cbc(), (unsigned char *)[password UTF8String], (int)password.length, NULL, NULL) < 0) {
  213. // PrivateKey
  214. NSString *privatekeyPath = [NSString stringWithFormat:@"%@/%@", directory, fileNamePrivateKey];
  215. f = fopen([privatekeyPath fileSystemRepresentation], "wb");
  216. if (PEM_write_PrivateKey(f, pkey, NULL, NULL, 0, NULL, NULL) < 0) {
  217. // Error
  218. fclose(f);
  219. return NO;
  220. }
  221. NSLog(@"[LOG] Saved privatekey to %@", privatekeyPath);
  222. fclose(f);
  223. // CSR Request sha256
  224. NSString *csrPath = [NSString stringWithFormat:@"%@/%@", directory, fileNameCSR];
  225. f = fopen([csrPath fileSystemRepresentation], "wb");
  226. X509_REQ *certreq = X509_to_X509_REQ(x509, pkey, EVP_sha256());
  227. if (PEM_write_X509_REQ(f, certreq) < 0) {
  228. // Error
  229. fclose(f);
  230. return NO;
  231. }
  232. NSLog(@"[LOG] Saved csr to %@", csrPath);
  233. fclose(f);
  234. return YES;
  235. }
  236. - (BOOL)saveP12WithCert:(X509 *)x509 key:(EVP_PKEY *)pkey directory:(NSString *)directory finished:(void (^)(NSError *))finished
  237. {
  238. //PKCS12 * p12 = PKCS12_create([password UTF8String], NULL, pkey, x509, NULL, 0, 0, PKCS12_DEFAULT_ITER, 1, NID_key_usage);
  239. PKCS12 *p12 = PKCS12_create(NULL, NULL, pkey, x509, NULL, 0, 0, PKCS12_DEFAULT_ITER, 1, NID_key_usage);
  240. NSString *path = [NSString stringWithFormat:@"%@/certificate.p12", directory];
  241. FILE *f = fopen([path fileSystemRepresentation], "wb");
  242. if (i2d_PKCS12_fp(f, p12) != 1) {
  243. fclose(f);
  244. return NO;
  245. }
  246. NSLog(@"[LOG] Saved p12 to %@", path);
  247. fclose(f);
  248. return YES;
  249. }
  250. #
  251. #pragma mark - Create CSR & Encrypt/Decrypt Private Key
  252. #
  253. - (NSString *)createCSR:(NSString *)userId directory:(NSString *)directory
  254. {
  255. // Create Certificate, if do not exists
  256. if (!_csrData) {
  257. if (![self generateCertificateX509WithUserId:userId directory:directory])
  258. return nil;
  259. }
  260. NSString *csr = [[NSString alloc] initWithData:_csrData encoding:NSUTF8StringEncoding];
  261. return csr;
  262. }
  263. - (NSString *)encryptPrivateKey:(NSString *)userId directory:(NSString *)directory passphrase:(NSString *)passphrase privateKey:(NSString **)privateKey iterationCount:(unsigned int)iterationCount
  264. {
  265. NSMutableData *cipher = [NSMutableData new];
  266. if (!_privateKeyData) {
  267. if (![self generateCertificateX509WithUserId:userId directory:directory])
  268. return nil;
  269. }
  270. NSMutableData *key = [NSMutableData dataWithLength:PBKDF2_KEY_LENGTH/8];
  271. NSData *salt = [self generateSalt:AES_SALT_LENGTH];
  272. // Remove all whitespaces from passphrase
  273. passphrase = [passphrase stringByReplacingOccurrencesOfString:@" " withString:@""];
  274. CCKeyDerivationPBKDF(kCCPBKDF2, passphrase.UTF8String, passphrase.length, salt.bytes, salt.length, kCCPRFHmacAlgSHA1, iterationCount, key.mutableBytes, key.length);
  275. NSData *initializationVector = [self generateIV:AES_IVEC_LENGTH];
  276. NSData *authenticationTag = [NSData new];
  277. NSString *pkEncoded = [_privateKeyData base64EncodedStringWithOptions:0];
  278. NSData *pkEncodedData = [pkEncoded dataUsingEncoding:NSUTF8StringEncoding];
  279. BOOL result = [self encryptData:pkEncodedData cipher:&cipher key:key keyLen:AES_KEY_256_LENGTH initializationVector:initializationVector authenticationTag:&authenticationTag];
  280. if (result && cipher) {
  281. NSString *cipherString = [cipher base64EncodedStringWithOptions:0];
  282. NSString *initializationVectorString = [initializationVector base64EncodedStringWithOptions:0];
  283. NSString *saltString = [salt base64EncodedStringWithOptions:0];
  284. NSString *encryptPrivateKey = [NSString stringWithFormat:@"%@%@%@%@%@", cipherString, IV_DELIMITER_ENCODED, initializationVectorString, IV_DELIMITER_ENCODED, saltString];
  285. *privateKey = [[NSString alloc] initWithData:_privateKeyData encoding:NSUTF8StringEncoding];
  286. return encryptPrivateKey;
  287. } else {
  288. return nil;
  289. }
  290. }
  291. - (NSData *)decryptPrivateKey:(NSString *)privateKey passphrase:(NSString *)passphrase publicKey:(NSString *)publicKey iterationCount:(unsigned int)iterationCount
  292. {
  293. NSMutableData *plain = [NSMutableData new];
  294. // Key
  295. NSMutableData *key = [NSMutableData dataWithLength:PBKDF2_KEY_LENGTH/8];
  296. // Split
  297. NSArray *cipherArray = [privateKey componentsSeparatedByString:IV_DELIMITER_ENCODED];
  298. if (cipherArray.count != 3) {
  299. cipherArray = [privateKey componentsSeparatedByString:IV_DELIMITER_ENCODED_OLD];
  300. if (cipherArray.count != 3) {
  301. return nil;
  302. }
  303. }
  304. NSData *cipher = [[NSData alloc] initWithBase64EncodedString:cipherArray[0] options:0];
  305. NSString *authenticationTagString = [privateKey substringWithRange:NSMakeRange([(NSString *)cipherArray[0] length] - AES_GCM_TAG_LENGTH, AES_GCM_TAG_LENGTH)];
  306. NSData *authenticationTag = [[NSData alloc] initWithBase64EncodedString:authenticationTagString options:0];
  307. NSData *initializationVector = [[NSData alloc] initWithBase64EncodedString:cipherArray[1] options:0];
  308. NSData *salt = [[NSData alloc] initWithBase64EncodedString:cipherArray[2] options:0];
  309. // Remove Authentication Tag
  310. cipher = [cipher subdataWithRange:NSMakeRange(0, cipher.length - AES_GCM_TAG_LENGTH)];
  311. // Remove all whitespaces from passphrase
  312. passphrase = [passphrase stringByReplacingOccurrencesOfString:@" " withString:@""];
  313. CCKeyDerivationPBKDF(kCCPBKDF2, passphrase.UTF8String, passphrase.length, salt.bytes, salt.length, kCCPRFHmacAlgSHA1, iterationCount, key.mutableBytes, key.length);
  314. BOOL result = [self decryptData:cipher plain:&plain key:key keyLen:AES_KEY_256_LENGTH initializationVector:initializationVector authenticationTag:authenticationTag];
  315. if (result && plain) {
  316. return plain;
  317. }
  318. return nil;
  319. }
  320. #
  321. #pragma mark - Encrypt / Decrypt file material
  322. #
  323. - (NSString *)encryptPayloadFile:(NSData *)encrypted key:(NSString *)key
  324. {
  325. NSMutableData *cipher;
  326. NSData *authenticationTag = [NSData new];
  327. // Key
  328. NSData *keyData = [[NSData alloc] initWithBase64EncodedString:key options:0];
  329. // Initialization Vector
  330. NSData *initializationVector = [self generateIV:AES_IVEC_LENGTH];
  331. BOOL result = [self encryptData:encrypted cipher:&cipher key:keyData keyLen:AES_KEY_128_LENGTH initializationVector:initializationVector authenticationTag:&authenticationTag];
  332. if (cipher != nil && result) {
  333. NSString *cipherString = [cipher base64EncodedStringWithOptions:0];
  334. NSString *initializationVectorString = [initializationVector base64EncodedStringWithOptions:0];
  335. NSString *payload = [NSString stringWithFormat:@"%@%@%@", cipherString, IV_DELIMITER_ENCODED, initializationVectorString];
  336. return payload;
  337. }
  338. return nil;
  339. }
  340. - (NSString *)encryptPayloadFile:(NSData *)encrypted key:(NSString *)key initializationVector:(NSString **)initializationVector authenticationTag:(NSString **)authenticationTag
  341. {
  342. NSMutableData *cipher;
  343. NSData *authenticationTagData = [NSData new];
  344. // Key
  345. NSData *keyData = [[NSData alloc] initWithBase64EncodedString:key options:0];
  346. // Initialization Vector
  347. NSData *initializationVectorData = [self generateIV:AES_IVEC_LENGTH];
  348. BOOL result = [self encryptData:encrypted cipher:&cipher key:keyData keyLen:AES_KEY_128_LENGTH initializationVector:initializationVectorData authenticationTag:&authenticationTagData];
  349. if (cipher != nil && result) {
  350. *initializationVector = [initializationVectorData base64EncodedStringWithOptions:0];
  351. *authenticationTag = [authenticationTagData base64EncodedStringWithOptions:0];
  352. NSString *payload = [cipher base64EncodedStringWithOptions:0];
  353. return payload;
  354. }
  355. return nil;
  356. }
  357. - (NSData *)decryptPayloadFile:(NSString *)encrypted key:(NSString *)key
  358. {
  359. NSMutableData *plain;
  360. NSRange range = [encrypted rangeOfString:IV_DELIMITER_ENCODED];
  361. if (range.location == NSNotFound) {
  362. range = [encrypted rangeOfString:IV_DELIMITER_ENCODED_OLD];
  363. if (range.location == NSNotFound) {
  364. return nil;
  365. }
  366. }
  367. // Cipher
  368. NSString *cipher = [encrypted substringToIndex:(range.location)];
  369. NSData *cipherData = [[NSData alloc] initWithBase64EncodedString:cipher options:0];
  370. // Key
  371. NSData *keyData = [[NSData alloc] initWithBase64EncodedString:key options:0];
  372. // Initialization Vector
  373. NSString *initializationVector = [encrypted substringWithRange:NSMakeRange(range.location + range.length, encrypted.length - (range.location + range.length))];
  374. NSData *initializationVectorData = [[NSData alloc] initWithBase64EncodedString:initializationVector options:0];
  375. // Authentication Tag
  376. NSString *authenticationTag = [cipher substringWithRange:NSMakeRange(cipher.length - AES_GCM_TAG_LENGTH, AES_GCM_TAG_LENGTH)];
  377. NSData *authenticationTagData = [[NSData alloc] initWithBase64EncodedString:authenticationTag options:0];
  378. // Remove Authentication Tag
  379. cipherData = [cipherData subdataWithRange:NSMakeRange(0, cipherData.length - AES_GCM_TAG_LENGTH)];
  380. BOOL result = [self decryptData:cipherData plain:&plain key:keyData keyLen:AES_KEY_128_LENGTH initializationVector:initializationVectorData authenticationTag:authenticationTagData];
  381. if (plain != nil && result) {
  382. return plain;
  383. }
  384. return nil;
  385. }
  386. - (NSData *)decryptPayloadFile:(NSString *)encrypted key:(NSString *)key initializationVector:(NSString *)initializationVector authenticationTag:(NSString *)authenticationTag
  387. {
  388. NSMutableData *plain;
  389. // Remove initializationVector Tag if exists [ANDROID]
  390. NSString *android = [@"|" stringByAppendingString: initializationVector];
  391. encrypted = [encrypted stringByReplacingOccurrencesOfString:android withString:@""];
  392. NSData *cipher = [[NSData alloc] initWithBase64EncodedString:encrypted options:0];
  393. NSData *keyData = [[NSData alloc] initWithBase64EncodedString:key options:0];
  394. NSData *initializationVectorData = [[NSData alloc] initWithBase64EncodedString:initializationVector options:0];
  395. NSData *authenticationTagData = [[NSData alloc] initWithBase64EncodedString:authenticationTag options:0];
  396. // Remove Authentication Tag
  397. cipher = [cipher subdataWithRange:NSMakeRange(0, cipher.length - AES_GCM_TAG_LENGTH)];
  398. BOOL result = [self decryptData:cipher plain:&plain key:keyData keyLen:AES_KEY_128_LENGTH initializationVector:initializationVectorData authenticationTag:authenticationTagData];
  399. if (plain != nil && result && plain.length > 0) {
  400. return plain;
  401. }
  402. return nil;
  403. }
  404. #
  405. #pragma mark - Encrypt / Decrypt file
  406. #
  407. - (BOOL)encryptFile:(NSString *)fileName fileNameIdentifier:(NSString *)fileNameIdentifier directory:(NSString *)directory key:(NSString **)key initializationVector:(NSString **)initializationVector authenticationTag:(NSString **)authenticationTag
  408. {
  409. NSData *authenticationTagData;
  410. NSData *keyData = [self generateKey:AES_KEY_128_LENGTH];
  411. NSData *initializationVectorData = [self generateIV:AES_IVEC_LENGTH];
  412. BOOL result = [self encryptFile:[NSString stringWithFormat:@"%@/%@", directory, fileName] fileNameCipher:[NSString stringWithFormat:@"%@/%@", directory, fileNameIdentifier] key:keyData keyLen:AES_KEY_128_LENGTH initializationVector:initializationVectorData authenticationTag:&authenticationTagData];
  413. if (result) {
  414. *key = [keyData base64EncodedStringWithOptions:0];
  415. *initializationVector = [initializationVectorData base64EncodedStringWithOptions:0];
  416. *authenticationTag = [authenticationTagData base64EncodedStringWithOptions:0];
  417. if (key == nil || initializationVector == nil || authenticationTag == nil) {
  418. return false;
  419. } else {
  420. return true;
  421. }
  422. }
  423. return false;
  424. }
  425. - (BOOL)decryptFile:(NSString *)fileName fileNameView:(NSString *)fileNameView ocId:(NSString *)ocId key:(NSString *)key initializationVector:(NSString *)initializationVector authenticationTag:(NSString *)authenticationTag
  426. {
  427. NSData *keyData = [[NSData alloc] initWithBase64EncodedString:key options:0];
  428. NSData *initializationVectorData = [[NSData alloc] initWithBase64EncodedString:initializationVector options:0];
  429. NSData *authenticationTagData = [[NSData alloc] initWithBase64EncodedString:authenticationTag options:0];
  430. return [self decryptFile:[CCUtility getDirectoryProviderStorageOcId:ocId fileNameView:fileName] fileNamePlain:[CCUtility getDirectoryProviderStorageOcId:ocId fileNameView:fileNameView] key:keyData keyLen:AES_KEY_128_LENGTH initializationVector:initializationVectorData authenticationTag:authenticationTagData];
  431. }
  432. // -----------------------------------------------------------------------------------------------------------------------------------------------------------------------
  433. // -----------------------------------------------------------------------------------------------------------------------------------------------------------------------
  434. #
  435. #pragma mark - OPENSSL ENCRYPT/DECRYPT
  436. #
  437. #
  438. #pragma mark - Encrypt/Decrypt asymmetric
  439. #
  440. - (NSData *)encryptAsymmetricData:(NSData *)plainData certificate:(NSString *)certificate
  441. {
  442. EVP_PKEY *key = NULL;
  443. int status = 0;
  444. unsigned char *pKey = (unsigned char *)[certificate UTF8String];
  445. // Extract real publicKey
  446. BIO *bio = BIO_new_mem_buf(pKey, -1);
  447. if (!bio)
  448. return nil;
  449. X509 *x509 = PEM_read_bio_X509(bio, NULL, 0, NULL);
  450. if (!x509)
  451. return nil;
  452. key = X509_get_pubkey(x509);
  453. if (!key)
  454. return nil;
  455. EVP_PKEY_CTX *ctx = EVP_PKEY_CTX_new(key, NULL);
  456. if (!ctx)
  457. return nil;
  458. status = EVP_PKEY_encrypt_init(ctx);
  459. if (status <= 0)
  460. return nil;
  461. status = EVP_PKEY_CTX_set_rsa_padding(ctx, RSA_PKCS1_OAEP_PADDING);
  462. if (status <= 0)
  463. return nil;
  464. status = EVP_PKEY_CTX_set_rsa_oaep_md(ctx, EVP_sha256());
  465. if (status <= 0)
  466. return nil;
  467. status = EVP_PKEY_CTX_set_rsa_mgf1_md(ctx, EVP_sha256());
  468. if (status <= 0)
  469. return nil;
  470. unsigned long outLen = 0;
  471. status = EVP_PKEY_encrypt(ctx, NULL, &outLen, [plainData bytes], (int)[plainData length]);
  472. if (status <= 0 || outLen == 0)
  473. return nil;
  474. unsigned char *out = (unsigned char *) malloc(outLen);
  475. status = EVP_PKEY_encrypt(ctx, out, &outLen, [plainData bytes], (int)[plainData length]);
  476. if (status <= 0)
  477. return nil;
  478. NSData *outData = [[NSData alloc] initWithBytes:out length:outLen];
  479. if (out)
  480. free(out);
  481. return outData;
  482. }
  483. - (NSData *)encryptAsymmetricData:(NSData *)plainData privateKey:(NSString *)privateKey
  484. {
  485. EVP_PKEY *key = NULL;
  486. int status = 0;
  487. unsigned char *pKey = (unsigned char *)[privateKey UTF8String];
  488. BIO *bio = BIO_new_mem_buf(pKey, -1);
  489. if (!bio)
  490. return nil;
  491. key = PEM_read_bio_PrivateKey(bio, NULL, NULL, NULL);
  492. if (!key)
  493. return nil;
  494. EVP_PKEY_CTX *ctx = EVP_PKEY_CTX_new(key, NULL);
  495. if (!ctx)
  496. return nil;
  497. status = EVP_PKEY_encrypt_init(ctx);
  498. if (status <= 0)
  499. return nil;
  500. status = EVP_PKEY_CTX_set_rsa_padding(ctx, RSA_PKCS1_OAEP_PADDING);
  501. if (status <= 0)
  502. return nil;
  503. status = EVP_PKEY_CTX_set_rsa_oaep_md(ctx, EVP_sha256());
  504. if (status <= 0)
  505. return nil;
  506. status = EVP_PKEY_CTX_set_rsa_mgf1_md(ctx, EVP_sha256());
  507. if (status <= 0)
  508. return nil;
  509. unsigned long outLen = 0;
  510. status = EVP_PKEY_encrypt(ctx, NULL, &outLen, [plainData bytes], (int)[plainData length]);
  511. if (status <= 0 || outLen == 0)
  512. return nil;
  513. unsigned char *out = (unsigned char *) malloc(outLen);
  514. status = EVP_PKEY_encrypt(ctx, out, &outLen, [plainData bytes], (int)[plainData length]);
  515. if (status <= 0)
  516. return nil;
  517. NSData *outData = [[NSData alloc] initWithBytes:out length:outLen];
  518. if (out)
  519. free(out);
  520. return outData;
  521. }
  522. - (NSData *)decryptAsymmetricData:(NSData *)cipherData privateKey:(NSString *)privateKey
  523. {
  524. unsigned char *pKey = (unsigned char *)[privateKey UTF8String];
  525. int status = 0;
  526. BIO *bio = BIO_new_mem_buf(pKey, -1);
  527. if (!bio)
  528. return nil;
  529. EVP_PKEY *key = PEM_read_bio_PrivateKey(bio, NULL, NULL, NULL);
  530. if (!key)
  531. return nil;
  532. EVP_PKEY_CTX *ctx = EVP_PKEY_CTX_new(key, NULL);
  533. if (!ctx)
  534. return nil;
  535. status = EVP_PKEY_decrypt_init(ctx);
  536. if (status <= 0)
  537. return nil;
  538. status = EVP_PKEY_CTX_set_rsa_padding(ctx, RSA_PKCS1_OAEP_PADDING);
  539. if (status <= 0)
  540. return nil;
  541. status = EVP_PKEY_CTX_set_rsa_oaep_md(ctx, EVP_sha256());
  542. if (status <= 0)
  543. return nil;
  544. status = EVP_PKEY_CTX_set_rsa_mgf1_md(ctx, EVP_sha256());
  545. if (status <= 0)
  546. return nil;
  547. unsigned long outLen = 0;
  548. status = EVP_PKEY_decrypt(ctx, NULL, &outLen, [cipherData bytes], (int)[cipherData length]);
  549. if (status <= 0 || outLen == 0)
  550. return nil;
  551. unsigned char *out = (unsigned char *) malloc(outLen);
  552. status = EVP_PKEY_decrypt(ctx, out, &outLen, [cipherData bytes], (int)[cipherData length]);
  553. if (status <= 0)
  554. return nil;
  555. NSData *outData = [[NSData alloc] initWithBytes:out length:outLen];
  556. if (out)
  557. free(out);
  558. return outData;
  559. }
  560. #
  561. #pragma mark - AES/GCM/NoPadding
  562. #
  563. // Encryption data using GCM mode
  564. - (BOOL)encryptData:(NSData *)plain cipher:(NSMutableData **)cipher key:(NSData *)key keyLen:(int)keyLen initializationVector:(NSData *)initializationVector authenticationTag:(NSData **)authenticationTag
  565. {
  566. int status = 0;
  567. int len = 0;
  568. // set up key
  569. len = keyLen;
  570. unsigned char cKey[len];
  571. bzero(cKey, sizeof(cKey));
  572. [key getBytes:cKey length:len];
  573. // set up ivec
  574. len = AES_IVEC_LENGTH;
  575. unsigned char cIV[len];
  576. bzero(cIV, sizeof(cIV));
  577. [initializationVector getBytes:cIV length:len];
  578. // set up tag
  579. len = AES_GCM_TAG_LENGTH;
  580. unsigned char cTag[len];
  581. bzero(cTag, sizeof(cTag));
  582. // Create and initialise the context
  583. EVP_CIPHER_CTX *ctx = EVP_CIPHER_CTX_new();
  584. if (!ctx)
  585. return NO;
  586. // Initialise the encryption operation
  587. if (keyLen == AES_KEY_128_LENGTH)
  588. status = EVP_EncryptInit_ex(ctx, EVP_aes_128_gcm(), NULL, NULL, NULL);
  589. else if (keyLen == AES_KEY_256_LENGTH)
  590. status = EVP_EncryptInit_ex(ctx, EVP_aes_256_gcm(), NULL, NULL, NULL);
  591. if (status <= 0) {
  592. EVP_CIPHER_CTX_free(ctx);
  593. return NO;
  594. }
  595. // Set IV length. Not necessary if this is 12 bytes (96 bits)
  596. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, (int)sizeof(cIV), NULL);
  597. if (status <= 0) {
  598. EVP_CIPHER_CTX_free(ctx);
  599. return NO;
  600. }
  601. // Initialise key and IV
  602. status = EVP_EncryptInit_ex (ctx, NULL, NULL, cKey, cIV);
  603. if (status <= 0) {
  604. EVP_CIPHER_CTX_free(ctx);
  605. return NO;
  606. }
  607. // Provide the message to be encrypted, and obtain the encrypted output
  608. *cipher = [NSMutableData dataWithLength:[plain length]];
  609. unsigned char * cCipher = [*cipher mutableBytes];
  610. int cCipherLen = 0;
  611. status = EVP_EncryptUpdate(ctx, cCipher, &cCipherLen, [plain bytes], (int)[plain length]);
  612. if (status <= 0) {
  613. EVP_CIPHER_CTX_free(ctx);
  614. return NO;
  615. }
  616. // Finalise the encryption
  617. status = EVP_EncryptFinal_ex(ctx, cCipher, &cCipherLen);
  618. if (status <= 0) {
  619. EVP_CIPHER_CTX_free(ctx);
  620. return NO;
  621. }
  622. // Get the tag
  623. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_GET_TAG, (int)sizeof(cTag), cTag);
  624. *authenticationTag = [NSData dataWithBytes:cTag length:sizeof(cTag)];
  625. // Append TAG
  626. [*cipher appendData:*authenticationTag];
  627. EVP_CIPHER_CTX_free(ctx);
  628. return status; // OpenSSL uses 1 for success
  629. }
  630. // Encryption file using GCM mode
  631. - (BOOL)encryptFile:(NSString *)fileName fileNameCipher:(NSString *)fileNameCipher key:(NSData *)key keyLen:(int)keyLen initializationVector:(NSData *)initializationVector authenticationTag:(NSData **)authenticationTag
  632. {
  633. int status = 0;
  634. int len = 0;
  635. // set up key
  636. len = keyLen;
  637. unsigned char cKey[len];
  638. bzero(cKey, sizeof(cKey));
  639. [key getBytes:cKey length:len];
  640. // set up ivec
  641. len = AES_IVEC_LENGTH;
  642. unsigned char cIV[len];
  643. bzero(cIV, sizeof(cIV));
  644. [initializationVector getBytes:cIV length:len];
  645. // set up tag
  646. len = AES_GCM_TAG_LENGTH;
  647. unsigned char cTag[len];
  648. bzero(cTag, sizeof(cTag));
  649. // Create and initialise the context
  650. EVP_CIPHER_CTX *ctx = EVP_CIPHER_CTX_new();
  651. if (!ctx) {
  652. return NO;
  653. }
  654. // Initialise the encryption operation
  655. if (keyLen == AES_KEY_128_LENGTH)
  656. status = EVP_EncryptInit_ex(ctx, EVP_aes_128_gcm(), NULL, NULL, NULL);
  657. else if (keyLen == AES_KEY_256_LENGTH)
  658. status = EVP_EncryptInit_ex(ctx, EVP_aes_256_gcm(), NULL, NULL, NULL);
  659. if (status <= 0) {
  660. EVP_CIPHER_CTX_free(ctx);
  661. return NO;
  662. }
  663. // Set IV length. Not necessary if this is 12 bytes (96 bits)
  664. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, (int)sizeof(cIV), NULL);
  665. if (status <= 0) {
  666. EVP_CIPHER_CTX_free(ctx);
  667. return NO;
  668. }
  669. // Initialise key and IV
  670. status = EVP_EncryptInit_ex (ctx, NULL, NULL, cKey, cIV);
  671. if (status <= 0) {
  672. EVP_CIPHER_CTX_free(ctx);
  673. return NO;
  674. }
  675. NSInputStream *inStream = [NSInputStream inputStreamWithFileAtPath:fileName];
  676. [inStream open];
  677. NSOutputStream *outStream = [NSOutputStream outputStreamToFileAtPath:fileNameCipher append:false];
  678. [outStream open];
  679. Byte buffer[streamBuffer];
  680. NSInteger totalNumberOfBytesWritten = 0;
  681. int cCipherLen = 0;
  682. unsigned char *cCipher;
  683. while ([inStream hasBytesAvailable]) {
  684. @autoreleasepool {
  685. NSInteger bytesRead = [inStream read:buffer maxLength:streamBuffer];
  686. if (bytesRead > 0) {
  687. cCipher = [[NSMutableData dataWithLength:bytesRead] mutableBytes];
  688. status = EVP_EncryptUpdate(ctx, cCipher, &cCipherLen, [[NSData dataWithBytes:buffer length:bytesRead] bytes], (int)bytesRead);
  689. if (status <= 0) {
  690. [inStream close];
  691. [outStream close];
  692. EVP_CIPHER_CTX_free(ctx);
  693. return NO;
  694. }
  695. if ([outStream hasSpaceAvailable]) {
  696. totalNumberOfBytesWritten = [outStream write:cCipher maxLength:cCipherLen];
  697. if (totalNumberOfBytesWritten != cCipherLen) {
  698. [inStream close];
  699. [outStream close];
  700. EVP_CIPHER_CTX_free(ctx);
  701. return NO;
  702. }
  703. }
  704. }
  705. }
  706. }
  707. [inStream close];
  708. status = EVP_EncryptFinal_ex(ctx, cCipher, &cCipherLen);
  709. if (status <= 0) {
  710. [outStream close];
  711. EVP_CIPHER_CTX_free(ctx);
  712. return NO;
  713. }
  714. // Get the tag
  715. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_GET_TAG, (int)sizeof(cTag), cTag);
  716. if (status <= 0) {
  717. [outStream close];
  718. EVP_CIPHER_CTX_free(ctx);
  719. return NO;
  720. }
  721. *authenticationTag = [NSData dataWithBytes:cTag length:sizeof(cTag)];
  722. // Append TAG
  723. if ([outStream hasSpaceAvailable]) {
  724. totalNumberOfBytesWritten = [outStream write:cTag maxLength:sizeof(cTag)];
  725. if (totalNumberOfBytesWritten != sizeof(cTag)) {
  726. status = NO;
  727. }
  728. } else {
  729. status = NO;
  730. }
  731. [outStream close];
  732. EVP_CIPHER_CTX_free(ctx);
  733. return status; // OpenSSL uses 1 for success
  734. }
  735. // Decryption data using GCM mode
  736. - (BOOL)decryptData:(NSData *)cipher plain:(NSMutableData **)plain key:(NSData *)key keyLen:(int)keyLen initializationVector:(NSData *)initializationVector authenticationTag:(NSData *)authenticationTag
  737. {
  738. int status = 0;
  739. int len = 0;
  740. // set up key
  741. len = keyLen;
  742. unsigned char cKey[len];
  743. bzero(cKey, sizeof(cKey));
  744. [key getBytes:cKey length:len];
  745. // set up ivec
  746. len = (int)[initializationVector length];
  747. unsigned char cIV[len];
  748. bzero(cIV, sizeof(cIV));
  749. [initializationVector getBytes:cIV length:len];
  750. // set up tag
  751. len = (int)[authenticationTag length];;
  752. unsigned char cTag[len];
  753. bzero(cTag, sizeof(cTag));
  754. [authenticationTag getBytes:cTag length:len];
  755. // Create and initialise the context
  756. EVP_CIPHER_CTX *ctx = EVP_CIPHER_CTX_new();
  757. if (!ctx)
  758. return NO;
  759. // Initialise the decryption operation
  760. if (keyLen == AES_KEY_128_LENGTH)
  761. status = EVP_DecryptInit_ex(ctx, EVP_aes_128_gcm(), NULL, NULL, NULL);
  762. else if (keyLen == AES_KEY_256_LENGTH)
  763. status = EVP_DecryptInit_ex(ctx, EVP_aes_256_gcm(), NULL, NULL, NULL);
  764. if (status <= 0) {
  765. EVP_CIPHER_CTX_free(ctx);
  766. return NO;
  767. }
  768. // Set IV length. Not necessary if this is 12 bytes (96 bits)
  769. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, (int)sizeof(cIV), NULL);
  770. if (status <= 0) {
  771. EVP_CIPHER_CTX_free(ctx);
  772. return NO;
  773. }
  774. // Initialise key and IV
  775. status = EVP_DecryptInit_ex(ctx, NULL, NULL, cKey, cIV);
  776. if (status <= 0) {
  777. EVP_CIPHER_CTX_free(ctx);
  778. return NO;
  779. }
  780. // Provide the message to be decrypted, and obtain the plaintext output
  781. *plain = [NSMutableData dataWithLength:([cipher length])];
  782. int cPlainLen = 0;
  783. unsigned char * cPlain = [*plain mutableBytes];
  784. status = EVP_DecryptUpdate(ctx, cPlain, &cPlainLen, [cipher bytes], (int)([cipher length]));
  785. if (status <= 0) {
  786. EVP_CIPHER_CTX_free(ctx);
  787. return NO;
  788. }
  789. // Tag is the last 16 bytes
  790. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_TAG, (int)sizeof(cTag), cTag);
  791. if (status <= 0) {
  792. EVP_CIPHER_CTX_free(ctx);
  793. return NO;
  794. }
  795. // Finalise the encryption
  796. EVP_DecryptFinal_ex(ctx,NULL, &cPlainLen);
  797. // Free
  798. EVP_CIPHER_CTX_free(ctx);
  799. return status; // OpenSSL uses 1 for success
  800. }
  801. // Decryption file using GCM mode
  802. - (BOOL)decryptFile:(NSString *)fileName fileNamePlain:(NSString *)fileNamePlain key:(NSData *)key keyLen:(int)keyLen initializationVector:(NSData *)initializationVector authenticationTag:(NSData *)authenticationTag
  803. {
  804. int status = 0;
  805. int len = 0;
  806. // set up key
  807. len = keyLen;
  808. unsigned char cKey[len];
  809. bzero(cKey, sizeof(cKey));
  810. [key getBytes:cKey length:len];
  811. // set up ivec
  812. len = (int)[initializationVector length];
  813. unsigned char cIV[len];
  814. bzero(cIV, sizeof(cIV));
  815. [initializationVector getBytes:cIV length:len];
  816. // set up tag
  817. len = (int)[authenticationTag length];;
  818. unsigned char cTag[len];
  819. bzero(cTag, sizeof(cTag));
  820. [authenticationTag getBytes:cTag length:len];
  821. // Create and initialise the context
  822. EVP_CIPHER_CTX *ctx = EVP_CIPHER_CTX_new();
  823. if (!ctx)
  824. return NO;
  825. // Initialise the decryption operation
  826. if (keyLen == AES_KEY_128_LENGTH)
  827. status = EVP_DecryptInit_ex(ctx, EVP_aes_128_gcm(), NULL, NULL, NULL);
  828. else if (keyLen == AES_KEY_256_LENGTH)
  829. status = EVP_DecryptInit_ex(ctx, EVP_aes_256_gcm(), NULL, NULL, NULL);
  830. if (status <= 0) {
  831. EVP_CIPHER_CTX_free(ctx);
  832. return NO;
  833. }
  834. // Set IV length. Not necessary if this is 12 bytes (96 bits)
  835. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, (int)sizeof(cIV), NULL);
  836. if (status <= 0) {
  837. EVP_CIPHER_CTX_free(ctx);
  838. return NO;
  839. }
  840. // Initialise key and IV
  841. status = EVP_DecryptInit_ex(ctx, NULL, NULL, cKey, cIV);
  842. if (status <= 0) {
  843. EVP_CIPHER_CTX_free(ctx);
  844. return NO;
  845. }
  846. NSInputStream *inStream = [NSInputStream inputStreamWithFileAtPath:fileName];
  847. [inStream open];
  848. NSOutputStream *outStream = [NSOutputStream outputStreamToFileAtPath:fileNamePlain append:false];
  849. [outStream open];
  850. Byte buffer[streamBuffer];
  851. NSInteger totalNumberOfBytesWritten = 0;
  852. int cPlainLen = 0;
  853. unsigned char *cPlain;
  854. while ([inStream hasBytesAvailable]) {
  855. @autoreleasepool {
  856. NSInteger bytesRead = [inStream read:buffer maxLength:streamBuffer];
  857. if (bytesRead > 0) {
  858. cPlain = [[NSMutableData dataWithLength:bytesRead] mutableBytes];
  859. status = EVP_DecryptUpdate(ctx, cPlain, &cPlainLen, [[NSData dataWithBytes:buffer length:bytesRead] bytes], (int)bytesRead);
  860. if (status <= 0) {
  861. [inStream close];
  862. [outStream close];
  863. EVP_CIPHER_CTX_free(ctx);
  864. return NO;
  865. }
  866. if ([outStream hasSpaceAvailable]) {
  867. totalNumberOfBytesWritten = [outStream write:cPlain maxLength:cPlainLen];
  868. if (totalNumberOfBytesWritten != cPlainLen) {
  869. [inStream close];
  870. [outStream close];
  871. EVP_CIPHER_CTX_free(ctx);
  872. return NO;
  873. }
  874. }
  875. }
  876. }
  877. }
  878. [inStream close];
  879. [outStream close];
  880. // Tag is the last 16 bytes
  881. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_TAG, (int)sizeof(cTag), cTag);
  882. if (status <= 0)
  883. return NO;
  884. // Finalise the encryption
  885. EVP_DecryptFinal_ex(ctx,NULL, &cPlainLen);
  886. // Free
  887. EVP_CIPHER_CTX_free(ctx);
  888. return status; // OpenSSL uses 1 for success
  889. }
  890. #
  891. #pragma mark - CMS
  892. #
  893. - (NSData *)generateSignatureCMS:(NSData *)data certificate:(NSString *)certificate privateKey:(NSString *)privateKey userId:(NSString *)userId
  894. {
  895. unsigned char *pKey = (unsigned char *)[privateKey UTF8String];
  896. unsigned char *certKey = (unsigned char *)[certificate UTF8String];
  897. BIO *printBIO = BIO_new_fp(stdout, BIO_NOCLOSE);
  898. BIO *certKeyBIO = BIO_new_mem_buf(certKey, -1);
  899. if (!certKeyBIO)
  900. return nil;
  901. X509 *x509 = PEM_read_bio_X509(certKeyBIO, NULL, 0, NULL);
  902. if (!x509)
  903. return nil;
  904. BIO *pkeyBIO = BIO_new_mem_buf(pKey, -1);
  905. EVP_PKEY *key = PEM_read_bio_PrivateKey(pkeyBIO, NULL, NULL, NULL);
  906. if (!key)
  907. return nil;
  908. BIO *dataBIO = BIO_new_mem_buf((void*)data.bytes, (int)data.length);
  909. CMS_ContentInfo *contentInfo = CMS_sign(x509, key, NULL, dataBIO, CMS_DETACHED);
  910. if (contentInfo == nil)
  911. return nil;
  912. CMS_ContentInfo_print_ctx(printBIO, contentInfo, 0, NULL);
  913. PEM_write_bio_CMS(printBIO, contentInfo);
  914. BIO *i2dCmsBioOut = BIO_new(BIO_s_mem());
  915. if (i2d_CMS_bio(i2dCmsBioOut, contentInfo) != 1)
  916. return nil;
  917. int len = BIO_pending(i2dCmsBioOut);
  918. char *keyBytes = malloc(len);
  919. BIO_read(i2dCmsBioOut, keyBytes, len);
  920. NSData *i2dCmsData = [NSData dataWithBytes:keyBytes length:len];
  921. BIO_free(printBIO);
  922. BIO_free(certKeyBIO);
  923. BIO_free(pkeyBIO);
  924. BIO_free(dataBIO);
  925. BIO_free(i2dCmsBioOut);
  926. return i2dCmsData;
  927. }
  928. /*
  929. - (BOOL)verifySignatureCMS:(NSData *)cmsContent data:(NSData *)data publicKey:(NSString *)publicKey userId:(NSString *)userId
  930. {
  931. BIO *dataBIO = BIO_new_mem_buf((void*)data.bytes, (int)data.length);
  932. BIO *printBIO = BIO_new_fp(stdout, BIO_NOCLOSE);
  933. BIO *cmsBIO = BIO_new_mem_buf(cmsContent.bytes, (int)cmsContent.length);
  934. CMS_ContentInfo *contentInfo = d2i_CMS_bio(cmsBIO, NULL);
  935. unsigned char *publicKeyUTF8 = (unsigned char *)[publicKey UTF8String];
  936. BIO *publicKeyBIO = BIO_new_mem_buf(publicKeyUTF8, -1);
  937. EVP_PKEY *pkey = PEM_read_bio_PUBKEY(publicKeyBIO, NULL, NULL, NULL);
  938. CMS_ContentInfo_print_ctx(printBIO, contentInfo, 0, NULL);
  939. BOOL verifyResult = CMS_verify(contentInfo, NULL, NULL, dataBIO, NULL, CMS_DETACHED | CMS_NO_SIGNER_CERT_VERIFY);
  940. if (verifyResult) {
  941. STACK_OF(X509) *signers = CMS_get0_signers(contentInfo);
  942. int numSigners = sk_X509_num(signers);
  943. for (int i = 0; i < numSigners; ++i) {
  944. X509 *signer = sk_X509_value(signers, i);
  945. int result = X509_verify(signer, pkey);
  946. if (result <= 0) {
  947. verifyResult = false;
  948. break;
  949. }
  950. int cnDataLength = X509_NAME_get_text_by_NID(X509_get_subject_name(signer), NID_commonName, 0, 0);
  951. cnDataLength += 1;
  952. NSMutableData* cnData = [NSMutableData dataWithLength:cnDataLength];
  953. X509_NAME_get_text_by_NID(X509_get_subject_name(signer), NID_commonName, [cnData mutableBytes], cnDataLength);
  954. NSString *cn = [[NSString alloc] initWithCString:[cnData mutableBytes] encoding:NSUTF8StringEncoding];
  955. if ([userId isEqualToString:cn]) {
  956. verifyResult = true;
  957. break;
  958. } else {
  959. verifyResult = false;
  960. }
  961. }
  962. if (signers) {
  963. sk_X509_free(signers);
  964. }
  965. signers = NULL;
  966. }
  967. BIO_free(dataBIO);
  968. BIO_free(printBIO);
  969. BIO_free(cmsBIO);
  970. BIO_free(publicKeyBIO);
  971. return verifyResult;
  972. }
  973. */
  974. - (BOOL)verifySignatureCMS:(NSData *)cmsContent data:(NSData *)data certificates:(NSArray*)certificates
  975. {
  976. BIO *dataBIO = BIO_new_mem_buf((void*)data.bytes, (int)data.length);
  977. BIO *printBIO = BIO_new_fp(stdout, BIO_NOCLOSE);
  978. BIO *cmsBIO = BIO_new_mem_buf(cmsContent.bytes, (int)cmsContent.length);
  979. CMS_ContentInfo *contentInfo = d2i_CMS_bio(cmsBIO, NULL);
  980. CMS_ContentInfo_print_ctx(printBIO, contentInfo, 0, NULL);
  981. BOOL verifyResult = CMS_verify(contentInfo, NULL, NULL, dataBIO, NULL, CMS_DETACHED | CMS_NO_SIGNER_CERT_VERIFY);
  982. struct stack_st_CMS_SignerInfo* signerInfos = CMS_get0_SignerInfos(contentInfo);
  983. if (verifyResult) {
  984. STACK_OF(X509) *signers = CMS_get0_signers(contentInfo);
  985. int numSigners = sk_X509_num(signers);
  986. for (NSString *certificate in certificates) {
  987. const char *ptrCertificate = [certificate cStringUsingEncoding:NSUTF8StringEncoding];
  988. BIO *certBio = BIO_new(BIO_s_mem());
  989. BIO_write(certBio, ptrCertificate,(unsigned int)strlen(ptrCertificate));
  990. X509 *certX509 = PEM_read_bio_X509(certBio, NULL, NULL, NULL);
  991. if (!certX509) {
  992. continue;
  993. }
  994. for (int i = 0; i < numSigners; ++i) {
  995. struct CMS_SignerInfo_st *signerInfo = sk_CMS_SignerInfo_value(signerInfos, i);
  996. if (CMS_SignerInfo_cert_cmp(signerInfo, certX509) == 0) {
  997. return true;
  998. }
  999. }
  1000. }
  1001. }
  1002. BIO_free(dataBIO);
  1003. BIO_free(printBIO);
  1004. BIO_free(cmsBIO);
  1005. return verifyResult;
  1006. }
  1007. #
  1008. #pragma mark - Utility
  1009. #
  1010. - (void)Encodedkey:(NSString **)key initializationVector:(NSString **)initializationVector
  1011. {
  1012. NSData *keyData = [self generateKey:AES_KEY_128_LENGTH];
  1013. NSData *ivData = [self generateIV:AES_IVEC_LENGTH];
  1014. *key = [keyData base64EncodedStringWithOptions:0];
  1015. *initializationVector = [ivData base64EncodedStringWithOptions:0];
  1016. }
  1017. - (NSString *)createSHA256:(NSData *)data
  1018. {
  1019. uint8_t digest[CC_SHA256_DIGEST_LENGTH];
  1020. CC_SHA256(data.bytes, (unsigned int)data.length, digest);
  1021. NSMutableString* output = [NSMutableString stringWithCapacity:CC_SHA256_DIGEST_LENGTH * 2];
  1022. for(int i = 0; i < CC_SHA256_DIGEST_LENGTH; i++)
  1023. [output appendFormat:@"%02x", digest[i]];
  1024. return output;
  1025. }
  1026. - (NSString *)createSHA512:(NSString *)string
  1027. {
  1028. const char *cstr = [string cStringUsingEncoding:NSUTF8StringEncoding];
  1029. NSData *data = [NSData dataWithBytes:cstr length:string.length];
  1030. uint8_t digest[CC_SHA512_DIGEST_LENGTH];
  1031. CC_SHA512(data.bytes, (unsigned int)data.length, digest);
  1032. NSMutableString* output = [NSMutableString stringWithCapacity:CC_SHA512_DIGEST_LENGTH * 2];
  1033. for(int i = 0; i < CC_SHA512_DIGEST_LENGTH; i++)
  1034. [output appendFormat:@"%02x", digest[i]];
  1035. return output;
  1036. }
  1037. - (NSData *)generateIV:(int)length
  1038. {
  1039. NSMutableData *ivData = [NSMutableData dataWithLength:length];
  1040. (void)SecRandomCopyBytes(kSecRandomDefault, length, ivData.mutableBytes);
  1041. return ivData;
  1042. }
  1043. - (NSData *)generateSalt:(int)length
  1044. {
  1045. NSMutableData *saltData = [NSMutableData dataWithLength:length];
  1046. (void)SecRandomCopyBytes(kSecRandomDefault, length, saltData.mutableBytes);
  1047. return saltData;
  1048. }
  1049. - (NSData *)generateKey:(int)length
  1050. {
  1051. NSMutableData *keyData = [NSMutableData dataWithLength:length];
  1052. unsigned char *pKeyData = [keyData mutableBytes];
  1053. RAND_bytes(pKeyData, length);
  1054. return keyData;
  1055. }
  1056. - (NSData *)generateKey
  1057. {
  1058. NSMutableData *keyData = [NSMutableData dataWithLength:AES_KEY_128_LENGTH];
  1059. unsigned char *pKeyData = [keyData mutableBytes];
  1060. RAND_bytes(pKeyData, AES_KEY_128_LENGTH);
  1061. return keyData;
  1062. }
  1063. - (NSString *)getSHA1:(NSString *)input
  1064. {
  1065. const char *cstr = [input cStringUsingEncoding:NSUTF8StringEncoding];
  1066. NSData *data = [NSData dataWithBytes:cstr length:input.length];
  1067. uint8_t digest[CC_SHA1_DIGEST_LENGTH];
  1068. CC_SHA1(data.bytes, (unsigned int)data.length, digest);
  1069. NSMutableString* output = [NSMutableString stringWithCapacity:CC_SHA1_DIGEST_LENGTH * 2];
  1070. for(int i = 0; i < CC_SHA1_DIGEST_LENGTH; i++)
  1071. [output appendFormat:@"%02x", digest[i]];
  1072. return output;
  1073. }
  1074. - (NSData *)hashValueMD5OfData:(NSData *)data
  1075. {
  1076. MD5_CTX md5Ctx;
  1077. unsigned char hashValue[MD5_DIGEST_LENGTH];
  1078. if(!MD5_Init(&md5Ctx)) {
  1079. return nil;
  1080. }
  1081. if (!MD5_Update(&md5Ctx, data.bytes, data.length)) {
  1082. return nil;
  1083. }
  1084. if (!MD5_Final(hashValue, &md5Ctx)) {
  1085. return nil;
  1086. }
  1087. return [NSData dataWithBytes:hashValue length:MD5_DIGEST_LENGTH];
  1088. }
  1089. - (NSString *)hexadecimalString:(NSData *)input
  1090. {
  1091. const unsigned char *dataBuffer = (const unsigned char *) [input bytes];
  1092. if (!dataBuffer) {
  1093. return [NSString string];
  1094. }
  1095. NSUInteger dataLength = [input length];
  1096. NSMutableString *hexString = [NSMutableString stringWithCapacity:(dataLength * 2)];
  1097. for (int i = 0; i < dataLength; ++i) {
  1098. [hexString appendString:[NSString stringWithFormat:@"%02lx", (unsigned long) dataBuffer[i]]];
  1099. }
  1100. return [NSString stringWithString:hexString];
  1101. }
  1102. - (NSString *)derToPemPrivateKey:(NSString *)input
  1103. {
  1104. NSInteger substringLength = 65;
  1105. NSMutableString *result = [NSMutableString stringWithString: input];
  1106. for(long i=substringLength;i<=input.length;i++) {
  1107. [result insertString: @"\n" atIndex: i];
  1108. i+=substringLength;
  1109. }
  1110. [result insertString: @"-----BEGIN PRIVATE KEY-----\n" atIndex: 0];
  1111. [result appendString:@"\n-----END PRIVATE KEY-----\n"];
  1112. return result;
  1113. }
  1114. - (NSString *)pubKeyToString:(EVP_PKEY *)pubkey
  1115. {
  1116. char *buf[256];
  1117. FILE *pFile;
  1118. NSString *pkey_string;
  1119. pFile = fmemopen(buf, sizeof(buf), "w");
  1120. PEM_write_PUBKEY(pFile,pubkey);
  1121. fputc('\0', pFile);
  1122. fclose(pFile);
  1123. pkey_string = [NSString stringWithUTF8String:(char *)buf];
  1124. return pkey_string;
  1125. }
  1126. @end