use std::net::{IpAddr, Ipv4Addr};
use netlink_packet_core::{Emitable, Parseable};
use crate::{
link::{
link_flag::LinkFlags, AmtMode, InfoAmt, InfoData, InfoKind,
LinkAttribute, LinkHeader, LinkInfo, LinkLayerType, LinkMessage,
LinkMessageBuffer,
},
AddressFamily,
};
#[test]
fn test_parsing_link_amt_gateway() {
#[rustfmt::skip]
let raw = vec![
0x00, 0x00, 0x01, 0x00, 0x04, 0x00, 0x00, 0x00, 0x43, 0x10, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x48, 0x00, 0x12, 0x00,
0x08, 0x00, 0x01, 0x00, 0x61, 0x6d, 0x74, 0x00,
0x3c, 0x00, 0x02, 0x00,
0x08, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00,
0x06, 0x00, 0x02, 0x00, 0x08, 0xdc, 0x00, 0x00,
0x06, 0x00, 0x03, 0x00, 0x08, 0xdc, 0x00, 0x00,
0x08, 0x00, 0x04, 0x00, 0x03, 0x00, 0x00, 0x00,
0x08, 0x00, 0x05, 0x00, 0x0a, 0x00, 0x00, 0x01,
0x08, 0x00, 0x07, 0x00, 0x0a, 0x00, 0x00, 0x02,
0x08, 0x00, 0x08, 0x00, 0x80, 0x00, 0x00, 0x00,
];
let expected = LinkMessage {
header: LinkHeader {
interface_family: AddressFamily::Unspec,
index: 4,
link_layer_type: LinkLayerType::Ether,
flags: LinkFlags::Up
| LinkFlags::Broadcast
| LinkFlags::Running
| LinkFlags::Multicast
| LinkFlags::LowerUp,
change_mask: LinkFlags::empty(),
},
attributes: vec![LinkAttribute::LinkInfo(vec![
LinkInfo::Kind(InfoKind::Amt),
LinkInfo::Data(InfoData::Amt(vec![
InfoAmt::Mode(AmtMode::Gateway),
InfoAmt::RelayPort(2268),
InfoAmt::GatewayPort(2268),
InfoAmt::Link(3),
InfoAmt::LocalIp(IpAddr::V4(Ipv4Addr::new(10, 0, 0, 1))),
InfoAmt::DiscoveryIp(IpAddr::V4(Ipv4Addr::new(10, 0, 0, 2))),
InfoAmt::MaxTunnels(128),
])),
])],
};
assert_eq!(
expected,
LinkMessage::parse(&LinkMessageBuffer::new(&raw)).unwrap()
);
let mut buf = vec![0; expected.buffer_len()];
expected.emit(&mut buf);
assert_eq!(buf, raw);
}
#[test]
fn test_parsing_link_amt_relay() {
#[rustfmt::skip]
let raw = vec![
0x00, 0x00, 0x01, 0x00,
0x05, 0x00, 0x00, 0x00,
0x43, 0x10, 0x01, 0x00,
0x00, 0x00, 0x00, 0x00,
0x30, 0x00, 0x12, 0x00,
0x08, 0x00, 0x01, 0x00, 0x61, 0x6d, 0x74, 0x00,
0x24, 0x00, 0x02, 0x00,
0x08, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x00,
0x06, 0x00, 0x02, 0x00, 0x08, 0xdc, 0x00, 0x00,
0x08, 0x00, 0x04, 0x00, 0x05, 0x00, 0x00, 0x00,
0x08, 0x00, 0x05, 0x00, 0x0a, 0x00, 0x00, 0x02,
];
let expected = LinkMessage {
header: LinkHeader {
interface_family: AddressFamily::Unspec,
index: 5,
link_layer_type: LinkLayerType::Ether,
flags: LinkFlags::Up
| LinkFlags::Broadcast
| LinkFlags::Running
| LinkFlags::Multicast
| LinkFlags::LowerUp,
change_mask: LinkFlags::empty(),
},
attributes: vec![LinkAttribute::LinkInfo(vec![
LinkInfo::Kind(InfoKind::Amt),
LinkInfo::Data(InfoData::Amt(vec![
InfoAmt::Mode(AmtMode::Relay),
InfoAmt::RelayPort(2268),
InfoAmt::Link(5),
InfoAmt::LocalIp(IpAddr::V4(Ipv4Addr::new(10, 0, 0, 2))),
])),
])],
};
assert_eq!(
expected,
LinkMessage::parse(&LinkMessageBuffer::new(&raw)).unwrap()
);
let mut buf = vec![0; expected.buffer_len()];
expected.emit(&mut buf);
assert_eq!(buf, raw);
}
#[test]
fn test_parsing_link_amt_minimal() {
#[rustfmt::skip]
let raw = vec![
0x00, 0x00, 0x01, 0x00,
0x07, 0x00, 0x00, 0x00,
0x43, 0x10, 0x01, 0x00,
0x00, 0x00, 0x00, 0x00,
0x20, 0x00, 0x12, 0x00,
0x08, 0x00, 0x01, 0x00, 0x61, 0x6d, 0x74, 0x00,
0x14, 0x00, 0x02, 0x00,
0x08, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00,
0x08, 0x00, 0x04, 0x00, 0x07, 0x00, 0x00, 0x00,
];
let expected = LinkMessage {
header: LinkHeader {
interface_family: AddressFamily::Unspec,
index: 7,
link_layer_type: LinkLayerType::Ether,
flags: LinkFlags::Up
| LinkFlags::Broadcast
| LinkFlags::Running
| LinkFlags::Multicast
| LinkFlags::LowerUp,
change_mask: LinkFlags::empty(),
},
attributes: vec![LinkAttribute::LinkInfo(vec![
LinkInfo::Kind(InfoKind::Amt),
LinkInfo::Data(InfoData::Amt(vec![
InfoAmt::Mode(AmtMode::Gateway),
InfoAmt::Link(7),
])),
])],
};
assert_eq!(
expected,
LinkMessage::parse(&LinkMessageBuffer::new(&raw)).unwrap()
);
let mut buf = vec![0; expected.buffer_len()];
expected.emit(&mut buf);
assert_eq!(buf, raw);
}