1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192

//! Each packet transporting a function management message shall be of service type 8.
use super::*;
use crate::sp::alloc::borrow::ToOwned;
use crate::sp::alloc::string::ToString;
use crate::{sp::PEC_LEN, FuncId, FUNC_ID_LEN};
use tc::{TcData, TcPacket, TcPacketHeader};

pub(crate) const SERVICE_TYPE: u8 = 8;

/// User data. Used as application data on TC and source data as TM.
pub struct Service8_1 {
    func_id: crate::FuncId,
    n: u8,
    /// Contains argument id and argument value.
    args_field: Vec<u8>,
}

impl TcData for Service8_1 {
    // empty
}

/// Implementations of SpacePacket specific to PUS and TC[8,1]
///
/// # Errors
///
/// If not a valid CCSDS 133. 0-B-1 packet for TC[8,1].
/// See page 483 of ECSS-E-ST-70-41C.
impl SpacePacket<TcPacket<Service8_1>> {
    const MES_SUBTYPE: u8 = 1;

    pub fn from_bytes(buffer: &[u8]) -> Result<Self, Error> {
        if buffer.len() < 7 {
            return Err(Error::InvalidPacket);
        }
        let primary_header = PrimaryHeader::from_bytes(&buffer[..PrimaryHeader::PH_LEN])?;
        // If the primary header is not defined properly, give an error accordingly.
        // It has to be have sec_header_flag set, version no to 0, and for TC type_flag should be set.
        if !primary_header.sec_header_flag
            || primary_header.ver_no != 0
            || !primary_header.type_flag
        {
            return Err(Error::InvalidPacket);
        };
        let sec_header = TcPacketHeader::from_bytes(
            &buffer[PrimaryHeader::PH_LEN..PrimaryHeader::PH_LEN + TcPacketHeader::TC_HEADER_LEN],
        )?;
        if sec_header.service_type != SERVICE_TYPE
            || sec_header.message_subtype != SpacePacket::MES_SUBTYPE
        {
            return Err(Error::InvalidPacket);
        }
        // slice for the range of func_len
        let range_start = TcPacketHeader::TC_HEADER_LEN + PrimaryHeader::PH_LEN;
        let func_range = range_start..range_start + FUNC_ID_LEN;
        let func_id_slice = &buffer[func_range];
        if !func_id_slice.is_ascii() {
            return Err(Error::InvalidPacket);
        }
        let func_fixed_slice = array_ref![buffer, range_start, FUNC_ID_LEN];
        let func_id = FuncId::from_byte_string(func_fixed_slice);
        if func_id.is_err() {
            return Err(Error::InvalidPacket);
        }
        let func_id = func_id.unwrap();
        let range_start = range_start + FUNC_ID_LEN + 1;
        let args_field = buffer[range_start..(buffer.len() - PEC_LEN)].to_vec();
        let app_data = Service8_1 {
            func_id,
            n: buffer[range_start - 1],
            args_field,
        };

        let packet_error_control = BigEndian::read_u16(&buffer[(buffer.len() - PEC_LEN)..]);
        Ok(SpacePacket {
            primary_header,
            data: TcPacket::<Service8_1> {
                header: sec_header,
                user_data: TxUserData::<Service8_1> {
                    packet_error_control,
                    data: app_data,
                },
            },
        })
    }

    pub fn new_service_8_1(
        apid: u16,
        packet_name: u16,
        func_id: String,
        n: u8,
        args_field: Vec<u8>,
    ) -> Result<Self, Error> {
        SpacePacket::<TcPacket<Service8_1>>::new(apid, packet_name, func_id, n, args_field)
    }

    ///
    /// # Errors
    ///
    /// returns error if func_id.len() > FUNC_ID_LEN
    /// or returns error if Primary or Secondary Headers are invalid
    ///
    pub fn new(
        apid: u16,
        packet_name: u16,
        func_id: String,
        n: u8,
        args_field: Vec<u8>,
    ) -> Result<Self, Error> {
        // +1 is for N field
        let data_len: usize = PrimaryHeader::PH_LEN
            + TcPacketHeader::TC_HEADER_LEN
            + crate::sp::PEC_LEN
            + FUNC_ID_LEN
            + 1
            + args_field.len();
        let primary_header = PrimaryHeader::new(
            PrimaryHeader::VER_NO,
            true,
            true,
            apid,
            (true, true),
            packet_name,
            (data_len - PrimaryHeader::PH_LEN - 1) as u16,
        )?;
        let sec_header = TcPacketHeader::new((false, false, false, false), 8, 1, 0)?;
        if !func_id.is_ascii() || func_id.len() > FUNC_ID_LEN {
            return Err(Error::InvalidFuncId(func_id.as_str().to_owned()));
        }
        // Convert to fixed size string structure
        let func_id = FuncId::from(func_id.as_str());
        if func_id.is_err() {
            return Err(Error::CapacityError);
        }
        let mut func_id = func_id.unwrap();
        // Add 0 for remaining parts
        for _i in func_id.len()..FUNC_ID_LEN {
            func_id.push(0 as char);
        }
        Ok(SpacePacket {
            primary_header,
            data: TcPacket {
                header: sec_header,
                user_data: TxUserData::<Service8_1> {
                    // TODO don't ignore
                    packet_error_control: 0,
                    data: Service8_1 {
                        func_id,
                        n,
                        args_field,
                    },
                },
            },
        })
    }
    /// Encodes the object to a byte vector
    pub fn to_bytes(&self) -> Vec<u8> {
        let arr_len = PrimaryHeader::PH_LEN + 1 + self.primary_header.data_len as usize;
        let mut bytes = Vec::with_capacity(arr_len);
        bytes.extend(self.primary_header.to_bytes().to_vec());
        bytes.extend(self.data.header.to_bytes().to_vec());
        bytes.extend(self.data.user_data.data.func_id.bytes());
        for _i in self.data.user_data.data.func_id.len()..FUNC_ID_LEN {
            bytes.push(0);
        }
        bytes.push(self.data.user_data.data.n);
        bytes.extend(self.data.user_data.data.args_field.to_vec());
        // add the two bytes then modify them to the true value.
        bytes.push(0);
        bytes.push(0);
        let pec_start = arr_len - PEC_LEN;
        BigEndian::write_u16(
            &mut bytes[pec_start..],
            self.data.user_data.packet_error_control,
        );
        bytes
    }

    pub fn exec_func(
        &self,
        func_map: &hashbrown::HashMap<FuncId, fn(&Vec<u8>) -> Result<(), Error>>,
    ) -> Result<(), Error> {
        let func_id = self.data.user_data.data.func_id;
        if let Some(to_exec) = func_map.get(func_id.as_str()) {
            to_exec(&self.data.user_data.data.args_field)?;
            Ok(())
        } else {
            Err(Error::InvalidFuncId(
                func_id.as_str().to_string().to_owned(),
            ))
        }
    }
}