2024-04-04 15:18:53 +02:00
|
|
|
use crate::pus::create_verification_reporter;
|
2024-05-02 11:14:05 +02:00
|
|
|
use log::info;
|
2024-04-04 15:18:53 +02:00
|
|
|
use satrs::event_man::{EventMessage, EventMessageU32};
|
|
|
|
use satrs::pool::SharedStaticMemoryPool;
|
2024-02-12 15:51:37 +01:00
|
|
|
use satrs::pus::test::PusService17TestHandler;
|
2024-04-04 15:18:53 +02:00
|
|
|
use satrs::pus::verification::{FailParams, VerificationReporter, VerificationReportingProvider};
|
2024-02-12 15:51:37 +01:00
|
|
|
use satrs::pus::{
|
2024-05-01 21:13:08 +02:00
|
|
|
DirectPusPacketHandlerResult, EcssTcAndToken, EcssTcInMemConverter, EcssTcInVecConverter,
|
|
|
|
EcssTmSender, MpscTcReceiver, MpscTmAsVecSender, PusServiceHelper,
|
2024-02-07 18:10:47 +01:00
|
|
|
};
|
2024-05-01 21:13:08 +02:00
|
|
|
use satrs::pus::{EcssTcInSharedStoreConverter, PartialPusHandlingError};
|
2024-02-12 15:51:37 +01:00
|
|
|
use satrs::spacepackets::ecss::tc::PusTcReader;
|
2024-05-02 11:14:05 +02:00
|
|
|
use satrs::spacepackets::ecss::{PusPacket, PusServiceId};
|
2024-04-16 11:04:22 +02:00
|
|
|
use satrs::tmtc::{PacketAsVec, PacketSenderWithSharedPool};
|
2024-04-04 15:18:53 +02:00
|
|
|
use satrs_example::config::components::PUS_TEST_SERVICE;
|
|
|
|
use satrs_example::config::{tmtc_err, TEST_EVENT};
|
|
|
|
use std::sync::mpsc;
|
2023-07-03 18:22:08 +02:00
|
|
|
|
2024-05-02 11:14:05 +02:00
|
|
|
use super::{DirectPusService, HandlingStatus};
|
2024-04-10 11:54:05 +02:00
|
|
|
|
2024-02-07 18:10:47 +01:00
|
|
|
pub fn create_test_service_static(
|
2024-04-16 11:04:22 +02:00
|
|
|
tm_sender: PacketSenderWithSharedPool,
|
2024-02-07 18:10:47 +01:00
|
|
|
tc_pool: SharedStaticMemoryPool,
|
2024-04-24 14:30:45 +02:00
|
|
|
event_sender: mpsc::SyncSender<EventMessageU32>,
|
2024-02-07 18:10:47 +01:00
|
|
|
pus_test_rx: mpsc::Receiver<EcssTcAndToken>,
|
2024-04-16 11:04:22 +02:00
|
|
|
) -> TestCustomServiceWrapper<PacketSenderWithSharedPool, EcssTcInSharedStoreConverter> {
|
2024-02-07 18:10:47 +01:00
|
|
|
let pus17_handler = PusService17TestHandler::new(PusServiceHelper::new(
|
2024-04-04 15:18:53 +02:00
|
|
|
PUS_TEST_SERVICE.id(),
|
|
|
|
pus_test_rx,
|
|
|
|
tm_sender,
|
|
|
|
create_verification_reporter(PUS_TEST_SERVICE.id(), PUS_TEST_SERVICE.apid),
|
2024-02-07 18:10:47 +01:00
|
|
|
EcssTcInSharedStoreConverter::new(tc_pool, 2048),
|
|
|
|
));
|
2024-04-04 15:18:53 +02:00
|
|
|
TestCustomServiceWrapper {
|
|
|
|
handler: pus17_handler,
|
2024-02-07 18:10:47 +01:00
|
|
|
test_srv_event_sender: event_sender,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn create_test_service_dynamic(
|
2024-04-16 11:04:22 +02:00
|
|
|
tm_funnel_tx: mpsc::Sender<PacketAsVec>,
|
2024-04-24 14:30:45 +02:00
|
|
|
event_sender: mpsc::SyncSender<EventMessageU32>,
|
2024-02-07 18:10:47 +01:00
|
|
|
pus_test_rx: mpsc::Receiver<EcssTcAndToken>,
|
2024-04-04 15:18:53 +02:00
|
|
|
) -> TestCustomServiceWrapper<MpscTmAsVecSender, EcssTcInVecConverter> {
|
2024-02-07 18:10:47 +01:00
|
|
|
let pus17_handler = PusService17TestHandler::new(PusServiceHelper::new(
|
2024-04-04 15:18:53 +02:00
|
|
|
PUS_TEST_SERVICE.id(),
|
|
|
|
pus_test_rx,
|
|
|
|
tm_funnel_tx,
|
|
|
|
create_verification_reporter(PUS_TEST_SERVICE.id(), PUS_TEST_SERVICE.apid),
|
2024-02-07 18:10:47 +01:00
|
|
|
EcssTcInVecConverter::default(),
|
|
|
|
));
|
2024-04-04 15:18:53 +02:00
|
|
|
TestCustomServiceWrapper {
|
|
|
|
handler: pus17_handler,
|
2024-02-07 18:10:47 +01:00
|
|
|
test_srv_event_sender: event_sender,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-04-16 11:04:22 +02:00
|
|
|
pub struct TestCustomServiceWrapper<TmSender: EcssTmSender, TcInMemConverter: EcssTcInMemConverter>
|
|
|
|
{
|
2024-04-04 15:18:53 +02:00
|
|
|
pub handler:
|
|
|
|
PusService17TestHandler<MpscTcReceiver, TmSender, TcInMemConverter, VerificationReporter>,
|
2024-04-24 14:30:45 +02:00
|
|
|
pub test_srv_event_sender: mpsc::SyncSender<EventMessageU32>,
|
2023-07-03 18:22:08 +02:00
|
|
|
}
|
|
|
|
|
2024-05-02 11:14:05 +02:00
|
|
|
impl<TmSender: EcssTmSender, TcInMemConverter: EcssTcInMemConverter> DirectPusService
|
|
|
|
for TestCustomServiceWrapper<TmSender, TcInMemConverter>
|
|
|
|
{
|
|
|
|
const SERVICE_ID: u8 = PusServiceId::Test as u8;
|
|
|
|
|
|
|
|
const SERVICE_STR: &'static str = "test";
|
|
|
|
}
|
|
|
|
|
2024-04-16 11:04:22 +02:00
|
|
|
impl<TmSender: EcssTmSender, TcInMemConverter: EcssTcInMemConverter>
|
2024-04-04 15:18:53 +02:00
|
|
|
TestCustomServiceWrapper<TmSender, TcInMemConverter>
|
2024-02-26 11:41:42 +01:00
|
|
|
{
|
2024-05-02 11:14:05 +02:00
|
|
|
pub fn poll_and_handle_next_tc(&mut self, timestamp: &[u8]) -> HandlingStatus {
|
|
|
|
let error_handler = |partial_error: &PartialPusHandlingError| {
|
|
|
|
log::warn!(
|
|
|
|
"PUS {}({}) partial error: {:?}",
|
|
|
|
Self::SERVICE_ID,
|
|
|
|
Self::SERVICE_STR,
|
|
|
|
partial_error
|
|
|
|
);
|
2024-05-01 21:13:08 +02:00
|
|
|
};
|
|
|
|
let res = self
|
|
|
|
.handler
|
2024-05-02 11:14:05 +02:00
|
|
|
.poll_and_handle_next_tc(error_handler, timestamp);
|
|
|
|
if let Err(e) = res {
|
|
|
|
log::warn!(
|
|
|
|
"PUS {}({}) error: {:?}",
|
|
|
|
Self::SERVICE_ID,
|
|
|
|
Self::SERVICE_STR,
|
|
|
|
e
|
|
|
|
);
|
2024-04-10 11:54:05 +02:00
|
|
|
return HandlingStatus::HandledOne;
|
2023-07-04 21:13:26 +02:00
|
|
|
}
|
|
|
|
match res.unwrap() {
|
2024-05-01 21:13:08 +02:00
|
|
|
DirectPusPacketHandlerResult::Handled(handling_status) => {
|
|
|
|
if handling_status == HandlingStatus::HandledOne {
|
|
|
|
info!("Received PUS ping command TC[17,1]");
|
|
|
|
info!("Sent ping reply PUS TM[17,2]");
|
|
|
|
}
|
|
|
|
return handling_status;
|
2023-07-04 22:26:41 +02:00
|
|
|
}
|
2024-05-01 21:13:08 +02:00
|
|
|
DirectPusPacketHandlerResult::SubserviceNotImplemented(subservice, _) => {
|
2024-05-02 11:14:05 +02:00
|
|
|
log::warn!(
|
|
|
|
"PUS {}({}) subservice {} not implemented",
|
|
|
|
Self::SERVICE_ID,
|
|
|
|
Self::SERVICE_STR,
|
|
|
|
subservice
|
|
|
|
);
|
2023-07-05 14:25:51 +02:00
|
|
|
}
|
2024-05-01 21:13:08 +02:00
|
|
|
DirectPusPacketHandlerResult::CustomSubservice(subservice, token) => {
|
2024-02-03 13:41:51 +01:00
|
|
|
let (tc, _) = PusTcReader::new(
|
2024-04-04 15:18:53 +02:00
|
|
|
self.handler
|
2024-02-03 13:41:51 +01:00
|
|
|
.service_helper
|
|
|
|
.tc_in_mem_converter
|
|
|
|
.tc_slice_raw(),
|
|
|
|
)
|
|
|
|
.unwrap();
|
2023-07-05 11:58:43 +02:00
|
|
|
if subservice == 128 {
|
2024-05-02 11:14:05 +02:00
|
|
|
info!("generating test event");
|
2023-07-04 21:13:26 +02:00
|
|
|
self.test_srv_event_sender
|
2024-04-04 15:18:53 +02:00
|
|
|
.send(EventMessage::new(PUS_TEST_SERVICE.id(), TEST_EVENT.into()))
|
2023-07-04 21:13:26 +02:00
|
|
|
.expect("Sending test event failed");
|
2024-05-02 11:14:05 +02:00
|
|
|
match self.handler.service_helper.verif_reporter().start_success(
|
|
|
|
self.handler.service_helper.tm_sender(),
|
|
|
|
token,
|
|
|
|
timestamp,
|
|
|
|
) {
|
|
|
|
Ok(started_token) => {
|
|
|
|
if let Err(e) = self
|
|
|
|
.handler
|
|
|
|
.service_helper
|
|
|
|
.verif_reporter()
|
|
|
|
.completion_success(
|
|
|
|
self.handler.service_helper.tm_sender(),
|
|
|
|
started_token,
|
|
|
|
timestamp,
|
|
|
|
)
|
|
|
|
{
|
|
|
|
error_handler(&PartialPusHandlingError::Verification(e));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Err(e) => {
|
|
|
|
error_handler(&PartialPusHandlingError::Verification(e));
|
|
|
|
}
|
|
|
|
}
|
2023-07-04 21:13:26 +02:00
|
|
|
} else {
|
|
|
|
let fail_data = [tc.subservice()];
|
2024-04-04 15:18:53 +02:00
|
|
|
self.handler
|
2024-02-03 13:41:51 +01:00
|
|
|
.service_helper
|
2024-04-04 15:18:53 +02:00
|
|
|
.verif_reporter()
|
2023-07-05 21:08:04 +02:00
|
|
|
.start_failure(
|
2024-04-04 15:18:53 +02:00
|
|
|
self.handler.service_helper.tm_sender(),
|
2023-07-04 21:13:26 +02:00
|
|
|
token,
|
2023-07-05 21:08:04 +02:00
|
|
|
FailParams::new(
|
2024-05-02 11:14:05 +02:00
|
|
|
timestamp,
|
2023-07-05 21:08:04 +02:00
|
|
|
&tmtc_err::INVALID_PUS_SUBSERVICE,
|
2024-02-20 14:33:21 +01:00
|
|
|
&fail_data,
|
2023-07-05 21:08:04 +02:00
|
|
|
),
|
2023-07-04 21:13:26 +02:00
|
|
|
)
|
|
|
|
.expect("Sending start failure verification failed");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2024-04-10 11:54:05 +02:00
|
|
|
HandlingStatus::HandledOne
|
2023-07-04 21:13:26 +02:00
|
|
|
}
|
2023-07-04 18:51:54 +02:00
|
|
|
}
|