aes_testutils.c

// Copyright lowRISC contributors (OpenTitan project).
// Licensed under the Apache License, Version 2.0, see LICENSE for details.
// SPDX-License-Identifier: Apache-2.0
 
#include "sw/device/lib/testing/aes_testutils.h"
 
#include "hw/ip/aes/model/aes_modes.h"
#include "hw/top/dt/alert_handler.h"
#include "sw/device/lib/dif/dif_aes.h"
#include "sw/device/lib/dif/dif_alert_handler.h"
#include "sw/device/lib/testing/test_framework/check.h"
 
#ifdef AES_TESTUTILS_HAS_EDN_AND_CSRNG
#include "hw/top/dt/csrng.h"
#include "sw/device/lib/dif/dif_csrng_shared.h"
#include "sw/device/lib/testing/csrng_testutils.h"
 
#include "hw/top/csrng_regs.h"  // Generated
#endif
 
// `extern` declarations to give the inline functions in the
// corresponding header a link location.
 
extern bool aes_testutils_get_status(dif_aes_t *aes, dif_aes_status_t flag);
 
// The mask share, used to mask kKey. Note that the masking should not be done
// manually. Software is expected to get the key in two shares right from the
// beginning.
static const uint8_t kKeyShare1[] = {
    0x0f, 0x1f, 0x2f, 0x3F, 0x4f, 0x5f, 0x6f, 0x7f, 0x8f, 0x9f, 0xaf,
    0xbf, 0xcf, 0xdf, 0xef, 0xff, 0x0a, 0x1a, 0x2a, 0x3a, 0x4a, 0x5a,
    0x6a, 0x7a, 0x8a, 0x9a, 0xaa, 0xba, 0xca, 0xda, 0xea, 0xfa,
};
 
dif_aes_key_share_t key;
 
enum {
  kAesTestutilsTimeout = (10 * 1000 * 1000),
};
 
#ifdef AES_TESTUTILS_HAS_EDN_AND_CSRNG
/**
 * Constants for switching AES masking off.
 */
enum {
  kCsrngBlockLen = 4,
  kCsrngKeyLen = 8,
  kEdnSeedMaterialLen = 12,
};
 
// CSRNG needs to constantly output these bits to EDN. If reseeded to an all-
// zero value the AES masking PRNG will output an all-zero vector. It will
// further keep this value if the CTRL_AUX_SHADOWED.FORCE_MASKS bit is set.
const uint32_t kAesMaskingPrngZeroOutputSeed[kCsrngBlockLen] = {
    0x00000000, 0x000000000, 0x00000000, 0x00000000};
 
// Seed material for instantiate command. The CTR_DRBG construction
// implemented by CSRNG produces
//
// key = 00 01 02 03 04 05 06 07 - 08 09 0a 0b 0c 0d 0e 0f
//       10 11 12 13 14 15 16 17 - 18 19 1a 1b 1c 1d 1e 1f
//
//   V = 6d 9f 08 eb 2a 2e 27 7a - b4 89 84 cf f1 ab 9a 09
//
// from this seed material upon instantiate. The key is arbitrarily chosen.
// Encrypting V using this key then gives the required
// kAesMaskingPrngZeroOutputSeed above.
const uint32_t kEdnSeedMaterialInstantiate[kEdnSeedMaterialLen] = {
    0x84adaf86, 0x652b7141, 0x1d880d0e, 0x1fff0b21, 0xa6ee8307, 0x1f57dfc8,
    0x59757d79, 0xdeb6522e, 0xc8c67d84, 0xa16abefa, 0xc34030be, 0x530e88f8};
 
// V and key after instantiate.
const uint32_t kCsrngVInstantiate[kCsrngBlockLen] = {0xf1ab9a08, 0xb48984cf,
                                                     0x2a2e277a, 0x6d9f08eb};
const uint32_t kCsrngKeyInstantiate[kCsrngKeyLen] = {
    0x1c1d1e1f, 0x18191a1b, 0x14151617, 0x10111213,
    0x0c0d0e0f, 0x08090a0b, 0x04050607, 0x00010203};
 
// V and key after generate.
const uint32_t kCsrngVGenerate[kCsrngBlockLen] = {0x654600bd, 0xf0c32787,
                                                  0x3eb52114, 0x8a1e0dce};
const uint32_t kCsrngKeyGenerate[kCsrngKeyLen] = {
    0xff6589b5, 0x4bb8e5f9, 0x62847098, 0x1e9f9cd1,
    0x3c005fbd, 0x9a1b6e70, 0xe30eb080, 0x71dea927};
 
// Seed material for reseed command. After one generate, this seed material
// will bring the key and V of CSRNG back to the state after instantiate.
// I.e., one can again run one generate to produce the seed required for AES
// (see kAesMaskingPrngZeroOutputSeed).
const uint32_t kEdnSeedMaterialReseed[kEdnSeedMaterialLen] = {
    0x96994362, 0x7ef8f0b9, 0x5b5332dc, 0xd0df9b12, 0x96dfbaa9, 0xac0b5af7,
    0xec2504be, 0xb00fb68c, 0xf37e0a7f, 0x88172eec, 0x4e4b5f58, 0xfec120c0};
 
status_t aes_testutils_masking_prng_zero_output_seed(const dif_csrng_t *csrng,
                                                     const dif_edn_t *edn0) {
  // Shutdown EDN0 and CSRNG
  TRY(dif_edn_stop(edn0));
  TRY(dif_csrng_stop(csrng));
 
  // Re-enable CSRNG
  TRY(dif_csrng_configure(csrng));
 
  // Re-enable EDN0 and configure it to produce the seed that if loaded into AES
  // causes the AES masking PRNG to output and all-zero output.
  dif_edn_auto_params_t edn0_params = {
      .instantiate_cmd =
          {
              .cmd = csrng_cmd_header_build(kCsrngAppCmdInstantiate,
                                            kDifCsrngEntropySrcToggleDisable,
                                            kEdnSeedMaterialLen,
                                            /*generate_len=*/0),
              .seed_material =
                  {
                      .len = kEdnSeedMaterialLen,
                  },
          },
      .reseed_cmd =
          {
              .cmd = csrng_cmd_header_build(kCsrngAppCmdReseed,
                                            kDifCsrngEntropySrcToggleDisable,
                                            kEdnSeedMaterialLen,
                                            /*generate_len=*/0),
              .seed_material =
                  {
                      .len = kEdnSeedMaterialLen,
                  },
          },
      .generate_cmd =
          {
              .cmd = csrng_cmd_header_build(kCsrngAppCmdGenerate,
                                            kDifCsrngEntropySrcToggleDisable,
                                            /*cmd_len=*/0,
                                            /*generate_len=*/1),
              .seed_material =
                  {
                      .len = 0,
                  },
          },
      .reseed_interval = 1,  // Reseed after every single generate.
  };
  memcpy(edn0_params.instantiate_cmd.seed_material.data,
         kEdnSeedMaterialInstantiate, sizeof(kEdnSeedMaterialInstantiate));
  memcpy(edn0_params.reseed_cmd.seed_material.data, kEdnSeedMaterialReseed,
         sizeof(kEdnSeedMaterialReseed));
  TRY(dif_edn_set_auto_mode(edn0, edn0_params));
  return OK_STATUS();
}
 
status_t aes_testutils_csrng_kat(const dif_csrng_t *csrng) {
  // Instantiate CSRNG with seed material suitable for switching the AES masking
  // off.
  dif_csrng_seed_material_t seed_material_instantiate = {
      .seed_material_len = 12,
  };
  memcpy(seed_material_instantiate.seed_material, kEdnSeedMaterialInstantiate,
         sizeof(kEdnSeedMaterialInstantiate));
  dif_csrng_internal_state_t expected_state_instantiate = {
      .reseed_counter = 0,
      .instantiated = true,
      .fips_compliance = false,
  };
  memcpy(expected_state_instantiate.v, kCsrngVInstantiate,
         sizeof(kCsrngVInstantiate));
  memcpy(expected_state_instantiate.key, kCsrngKeyInstantiate,
         sizeof(kCsrngKeyInstantiate));
  TRY(csrng_testutils_kat_instantiate(csrng, false, &seed_material_instantiate,
                                      &expected_state_instantiate));
 
  // Generate one block containing the required seed for the AES masking PRNG
  // to output an all-zero vector.
  dif_csrng_internal_state_t expected_state_generate = {
      .reseed_counter = 1,
      .instantiated = true,
      .fips_compliance = false,
  };
  memcpy(expected_state_generate.v, kCsrngVGenerate, sizeof(kCsrngVGenerate));
  memcpy(expected_state_generate.key, kCsrngKeyGenerate,
         sizeof(kCsrngKeyGenerate));
  TRY(csrng_testutils_kat_generate(csrng, 1, kCsrngBlockLen, NULL,
                                   kAesMaskingPrngZeroOutputSeed,
                                   &expected_state_generate));
 
  // Reseed the CSRNG instance to produce the required seed for the AES masking
  // PRNG to output an all-zero vector upon the next generate command.
  dif_csrng_seed_material_t seed_material_reseed = {
      .seed_material_len = 12,
  };
  memcpy(seed_material_reseed.seed_material, kEdnSeedMaterialReseed,
         sizeof(kEdnSeedMaterialReseed));
  dif_csrng_internal_state_t expected_state_reseed = {
      .reseed_counter = 0,
      .instantiated = true,
      .fips_compliance = false,
  };
  memcpy(expected_state_reseed.v, kCsrngVInstantiate,
         sizeof(kCsrngVInstantiate));
  memcpy(expected_state_reseed.key, kCsrngKeyInstantiate,
         sizeof(kCsrngKeyInstantiate));
  TRY(csrng_testutils_kat_reseed(csrng, &seed_material_reseed,
                                 &expected_state_reseed));
 
  // We're about to generate the same `genbits` again which will trigger a
  // `csrng_recov_alert`. This is expected, so we must disable this alert if it
  // is currently enabled and restore it after.
  dif_alert_handler_t alert_handler;
  TRY(dif_alert_handler_init_from_dt(kDtAlertHandler, &alert_handler));
 
  dt_alert_id_t csrng_recov_alert =
      dt_csrng_alert_to_alert_id(kDtCsrng, kDtCsrngAlertRecovAlert);
 
  dif_toggle_t was_enabled = kDifToggleDisabled;
  TRY(dif_alert_handler_alert_is_enabled(&alert_handler, csrng_recov_alert,
                                         &was_enabled));
 
  if (was_enabled == kDifToggleEnabled) {
    TRY(dif_alert_handler_alert_set_enabled(&alert_handler, csrng_recov_alert,
                                            kDifToggleDisabled));
  }
 
  // Generate one block containing the required seed for the AES masking PRNG
  // to output an all-zero vector.
  TRY(csrng_testutils_kat_generate(csrng, 1, kCsrngBlockLen, NULL,
                                   kAesMaskingPrngZeroOutputSeed,
                                   &expected_state_generate));
 
  // Check that the correct recoverable alert fired even if the alert was
  // ignored.
  uint32_t recov_alerts = 0;
  TRY(dif_csrng_get_recoverable_alerts(csrng, &recov_alerts));
  TRY_CHECK(recov_alerts == kDifCsrngRecoverableAlertRepeatedGenBits);
  TRY(dif_csrng_clear_recoverable_alerts(csrng));
 
  // Restore the alert if it was enabled.
  if (was_enabled == kDifToggleEnabled) {
    TRY(dif_alert_handler_alert_set_enabled(&alert_handler, csrng_recov_alert,
                                            kDifToggleEnabled));
  }
 
  return OK_STATUS();
}
#endif
 
status_t aes_testutils_setup_encryption(dif_aes_transaction_t transaction,
                                        dif_aes_t *aes) {
  // Mask the key. Note that this should not be done manually. Software is
  // expected to get the key in two shares right from the beginning.
  uint8_t key_share0[sizeof(kAesModesKey256)];
  for (int i = 0; i < sizeof(kAesModesKey256); ++i) {
    key_share0[i] = kAesModesKey256[i] ^ kKeyShare1[i];
  }
 
  // "Convert" key share byte arrays to `dif_aes_key_share_t`.
  memcpy(key.share0, key_share0, sizeof(key.share0));
  memcpy(key.share1, kKeyShare1, sizeof(key.share1));
 
  AES_TESTUTILS_WAIT_FOR_STATUS(aes, kDifAesStatusIdle, true,
                                kAesTestutilsTimeout);
  CHECK_DIF_OK(dif_aes_start(aes, &transaction, &key, NULL));
 
  // "Convert" plain data byte arrays to `dif_aes_data_t`.
  dif_aes_data_t in_data_plain;
  memcpy(in_data_plain.data, kAesModesPlainText, sizeof(in_data_plain.data));
 
  // Load the plain text to trigger the encryption operation.
  AES_TESTUTILS_WAIT_FOR_STATUS(aes, kDifAesStatusIdle, true,
                                kAesTestutilsTimeout);
  AES_TESTUTILS_WAIT_FOR_STATUS(aes, kDifAesStatusInputReady, true,
                                kAesTestutilsTimeout);
  CHECK_DIF_OK(dif_aes_load_data(aes, in_data_plain));
 
  return OK_STATUS();
}
 
status_t aes_testutils_decrypt_ciphertext(dif_aes_transaction_t transaction,
                                          dif_aes_t *aes) {
  // Read out the produced cipher text.
  dif_aes_data_t out_data;
  CHECK_DIF_OK(dif_aes_read_output(aes, &out_data));
 
  // Finish the ECB encryption transaction.
  CHECK_DIF_OK(dif_aes_end(aes));
  CHECK_ARRAYS_EQ((uint8_t *)out_data.data, kAesModesCipherTextEcb256,
                  sizeof(out_data.data));
 
  // Setup ECB decryption transaction.
  transaction.operation = kDifAesOperationDecrypt;
  CHECK_DIF_OK(dif_aes_start(aes, &transaction, &key, NULL));
 
  // Load the previously produced cipher text to start the decryption operation.
  AES_TESTUTILS_WAIT_FOR_STATUS(aes, kDifAesStatusInputReady, true,
                                kAesTestutilsTimeout);
  CHECK_DIF_OK(dif_aes_load_data(aes, out_data));
 
  // Read out the produced plain text.
  AES_TESTUTILS_WAIT_FOR_STATUS(aes, kDifAesStatusOutputValid, true,
                                kAesTestutilsTimeout);
  CHECK_DIF_OK(dif_aes_read_output(aes, &out_data));
 
  // Finish the ECB encryption transaction.
  CHECK_DIF_OK(dif_aes_end(aes));
 
  CHECK_ARRAYS_EQ((uint8_t *)out_data.data, kAesModesPlainText,
                  sizeof(out_data.data));
  return OK_STATUS();
}