mod peer_index_table;
mod rib_afi_entries;
use crate::error::ParserError;
use crate::messages::table_dump_v2::peer_index_table::parse_peer_index_table;
use crate::messages::table_dump_v2::rib_afi_entries::parse_rib_afi_entries;
use crate::models::*;
use bytes::Bytes;
use std::convert::TryFrom;
pub fn parse_table_dump_v2_message(
sub_type: u16,
mut input: Bytes,
) -> Result<TableDumpV2Message, ParserError> {
let v2_type: TableDumpV2Type = TableDumpV2Type::try_from(sub_type)?;
let msg: TableDumpV2Message = match v2_type {
TableDumpV2Type::PeerIndexTable => {
TableDumpV2Message::PeerIndexTable(parse_peer_index_table(&mut input)?)
}
TableDumpV2Type::RibIpv4Unicast
| TableDumpV2Type::RibIpv4Multicast
| TableDumpV2Type::RibIpv6Unicast
| TableDumpV2Type::RibIpv6Multicast
| TableDumpV2Type::RibIpv4UnicastAddPath
| TableDumpV2Type::RibIpv4MulticastAddPath
| TableDumpV2Type::RibIpv6UnicastAddPath
| TableDumpV2Type::RibIpv6MulticastAddPath => {
TableDumpV2Message::RibAfi(parse_rib_afi_entries(&mut input, v2_type)?)
}
TableDumpV2Type::RibGeneric
| TableDumpV2Type::RibGenericAddPath
| TableDumpV2Type::GeoPeerTable => {
return Err(ParserError::Unsupported(
"TableDumpV2 RibGeneric and GeoPeerTable is not currently supported".to_string(),
))
}
};
Ok(msg)
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_unsupported_type() {
let msg = parse_table_dump_v2_message(7, Bytes::new());
assert!(msg.is_err());
}
}