String+AES.swift

String+AES.swift

import Foundation
import CryptoSwift

extension String {
 
    func aesEncrypt(key: String, iv: String) -> String {
        let data = self.dataUsingEncoding(NSUTF8StringEncoding)
        let enc = AES(key: key, iv: iv, blockMode:.CBC)?.encrypt(data!.arrayOfBytes(), padding: PKCS7())
        let encData = NSData(bytes: enc!, length: Int(enc!.count))
        let base64String: String = encData.base64EncodedStringWithOptions(NSDataBase64EncodingOptions(rawValue: 0));
        let result = String(base64String)
        return result
    }
    
    func aesDecrypt(key: String, iv: String) -> String {
        let data = NSData(base64EncodedString: self, options: NSDataBase64DecodingOptions(rawValue: 0))
        let dec = AES(key: key, iv: iv, blockMode:.CBC)?.decrypt(data!.arrayOfBytes(), padding: PKCS7())
        let decData = NSData(bytes: dec!, length: Int(dec!.count))
        var result = NSString(data: decData, encoding: NSUTF8StringEncoding)
        return String(result!)
    }
}