-
Notifications
You must be signed in to change notification settings - Fork 22
/
AES.swift
72 lines (55 loc) · 2.13 KB
/
AES.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
//
// AES.swift
// CommonCryptoSwift
//
// Created by Khoa Pham on 27/07/16.
// Copyright © 2016 Fantageek. All rights reserved.
//
import Foundation
import CCommonCrypto
public struct AES {
// MARK: - Data
public static func encrypt(_ data: Data, key: Data) -> Data? {
return perform(data, key: key, encrypting: true)
}
public static func decrypt(_ data: Data, key: Data) -> Data? {
return perform(data, key: key, encrypting: false)
}
// MARK: - String
public static func encrypt(_ string: String, key: String) -> String? {
guard let data = string.data(using: String.Encoding.utf8),
let keyData = key.data(using: String.Encoding.utf8) else { return nil }
let encrypted = perform(data, key: keyData, encrypting: true)
return encrypted?.base64EncodedString(options: .lineLength64Characters)
}
public static func decrypt(_ string: String, key: String) -> String? {
guard let data = Data(base64Encoded: string, options: .ignoreUnknownCharacters),
let keyData = key.data(using: String.Encoding.utf8) else { return nil }
guard let decrypted = perform(data, key: keyData, encrypting: false) else { return nil }
return String(data: decrypted, encoding: String.Encoding.utf8)
}
// MARK: - Private
fileprivate static func perform(_ data: Data, key: Data, encrypting: Bool) -> Data? {
guard let out = NSMutableData(length: data.count + kCCBlockSizeAES128) else { return nil }
let hashData = Hash.SHA384(key)
let hashKeyData = hashData.subdata(in: 0..<32)
let ivData = hashData.subdata(in: 32..<48)
let operation = encrypting ? kCCEncrypt : kCCDecrypt
var dataOutMovedLength: size_t = 0
let status = CCCrypt(
CCOperation(operation),
CCAlgorithm(kCCAlgorithmAES128),
CCOptions(kCCOptionPKCS7Padding),
(hashKeyData as NSData).bytes,
kCCKeySizeAES128,
(ivData as NSData).bytes,
(data as NSData).bytes,
size_t(data.count),
out.mutableBytes,
size_t(out.length),
&dataOutMovedLength)
guard Int(status) == Int(kCCSuccess) else { return nil }
out.length = dataOutMovedLength
return out as Data
}
}