dust-devil-core 1.1.0

Shared sources used in the dust-devil socks5 server and monitoring clients.
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
39
40
41
42
43
44
45
46
47

use std::io::{Error, ErrorKind};

use tokio::io::AsyncReadExt;

use crate::serialize::{ByteRead, ByteWrite};

use super::SandstormCommandType;

/// A Sandstorm meow (ping) request.
pub struct MeowRequest;

/// A Sandstorm meow (ping) response.
pub struct MeowResponse;

impl ByteRead for MeowRequest {
    async fn read<R: tokio::io::AsyncRead + Unpin + ?Sized>(_reader: &mut R) -> Result<Self, Error> {
        Ok(Self)
    }
}

impl ByteWrite for MeowRequest {
    async fn write<W: tokio::io::AsyncWrite + Unpin + ?Sized>(&self, writer: &mut W) -> Result<(), Error> {
        SandstormCommandType::Meow.write(writer).await
    }
}

impl ByteRead for MeowResponse {
    async fn read<R: tokio::io::AsyncRead + Unpin + ?Sized>(reader: &mut R) -> Result<Self, Error> {
        let mut meow = [0u8; 4];
        reader.read_exact(&mut meow).await?;

        if meow == [b'M', b'E', b'O', b'W'] {
            Ok(Self)
        } else {
            Err(Error::new(
                ErrorKind::InvalidData,
                "Server responded to meow, but did not say MEOW!",
            ))
        }
    }
}

impl ByteWrite for MeowResponse {
    async fn write<W: tokio::io::AsyncWrite + Unpin + ?Sized>(&self, writer: &mut W) -> Result<(), Error> {
        (SandstormCommandType::Meow, b'M', b'E', b'O', b'W').write(writer).await
    }
}