Skip to content

Commit 40367b4

Browse files
committed
Made sphinx packet builder more generic
1 parent 854a821 commit 40367b4

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/packet/builder.rs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,9 @@ impl<'a> SphinxPacketBuilder<'a> {
2828
self
2929
}
3030

31-
pub fn build_packet(
31+
pub fn build_packet<M: AsRef<[u8]>>(
3232
&self,
33-
message: Vec<u8>,
33+
message: M,
3434
route: &[Node],
3535
destination: &Destination,
3636
delays: &[Delay],
@@ -41,7 +41,8 @@ impl<'a> SphinxPacketBuilder<'a> {
4141
};
4242

4343
// no need to check if plaintext has correct length as this check is already performed in payload encapsulation
44-
let payload = Payload::encapsulate_message(&message, &payload_keys, self.payload_size)?;
44+
let payload =
45+
Payload::encapsulate_message(message.as_ref(), &payload_keys, self.payload_size)?;
4546
Ok(SphinxPacket { header, payload })
4647
}
4748
}

0 commit comments

Comments
 (0)