From ea7d0023846f63f8a8c804508883180289b3f80e Mon Sep 17 00:00:00 2001
From: Tyler Fanelli <tfanelli@redhat.com>
Date: Fri, 16 Feb 2024 00:07:29 -0500
Subject: [PATCH] lib: Add bergamo and siena support to Generation from String

Signed-off-by: Tyler Fanelli <tfanelli@redhat.com>
---
 src/lib.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index d7aa6d23..178ed282 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -350,6 +350,12 @@ impl TryFrom<String> for Generation {
             #[cfg(any(feature = "sev", feature = "snp"))]
             "genoa" => Ok(Self::Genoa),
 
+            #[cfg(any(feature = "sev", feature = "snp"))]
+            "bergamo" => Ok(Self::Genoa),
+
+            #[cfg(any(feature = "sev", feature = "snp"))]
+            "siena" => Ok(Self::Genoa),
+
             _ => Err(()),
         }
     }