hc
2024-03-25 edb30157bad0c0001c32b854271ace01d3b9a16a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
/** @file
 
  This driver produces an EFI_RNG_PROTOCOL instance for the Broadcom 2838 RNG
 
  Copyright (C) 2019, Pete Batard <pete@akeo.ie>
  Copyright (C) 2019, Linaro Ltd. All rights reserved.<BR>
 
  SPDX-License-Identifier: BSD-2-Clause-Patent
 
**/
 
#include <Library/BaseLib.h>
#include <Library/BaseMemoryLib.h>
#include <Library/DebugLib.h>
#include <Library/IoLib.h>
#include <Library/PcdLib.h>
#include <Library/TimerLib.h>
#include <Library/UefiBootServicesTableLib.h>
 
#include <IndustryStandard/Bcm2838Rng.h>
 
#include <Protocol/Rng.h>
 
#define RNG_WARMUP_COUNT        0x40000
#define RNG_MAX_RETRIES         0x100         // arbitrary upper bound
 
/**
  Returns information about the random number generation implementation.
 
  @param[in]     This                 A pointer to the EFI_RNG_PROTOCOL
                                      instance.
  @param[in,out] RNGAlgorithmListSize On input, the size in bytes of
                                      RNGAlgorithmList.
                                      On output with a return code of
                                      EFI_SUCCESS, the size in bytes of the
                                      data returned in RNGAlgorithmList. On
                                      output with a return code of
                                      EFI_BUFFER_TOO_SMALL, the size of
                                      RNGAlgorithmList required to obtain the
                                      list.
  @param[out] RNGAlgorithmList        A caller-allocated memory buffer filled
                                      by the driver with one EFI_RNG_ALGORITHM
                                      element for each supported RNG algorithm.
                                      The list must not change across multiple
                                      calls to the same driver. The first
                                      algorithm in the list is the default
                                      algorithm for the driver.
 
  @retval EFI_SUCCESS                 The RNG algorithm list was returned
                                      successfully.
  @retval EFI_UNSUPPORTED             The services is not supported by this
                                      driver.
  @retval EFI_DEVICE_ERROR            The list of algorithms could not be
                                      retrieved due to a hardware or firmware
                                      error.
  @retval EFI_INVALID_PARAMETER       One or more of the parameters are
                                      incorrect.
  @retval EFI_BUFFER_TOO_SMALL        The buffer RNGAlgorithmList is too small
                                      to hold the result.
 
**/
STATIC
EFI_STATUS
EFIAPI
Bcm2838RngGetInfo (
  IN      EFI_RNG_PROTOCOL        *This,
  IN OUT  UINTN                   *RNGAlgorithmListSize,
  OUT     EFI_RNG_ALGORITHM       *RNGAlgorithmList
  )
{
  if (This == NULL || RNGAlgorithmListSize == NULL) {
    return EFI_INVALID_PARAMETER;
  }
 
  if (*RNGAlgorithmListSize < sizeof (EFI_RNG_ALGORITHM)) {
    *RNGAlgorithmListSize = sizeof (EFI_RNG_ALGORITHM);
    return EFI_BUFFER_TOO_SMALL;
  }
 
  if (RNGAlgorithmList == NULL) {
    return EFI_INVALID_PARAMETER;
  }
 
  *RNGAlgorithmListSize = sizeof (EFI_RNG_ALGORITHM);
  CopyGuid (RNGAlgorithmList, &gEfiRngAlgorithmRaw);
 
  return EFI_SUCCESS;
}
 
/**
  Read a single random value, in either FIFO or regular mode.
 
  @param[in]  Val                     A pointer to the 32-bit word that is to
                                      be filled with a random value.
 
  @retval EFI_SUCCESS                 A random value was successfully read.
  @retval EFI_NOT_READY               The number of retries elapsed before a
                                      random value was generated.
 
**/
STATIC
EFI_STATUS
EFIAPI
Bcm2838RngReadValue (
  IN OUT  UINT32                  *Val
)
{
  UINT32 Avail;
  UINT32 i;
 
  ASSERT (Val != NULL);
 
  Avail = MmioRead32 (RNG_FIFO_COUNT) & RNG_FIFO_DATA_AVAIL_MASK;
 
  //
  // If we don't have a value ready, wait 1 us and retry.
  //
  // Empirical testing on the platform this driver is designed to be used
  // with shows that, unless you set a large divisor for the sample rate,
  // random bits should be generated around the MHz frequency.
  // Therefore a retry that doesn't expire until at least RNG_MAX_RETRIES
  // microseconds should give us ample time to obtain a value. Besides,
  // even outside of calling MicroSecondDelay (), we expect MMIO reads to
  // be slow anyway...
  //
  // On the other hand, we may run into a timeout here if the warmup period
  // has not been completed since the RNG locks RNG_FIFO_COUNT to zero
  // until then. However, with the values we use for the target platform,
  // (RPi4) you'd need to start requesting random data within the first
  // 250 to 500 ms after driver instantiation for this to happen.
  //
  for (i = 0; Avail < 1 && i < RNG_MAX_RETRIES; i++) {
    MicroSecondDelay (1);
    Avail = MmioRead32 (RNG_FIFO_COUNT) & RNG_FIFO_DATA_AVAIL_MASK;
  }
  if (Avail < 1) {
    return EFI_NOT_READY;
  }
 
  *Val = MmioRead32 (RNG_FIFO_DATA);
 
  return EFI_SUCCESS;
}
 
/**
  Produces and returns an RNG value using either the default or specified RNG
  algorithm.
 
  @param[in]  This                    A pointer to the EFI_RNG_PROTOCOL
                                      instance.
  @param[in]  RNGAlgorithm            A pointer to the EFI_RNG_ALGORITHM that
                                      identifies the RNG algorithm to use. May
                                      be NULL in which case the function will
                                      use its default RNG algorithm.
  @param[in]  RNGValueLength          The length in bytes of the memory buffer
                                      pointed to by RNGValue. The driver shall
                                      return exactly this numbers of bytes.
  @param[out] RNGValue                A caller-allocated memory buffer filled
                                      by the driver with the resulting RNG
                                      value.
 
  @retval EFI_SUCCESS                 The RNG value was returned successfully.
  @retval EFI_UNSUPPORTED             The algorithm specified by RNGAlgorithm
                                      is not supported by this driver.
  @retval EFI_DEVICE_ERROR            An RNG value could not be retrieved due
                                      to a hardware or firmware error.
  @retval EFI_NOT_READY               There is not enough random data available
                                      to satisfy the length requested by
                                      RNGValueLength.
  @retval EFI_INVALID_PARAMETER       RNGValue is NULL or RNGValueLength is
                                      zero.
 
**/
STATIC
EFI_STATUS
EFIAPI
Bcm2838RngGetRNG (
  IN EFI_RNG_PROTOCOL            *This,
  IN EFI_RNG_ALGORITHM           *RNGAlgorithm, OPTIONAL
  IN UINTN                       RNGValueLength,
  OUT UINT8                      *RNGValue
  )
{
  EFI_STATUS      Status;
  UINT32          Val;
 
  if (This == NULL || RNGValueLength == 0 || RNGValue == NULL) {
    return EFI_INVALID_PARAMETER;
  }
 
  //
  // We only support the raw algorithm, so reject requests for anything else
  //
  if (RNGAlgorithm != NULL &&
      !CompareGuid (RNGAlgorithm, &gEfiRngAlgorithmRaw)) {
    return EFI_UNSUPPORTED;
  }
 
  //
  // The Linux driver from Broadcom checks RNG_BIT_COUNT here to ensure that
  // the warmup threshold has been reached, but our testing shows that this is
  // pointless as RNG_BIT_COUNT is increased even during warmup and, as long
  // as warmup isn't complete, the FIFO reports that no values are available.
  //
  // Also note that RNG_BIT_COUNT doesn't roll over. Once it reaches 0xFFFFFFFF
  // it just stays there...
  //
 
  while (RNGValueLength >= sizeof (UINT32)) {
    Status = Bcm2838RngReadValue (&Val);
    if (EFI_ERROR (Status)) {
      return Status;
    }
    WriteUnaligned32 ((VOID *)RNGValue, Val);
    RNGValue += sizeof (UINT32);
    RNGValueLength -= sizeof (UINT32);
  }
 
  if (RNGValueLength > 0) {
    Status = Bcm2838RngReadValue (&Val);
    if (EFI_ERROR (Status)) {
      return Status;
    }
    while (RNGValueLength--) {
      *RNGValue++ = (UINT8)Val;
      Val >>= 8;
    }
  }
  return EFI_SUCCESS;
}
 
STATIC EFI_RNG_PROTOCOL mBcm2838RngProtocol = {
  Bcm2838RngGetInfo,
  Bcm2838RngGetRNG
};
 
//
// Entry point of this driver.
//
EFI_STATUS
EFIAPI
Bcm2838RngEntryPoint (
  IN EFI_HANDLE       ImageHandle,
  IN EFI_SYSTEM_TABLE *SystemTable
  )
{
  EFI_STATUS      Status;
 
  Status = gBS->InstallMultipleProtocolInterfaces (&ImageHandle,
                  &gEfiRngProtocolGuid, &mBcm2838RngProtocol,
                  NULL);
  ASSERT_EFI_ERROR (Status);
 
  //
  // Set the warm up bit count.
  //
  // This results in the RNG holding off from populating any value into the
  // FIFO until the value below has been reached in RNG_BIT_COUNT.
  //
  MmioWrite32 (RNG_BIT_COUNT_THRESHOLD, RNG_WARMUP_COUNT);
 
  //
  // We would disable RNG interrupts here... if we had access to the datasheet.
  //
 
  //
  // Enable the RNG with a sample rate divisor of 3.
  //
  // For a Bcm2711 running at 1.5 GHz, this should generate random bits at a
  // rate of about 1 every microsecond (1 MHz).
  //
  // Note that The divisor is a power of two index, meaning that each
  // consecutive divisor halves the frequency at which bits are being
  // produced. Zero is a special value that seems to set the rate somewhere
  // between divisors 1 and 2.
  //
  // Also note that Broadcom set the whole ENABLE_MASK in the Linux driver,
  // instead of single bits, which may be unintended. But since we don't have
  // any public documentation on what each of these bits do, we follow suit.
  //
  MmioWrite32 (RNG_CTRL,
    RNG_CTRL_ENABLE_MASK | (3 << RNG_CTRL_SAMPLE_RATE_DIVISOR_SHIFT));
 
  return EFI_SUCCESS;
}