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}