88 lines
3.1 KiB
Swift
88 lines
3.1 KiB
Swift
//
|
|
// XORTests.swift
|
|
// TunnelKitOpenVPNTests
|
|
//
|
|
// Created by Davide De Rosa on 11/4/22.
|
|
// Copyright (c) 2022 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
|
|
import TunnelKitCore
|
|
import TunnelKitOpenVPNProtocol
|
|
import CTunnelKitOpenVPNProtocol
|
|
|
|
final class XORTests: XCTestCase {
|
|
private let mask = Data(hex: "f76dab30")
|
|
|
|
override func setUpWithError() throws {
|
|
// Put setup code here. This method is called before the invocation of each test method in the class.
|
|
}
|
|
|
|
override func tearDownWithError() throws {
|
|
// Put teardown code here. This method is called after the invocation of each test method in the class.
|
|
}
|
|
|
|
func testMask() throws {
|
|
let processor = XORProcessor(method: .xormask(mask: mask))
|
|
processor.assertReversible(try SecureRandom.data(length: 1000))
|
|
}
|
|
|
|
func testPtrPos() throws {
|
|
let processor = XORProcessor(method: .xorptrpos)
|
|
processor.assertReversible(try SecureRandom.data(length: 1000))
|
|
}
|
|
|
|
func testReverse() throws {
|
|
let processor = XORProcessor(method: .reverse)
|
|
processor.assertReversible(try SecureRandom.data(length: 1000))
|
|
}
|
|
|
|
func testObfuscate() throws {
|
|
let processor = XORProcessor(method: .obfuscate(mask: mask))
|
|
processor.assertReversible(try SecureRandom.data(length: 1000))
|
|
}
|
|
|
|
func testPacketStream() throws {
|
|
let data = try SecureRandom.data(length: 10000)
|
|
PacketStream.assertReversible(data, method: .none)
|
|
PacketStream.assertReversible(data, method: .mask, mask: mask)
|
|
PacketStream.assertReversible(data, method: .ptrPos)
|
|
PacketStream.assertReversible(data, method: .reverse)
|
|
PacketStream.assertReversible(data, method: .obfuscate, mask: mask)
|
|
}
|
|
}
|
|
|
|
private extension XORProcessor {
|
|
func assertReversible(_ data: Data) {
|
|
let xored = processPacket(data, outbound: true)
|
|
XCTAssertEqual(processPacket(xored, outbound: false), data)
|
|
}
|
|
}
|
|
|
|
private extension PacketStream {
|
|
static func assertReversible(_ data: Data, method: XORMethodNative, mask: Data? = nil) {
|
|
var until = 0
|
|
let outStream = PacketStream.outboundStream(fromPacket: data, xorMethod: method, xorMask: mask)
|
|
let inStream = PacketStream.packets(fromInboundStream: outStream, until: &until, xorMethod: method, xorMask: mask)
|
|
let originalData = Data(inStream.joined())
|
|
XCTAssertEqual(data.toHex(), originalData.toHex())
|
|
}
|
|
}
|