1 // SPDX-License-Identifier: GPL-2.0-or-later 2 /* 3 * Copyright 2025 Google LLC 4 */ 5 #include <crypto/blake2s.h> 6 #include "blake2s-testvecs.h" 7 8 /* 9 * The following are compatibility functions that present BLAKE2s as an unkeyed 10 * hash function that produces hashes of fixed length BLAKE2S_HASH_SIZE, so that 11 * hash-test-template.h can be reused to test it. 12 */ 13 14 static void blake2s_default(const u8 *data, size_t len, 15 u8 out[BLAKE2S_HASH_SIZE]) 16 { 17 blake2s(out, data, NULL, BLAKE2S_HASH_SIZE, len, 0); 18 } 19 20 static void blake2s_init_default(struct blake2s_state *state) 21 { 22 blake2s_init(state, BLAKE2S_HASH_SIZE); 23 } 24 25 /* 26 * Generate the HASH_KUNIT_CASES using hash-test-template.h. These test BLAKE2s 27 * with a key length of 0 and a hash length of BLAKE2S_HASH_SIZE. 28 */ 29 #define HASH blake2s_default 30 #define HASH_CTX blake2s_state 31 #define HASH_SIZE BLAKE2S_HASH_SIZE 32 #define HASH_INIT blake2s_init_default 33 #define HASH_UPDATE blake2s_update 34 #define HASH_FINAL blake2s_final 35 #include "hash-test-template.h" 36 37 /* 38 * BLAKE2s specific test case which tests all possible combinations of key 39 * length and hash length. 40 */ 41 static void test_blake2s_all_key_and_hash_lens(struct kunit *test) 42 { 43 const size_t data_len = 100; 44 u8 *data = &test_buf[0]; 45 u8 *key = data + data_len; 46 u8 *hash = key + BLAKE2S_KEY_SIZE; 47 struct blake2s_state main_state; 48 u8 main_hash[BLAKE2S_HASH_SIZE]; 49 50 rand_bytes_seeded_from_len(data, data_len); 51 blake2s_init(&main_state, BLAKE2S_HASH_SIZE); 52 for (int key_len = 0; key_len <= BLAKE2S_KEY_SIZE; key_len++) { 53 rand_bytes_seeded_from_len(key, key_len); 54 for (int out_len = 1; out_len <= BLAKE2S_HASH_SIZE; out_len++) { 55 blake2s(hash, data, key, out_len, data_len, key_len); 56 blake2s_update(&main_state, hash, out_len); 57 } 58 } 59 blake2s_final(&main_state, main_hash); 60 KUNIT_ASSERT_MEMEQ(test, main_hash, blake2s_keyed_testvec_consolidated, 61 BLAKE2S_HASH_SIZE); 62 } 63 64 /* 65 * BLAKE2s specific test case which tests using a guarded buffer for all allowed 66 * key lengths. Also tests both blake2s() and blake2s_init_key(). 67 */ 68 static void test_blake2s_with_guarded_key_buf(struct kunit *test) 69 { 70 const size_t data_len = 100; 71 72 rand_bytes(test_buf, data_len); 73 for (int key_len = 0; key_len <= BLAKE2S_KEY_SIZE; key_len++) { 74 u8 key[BLAKE2S_KEY_SIZE]; 75 u8 *guarded_key = &test_buf[TEST_BUF_LEN - key_len]; 76 u8 hash1[BLAKE2S_HASH_SIZE]; 77 u8 hash2[BLAKE2S_HASH_SIZE]; 78 struct blake2s_state state; 79 80 rand_bytes(key, key_len); 81 memcpy(guarded_key, key, key_len); 82 83 blake2s(hash1, test_buf, key, 84 BLAKE2S_HASH_SIZE, data_len, key_len); 85 blake2s(hash2, test_buf, guarded_key, 86 BLAKE2S_HASH_SIZE, data_len, key_len); 87 KUNIT_ASSERT_MEMEQ(test, hash1, hash2, BLAKE2S_HASH_SIZE); 88 89 blake2s_init_key(&state, BLAKE2S_HASH_SIZE, 90 guarded_key, key_len); 91 blake2s_update(&state, test_buf, data_len); 92 blake2s_final(&state, hash2); 93 KUNIT_ASSERT_MEMEQ(test, hash1, hash2, BLAKE2S_HASH_SIZE); 94 } 95 } 96 97 /* 98 * BLAKE2s specific test case which tests using a guarded output buffer for all 99 * allowed output lengths. 100 */ 101 static void test_blake2s_with_guarded_out_buf(struct kunit *test) 102 { 103 const size_t data_len = 100; 104 105 rand_bytes(test_buf, data_len); 106 for (int out_len = 1; out_len <= BLAKE2S_HASH_SIZE; out_len++) { 107 u8 hash[BLAKE2S_HASH_SIZE]; 108 u8 *guarded_hash = &test_buf[TEST_BUF_LEN - out_len]; 109 110 blake2s(hash, test_buf, NULL, out_len, data_len, 0); 111 blake2s(guarded_hash, test_buf, NULL, out_len, data_len, 0); 112 KUNIT_ASSERT_MEMEQ(test, hash, guarded_hash, out_len); 113 } 114 } 115 116 static struct kunit_case blake2s_test_cases[] = { 117 HASH_KUNIT_CASES, 118 KUNIT_CASE(test_blake2s_all_key_and_hash_lens), 119 KUNIT_CASE(test_blake2s_with_guarded_key_buf), 120 KUNIT_CASE(test_blake2s_with_guarded_out_buf), 121 KUNIT_CASE(benchmark_hash), 122 {}, 123 }; 124 125 static struct kunit_suite blake2s_test_suite = { 126 .name = "blake2s", 127 .test_cases = blake2s_test_cases, 128 .suite_init = hash_suite_init, 129 .suite_exit = hash_suite_exit, 130 }; 131 kunit_test_suite(blake2s_test_suite); 132 133 MODULE_DESCRIPTION("KUnit tests and benchmark for BLAKE2s"); 134 MODULE_LICENSE("GPL"); 135