mirror of
https://github.com/Ed94/Odin.git
synced 2026-07-31 20:00:09 +00:00
core/crypto/hkdf: Initial import
This commit is contained in:
@@ -5,15 +5,82 @@ import "core:fmt"
|
||||
import "core:testing"
|
||||
|
||||
import "core:crypto/hash"
|
||||
import "core:crypto/hkdf"
|
||||
import "core:crypto/pbkdf2"
|
||||
|
||||
@(test)
|
||||
test_kdf :: proc(t: ^testing.T) {
|
||||
log(t, "Testing KDFs")
|
||||
|
||||
test_hkdf(t)
|
||||
test_pbkdf2(t)
|
||||
}
|
||||
|
||||
@(test)
|
||||
test_hkdf :: proc(t: ^testing.T) {
|
||||
log(t, "Testing HKDF")
|
||||
|
||||
tmp: [128]byte // Good enough.
|
||||
|
||||
test_vectors := []struct {
|
||||
algo: hash.Algorithm,
|
||||
ikm: string,
|
||||
salt: string,
|
||||
info: string,
|
||||
okm: string,
|
||||
} {
|
||||
// SHA-256
|
||||
// - https://www.rfc-editor.org/rfc/rfc5869
|
||||
{
|
||||
hash.Algorithm.SHA256,
|
||||
"0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b",
|
||||
"000102030405060708090a0b0c",
|
||||
"f0f1f2f3f4f5f6f7f8f9",
|
||||
"3cb25f25faacd57a90434f64d0362f2a2d2d0a90cf1a5a4c5db02d56ecc4c5bf34007208d5b887185865",
|
||||
},
|
||||
{
|
||||
hash.Algorithm.SHA256,
|
||||
"000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f404142434445464748494a4b4c4d4e4f",
|
||||
"606162636465666768696a6b6c6d6e6f707172737475767778797a7b7c7d7e7f808182838485868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fa0a1a2a3a4a5a6a7a8a9aaabacadaeaf",
|
||||
"b0b1b2b3b4b5b6b7b8b9babbbcbdbebfc0c1c2c3c4c5c6c7c8c9cacbcccdcecfd0d1d2d3d4d5d6d7d8d9dadbdcdddedfe0e1e2e3e4e5e6e7e8e9eaebecedeeeff0f1f2f3f4f5f6f7f8f9fafbfcfdfeff",
|
||||
"b11e398dc80327a1c8e7f78c596a49344f012eda2d4efad8a050cc4c19afa97c59045a99cac7827271cb41c65e590e09da3275600c2f09b8367793a9aca3db71cc30c58179ec3e87c14c01d5c1f3434f1d87",
|
||||
},
|
||||
{
|
||||
hash.Algorithm.SHA256,
|
||||
"0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b",
|
||||
"",
|
||||
"",
|
||||
"8da4e775a563c18f715f802a063c5a31b8a11f5c5ee1879ec3454e5f3c738d2d9d201395faa4b61a96c8",
|
||||
},
|
||||
}
|
||||
for v, _ in test_vectors {
|
||||
algo_name := hash.ALGORITHM_NAMES[v.algo]
|
||||
dst := tmp[:len(v.okm) / 2]
|
||||
|
||||
ikm, _ := hex.decode(transmute([]byte)(v.ikm), context.temp_allocator)
|
||||
salt, _ := hex.decode(transmute([]byte)(v.salt), context.temp_allocator)
|
||||
info, _ := hex.decode(transmute([]byte)(v.info), context.temp_allocator)
|
||||
|
||||
hkdf.extract_and_expand(v.algo, salt, ikm, info, dst)
|
||||
|
||||
dst_str := string(hex.encode(dst, context.temp_allocator))
|
||||
|
||||
expect(
|
||||
t,
|
||||
dst_str == v.okm,
|
||||
fmt.tprintf(
|
||||
"HKDF-%s: Expected: %s for input of (%s, %s, %s), but got %s instead",
|
||||
algo_name,
|
||||
v.okm,
|
||||
v.ikm,
|
||||
v.salt,
|
||||
v.info,
|
||||
dst_str,
|
||||
),
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
@(test)
|
||||
test_pbkdf2 :: proc(t: ^testing.T) {
|
||||
log(t, "Testing PBKDF2")
|
||||
|
||||
Reference in New Issue
Block a user