chore: Allow exporting keys to slices for import

wireapp · Aug 13, 2024 · 020f919 · 020f919
1 parent ab7d4a4
commit 020f919
Showing 1 changed file with 19 additions and 9 deletions.
diff --git a/src/internal/keys.rs b/src/internal/keys.rs
@@ -255,15 +255,17 @@ impl PreKeyBundle {
 
     pub fn verify(&self) -> PreKeyAuth {
         match self.signature {
-            Some(ref sig) => if self
-                .identity_key
-                .public_key
-                .verify(sig, &self.public_key.pub_edward.0)
-            {
-                PreKeyAuth::Valid
-            } else {
-                PreKeyAuth::Invalid
-            },
+            Some(ref sig) => {
+                if self
+                    .identity_key
+                    .public_key
+                    .verify(sig, &self.public_key.pub_edward.0)
+                {
+                    PreKeyAuth::Valid
+                } else {
+                    PreKeyAuth::Invalid
+                }
+            }
             None => PreKeyAuth::Unknown,
         }
     }
@@ -442,6 +444,10 @@ impl SecretKey {
         }
     }
 
+    pub fn as_slice(&self) -> &[u8] {
+        self.sec_edward.0.as_slice()
+    }
+
     pub fn shared_secret(&self, p: &PublicKey) -> Result<[u8; 32], Zero> {
         ecdh::scalarmult(&self.sec_curve, &p.pub_curve)
             .map(|ge| ge.0)
@@ -505,6 +511,10 @@ impl PublicKey {
         sign::verify_detached(&s.sig, m, &self.pub_edward)
     }
 
+    pub fn as_slice(&self) -> &[u8] {
+        self.pub_edward.0.as_slice()
+    }
+
     pub fn fingerprint(&self) -> String {
         fmt_hex(&self.pub_edward.0)
     }