slip: ip_packet_addr: clarify arg name

[hippotat.git] / src / slip.rs

diff --git a/src/slip.rs b/src/slip.rs
index 2d376a3ef54df6213fe5a76e49094a8a71fe31c0..fbbbf3dece7317a5f547f0b7c60f5dbdf888656a 100644 (file)
--- a/src/slip.rs
+++ b/src/slip.rs
@@ -24,39 +24,44 @@ impl SlipMime for Slip2Mime { const CONV_TO: Option<bool> = Some(true); }
 impl SlipMime for Mime2Slip { const CONV_TO: Option<bool> = Some(false); }
 impl SlipMime for SlipNoConv { const CONV_TO: Option<bool> = None; }
 
-#[derive(Debug)]
-pub struct ErrorOnlyBad;
-
-#[throws(ErrorOnlyBad)]
-pub fn checkn<AC, EH, OUT, ACR, M: SlipMime+Copy>(
+#[derive(Debug,Error)]
+pub enum SlipFramesError<E> where E: std::error::Error + 'static {
+  #[error("only bad IP datagrams")] ErrorOnlyBad,
+  #[error("{0}")] Other(#[from] E),
+}
+  
+#[throws(SlipFramesError<EHE>)]
+pub async fn checkn<AC, EH, EHE, OUT, FOUT, ACR, M: SlipMime+Copy>(
   mime: M,
   mtu: u32,
   data: &[u8],
   addr_chk: AC,
   mut out: OUT,
   mut error_handler: EH
-) where AC: Fn(&[u8]) -> Result<ACR, PacketError> + Copy,
-        OUT: FnMut((Box<[u8]>, ACR)) -> Result<(), PacketError>,
-        EH: FnMut(PacketError),
+) where AC: Fn(&[u8]) -> Result<ACR, PacketError> + Copy + Send,
+        OUT: FnMut((Box<[u8]>, ACR)) -> FOUT + Send,
+        FOUT: Future<Output=Result<(), PacketError>> + Send,
+        EH: FnMut(PacketError) -> Result<(), SlipFramesError<EHE>> + Send,
+        EHE: std::error::Error + Send + 'static,
 {
   //  eprintln!("before: {:?}", DumpHex(data));
   if data.is_empty() { return }
   let mut ok = false;
   let mut err = false;
   for packet in data.split(|&c| c == SLIP_END) {
-    match (||{
+    match async {
       let checked = check1(mime, mtu, packet, addr_chk);
       if matches!(checked, Err(PacketError::Empty)) { return Ok::<_,PE>(()) }
-      out(checked?)?;
+      out(checked?).await?;
       ok = true;
       Ok::<_,PE>(())
-    })() {
+    }.await {
       Ok(()) => { },
-      Err(e) => { err=true; error_handler(e); },
+      Err(e) => { err=true; error_handler(e)?; },
     }
   }
 //  eprintln!(" after: {:?}", DumpHex(data));
-  if err && !ok { throw!(ErrorOnlyBad) }
+  if err && !ok { throw!(SlipFramesError::ErrorOnlyBad) }
 }
 
 #[throws(PacketError)]
@@ -162,20 +167,20 @@ impl From<Frames> for FramesData {
 const HEADER_FOR_ADDR: usize = 40;
 
 #[throws(PacketError)]
-pub fn ip_packet_addr<const DST: bool>(packet: &[u8]) -> IpAddr {
-  let vsn = (packet.get(0).ok_or_else(|| PE::Empty)? & 0xf0) >> 4;
+pub fn ip_packet_addr<const DST: bool>(header: &[u8]) -> IpAddr {
+  let vsn = (header.get(0).ok_or_else(|| PE::Empty)? & 0xf0) >> 4;
   match vsn {
-    4 if packet.len() >= 20 => {
-      let slice = &packet[if DST { 16 } else { 12 }..][0..4];
+    4 if header.len() >= 20 => {
+      let slice = &header[if DST { 16 } else { 12 }..][0..4];
       Ipv4Addr::from(*<&[u8;4]>::try_from(slice).unwrap()).into()
     },
 
-    6 if packet.len() >= 40 => {
-      let slice = &packet[if DST { 24 } else { 8 }..][0..16];
+    6 if header.len() >= 40 => {
+      let slice = &header[if DST { 24 } else { 8 }..][0..16];
       Ipv6Addr::from(*<&[u8;16]>::try_from(slice).unwrap()).into()
     },
 
-    _ => throw!(PE::Truncated{ vsn, len: packet.len() }),
+    _ => throw!(PE::Truncated{ vsn, len: header.len() }),
   }
 }