123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280 |
- //
- // Access.swift
- // CryptoSwift
- //
- // Created by Marcin Krzyzanowski on 06/08/16.
- // Copyright © 2016 Marcin Krzyzanowski. All rights reserved.
- //
- import XCTest
- import Foundation
- // import without @testable to test public API
- import CryptoSwift
- class Access: XCTestCase {
- let cipher = try! AES(key: "secret0key000000", iv: "0123456789012345")
- let authenticator = HMAC(key: Array<UInt8>(hex: "b1b2b3b3b3b3b3b3b1b2b3b3b3b3b3b3"), variant: .sha1)
- func testChecksum() {
- let _ = Checksum.crc32([1, 2, 3])
- let _ = Checksum.crc16([1, 2, 3])
- }
- func testRandomIV() {
- let _ = AES.randomIV(AES.blockSize)
- let _ = ChaCha20.randomIV(ChaCha20.blockSize)
- }
- func testDigest() {
- let _ = Digest.md5([1, 2, 3])
- let _ = Digest.sha1([1, 2, 3])
- let _ = Digest.sha224([1, 2, 3])
- let _ = Digest.sha256([1, 2, 3])
- let _ = Digest.sha384([1, 2, 3])
- let _ = Digest.sha512([1, 2, 3])
- let _ = Digest.sha3([1, 2, 3], variant: .sha224)
- let _ = SHA1().calculate(for: [0])
- let _ = SHA2(variant: .sha224).calculate(for: [0])
- let _ = SHA3(variant: .sha256).calculate(for: [0])
- let _ = MD5().calculate(for: [0])
- }
- func testArrayExtension() {
- let array = Array<UInt8>(hex: "b1b2b3b3b3b3b3b3b1b2b3b3b3b3b3b3")
- let _ = array.toHexString()
- let _ = array.md5()
- let _ = array.sha1()
- let _ = array.sha256()
- let _ = array.sha384()
- let _ = array.sha512()
- let _ = array.sha2(.sha224)
- let _ = array.sha3(.sha224)
- let _ = array.crc32()
- let _ = array.crc16()
- do {
- let _ = try array.encrypt(cipher: cipher)
- let _ = try array.decrypt(cipher: cipher)
- let _ = try array.authenticate(with: authenticator)
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testCollectionExtension() {
- // nothing public
- }
- func testStringExtension() {
- let string = "foofoobarbar"
- let _ = string.md5()
- let _ = string.sha1()
- let _ = string.sha224()
- let _ = string.sha256()
- let _ = string.sha384()
- let _ = string.sha512()
- let _ = string.sha3(.sha224)
- let _ = string.crc16()
- let _ = string.crc32()
- do {
- let _ = try string.encrypt(cipher: cipher)
- let _ = try string.authenticate(with: authenticator)
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testStringFoundationExtension() {
- let string = "aPf/i9th9iX+vf49eR7PYk2q7S5xmm3jkRLejgzHNJs="
- do {
- let _ = try string.decryptBase64ToString(cipher: cipher)
- let _ = try string.decryptBase64(cipher: cipher)
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testIntExtension() {
- // nothing public
- }
- func testUInt16Extension() {
- // nothing public
- }
- func testUInt32Extension() {
- // nothing public
- }
- func testUInt64Extension() {
- // nothing public
- }
- func testUInt8Extension() {
- // nothing public
- }
- func testDataExtension() {
- let data = Data(bytes: [1, 2, 3, 4, 5, 6, 7, 8, 1, 2, 3, 4, 5, 6, 7, 8])
- let _ = data.checksum()
- let _ = data.md5()
- let _ = data.sha1()
- let _ = data.sha224()
- let _ = data.sha256()
- let _ = data.sha384()
- let _ = data.sha512()
- let _ = data.sha3(.sha224)
- let _ = data.crc16()
- let _ = data.crc32()
- let _ = data.bytes
- let _ = data.toHexString()
- do {
- let _ = try data.encrypt(cipher: cipher)
- let _ = try data.decrypt(cipher: cipher)
- let _ = try data.authenticate(with: authenticator)
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testPadding() {
- // PKCS7
- let _ = PKCS7().add(to: [1, 2, 3], blockSize: 16)
- let _ = PKCS7().remove(from: [1, 2, 3], blockSize: 16)
- // NoPadding
- let _ = NoPadding().add(to: [1, 2, 3], blockSize: 16)
- let _ = NoPadding().remove(from: [1, 2, 3], blockSize: 16)
- // ZeroPadding
- let _ = ZeroPadding().add(to: [1, 2, 3], blockSize: 16)
- let _ = ZeroPadding().remove(from: [1, 2, 3], blockSize: 16)
- }
- func testPBKDF() {
- do {
- let _ = PKCS5.PBKDF1.Variant.md5
- let _ = try PKCS5.PBKDF1(password: [1, 2, 3, 4, 5, 6, 7], salt: [1, 2, 3, 4, 5, 6, 7, 8]).calculate()
- let _ = try PKCS5.PBKDF2(password: [1, 2, 3, 4, 5, 6, 7], salt: [1, 2, 3, 4]).calculate()
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testAuthenticators() {
- do {
- let _ = try HMAC(key: Array<UInt8>(hex: "b1b2b3b3b3b3b3b3b1b2b3b3b3b3b3b3"), variant: .sha1).authenticate([1, 2, 3])
- let _ = try Poly1305(key: [1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2]).authenticate([1, 2, 3])
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testAES() {
- do {
- let cipher = try AES(key: "secret0key000000", iv: "0123456789012345")
- var encryptor = cipher.makeEncryptor()
- let _ = try encryptor.update(withBytes: [1, 2, 3])
- let _ = try encryptor.finish()
- var decryptor = cipher.makeDecryptor()
- let _ = try decryptor.update(withBytes: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16])
- let _ = try decryptor.finish()
- let enc = try cipher.encrypt([1, 2, 3])
- let _ = try cipher.decrypt(enc)
- let _ = try AES(key: [1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6], iv: [1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6], blockMode: .CBC, padding: NoPadding())
- let _ = AES.Variant.aes128
- let _ = AES.blockSize
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testBlowfish() {
- do {
- let cipher = try Blowfish(key: "secret0key000000", iv: "01234567")
- let enc = try cipher.encrypt([1, 2, 3])
- let _ = try cipher.decrypt(enc)
- let _ = try Blowfish(key: [1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6], iv: [1, 2, 3, 4, 5, 6, 7, 8], blockMode: .CBC, padding: NoPadding())
- let _ = Blowfish.blockSize
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testRabbit() {
- do {
- let rabbit = try Rabbit(key: [0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00])
- let enc = rabbit.encrypt([1, 2, 3])
- let _ = rabbit.decrypt(enc)
- XCTAssertThrowsError(try Rabbit(key: "123"))
- let _ = Rabbit.blockSize
- let _ = Rabbit.keySize
- let _ = Rabbit.ivSize
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testChaCha20() {
- let key: Array<UInt8> = [0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00]
- let iv: Array<UInt8> = [0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00]
- do {
- let _ = ChaCha20.blockSize
- let chacha20 = try ChaCha20(key: key, iv: iv)
- let enc = try chacha20.encrypt([1, 3, 4])
- let _ = try chacha20.decrypt(enc)
- XCTAssertThrowsError(try ChaCha20(key: "123", iv: "12345678"))
- let _ = chacha20.makeEncryptor()
- let _ = chacha20.makeDecryptor()
- } catch {
- XCTFail(error.localizedDescription)
- }
- }
- func testUpdatable() {
- // TODO:
- }
- static let allTests = [
- ("testChecksum", testChecksum),
- ("testDigest", testDigest),
- ("testArrayExtension", testArrayExtension),
- ("testCollectionExtension", testCollectionExtension),
- ("testStringExtension", testStringExtension),
- ("testStringFoundationExtension", testStringFoundationExtension),
- ("testIntExtension", testIntExtension),
- ("testUInt16Extension", testUInt16Extension),
- ("testUInt32Extension", testUInt32Extension),
- ("testUInt64Extension", testUInt64Extension),
- ("testUInt8Extension", testUInt8Extension),
- ("testDataExtension", testDataExtension),
- ("testPadding", testPadding),
- ("testPBKDF", testPBKDF),
- ("testAuthenticators", testAuthenticators),
- ("testAES", testAES),
- ("testBlowfish", testBlowfish),
- ("testRabbit", testRabbit),
- ("testChaCha20", testChaCha20),
- ("testUpdatable", testUpdatable),
- ("testRandomIV", testRandomIV),
- ]
- }
|