netlink-packet-core 0.9.0

// SPDX-License-Identifier: MIT

use std::{
    mem::size_of,
    net::{IpAddr, Ipv4Addr, Ipv6Addr},
};

use crate::DecodeError;

pub fn parse_mac(payload: &[u8]) -> Result<[u8; 6], DecodeError> {
    if payload.len() != 6 {
        return Err(DecodeError::invalid_mac_address(payload.len()));
    }
    let mut address: [u8; 6] = [0; 6];
    for (i, byte) in payload.iter().enumerate() {
        address[i] = *byte;
    }
    Ok(address)
}

pub fn parse_ipv6(payload: &[u8]) -> Result<[u8; 16], DecodeError> {
    if payload.len() != 16 {
        return Err(DecodeError::invalid_ip_address(payload.len()));
    }
    let mut address: [u8; 16] = [0; 16];
    for (i, byte) in payload.iter().enumerate() {
        address[i] = *byte;
    }
    Ok(address)
}

pub fn parse_ip(payload: &[u8]) -> Result<IpAddr, DecodeError> {
    match payload.len() {
        4 => Ok(
            Ipv4Addr::new(payload[0], payload[1], payload[2], payload[3])
                .into(),
        ),
        16 => Ok(Ipv6Addr::from([
            payload[0],
            payload[1],
            payload[2],
            payload[3],
            payload[4],
            payload[5],
            payload[6],
            payload[7],
            payload[8],
            payload[9],
            payload[10],
            payload[11],
            payload[12],
            payload[13],
            payload[14],
            payload[15],
        ])
        .into()),
        other => Err(DecodeError::invalid_ip_address(other)),
    }
}

pub fn parse_string(payload: &[u8]) -> Result<String, DecodeError> {
    if payload.is_empty() {
        return Ok(String::new());
    }
    // iproute2 is a bit inconsistent with null-terminated strings.
    let slice = if payload[payload.len() - 1] == 0 {
        &payload[..payload.len() - 1]
    } else {
        &payload[..payload.len()]
    };
    let s = String::from_utf8(slice.to_vec())?;
    Ok(s)
}

pub fn parse_u8(payload: &[u8]) -> Result<u8, DecodeError> {
    if payload.len() != 1 {
        return Err(DecodeError::invalid_number(1, payload.len()));
    }
    Ok(payload[0])
}

pub fn parse_i8(payload: &[u8]) -> Result<i8, DecodeError> {
    if payload.len() != 1 {
        return Err(DecodeError::invalid_number(1, payload.len()));
    }
    Ok(payload[0] as i8)
}

macro_rules! gen_int_parser {
    (
        $(
            $parse:ident, $parse_be:ident, $parse_le:ident,
            $emit:ident, $emit_le:ident, $emit_be:ident,
            $data_type:ty,
        )+
    ) => {
        $(
            pub fn $parse(payload: &[u8]) -> Result<$data_type, DecodeError> {
                if payload.len() != size_of::<$data_type>() {
                    return Err(DecodeError::invalid_number(
                        size_of::<$data_type>(),
                        payload.len(),
                    ));
                }
                let mut data = [0u8; size_of::<$data_type>()];
                data.copy_from_slice(payload);
                Ok(<$data_type>::from_ne_bytes(data))
            }

            pub fn $parse_be(payload: &[u8]) -> Result<$data_type, DecodeError> {
                if payload.len() != size_of::<$data_type>() {
                    return Err(DecodeError::invalid_number(
                        size_of::<$data_type>(),
                        payload.len(),
                    ));
                }
                let mut data = [0u8; size_of::<$data_type>()];
                data.copy_from_slice(payload);
                Ok(<$data_type>::from_be_bytes(data))
            }

            pub fn $parse_le(payload: &[u8]) -> Result<$data_type, DecodeError> {
                if payload.len() != size_of::<$data_type>() {
                    return Err(DecodeError::invalid_number(
                        size_of::<$data_type>(),
                        payload.len(),
                    ));
                }
                let mut data = [0u8; size_of::<$data_type>()];
                data.copy_from_slice(payload);
                Ok(<$data_type>::from_le_bytes(data))
            }

            pub fn $emit(
                buf: &mut [u8],
                value: $data_type,
            ) -> Result<(), DecodeError> {
                if buf.len() < size_of::<$data_type>() {
                    return Err(DecodeError::buffer_too_small(
                        buf.len(),
                        size_of::<$data_type>(),
                    ));
                }
                buf[..size_of::<$data_type>()]
                    .copy_from_slice(&value.to_ne_bytes());
                Ok(())
            }

            pub fn $emit_le(
                buf: &mut [u8],
                value: $data_type,
            ) -> Result<(), DecodeError> {
                if buf.len() < size_of::<$data_type>() {
                    return Err(DecodeError::buffer_too_small(
                        buf.len(),
                        size_of::<$data_type>(),
                    ));
                }
                buf[..size_of::<$data_type>()]
                    .copy_from_slice(&value.to_le_bytes());
                Ok(())
            }

            pub fn $emit_be(
                buf: &mut [u8],
                value: $data_type,
            ) -> Result<(), DecodeError> {
                if buf.len() < size_of::<$data_type>() {
                    return Err(DecodeError::buffer_too_small(
                        buf.len(),
                        size_of::<$data_type>(),
                    ));
                }
                buf[..size_of::<$data_type>()]
                    .copy_from_slice(&value.to_be_bytes());
                Ok(())
            }
        )+
    }
}

gen_int_parser!(
    parse_u16,
    parse_u16_be,
    parse_u16_le,
    emit_u16,
    emit_u16_le,
    emit_u16_be,
    u16,
    parse_u32,
    parse_u32_be,
    parse_u32_le,
    emit_u32,
    emit_u32_le,
    emit_u32_be,
    u32,
    parse_u64,
    parse_u64_be,
    parse_u64_le,
    emit_u64,
    emit_u64_le,
    emit_u64_be,
    u64,
    parse_u128,
    parse_u128_be,
    parse_u128_le,
    emit_u128,
    emit_u128_le,
    emit_u128_be,
    u128,
    parse_i16,
    parse_i16_be,
    parse_i16_le,
    emit_i16,
    emit_i16_le,
    emit_i16_be,
    i16,
    parse_i32,
    parse_i32_be,
    parse_i32_le,
    emit_i32,
    emit_i32_le,
    emit_i32_be,
    i32,
    parse_i64,
    parse_i64_be,
    parse_i64_le,
    emit_i64,
    emit_i64_le,
    emit_i64_be,
    i64,
    parse_i128,
    parse_i128_be,
    parse_i128_le,
    emit_i128,
    emit_i128_le,
    emit_i128_be,
    i128,
);