use std::io::Cursor;
use byteorder::{BigEndian, ReadBytesExt, WriteBytesExt};
use crate::error::Result;
use crate::net::constants::ACKED_PACKET_HEADER;
use super::{HeaderReader, HeaderWriter};
#[derive(Copy, Clone, Debug)]
pub struct AckedPacketHeader {
pub seq: u16,
ack_seq: u16,
ack_field: u32,
}
impl AckedPacketHeader {
pub fn new(seq_num: u16, last_seq: u16, bit_field: u32) -> AckedPacketHeader {
AckedPacketHeader {
seq: seq_num,
ack_seq: last_seq,
ack_field: bit_field,
}
}
#[allow(dead_code)]
pub fn sequence(&self) -> u16 {
self.seq
}
pub fn ack_field(&self) -> u32 {
self.ack_field
}
pub fn ack_seq(&self) -> u16 {
self.ack_seq
}
}
impl HeaderWriter for AckedPacketHeader {
type Output = Result<()>;
fn parse(&self, buffer: &mut Vec<u8>) -> Self::Output {
buffer.write_u16::<BigEndian>(self.seq)?;
buffer.write_u16::<BigEndian>(self.ack_seq)?;
buffer.write_u32::<BigEndian>(self.ack_field)?;
Ok(())
}
}
impl HeaderReader for AckedPacketHeader {
type Header = Result<AckedPacketHeader>;
fn read(rdr: &mut Cursor<&[u8]>) -> Self::Header {
let seq = rdr.read_u16::<BigEndian>()?;
let ack_seq = rdr.read_u16::<BigEndian>()?;
let ack_field = rdr.read_u32::<BigEndian>()?;
Ok(AckedPacketHeader {
seq,
ack_seq,
ack_field,
})
}
fn size() -> u8 {
ACKED_PACKET_HEADER
}
}
#[cfg(test)]
mod tests {
use std::io::Cursor;
use crate::net::constants::ACKED_PACKET_HEADER;
use crate::packet::header::{AckedPacketHeader, HeaderReader, HeaderWriter};
#[test]
fn serialize() {
let mut buffer = Vec::new();
let header = AckedPacketHeader::new(1, 2, 3);
assert![header.parse(&mut buffer).is_ok()];
assert_eq!(buffer[1], 1);
assert_eq!(buffer[3], 2);
assert_eq!(buffer[7], 3);
assert_eq!(buffer.len() as u8, AckedPacketHeader::size());
}
#[test]
fn deserialize() {
let buffer = vec![0, 1, 0, 2, 0, 0, 0, 3];
let mut cursor = Cursor::new(buffer.as_slice());
let header = AckedPacketHeader::read(&mut cursor).unwrap();
assert_eq!(header.sequence(), 1);
assert_eq!(header.ack_seq(), 2);
assert_eq!(header.ack_field(), 3);
}
#[test]
fn size() {
assert_eq!(AckedPacketHeader::size(), ACKED_PACKET_HEADER);
}
}