internet 0.1.0

Network library for rust
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

//! Codec for structures.

use crate::Cursor;
use crate::{Buf, BufMut, BufResult};

/// A `Codec` trait provides encoding and decoding.
pub trait Codec<C = ()>: Sized {
    /// Encodes `self` using `writer` and returns `()` or [`BufError`] if error.
    ///
    /// [`BufError`]: crate::BufError
    fn encode<W: BufMut>(&self, writer: &mut Cursor<W>, context: C) -> BufResult<()>;

    /// Decodes bytes from `reader` and returns `Self` or [`BufError`] if error.
    ///
    /// [`BufError`]: crate::BufError
    fn decode<R: Buf>(reader: &mut Cursor<R>, context: C) -> BufResult<Self>;
}

impl Codec for u8 {
    #[inline]
    fn encode<W: BufMut>(&self, writer: &mut Cursor<W>, _: ()) -> BufResult<()> {
        writer.write_u8(*self)
    }

    #[inline]
    fn decode<R: Buf>(reader: &mut Cursor<R>, _: ()) -> BufResult<Self> {
        reader.read_u8()
    }
}

impl Codec for u16 {
    #[inline]
    fn encode<W: BufMut>(&self, writer: &mut Cursor<W>, _: ()) -> BufResult<()> {
        writer.write_u16_be(*self)
    }

    #[inline]
    fn decode<R: Buf>(reader: &mut Cursor<R>, _: ()) -> BufResult<Self> {
        reader.read_u16_be()
    }
}

impl Codec for u32 {
    #[inline]
    fn encode<W: BufMut>(&self, writer: &mut Cursor<W>, _: ()) -> BufResult<()> {
        writer.write_u32_be(*self)
    }

    #[inline]
    fn decode<R: Buf>(reader: &mut Cursor<R>, _: ()) -> BufResult<Self> {
        reader.read_u32_be()
    }
}

impl Codec for u64 {
    #[inline]
    fn encode<W: BufMut>(&self, writer: &mut Cursor<W>, _: ()) -> BufResult<()> {
        writer.write_array(&self.to_be_bytes())
    }

    #[inline]
    fn decode<R: Buf>(reader: &mut Cursor<R>, _: ()) -> BufResult<Self> {
        let bytes = reader.read_array::<8>()?;
        Ok(u64::from_be_bytes(bytes))
    }
}

impl<const N: usize> Codec for [u8; N] {
    #[inline]
    fn encode<W: BufMut>(&self, writer: &mut Cursor<W>, _: ()) -> BufResult<()> {
        writer.write_array(self)
    }

    #[inline]
    fn decode<R: Buf>(reader: &mut Cursor<R>, _: ()) -> BufResult<Self> {
        reader.read_array::<N>()
    }
}

#[cfg(feature = "alloc")]
impl<T: Codec> Codec for Vec<T> {
    fn encode<W: BufMut>(&self, writer: &mut Cursor<W>, _: ()) -> BufResult<()> {
        for item in self.iter() {
            item.encode(writer, ())?;
        }
        Ok(())
    }

    fn decode<R: Buf>(reader: &mut Cursor<R>, _: ()) -> BufResult<Self> {
        let mut vec = Vec::new();

        while !reader.is_eof() {
            vec.push(T::decode(reader, ())?);
        }

        Ok(vec)
    }
}

#[cfg(feature = "alloc")]
impl<T: Codec> Codec<usize> for Vec<T> {
    fn encode<W: BufMut>(&self, writer: &mut Cursor<W>, _: usize) -> BufResult<()> {
        for item in self.iter() {
            item.encode(writer, ())?;
        }
        Ok(())
    }

    fn decode<R: Buf>(reader: &mut Cursor<R>, to_decode: usize) -> BufResult<Self> {
        let mut vec = Vec::new();

        for _ in 0..to_decode {
            if reader.is_eof() {
                use crate::BufError;

                return Err(BufError::OutOfBounds);
            }
            vec.push(T::decode(reader, ())?);
        }

        Ok(vec)
    }
}