use crate::crc::{CRC_CCITT_FALSE, CRC_CCITT_FALSE_NO_TABLE};
use crate::ecss::tc::{AckFlags, ACK_ALL};
use crate::ecss::{
crc_from_raw_data, sp_header_impls, user_data_from_raw,
verify_crc16_ccitt_false_from_raw_to_pus_error, MessageTypeId, PusError, PusPacket, PusVersion,
WritablePusPacket,
};
use crate::util::{UnsignedByteField, UnsignedEnum};
use crate::SpHeader;
use crate::{ByteConversionError, CcsdsPacket, PacketType, SequenceFlags, CCSDS_HEADER_LEN};
use arbitrary_int::{u11, u14, u3, u4};
use core::mem::size_of;
use delegate::delegate;
use num_enum::{IntoPrimitive, TryFromPrimitive};
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
use zerocopy::IntoBytes;
#[cfg(feature = "alloc")]
use alloc::vec::Vec;
use super::verify_crc16_ccitt_false_from_raw_to_pus_error_no_table;
const PUS_VERSION: PusVersion = PusVersion::PusA;
pub trait IsPusTelecommand {}
#[derive(Debug, Eq, PartialEq, Copy, Clone, IntoPrimitive, TryFromPrimitive)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "defmt", derive(defmt::Format))]
#[repr(u8)]
enum AckOpts {
Acceptance = 0b1000,
Start = 0b0100,
Progress = 0b0010,
Completion = 0b0001,
}
pub const MAX_SEC_HEADER_LEN: usize = 18;
pub const MAX_SPARE_BYTES: usize = 7;
#[derive(Debug, Eq, PartialEq, Copy, Clone, thiserror::Error)]
#[error("invalid number of spare bytes, must be between 0 and 7")]
pub struct InvalidNumberOfSpareBytesError;
#[derive(Debug, Eq, PartialEq, Copy, Clone, thiserror::Error)]
#[error("invalid version, expected PUS A (1), got {0}")]
pub struct VersionError(pub u8);
pub trait GenericPusTcSecondaryHeader {
fn pus_version(&self) -> Result<PusVersion, u4>;
fn ack_flags(&self) -> AckFlags;
fn service(&self) -> u8;
fn subservice(&self) -> u8;
fn source_id(&self) -> Option<UnsignedByteField>;
fn spare_bytes(&self) -> usize;
}
#[derive(PartialEq, Eq, Copy, Clone, Debug)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "defmt", derive(defmt::Format))]
pub struct PusTcSecondaryHeader {
pub service: u8,
pub subservice: u8,
pub source_id: Option<UnsignedByteField>,
pub ack: AckFlags,
pub version: PusVersion,
spare_bytes: usize,
}
impl GenericPusTcSecondaryHeader for PusTcSecondaryHeader {
#[inline]
fn pus_version(&self) -> Result<PusVersion, u4> {
Ok(self.version)
}
#[inline]
fn ack_flags(&self) -> AckFlags {
self.ack
}
#[inline]
fn service(&self) -> u8 {
self.service
}
#[inline]
fn subservice(&self) -> u8 {
self.subservice
}
#[inline]
fn source_id(&self) -> Option<UnsignedByteField> {
self.source_id
}
#[inline]
fn spare_bytes(&self) -> usize {
self.spare_bytes
}
}
impl PusTcSecondaryHeader {
#[inline]
pub fn new_simple(service: u8, subservice: u8) -> Self {
PusTcSecondaryHeader {
service,
subservice,
ack: ACK_ALL,
source_id: None,
version: PUS_VERSION,
spare_bytes: 0,
}
}
#[inline]
pub fn new(
service: u8,
subservice: u8,
ack: AckFlags,
source_id: Option<UnsignedByteField>,
spare_bytes: usize,
) -> Self {
PusTcSecondaryHeader {
service,
subservice,
ack,
source_id,
version: PUS_VERSION,
spare_bytes,
}
}
pub fn set_spare_bytes(&mut self, spare_bytes: usize) {
if spare_bytes > 7 {
return;
}
self.spare_bytes = spare_bytes;
}
pub fn written_len(&self) -> usize {
let mut len = 3 + self.spare_bytes;
if let Some(source_id) = self.source_id {
len += source_id.size();
}
len
}
#[cfg(feature = "alloc")]
pub fn to_vec(&self) -> Result<Vec<u8>, ByteConversionError> {
let mut buf = alloc::vec![0; self.written_len()];
self.write_to_be_bytes(&mut buf)?;
Ok(buf)
}
pub fn write_to_be_bytes(&self, buf: &mut [u8]) -> Result<usize, ByteConversionError> {
if buf.len() < self.written_len() {
return Err(ByteConversionError::ToSliceTooSmall {
found: buf.len(),
expected: self.written_len(),
});
}
let mut current_idx = 0;
buf[0] = ((PUS_VERSION as u8) << 4) | self.ack.raw_value().value();
buf[1] = self.service;
buf[2] = self.subservice;
current_idx += 3;
if let Some(source_id) = self.source_id {
current_idx += source_id
.write_to_be_bytes(&mut buf[current_idx..current_idx + source_id.size()])
.unwrap();
}
if self.spare_bytes > 0 {
buf[current_idx..current_idx + self.spare_bytes].fill(0);
current_idx += self.spare_bytes;
}
Ok(current_idx)
}
pub fn from_bytes(
data: &[u8],
source_id_size: Option<usize>,
spare_bytes: usize,
) -> Result<Self, PusError> {
let expected_len = 3 + source_id_size.unwrap_or(0) + spare_bytes;
if data.len() < expected_len {
return Err(PusError::ByteConversion(
ByteConversionError::FromSliceTooSmall {
found: data.len(),
expected: expected_len,
},
));
}
let version = u4::new((data[0] >> 4) & 0b111);
if version != PusVersion::PusA.raw_value() {
return Err(PusError::VersionNotSupported(version));
}
let ack_flags_raw = u4::new(data[0] & 0b1111);
let service = data[1];
let subservice = data[2];
let mut source_id = None;
if let Some(source_id_len) = source_id_size {
source_id = Some(
UnsignedByteField::new_from_be_bytes(source_id_len, &data[3..3 + source_id_len])
.unwrap(),
);
}
Ok(Self {
service,
subservice,
source_id,
ack: AckFlags::new_with_raw_value(ack_flags_raw),
version: PusVersion::PusA,
spare_bytes,
})
}
}
#[derive(Copy, Clone, Debug, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "defmt", derive(defmt::Format))]
pub struct PusTcCreator<'app_data> {
sp_header: SpHeader,
pub sec_header: PusTcSecondaryHeader,
app_data: &'app_data [u8],
}
impl<'app_data> PusTcCreator<'app_data> {
#[inline]
pub fn new(
mut sp_header: SpHeader,
sec_header: PusTcSecondaryHeader,
app_data: &'app_data [u8],
set_ccsds_len: bool,
) -> Self {
sp_header.set_packet_type(PacketType::Tc);
sp_header.set_sec_header_flag();
let mut pus_tc = Self {
sp_header,
app_data,
sec_header,
};
if set_ccsds_len {
pus_tc.update_ccsds_data_len();
}
pus_tc
}
#[inline]
pub fn new_simple(
sph: SpHeader,
service: u8,
subservice: u8,
app_data: &'app_data [u8],
set_ccsds_len: bool,
) -> Self {
Self::new(
sph,
PusTcSecondaryHeader::new(service, subservice, ACK_ALL, None, 0),
app_data,
set_ccsds_len,
)
}
#[inline]
pub fn new_no_app_data(
sp_header: SpHeader,
sec_header: PusTcSecondaryHeader,
set_ccsds_len: bool,
) -> Self {
Self::new(sp_header, sec_header, &[], set_ccsds_len)
}
#[inline]
pub fn sp_header(&self) -> &SpHeader {
&self.sp_header
}
#[inline]
pub fn sp_header_mut(&mut self) -> &mut SpHeader {
&mut self.sp_header
}
#[inline]
pub fn set_ack_field(&mut self, ack: AckFlags) {
self.sec_header.ack = ack;
}
#[inline]
pub fn set_source_id(&mut self, source_id: Option<UnsignedByteField>) {
self.sec_header.source_id = source_id;
}
#[inline]
pub fn app_data(&'app_data self) -> &'app_data [u8] {
self.user_data()
}
sp_header_impls!();
#[inline]
pub fn update_ccsds_data_len(&mut self) {
self.sp_header.data_len =
self.len_written() as u16 - size_of::<crate::zc::SpHeader>() as u16 - 1;
}
pub fn calc_own_crc16(&self) -> u16 {
let mut digest = CRC_CCITT_FALSE.digest();
let sph_zc = crate::zc::SpHeader::from(self.sp_header);
digest.update(sph_zc.as_bytes());
let mut buf: [u8; MAX_SEC_HEADER_LEN] = [0; MAX_SEC_HEADER_LEN];
let sec_header_len = self.sec_header.write_to_be_bytes(&mut buf).unwrap();
digest.update(&buf[0..sec_header_len]);
digest.update(self.app_data);
digest.finalize()
}
pub fn calc_own_crc16_no_table(&self) -> u16 {
let mut digest = CRC_CCITT_FALSE_NO_TABLE.digest();
let sph_zc = crate::zc::SpHeader::from(self.sp_header);
digest.update(sph_zc.as_bytes());
let mut buf: [u8; MAX_SEC_HEADER_LEN] = [0; MAX_SEC_HEADER_LEN];
let sec_header_len = self.sec_header.write_to_be_bytes(&mut buf).unwrap();
digest.update(&buf[0..sec_header_len]);
digest.update(self.app_data);
digest.finalize()
}
#[cfg(feature = "alloc")]
pub fn append_to_vec(&self, vec: &mut Vec<u8>) -> usize {
let sph_zc = crate::zc::SpHeader::from(self.sp_header);
let start_idx = vec.len();
vec.extend_from_slice(sph_zc.as_bytes());
vec.extend_from_slice(self.sec_header.to_vec().unwrap().as_slice());
vec.extend_from_slice(self.app_data);
let mut digest = CRC_CCITT_FALSE.digest();
let mut appended_len =
CCSDS_HEADER_LEN + self.sec_header.written_len() + self.app_data.len();
digest.update(&vec[start_idx..start_idx + appended_len]);
vec.extend_from_slice(&digest.finalize().to_be_bytes());
appended_len += 2;
appended_len
}
pub fn write_to_bytes(&self, slice: &mut [u8]) -> Result<usize, ByteConversionError> {
let writer_unfinalized = self.common_write(slice)?;
Ok(writer_unfinalized.finalize())
}
pub fn write_to_bytes_crc_no_table(
&self,
slice: &mut [u8],
) -> Result<usize, ByteConversionError> {
let writer_unfinalized = self.common_write(slice)?;
Ok(writer_unfinalized.finalize_crc_no_table())
}
pub fn write_to_bytes_no_crc(&self, slice: &mut [u8]) -> Result<usize, ByteConversionError> {
let writer_unfinalized = self.common_write(slice)?;
Ok(writer_unfinalized.finalize_no_crc())
}
fn common_write<'a>(
&self,
slice: &'a mut [u8],
) -> Result<PusTcCreatorWithReservedAppData<'a>, ByteConversionError> {
if self.len_written() > slice.len() {
return Err(ByteConversionError::ToSliceTooSmall {
found: slice.len(),
expected: self.len_written(),
});
}
let mut writer_unfinalized = PusTcCreatorWithReservedAppData::write_to_bytes_partially(
slice,
self.sp_header,
self.sec_header,
self.app_data.len(),
)?;
writer_unfinalized
.app_data_mut()
.copy_from_slice(self.app_data);
Ok(writer_unfinalized)
}
}
impl WritablePusPacket for PusTcCreator<'_> {
#[inline]
fn len_written(&self) -> usize {
CCSDS_HEADER_LEN + self.sec_header.written_len() + self.app_data.len() + 2
}
fn has_checksum(&self) -> bool {
true
}
fn write_to_bytes_no_checksum(&self, slice: &mut [u8]) -> Result<usize, PusError> {
Ok(Self::write_to_bytes_no_crc(self, slice)?)
}
fn write_to_bytes(&self, slice: &mut [u8]) -> Result<usize, PusError> {
Ok(Self::write_to_bytes(self, slice)?)
}
fn write_to_bytes_checksum_no_table(&self, slice: &mut [u8]) -> Result<usize, PusError> {
Ok(Self::write_to_bytes_crc_no_table(self, slice)?)
}
}
impl CcsdsPacket for PusTcCreator<'_> {
delegate!(to self.sp_header {
#[inline]
fn ccsds_version(&self) -> u3;
#[inline]
fn packet_id(&self) -> crate::PacketId;
#[inline]
fn psc(&self) -> crate::PacketSequenceControl;
#[inline]
fn data_len(&self) -> u16;
});
}
impl PusPacket for PusTcCreator<'_> {
delegate!(to self.sec_header {
#[inline]
fn pus_version(&self) -> Result<PusVersion, u4>;
});
#[inline]
fn message_type_id(&self) -> MessageTypeId {
MessageTypeId {
type_id: self.service_type_id(),
subtype_id: self.message_subtype_id(),
}
}
#[inline]
fn service_type_id(&self) -> u8 {
self.sec_header.service
}
#[inline]
fn message_subtype_id(&self) -> u8 {
self.sec_header.subservice
}
#[inline]
fn user_data(&self) -> &[u8] {
self.app_data
}
#[inline]
fn checksum(&self) -> Option<u16> {
Some(self.calc_own_crc16())
}
}
impl GenericPusTcSecondaryHeader for PusTcCreator<'_> {
delegate!(to self.sec_header {
#[inline]
fn pus_version(&self) -> Result<PusVersion, u4>;
#[inline]
fn service(&self) -> u8;
#[inline]
fn subservice(&self) -> u8;
#[inline]
fn source_id(&self) -> Option<UnsignedByteField>;
#[inline]
fn ack_flags(&self) -> AckFlags;
#[inline]
fn spare_bytes(&self) -> usize;
});
}
impl IsPusTelecommand for PusTcCreator<'_> {}
pub struct PusTcCreatorWithReservedAppData<'buf> {
buf: &'buf mut [u8],
app_data_offset: usize,
full_len: usize,
}
impl<'buf> PusTcCreatorWithReservedAppData<'buf> {
#[inline]
pub fn new(
buf: &'buf mut [u8],
mut sp_header: SpHeader,
sec_header: PusTcSecondaryHeader,
app_data_len: usize,
) -> Result<Self, ByteConversionError> {
sp_header.set_packet_type(PacketType::Tc);
sp_header.set_sec_header_flag();
let len_written = CCSDS_HEADER_LEN + sec_header.written_len() + app_data_len + 2;
if len_written > buf.len() {
return Err(ByteConversionError::ToSliceTooSmall {
found: buf.len(),
expected: len_written,
});
}
sp_header.data_len = len_written as u16 - size_of::<crate::zc::SpHeader>() as u16 - 1;
Self::write_to_bytes_partially(buf, sp_header, sec_header, app_data_len)
}
fn write_to_bytes_partially(
buf: &'buf mut [u8],
sp_header: SpHeader,
sec_header: PusTcSecondaryHeader,
app_data_len: usize,
) -> Result<Self, ByteConversionError> {
let mut curr_idx = 0;
sp_header.write_to_be_bytes(&mut buf[0..CCSDS_HEADER_LEN])?;
curr_idx += CCSDS_HEADER_LEN;
curr_idx += sec_header
.write_to_be_bytes(&mut buf[curr_idx..curr_idx + sec_header.written_len()])?;
let app_data_offset = curr_idx;
curr_idx += app_data_len;
Ok(Self {
buf,
app_data_offset,
full_len: curr_idx + 2,
})
}
#[inline]
pub const fn len_written(&self) -> usize {
self.full_len
}
#[inline]
pub fn app_data_mut(&mut self) -> &mut [u8] {
&mut self.buf[self.app_data_offset..self.full_len - 2]
}
#[inline]
pub fn app_data(&self) -> &[u8] {
&self.buf[self.app_data_offset..self.full_len - 2]
}
#[inline]
pub fn app_data_len(&self) -> usize {
self.full_len - 2 - self.app_data_offset
}
pub fn finalize(self) -> usize {
let mut digest = CRC_CCITT_FALSE.digest();
digest.update(&self.buf[0..self.full_len - 2]);
self.buf[self.full_len - 2..self.full_len]
.copy_from_slice(&digest.finalize().to_be_bytes());
self.full_len
}
pub fn finalize_crc_no_table(self) -> usize {
let mut digest = CRC_CCITT_FALSE_NO_TABLE.digest();
digest.update(&self.buf[0..self.full_len - 2]);
self.buf[self.full_len - 2..self.full_len]
.copy_from_slice(&digest.finalize().to_be_bytes());
self.full_len
}
#[inline]
pub fn finalize_no_crc(self) -> usize {
self.full_len - 2
}
}
#[derive(Eq, Copy, Clone, Debug)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "defmt", derive(defmt::Format))]
pub struct PusTcReader<'raw_data> {
#[cfg_attr(feature = "serde", serde(skip))]
raw_data: &'raw_data [u8],
sp_header: SpHeader,
sec_header: PusTcSecondaryHeader,
app_data: &'raw_data [u8],
crc16: u16,
}
impl<'raw_data> PusTcReader<'raw_data> {
pub fn new(
slice: &'raw_data [u8],
source_id_size: Option<usize>,
spare_bytes: usize,
) -> Result<Self, PusError> {
let pus_tc = Self::new_no_crc_check(slice, source_id_size, spare_bytes)?;
verify_crc16_ccitt_false_from_raw_to_pus_error(pus_tc.raw_data(), pus_tc.crc16())?;
Ok(pus_tc)
}
pub fn new_crc_no_table(
slice: &'raw_data [u8],
source_id_size: Option<usize>,
spare_bytes: usize,
) -> Result<Self, PusError> {
let pus_tc = Self::new_no_crc_check(slice, source_id_size, spare_bytes)?;
verify_crc16_ccitt_false_from_raw_to_pus_error_no_table(pus_tc.raw_data(), pus_tc.crc16())?;
Ok(pus_tc)
}
pub fn new_no_crc_check(
slice: &'raw_data [u8],
source_id_size: Option<usize>,
spare_bytes: usize,
) -> Result<Self, PusError> {
let raw_data_len = slice.len();
if raw_data_len < CCSDS_HEADER_LEN {
return Err(ByteConversionError::FromSliceTooSmall {
found: raw_data_len,
expected: CCSDS_HEADER_LEN,
}
.into());
}
let mut current_idx = 0;
let (sp_header, _) = SpHeader::from_be_bytes(&slice[0..CCSDS_HEADER_LEN])?;
current_idx += CCSDS_HEADER_LEN;
let total_len = sp_header.packet_len();
if raw_data_len < total_len {
return Err(ByteConversionError::FromSliceTooSmall {
found: raw_data_len,
expected: total_len,
}
.into());
}
let sec_header =
PusTcSecondaryHeader::from_bytes(&slice[current_idx..], source_id_size, spare_bytes)?;
current_idx += sec_header.written_len();
let raw_data = &slice[0..total_len];
Ok(Self {
sp_header,
sec_header,
raw_data,
app_data: user_data_from_raw(current_idx, total_len, slice, true)?,
crc16: crc_from_raw_data(raw_data)?,
})
}
#[inline]
pub fn app_data(&self) -> &[u8] {
self.user_data()
}
#[inline]
pub fn raw_data(&self) -> &[u8] {
self.raw_data
}
#[inline]
pub fn len_packed(&self) -> usize {
self.sp_header.packet_len()
}
#[inline]
pub fn sp_header(&self) -> &SpHeader {
&self.sp_header
}
#[inline]
pub fn crc16(&self) -> u16 {
self.crc16
}
}
impl PartialEq for PusTcReader<'_> {
#[inline]
fn eq(&self, other: &Self) -> bool {
self.raw_data == other.raw_data
}
}
impl CcsdsPacket for PusTcReader<'_> {
delegate!(to self.sp_header {
#[inline]
fn ccsds_version(&self) -> u3;
#[inline]
fn packet_id(&self) -> crate::PacketId;
#[inline]
fn psc(&self) -> crate::PacketSequenceControl;
#[inline]
fn data_len(&self) -> u16;
});
}
impl PusPacket for PusTcReader<'_> {
delegate!(to self.sec_header {
#[inline]
fn pus_version(&self) -> Result<PusVersion, u4>;
});
#[inline]
fn message_type_id(&self) -> MessageTypeId {
MessageTypeId {
type_id: self.service_type_id(),
subtype_id: self.message_subtype_id(),
}
}
#[inline]
fn service_type_id(&self) -> u8 {
self.sec_header.service
}
#[inline]
fn message_subtype_id(&self) -> u8 {
self.sec_header.subservice
}
#[inline]
fn has_checksum(&self) -> bool {
true
}
#[inline]
fn user_data(&self) -> &[u8] {
self.app_data
}
#[inline]
fn checksum(&self) -> Option<u16> {
Some(self.crc16)
}
}
impl GenericPusTcSecondaryHeader for PusTcReader<'_> {
delegate!(to self.sec_header {
#[inline]
fn pus_version(&self) -> Result<PusVersion, u4>;
#[inline]
fn service(&self) -> u8;
#[inline]
fn subservice(&self) -> u8;
#[inline]
fn source_id(&self) -> Option<UnsignedByteField>;
#[inline]
fn ack_flags(&self) -> AckFlags;
#[inline]
fn spare_bytes(&self) -> usize;
});
}
impl IsPusTelecommand for PusTcReader<'_> {}
impl PartialEq<PusTcCreator<'_>> for PusTcReader<'_> {
fn eq(&self, other: &PusTcCreator) -> bool {
self.sp_header == other.sp_header
&& self.sec_header == other.sec_header
&& self.app_data == other.app_data
}
}
impl PartialEq<PusTcReader<'_>> for PusTcCreator<'_> {
fn eq(&self, other: &PusTcReader) -> bool {
self.sp_header == other.sp_header
&& self.sec_header == other.sec_header
&& self.app_data == other.app_data
}
}
#[cfg(all(test, feature = "std"))]
mod tests {
use std::println;
use super::*;
use crate::ecss::{PusError, PusPacket, WritablePusPacket};
use crate::util::{UnsignedByteFieldU16, UnsignedByteFieldU8};
use crate::{ByteConversionError, SpHeader};
use crate::{CcsdsPacket, SequenceFlags};
use alloc::string::ToString;
use alloc::vec::Vec;
use arbitrary_int::{u11, u14};
#[cfg(feature = "serde")]
use postcard::{from_bytes, to_allocvec};
fn base_ping_tc_full_ctor() -> PusTcCreator<'static> {
let sph = SpHeader::new_for_unseg_tc(u11::new(0x02), u14::new(0x34), 0);
let tc_header = PusTcSecondaryHeader::new_simple(17, 1);
PusTcCreator::new_no_app_data(sph, tc_header, true)
}
fn base_ping_tc_simple_ctor() -> PusTcCreator<'static> {
let sph = SpHeader::new_for_unseg_tc(u11::new(0x02), u14::new(0x34), 0);
PusTcCreator::new_simple(sph, 17, 1, &[], true)
}
fn base_ping_tc_simple_ctor_with_app_data(app_data: &'static [u8]) -> PusTcCreator<'static> {
let sph = SpHeader::new_for_unseg_tc(u11::new(0x02), u14::new(0x34), 0);
PusTcCreator::new_simple(sph, 17, 1, app_data, true)
}
#[test]
fn test_tc_fields() {
let pus_tc = base_ping_tc_full_ctor();
verify_test_tc(&pus_tc, false, 11);
}
#[test]
fn test_serialization() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
let size = pus_tc
.write_to_bytes(test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
assert_eq!(size, 11);
assert_eq!(
pus_tc.checksum().unwrap(),
u16::from_be_bytes(test_buf[size - 2..size].try_into().unwrap())
);
}
#[test]
fn test_serialization_with_trait_1() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
let size = WritablePusPacket::write_to_bytes(&pus_tc, test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
assert_eq!(size, 11);
assert_eq!(
pus_tc.checksum().unwrap(),
u16::from_be_bytes(test_buf[size - 2..size].try_into().unwrap())
);
}
#[test]
fn test_serialization_with_trait_2() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
let size =
WritablePusPacket::write_to_bytes_checksum_no_table(&pus_tc, test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
assert_eq!(size, 11);
assert_eq!(
pus_tc.checksum().unwrap(),
u16::from_be_bytes(test_buf[size - 2..size].try_into().unwrap())
);
}
#[test]
fn test_serialization_crc_no_table() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
let size = pus_tc
.write_to_bytes_crc_no_table(test_buf.as_mut_slice())
.expect("error writing tc to buffer");
assert_eq!(size, 11);
assert_eq!(
pus_tc.checksum().unwrap(),
u16::from_be_bytes(test_buf[size - 2..size].try_into().unwrap())
);
}
#[test]
fn test_serialization_no_crc() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
let size = pus_tc
.write_to_bytes_no_crc(test_buf.as_mut_slice())
.expect("error writing tc to buffer");
assert_eq!(size, 9);
assert_eq!(test_buf[10], 0);
assert_eq!(test_buf[11], 0);
}
#[test]
fn test_serialization_no_crc_with_trait() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
let size = WritablePusPacket::write_to_bytes_no_checksum(&pus_tc, test_buf.as_mut_slice())
.expect("error writing tc to buffer");
assert_eq!(size, 9);
assert_eq!(test_buf[9], 0);
assert_eq!(test_buf[10], 0);
}
#[test]
fn test_deserialization() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
let size = pus_tc
.write_to_bytes(test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
assert_eq!(size, 11);
let tc_from_raw = PusTcReader::new(&test_buf, None, 0)
.expect("Creating PUS TC struct from raw buffer failed");
assert_eq!(tc_from_raw.packet_len(), 11);
verify_test_tc_with_reader(&tc_from_raw, false, 11);
assert!(tc_from_raw.user_data().is_empty());
verify_test_tc_raw(&test_buf);
verify_crc_no_app_data(&test_buf);
}
#[test]
fn test_deserialization_alt_ctor() {
let sph = SpHeader::new_for_unseg_tc(u11::new(0x02), u14::new(0x34), 0);
let tc_header = PusTcSecondaryHeader::new_simple(17, 1);
let mut test_buf: [u8; 32] = [0; 32];
let mut pus_tc =
PusTcCreatorWithReservedAppData::new(&mut test_buf, sph, tc_header, 0).unwrap();
assert_eq!(pus_tc.len_written(), 11);
assert_eq!(pus_tc.app_data_len(), 0);
assert_eq!(pus_tc.app_data(), &[]);
assert_eq!(pus_tc.app_data_mut(), &[]);
let size = pus_tc.finalize();
assert_eq!(size, 11);
let tc_from_raw = PusTcReader::new(&test_buf, None, 0)
.expect("Creating PUS TC struct from raw buffer failed");
assert_eq!(tc_from_raw.packet_len(), 11);
verify_test_tc_with_reader(&tc_from_raw, false, 11);
assert!(tc_from_raw.user_data().is_empty());
verify_test_tc_raw(&test_buf);
verify_crc_no_app_data(&test_buf);
}
#[test]
fn test_deserialization_no_table() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
let size = pus_tc
.write_to_bytes(test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
assert_eq!(size, 11);
let tc_from_raw = PusTcReader::new_crc_no_table(&test_buf, None, 0)
.expect("Creating PUS TC struct from raw buffer failed");
assert_eq!(tc_from_raw.packet_len(), 11);
verify_test_tc_with_reader(&tc_from_raw, false, 11);
assert!(tc_from_raw.user_data().is_empty());
verify_test_tc_raw(&test_buf);
verify_crc_no_app_data(&test_buf);
}
#[test]
fn test_writing_into_vec() {
let pus_tc = base_ping_tc_simple_ctor();
let tc_vec = pus_tc.to_vec().expect("Error writing TC to buffer");
assert_eq!(tc_vec.len(), 11);
let tc_from_raw = PusTcReader::new(tc_vec.as_slice(), None, 0)
.expect("Creating PUS TC struct from raw buffer failed");
assert_eq!(tc_from_raw.packet_len(), 11);
verify_test_tc_with_reader(&tc_from_raw, false, 11);
assert!(tc_from_raw.user_data().is_empty());
verify_test_tc_raw(&tc_vec);
verify_crc_no_app_data(&tc_vec);
}
#[test]
fn test_update_func() {
let sph = SpHeader::new_for_unseg_tc(u11::new(0x02), u14::new(0x34), 0);
let mut tc = PusTcCreator::new_simple(sph, 17, 1, &[], false);
assert_eq!(tc.data_len(), 0);
tc.update_ccsds_data_len();
assert_eq!(tc.data_len(), 4);
}
#[test]
fn test_deserialization_with_app_data() {
let pus_tc = base_ping_tc_simple_ctor_with_app_data(&[1, 2, 3]);
let mut test_buf: [u8; 32] = [0; 32];
let size = pus_tc
.write_to_bytes(test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
assert_eq!(size, 14);
let tc_from_raw = PusTcReader::new(&test_buf, None, 0)
.expect("Creating PUS TC struct from raw buffer failed");
assert_eq!(tc_from_raw.packet_len(), 14);
verify_test_tc_with_reader(&tc_from_raw, true, 14);
let user_data = tc_from_raw.user_data();
assert_eq!(tc_from_raw.user_data(), tc_from_raw.app_data());
assert_eq!(tc_from_raw.raw_data(), &test_buf[..size]);
assert_eq!(
tc_from_raw.checksum().unwrap(),
u16::from_be_bytes(test_buf[size - 2..size].try_into().unwrap())
);
assert_eq!(user_data[0], 1);
assert_eq!(user_data[1], 2);
assert_eq!(user_data[2], 3);
}
#[test]
fn test_reader_eq() {
let pus_tc = base_ping_tc_simple_ctor_with_app_data(&[1, 2, 3]);
let mut test_buf: [u8; 32] = [0; 32];
pus_tc
.write_to_bytes(test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
let tc_from_raw_0 = PusTcReader::new(&test_buf, None, 0)
.expect("Creating PUS TC struct from raw buffer failed");
let tc_from_raw_1 = PusTcReader::new(&test_buf, None, 0)
.expect("Creating PUS TC struct from raw buffer failed");
assert_eq!(tc_from_raw_0, tc_from_raw_1);
}
#[test]
fn test_vec_ser_deser() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_vec = Vec::new();
let size = pus_tc.append_to_vec(&mut test_vec);
assert_eq!(size, 11);
println!("Test vector: {test_vec:x?}");
verify_test_tc_raw(&test_vec.as_slice());
verify_crc_no_app_data(&test_vec.as_slice());
}
#[test]
fn test_incorrect_crc() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
pus_tc
.write_to_bytes(test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
test_buf[9] = 0;
test_buf[10] = 0;
let res = PusTcReader::new(&test_buf, None, 0);
assert!(res.is_err());
let err = res.unwrap_err();
if let PusError::ChecksumFailure(crc) = err {
assert_eq!(crc, 0);
assert_eq!(
err.to_string(),
"checksum verification for crc16 0x0000 failed"
);
} else {
panic!("unexpected error {err}");
}
}
#[test]
fn test_manual_crc_calculation() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
pus_tc.calc_own_crc16();
pus_tc
.write_to_bytes(test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
verify_test_tc_raw(&test_buf);
verify_crc_no_app_data(&test_buf);
}
#[test]
fn test_with_application_data_vec() {
let pus_tc = base_ping_tc_simple_ctor_with_app_data(&[1, 2, 3]);
verify_test_tc(&pus_tc, true, 14);
let mut test_vec = Vec::new();
let size = pus_tc.append_to_vec(&mut test_vec);
assert_eq!(test_vec[9], 1);
assert_eq!(test_vec[10], 2);
assert_eq!(test_vec[11], 3);
assert_eq!(size, 14);
}
#[test]
fn test_write_buf_too_small() {
let pus_tc = base_ping_tc_simple_ctor();
let mut test_buf = [0; 10];
let res = pus_tc.write_to_bytes(test_buf.as_mut_slice());
assert!(res.is_err());
let err = res.unwrap_err();
assert_eq!(
err,
ByteConversionError::ToSliceTooSmall {
found: 10,
expected: 11
}
);
assert_eq!(
err.to_string(),
"target slice with size 10 is too small, expected size of at least 11"
);
}
#[test]
fn test_with_application_data_buf() {
let pus_tc = base_ping_tc_simple_ctor_with_app_data(&[1, 2, 3]);
verify_test_tc(&pus_tc, true, 14);
let mut test_buf: [u8; 32] = [0; 32];
let size = pus_tc
.write_to_bytes(test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
assert_eq!(test_buf[9], 1);
assert_eq!(test_buf[10], 2);
assert_eq!(test_buf[11], 3);
assert_eq!(size, 14);
}
#[test]
fn test_custom_setters() {
let mut pus_tc = base_ping_tc_simple_ctor();
let mut test_buf: [u8; 32] = [0; 32];
pus_tc.set_apid(u11::new(0x7ff));
pus_tc.set_seq_count(u14::new(0x3fff));
pus_tc.set_ack_field(AckFlags::new_with_raw_value(u4::new(0b11)));
let source_id = UnsignedByteFieldU16::new(0xffff).into();
pus_tc.set_source_id(Some(source_id));
pus_tc.set_seq_flags(SequenceFlags::Unsegmented);
assert_eq!(pus_tc.source_id(), Some(source_id));
assert_eq!(pus_tc.seq_count().value(), 0x3fff);
assert_eq!(pus_tc.ack_flags().raw_value().value(), 0b11);
assert_eq!(pus_tc.apid().value(), 0x7ff);
assert_eq!(pus_tc.sequence_flags(), SequenceFlags::Unsegmented);
pus_tc.calc_own_crc16();
pus_tc
.write_to_bytes(test_buf.as_mut_slice())
.expect("Error writing TC to buffer");
assert_eq!(test_buf[0], 0x1f);
assert_eq!(test_buf[1], 0xff);
assert_eq!(test_buf[2], 0xff);
assert_eq!(test_buf[3], 0xff);
assert_eq!(test_buf[6], 0x13);
assert_eq!(test_buf[9], 0xff);
assert_eq!(test_buf[10], 0xff);
}
fn verify_test_tc(tc: &PusTcCreator, has_user_data: bool, exp_full_len: usize) {
verify_test_tc_generic(tc);
if !has_user_data {
assert!(tc.user_data().is_empty());
}
let mut comp_header =
SpHeader::new_for_unseg_tc(u11::new(0x02), u14::new(0x34), exp_full_len as u16 - 7);
comp_header.set_sec_header_flag();
assert_eq!(*tc.sp_header(), comp_header);
}
fn verify_test_tc_with_reader(tc: &PusTcReader, has_user_data: bool, exp_full_len: usize) {
verify_test_tc_generic(tc);
if !has_user_data {
assert!(tc.user_data().is_empty());
}
assert_eq!(tc.len_packed(), exp_full_len);
let mut comp_header =
SpHeader::new_for_unseg_tc(u11::new(0x02), u14::new(0x34), exp_full_len as u16 - 7);
comp_header.set_sec_header_flag();
assert_eq!(*tc.sp_header(), comp_header);
}
fn verify_test_tc_generic(tc: &(impl PusPacket + GenericPusTcSecondaryHeader)) {
assert_eq!(PusPacket::service_type_id(tc), 17);
assert_eq!(GenericPusTcSecondaryHeader::service(tc), 17);
assert_eq!(PusPacket::message_subtype_id(tc), 1);
assert_eq!(GenericPusTcSecondaryHeader::subservice(tc), 1);
assert!(tc.sec_header_flag());
assert_eq!(PusPacket::pus_version(tc).unwrap(), PusVersion::PusA);
assert_eq!(tc.seq_count().value(), 0x34);
assert!(tc.source_id().is_none());
assert_eq!(tc.apid().value(), 0x02);
assert_eq!(tc.ack_flags(), ACK_ALL);
assert_eq!(PusPacket::pus_version(tc).unwrap(), PusVersion::PusA);
assert_eq!(
GenericPusTcSecondaryHeader::pus_version(tc).unwrap(),
PusVersion::PusA
);
}
fn verify_test_tc_raw(slice: &impl AsRef<[u8]>) {
let slice = slice.as_ref();
assert_eq!(slice[0], 0x18);
assert_eq!(slice[1], 0x02);
assert_eq!(slice[2], 0xc0);
assert_eq!(slice[3], 0x34);
assert_eq!(slice[4], 0x00);
assert_eq!(slice[5], 0x04);
assert_eq!(slice[6], 0x1f);
assert_eq!(slice[7], 0x11);
assert_eq!(slice[8], 0x01);
}
fn verify_crc_no_app_data(slice: &impl AsRef<[u8]>) {
let slice = slice.as_ref();
assert_eq!(slice[9], 0x37);
assert_eq!(slice[10], 0x2d);
}
#[test]
fn partial_eq_pus_tc() {
let pus_tc_1 = base_ping_tc_simple_ctor();
let pus_tc_2 = base_ping_tc_full_ctor();
assert_eq!(pus_tc_1, pus_tc_2);
}
#[test]
fn partial_eq_serialized_vs_derialized() {
let pus_tc = base_ping_tc_simple_ctor();
let mut buf = [0; 32];
pus_tc.write_to_bytes(&mut buf).unwrap();
assert_eq!(pus_tc, PusTcReader::new(&buf, None, 0).unwrap());
assert_eq!(PusTcReader::new(&buf, None, 0).unwrap(), pus_tc);
}
#[test]
fn test_ack_opts_from_raw() {
let ack_opts_raw = AckOpts::Start as u8;
let ack_opts = AckOpts::try_from(ack_opts_raw).unwrap();
assert_eq!(ack_opts, AckOpts::Start);
}
#[test]
fn test_reader_buf_too_small() {
let app_data = &[1, 2, 3, 4];
let pus_tc = base_ping_tc_simple_ctor_with_app_data(app_data);
let mut buf = [0; 32];
let written_len = pus_tc.write_to_bytes(&mut buf).unwrap();
let error = PusTcReader::new(&buf[0..7], None, 0);
assert!(error.is_err());
let error = error.unwrap_err();
if let PusError::ByteConversion(ByteConversionError::FromSliceTooSmall {
found,
expected,
}) = error
{
assert_eq!(found, 7);
assert_eq!(expected, written_len);
} else {
panic!("unexpected error {error}")
}
}
#[test]
fn test_reader_input_too_small() {
let buf: [u8; 5] = [0; 5];
let error = PusTcReader::new(&buf, None, 0);
assert!(error.is_err());
let error = error.unwrap_err();
if let PusError::ByteConversion(ByteConversionError::FromSliceTooSmall {
found,
expected,
}) = error
{
assert_eq!(found, 5);
assert_eq!(expected, 6);
} else {
panic!("unexpected error {error}")
}
}
#[test]
fn test_with_source_id_and_spare_bytes() {
let sph = SpHeader::new_for_unseg_tc(u11::new(0x02), u14::new(0x34), 0);
let source_id = UnsignedByteFieldU8::new(5).into();
let tc_header = PusTcSecondaryHeader::new(17, 1, AckFlags::ALL, Some(source_id), 2);
let creator = PusTcCreator::new(sph, tc_header, &[1, 2, 3], true);
assert_eq!(creator.len_written(), 17);
let mut buf: [u8; 32] = [0; 32];
assert_eq!(creator.write_to_bytes(&mut buf).unwrap(), 17);
assert_eq!(buf[9], 5);
assert_eq!(buf[10], 0);
assert_eq!(buf[11], 0);
assert_eq!(buf[12], 1);
assert_eq!(buf[13], 2);
assert_eq!(buf[14], 3);
let tc_reader = PusTcReader::new(&buf, Some(1), 2)
.expect("Creating PUS TC struct from raw buffer failed");
assert_eq!(tc_reader.sp_header(), creator.sp_header());
assert_eq!(tc_reader.app_data(), creator.app_data());
assert_eq!(tc_reader.source_id(), Some(source_id));
assert_eq!(creator.spare_bytes(), 2);
}
#[test]
#[cfg(feature = "serde")]
fn test_serialization_tc_serde() {
let pus_tc = base_ping_tc_simple_ctor();
let output = to_allocvec(&pus_tc).unwrap();
let output_converted_back: PusTcCreator = from_bytes(&output).unwrap();
assert_eq!(output_converted_back, pus_tc);
}
}