| 1 | // SPDX-License-Identifier: GPL-2.0 |
|---|
| 2 | #include <kunit/test.h> |
|---|
| 3 | |
|---|
| 4 | #include "protocol.h" |
|---|
| 5 | |
|---|
| 6 | struct test_case { |
|---|
| 7 | char *key; |
|---|
| 8 | char *msg; |
|---|
| 9 | char *result; |
|---|
| 10 | }; |
|---|
| 11 | |
|---|
| 12 | /* we can't reuse RFC 4231 test vectors, as we have constraint on the |
|---|
| 13 | * input and key size. |
|---|
| 14 | */ |
|---|
| 15 | static struct test_case tests[] = { |
|---|
| 16 | { |
|---|
| 17 | .key = "0b0b0b0b0b0b0b0b" , |
|---|
| 18 | .msg = "48692054" , |
|---|
| 19 | .result = "8385e24fb4235ac37556b6b886db106284a1da671699f46db1f235ec622dcafa" , |
|---|
| 20 | }, |
|---|
| 21 | { |
|---|
| 22 | .key = "aaaaaaaaaaaaaaaa" , |
|---|
| 23 | .msg = "dddddddd" , |
|---|
| 24 | .result = "2c5e219164ff1dca1c4a92318d847bb6b9d44492984e1eb71aff9022f71046e9" , |
|---|
| 25 | }, |
|---|
| 26 | { |
|---|
| 27 | .key = "0102030405060708" , |
|---|
| 28 | .msg = "cdcdcdcd" , |
|---|
| 29 | .result = "e73b9ba9969969cefb04aa0d6df18ec2fcc075b6f23b4d8c4da736a5dbbc6e7d" , |
|---|
| 30 | }, |
|---|
| 31 | }; |
|---|
| 32 | |
|---|
| 33 | static void mptcp_crypto_test_basic(struct kunit *test) |
|---|
| 34 | { |
|---|
| 35 | char hmac[32], hmac_hex[65]; |
|---|
| 36 | u32 nonce1, nonce2; |
|---|
| 37 | u64 key1, key2; |
|---|
| 38 | u8 msg[8]; |
|---|
| 39 | int i, j; |
|---|
| 40 | |
|---|
| 41 | for (i = 0; i < ARRAY_SIZE(tests); ++i) { |
|---|
| 42 | /* mptcp hmap will convert to be before computing the hmac */ |
|---|
| 43 | key1 = be64_to_cpu(*((__be64 *)&tests[i].key[0])); |
|---|
| 44 | key2 = be64_to_cpu(*((__be64 *)&tests[i].key[8])); |
|---|
| 45 | nonce1 = be32_to_cpu(*((__be32 *)&tests[i].msg[0])); |
|---|
| 46 | nonce2 = be32_to_cpu(*((__be32 *)&tests[i].msg[4])); |
|---|
| 47 | |
|---|
| 48 | put_unaligned_be32(val: nonce1, p: &msg[0]); |
|---|
| 49 | put_unaligned_be32(val: nonce2, p: &msg[4]); |
|---|
| 50 | |
|---|
| 51 | mptcp_crypto_hmac_sha(key1, key2, msg, len: 8, hmac); |
|---|
| 52 | for (j = 0; j < 32; ++j) |
|---|
| 53 | sprintf(buf: &hmac_hex[j << 1], fmt: "%02x" , hmac[j] & 0xff); |
|---|
| 54 | hmac_hex[64] = 0; |
|---|
| 55 | |
|---|
| 56 | KUNIT_EXPECT_STREQ(test, &hmac_hex[0], tests[i].result); |
|---|
| 57 | } |
|---|
| 58 | } |
|---|
| 59 | |
|---|
| 60 | static struct kunit_case mptcp_crypto_test_cases[] = { |
|---|
| 61 | KUNIT_CASE(mptcp_crypto_test_basic), |
|---|
| 62 | {} |
|---|
| 63 | }; |
|---|
| 64 | |
|---|
| 65 | static struct kunit_suite mptcp_crypto_suite = { |
|---|
| 66 | .name = "mptcp-crypto" , |
|---|
| 67 | .test_cases = mptcp_crypto_test_cases, |
|---|
| 68 | }; |
|---|
| 69 | |
|---|
| 70 | kunit_test_suite(mptcp_crypto_suite); |
|---|
| 71 | |
|---|
| 72 | MODULE_LICENSE("GPL" ); |
|---|
| 73 | MODULE_DESCRIPTION("KUnit tests for MPTCP Crypto" ); |
|---|
| 74 | |
|---|
