turn-types 0.7.1

// Copyright (C) 2025 Matthew Waters <matthew@centricular.com>
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
//
// SPDX-License-Identifier: MIT OR Apache-2.0

//! TURN [`ChannelData`] messages.
//!
//! [`ChannelData`] is used as an optional more efficient data transfer mechanism between a TURN
//! server and a TURN client. A [`ChannelData`] message contains a simple 4-byte header that
//! contains the channel identifier and the length of the data.

use stun_types::message::StunParseError;

/// A [`ChannelData`] message.
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
pub struct ChannelData<'a> {
    id: u16,
    data: &'a [u8],
}

impl<'a> ChannelData<'a> {
    /// Construct a new [`ChannelData`] with the provided identifer and byte sequence.
    pub fn new(id: u16, data: &'a [u8]) -> Self {
        Self { id, data }
    }

    /// The channel identifier stored in this piece of data.
    pub fn id(&self) -> u16 {
        self.id
    }

    /// The sequence of bytes in this message.
    pub fn data(&self) -> &[u8] {
        self.data
    }

    /// Parse a sequence of bytes into a [`ChannelData`].  Returns appropriate errors on failure.
    ///
    /// # Examples
    /// ```
    /// # use turn_types::channel::*;
    /// let data = [4; 3];
    /// let channel = ChannelData::new(0x4000, &data);
    /// let mut output = [0; 7];
    /// assert_eq!(7, channel.write_into_unchecked(&mut output));
    /// let parsed = ChannelData::parse(&output).unwrap();
    /// assert_eq!(parsed.id(), channel.id());
    /// assert_eq!(parsed.data(), channel.data());
    /// ```
    pub fn parse(data: &'a [u8]) -> Result<Self, StunParseError> {
        let (id, len) = Self::parse_header(data)?;

        if len + 4 > data.len() {
            return Err(stun_types::message::StunParseError::Truncated {
                expected: 4 + len,
                actual: data.len(),
            });
        }

        Ok(ChannelData {
            id,
            data: &data[4..4 + len],
        })
    }

    /// Parse the header of an [`ChannelData`] returning the channel ID and the length of the
    /// contained data (without the 4 byte header).
    ///
    /// # Examples
    /// ```
    /// # use turn_types::channel::*;
    /// let data = [4; 3];
    /// let channel = ChannelData::new(0x4000, &data);
    /// let mut output = [0; 7];
    /// assert_eq!(7, channel.write_into_unchecked(&mut output));
    /// let (id, len) = ChannelData::parse_header(&output).unwrap();
    /// assert_eq!(id, channel.id());
    /// assert_eq!(len, 3);
    /// ```
    pub fn parse_header(data: &[u8]) -> Result<(u16, usize), StunParseError> {
        if data.len() < 4 {
            return Err(stun_types::message::StunParseError::Truncated {
                expected: 4,
                actual: data.len(),
            });
        }
        let id = u16::from_be_bytes([data[0], data[1]]);
        let len = u16::from_be_bytes([data[2], data[3]]) as usize;

        if !(0x4000..=0xFFFE).contains(&id) {
            return Err(stun_types::message::StunParseError::InvalidAttributeData);
        }

        Ok((id, len))
    }

    /// Write this [`ChannelData`] into the provided destination slice.
    ///
    /// The destination slice must have size `ChannelData::data().len() + 4`.
    pub fn write_into_unchecked(self, dest: &mut [u8]) -> usize {
        dest[..2].copy_from_slice(self.id.to_be_bytes().as_ref());
        dest[2..4].copy_from_slice((self.data.len() as u16).to_be_bytes().as_ref());
        dest[4..].copy_from_slice(self.data);
        self.data.len() + 4
    }
}

impl core::fmt::Display for ChannelData<'_> {
    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
        write!(
            f,
            "ChannelData(id: {}, data of {} bytes)",
            self.id,
            self.data.len()
        )
    }
}

impl AsRef<[u8]> for ChannelData<'_> {
    fn as_ref(&self) -> &[u8] {
        self.data
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn channel_data_parse_invalid_id() {
        let data = [0x00, 0x00, 0x00, 0x00];
        assert!(matches!(
            ChannelData::parse(&data),
            Err(StunParseError::InvalidAttributeData)
        ));
    }

    #[test]
    fn channel_data_parse_empty() {
        let data = [0x40, 0x00, 0x00, 0x00];
        let channel = ChannelData::parse(&data).unwrap();
        assert_eq!(channel.data(), &[]);
    }

    #[test]
    fn channel_data_parse_truncated_data() {
        let data = [0x40, 0x00, 0x00, 0x01];
        let Err(StunParseError::Truncated { expected, actual }) = ChannelData::parse(&data) else {
            unreachable!();
        };
        assert_eq!(expected, 5);
        assert_eq!(actual, 4);
        assert_eq!(ChannelData::parse_header(&data).unwrap(), (0x4000, 1));
    }

    #[test]
    fn channel_data_parse_truncated_header() {
        let data = [0x40, 0x00, 0x00];
        let Err(StunParseError::Truncated { expected, actual }) = ChannelData::parse(&data) else {
            unreachable!();
        };
        assert_eq!(expected, 4);
        assert_eq!(actual, 3);
    }

    static CHANNEL_SINGLE_BYTE: [u8; 5] = [0x40, 0x00, 0x00, 0x01, 0x42];

    #[test]
    fn channel_data_parse_success() {
        let channel = ChannelData::parse(&CHANNEL_SINGLE_BYTE).unwrap();
        assert_eq!(channel.data(), &[0x42]);
        assert_eq!(channel.as_ref(), &[0x42]);
    }

    #[test]
    fn channel_data_display() {
        let channel = ChannelData::parse(&CHANNEL_SINGLE_BYTE).unwrap();
        assert_eq!(
            &alloc::format!("{channel}"),
            "ChannelData(id: 16384, data of 1 bytes)"
        );
    }
}