1 files changed, 16 insertions, 3 deletions
diff --git a/oxish-proto/src/auth.rs b/oxish-proto/src/auth.rs
index ce328be..153f037 100644
--- a/oxish-proto/src/auth.rs
+++ b/oxish-proto/src/auth.rs
@@ -1,4 +1,4 @@
-use core::str;
+use core::{ops::Deref, str};
 use std::borrow::Cow;
 
 use tracing::warn;
@@ -276,7 +276,7 @@ pub struct SignatureData<'a> {
 
 impl<'a> SignatureData<'a> {
     /// Build the data that the client signs for public key authentication (RFC 4252 Section 7)
-    pub fn encode(&self) -> Vec<u8> {
+    pub fn encode(&self) -> SignatureInput {
         let mut buf = Vec::new();
         self.session_id.encode(&mut buf);
         MessageType::UserAuthRequest.encode(&mut buf);
@@ -286,7 +286,20 @@ impl<'a> SignatureData<'a> {
         true.encode(&mut buf);
         self.algorithm.encode(&mut buf);
         self.public_key.encode(&mut buf);
-        buf
+        SignatureInput(buf)
+    }
+}
+
+/// Encoded signature input for public key authentication
+///
+/// Constructed by [`SignatureData::encode()`].
+pub struct SignatureInput(Vec<u8>);
+
+impl Deref for SignatureInput {
+    type Target = [u8];
+
+    fn deref(&self) -> &Self::Target {
+        &self.0
     }
 }