socks5_protocol_async/protocol/
auth_method_negotiation_reply.rs

1use super::constant;
2use failure::Error;
3use futures_io::AsyncWrite;
4use futures_util::AsyncWriteExt;
5
6#[derive(Debug, Clone, PartialEq, Eq)]
7pub struct AuthMethodNegotiationReply {
8    pub selected_method: u8,
9}
10
11impl AuthMethodNegotiationReply {
12    pub async fn write<AW>(&self, writer: &mut AW) -> Result<(), Error>
13    where
14        AW: AsyncWrite + Unpin,
15    {
16        let buf = [constant::protocol_version::SOCKS5, self.selected_method];
17        writer.write_all(&buf).await?;
18        Ok(())
19    }
20}
21
22#[cfg(test)]
23mod test {
24    use super::*;
25    use crate::test_util::*;
26    use futures_util::io::Cursor;
27
28    #[test]
29    fn happy_path() {
30        let mut writer = [0u8; 2];
31        let mut writer = Cursor::new(&mut writer[..]);
32        let res = AuthMethodNegotiationReply {
33            selected_method: 0xFF,
34        };
35        let future = res.write(&mut writer);
36        let result = extract_future_output(future);
37        assert_eq!(result.unwrap(), ());
38        assert_eq!(writer.into_inner(), [5, 0xFF])
39    }
40
41    #[test]
42    fn unable_to_write_whole_thing() {
43        let mut writer = [0u8; 1];
44        let mut writer = Cursor::new(&mut writer[..]);
45        let res = AuthMethodNegotiationReply {
46            selected_method: 0xFF,
47        };
48        let future = res.write(&mut writer);
49        let result = extract_future_output(future);
50        let err = result.unwrap_err();
51        let err = err.downcast::<std::io::Error>().unwrap();
52        assert_eq!(err.kind(), std::io::ErrorKind::WriteZero);
53        assert_eq!(writer.into_inner(), [5])
54    }
55}

socks5_protocol_async/protocol/auth_method_negotiation_reply.rs

socks5_protocol_async/protocol/
auth_method_negotiation_reply.rs