use anyhow::Context;
use netlink_packet_utils::{
DecodeError, Emitable, Parseable, ParseableParametrized,
};
use netlink_packet_core::{
NetlinkDeserializable, NetlinkHeader, NetlinkPayload, NetlinkSerializable,
};
use crate::{
address::{AddressHeader, AddressMessage, AddressMessageBuffer},
link::{LinkMessage, LinkMessageBuffer},
neighbour::{NeighbourMessage, NeighbourMessageBuffer},
neighbour_table::{NeighbourTableMessage, NeighbourTableMessageBuffer},
nsid::{NsidMessage, NsidMessageBuffer},
route::{RouteHeader, RouteMessage, RouteMessageBuffer},
rule::{RuleMessage, RuleMessageBuffer},
tc::{TcMessage, TcMessageBuffer},
};
const RTM_NEWLINK: u16 = 16;
const RTM_DELLINK: u16 = 17;
const RTM_GETLINK: u16 = 18;
const RTM_SETLINK: u16 = 19;
const RTM_NEWADDR: u16 = 20;
const RTM_DELADDR: u16 = 21;
const RTM_GETADDR: u16 = 22;
const RTM_NEWROUTE: u16 = 24;
const RTM_DELROUTE: u16 = 25;
const RTM_GETROUTE: u16 = 26;
const RTM_NEWNEIGH: u16 = 28;
const RTM_DELNEIGH: u16 = 29;
const RTM_GETNEIGH: u16 = 30;
const RTM_NEWRULE: u16 = 32;
const RTM_DELRULE: u16 = 33;
const RTM_GETRULE: u16 = 34;
const RTM_NEWQDISC: u16 = 36;
const RTM_DELQDISC: u16 = 37;
const RTM_GETQDISC: u16 = 38;
const RTM_NEWTCLASS: u16 = 40;
const RTM_DELTCLASS: u16 = 41;
const RTM_GETTCLASS: u16 = 42;
const RTM_NEWTFILTER: u16 = 44;
const RTM_DELTFILTER: u16 = 45;
const RTM_GETTFILTER: u16 = 46;
const RTM_NEWNEIGHTBL: u16 = 64;
const RTM_GETNEIGHTBL: u16 = 66;
const RTM_SETNEIGHTBL: u16 = 67;
const RTM_NEWNSID: u16 = 88;
const RTM_DELNSID: u16 = 89;
const RTM_GETNSID: u16 = 90;
const RTM_NEWCHAIN: u16 = 100;
const RTM_DELCHAIN: u16 = 101;
const RTM_GETCHAIN: u16 = 102;
const RTM_NEWLINKPROP: u16 = 108;
const RTM_DELLINKPROP: u16 = 109;
buffer!(RouteNetlinkMessageBuffer);
impl<'a, T: AsRef<[u8]> + ?Sized>
ParseableParametrized<RouteNetlinkMessageBuffer<&'a T>, u16>
for RouteNetlinkMessage
{
fn parse_with_param(
buf: &RouteNetlinkMessageBuffer<&'a T>,
message_type: u16,
) -> Result<Self, DecodeError> {
let message = match message_type {
RTM_NEWLINK | RTM_GETLINK | RTM_DELLINK | RTM_SETLINK => {
let msg = match LinkMessageBuffer::new_checked(&buf.inner()) {
Ok(buf) => LinkMessage::parse(&buf)
.context("invalid link message")?,
Err(e) => {
if buf.inner().len() == 4 && message_type == RTM_GETLINK
{
let mut msg = LinkMessage::default();
msg.header.interface_family = buf.inner()[0].into();
msg
} else {
return Err(e);
}
}
};
match message_type {
RTM_NEWLINK => RouteNetlinkMessage::NewLink(msg),
RTM_GETLINK => RouteNetlinkMessage::GetLink(msg),
RTM_DELLINK => RouteNetlinkMessage::DelLink(msg),
RTM_SETLINK => RouteNetlinkMessage::SetLink(msg),
_ => unreachable!(),
}
}
RTM_NEWADDR | RTM_GETADDR | RTM_DELADDR => {
let msg = match AddressMessageBuffer::new_checked(&buf.inner())
{
Ok(buf) => AddressMessage::parse(&buf)
.context("invalid link message")?,
Err(e) => {
if buf.inner().len() == 4 && message_type == RTM_GETADDR
{
let mut msg = AddressMessage {
header: AddressHeader::default(),
attributes: vec![],
};
msg.header.family = buf.inner()[0].into();
msg
} else {
return Err(e);
}
}
};
match message_type {
RTM_NEWADDR => RouteNetlinkMessage::NewAddress(msg),
RTM_GETADDR => RouteNetlinkMessage::GetAddress(msg),
RTM_DELADDR => RouteNetlinkMessage::DelAddress(msg),
_ => unreachable!(),
}
}
RTM_NEWNEIGH | RTM_GETNEIGH | RTM_DELNEIGH => {
let err = "invalid neighbour message";
let msg = NeighbourMessage::parse(
&NeighbourMessageBuffer::new_checked(&buf.inner())
.context(err)?,
)
.context(err)?;
match message_type {
RTM_GETNEIGH => RouteNetlinkMessage::GetNeighbour(msg),
RTM_NEWNEIGH => RouteNetlinkMessage::NewNeighbour(msg),
RTM_DELNEIGH => RouteNetlinkMessage::DelNeighbour(msg),
_ => unreachable!(),
}
}
RTM_NEWNEIGHTBL | RTM_GETNEIGHTBL | RTM_SETNEIGHTBL => {
let err = "invalid neighbour table message";
let msg = NeighbourTableMessage::parse(
&NeighbourTableMessageBuffer::new_checked(&buf.inner())
.context(err)?,
)
.context(err)?;
match message_type {
RTM_GETNEIGHTBL => {
RouteNetlinkMessage::GetNeighbourTable(msg)
}
RTM_NEWNEIGHTBL => {
RouteNetlinkMessage::NewNeighbourTable(msg)
}
RTM_SETNEIGHTBL => {
RouteNetlinkMessage::SetNeighbourTable(msg)
}
_ => unreachable!(),
}
}
RTM_NEWROUTE | RTM_GETROUTE | RTM_DELROUTE => {
let msg = match RouteMessageBuffer::new_checked(&buf.inner()) {
Ok(buf) => RouteMessage::parse(&buf)
.context("invalid route message")?,
Err(e) => {
if (buf.inner().len() == 4 || buf.inner().len() == 1)
&& message_type == RTM_GETROUTE
{
let mut msg = RouteMessage {
header: RouteHeader::default(),
attributes: vec![],
};
msg.header.address_family = buf.inner()[0].into();
msg
} else {
return Err(e);
}
}
};
match message_type {
RTM_NEWROUTE => RouteNetlinkMessage::NewRoute(msg),
RTM_GETROUTE => RouteNetlinkMessage::GetRoute(msg),
RTM_DELROUTE => RouteNetlinkMessage::DelRoute(msg),
_ => unreachable!(),
}
}
RTM_NEWRULE | RTM_GETRULE | RTM_DELRULE => {
let err = "invalid fib rule message";
let msg = RuleMessage::parse(
&RuleMessageBuffer::new_checked(&buf.inner())
.context(err)?,
)
.context(err)?;
match message_type {
RTM_NEWRULE => RouteNetlinkMessage::NewRule(msg),
RTM_DELRULE => RouteNetlinkMessage::DelRule(msg),
RTM_GETRULE => RouteNetlinkMessage::GetRule(msg),
_ => unreachable!(),
}
}
RTM_NEWQDISC | RTM_DELQDISC | RTM_GETQDISC | RTM_NEWTCLASS
| RTM_DELTCLASS | RTM_GETTCLASS | RTM_NEWTFILTER
| RTM_DELTFILTER | RTM_GETTFILTER | RTM_NEWCHAIN | RTM_DELCHAIN
| RTM_GETCHAIN => {
let err = "invalid tc message";
let msg = TcMessage::parse(
&TcMessageBuffer::new_checked(&buf.inner()).context(err)?,
)
.context(err)?;
match message_type {
RTM_NEWQDISC => {
RouteNetlinkMessage::NewQueueDiscipline(msg)
}
RTM_DELQDISC => {
RouteNetlinkMessage::DelQueueDiscipline(msg)
}
RTM_GETQDISC => {
RouteNetlinkMessage::GetQueueDiscipline(msg)
}
RTM_NEWTCLASS => RouteNetlinkMessage::NewTrafficClass(msg),
RTM_DELTCLASS => RouteNetlinkMessage::DelTrafficClass(msg),
RTM_GETTCLASS => RouteNetlinkMessage::GetTrafficClass(msg),
RTM_NEWTFILTER => {
RouteNetlinkMessage::NewTrafficFilter(msg)
}
RTM_DELTFILTER => {
RouteNetlinkMessage::DelTrafficFilter(msg)
}
RTM_GETTFILTER => {
RouteNetlinkMessage::GetTrafficFilter(msg)
}
RTM_NEWCHAIN => RouteNetlinkMessage::NewTrafficChain(msg),
RTM_DELCHAIN => RouteNetlinkMessage::DelTrafficChain(msg),
RTM_GETCHAIN => RouteNetlinkMessage::GetTrafficChain(msg),
_ => unreachable!(),
}
}
RTM_NEWNSID | RTM_GETNSID | RTM_DELNSID => {
let err = "invalid nsid message";
let msg = NsidMessage::parse(
&NsidMessageBuffer::new_checked(&buf.inner())
.context(err)?,
)
.context(err)?;
match message_type {
RTM_NEWNSID => RouteNetlinkMessage::NewNsId(msg),
RTM_DELNSID => RouteNetlinkMessage::DelNsId(msg),
RTM_GETNSID => RouteNetlinkMessage::GetNsId(msg),
_ => unreachable!(),
}
}
_ => {
return Err(
format!("Unknown message type: {message_type}").into()
)
}
};
Ok(message)
}
}
#[derive(Debug, PartialEq, Eq, Clone)]
#[non_exhaustive]
pub enum RouteNetlinkMessage {
NewLink(LinkMessage),
DelLink(LinkMessage),
GetLink(LinkMessage),
SetLink(LinkMessage),
NewLinkProp(LinkMessage),
DelLinkProp(LinkMessage),
NewAddress(AddressMessage),
DelAddress(AddressMessage),
GetAddress(AddressMessage),
NewNeighbour(NeighbourMessage),
GetNeighbour(NeighbourMessage),
DelNeighbour(NeighbourMessage),
NewNeighbourTable(NeighbourTableMessage),
GetNeighbourTable(NeighbourTableMessage),
SetNeighbourTable(NeighbourTableMessage),
NewRoute(RouteMessage),
DelRoute(RouteMessage),
GetRoute(RouteMessage),
NewQueueDiscipline(TcMessage),
DelQueueDiscipline(TcMessage),
GetQueueDiscipline(TcMessage),
NewTrafficClass(TcMessage),
DelTrafficClass(TcMessage),
GetTrafficClass(TcMessage),
NewTrafficFilter(TcMessage),
DelTrafficFilter(TcMessage),
GetTrafficFilter(TcMessage),
NewTrafficChain(TcMessage),
DelTrafficChain(TcMessage),
GetTrafficChain(TcMessage),
NewNsId(NsidMessage),
DelNsId(NsidMessage),
GetNsId(NsidMessage),
NewRule(RuleMessage),
DelRule(RuleMessage),
GetRule(RuleMessage),
}
impl RouteNetlinkMessage {
pub fn is_new_link(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewLink(_))
}
pub fn is_del_link(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelLink(_))
}
pub fn is_get_link(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetLink(_))
}
pub fn is_set_link(&self) -> bool {
matches!(self, RouteNetlinkMessage::SetLink(_))
}
pub fn is_new_address(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewAddress(_))
}
pub fn is_del_address(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelAddress(_))
}
pub fn is_get_address(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetAddress(_))
}
pub fn is_get_neighbour(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetNeighbour(_))
}
pub fn is_new_route(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewRoute(_))
}
pub fn is_new_neighbour(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewNeighbour(_))
}
pub fn is_get_route(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetRoute(_))
}
pub fn is_del_neighbour(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelNeighbour(_))
}
pub fn is_new_neighbour_table(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewNeighbourTable(_))
}
pub fn is_get_neighbour_table(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetNeighbourTable(_))
}
pub fn is_set_neighbour_table(&self) -> bool {
matches!(self, RouteNetlinkMessage::SetNeighbourTable(_))
}
pub fn is_del_route(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelRoute(_))
}
pub fn is_new_qdisc(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewQueueDiscipline(_))
}
pub fn is_del_qdisc(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelQueueDiscipline(_))
}
pub fn is_get_qdisc(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetQueueDiscipline(_))
}
pub fn is_new_class(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewTrafficClass(_))
}
pub fn is_del_class(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelTrafficClass(_))
}
pub fn is_get_class(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetTrafficClass(_))
}
pub fn is_new_filter(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewTrafficFilter(_))
}
pub fn is_del_filter(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelTrafficFilter(_))
}
pub fn is_get_filter(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetTrafficFilter(_))
}
pub fn is_new_chain(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewTrafficChain(_))
}
pub fn is_del_chain(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelTrafficChain(_))
}
pub fn is_get_chain(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetTrafficChain(_))
}
pub fn is_new_nsid(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewNsId(_))
}
pub fn is_get_nsid(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetNsId(_))
}
pub fn is_del_nsid(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelNsId(_))
}
pub fn is_get_rule(&self) -> bool {
matches!(self, RouteNetlinkMessage::GetRule(_))
}
pub fn is_new_rule(&self) -> bool {
matches!(self, RouteNetlinkMessage::NewRule(_))
}
pub fn is_del_rule(&self) -> bool {
matches!(self, RouteNetlinkMessage::DelRule(_))
}
pub fn message_type(&self) -> u16 {
use self::RouteNetlinkMessage::*;
match self {
NewLink(_) => RTM_NEWLINK,
DelLink(_) => RTM_DELLINK,
GetLink(_) => RTM_GETLINK,
SetLink(_) => RTM_SETLINK,
NewLinkProp(_) => RTM_NEWLINKPROP,
DelLinkProp(_) => RTM_DELLINKPROP,
NewAddress(_) => RTM_NEWADDR,
DelAddress(_) => RTM_DELADDR,
GetAddress(_) => RTM_GETADDR,
GetNeighbour(_) => RTM_GETNEIGH,
NewNeighbour(_) => RTM_NEWNEIGH,
DelNeighbour(_) => RTM_DELNEIGH,
GetNeighbourTable(_) => RTM_GETNEIGHTBL,
NewNeighbourTable(_) => RTM_NEWNEIGHTBL,
SetNeighbourTable(_) => RTM_SETNEIGHTBL,
NewRoute(_) => RTM_NEWROUTE,
DelRoute(_) => RTM_DELROUTE,
GetRoute(_) => RTM_GETROUTE,
NewQueueDiscipline(_) => RTM_NEWQDISC,
DelQueueDiscipline(_) => RTM_DELQDISC,
GetQueueDiscipline(_) => RTM_GETQDISC,
NewTrafficClass(_) => RTM_NEWTCLASS,
DelTrafficClass(_) => RTM_DELTCLASS,
GetTrafficClass(_) => RTM_GETTCLASS,
NewTrafficFilter(_) => RTM_NEWTFILTER,
DelTrafficFilter(_) => RTM_DELTFILTER,
GetTrafficFilter(_) => RTM_GETTFILTER,
NewTrafficChain(_) => RTM_NEWCHAIN,
DelTrafficChain(_) => RTM_DELCHAIN,
GetTrafficChain(_) => RTM_GETCHAIN,
GetNsId(_) => RTM_GETNSID,
NewNsId(_) => RTM_NEWNSID,
DelNsId(_) => RTM_DELNSID,
GetRule(_) => RTM_GETRULE,
NewRule(_) => RTM_NEWRULE,
DelRule(_) => RTM_DELRULE,
}
}
}
impl Emitable for RouteNetlinkMessage {
#[rustfmt::skip]
fn buffer_len(&self) -> usize {
use self::RouteNetlinkMessage::*;
match self {
| NewLink(ref msg)
| DelLink(ref msg)
| GetLink(ref msg)
| SetLink(ref msg)
| NewLinkProp(ref msg)
| DelLinkProp(ref msg)
=> msg.buffer_len(),
| NewAddress(ref msg)
| DelAddress(ref msg)
| GetAddress(ref msg)
=> msg.buffer_len(),
| NewNeighbour(ref msg)
| GetNeighbour(ref msg)
| DelNeighbour(ref msg)
=> msg.buffer_len(),
| NewNeighbourTable(ref msg)
| GetNeighbourTable(ref msg)
| SetNeighbourTable(ref msg)
=> msg.buffer_len(),
| NewRoute(ref msg)
| DelRoute(ref msg)
| GetRoute(ref msg)
=> msg.buffer_len(),
| NewQueueDiscipline(ref msg)
| DelQueueDiscipline(ref msg)
| GetQueueDiscipline(ref msg)
| NewTrafficClass(ref msg)
| DelTrafficClass(ref msg)
| GetTrafficClass(ref msg)
| NewTrafficFilter(ref msg)
| DelTrafficFilter(ref msg)
| GetTrafficFilter(ref msg)
| NewTrafficChain(ref msg)
| DelTrafficChain(ref msg)
| GetTrafficChain(ref msg)
=> msg.buffer_len(),
| NewNsId(ref msg)
| DelNsId(ref msg)
| GetNsId(ref msg)
=> msg.buffer_len(),
| NewRule(ref msg)
| DelRule(ref msg)
| GetRule(ref msg)
=> msg.buffer_len()
}
}
#[rustfmt::skip]
fn emit(&self, buffer: &mut [u8]) {
use self::RouteNetlinkMessage::*;
match self {
| NewLink(ref msg)
| DelLink(ref msg)
| GetLink(ref msg)
| SetLink(ref msg)
| NewLinkProp(ref msg)
| DelLinkProp(ref msg)
=> msg.emit(buffer),
| NewAddress(ref msg)
| DelAddress(ref msg)
| GetAddress(ref msg)
=> msg.emit(buffer),
| GetNeighbour(ref msg)
| NewNeighbour(ref msg)
| DelNeighbour(ref msg)
=> msg.emit(buffer),
| GetNeighbourTable(ref msg)
| NewNeighbourTable(ref msg)
| SetNeighbourTable(ref msg)
=> msg.emit(buffer),
| NewRoute(ref msg)
| DelRoute(ref msg)
| GetRoute(ref msg)
=> msg.emit(buffer),
| NewQueueDiscipline(ref msg)
| DelQueueDiscipline(ref msg)
| GetQueueDiscipline(ref msg)
| NewTrafficClass(ref msg)
| DelTrafficClass(ref msg)
| GetTrafficClass(ref msg)
| NewTrafficFilter(ref msg)
| DelTrafficFilter(ref msg)
| GetTrafficFilter(ref msg)
| NewTrafficChain(ref msg)
| DelTrafficChain(ref msg)
| GetTrafficChain(ref msg)
=> msg.emit(buffer),
| NewNsId(ref msg)
| DelNsId(ref msg)
| GetNsId(ref msg)
=> msg.emit(buffer),
| NewRule(ref msg)
| DelRule(ref msg)
| GetRule(ref msg)
=> msg.emit(buffer)
}
}
}
impl NetlinkSerializable for RouteNetlinkMessage {
fn message_type(&self) -> u16 {
self.message_type()
}
fn buffer_len(&self) -> usize {
<Self as Emitable>::buffer_len(self)
}
fn serialize(&self, buffer: &mut [u8]) {
self.emit(buffer)
}
}
impl NetlinkDeserializable for RouteNetlinkMessage {
type Error = DecodeError;
fn deserialize(
header: &NetlinkHeader,
payload: &[u8],
) -> Result<Self, Self::Error> {
let buf = RouteNetlinkMessageBuffer::new(payload);
match RouteNetlinkMessage::parse_with_param(&buf, header.message_type) {
Err(e) => Err(e),
Ok(message) => Ok(message),
}
}
}
impl From<RouteNetlinkMessage> for NetlinkPayload<RouteNetlinkMessage> {
fn from(message: RouteNetlinkMessage) -> Self {
NetlinkPayload::InnerMessage(message)
}
}