NCEndToEndEncryption.m 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995
  1. //
  2. // NCEndToEndEncryption.m
  3. // Nextcloud
  4. //
  5. // Created by Marino Faggiana on 19/09/17.
  6. // Copyright © 2017 TWS. All rights reserved.
  7. //
  8. // Author Marino Faggiana <m.faggiana@twsweb.it>
  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/x509.h>
  29. #import <openssl/bio.h>
  30. #import <openssl/err.h>
  31. #import <openssl/pem.h>
  32. #import <openssl/rsa.h>
  33. #import <openssl/pkcs12.h>
  34. #import <openssl/ssl.h>
  35. #import <openssl/err.h>
  36. #import <openssl/bn.h>
  37. #import <openssl/md5.h>
  38. #import <openssl/rand.h>
  39. #import <openssl/engine.h>
  40. #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);
  41. #define IV_DELIMITER_ENCODED @"fA==" // "|" base64 encoded
  42. #define PBKDF2_INTERACTION_COUNT 1024
  43. #define PBKDF2_KEY_LENGTH 256
  44. #define PBKDF2_SALT @"$4$YmBjm3hk$Qb74D5IUYwghUmzsMqeNFx5z0/8$"
  45. #define ASYMMETRIC_STRING_TEST @"Nextcloud a safe home for all your data"
  46. #define fileNameCertificate @"cert.pem"
  47. #define fileNameCSR @"csr.pem"
  48. #define fileNamePrivateKey @"privateKey.pem"
  49. #define fileNamePubliceKey @"publicKey.pem"
  50. #define AES_KEY_128_LENGTH 16
  51. #define AES_KEY_256_LENGTH 32
  52. #define AES_IVEC_LENGTH 16
  53. #define AES_GCM_TAG_LENGTH 16
  54. @interface NCEndToEndEncryption ()
  55. {
  56. NSData *_privateKeyData;
  57. NSData *_publicKeyData;
  58. NSData *_csrData;
  59. }
  60. @end
  61. @implementation NCEndToEndEncryption
  62. //Singleton
  63. + (instancetype)sharedManager {
  64. static NCEndToEndEncryption *NCEndToEndEncryption = nil;
  65. static dispatch_once_t onceToken;
  66. dispatch_once(&onceToken, ^{
  67. NCEndToEndEncryption = [self new];
  68. });
  69. return NCEndToEndEncryption;
  70. }
  71. #
  72. #pragma mark - Generate Certificate X509 - CSR - Private Key
  73. #
  74. - (BOOL)generateCertificateX509WithUserID:(NSString *)userID directoryUser:(NSString *)directoryUser
  75. {
  76. OPENSSL_init_ssl(0, NULL);
  77. OPENSSL_init_crypto(0, NULL);
  78. X509 *x509;
  79. x509 = X509_new();
  80. EVP_PKEY *pkey;
  81. NSError *keyError;
  82. pkey = [self generateRSAKey:&keyError];
  83. if (keyError) {
  84. return NO;
  85. }
  86. X509_set_pubkey(x509, pkey);
  87. EVP_PKEY_free(pkey);
  88. // Set Serial Number
  89. ASN1_INTEGER_set(X509_get_serialNumber(x509), 123);
  90. // Set Valididity Date Range
  91. long notBefore = [[NSDate date] timeIntervalSinceDate:[NSDate date]];
  92. long notAfter = [[[NSDate date] dateByAddingTimeInterval:60*60*24*365*10] timeIntervalSinceDate:[NSDate date]]; // 10 year
  93. X509_gmtime_adj((ASN1_TIME *)X509_get0_notBefore(x509), notBefore);
  94. X509_gmtime_adj((ASN1_TIME *)X509_get0_notAfter(x509), notAfter);
  95. X509_NAME *name = X509_get_subject_name(x509);
  96. // Now to add the subject name fields to the certificate
  97. // I use a macro here to make it cleaner.
  98. const unsigned char *cUserID = (const unsigned char *) [userID cStringUsingEncoding:NSUTF8StringEncoding];
  99. // Common Name = UserID.
  100. addName("CN", cUserID);
  101. // The organizational unit for the cert. Usually this is a department.
  102. addName("OU", "Certificate Authority");
  103. // The organization of the cert.
  104. addName("O", "Nextcloud");
  105. // The city of the organization.
  106. addName("L", "Vicenza");
  107. // The state/province of the organization.
  108. addName("S", "Italy");
  109. // The country (ISO 3166) of the organization
  110. addName("C", "IT");
  111. X509_set_issuer_name(x509, name);
  112. /*
  113. for (SANObject * san in self.options.sans) {
  114. if (!san.value || san.value.length <= 0) {
  115. continue;
  116. }
  117. NSString * prefix = san.type == SANObjectTypeIP ? @"IP:" : @"DNS:";
  118. NSString * value = [NSString stringWithFormat:@"%@%@", prefix, san.value];
  119. NSLog(@"Add subjectAltName %@", value);
  120. X509_EXTENSION * extension = NULL;
  121. ASN1_STRING * asnValue = ASN1_STRING_new();
  122. ASN1_STRING_set(asnValue, (const unsigned char *)[value UTF8String], (int)value.length);
  123. X509_EXTENSION_create_by_NID(&extension, NID_subject_alt_name, 0, asnValue);
  124. X509_add_ext(x509, extension, -1);
  125. }
  126. */
  127. // Specify the encryption algorithm of the signature.
  128. // SHA256 should suit your needs.
  129. if (X509_sign(x509, pkey, EVP_sha256()) < 0) {
  130. return NO;
  131. }
  132. X509_print_fp(stdout, x509);
  133. // Extract CSR, publicKey, privateKey
  134. int len;
  135. char *keyBytes;
  136. // CSR
  137. BIO *csrBIO = BIO_new(BIO_s_mem());
  138. X509_REQ *certReq = X509_to_X509_REQ(x509, pkey, EVP_sha256());
  139. PEM_write_bio_X509_REQ(csrBIO, certReq);
  140. len = BIO_pending(csrBIO);
  141. keyBytes = malloc(len);
  142. BIO_read(csrBIO, keyBytes, len);
  143. _csrData = [NSData dataWithBytes:keyBytes length:len];
  144. NSLog(@"[LOG] \n%@", [[NSString alloc] initWithData:_csrData encoding:NSUTF8StringEncoding]);
  145. // PublicKey
  146. BIO *publicKeyBIO = BIO_new(BIO_s_mem());
  147. PEM_write_bio_PUBKEY(publicKeyBIO, pkey);
  148. len = BIO_pending(publicKeyBIO);
  149. keyBytes = malloc(len);
  150. BIO_read(publicKeyBIO, keyBytes, len);
  151. _publicKeyData = [NSData dataWithBytes:keyBytes length:len];
  152. NSLog(@"[LOG] \n%@", [[NSString alloc] initWithData:_publicKeyData encoding:NSUTF8StringEncoding]);
  153. // PrivateKey
  154. BIO *privateKeyBIO = BIO_new(BIO_s_mem());
  155. PEM_write_bio_PKCS8PrivateKey(privateKeyBIO, pkey, NULL, NULL, 0, NULL, NULL);
  156. len = BIO_pending(privateKeyBIO);
  157. keyBytes = malloc(len);
  158. BIO_read(privateKeyBIO, keyBytes, len);
  159. _privateKeyData = [NSData dataWithBytes:keyBytes length:len];
  160. NSLog(@"[LOG] \n%@", [[NSString alloc] initWithData:_privateKeyData encoding:NSUTF8StringEncoding]);
  161. if(keyBytes)
  162. free(keyBytes);
  163. #ifdef DEBUG
  164. // Save to disk [DEBUG MODE]
  165. [self saveToDiskPEMWithCert:x509 key:pkey directoryUser:directoryUser];
  166. #endif
  167. return YES;
  168. }
  169. - (EVP_PKEY *)generateRSAKey:(NSError **)error
  170. {
  171. EVP_PKEY *pkey = EVP_PKEY_new();
  172. if (!pkey) {
  173. return NULL;
  174. }
  175. BIGNUM *bigNumber = BN_new();
  176. int exponent = RSA_F4;
  177. RSA *rsa = RSA_new();
  178. if (BN_set_word(bigNumber, exponent) < 0) {
  179. goto cleanup;
  180. }
  181. if (RSA_generate_key_ex(rsa, 2048, bigNumber, NULL) < 0) {
  182. goto cleanup;
  183. }
  184. if (!EVP_PKEY_set1_RSA(pkey, rsa)) {
  185. goto cleanup;
  186. }
  187. cleanup:
  188. RSA_free(rsa);
  189. BN_free(bigNumber);
  190. return pkey;
  191. }
  192. - (BOOL)saveToDiskPEMWithCert:(X509 *)x509 key:(EVP_PKEY *)pkey directoryUser:(NSString *)directoryUser
  193. {
  194. FILE *f;
  195. // Certificate
  196. NSString *certificatePath = [NSString stringWithFormat:@"%@/%@", directoryUser, fileNameCertificate];
  197. f = fopen([certificatePath fileSystemRepresentation], "wb");
  198. if (PEM_write_X509(f, x509) < 0) {
  199. // Error writing to disk.
  200. fclose(f);
  201. return NO;
  202. }
  203. NSLog(@"[LOG] Saved cert to %@", certificatePath);
  204. fclose(f);
  205. // PublicKey
  206. NSString *publicKeyPath = [NSString stringWithFormat:@"%@/%@", directoryUser, fileNamePubliceKey];
  207. f = fopen([publicKeyPath fileSystemRepresentation], "wb");
  208. if (PEM_write_PUBKEY(f, pkey) < 0) {
  209. // Error
  210. fclose(f);
  211. return NO;
  212. }
  213. NSLog(@"[LOG] Saved publicKey to %@", publicKeyPath);
  214. fclose(f);
  215. // Here you write the private key (pkey) to disk. OpenSSL will encrypt the
  216. // file using the password and cipher you provide.
  217. //if (PEM_write_PrivateKey(f, pkey, EVP_des_ede3_cbc(), (unsigned char *)[password UTF8String], (int)password.length, NULL, NULL) < 0) {
  218. // PrivateKey
  219. NSString *privatekeyPath = [NSString stringWithFormat:@"%@/%@", directoryUser, fileNamePrivateKey];
  220. f = fopen([privatekeyPath fileSystemRepresentation], "wb");
  221. if (PEM_write_PrivateKey(f, pkey, NULL, NULL, 0, NULL, NULL) < 0) {
  222. // Error
  223. fclose(f);
  224. return NO;
  225. }
  226. NSLog(@"[LOG] Saved privatekey to %@", privatekeyPath);
  227. fclose(f);
  228. // CSR Request sha256
  229. NSString *csrPath = [NSString stringWithFormat:@"%@/%@", directoryUser, fileNameCSR];
  230. f = fopen([csrPath fileSystemRepresentation], "wb");
  231. X509_REQ *certreq = X509_to_X509_REQ(x509, pkey, EVP_sha256());
  232. if (PEM_write_X509_REQ(f, certreq) < 0) {
  233. // Error
  234. fclose(f);
  235. return NO;
  236. }
  237. NSLog(@"[LOG] Saved csr to %@", csrPath);
  238. fclose(f);
  239. return YES;
  240. }
  241. - (BOOL)saveP12WithCert:(X509 *)x509 key:(EVP_PKEY *)pkey directoryUser:(NSString *)directoryUser finished:(void (^)(NSError *))finished
  242. {
  243. //PKCS12 * p12 = PKCS12_create([password UTF8String], NULL, pkey, x509, NULL, 0, 0, PKCS12_DEFAULT_ITER, 1, NID_key_usage);
  244. PKCS12 *p12 = PKCS12_create(NULL, NULL, pkey, x509, NULL, 0, 0, PKCS12_DEFAULT_ITER, 1, NID_key_usage);
  245. NSString *path = [NSString stringWithFormat:@"%@/certificate.p12", directoryUser];
  246. FILE *f = fopen([path fileSystemRepresentation], "wb");
  247. if (i2d_PKCS12_fp(f, p12) != 1) {
  248. fclose(f);
  249. return NO;
  250. }
  251. NSLog(@"[LOG] Saved p12 to %@", path);
  252. fclose(f);
  253. return YES;
  254. }
  255. #
  256. #pragma mark - Create CSR & Encrypt Private Key
  257. #
  258. - (NSString *)createCSR:(NSString *)userID directoryUser:(NSString *)directoryUser
  259. {
  260. // Create Certificate, if do not exists
  261. if (!_csrData) {
  262. if (![self generateCertificateX509WithUserID:userID directoryUser:directoryUser])
  263. return nil;
  264. }
  265. NSString *csr = [[NSString alloc] initWithData:_csrData encoding:NSUTF8StringEncoding];
  266. return csr;
  267. }
  268. - (NSString *)encryptPrivateKey:(NSString *)userID directoryUser: (NSString *)directoryUser passphrase:(NSString *)passphrase privateKey:(NSString **)privateKey
  269. {
  270. NSMutableData *privateKeyCipherData = [NSMutableData new];
  271. if (!_privateKeyData) {
  272. if (![self generateCertificateX509WithUserID:userID directoryUser:directoryUser])
  273. return nil;
  274. }
  275. NSMutableData *keyData = [NSMutableData dataWithLength:PBKDF2_KEY_LENGTH/8];
  276. NSData *saltData = [PBKDF2_SALT dataUsingEncoding:NSUTF8StringEncoding];
  277. // Remove all whitespaces from passphrase
  278. passphrase = [passphrase stringByReplacingOccurrencesOfString:@" " withString:@""];
  279. CCKeyDerivationPBKDF(kCCPBKDF2, passphrase.UTF8String, passphrase.length, saltData.bytes, saltData.length, kCCPRFHmacAlgSHA1, PBKDF2_INTERACTION_COUNT, keyData.mutableBytes, keyData.length);
  280. NSData *ivData = [self generateIV:AES_IVEC_LENGTH];
  281. NSData *tagData = [NSData new];
  282. /* ENCODE 64 privateKey JAVA compatibility */
  283. NSString *privateKeyBase64 = [_privateKeyData base64EncodedStringWithOptions:0];
  284. NSData *privateKeyBase64Data = [privateKeyBase64 dataUsingEncoding:NSUTF8StringEncoding];
  285. /* --------------------------------------- */
  286. BOOL result = [self encryptData:privateKeyBase64Data cipherData:&privateKeyCipherData keyData:keyData keyLen:AES_KEY_256_LENGTH ivData:ivData tagData:&tagData];
  287. if (result && privateKeyCipherData) {
  288. NSString *privateKeyCipherBase64;
  289. NSString *initVectorBase64;
  290. NSString *privateKeyCipherWithInitVectorBase64;
  291. privateKeyCipherBase64 = [privateKeyCipherData base64EncodedStringWithOptions:0];
  292. initVectorBase64 = [ivData base64EncodedStringWithOptions:0];
  293. privateKeyCipherWithInitVectorBase64 = [NSString stringWithFormat:@"%@%@%@", privateKeyCipherBase64, IV_DELIMITER_ENCODED, initVectorBase64];
  294. *privateKey = [[NSString alloc] initWithData:_privateKeyData encoding:NSUTF8StringEncoding];
  295. return privateKeyCipherWithInitVectorBase64;
  296. } else {
  297. return nil;
  298. }
  299. }
  300. #
  301. #pragma mark - Decrypt Private Key
  302. #
  303. - (NSString *)decryptPrivateKey:(NSString *)privateKeyCipher passphrase:(NSString *)passphrase publicKey:(NSString *)publicKey
  304. {
  305. NSMutableData *privateKeyData = [NSMutableData new];
  306. NSString *privateKey;
  307. // Key (data)
  308. NSMutableData *keyData = [NSMutableData dataWithLength:PBKDF2_KEY_LENGTH/8];
  309. NSData *saltData = [PBKDF2_SALT dataUsingEncoding:NSUTF8StringEncoding];
  310. // Remove all whitespaces from passphrase
  311. passphrase = [passphrase stringByReplacingOccurrencesOfString:@" " withString:@""];
  312. CCKeyDerivationPBKDF(kCCPBKDF2, passphrase.UTF8String, passphrase.length, saltData.bytes, saltData.length, kCCPRFHmacAlgSHA1, PBKDF2_INTERACTION_COUNT, keyData.mutableBytes, keyData.length);
  313. // Find range for IV_DELIMITER_ENCODED
  314. NSRange range = [privateKeyCipher rangeOfString:IV_DELIMITER_ENCODED];
  315. // Init Vector
  316. NSString *ivBase64 = [privateKeyCipher substringFromIndex:(range.location + range.length)];
  317. NSData *ivData = [[NSData alloc] initWithBase64EncodedString:ivBase64 options:0];
  318. // TAG
  319. NSString *tagBase64 = [privateKeyCipher substringWithRange:NSMakeRange(range.location - AES_GCM_TAG_LENGTH, AES_GCM_TAG_LENGTH)];
  320. NSData *tagData = [[NSData alloc] initWithBase64EncodedString:tagBase64 options:0];
  321. // PrivateKey
  322. NSString *privateKeyCipherBase64 = [privateKeyCipher substringToIndex:(range.location)];
  323. NSData *privateKeyCipherData = [[NSData alloc] initWithBase64EncodedString:privateKeyCipherBase64 options:0];
  324. // Decrypt
  325. BOOL result = [self decryptData:privateKeyCipherData plainData:&privateKeyData keyData:keyData keyLen:AES_KEY_256_LENGTH ivData:ivData tagData:tagData];
  326. if (result && privateKeyData)
  327. /* DENCODE 64 privateKey JAVA compatibility */
  328. privateKey = [self base64Decode:privateKeyData];
  329. /* ---------------------------------------- */
  330. if (privateKey) {
  331. NSData *encryptData = [self encryptAsymmetricString:ASYMMETRIC_STRING_TEST publicKey:publicKey];
  332. if (!encryptData)
  333. return nil;
  334. NSString *decryptString = [self decryptAsymmetricData:encryptData privateKey:privateKey];
  335. if (decryptString && [decryptString isEqualToString:ASYMMETRIC_STRING_TEST])
  336. return privateKey;
  337. else
  338. return nil;
  339. return privateKey;
  340. } else {
  341. return nil;
  342. }
  343. }
  344. #
  345. #pragma mark - Asymmetric Encrypt/Decrypt String
  346. #
  347. - (NSData *)encryptAsymmetricString:(NSString *)plain publicKey:(NSString *)publicKey
  348. {
  349. unsigned char *pKey = (unsigned char *)[publicKey UTF8String];
  350. ENGINE *eng = NULL;
  351. int status = 0;
  352. // Extract real publicKey
  353. BIO *bio = BIO_new_mem_buf(pKey, -1);
  354. if (!bio)
  355. return nil;
  356. X509 *x509 = PEM_read_bio_X509(bio, NULL, 0, NULL);
  357. if (!x509)
  358. return nil;
  359. EVP_PKEY *key = X509_get_pubkey(x509);
  360. if (!key)
  361. return nil;
  362. EVP_PKEY_CTX *ctx = EVP_PKEY_CTX_new(key, eng);
  363. if (!ctx)
  364. return nil;
  365. status = EVP_PKEY_encrypt_init(ctx);
  366. if (status <= 0)
  367. return nil;
  368. status = EVP_PKEY_CTX_set_rsa_padding(ctx, RSA_PKCS1_OAEP_PADDING);
  369. if (status <= 0)
  370. return nil;
  371. status = EVP_PKEY_CTX_set_rsa_oaep_md(ctx, EVP_sha256());
  372. if (status <= 0)
  373. return nil;
  374. status = EVP_PKEY_CTX_set_rsa_mgf1_md(ctx, EVP_sha256());
  375. if (status <= 0)
  376. return nil;
  377. unsigned long outLen = 0;
  378. NSData *plainData = [plain dataUsingEncoding:NSUTF8StringEncoding];
  379. status = EVP_PKEY_encrypt(ctx, NULL, &outLen, [plainData bytes], (int)[plainData length]);
  380. if (status <= 0 || outLen == 0)
  381. return nil;
  382. unsigned char *out = (unsigned char *) malloc(outLen);
  383. status = EVP_PKEY_encrypt(ctx, out, &outLen, [plainData bytes], (int)[plainData length]);
  384. if (status <= 0)
  385. return nil;
  386. NSData *outData = [[NSData alloc] initWithBytes:out length:outLen];
  387. if (out)
  388. free(out);
  389. return outData;
  390. }
  391. - (NSString *)decryptAsymmetricData:(NSData *)cipherData privateKey:(NSString *)privateKey
  392. {
  393. unsigned char *pKey = (unsigned char *)[privateKey UTF8String];
  394. ENGINE *eng = ENGINE_get_default_RSA();
  395. int status = 0;
  396. BIO *bio = BIO_new_mem_buf(pKey, -1);
  397. if (!bio)
  398. return nil;
  399. EVP_PKEY *key = PEM_read_bio_PrivateKey(bio, NULL, NULL, NULL);
  400. if (!key)
  401. return nil;
  402. EVP_PKEY_CTX *ctx = EVP_PKEY_CTX_new(key, eng);
  403. if (!ctx)
  404. return nil;
  405. status = EVP_PKEY_decrypt_init(ctx);
  406. if (status <= 0)
  407. return nil;
  408. status = EVP_PKEY_CTX_set_rsa_padding(ctx, RSA_PKCS1_OAEP_PADDING);
  409. if (status <= 0)
  410. return nil;
  411. status = EVP_PKEY_CTX_set_rsa_oaep_md(ctx, EVP_sha256());
  412. if (status <= 0)
  413. return nil;
  414. status = EVP_PKEY_CTX_set_rsa_mgf1_md(ctx, EVP_sha256());
  415. if (status <= 0)
  416. return nil;
  417. unsigned long outLen = 0;
  418. status = EVP_PKEY_decrypt(ctx, NULL, &outLen, [cipherData bytes], (int)[cipherData length]);
  419. if (status <= 0 || outLen == 0)
  420. return nil;
  421. unsigned char *out = (unsigned char *) malloc(outLen);
  422. status = EVP_PKEY_decrypt(ctx, out, &outLen, [cipherData bytes], (int)[cipherData length]);
  423. if (status <= 0)
  424. return nil;
  425. NSString *outString = [[NSString alloc] initWithBytes:out length:outLen encoding:NSUTF8StringEncoding];
  426. if (out)
  427. free(out);
  428. return outString;
  429. }
  430. #
  431. #pragma mark - Encrypt / Decrypt Metadata
  432. #
  433. - (NSString *)decryptMetadata:(NSString *)encrypted privateKey:(NSString *)privateKey initializationVector:(NSString *)initializationVector authenticationTag:(NSString *)authenticationTag
  434. {
  435. NSMutableData *plainData;
  436. NSData *keyData = [privateKey dataUsingEncoding:NSUTF8StringEncoding];
  437. NSData *ivData = [initializationVector dataUsingEncoding:NSUTF8StringEncoding];
  438. // Tag
  439. NSRange range = [encrypted rangeOfString:IV_DELIMITER_ENCODED];
  440. authenticationTag = [encrypted substringWithRange:NSMakeRange(range.location - AES_GCM_TAG_LENGTH, AES_GCM_TAG_LENGTH)];
  441. NSData *tagData = [authenticationTag dataUsingEncoding:NSUTF8StringEncoding];
  442. // Cipher
  443. NSString *cipher = [encrypted substringToIndex:(range.location)];
  444. NSData *cipherData = [cipher dataUsingEncoding:NSUTF8StringEncoding];
  445. BOOL result = [self decryptData:cipherData plainData:&plainData keyData:keyData keyLen:AES_KEY_128_LENGTH ivData:ivData tagData:tagData];
  446. if (plainData != nil && result) {
  447. NSString *plain = [plainData base64EncodedStringWithOptions:0];
  448. return plain;
  449. } else {
  450. return nil;
  451. }
  452. }
  453. #
  454. #pragma mark - Encrypt / Decrypt file
  455. #
  456. - (BOOL)encryptFileName:(NSString *)fileName fileNameIdentifier:(NSString *)fileNameIdentifier directoryUser:(NSString *)directoryUser key:(NSString **)key initializationVector:(NSString **)initializationVector authenticationTag:(NSString **)authenticationTag
  457. {
  458. NSMutableData *cipherData;
  459. NSData *tagData;
  460. NSData *plainData;
  461. plainData = [[NSFileManager defaultManager] contentsAtPath:[NSString stringWithFormat:@"%@/%@", directoryUser, fileName]];
  462. if (plainData == nil)
  463. return false;
  464. NSData *keyData = [self generateKey:AES_KEY_128_LENGTH];
  465. NSData *ivData = [self generateIV:AES_IVEC_LENGTH];
  466. BOOL result = [self encryptData:plainData cipherData:&cipherData keyData:keyData keyLen:AES_KEY_128_LENGTH ivData:ivData tagData:&tagData];
  467. if (cipherData != nil && result) {
  468. [cipherData writeToFile:[NSString stringWithFormat:@"%@/%@", directoryUser, fileNameIdentifier] atomically:YES];
  469. *key = [keyData base64EncodedStringWithOptions:0];
  470. *initializationVector = [ivData base64EncodedStringWithOptions:0];
  471. *authenticationTag = [tagData base64EncodedStringWithOptions:0];
  472. return true;
  473. }
  474. return false;
  475. }
  476. /*
  477. - (void)decryptMetadata:(NSString *)metadata activeUrl:(NSString *)activeUrl
  478. {
  479. NSMutableData *plainData;
  480. NSData *cipherData = [[NSFileManager defaultManager] contentsAtPath:[NSString stringWithFormat:@"%@/%@", activeUrl, metadata.fileID]];
  481. NSData *keyData = [[NSData alloc] initWithBase64EncodedString:@"WANM0gRv+DhaexIsI0T3Lg==" options:0];
  482. NSData *ivData = [[NSData alloc] initWithBase64EncodedString:@"gKm3n+mJzeY26q4OfuZEqg==" options:0];
  483. NSData *tagData = [[NSData alloc] initWithBase64EncodedString:@"PboI9tqHHX3QeAA22PIu4w==" options:0];
  484. BOOL result = [self decryptData:cipherData plainData:&plainData keyData:keyData keyLen:AES_KEY_128_LENGTH ivData:ivData tagData:tagData];
  485. if (plainData != nil && result) {
  486. [plainData writeToFile:[NSString stringWithFormat:@"%@/%@", activeUrl, @"decrypted"] atomically:YES];
  487. }
  488. }
  489. - (NSString *)decryptMetadata:(NSString *)cipher key:(NSString *)key iv:(NSString *)iv tag:(NSString *)tag
  490. {
  491. NSMutableData *plainData;
  492. NSData *cipherData = [cipher dataUsingEncoding:NSUTF8StringEncoding];
  493. NSData *keyData = [key dataUsingEncoding:NSUTF8StringEncoding];
  494. NSData *ivData = [iv dataUsingEncoding:NSUTF8StringEncoding];
  495. NSData *tagData = [tag dataUsingEncoding:NSUTF8StringEncoding];
  496. BOOL result = [self decryptData:cipherData plainData:&plainData keyData:keyData keyLen:AES_KEY_128_LENGTH ivData:ivData tagData:tagData];
  497. if (plainData != nil && result)
  498. return [[NSString alloc] initWithData:plainData encoding:NSUTF8StringEncoding];
  499. else
  500. return nil;
  501. }
  502. */
  503. #
  504. #pragma mark - AES/GCM/NoPadding
  505. #
  506. // Encryption using GCM mode
  507. - (BOOL)encryptData:(NSData *)plainData cipherData:(NSMutableData **)cipherData keyData:(NSData *)keyData keyLen:(int)keyLen ivData:(NSData *)ivData tagData:(NSData **)tagData
  508. {
  509. int status = 0;
  510. int len = 0;
  511. NSData *printData;
  512. // set up key
  513. len = keyLen;
  514. unsigned char cKey[len];
  515. bzero(cKey, sizeof(cKey));
  516. [keyData getBytes:cKey length:len];
  517. // ----- DEBUG Print -----
  518. printData = [NSData dataWithBytes:cKey length:len];
  519. NSLog(@"Key %@", [printData base64EncodedStringWithOptions:0]);
  520. // -----------------------
  521. // set up ivec
  522. len = AES_IVEC_LENGTH;
  523. unsigned char cIV[len];
  524. bzero(cIV, sizeof(cIV));
  525. [ivData getBytes:cIV length:len];
  526. // ----- DEBUG Print -----
  527. printData = [NSData dataWithBytes:cIV length:len];
  528. NSLog(@"IV %@", [printData base64EncodedStringWithOptions:0]);
  529. // -----------------------
  530. // set up tag
  531. len = AES_GCM_TAG_LENGTH;
  532. unsigned char cTag[len];
  533. bzero(cTag, sizeof(cTag));
  534. // Create and initialise the context
  535. EVP_CIPHER_CTX *ctx = EVP_CIPHER_CTX_new();
  536. if (! ctx)
  537. return NO;
  538. // Initialise the encryption operation
  539. if (keyLen == AES_KEY_128_LENGTH)
  540. status = EVP_EncryptInit_ex (ctx, EVP_aes_128_gcm(), NULL, NULL, NULL);
  541. else if (keyLen == AES_KEY_256_LENGTH)
  542. status = EVP_EncryptInit_ex (ctx, EVP_aes_256_gcm(), NULL, NULL, NULL);
  543. if (status <= 0)
  544. return NO;
  545. // Set IV length. Not necessary if this is 12 bytes (96 bits)
  546. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, (int)sizeof(cIV), NULL);
  547. if (status <= 0)
  548. return NO;
  549. // Initialise key and IV
  550. status = EVP_EncryptInit_ex (ctx, NULL, NULL, cKey, cIV);
  551. if (status <= 0)
  552. return NO;
  553. // Provide the message to be encrypted, and obtain the encrypted output
  554. *cipherData = [NSMutableData dataWithLength:[plainData length]];
  555. unsigned char * ctBytes = [*cipherData mutableBytes];
  556. int pCipherLen = 0;
  557. status = EVP_EncryptUpdate(ctx, ctBytes, &pCipherLen, [plainData bytes], (int)[plainData length]);
  558. if (status <= 0)
  559. return NO;
  560. //Finalise the encryption
  561. len = pCipherLen;
  562. status = EVP_EncryptFinal_ex(ctx, ctBytes+pCipherLen, &len);
  563. if (status <= 0)
  564. return NO;
  565. //Get the tag
  566. status = EVP_CIPHER_CTX_ctrl (ctx, EVP_CTRL_GCM_GET_TAG, (int)sizeof(cTag), cTag);
  567. *tagData = [NSData dataWithBytes:cTag length:sizeof(cTag)];
  568. // add TAG JAVA compatibility
  569. [*cipherData appendData:*tagData];
  570. // --------------------------
  571. // Free
  572. EVP_CIPHER_CTX_free(ctx);
  573. return status; // OpenSSL uses 1 for success
  574. }
  575. // Decryption using GCM mode
  576. - (BOOL)decryptData:(NSData *)cipherData plainData:(NSMutableData **)plainData keyData:(NSData *)keyData keyLen:(int)keyLen ivData:(NSData *)ivData tagData:(NSData *)tagData
  577. {
  578. int status = 0;
  579. int len = 0;
  580. NSData *printData;
  581. // set up key
  582. len = keyLen;
  583. unsigned char cKey[len];
  584. bzero(cKey, sizeof(cKey));
  585. [keyData getBytes:cKey length:len];
  586. // ----- DEBUG Print -----
  587. printData = [NSData dataWithBytes:cKey length:len];
  588. NSLog(@"Key %@", [printData base64EncodedStringWithOptions:0]);
  589. // -----------------------
  590. // set up ivec
  591. len = (int)[ivData length];
  592. unsigned char cIV[len];
  593. bzero(cIV, sizeof(cIV));
  594. [ivData getBytes:cIV length:len];
  595. // ----- DEBUG Print -----
  596. printData = [NSData dataWithBytes:cIV length:len];
  597. NSLog(@"IV %@", [printData base64EncodedStringWithOptions:0]);
  598. // -----------------------
  599. // set up tag
  600. len = (int)[tagData length];;
  601. unsigned char cTag[len];
  602. bzero(cTag, sizeof(cTag));
  603. [tagData getBytes:cTag length:len];
  604. // ----- DEBUG Print -----
  605. printData = [NSData dataWithBytes:cTag length:len];
  606. NSLog(@"Tag %@", [printData base64EncodedStringWithOptions:0]);
  607. // -----------------------
  608. // Create and initialise the context
  609. EVP_CIPHER_CTX *ctx = EVP_CIPHER_CTX_new();
  610. if (! ctx)
  611. return NO;
  612. // Initialise the decryption operation
  613. if (keyLen == AES_KEY_128_LENGTH)
  614. status = EVP_DecryptInit_ex(ctx, EVP_aes_128_gcm(), NULL, NULL, NULL);
  615. else if (keyLen == AES_KEY_256_LENGTH)
  616. status = EVP_DecryptInit_ex(ctx, EVP_aes_256_gcm(), NULL, NULL, NULL);
  617. if (status <= 0)
  618. return NO;
  619. // Set IV length. Not necessary if this is 12 bytes (96 bits)
  620. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, (int)sizeof(cIV), NULL);
  621. if (status <= 0)
  622. return NO;
  623. // Initialise key and IV
  624. status = EVP_DecryptInit_ex(ctx, NULL, NULL, cKey, cIV);
  625. if (status <= 0)
  626. return NO;
  627. // remove TAG JAVA compatibility
  628. cipherData = [cipherData subdataWithRange:NSMakeRange(0, cipherData.length - 16)];
  629. // -----------------------------
  630. // Provide the message to be decrypted, and obtain the plaintext output
  631. *plainData = [NSMutableData dataWithLength:([cipherData length])];
  632. int pPlainLen = 0;
  633. unsigned char * pPlain = [*plainData mutableBytes];
  634. status = EVP_DecryptUpdate(ctx, pPlain, &pPlainLen, [cipherData bytes], (int)([cipherData length]));
  635. if (status <= 0)
  636. return NO;
  637. // Tag is the last 16 bytes
  638. status = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_TAG, (int)sizeof(cTag), cTag);
  639. if (status <= 0)
  640. return NO;
  641. //Finalise the encryption
  642. EVP_DecryptFinal_ex(ctx,NULL, &pPlainLen);
  643. // Free'
  644. EVP_CIPHER_CTX_free(ctx);
  645. return status; // OpenSSL uses 1 for success
  646. }
  647. #
  648. #pragma mark - Utility
  649. #
  650. - (NSString *)createSHA512:(NSString *)string
  651. {
  652. const char *cstr = [string cStringUsingEncoding:NSUTF8StringEncoding];
  653. NSData *data = [NSData dataWithBytes:cstr length:string.length];
  654. uint8_t digest[CC_SHA512_DIGEST_LENGTH];
  655. CC_SHA512(data.bytes, (unsigned int)data.length, digest);
  656. NSMutableString* output = [NSMutableString stringWithCapacity:CC_SHA512_DIGEST_LENGTH * 2];
  657. for(int i = 0; i < CC_SHA512_DIGEST_LENGTH; i++)
  658. [output appendFormat:@"%02x", digest[i]];
  659. return output;
  660. }
  661. - (NSData *)generateIV:(int)length
  662. {
  663. NSMutableData *ivData = [NSMutableData dataWithLength:length];
  664. (void)SecRandomCopyBytes(kSecRandomDefault, length, ivData.mutableBytes);
  665. return ivData;
  666. }
  667. - (NSData *)generateKey:(int)length
  668. {
  669. NSMutableData *keyData = [NSMutableData dataWithLength:length];
  670. unsigned char *pKeyData = [keyData mutableBytes];
  671. RAND_bytes(pKeyData, length);
  672. return keyData;
  673. }
  674. - (NSString *)getMD5:(NSString *)input
  675. {
  676. // Create pointer to the string as UTF8
  677. const char *ptr = [input cStringUsingEncoding:NSUTF8StringEncoding];
  678. // Create byte array of unsigned chars
  679. unsigned char md5Buffer[CC_MD5_DIGEST_LENGTH];
  680. // Create 16 byte MD5 hash value, store in buffer
  681. CC_MD5(ptr, (unsigned int)strlen(ptr), md5Buffer);
  682. // Convert MD5 value in the buffer to NSString of hex values
  683. NSMutableString *output = [NSMutableString stringWithCapacity:CC_MD5_DIGEST_LENGTH * 2];
  684. for(int i = 0; i < CC_MD5_DIGEST_LENGTH; i++)
  685. [output appendFormat:@"%02x",md5Buffer[i]];
  686. return output;
  687. }
  688. - (NSString *)getSHA1:(NSString *)input
  689. {
  690. const char *cstr = [input cStringUsingEncoding:NSUTF8StringEncoding];
  691. NSData *data = [NSData dataWithBytes:cstr length:input.length];
  692. uint8_t digest[CC_SHA1_DIGEST_LENGTH];
  693. CC_SHA1(data.bytes, (unsigned int)data.length, digest);
  694. NSMutableString* output = [NSMutableString stringWithCapacity:CC_SHA1_DIGEST_LENGTH * 2];
  695. for(int i = 0; i < CC_SHA1_DIGEST_LENGTH; i++)
  696. [output appendFormat:@"%02x", digest[i]];
  697. return output;
  698. }
  699. - (NSData *)hashValueMD5OfData:(NSData *)data
  700. {
  701. MD5_CTX md5Ctx;
  702. unsigned char hashValue[MD5_DIGEST_LENGTH];
  703. if(!MD5_Init(&md5Ctx)) {
  704. return nil;
  705. }
  706. if (!MD5_Update(&md5Ctx, data.bytes, data.length)) {
  707. return nil;
  708. }
  709. if (!MD5_Final(hashValue, &md5Ctx)) {
  710. return nil;
  711. }
  712. return [NSData dataWithBytes:hashValue length:MD5_DIGEST_LENGTH];
  713. }
  714. - (NSString *)hexadecimalString:(NSData *)input
  715. {
  716. const unsigned char *dataBuffer = (const unsigned char *) [input bytes];
  717. if (!dataBuffer) {
  718. return [NSString string];
  719. }
  720. NSUInteger dataLength = [input length];
  721. NSMutableString *hexString = [NSMutableString stringWithCapacity:(dataLength * 2)];
  722. for (int i = 0; i < dataLength; ++i) {
  723. [hexString appendString:[NSString stringWithFormat:@"%02lx", (unsigned long) dataBuffer[i]]];
  724. }
  725. return [NSString stringWithString:hexString];
  726. }
  727. - (NSString *)base64Encode:(NSData *)input
  728. {
  729. void *bytes;
  730. BIO *buffer = BIO_new(BIO_s_mem());
  731. BIO *base64 = BIO_new(BIO_f_base64());
  732. buffer = BIO_push(base64, buffer);
  733. BIO_write(buffer, [input bytes], (int)[input length]);
  734. NSUInteger length = BIO_get_mem_data(buffer, &bytes);
  735. NSString *string = [[NSString alloc] initWithBytes:bytes length:length encoding:NSUTF8StringEncoding];
  736. BIO_free_all(buffer);
  737. return string;
  738. }
  739. - (NSString *)base64Decode:(NSData *)input
  740. {
  741. NSMutableData *data = [NSMutableData data];
  742. BIO *buffer = BIO_new_mem_buf((void *)[input bytes], (int)[input length]);
  743. BIO *base64 = BIO_new(BIO_f_base64());
  744. buffer = BIO_push(base64, buffer);
  745. BIO_set_flags(base64, BIO_FLAGS_BASE64_NO_NL);
  746. char chars[input.length];
  747. int length = BIO_read(buffer, chars, (int)sizeof(chars));
  748. while (length > 0) {
  749. [data appendBytes:chars length:length];
  750. length = BIO_read(buffer, chars, (int)sizeof(chars));
  751. }
  752. BIO_free_all(buffer);
  753. return [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
  754. }
  755. - (NSString *)derToPemPrivateKey:(NSString *)input
  756. {
  757. NSInteger substringLength = 65;
  758. NSMutableString *result = [NSMutableString stringWithString: input];
  759. for(long i=substringLength;i<=input.length;i++) {
  760. [result insertString: @"\n" atIndex: i];
  761. i+=substringLength;
  762. }
  763. [result insertString: @"-----BEGIN PRIVATE KEY-----\n" atIndex: 0];
  764. [result appendString:@"\n-----END PRIVATE KEY-----\n"];
  765. return result;
  766. }
  767. @end