ethrex-p2p 17.0.0

Peer-to-peer networking (discv4/discv5, RLPx, eth, snap) for the ethrex Ethereum client
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use crate::discv4::messages::{Message, Packet, PacketDecodeErr};

use bytes::BytesMut;
use secp256k1::SecretKey;
use tokio_util::codec::{Decoder, Encoder};

#[derive(Debug)]
pub struct Discv4Codec {
    signer: SecretKey,
}

impl Discv4Codec {
    pub fn new(signer: SecretKey) -> Self {
        Self { signer }
    }
}

impl Decoder for Discv4Codec {
    type Item = Packet;
    type Error = PacketDecodeErr;

    fn decode(&mut self, buf: &mut BytesMut) -> Result<Option<Self::Item>, Self::Error> {
        if !buf.is_empty() {
            Ok(Some(Packet::decode(&buf.split_to(buf.len()))?))
        } else {
            Ok(None)
        }
    }
}

impl Encoder<Message> for Discv4Codec {
    type Error = PacketDecodeErr;

    fn encode(&mut self, message: Message, buf: &mut BytesMut) -> Result<(), Self::Error> {
        message.encode_with_header(buf, &self.signer);
        Ok(())
    }
}