1 // SPDX-License-Identifier: GPL-2.0-only 2 /* Copyright 2025 Google LLC */ 3 4 #include <test_progs.h> 5 #include "bpf/libbpf_internal.h" 6 7 #define MAX_LEN 4096 8 9 /* Test libbpf_sha256() for all lengths from 0 to MAX_LEN inclusively. */ 10 void test_sha256(void) 11 { 12 /* 13 * The correctness of this value was verified by running this test with 14 * libbpf_sha256() replaced by OpenSSL's SHA256(). 15 */ 16 static const __u8 expected_digest_of_digests[SHA256_DIGEST_LENGTH] = { 17 0x62, 0x30, 0x0e, 0x1d, 0xea, 0x7f, 0xc4, 0x74, 18 0xfd, 0x8e, 0x64, 0x0b, 0xd8, 0x5f, 0xea, 0x04, 19 0xf3, 0xef, 0x77, 0x42, 0xc2, 0x01, 0xb8, 0x90, 20 0x6e, 0x19, 0x91, 0x1b, 0xca, 0xb3, 0x28, 0x42, 21 }; 22 __u64 seed = 0; 23 __u8 *data = NULL, *digests = NULL; 24 __u8 digest_of_digests[SHA256_DIGEST_LENGTH]; 25 size_t i; 26 27 data = malloc(MAX_LEN); 28 if (!ASSERT_OK_PTR(data, "malloc")) 29 goto out; 30 digests = malloc((MAX_LEN + 1) * SHA256_DIGEST_LENGTH); 31 if (!ASSERT_OK_PTR(digests, "malloc")) 32 goto out; 33 34 /* Generate MAX_LEN bytes of "random" data deterministically. */ 35 for (i = 0; i < MAX_LEN; i++) { 36 seed = (seed * 25214903917 + 11) & ((1ULL << 48) - 1); 37 data[i] = (__u8)(seed >> 16); 38 } 39 40 /* Calculate a digest for each length 0 through MAX_LEN inclusively. */ 41 for (i = 0; i <= MAX_LEN; i++) 42 libbpf_sha256(data, i, &digests[i * SHA256_DIGEST_LENGTH]); 43 44 /* Calculate and verify the digest of all the digests. */ 45 libbpf_sha256(digests, (MAX_LEN + 1) * SHA256_DIGEST_LENGTH, 46 digest_of_digests); 47 ASSERT_MEMEQ(digest_of_digests, expected_digest_of_digests, 48 SHA256_DIGEST_LENGTH, "digest_of_digests"); 49 out: 50 free(data); 51 free(digests); 52 } 53