1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
use std::io::Cursor;
use byteorder::{WriteBytesExt, BigEndian};
use crate::error::*;
use crate::buffer::{self, Buffer};
use crate::builder::{Builder as Build, Finalization};
use crate::icmp::checksum;
use crate::icmp::{echo, timestamp, information};
#[derive(Debug)]
pub struct Builder<B: Buffer = buffer::Dynamic> {
buffer: B,
finalizer: Finalization,
}
impl<B: Buffer> Build<B> for Builder<B> {
fn with(buffer: B) -> Result<Self> {
Ok(Builder {
buffer: buffer,
finalizer: Default::default(),
})
}
fn finalizer(&mut self) -> &mut Finalization {
&mut self.finalizer
}
fn build(self) -> Result<B::Inner> {
Err(Error::InvalidPacket)
}
}
impl Default for Builder<buffer::Dynamic> {
fn default() -> Self {
Builder::with(buffer::Dynamic::default()).unwrap()
}
}
impl<B: Buffer> Builder<B> {
pub fn echo(self) -> Result<echo::Builder<B>> {
let mut echo = echo::Builder::with(self.buffer)?;
echo.finalizer().extend(self.finalizer);
Ok(echo)
}
pub fn information(self) -> Result<information::Builder<B>> {
let mut information = information::Builder::with(self.buffer)?;
information.finalizer().extend(self.finalizer);
Ok(information)
}
pub fn timestamp(self) -> Result<timestamp::Builder<B>> {
let mut timestamp = timestamp::Builder::with(self.buffer)?;
timestamp.finalizer().extend(self.finalizer);
Ok(timestamp)
}
}
pub(in crate::icmp) fn prepare<B: Buffer>(finalizer: &mut Finalization, buffer: &B) {
let offset = buffer.offset();
let length = buffer.length();
finalizer.add(move |out| {
let checksum = checksum(&out[offset .. offset + length]);
Cursor::new(&mut out[offset + 2 ..])
.write_u16::<BigEndian>(checksum)?;
Ok(())
});
}
#[cfg(test)]
mod test {
use crate::builder::Builder;
use crate::icmp;
#[test]
fn simple() {
let packet = icmp::Builder::default()
.echo().unwrap().request().unwrap()
.identifier(42).unwrap()
.sequence(2).unwrap()
.payload(b"test").unwrap()
.build().unwrap();
let packet = icmp::Packet::new(packet).unwrap();
assert_eq!(packet.kind(), icmp::Kind::EchoRequest);
}
}