warg_crypto/
encoding.rs

1use signature::Error as SignatureError;
2
3use crate::signing;
4use crate::{ByteVisitor, VisitBytes};
5
6pub trait Encode {
7    fn encode(&self) -> Vec<u8>;
8}
9
10#[derive(Default)]
11struct EncodingVisitor {
12    bytes: Vec<u8>,
13}
14
15impl ByteVisitor for EncodingVisitor {
16    fn visit_bytes(&mut self, bytes: impl AsRef<[u8]>) {
17        self.bytes.extend(bytes.as_ref())
18    }
19}
20
21impl<T> Encode for T
22where
23    T: VisitBytes,
24{
25    fn encode(&self) -> Vec<u8> {
26        let mut visitor = EncodingVisitor::default();
27        self.visit(&mut visitor);
28        visitor.bytes
29    }
30}
31
32pub trait Signable: Encode {
33    const PREFIX: &'static [u8];
34
35    fn sign(
36        &self,
37        private_key: &signing::PrivateKey,
38    ) -> Result<signing::Signature, SignatureError> {
39        let prefixed_content = [Self::PREFIX, b":", self.encode().as_slice()].concat();
40        private_key.sign(&prefixed_content)
41    }
42
43    fn verify(
44        public_key: &signing::PublicKey,
45        msg: &[u8],
46        signature: &signing::Signature,
47    ) -> Result<(), SignatureError> {
48        let prefixed_content = [Self::PREFIX, b":", msg].concat();
49        public_key.verify(&prefixed_content, signature)
50    }
51}
warg_crypto/encoding.rs

warg_crypto/
encoding.rs
