sctp_async/param/
mod.rs

1#[cfg(test)]
2mod param_test;
3
4pub(crate) mod param_chunk_list;
5pub(crate) mod param_forward_tsn_supported;
6pub(crate) mod param_header;
7pub(crate) mod param_heartbeat_info;
8pub(crate) mod param_outgoing_reset_request;
9pub(crate) mod param_random;
10pub(crate) mod param_reconfig_response;
11pub(crate) mod param_requested_hmac_algorithm;
12pub(crate) mod param_state_cookie;
13pub(crate) mod param_supported_extensions;
14pub(crate) mod param_type;
15
16use crate::error::{Error, Result};
17use crate::param::{
18    param_chunk_list::ParamChunkList, param_forward_tsn_supported::ParamForwardTsnSupported,
19    param_heartbeat_info::ParamHeartbeatInfo,
20    param_outgoing_reset_request::ParamOutgoingResetRequest, param_random::ParamRandom,
21    param_reconfig_response::ParamReconfigResponse,
22    param_requested_hmac_algorithm::ParamRequestedHmacAlgorithm,
23    param_state_cookie::ParamStateCookie, param_supported_extensions::ParamSupportedExtensions,
24};
25use param_header::*;
26use param_type::*;
27
28use bytes::{Buf, Bytes, BytesMut};
29use std::{any::Any, fmt};
30
31pub(crate) trait Param: fmt::Display + fmt::Debug {
32    fn header(&self) -> ParamHeader;
33    fn unmarshal(raw: &Bytes) -> Result<Self>
34    where
35        Self: Sized;
36    fn marshal_to(&self, buf: &mut BytesMut) -> Result<usize>;
37    fn value_length(&self) -> usize;
38    fn clone_to(&self) -> Box<dyn Param + Send + Sync>;
39    fn as_any(&self) -> &(dyn Any + Send + Sync);
40
41    fn marshal(&self) -> Result<Bytes> {
42        let capacity = PARAM_HEADER_LENGTH + self.value_length();
43        let mut buf = BytesMut::with_capacity(capacity);
44        self.marshal_to(&mut buf)?;
45        Ok(buf.freeze())
46    }
47}
48
49impl Clone for Box<dyn Param + Send + Sync> {
50    fn clone(&self) -> Box<dyn Param + Send + Sync> {
51        self.clone_to()
52    }
53}
54
55pub(crate) fn build_param(raw_param: &Bytes) -> Result<Box<dyn Param + Send + Sync>> {
56    if raw_param.len() < PARAM_HEADER_LENGTH {
57        return Err(Error::ErrParamHeaderTooShort);
58    }
59    let reader = &mut raw_param.slice(..2);
60    let t: ParamType = reader.get_u16().into();
61    match t {
62        ParamType::ForwardTsnSupp => Ok(Box::new(ParamForwardTsnSupported::unmarshal(raw_param)?)),
63        ParamType::SupportedExt => Ok(Box::new(ParamSupportedExtensions::unmarshal(raw_param)?)),
64        ParamType::Random => Ok(Box::new(ParamRandom::unmarshal(raw_param)?)),
65        ParamType::ReqHmacAlgo => Ok(Box::new(ParamRequestedHmacAlgorithm::unmarshal(raw_param)?)),
66        ParamType::ChunkList => Ok(Box::new(ParamChunkList::unmarshal(raw_param)?)),
67        ParamType::StateCookie => Ok(Box::new(ParamStateCookie::unmarshal(raw_param)?)),
68        ParamType::HeartbeatInfo => Ok(Box::new(ParamHeartbeatInfo::unmarshal(raw_param)?)),
69        ParamType::OutSsnResetReq => Ok(Box::new(ParamOutgoingResetRequest::unmarshal(raw_param)?)),
70        ParamType::ReconfigResp => Ok(Box::new(ParamReconfigResponse::unmarshal(raw_param)?)),
71        _ => Err(Error::ErrParamTypeUnhandled),
72    }
73}
sctp_async/param/mod.rs

sctp_async/param/
mod.rs
