use byteorder::{BigEndian, ReadBytesExt};
use std::io::{Error, Read};
use std::net::{IpAddr, Ipv4Addr, Ipv6Addr};
use crate::Header;
use crate::AFI;
#[derive(Debug)]
pub struct OSPFv2 {
pub remote: Ipv4Addr,
pub local: Ipv4Addr,
pub message: Vec<u8>,
}
impl OSPFv2 {
pub fn parse(header: &Header, stream: &mut Read) -> Result<OSPFv2, Error> {
let length = (header.length - 2 * AFI::IPV4.size()) as usize;
let mut record = OSPFv2 {
remote: Ipv4Addr::from(stream.read_u32::<BigEndian>()?),
local: Ipv4Addr::from(stream.read_u32::<BigEndian>()?),
message: vec![0; length as usize],
};
stream.read_exact(&mut record.message)?;
Ok(record)
}
}
#[derive(Debug)]
pub struct OSPFv3 {
pub remote: IpAddr,
pub local: IpAddr,
pub message: Vec<u8>,
}
impl OSPFv3 {
pub fn parse(header: &Header, stream: &mut Read) -> Result<OSPFv3, Error> {
let mut record = match AFI::from(stream.read_u16::<BigEndian>()?)? {
AFI::IPV4 => {
let length = (header.length - 2 * AFI::IPV4.size()) as usize;
OSPFv3 {
remote: IpAddr::V4(Ipv4Addr::from(stream.read_u32::<BigEndian>()?)),
local: IpAddr::V4(Ipv4Addr::from(stream.read_u32::<BigEndian>()?)),
message: vec![0; length as usize],
}
}
AFI::IPV6 => {
let length: usize = (header.length - 2 * AFI::IPV6.size()) as usize;
OSPFv3 {
remote: IpAddr::V6(Ipv6Addr::from(stream.read_u128::<BigEndian>()?)),
local: IpAddr::V6(Ipv6Addr::from(stream.read_u128::<BigEndian>()?)),
message: vec![0; length as usize],
}
}
};
stream.read_exact(&mut record.message)?;
Ok(record)
}
}