xref: /xnu-11417.140.69/tests/entropy.c (revision 43a90889846e00bfb5cf1d255cdc0a701a1e05a4)
1*43a90889SApple OSS Distributions #include <stdlib.h>
2*43a90889SApple OSS Distributions #include <sys/sysctl.h>
3*43a90889SApple OSS Distributions #include <darwintest.h>
4*43a90889SApple OSS Distributions #include <perfdata/perfdata.h>
5*43a90889SApple OSS Distributions 
6*43a90889SApple OSS Distributions typedef uint32_t entropy_sample_t;
7*43a90889SApple OSS Distributions 
8*43a90889SApple OSS Distributions T_GLOBAL_META(T_META_NAMESPACE("xnu.crypto"),
9*43a90889SApple OSS Distributions     T_META_RADAR_COMPONENT_NAME("xnu"),
10*43a90889SApple OSS Distributions     T_META_RADAR_COMPONENT_VERSION("crypto"));
11*43a90889SApple OSS Distributions 
12*43a90889SApple OSS Distributions T_DECL(entropy_collect, "Collect entropy for offline analysis",
13*43a90889SApple OSS Distributions     T_META_REQUIRES_SYSCTL_EQ("kern.development", 1),
14*43a90889SApple OSS Distributions     T_META_BOOTARGS_SET("entropy-analysis-sample-count=1000"))
15*43a90889SApple OSS Distributions {
16*43a90889SApple OSS Distributions 	int ret;
17*43a90889SApple OSS Distributions 	uint32_t entropy_size = 0;
18*43a90889SApple OSS Distributions 	size_t size = sizeof(entropy_size);
19*43a90889SApple OSS Distributions 
20*43a90889SApple OSS Distributions 	ret = sysctlbyname("kern.entropy.analysis.buffer_size", &entropy_size, &size, NULL, 0);
21*43a90889SApple OSS Distributions 	T_ASSERT_POSIX_SUCCESS(ret, "sysctlbyname kern.entropy.analysis.buffer_size");
22*43a90889SApple OSS Distributions 
23*43a90889SApple OSS Distributions 	uint32_t entropy_count = entropy_size / sizeof(entropy_sample_t);
24*43a90889SApple OSS Distributions 	entropy_sample_t *entropy = calloc(entropy_count, sizeof(entropy_sample_t));
25*43a90889SApple OSS Distributions 	size = entropy_size;
26*43a90889SApple OSS Distributions 
27*43a90889SApple OSS Distributions 	ret = sysctlbyname("kern.entropy.analysis.buffer", entropy, &size, NULL, 0);
28*43a90889SApple OSS Distributions 	T_ASSERT_POSIX_SUCCESS(ret, "sysctlbyname kern.entropy.analysis.buffer");
29*43a90889SApple OSS Distributions 
30*43a90889SApple OSS Distributions 	// This test is not an entropy assessment. We're just checking to
31*43a90889SApple OSS Distributions 	// make sure the machinery of the entropy collection sysctl seems
32*43a90889SApple OSS Distributions 	// to be working.
33*43a90889SApple OSS Distributions 	for (uint32_t i = 0; i < entropy_count; i += 1) {
34*43a90889SApple OSS Distributions 		T_QUIET; T_EXPECT_NE(entropy[i], 0, "entropy buffer null sample %u", i);
35*43a90889SApple OSS Distributions 	}
36*43a90889SApple OSS Distributions 
37*43a90889SApple OSS Distributions 	free(entropy);
38*43a90889SApple OSS Distributions }
39*43a90889SApple OSS Distributions 
40*43a90889SApple OSS Distributions T_DECL(entropy_filter_rate, "Sample entropy filter rate")
41*43a90889SApple OSS Distributions {
42*43a90889SApple OSS Distributions 	int ret;
43*43a90889SApple OSS Distributions 	uint64_t total_sample_count = 0;
44*43a90889SApple OSS Distributions 	uint64_t rejected_sample_count = 0;
45*43a90889SApple OSS Distributions 	size_t size = sizeof(total_sample_count);
46*43a90889SApple OSS Distributions 
47*43a90889SApple OSS Distributions 	ret = sysctlbyname("kern.entropy.filter.total_sample_count", &total_sample_count, &size, NULL, 0);
48*43a90889SApple OSS Distributions 	T_ASSERT_POSIX_SUCCESS(ret, "kern.entropy.filter.total_sample_count");
49*43a90889SApple OSS Distributions 
50*43a90889SApple OSS Distributions 	size = sizeof(rejected_sample_count);
51*43a90889SApple OSS Distributions 	ret = sysctlbyname("kern.entropy.filter.rejected_sample_count", &rejected_sample_count, &size, NULL, 0);
52*43a90889SApple OSS Distributions 	T_ASSERT_POSIX_SUCCESS(ret, "kern.entropy.filter.rejected_sample_count");
53*43a90889SApple OSS Distributions 
54*43a90889SApple OSS Distributions 	double rejection_rate = (double) rejected_sample_count / (double) total_sample_count;
55*43a90889SApple OSS Distributions 
56*43a90889SApple OSS Distributions 	pdwriter_t writer = pdwriter_open_tmp("xnu", "entropy_filter_rate", 0, 0, NULL, 0);
57*43a90889SApple OSS Distributions 	T_ASSERT_NOTNULL(writer, "pdwriter_open_tmp");
58*43a90889SApple OSS Distributions 
59*43a90889SApple OSS Distributions 	pdwriter_new_value(writer, "Rejection Rate", PDUNIT_CUSTOM(rejectrate), rejection_rate);
60*43a90889SApple OSS Distributions 
61*43a90889SApple OSS Distributions 	pdwriter_close(writer);
62*43a90889SApple OSS Distributions }
63