1*e7776783SApple OSS Distributions /*
2*e7776783SApple OSS Distributions * Copyright (c) 2019 Apple Inc. All rights reserved.
3*e7776783SApple OSS Distributions *
4*e7776783SApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5*e7776783SApple OSS Distributions *
6*e7776783SApple OSS Distributions * This file contains Original Code and/or Modifications of Original Code
7*e7776783SApple OSS Distributions * as defined in and that are subject to the Apple Public Source License
8*e7776783SApple OSS Distributions * Version 2.0 (the 'License'). You may not use this file except in
9*e7776783SApple OSS Distributions * compliance with the License. The rights granted to you under the License
10*e7776783SApple OSS Distributions * may not be used to create, or enable the creation or redistribution of,
11*e7776783SApple OSS Distributions * unlawful or unlicensed copies of an Apple operating system, or to
12*e7776783SApple OSS Distributions * circumvent, violate, or enable the circumvention or violation of, any
13*e7776783SApple OSS Distributions * terms of an Apple operating system software license agreement.
14*e7776783SApple OSS Distributions *
15*e7776783SApple OSS Distributions * Please obtain a copy of the License at
16*e7776783SApple OSS Distributions * http://www.opensource.apple.com/apsl/ and read it before using this file.
17*e7776783SApple OSS Distributions *
18*e7776783SApple OSS Distributions * The Original Code and all software distributed under the License are
19*e7776783SApple OSS Distributions * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20*e7776783SApple OSS Distributions * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21*e7776783SApple OSS Distributions * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22*e7776783SApple OSS Distributions * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23*e7776783SApple OSS Distributions * Please see the License for the specific language governing rights and
24*e7776783SApple OSS Distributions * limitations under the License.
25*e7776783SApple OSS Distributions *
26*e7776783SApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27*e7776783SApple OSS Distributions */
28*e7776783SApple OSS Distributions
29*e7776783SApple OSS Distributions #include <sys/sysctl.h>
30*e7776783SApple OSS Distributions #include <sys/malloc.h>
31*e7776783SApple OSS Distributions #include <pexpert/pexpert.h>
32*e7776783SApple OSS Distributions #include <kern/kalloc.h>
33*e7776783SApple OSS Distributions #include <kern/percpu.h>
34*e7776783SApple OSS Distributions #include <prng/entropy.h>
35*e7776783SApple OSS Distributions #include <libkern/section_keywords.h>
36*e7776783SApple OSS Distributions
37*e7776783SApple OSS Distributions SYSCTL_NODE(_kern, OID_AUTO, entropy, CTLFLAG_RD, 0, NULL);
38*e7776783SApple OSS Distributions SYSCTL_NODE(_kern_entropy, OID_AUTO, health, CTLFLAG_RD, 0, NULL);
39*e7776783SApple OSS Distributions SYSCTL_INT(_kern_entropy_health, OID_AUTO, startup_done, CTLFLAG_RD, &entropy_health_startup_done, 0, NULL);
40*e7776783SApple OSS Distributions
41*e7776783SApple OSS Distributions SYSCTL_NODE(_kern_entropy_health, OID_AUTO, repetition_count_test, CTLFLAG_RD, 0, NULL);
42*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_health_repetition_count_test, OID_AUTO, reset_count, CTLFLAG_RD, &entropy_health_rct_stats.reset_count, 0, NULL);
43*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_health_repetition_count_test, OID_AUTO, failure_count, CTLFLAG_RD, &entropy_health_rct_stats.failure_count, 0, NULL);
44*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_health_repetition_count_test, OID_AUTO, max_observation_count, CTLFLAG_RD, &entropy_health_rct_stats.max_observation_count, 0, NULL);
45*e7776783SApple OSS Distributions
46*e7776783SApple OSS Distributions SYSCTL_NODE(_kern_entropy_health, OID_AUTO, adaptive_proportion_test, CTLFLAG_RD, 0, NULL);
47*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_health_adaptive_proportion_test, OID_AUTO, reset_count, CTLFLAG_RD, &entropy_health_apt_stats.reset_count, 0, NULL);
48*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_health_adaptive_proportion_test, OID_AUTO, failure_count, CTLFLAG_RD, &entropy_health_apt_stats.failure_count, 0, NULL);
49*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_health_adaptive_proportion_test, OID_AUTO, max_observation_count, CTLFLAG_RD, &entropy_health_apt_stats.max_observation_count, 0, NULL);
50*e7776783SApple OSS Distributions
51*e7776783SApple OSS Distributions SYSCTL_NODE(_kern_entropy, OID_AUTO, filter, CTLFLAG_RD, 0,
52*e7776783SApple OSS Distributions "Entropy filter information");
53*e7776783SApple OSS Distributions SYSCTL_QUAD(_kern_entropy_filter, OID_AUTO, total_sample_count, CTLFLAG_RD, &entropy_filter_total_sample_count,
54*e7776783SApple OSS Distributions "The total number of samples processed (i.e. the number of interrupts)");
55*e7776783SApple OSS Distributions SYSCTL_QUAD(_kern_entropy_filter, OID_AUTO, accepted_sample_count, CTLFLAG_RD, &entropy_filter_accepted_sample_count,
56*e7776783SApple OSS Distributions "The number of samples that passed the filter");
57*e7776783SApple OSS Distributions SYSCTL_QUAD(_kern_entropy_filter, OID_AUTO, rejected_sample_count, CTLFLAG_RD, &entropy_filter_rejected_sample_count,
58*e7776783SApple OSS Distributions "The number of samples that were rejected by the filters");
59*e7776783SApple OSS Distributions
60*e7776783SApple OSS Distributions SYSCTL_NODE(_kern_entropy, OID_AUTO, analysis, CTLFLAG_RD, 0,
61*e7776783SApple OSS Distributions "Subsystem to collect entropy samples for offline analysis");
62*e7776783SApple OSS Distributions
63*e7776783SApple OSS Distributions static int entropy_analysis_supported = ENTROPY_ANALYSIS_SUPPORTED;
64*e7776783SApple OSS Distributions SYSCTL_INT(_kern_entropy_analysis, OID_AUTO, supported, CTLFLAG_RD, &entropy_analysis_supported, 0,
65*e7776783SApple OSS Distributions "1 if the kernel was built with entropy analysis support; 0 otherwise");
66*e7776783SApple OSS Distributions
67*e7776783SApple OSS Distributions #if ENTROPY_ANALYSIS_SUPPORTED
68*e7776783SApple OSS Distributions
69*e7776783SApple OSS Distributions SYSCTL_INT(_kern_entropy_analysis, OID_AUTO, enabled, CTLFLAG_RD, &entropy_analysis_enabled, 0,
70*e7776783SApple OSS Distributions "1 if entropy analysis is enabled (via boot arg); 0 otherwise");
71*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_analysis, OID_AUTO, max_sample_count, CTLFLAG_RD | CTLFLAG_NOAUTO, &entropy_analysis_max_sample_count, 0,
72*e7776783SApple OSS Distributions "Target count of samples to be collected");
73*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_analysis, OID_AUTO, sample_count, CTLFLAG_RD | CTLFLAG_NOAUTO, &entropy_analysis_sample_count, 0,
74*e7776783SApple OSS Distributions "Current count of samples collected");
75*e7776783SApple OSS Distributions
76*e7776783SApple OSS Distributions static int entropy_analysis_sample_size = sizeof(entropy_sample_t);
77*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_analysis, OID_AUTO, sample_size, CTLFLAG_RD | CTLFLAG_NOAUTO, &entropy_analysis_sample_size, 0,
78*e7776783SApple OSS Distributions "Size (in bytes) of a single sample");
79*e7776783SApple OSS Distributions
80*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_analysis, OID_AUTO, buffer_size, CTLFLAG_RD | CTLFLAG_NOAUTO, &entropy_analysis_buffer_size, 0,
81*e7776783SApple OSS Distributions "Size (in bytes) of the buffer of samples");
82*e7776783SApple OSS Distributions
83*e7776783SApple OSS Distributions SYSCTL_UINT(_kern_entropy_analysis, OID_AUTO, filter_size, CTLFLAG_RD | CTLFLAG_NOAUTO, &entropy_analysis_filter_size, 0,
84*e7776783SApple OSS Distributions "Size (in bytes) of the filter bitmap");
85*e7776783SApple OSS Distributions
86*e7776783SApple OSS Distributions static int
87*e7776783SApple OSS Distributions sysctl_entropy_collect SYSCTL_HANDLER_ARGS
88*e7776783SApple OSS Distributions {
89*e7776783SApple OSS Distributions if (req->oldptr == USER_ADDR_NULL) {
90*e7776783SApple OSS Distributions return SYSCTL_OUT(req, NULL, entropy_analysis_buffer_size);
91*e7776783SApple OSS Distributions }
92*e7776783SApple OSS Distributions
93*e7776783SApple OSS Distributions return SYSCTL_OUT(req, entropy_analysis_buffer, entropy_analysis_buffer_size);
94*e7776783SApple OSS Distributions }
95*e7776783SApple OSS Distributions
96*e7776783SApple OSS Distributions SYSCTL_PROC(_kern_entropy_analysis, OID_AUTO, buffer,
97*e7776783SApple OSS Distributions CTLTYPE_OPAQUE | CTLFLAG_RD | CTLFLAG_NOAUTO,
98*e7776783SApple OSS Distributions NULL, 0, sysctl_entropy_collect, "-",
99*e7776783SApple OSS Distributions "The buffer of samples");
100*e7776783SApple OSS Distributions
101*e7776783SApple OSS Distributions static int
102*e7776783SApple OSS Distributions sysctl_entropy_filter SYSCTL_HANDLER_ARGS
103*e7776783SApple OSS Distributions {
104*e7776783SApple OSS Distributions if (req->oldptr == USER_ADDR_NULL) {
105*e7776783SApple OSS Distributions return SYSCTL_OUT(req, NULL, entropy_analysis_filter_size);
106*e7776783SApple OSS Distributions }
107*e7776783SApple OSS Distributions
108*e7776783SApple OSS Distributions // There is one bit in the bitmap for each sample.
109*e7776783SApple OSS Distributions unsigned filter_nbits = entropy_analysis_max_sample_count;
110*e7776783SApple OSS Distributions
111*e7776783SApple OSS Distributions bitmap_t *filter = bitmap_alloc(filter_nbits);
112*e7776783SApple OSS Distributions if (!filter) {
113*e7776783SApple OSS Distributions return ENOMEM;
114*e7776783SApple OSS Distributions }
115*e7776783SApple OSS Distributions
116*e7776783SApple OSS Distributions entropy_filter(entropy_analysis_max_sample_count, entropy_analysis_buffer, BITMAP_LEN(filter_nbits), filter);
117*e7776783SApple OSS Distributions
118*e7776783SApple OSS Distributions int status = SYSCTL_OUT(req, filter, entropy_analysis_filter_size);
119*e7776783SApple OSS Distributions
120*e7776783SApple OSS Distributions bitmap_free(filter, filter_nbits);
121*e7776783SApple OSS Distributions return status;
122*e7776783SApple OSS Distributions }
123*e7776783SApple OSS Distributions
124*e7776783SApple OSS Distributions SYSCTL_PROC(_kern_entropy_analysis, OID_AUTO, filter,
125*e7776783SApple OSS Distributions CTLTYPE_OPAQUE | CTLFLAG_RD | CTLFLAG_NOAUTO,
126*e7776783SApple OSS Distributions NULL, 0, sysctl_entropy_filter, "-",
127*e7776783SApple OSS Distributions "The bitmap of filtered samples");
128*e7776783SApple OSS Distributions
129*e7776783SApple OSS Distributions __startup_func
130*e7776783SApple OSS Distributions static void
entropy_analysis_sysctl_startup(void)131*e7776783SApple OSS Distributions entropy_analysis_sysctl_startup(void)
132*e7776783SApple OSS Distributions {
133*e7776783SApple OSS Distributions uint32_t sample_count = 0;
134*e7776783SApple OSS Distributions if (__improbable(PE_parse_boot_argn(ENTROPY_ANALYSIS_BOOTARG, &sample_count, sizeof(sample_count)))) {
135*e7776783SApple OSS Distributions sysctl_register_oid_early(&sysctl__kern_entropy_analysis_max_sample_count);
136*e7776783SApple OSS Distributions sysctl_register_oid_early(&sysctl__kern_entropy_analysis_sample_count);
137*e7776783SApple OSS Distributions sysctl_register_oid_early(&sysctl__kern_entropy_analysis_sample_size);
138*e7776783SApple OSS Distributions sysctl_register_oid_early(&sysctl__kern_entropy_analysis_buffer_size);
139*e7776783SApple OSS Distributions sysctl_register_oid_early(&sysctl__kern_entropy_analysis_buffer);
140*e7776783SApple OSS Distributions sysctl_register_oid_early(&sysctl__kern_entropy_analysis_filter_size);
141*e7776783SApple OSS Distributions sysctl_register_oid_early(&sysctl__kern_entropy_analysis_filter);
142*e7776783SApple OSS Distributions }
143*e7776783SApple OSS Distributions }
144*e7776783SApple OSS Distributions STARTUP(SYSCTL, STARTUP_RANK_MIDDLE, entropy_analysis_sysctl_startup);
145*e7776783SApple OSS Distributions
146*e7776783SApple OSS Distributions #endif // ENTROPY_ANALYSIS_SUPPORTED
147