76 lines
2.5 KiB
Swift
76 lines
2.5 KiB
Swift
|
//
|
||
|
|
// CryptoCTRTests.swift
|
||
|
|
// TunnelKitOpenVPNTests
|
||
|
|
//
|
||
|
|
// Created by Davide De Rosa on 12/12/23.
|
||
|
|
// Copyright (c) 2023 Davide De Rosa. All rights reserved.
|
||
|
|
//
|
||
|
|
// https://github.com/passepartoutvpn
|
||
|
|
//
|
||
|
|
// This file is part of TunnelKit.
|
||
|
|
//
|
||
|
|
// TunnelKit is free software: you can redistribute it and/or modify
|
||
|
|
// it under the terms of the GNU General Public License as published by
|
||
|
|
// the Free Software Foundation, either version 3 of the License, or
|
||
|
|
// (at your option) any later version.
|
||
|
|
//
|
||
|
|
// TunnelKit is distributed in the hope that it will be useful,
|
||
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
// GNU General Public License for more details.
|
||
|
|
//
|
||
|
|
// You should have received a copy of the GNU General Public License
|
||
|
|
// along with TunnelKit. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
//
|
||
|
|
|
||
|
|
import XCTest
|
||
|
|
@testable import TunnelKitCore
|
||
|
|
@testable import TunnelKitOpenVPNCore
|
||
|
|
import CTunnelKitCore
|
||
|
|
import CTunnelKitOpenVPNProtocol
|
||
|
|
|
||
|
|
class CryptoCTRTests: XCTestCase {
|
||
|
|
private let cipherKey = ZeroingData(count: 32)
|
||
|
|
|
||
|
|
private let hmacKey = ZeroingData(count: 32)
|
||
|
|
|
||
|
|
private let plainData = Data(hex: "00112233ffddaa")
|
||
|
|
|
||
|
|
func test_givenData_whenEncrypt_thenDecrypts() {
|
||
|
|
let encryptedData: Data
|
||
|
|
var flags = cryptoFlags
|
||
|
|
|
||
|
|
let sut1 = CryptoCTR(cipherName: "aes-128-ctr", digestName: "sha256")
|
||
|
|
sut1.configureEncryption(withCipherKey: cipherKey, hmacKey: hmacKey)
|
||
|
|
do {
|
||
|
|
encryptedData = try sut1.encryptData(plainData, flags: &flags)
|
||
|
|
} catch {
|
||
|
|
XCTFail("Cannot encrypt: \(error)")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
let sut2 = CryptoCTR(cipherName: "aes-128-ctr", digestName: "sha256")
|
||
|
|
sut2.configureDecryption(withCipherKey: cipherKey, hmacKey: hmacKey)
|
||
|
|
do {
|
||
|
|
let returnedData = try sut2.decryptData(encryptedData, flags: &flags)
|
||
|
|
XCTAssertEqual(returnedData, plainData)
|
||
|
|
} catch {
|
||
|
|
XCTFail("Cannot decrypt: \(error)")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private var cryptoFlags: CryptoFlags {
|
||
|
|
let packetId: [UInt8] = [0x56, 0x34, 0x12, 0x00]
|
||
|
|
let ad: [UInt8] = [0x00, 0x12, 0x34, 0x56]
|
||
|
|
return packetId.withUnsafeBufferPointer { iv in
|
||
|
|
ad.withUnsafeBufferPointer { ad in
|
||
|
|
CryptoFlags(iv: iv.baseAddress,
|
||
|
|
ivLength: packetId.count,
|
||
|
|
ad: ad.baseAddress,
|
||
|
|
adLength: ad.count,
|
||
|
|
forTesting: true)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|