118 lines
3.8 KiB
Rust
118 lines
3.8 KiB
Rust
#![feature(test)]
|
|
|
|
mod wire {
|
|
use smoltcp::phy::ChecksumCapabilities;
|
|
use smoltcp::wire::{IpAddress, IpProtocol};
|
|
#[cfg(feature = "proto-ipv4")]
|
|
use smoltcp::wire::{Ipv4Address, Ipv4Packet, Ipv4Repr};
|
|
#[cfg(feature = "proto-ipv6")]
|
|
use smoltcp::wire::{Ipv6Address, Ipv6Packet, Ipv6Repr};
|
|
use smoltcp::wire::{TcpControl, TcpPacket, TcpRepr, TcpSeqNumber};
|
|
use smoltcp::wire::{UdpPacket, UdpRepr};
|
|
|
|
extern crate test;
|
|
|
|
#[cfg(feature = "proto-ipv6")]
|
|
const SRC_ADDR: IpAddress = IpAddress::Ipv6(Ipv6Address([
|
|
0xfe, 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
|
|
]));
|
|
#[cfg(feature = "proto-ipv6")]
|
|
const DST_ADDR: IpAddress = IpAddress::Ipv6(Ipv6Address([
|
|
0xfe, 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2,
|
|
]));
|
|
|
|
#[cfg(all(not(feature = "proto-ipv6"), feature = "proto-ipv4"))]
|
|
const SRC_ADDR: IpAddress = IpAddress::Ipv4(Ipv4Address([192, 168, 1, 1]));
|
|
#[cfg(all(not(feature = "proto-ipv6"), feature = "proto-ipv4"))]
|
|
const DST_ADDR: IpAddress = IpAddress::Ipv4(Ipv4Address([192, 168, 1, 2]));
|
|
|
|
#[bench]
|
|
#[cfg(any(feature = "proto-ipv6", feature = "proto-ipv4"))]
|
|
fn bench_emit_tcp(b: &mut test::Bencher) {
|
|
static PAYLOAD_BYTES: [u8; 400] = [0x2a; 400];
|
|
let repr = TcpRepr {
|
|
src_port: 48896,
|
|
dst_port: 80,
|
|
control: TcpControl::Syn,
|
|
seq_number: TcpSeqNumber(0x01234567),
|
|
ack_number: None,
|
|
window_len: 0x0123,
|
|
window_scale: None,
|
|
max_seg_size: None,
|
|
sack_permitted: false,
|
|
sack_ranges: [None, None, None],
|
|
payload: &PAYLOAD_BYTES,
|
|
};
|
|
let mut bytes = vec![0xa5; repr.buffer_len()];
|
|
|
|
b.iter(|| {
|
|
let mut packet = TcpPacket::new_unchecked(&mut bytes);
|
|
repr.emit(
|
|
&mut packet,
|
|
&SRC_ADDR,
|
|
&DST_ADDR,
|
|
&ChecksumCapabilities::default(),
|
|
);
|
|
});
|
|
}
|
|
|
|
#[bench]
|
|
#[cfg(any(feature = "proto-ipv6", feature = "proto-ipv4"))]
|
|
fn bench_emit_udp(b: &mut test::Bencher) {
|
|
static PAYLOAD_BYTES: [u8; 400] = [0x2a; 400];
|
|
let repr = UdpRepr {
|
|
src_port: 48896,
|
|
dst_port: 80,
|
|
};
|
|
let mut bytes = vec![0xa5; repr.header_len() + PAYLOAD_BYTES.len()];
|
|
|
|
b.iter(|| {
|
|
let mut packet = UdpPacket::new_unchecked(&mut bytes);
|
|
repr.emit(
|
|
&mut packet,
|
|
&SRC_ADDR,
|
|
&DST_ADDR,
|
|
PAYLOAD_BYTES.len(),
|
|
|buf| buf.copy_from_slice(&PAYLOAD_BYTES),
|
|
&ChecksumCapabilities::default(),
|
|
);
|
|
});
|
|
}
|
|
|
|
#[bench]
|
|
#[cfg(feature = "proto-ipv4")]
|
|
fn bench_emit_ipv4(b: &mut test::Bencher) {
|
|
let repr = Ipv4Repr {
|
|
src_addr: Ipv4Address([192, 168, 1, 1]),
|
|
dst_addr: Ipv4Address([192, 168, 1, 2]),
|
|
protocol: IpProtocol::Tcp,
|
|
payload_len: 100,
|
|
hop_limit: 64,
|
|
};
|
|
let mut bytes = vec![0xa5; repr.buffer_len()];
|
|
|
|
b.iter(|| {
|
|
let mut packet = Ipv4Packet::new_unchecked(&mut bytes);
|
|
repr.emit(&mut packet, &ChecksumCapabilities::default());
|
|
});
|
|
}
|
|
|
|
#[bench]
|
|
#[cfg(feature = "proto-ipv6")]
|
|
fn bench_emit_ipv6(b: &mut test::Bencher) {
|
|
let repr = Ipv6Repr {
|
|
src_addr: Ipv6Address([0xfe, 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1]),
|
|
dst_addr: Ipv6Address([0xfe, 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2]),
|
|
next_header: IpProtocol::Tcp,
|
|
payload_len: 100,
|
|
hop_limit: 64,
|
|
};
|
|
let mut bytes = vec![0xa5; repr.buffer_len()];
|
|
|
|
b.iter(|| {
|
|
let mut packet = Ipv6Packet::new_unchecked(&mut bytes);
|
|
repr.emit(&mut packet);
|
|
});
|
|
}
|
|
}
|