From 40cc4970831f112d2c482f62ec7458e70657900e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sosth=C3=A8ne=20Gu=C3=A9don?= <sosthene@nitrokey.com>
Date: Wed, 23 Nov 2022 11:59:11 +0100
Subject: [PATCH] Add support for generate asymmetric keypair

---
 src/lib.rs | 40 +++++++++++++++++++++++++++++++++++++---
 1 file changed, 37 insertions(+), 3 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index c6b9b34..1e18c8f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -617,7 +617,7 @@ impl<'a, T: trussed::Client + trussed::client::Ed255> LoadedAuthenticator<'a, T>
         &mut self,
         reference: AsymmetricKeyReference,
         data: &[u8],
-        reply: Reply<'_, R>,
+        mut reply: Reply<'_, R>,
     ) -> Result {
         if !self
             .state
@@ -712,10 +712,44 @@ impl<'a, T: trussed::Client + trussed::client::Ed255> LoadedAuthenticator<'a, T>
 
         match parsed_mechanism {
             AsymmetricAlgorithms::P256 => {
-                todo!()
+                let serialized_key = syscall!(self.trussed.serialize_key(
+                    trussed::types::Mechanism::P256,
+                    public_key,
+                    trussed::types::KeySerialization::Raw
+                ))
+                .serialized_key;
+                reply.expand(&[0x7F, 0x49])?;
+                let offset = reply.len();
+                reply.expand(&[0x86])?;
+                reply.append_len(serialized_key.len() + 1)?;
+                reply.expand(&[0x04])?;
+                reply.expand(&serialized_key)?;
+                reply.prepend_len(offset)?;
             }
             AsymmetricAlgorithms::Rsa2048 | AsymmetricAlgorithms::Rsa4096 => {
-                todo!()
+                reply.expand(&[0x7F, 0x49])?;
+                let offset = reply.len();
+                let serialized_e = syscall!(self.trussed.serialize_key(
+                    trussed::types::Mechanism::P256,
+                    public_key,
+                    trussed::types::KeySerialization::RsaE
+                ))
+                .serialized_key;
+                reply.expand(&[0x81])?;
+                reply.append_len(serialized_e.len())?;
+                reply.expand(&serialized_e)?;
+
+                let serialized_n = syscall!(self.trussed.serialize_key(
+                    trussed::types::Mechanism::P256,
+                    public_key,
+                    trussed::types::KeySerialization::RsaN
+                ))
+                .serialized_key;
+                reply.expand(&[0x82])?;
+                reply.append_len(serialized_n.len())?;
+                reply.expand(&serialized_n)?;
+
+                reply.prepend_len(offset)?;
             }
         };