drivers: Add driver for ml_dsa87 peripheral

This adds both a driver and some unit tests.

Some TODOs are left in the driver as it is intended to hook this
peripheral to the keyvault in the future.

Signed-off-by: Arthur Heymans <arthur.heymans@9elements.com>

builder/src/firmware.rs

@@ -164,6 +164,11 @@ pub mod driver_tests {
         ..BASE_FWID
     };
 
+    pub const ML_DSA87: FwId = FwId {
+        bin_name: "ml_dsa87",
+        ..BASE_FWID
+    };
+
     pub const ECC384: FwId = FwId {
         bin_name: "ecc384",
         ..BASE_FWID
@@ -400,6 +405,7 @@ pub const REGISTERED_FW: &[&FwId] = &[
     &hw_model_tests::TEST_UNITIALIZED_READ,
     &hw_model_tests::TEST_PCR_EXTEND,
     &driver_tests::DOE,
+    &driver_tests::ML_DSA87,
     &driver_tests::ECC384,
     &driver_tests::ECC384_SIGN_VALIDATION_FAILURE,
     &driver_tests::ERROR_REPORTER,

drivers/src/array.rs

@@ -160,6 +160,9 @@ pub type Array4x8 = Array4xN<8, 32>;
 pub type Array4x12 = Array4xN<12, 48>;
 pub type Array4x16 = Array4xN<16, 64>;
 pub type Array4x32 = Array4xN<32, 128>;
+pub type Array4x648 = Array4xN<648, 2592>;
+pub type Array4x1157 = Array4xN<1157, 4628>;
+pub type Array4x1224 = Array4xN<1224, 4896>;
 
 #[cfg(test)]
 mod tests {

drivers/src/lib.rs

@@ -37,6 +37,7 @@ mod kv_access;
 mod lms;
 mod mailbox;
 pub mod memory_layout;
+mod ml_dsa87;
 mod okref;
 mod pcr_bank;
 pub mod pcr_log;
@@ -52,7 +53,10 @@ mod soc_ifc;
 mod trng;
 mod trng_ext;
 
-pub use array::{Array4x12, Array4x16, Array4x4, Array4x5, Array4x8, Array4xN};
+pub use array::{
+    Array4x1157, Array4x12, Array4x1224, Array4x16, Array4x4, Array4x5, Array4x648, Array4x8,
+    Array4xN,
+};
 pub use array_concat::array_concat3;
 pub use bounded_address::{BoundedAddr, MemBounds, RomAddr};
 pub use caliptra_error::{CaliptraError, CaliptraResult};
@@ -82,6 +86,10 @@ pub use lms::{
     Sha256Digest, D_INTR, D_LEAF, D_MESG, D_PBLC,
 };
 pub use mailbox::{Mailbox, MailboxRecvTxn, MailboxSendTxn};
+pub use ml_dsa87::{
+    MlDsa87, MlDsa87MsgScalar, MlDsa87PublicKey, MlDsa87PublicKeyScalar, MlDsa87Result,
+    MlDsa87SecretKey, MlDsa87SecretKeyScalar, MlDsa87Seed, MlDsa87SignatureScalar,
+};
 pub use okref::okmutref;
 pub use okref::okref;
 pub use pcr_bank::{PcrBank, PcrId};