Made sphinx packet builder more generic

nymtech · Nov 9, 2022 · cc8e4a5 · cc8e4a5
1 parent bec341d
commit cc8e4a5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/packet/builder.rs b/src/packet/builder.rs
@@ -28,9 +28,9 @@ impl<'a> SphinxPacketBuilder<'a> {
         self
     }
 
-    pub fn build_packet(
+    pub fn build_packet<M: AsRef<[u8]>>(
         &self,
-        message: Vec<u8>,
+        message: M,
         route: &[Node],
         destination: &Destination,
         delays: &[Delay],
@@ -41,7 +41,8 @@ impl<'a> SphinxPacketBuilder<'a> {
         };
 
         // no need to check if plaintext has correct length as this check is already performed in payload encapsulation
-        let payload = Payload::encapsulate_message(&message, &payload_keys, self.payload_size)?;
+        let payload =
+            Payload::encapsulate_message(message.as_ref(), &payload_keys, self.payload_size)?;
         Ok(SphinxPacket { header, payload })
     }
 }