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

//! CONNACK

use std::io::{Read, Write};

use control::{FixedHeader, PacketType, ControlType};
use control::variable_header::{ConnackFlags, ConnectReturnCode};
use packet::{Packet, PacketError};
use {Encodable, Decodable};

/// `CONNACK` packet
#[derive(Debug, Eq, PartialEq)]
pub struct ConnackPacket {
    fixed_header: FixedHeader,
    flags: ConnackFlags,
    ret_code: ConnectReturnCode,
    payload: (),
}

impl ConnackPacket {
    pub fn new(session_present: bool, ret_code: ConnectReturnCode) -> ConnackPacket {
        ConnackPacket {
            fixed_header: FixedHeader::new(PacketType::with_default(ControlType::ConnectAcknowledgement),
                                           2),
            flags: ConnackFlags { session_present: session_present },
            ret_code: ret_code,
            payload: (),
        }
    }

    pub fn connack_flags(&self) -> ConnackFlags {
        self.flags
    }

    pub fn connect_return_code(&self) -> ConnectReturnCode {
        self.ret_code
    }
}

impl<'a> Packet<'a> for ConnackPacket {
    type Payload = ();

    fn fixed_header(&self) -> &FixedHeader {
        &self.fixed_header
    }

    fn payload(&self) -> &Self::Payload {
        &self.payload
    }

    fn encode_variable_headers<W: Write>(&self, writer: &mut W) -> Result<(), PacketError<'a, Self>> {
        try!(self.flags.encode(writer));
        try!(self.ret_code.encode(writer));
        Ok(())
    }

    fn encoded_variable_headers_length(&self) -> u32 {
        self.flags.encoded_length() + self.ret_code.encoded_length()
    }

    fn decode_packet<R: Read>(reader: &mut R, fixed_header: FixedHeader) -> Result<Self, PacketError<'a, Self>> {
        let flags: ConnackFlags = try!(Decodable::decode(reader));
        let code: ConnectReturnCode = try!(Decodable::decode(reader));

        Ok(ConnackPacket {
               fixed_header: fixed_header,
               flags: flags,
               ret_code: code,
               payload: (),
           })
    }
}

#[cfg(test)]
mod test {
    use super::*;

    use std::io::Cursor;

    use control::variable_header::ConnectReturnCode;
    use {Encodable, Decodable};

    #[test]
    pub fn test_connack_packet_basic() {
        let packet = ConnackPacket::new(false, ConnectReturnCode::IdentifierRejected);

        let mut buf = Vec::new();
        packet.encode(&mut buf).unwrap();

        let mut decode_buf = Cursor::new(buf);
        let decoded = ConnackPacket::decode(&mut decode_buf).unwrap();

        assert_eq!(packet, decoded);
    }
}