tentacli 15.2.0

Framework for building extensible network protocol clients via modular plugins.
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use async_trait::async_trait;
use std::sync::Arc;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::UdpSocket;
use tokio::net::tcp::{OwnedReadHalf, OwnedWriteHalf};

#[async_trait]
pub trait TransportRead: Send + Sync {
    async fn read(&mut self, buffer: &mut [u8]) -> anyhow::Result<usize>;
}

#[async_trait]
pub trait TransportWrite: Send + Sync {
    async fn write(&mut self, buffer: &mut [u8]) -> anyhow::Result<usize>;
}

pub struct TcpRead(pub OwnedReadHalf);
#[async_trait]
impl TransportRead for TcpRead {
    async fn read(&mut self, buffer: &mut [u8]) -> anyhow::Result<usize> {
        self.0.read(buffer).await.map_err(Into::into)
    }
}

pub struct TcpWrite(pub OwnedWriteHalf);
#[async_trait]
impl TransportWrite for TcpWrite {
    async fn write(&mut self, buffer: &mut [u8]) -> anyhow::Result<usize> {
        self.0.write(buffer).await.map_err(Into::into)
    }
}

pub struct UdpRead(pub Arc<UdpSocket>);
#[async_trait]
impl TransportRead for UdpRead {
    async fn read(&mut self, buffer: &mut [u8]) -> anyhow::Result<usize> {
        self.0.recv(buffer).await.map_err(Into::into)
    }
}

pub struct UdpWrite(pub Arc<UdpSocket>);
#[async_trait]
impl TransportWrite for UdpWrite {
    async fn write(&mut self, buffer: &mut [u8]) -> anyhow::Result<usize> {
        self.0.send(buffer).await.map_err(Into::into)
    }
}

#[allow(dead_code)]
pub enum Protocol {
    TCP,
    UDP,
}