import UIKit
import CommonCrypto

class MBAAESSwift: NSObject {


/// AES128加密
/// - Parameters:
///   - data: 需要加密的數(shù)據(jù)
///   - key: 加密 密鑰 16 為長(zhǎng)度
/// - Returns: 加密后的數(shù)據(jù)
@objc class func encrypt(data:Data,key:String) -> Data? {
    
    let keyLength = kCCKeySizeAES128 + 1
    
    var keyPtr:Array<CChar> = []
    
    for _:Int in 0 ..< keyLength {
        keyPtr.append(CChar(0))
    }

    if key.getCString(&keyPtr, maxLength: keyLength, encoding: .utf8) {
        
        let dataLength:Int = data.count
        
        let bufferSize:Int = dataLength + kCCBlockSizeAES128
        
        let buffer = malloc(bufferSize)
        
        var numBytesEncrypted:Int = 0
        
        let bytes = [UInt8](data)
        
        let cryptStatus:CCCryptorStatus = CCCrypt(CCOperation(kCCEncrypt), CCAlgorithm(kCCAlgorithmAES), CCOptions(kCCOptionPKCS7Padding | kCCOptionECBMode) , keyPtr, kCCKeySizeAES128, nil, bytes, dataLength, buffer, bufferSize, &numBytesEncrypted)
        
        var outData:Data?
        
        if cryptStatus == kCCSuccess {
            
            outData = Data.init(bytes: buffer!, count: numBytesEncrypted)
        }
        
        free(buffer)
        
        return outData
    }
    
    return nil;
}

/// AES 128 解密
/// - Parameters:
///   - data: 需要解密的數(shù)據(jù)
///   - key: 解密的密鑰 16位
/// - Returns: 解密后的數(shù)據(jù)
@objc class func decrypt(data:Data,key:String) -> Data? {

    let keyLength = kCCKeySizeAES128 + 1
    
    var keyPtr:Array<CChar> = []
    
    for _:Int in 0 ..< keyLength {
        keyPtr.append(CChar(0))
    }

    if key.getCString(&keyPtr, maxLength: keyLength, encoding: .utf8) {
        
        let dataLength:Int = data.count
        
        let bufferSize:Int = dataLength + kCCBlockSizeAES128
        
        let buffer = malloc(bufferSize)
        
        var numBytesDecrypted:Int = 0
        
        let bytes = [UInt8](data)
        
        let cryptStatus:CCCryptorStatus = CCCrypt(CCOperation(kCCDecrypt), CCAlgorithm(kCCAlgorithmAES), CCOptions(kCCOptionPKCS7Padding | kCCOptionECBMode) , keyPtr, kCCKeySizeAES128, nil, bytes, dataLength, buffer, bufferSize, &numBytesDecrypted)
        
        var outData:Data?
        
        if cryptStatus == kCCSuccess {
            
            outData = Data.init(bytes: buffer!, count: numBytesDecrypted)
        }
        
        free(buffer)

        return outData
    }
    
    return nil
}
}

加解密 key ,iv 參數(shù)，加解密模式ECB/CBC 和服務(wù)端協(xié)商