use crate::error::PfcpError;
use crate::ie::{Ie, IeType};
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct SdfFilter {
pub flow_description: String,
}
impl SdfFilter {
pub fn new(flow_description: &str) -> Self {
SdfFilter {
flow_description: flow_description.to_string(),
}
}
pub fn marshal(&self) -> Vec<u8> {
self.flow_description.as_bytes().to_vec()
}
pub fn unmarshal(payload: &[u8]) -> Result<Self, PfcpError> {
if payload.is_empty() {
return Err(PfcpError::invalid_length(
"SDF Filter",
IeType::SdfFilter,
1,
0,
));
}
let flow_description = String::from_utf8(payload.to_vec()).map_err(|e| {
PfcpError::encoding_error("SDF Filter", IeType::SdfFilter, e.utf8_error())
})?;
Ok(SdfFilter { flow_description })
}
pub fn to_ie(&self) -> Ie {
Ie::new(IeType::SdfFilter, self.marshal())
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_sdf_filter_marshal_unmarshal() {
let sdf = SdfFilter::new("permit in ip from any to 10.0.0.0/8");
let marshaled = sdf.marshal();
let unmarshaled = SdfFilter::unmarshal(&marshaled).unwrap();
assert_eq!(
unmarshaled.flow_description,
"permit in ip from any to 10.0.0.0/8"
);
}
#[test]
fn test_sdf_filter_unmarshal_empty() {
let result = SdfFilter::unmarshal(&[]);
assert!(result.is_err());
let err = result.unwrap_err();
assert!(matches!(err, PfcpError::InvalidLength { .. }));
}
}