async_arp/
request.rs

1use crate::error::{InputBuildError, ResponseTimeout};
2use pnet::{packet::arp::Arp, util::MacAddr};
3use std::net::Ipv4Addr;
4
5#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)]
6pub struct RequestInput {
7    pub sender_ip: Ipv4Addr,
8    pub sender_mac: MacAddr,
9    pub target_ip: Ipv4Addr,
10    pub target_mac: MacAddr,
11}
12
13#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Default)]
14pub struct RequestInputBuilder {
15    sender_ip: Option<Ipv4Addr>,
16    sender_mac: Option<MacAddr>,
17    target_ip: Option<Ipv4Addr>,
18    target_mac: Option<MacAddr>,
19}
20
21impl RequestInputBuilder {
22    pub fn new() -> Self {
23        Self::default()
24    }
25
26    pub fn with_sender_mac(mut self, sender_mac: MacAddr) -> Self {
27        self.sender_mac = Some(sender_mac);
28        self
29    }
30
31    pub fn with_sender_ip(mut self, sender_ip: Ipv4Addr) -> Self {
32        self.sender_ip = Some(sender_ip);
33        self
34    }
35
36    pub fn with_target_mac(mut self, target_mac: MacAddr) -> Self {
37        self.target_mac = Some(target_mac);
38        self
39    }
40
41    pub fn with_target_ip(mut self, target_ip: Ipv4Addr) -> Self {
42        self.target_ip = Some(target_ip);
43        self
44    }
45
46    pub fn build(&self) -> Result<RequestInput, InputBuildError> {
47        Ok(RequestInput {
48            target_mac: self.target_mac.ok_or(InputBuildError::MissingTargetMac)?,
49            target_ip: self.target_ip.ok_or(InputBuildError::MissingTargetIp)?,
50            sender_mac: self.sender_mac.ok_or(InputBuildError::MissingSenderMac)?,
51            sender_ip: self.sender_ip.ok_or(InputBuildError::MissingSenderIp)?,
52        })
53    }
54}
55
56#[derive(Clone, Debug)]
57pub struct RequestOutcome {
58    pub input: RequestInput,
59    pub response_result: Result<Arp, ResponseTimeout>,
60}
61
62impl RequestOutcome {
63    pub fn new(input: RequestInput, response_result: Result<Arp, ResponseTimeout>) -> Self {
64        Self {
65            input,
66            response_result,
67        }
68    }
69}