5#include "sw/device/lib/testing/keymgr_testutils.h"
15#include "sw/device/lib/testing/entropy_testutils.h"
16#include "sw/device/lib/testing/flash_ctrl_testutils.h"
17#include "sw/device/lib/testing/kmac_testutils.h"
18#include "sw/device/lib/testing/otp_ctrl_testutils.h"
19#include "sw/device/lib/testing/rstmgr_testutils.h"
20#include "sw/device/lib/testing/test_framework/check.h"
22#include "sw/device/silicon_creator/lib/drivers/retention_sram.h"
26#define MODULE_ID MAKE_MODULE_ID('k', 'm', 't')
30 kFlashInfoPartitionId = 0,
36 kFlashInfoPageIdCreatorSecret = 1,
39 kFlashInfoPageIdOwnerSecret = 2,
42const static char *kKeymgrStageNames[] = {
53 const keymgr_testutils_secret_t *data,
57 TRY(flash_ctrl_testutils_info_region_scrambled_setup(
58 flash, page_id, kFlashInfoBankId, kFlashInfoPartitionId, &address));
60 TRY(flash_ctrl_testutils_info_region_setup(
61 flash, page_id, kFlashInfoBankId, kFlashInfoPartitionId, &address));
64 TRY(flash_ctrl_testutils_erase_and_write_page(
65 flash, address, kFlashInfoPartitionId, data->value,
66 kDifFlashCtrlPartitionTypeInfo,
ARRAYSIZE(data->value)));
68 keymgr_testutils_secret_t readback_data;
69 TRY(flash_ctrl_testutils_read(
70 flash, address, kFlashInfoPartitionId, readback_data.value,
71 kDifFlashCtrlPartitionTypeInfo,
ARRAYSIZE(readback_data.value), 0));
72 TRY_CHECK(memcmp(data->value, readback_data.value,
sizeof(data->value)) == 0);
76status_t keymgr_testutils_flash_init(
78 const keymgr_testutils_secret_t *creator_secret,
79 const keymgr_testutils_secret_t *owner_secret) {
81 write_info_page(flash, kFlashInfoPageIdCreatorSecret, creator_secret,
83 write_info_page(flash, kFlashInfoPageIdOwnerSecret, owner_secret,
88static status_t check_lock_otp_partition(
void) {
90 TRY(dif_otp_ctrl_init(
99 LOG_INFO(
"OTP partition locked. Digest: %x-%x", ((uint32_t *)&digest)[0],
100 ((uint32_t *)&digest)[1]);
115 TRY(kmac_testutils_config(kmac,
true));
125 TRY(keymgr_testutils_startup(keymgr, kmac));
126 LOG_INFO(
"Keymgr entered CreatorRootKey State");
129 TRY(keymgr_testutils_generate_identity(
132 LOG_INFO(
"Keymgr generated identity at CreatorRootKey State");
136 TRY(keymgr_testutils_advance_state(keymgr, &kOwnerIntParams));
137 TRY(keymgr_testutils_check_state(keymgr,
139 LOG_INFO(
"Keymgr entered OwnerIntKey State");
145 TRY(keymgr_testutils_try_startup(keymgr, kmac, &keymgr_state));
148 TRY(keymgr_testutils_advance_state(keymgr, &kOwnerIntParams));
149 TRY(dif_keymgr_get_state(keymgr, &keymgr_state));
153 TRY(keymgr_testutils_advance_state(keymgr, &kOwnerRootKeyParams));
161 return keymgr_initialize_sival(keymgr, kmac);
164 return keymgr_initialize_sim_dv(keymgr, kmac);
169 TRY(dif_init(keymgr, kmac));
173 TRY(dif_keymgr_get_state(keymgr, keymgr_state));
177 LOG_INFO(
"Unexpected keymgr state: 0x%x", keymgr_state);
182 TRY(keymgr_testutils_startup(keymgr, kmac));
183 TRY(dif_keymgr_get_state(keymgr, keymgr_state));
189status_t keymgr_testutils_init_nvm_then_reset(
void) {
196 rstmgr_testutils_reason_get();
200 LOG_INFO(
"Powered up for the first time, program flash");
202 TRY(dif_flash_ctrl_init_state(
205 TRY(keymgr_testutils_flash_init(&flash, &kCreatorSecret, &kOwnerSecret));
207 TRY(check_lock_otp_partition());
210 LOG_INFO(
"Requesting a reset to make OTP partitions accessible to keymgr");
211 rstmgr_testutils_reason_clear();
212 TRY(dif_rstmgr_software_device_reset(&rstmgr));
215 wait_for_interrupt();
232 bool is_using_test_rom =
235 .reserved[
ARRAYSIZE((retention_sram_t){0}.creator.reserved) - 1] ==
238 TRY(keymgr_testutils_init_nvm_then_reset());
247 LOG_INFO(
"Initializing entropy complex in Auto mode");
249 TRY(entropy_testutils_auto_mode_init());
251 LOG_INFO(
"Powered up for the second time, actuate keymgr and perform test.");
253 TRY(dif_init(keymgr, kmac));
257 TRY(keymgr_testutils_advance_state(keymgr, NULL));
259 LOG_INFO(
"Keymgr entered Init State");
262 if (is_using_test_rom) {
263 LOG_INFO(
"Using test_rom, setting inputs and advancing state...");
264 TRY(keymgr_testutils_advance_state(keymgr, &kCreatorParams));
266 LOG_INFO(
"Using rom, only advancing state...");
267 TRY(dif_keymgr_advance_state_raw(keymgr));
268 TRY(keymgr_testutils_wait_for_operation_done(keymgr));
271 LOG_INFO(
"Keymgr entered CreatorRootKey State");
276 TRY(keymgr_testutils_generate_identity(
279 LOG_INFO(
"Keymgr generated identity at CreatorRootKey State");
284status_t keymgr_testutils_advance_state(
286 TRY(dif_keymgr_advance_state(keymgr, params));
287 return keymgr_testutils_wait_for_operation_done(keymgr);
290status_t keymgr_testutils_check_state(
const dif_keymgr_t *keymgr,
293 TRY(dif_keymgr_get_state(keymgr, &act_state));
294 TRY_CHECK(act_state == exp_state,
295 "Keymgr in unexpected state: %x, expected to be %x", act_state,
300status_t keymgr_testutils_generate_identity(
303 TRY(dif_keymgr_generate_identity_seed(keymgr, params));
304 return keymgr_testutils_wait_for_operation_done(keymgr);
307status_t keymgr_testutils_generate_versioned_key(
310 TRY(dif_keymgr_generate_versioned_key(keymgr, params));
311 return keymgr_testutils_wait_for_operation_done(keymgr);
314status_t keymgr_testutils_disable(
const dif_keymgr_t *keymgr) {
315 TRY(dif_keymgr_disable(keymgr));
316 return keymgr_testutils_wait_for_operation_done(keymgr);
319status_t keymgr_testutils_wait_for_operation_done(
const dif_keymgr_t *keymgr) {
322 TRY(dif_keymgr_get_status_codes(keymgr, &
status));
329status_t keymgr_testutils_max_key_version_get(
const dif_keymgr_t *keymgr,
330 uint32_t *max_key_version) {
332 TRY(dif_keymgr_get_state(keymgr, &keymgr_state));
337 LOG_INFO(
"Unexpected keymgr state: 0x%x", keymgr_state);
341 dif_keymgr_max_key_version_t versions;
342 TRY(dif_keymgr_read_max_key_version(keymgr, &versions));
344 switch (keymgr_state) {
361status_t keymgr_testutils_state_string_get(
const dif_keymgr_t *keymgr,
362 const char **stage_name) {
364 CHECK_DIF_OK(dif_keymgr_get_state(keymgr, &state));
366 if (state >=
ARRAYSIZE(kKeymgrStageNames)) {
371 *stage_name = kKeymgrStageNames[state];