import ThreemaEssentials
import XCTest
@testable
import ThreemaFramework
final class ThreemaKDFTests: XCTestCase {
// First three vectors taken from multidevice-kdf/test-vectors-blake2b.csv
static let testVectors: [TestVector] = [
TestVector(
secretKey:
"101692161c717bc3fe893b3dbcfe7424c725fd06624940a1046895fb83960240",
salt:
"492e",
personal:
"a390519d083d07c5",
derived:
"ae810e70c16cc45692c1d4fedf323ca2ca0218d90dc0f969ab1a7aeb6d3039a8"
),
TestVector(
secretKey:
"f8e2fcb4369c164e1cdfff82cb7a2c970f1b9a1553c143bf6aef588c1343c2da",
salt:
"aaf4ad",
personal:
"9f4c909b3b27f8e5",
derived:
"5f316d0be440fc40b60bd1c90aab60f1de6f9e2de57d9d0f24b3a3fa02eda76a"
),
TestVector(
secretKey:
"88f7c68a72c76747494fec2d9783e2948906d86b2458818b7e9ee7fce856cb72",
salt:
"00",
personal:
"ef2a504bdb217992",
derived:
"b4da86a0622262e4f8f8bce44aa6a3ee994b61997c1895af2803a98901914cc0"
),
]
func testDeriveKey() {
for testVector in ThreemaKDFTests.testVectors {
let kdf = ThreemaKDF(personal: testVector.personal)
let derived = kdf.deriveKey(salt: testVector.salt, key: testVector.secretKey)
XCTAssertEqual(derived, testVector.derived)
}
}
func testBlake2bHash() {
// From https://raw.githubusercontent.com/BLAKE2/BLAKE2/master/testvectors/blake2-kat.json
let data = Data(
BytesUtility
.toBytes(hexString:
"000102030405060708090a0b0c0d0e0f")!
)
let expectedDigest = Data(
BytesUtility
.toBytes(
hexString:
"bfbabbef45554ccfa0dc83752a19cc35d5920956b301d558d772282bc867009168e9e98606bb5ba73a385de5749228c925a85019b71f72fe29b3cd37ca52efe6"
)!
)
let digest = ThreemaKDF.hash(input: data, outputLen: .b64)!
XCTAssertEqual(digest, expectedDigest)
}
}
final class TestVector {
let secretKey: Data
let salt: Data
let personal: Data
let derived: Data
init(secretKey: String, salt: String, personal: String, derived: String) {
self.secretKey = Data(BytesUtility.toBytes(hexString: secretKey)!)
self.salt = Data(BytesUtility.toBytes(hexString: salt)!)
self.personal = Data(BytesUtility.toBytes(hexString: personal)!)
self.derived = Data(BytesUtility.toBytes(hexString: derived)!)
}
}
