Derive default for enums using attributes

Fraunhofer-AISEC · Sep 1, 2024 · d951ed7 · d951ed7
1 parent d407a3c
commit d951ed7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 14 deletions.
diff --git a/src/lm_ots/parameters.rs b/src/lm_ots/parameters.rs
@@ -12,21 +12,16 @@ use crate::{
 use crate::util::coef::coef_helper;
 
 /// Specifies the used Winternitz parameter.
-#[derive(Clone, Copy, PartialEq, Eq)]
+#[derive(Default, Clone, Copy, PartialEq, Eq)]
 pub enum LmotsAlgorithm {
+    #[default]
     LmotsReserved = 0,
     LmotsW1 = 1,
     LmotsW2 = 2,
     LmotsW4 = 3,
     LmotsW8 = 4,
 }
 
-impl Default for LmotsAlgorithm {
-    fn default() -> Self {
-        LmotsAlgorithm::LmotsReserved
-    }
-}
-
 impl From<u32> for LmotsAlgorithm {
     fn from(_type: u32) -> Self {
         match _type {

diff --git a/src/lms/parameters.rs b/src/lms/parameters.rs
@@ -3,8 +3,9 @@ use core::marker::PhantomData;
 use crate::hasher::{sha256::Sha256_256, HashChain};
 
 /// Specifies the used Tree height.
-#[derive(Clone, Copy)]
+#[derive(Default, Clone, Copy)]
 pub enum LmsAlgorithm {
+    #[default]
     LmsReserved = 0,
     #[cfg(test)]
     LmsH2 = 1,
@@ -15,12 +16,6 @@ pub enum LmsAlgorithm {
     LmsH25 = 9,
 }
 
-impl Default for LmsAlgorithm {
-    fn default() -> Self {
-        LmsAlgorithm::LmsReserved
-    }
-}
-
 impl From<u32> for LmsAlgorithm {
     fn from(_type: u32) -> Self {
         match _type {