#[cfg(feature = "async")]
use core::task::Waker;
use core::{cmp, fmt, mem};
#[cfg(feature = "async")]
use crate::socket::WakerRegistration;
use crate::socket::{Context, PollAt};
use crate::storage::{Assembler, RingBuffer};
use crate::time::{Duration, Instant};
use crate::wire::{
IpAddress, IpEndpoint, IpProtocol, IpRepr, TcpControl, TcpRepr, TcpSeqNumber, TCP_HEADER_LEN,
};
use crate::{Error, Result};
pub type SocketBuffer<'a> = RingBuffer<'a, u8>;
#[derive(Debug, PartialEq, Eq, Clone, Copy)]
#[cfg_attr(feature = "defmt", derive(defmt::Format))]
pub enum State {
Closed,
Listen,
SynSent,
SynReceived,
Established,
FinWait1,
FinWait2,
CloseWait,
Closing,
LastAck,
TimeWait,
}
impl fmt::Display for State {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match *self {
State::Closed => write!(f, "CLOSED"),
State::Listen => write!(f, "LISTEN"),
State::SynSent => write!(f, "SYN-SENT"),
State::SynReceived => write!(f, "SYN-RECEIVED"),
State::Established => write!(f, "ESTABLISHED"),
State::FinWait1 => write!(f, "FIN-WAIT-1"),
State::FinWait2 => write!(f, "FIN-WAIT-2"),
State::CloseWait => write!(f, "CLOSE-WAIT"),
State::Closing => write!(f, "CLOSING"),
State::LastAck => write!(f, "LAST-ACK"),
State::TimeWait => write!(f, "TIME-WAIT"),
}
}
}
const RTTE_INITIAL_RTT: u32 = 300;
const RTTE_INITIAL_DEV: u32 = 100;
const RTTE_MIN_MARGIN: u32 = 5;
const RTTE_MIN_RTO: u32 = 10;
const RTTE_MAX_RTO: u32 = 10000;
#[derive(Debug, Clone, Copy)]
#[cfg_attr(feature = "defmt", derive(defmt::Format))]
struct RttEstimator {
rtt: u32,
deviation: u32,
timestamp: Option<(Instant, TcpSeqNumber)>,
max_seq_sent: Option<TcpSeqNumber>,
rto_count: u8,
}
impl Default for RttEstimator {
fn default() -> Self {
Self {
rtt: RTTE_INITIAL_RTT,
deviation: RTTE_INITIAL_DEV,
timestamp: None,
max_seq_sent: None,
rto_count: 0,
}
}
}
impl RttEstimator {
fn retransmission_timeout(&self) -> Duration {
let margin = RTTE_MIN_MARGIN.max(self.deviation * 4);
let ms = (self.rtt + margin).max(RTTE_MIN_RTO).min(RTTE_MAX_RTO);
Duration::from_millis(ms as u64)
}
fn sample(&mut self, new_rtt: u32) {
self.rtt = (self.rtt * 7 + new_rtt + 7) / 8;
let diff = (self.rtt as i32 - new_rtt as i32).abs() as u32;
self.deviation = (self.deviation * 3 + diff + 3) / 4;
self.rto_count = 0;
let rto = self.retransmission_timeout().total_millis();
net_trace!(
"rtte: sample={:?} rtt={:?} dev={:?} rto={:?}",
new_rtt,
self.rtt,
self.deviation,
rto
);
}
fn on_send(&mut self, timestamp: Instant, seq: TcpSeqNumber) {
if self
.max_seq_sent
.map(|max_seq_sent| seq > max_seq_sent)
.unwrap_or(true)
{
self.max_seq_sent = Some(seq);
if self.timestamp.is_none() {
self.timestamp = Some((timestamp, seq));
net_trace!("rtte: sampling at seq={:?}", seq);
}
}
}
fn on_ack(&mut self, timestamp: Instant, seq: TcpSeqNumber) {
if let Some((sent_timestamp, sent_seq)) = self.timestamp {
if seq >= sent_seq {
self.sample((timestamp - sent_timestamp).total_millis() as u32);
self.timestamp = None;
}
}
}
fn on_retransmit(&mut self) {
if self.timestamp.is_some() {
net_trace!("rtte: abort sampling due to retransmit");
}
self.timestamp = None;
self.rto_count = self.rto_count.saturating_add(1);
if self.rto_count >= 3 {
self.rto_count = 0;
self.rtt = RTTE_MAX_RTO.min(self.rtt * 2);
let rto = self.retransmission_timeout().total_millis();
net_trace!(
"rtte: too many retransmissions, increasing: rtt={:?} dev={:?} rto={:?}",
self.rtt,
self.deviation,
rto
);
}
}
}
#[derive(Debug, Clone, Copy, PartialEq)]
#[cfg_attr(feature = "defmt", derive(defmt::Format))]
enum Timer {
Idle {
keep_alive_at: Option<Instant>,
},
Retransmit {
expires_at: Instant,
delay: Duration,
},
FastRetransmit,
Close {
expires_at: Instant,
},
}
const ACK_DELAY_DEFAULT: Duration = Duration::from_millis(10);
const CLOSE_DELAY: Duration = Duration::from_millis(10_000);
impl Timer {
fn new() -> Timer {
Timer::Idle {
keep_alive_at: None,
}
}
fn should_keep_alive(&self, timestamp: Instant) -> bool {
match *self {
Timer::Idle {
keep_alive_at: Some(keep_alive_at),
} if timestamp >= keep_alive_at => true,
_ => false,
}
}
fn should_retransmit(&self, timestamp: Instant) -> Option<Duration> {
match *self {
Timer::Retransmit { expires_at, delay } if timestamp >= expires_at => {
Some(timestamp - expires_at + delay)
}
Timer::FastRetransmit => Some(Duration::from_millis(0)),
_ => None,
}
}
fn should_close(&self, timestamp: Instant) -> bool {
match *self {
Timer::Close { expires_at } if timestamp >= expires_at => true,
_ => false,
}
}
fn poll_at(&self) -> PollAt {
match *self {
Timer::Idle {
keep_alive_at: Some(keep_alive_at),
} => PollAt::Time(keep_alive_at),
Timer::Idle {
keep_alive_at: None,
} => PollAt::Ingress,
Timer::Retransmit { expires_at, .. } => PollAt::Time(expires_at),
Timer::FastRetransmit => PollAt::Now,
Timer::Close { expires_at } => PollAt::Time(expires_at),
}
}
fn set_for_idle(&mut self, timestamp: Instant, interval: Option<Duration>) {
*self = Timer::Idle {
keep_alive_at: interval.map(|interval| timestamp + interval),
}
}
fn set_keep_alive(&mut self) {
if let Timer::Idle {
ref mut keep_alive_at,
} = *self
{
if keep_alive_at.is_none() {
*keep_alive_at = Some(Instant::from_millis(0))
}
}
}
fn rewind_keep_alive(&mut self, timestamp: Instant, interval: Option<Duration>) {
if let Timer::Idle {
ref mut keep_alive_at,
} = *self
{
*keep_alive_at = interval.map(|interval| timestamp + interval)
}
}
fn set_for_retransmit(&mut self, timestamp: Instant, delay: Duration) {
match *self {
Timer::Idle { .. } | Timer::FastRetransmit { .. } => {
*self = Timer::Retransmit {
expires_at: timestamp + delay,
delay: delay,
}
}
Timer::Retransmit { expires_at, delay } if timestamp >= expires_at => {
*self = Timer::Retransmit {
expires_at: timestamp + delay,
delay: delay * 2,
}
}
Timer::Retransmit { .. } => (),
Timer::Close { .. } => (),
}
}
fn set_for_fast_retransmit(&mut self) {
*self = Timer::FastRetransmit
}
fn set_for_close(&mut self, timestamp: Instant) {
*self = Timer::Close {
expires_at: timestamp + CLOSE_DELAY,
}
}
fn is_retransmit(&self) -> bool {
match *self {
Timer::Retransmit { .. } | Timer::FastRetransmit => true,
_ => false,
}
}
}
#[derive(Debug, PartialEq, Eq, Clone, Copy)]
enum AckDelayTimer {
Idle,
Waiting(Instant),
Immediate,
}
#[derive(Debug)]
pub struct TcpSocket<'a> {
state: State,
timer: Timer,
rtte: RttEstimator,
assembler: Assembler,
rx_buffer: SocketBuffer<'a>,
rx_fin_received: bool,
tx_buffer: SocketBuffer<'a>,
timeout: Option<Duration>,
keep_alive: Option<Duration>,
hop_limit: Option<u8>,
listen_address: IpAddress,
local_endpoint: IpEndpoint,
remote_endpoint: IpEndpoint,
local_seq_no: TcpSeqNumber,
remote_seq_no: TcpSeqNumber,
remote_last_seq: TcpSeqNumber,
remote_last_ack: Option<TcpSeqNumber>,
remote_last_win: u16,
remote_win_shift: u8,
remote_win_len: usize,
remote_win_scale: Option<u8>,
remote_has_sack: bool,
remote_mss: usize,
remote_last_ts: Option<Instant>,
local_rx_last_seq: Option<TcpSeqNumber>,
local_rx_last_ack: Option<TcpSeqNumber>,
local_rx_dup_acks: u8,
ack_delay: Option<Duration>,
ack_delay_timer: AckDelayTimer,
challenge_ack_timer: Instant,
nagle: bool,
#[cfg(feature = "async")]
rx_waker: WakerRegistration,
#[cfg(feature = "async")]
tx_waker: WakerRegistration,
}
const DEFAULT_MSS: usize = 536;
impl<'a> TcpSocket<'a> {
#[allow(unused_comparisons)] pub fn new<T>(rx_buffer: T, tx_buffer: T) -> TcpSocket<'a>
where
T: Into<SocketBuffer<'a>>,
{
let (rx_buffer, tx_buffer) = (rx_buffer.into(), tx_buffer.into());
let rx_capacity = rx_buffer.capacity();
if rx_capacity > (1 << 30) {
panic!("receiving buffer too large, cannot exceed 1 GiB")
}
let rx_cap_log2 = mem::size_of::<usize>() * 8 - rx_capacity.leading_zeros() as usize;
TcpSocket {
state: State::Closed,
timer: Timer::new(),
rtte: RttEstimator::default(),
assembler: Assembler::new(rx_buffer.capacity()),
tx_buffer: tx_buffer,
rx_buffer: rx_buffer,
rx_fin_received: false,
timeout: None,
keep_alive: None,
hop_limit: None,
listen_address: IpAddress::default(),
local_endpoint: IpEndpoint::default(),
remote_endpoint: IpEndpoint::default(),
local_seq_no: TcpSeqNumber::default(),
remote_seq_no: TcpSeqNumber::default(),
remote_last_seq: TcpSeqNumber::default(),
remote_last_ack: None,
remote_last_win: 0,
remote_win_len: 0,
remote_win_shift: rx_cap_log2.saturating_sub(16) as u8,
remote_win_scale: None,
remote_has_sack: false,
remote_mss: DEFAULT_MSS,
remote_last_ts: None,
local_rx_last_ack: None,
local_rx_last_seq: None,
local_rx_dup_acks: 0,
ack_delay: Some(ACK_DELAY_DEFAULT),
ack_delay_timer: AckDelayTimer::Idle,
challenge_ack_timer: Instant::from_secs(0),
nagle: true,
#[cfg(feature = "async")]
rx_waker: WakerRegistration::new(),
#[cfg(feature = "async")]
tx_waker: WakerRegistration::new(),
}
}
#[cfg(feature = "async")]
pub fn register_recv_waker(&mut self, waker: &Waker) {
self.rx_waker.register(waker)
}
#[cfg(feature = "async")]
pub fn register_send_waker(&mut self, waker: &Waker) {
self.tx_waker.register(waker)
}
pub fn timeout(&self) -> Option<Duration> {
self.timeout
}
pub fn ack_delay(&self) -> Option<Duration> {
self.ack_delay
}
pub fn nagle_enabled(&self) -> bool {
self.nagle
}
#[inline]
fn scaled_window(&self) -> u16 {
cmp::min(
self.rx_buffer.window() >> self.remote_win_shift as usize,
(1 << 16) - 1,
) as u16
}
pub fn set_timeout(&mut self, duration: Option<Duration>) {
self.timeout = duration
}
pub fn set_ack_delay(&mut self, duration: Option<Duration>) {
self.ack_delay = duration
}
pub fn set_nagle_enabled(&mut self, enabled: bool) {
self.nagle = enabled
}
pub fn keep_alive(&self) -> Option<Duration> {
self.keep_alive
}
pub fn set_keep_alive(&mut self, interval: Option<Duration>) {
self.keep_alive = interval;
if self.keep_alive.is_some() {
self.timer.set_keep_alive();
}
}
pub fn hop_limit(&self) -> Option<u8> {
self.hop_limit
}
pub fn set_hop_limit(&mut self, hop_limit: Option<u8>) {
if let Some(0) = hop_limit {
panic!("the time-to-live value of a packet must not be zero")
}
self.hop_limit = hop_limit
}
#[inline]
pub fn local_endpoint(&self) -> IpEndpoint {
self.local_endpoint
}
#[inline]
pub fn remote_endpoint(&self) -> IpEndpoint {
self.remote_endpoint
}
#[inline]
pub fn state(&self) -> State {
self.state
}
fn reset(&mut self) {
let rx_cap_log2 =
mem::size_of::<usize>() * 8 - self.rx_buffer.capacity().leading_zeros() as usize;
self.state = State::Closed;
self.timer = Timer::new();
self.rtte = RttEstimator::default();
self.assembler = Assembler::new(self.rx_buffer.capacity());
self.tx_buffer.clear();
self.rx_buffer.clear();
self.rx_fin_received = false;
self.keep_alive = None;
self.timeout = None;
self.hop_limit = None;
self.listen_address = IpAddress::default();
self.local_endpoint = IpEndpoint::default();
self.remote_endpoint = IpEndpoint::default();
self.local_seq_no = TcpSeqNumber::default();
self.remote_seq_no = TcpSeqNumber::default();
self.remote_last_seq = TcpSeqNumber::default();
self.remote_last_ack = None;
self.remote_last_win = 0;
self.remote_win_len = 0;
self.remote_win_scale = None;
self.remote_win_shift = rx_cap_log2.saturating_sub(16) as u8;
self.remote_mss = DEFAULT_MSS;
self.remote_last_ts = None;
self.ack_delay = Some(ACK_DELAY_DEFAULT);
self.ack_delay_timer = AckDelayTimer::Idle;
self.challenge_ack_timer = Instant::from_secs(0);
self.nagle = true;
#[cfg(feature = "async")]
{
self.rx_waker.wake();
self.tx_waker.wake();
}
}
pub fn listen<T>(&mut self, local_endpoint: T) -> Result<()>
where
T: Into<IpEndpoint>,
{
let local_endpoint = local_endpoint.into();
if local_endpoint.port == 0 {
return Err(Error::Unaddressable);
}
if self.is_open() {
return Err(Error::Illegal);
}
self.reset();
self.listen_address = local_endpoint.addr;
self.local_endpoint = local_endpoint;
self.remote_endpoint = IpEndpoint::default();
self.set_state(State::Listen);
Ok(())
}
pub fn connect<T, U>(
&mut self,
cx: &mut Context,
remote_endpoint: T,
local_endpoint: U,
) -> Result<()>
where
T: Into<IpEndpoint>,
U: Into<IpEndpoint>,
{
let remote_endpoint = remote_endpoint.into();
let local_endpoint = local_endpoint.into();
if self.is_open() {
return Err(Error::Illegal);
}
if !remote_endpoint.is_specified() {
return Err(Error::Unaddressable);
}
if local_endpoint.port == 0 {
return Err(Error::Unaddressable);
}
let local_addr = match local_endpoint.addr {
IpAddress::Unspecified => remote_endpoint.addr.as_unspecified(),
ip => ip,
};
let local_endpoint = IpEndpoint {
addr: local_addr,
..local_endpoint
};
self.reset();
self.local_endpoint = local_endpoint;
self.remote_endpoint = remote_endpoint;
self.set_state(State::SynSent);
let seq = Self::random_seq_no(cx);
self.local_seq_no = seq;
self.remote_last_seq = seq;
Ok(())
}
#[cfg(test)]
fn random_seq_no(_cx: &mut Context) -> TcpSeqNumber {
TcpSeqNumber(10000)
}
#[cfg(not(test))]
fn random_seq_no(cx: &mut Context) -> TcpSeqNumber {
TcpSeqNumber(cx.rand().rand_u32() as i32)
}
pub fn close(&mut self) {
match self.state {
State::Listen => self.set_state(State::Closed),
State::SynSent => self.set_state(State::Closed),
State::SynReceived | State::Established => self.set_state(State::FinWait1),
State::CloseWait => self.set_state(State::LastAck),
State::FinWait1
| State::FinWait2
| State::Closing
| State::TimeWait
| State::LastAck
| State::Closed => (),
}
}
pub fn abort(&mut self) {
self.set_state(State::Closed);
}
#[inline]
pub fn is_listening(&self) -> bool {
match self.state {
State::Listen => true,
_ => false,
}
}
#[inline]
pub fn is_open(&self) -> bool {
match self.state {
State::Closed => false,
State::TimeWait => false,
_ => true,
}
}
#[inline]
pub fn is_active(&self) -> bool {
match self.state {
State::Closed => false,
State::TimeWait => false,
State::Listen => false,
_ => true,
}
}
#[inline]
pub fn may_send(&self) -> bool {
match self.state {
State::Established => true,
State::CloseWait => true,
_ => false,
}
}
#[inline]
pub fn may_recv(&self) -> bool {
match self.state {
State::Established => true,
State::FinWait1 | State::FinWait2 => true,
_ if !self.rx_buffer.is_empty() => true,
_ => false,
}
}
#[inline]
pub fn can_send(&self) -> bool {
if !self.may_send() {
return false;
}
!self.tx_buffer.is_full()
}
#[inline]
pub fn recv_capacity(&self) -> usize {
self.rx_buffer.capacity()
}
#[inline]
pub fn send_capacity(&self) -> usize {
self.tx_buffer.capacity()
}
#[inline]
pub fn can_recv(&self) -> bool {
if !self.may_recv() {
return false;
}
!self.rx_buffer.is_empty()
}
fn send_impl<'b, F, R>(&'b mut self, f: F) -> Result<R>
where
F: FnOnce(&'b mut SocketBuffer<'a>) -> (usize, R),
{
if !self.may_send() {
return Err(Error::Illegal);
}
if self.tx_buffer.is_empty() {
self.remote_last_ts = None
}
let _old_length = self.tx_buffer.len();
let (size, result) = f(&mut self.tx_buffer);
if size > 0 {
#[cfg(any(test, feature = "verbose"))]
net_trace!(
"tcp:{}:{}: tx buffer: enqueueing {} octets (now {})",
self.local_endpoint,
self.remote_endpoint,
size,
_old_length + size
);
}
Ok(result)
}
pub fn send<'b, F, R>(&'b mut self, f: F) -> Result<R>
where
F: FnOnce(&'b mut [u8]) -> (usize, R),
{
self.send_impl(|tx_buffer| tx_buffer.enqueue_many_with(f))
}
pub fn send_slice(&mut self, data: &[u8]) -> Result<usize> {
self.send_impl(|tx_buffer| {
let size = tx_buffer.enqueue_slice(data);
(size, size)
})
}
fn recv_error_check(&mut self) -> Result<()> {
if !self.may_recv() {
if self.rx_fin_received {
return Err(Error::Finished);
}
return Err(Error::Illegal);
}
Ok(())
}
fn recv_impl<'b, F, R>(&'b mut self, f: F) -> Result<R>
where
F: FnOnce(&'b mut SocketBuffer<'a>) -> (usize, R),
{
self.recv_error_check()?;
let _old_length = self.rx_buffer.len();
let (size, result) = f(&mut self.rx_buffer);
self.remote_seq_no += size;
if size > 0 {
#[cfg(any(test, feature = "verbose"))]
net_trace!(
"tcp:{}:{}: rx buffer: dequeueing {} octets (now {})",
self.local_endpoint,
self.remote_endpoint,
size,
_old_length - size
);
}
Ok(result)
}
pub fn recv<'b, F, R>(&'b mut self, f: F) -> Result<R>
where
F: FnOnce(&'b mut [u8]) -> (usize, R),
{
self.recv_impl(|rx_buffer| rx_buffer.dequeue_many_with(f))
}
pub fn recv_slice(&mut self, data: &mut [u8]) -> Result<usize> {
self.recv_impl(|rx_buffer| {
let size = rx_buffer.dequeue_slice(data);
(size, size)
})
}
pub fn peek(&mut self, size: usize) -> Result<&[u8]> {
self.recv_error_check()?;
let buffer = self.rx_buffer.get_allocated(0, size);
if !buffer.is_empty() {
#[cfg(any(test, feature = "verbose"))]
net_trace!(
"tcp:{}:{}: rx buffer: peeking at {} octets",
self.local_endpoint,
self.remote_endpoint,
buffer.len()
);
}
Ok(buffer)
}
pub fn peek_slice(&mut self, data: &mut [u8]) -> Result<usize> {
let buffer = self.peek(data.len())?;
let data = &mut data[..buffer.len()];
data.copy_from_slice(buffer);
Ok(buffer.len())
}
pub fn send_queue(&self) -> usize {
self.tx_buffer.len()
}
pub fn recv_queue(&self) -> usize {
self.rx_buffer.len()
}
fn set_state(&mut self, state: State) {
if self.state != state {
if self.remote_endpoint.addr.is_unspecified() {
net_trace!(
"tcp:{}: state={}=>{}",
self.local_endpoint,
self.state,
state
);
} else {
net_trace!(
"tcp:{}:{}: state={}=>{}",
self.local_endpoint,
self.remote_endpoint,
self.state,
state
);
}
}
self.state = state;
#[cfg(feature = "async")]
{
self.rx_waker.wake();
self.tx_waker.wake();
}
}
pub(crate) fn reply(ip_repr: &IpRepr, repr: &TcpRepr) -> (IpRepr, TcpRepr<'static>) {
let reply_repr = TcpRepr {
src_port: repr.dst_port,
dst_port: repr.src_port,
control: TcpControl::None,
seq_number: TcpSeqNumber(0),
ack_number: None,
window_len: 0,
window_scale: None,
max_seg_size: None,
sack_permitted: false,
sack_ranges: [None, None, None],
payload: &[],
};
let ip_reply_repr = IpRepr::Unspecified {
src_addr: ip_repr.dst_addr(),
dst_addr: ip_repr.src_addr(),
protocol: IpProtocol::Tcp,
payload_len: reply_repr.buffer_len(),
hop_limit: 64,
};
(ip_reply_repr, reply_repr)
}
pub(crate) fn rst_reply(ip_repr: &IpRepr, repr: &TcpRepr) -> (IpRepr, TcpRepr<'static>) {
debug_assert!(repr.control != TcpControl::Rst);
let (ip_reply_repr, mut reply_repr) = Self::reply(ip_repr, repr);
reply_repr.control = TcpControl::Rst;
reply_repr.seq_number = repr.ack_number.unwrap_or_default();
if repr.control == TcpControl::Syn && repr.ack_number.is_none() {
reply_repr.ack_number = Some(repr.seq_number + repr.segment_len());
}
(ip_reply_repr, reply_repr)
}
fn ack_reply(&mut self, ip_repr: &IpRepr, repr: &TcpRepr) -> (IpRepr, TcpRepr<'static>) {
let (mut ip_reply_repr, mut reply_repr) = Self::reply(ip_repr, repr);
reply_repr.seq_number = self.remote_last_seq;
reply_repr.ack_number = Some(self.remote_seq_no + self.rx_buffer.len());
self.remote_last_ack = reply_repr.ack_number;
reply_repr.window_len = self.scaled_window();
self.remote_last_win = reply_repr.window_len;
if self.remote_has_sack {
net_debug!("sending sACK option with current assembler ranges");
reply_repr.sack_ranges[0] = None;
if let Some(last_seg_seq) = self.local_rx_last_seq.map(|s| s.0 as u32) {
reply_repr.sack_ranges[0] = self
.assembler
.iter_data(reply_repr.ack_number.map(|s| s.0 as usize).unwrap_or(0))
.map(|(left, right)| (left as u32, right as u32))
.find(|(left, right)| *left <= last_seg_seq && *right >= last_seg_seq);
}
if reply_repr.sack_ranges[0].is_none() {
reply_repr.sack_ranges[0] = self
.assembler
.iter_data(reply_repr.ack_number.map(|s| s.0 as usize).unwrap_or(0))
.map(|(left, right)| (left as u32, right as u32))
.next();
}
}
ip_reply_repr.set_payload_len(reply_repr.buffer_len());
(ip_reply_repr, reply_repr)
}
fn challenge_ack_reply(
&mut self,
cx: &mut Context,
ip_repr: &IpRepr,
repr: &TcpRepr,
) -> Option<(IpRepr, TcpRepr<'static>)> {
if cx.now() < self.challenge_ack_timer {
return None;
}
self.challenge_ack_timer = cx.now() + Duration::from_secs(1);
return Some(self.ack_reply(ip_repr, repr));
}
pub(crate) fn accepts(&self, _cx: &mut Context, ip_repr: &IpRepr, repr: &TcpRepr) -> bool {
if self.state == State::Closed {
return false;
}
if self.state == State::Listen && repr.ack_number.is_some() {
return false;
}
if self.local_endpoint.port != repr.dst_port {
return false;
}
if !self.local_endpoint.addr.is_unspecified()
&& self.local_endpoint.addr != ip_repr.dst_addr()
{
return false;
}
if self.remote_endpoint.port != 0 && self.remote_endpoint.port != repr.src_port {
return false;
}
if !self.remote_endpoint.addr.is_unspecified()
&& self.remote_endpoint.addr != ip_repr.src_addr()
{
return false;
}
true
}
pub(crate) fn process(
&mut self,
cx: &mut Context,
ip_repr: &IpRepr,
repr: &TcpRepr,
) -> Result<Option<(IpRepr, TcpRepr<'static>)>> {
debug_assert!(self.accepts(cx, ip_repr, repr));
let (sent_syn, sent_fin) = match self.state {
State::SynSent | State::SynReceived => (true, false),
State::FinWait1 | State::LastAck | State::Closing => (false, true),
_ => (false, false),
};
let control_len = (sent_syn as usize) + (sent_fin as usize);
match (self.state, repr.control, repr.ack_number) {
(State::SynSent, TcpControl::Rst, None) => {
net_debug!(
"tcp:{}:{}: unacceptable RST (expecting RST|ACK) \
in response to initial SYN",
self.local_endpoint,
self.remote_endpoint
);
return Err(Error::Dropped);
}
(State::SynSent, TcpControl::Rst, Some(ack_number)) => {
if ack_number != self.local_seq_no + 1 {
net_debug!(
"tcp:{}:{}: unacceptable RST|ACK in response to initial SYN",
self.local_endpoint,
self.remote_endpoint
);
return Err(Error::Dropped);
}
}
(_, TcpControl::Rst, _) => (),
(State::Listen, _, None) => (),
(State::Listen, _, Some(_)) => unreachable!(),
(_, _, None) => {
net_debug!(
"tcp:{}:{}: expecting an ACK",
self.local_endpoint,
self.remote_endpoint
);
return Err(Error::Dropped);
}
(State::SynSent, TcpControl::Syn, Some(ack_number)) => {
if ack_number != self.local_seq_no + 1 {
net_debug!(
"tcp:{}:{}: unacceptable SYN|ACK in response to initial SYN",
self.local_endpoint,
self.remote_endpoint
);
return Ok(Some(Self::rst_reply(ip_repr, repr)));
}
}
(State::SynSent, TcpControl::None, Some(ack_number)) => {
if ack_number == self.local_seq_no + 1 {
net_debug!(
"tcp:{}:{}: expecting a SYN|ACK, received an ACK with the right ack_number, ignoring.",
self.local_endpoint,
self.remote_endpoint
);
return Err(Error::Dropped);
}
net_debug!(
"tcp:{}:{}: expecting a SYN|ACK, received an ACK with the wrong ack_number, sending RST.",
self.local_endpoint,
self.remote_endpoint
);
return Ok(Some(Self::rst_reply(ip_repr, repr)));
}
(State::SynSent, _, _) => {
net_debug!(
"tcp:{}:{}: expecting a SYN|ACK",
self.local_endpoint,
self.remote_endpoint
);
return Err(Error::Dropped);
}
(State::SynReceived, _, Some(ack_number)) => {
if ack_number != self.local_seq_no + 1 {
net_debug!(
"tcp:{}:{}: unacceptable ACK in response to SYN|ACK",
self.local_endpoint,
self.remote_endpoint
);
return Ok(Some(Self::rst_reply(ip_repr, repr)));
}
}
(_, _, Some(ack_number)) => {
let unacknowledged = self.tx_buffer.len() + control_len;
let mut ack_min = self.local_seq_no;
let ack_max = self.local_seq_no + unacknowledged;
if sent_syn {
ack_min += 1;
}
if ack_number < ack_min {
net_debug!(
"tcp:{}:{}: duplicate ACK ({} not in {}...{})",
self.local_endpoint,
self.remote_endpoint,
ack_number,
ack_min,
ack_max
);
return Err(Error::Dropped);
}
if ack_number > ack_max {
net_debug!(
"tcp:{}:{}: unacceptable ACK ({} not in {}...{})",
self.local_endpoint,
self.remote_endpoint,
ack_number,
ack_min,
ack_max
);
return Ok(self.challenge_ack_reply(cx, ip_repr, repr));
}
}
}
let window_start = self.remote_seq_no + self.rx_buffer.len();
let window_end = self.remote_seq_no + self.rx_buffer.capacity();
let segment_start = repr.seq_number;
let segment_end = repr.seq_number + repr.segment_len();
let payload_offset;
match self.state {
State::Listen | State::SynSent => payload_offset = 0,
_ => {
let mut segment_in_window = true;
if window_start == window_end && segment_start != segment_end {
net_debug!(
"tcp:{}:{}: non-zero-length segment with zero receive window, \
will only send an ACK",
self.local_endpoint,
self.remote_endpoint
);
segment_in_window = false;
}
if segment_start == segment_end && segment_end == window_start - 1 {
net_debug!(
"tcp:{}:{}: received a keep-alive or window probe packet, \
will send an ACK",
self.local_endpoint,
self.remote_endpoint
);
segment_in_window = false;
} else if !((window_start <= segment_start && segment_start <= window_end)
&& (window_start <= segment_end && segment_end <= window_end))
{
net_debug!(
"tcp:{}:{}: segment not in receive window \
({}..{} not intersecting {}..{}), will send challenge ACK",
self.local_endpoint,
self.remote_endpoint,
segment_start,
segment_end,
window_start,
window_end
);
segment_in_window = false;
}
if segment_in_window {
payload_offset = (segment_start - window_start) as usize;
self.local_rx_last_seq = Some(repr.seq_number);
} else {
if self.state == State::TimeWait {
self.timer.set_for_close(cx.now());
}
return Ok(self.challenge_ack_reply(cx, ip_repr, repr));
}
}
}
let mut ack_len = 0;
let mut ack_of_fin = false;
let mut ack_all = false;
if repr.control != TcpControl::Rst {
if let Some(ack_number) = repr.ack_number {
let tx_buffer_start_seq = self.local_seq_no + (sent_syn as usize);
if ack_number >= tx_buffer_start_seq {
ack_len = ack_number - tx_buffer_start_seq;
if sent_fin && self.tx_buffer.len() + 1 == ack_len {
ack_len -= 1;
net_trace!(
"tcp:{}:{}: received ACK of FIN",
self.local_endpoint,
self.remote_endpoint
);
ack_of_fin = true;
}
ack_all = self.remote_last_seq == ack_number
}
self.rtte.on_ack(cx.now(), ack_number);
}
}
let mut control = repr.control;
control = control.quash_psh();
if control == TcpControl::Fin && window_start != segment_start {
control = TcpControl::None;
}
match (self.state, control) {
(State::Listen, TcpControl::Rst) => return Err(Error::Dropped),
(State::SynReceived, TcpControl::Rst) => {
net_trace!(
"tcp:{}:{}: received RST",
self.local_endpoint,
self.remote_endpoint
);
self.local_endpoint.addr = self.listen_address;
self.remote_endpoint = IpEndpoint::default();
self.set_state(State::Listen);
return Ok(None);
}
(_, TcpControl::Rst) => {
net_trace!(
"tcp:{}:{}: received RST",
self.local_endpoint,
self.remote_endpoint
);
self.set_state(State::Closed);
self.local_endpoint = IpEndpoint::default();
self.remote_endpoint = IpEndpoint::default();
return Ok(None);
}
(State::Listen, TcpControl::Syn) => {
net_trace!("tcp:{}: received SYN", self.local_endpoint);
if let Some(max_seg_size) = repr.max_seg_size {
if max_seg_size == 0 {
net_trace!(
"tcp:{}:{}: received SYNACK with zero MSS, ignoring",
self.local_endpoint,
self.remote_endpoint
);
return Ok(None);
}
self.remote_mss = max_seg_size as usize
}
self.local_endpoint = IpEndpoint::new(ip_repr.dst_addr(), repr.dst_port);
self.remote_endpoint = IpEndpoint::new(ip_repr.src_addr(), repr.src_port);
self.local_seq_no = Self::random_seq_no(cx);
self.remote_seq_no = repr.seq_number + 1;
self.remote_last_seq = self.local_seq_no;
self.remote_has_sack = repr.sack_permitted;
self.remote_win_scale = repr.window_scale;
if self.remote_win_scale.is_none() {
self.remote_win_shift = 0;
}
self.set_state(State::SynReceived);
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
(State::SynReceived, TcpControl::None) => {
self.set_state(State::Established);
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
(State::SynReceived, TcpControl::Fin) => {
self.remote_seq_no += 1;
self.rx_fin_received = true;
self.set_state(State::CloseWait);
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
(State::SynSent, TcpControl::Syn) => {
net_trace!(
"tcp:{}:{}: received SYN|ACK",
self.local_endpoint,
self.remote_endpoint
);
if let Some(max_seg_size) = repr.max_seg_size {
if max_seg_size == 0 {
net_trace!(
"tcp:{}:{}: received SYNACK with zero MSS, ignoring",
self.local_endpoint,
self.remote_endpoint
);
return Ok(None);
}
self.remote_mss = max_seg_size as usize;
}
self.local_endpoint = IpEndpoint::new(ip_repr.dst_addr(), repr.dst_port);
self.remote_seq_no = repr.seq_number + 1;
self.remote_last_seq = self.local_seq_no + 1;
self.remote_last_ack = Some(repr.seq_number);
self.remote_win_scale = repr.window_scale;
if self.remote_win_scale.is_none() {
self.remote_win_shift = 0;
}
self.set_state(State::Established);
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
(State::Established, TcpControl::None) => {
if !self.timer.is_retransmit() || ack_all {
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
}
(State::Established, TcpControl::Fin) => {
self.remote_seq_no += 1;
self.rx_fin_received = true;
self.set_state(State::CloseWait);
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
(State::FinWait1, TcpControl::None) => {
if ack_of_fin {
self.set_state(State::FinWait2);
}
if ack_all {
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
}
(State::FinWait1, TcpControl::Fin) => {
self.remote_seq_no += 1;
self.rx_fin_received = true;
if ack_of_fin {
self.set_state(State::TimeWait);
self.timer.set_for_close(cx.now());
} else {
self.set_state(State::Closing);
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
}
(State::FinWait2, TcpControl::None) => {
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
(State::FinWait2, TcpControl::Fin) => {
self.remote_seq_no += 1;
self.rx_fin_received = true;
self.set_state(State::TimeWait);
self.timer.set_for_close(cx.now());
}
(State::Closing, TcpControl::None) => {
if ack_of_fin {
self.set_state(State::TimeWait);
self.timer.set_for_close(cx.now());
} else {
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
}
(State::CloseWait, TcpControl::None) => {
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
(State::LastAck, TcpControl::None) => {
if ack_of_fin {
self.set_state(State::Closed);
self.local_endpoint = IpEndpoint::default();
self.remote_endpoint = IpEndpoint::default();
} else {
self.timer.set_for_idle(cx.now(), self.keep_alive);
}
}
_ => {
net_debug!(
"tcp:{}:{}: unexpected packet {}",
self.local_endpoint,
self.remote_endpoint,
repr
);
return Err(Error::Dropped);
}
}
self.remote_last_ts = Some(cx.now());
let scale = match repr.control {
TcpControl::Syn => 0,
_ => self.remote_win_scale.unwrap_or(0),
};
self.remote_win_len = (repr.window_len as usize) << (scale as usize);
if ack_len > 0 {
debug_assert!(self.tx_buffer.len() >= ack_len);
net_trace!(
"tcp:{}:{}: tx buffer: dequeueing {} octets (now {})",
self.local_endpoint,
self.remote_endpoint,
ack_len,
self.tx_buffer.len() - ack_len
);
self.tx_buffer.dequeue_allocated(ack_len);
#[cfg(feature = "async")]
self.tx_waker.wake();
}
if let Some(ack_number) = repr.ack_number {
match self.local_rx_last_ack {
Some(ref last_rx_ack)
if repr.payload.is_empty()
&& *last_rx_ack == ack_number
&& ack_number < self.remote_last_seq =>
{
self.local_rx_dup_acks = self.local_rx_dup_acks.saturating_add(1);
net_debug!(
"tcp:{}:{}: received duplicate ACK for seq {} (duplicate nr {}{})",
self.local_endpoint,
self.remote_endpoint,
ack_number,
self.local_rx_dup_acks,
if self.local_rx_dup_acks == u8::max_value() {
"+"
} else {
""
}
);
if self.local_rx_dup_acks == 3 {
self.timer.set_for_fast_retransmit();
net_debug!(
"tcp:{}:{}: started fast retransmit",
self.local_endpoint,
self.remote_endpoint
);
}
}
_ => {
if self.local_rx_dup_acks > 0 {
self.local_rx_dup_acks = 0;
net_debug!(
"tcp:{}:{}: reset duplicate ACK count",
self.local_endpoint,
self.remote_endpoint
);
}
self.local_rx_last_ack = Some(ack_number);
}
};
self.local_seq_no = ack_number;
if self.remote_last_seq < self.local_seq_no {
self.remote_last_seq = self.local_seq_no
}
}
let payload_len = repr.payload.len();
if payload_len == 0 {
return Ok(None);
}
let assembler_was_empty = self.assembler.is_empty();
match self.assembler.add(payload_offset, payload_len) {
Ok(()) => {
debug_assert!(self.assembler.total_size() == self.rx_buffer.capacity());
net_trace!(
"tcp:{}:{}: rx buffer: receiving {} octets at offset {}",
self.local_endpoint,
self.remote_endpoint,
payload_len,
payload_offset
);
let len_written = self
.rx_buffer
.write_unallocated(payload_offset, repr.payload);
debug_assert!(len_written == payload_len);
}
Err(_) => {
net_debug!(
"tcp:{}:{}: assembler: too many holes to add {} octets at offset {}",
self.local_endpoint,
self.remote_endpoint,
payload_len,
payload_offset
);
return Err(Error::Dropped);
}
}
if let Some(contig_len) = self.assembler.remove_front() {
debug_assert!(self.assembler.total_size() == self.rx_buffer.capacity());
net_trace!(
"tcp:{}:{}: rx buffer: enqueueing {} octets (now {})",
self.local_endpoint,
self.remote_endpoint,
contig_len,
self.rx_buffer.len() + contig_len
);
self.rx_buffer.enqueue_unallocated(contig_len);
#[cfg(feature = "async")]
self.rx_waker.wake();
}
if !self.assembler.is_empty() {
net_trace!(
"tcp:{}:{}: assembler: {}",
self.local_endpoint,
self.remote_endpoint,
self.assembler
);
}
if let Some(ack_delay) = self.ack_delay {
if self.ack_to_transmit() || self.window_to_update() {
self.ack_delay_timer = match self.ack_delay_timer {
AckDelayTimer::Idle => {
net_trace!(
"tcp:{}:{}: starting delayed ack timer",
self.local_endpoint,
self.remote_endpoint
);
AckDelayTimer::Waiting(cx.now() + ack_delay)
}
AckDelayTimer::Waiting(_) => {
net_trace!(
"tcp:{}:{}: delayed ack timer already started, forcing expiry",
self.local_endpoint,
self.remote_endpoint
);
AckDelayTimer::Immediate
}
AckDelayTimer::Immediate => {
net_trace!(
"tcp:{}:{}: delayed ack timer already force-expired",
self.local_endpoint,
self.remote_endpoint
);
AckDelayTimer::Immediate
}
};
}
}
if !self.assembler.is_empty() || !assembler_was_empty {
net_trace!(
"tcp:{}:{}: ACKing incoming segment",
self.local_endpoint,
self.remote_endpoint
);
Ok(Some(self.ack_reply(ip_repr, repr)))
} else {
Ok(None)
}
}
fn timed_out(&self, timestamp: Instant) -> bool {
match (self.remote_last_ts, self.timeout) {
(Some(remote_last_ts), Some(timeout)) => timestamp >= remote_last_ts + timeout,
(_, _) => false,
}
}
fn seq_to_transmit(&self, cx: &mut Context) -> bool {
let ip_header_len = match self.local_endpoint.addr {
#[cfg(feature = "proto-ipv4")]
IpAddress::Ipv4(_) => crate::wire::IPV4_HEADER_LEN,
#[cfg(feature = "proto-ipv6")]
IpAddress::Ipv6(_) => crate::wire::IPV6_HEADER_LEN,
IpAddress::Unspecified => unreachable!(),
};
let local_mss = cx.ip_mtu() - ip_header_len - TCP_HEADER_LEN;
let effective_mss = local_mss.min(self.remote_mss);
let data_in_flight = self.remote_last_seq != self.local_seq_no;
if matches!(self.state, State::SynSent | State::SynReceived) && !data_in_flight {
return true;
}
let max_send_seq =
self.local_seq_no + core::cmp::min(self.remote_win_len, self.tx_buffer.len());
let max_send = if max_send_seq >= self.remote_last_seq {
max_send_seq - self.remote_last_seq
} else {
0
};
let mut can_send = max_send != 0;
let can_send_full = max_send >= effective_mss;
let want_fin = match self.state {
State::FinWait1 => true,
State::Closing => true,
State::LastAck => true,
_ => false,
};
if self.nagle && data_in_flight && !can_send_full && !want_fin {
can_send = false;
}
let can_fin = want_fin && self.remote_last_seq == self.local_seq_no + self.tx_buffer.len();
can_send || can_fin
}
fn delayed_ack_expired(&self, timestamp: Instant) -> bool {
match self.ack_delay_timer {
AckDelayTimer::Idle => true,
AckDelayTimer::Waiting(t) => t <= timestamp,
AckDelayTimer::Immediate => true,
}
}
fn ack_to_transmit(&self) -> bool {
if let Some(remote_last_ack) = self.remote_last_ack {
remote_last_ack < self.remote_seq_no + self.rx_buffer.len()
} else {
false
}
}
fn window_to_update(&self) -> bool {
match self.state {
State::SynSent
| State::SynReceived
| State::Established
| State::FinWait1
| State::FinWait2 => self.scaled_window() > self.remote_last_win,
_ => false,
}
}
pub(crate) fn dispatch<F>(&mut self, cx: &mut Context, emit: F) -> Result<()>
where
F: FnOnce(&mut Context, (IpRepr, TcpRepr)) -> Result<()>,
{
if !self.remote_endpoint.is_specified() {
return Err(Error::Exhausted);
}
if self.remote_last_ts.is_none() {
self.remote_last_ts = Some(cx.now());
}
if self.timed_out(cx.now()) {
net_debug!(
"tcp:{}:{}: timeout exceeded",
self.local_endpoint,
self.remote_endpoint
);
self.set_state(State::Closed);
} else if !self.seq_to_transmit(cx) {
if let Some(retransmit_delta) = self.timer.should_retransmit(cx.now()) {
net_debug!(
"tcp:{}:{}: retransmitting at t+{}",
self.local_endpoint,
self.remote_endpoint,
retransmit_delta
);
self.remote_last_seq = self.local_seq_no;
self.timer.set_for_idle(cx.now(), self.keep_alive);
self.rtte.on_retransmit();
}
}
if self.seq_to_transmit(cx) {
net_trace!(
"tcp:{}:{}: outgoing segment will send data or flags",
self.local_endpoint,
self.remote_endpoint
);
} else if self.ack_to_transmit() && self.delayed_ack_expired(cx.now()) {
net_trace!(
"tcp:{}:{}: outgoing segment will acknowledge",
self.local_endpoint,
self.remote_endpoint
);
} else if self.window_to_update() && self.delayed_ack_expired(cx.now()) {
net_trace!(
"tcp:{}:{}: outgoing segment will update window",
self.local_endpoint,
self.remote_endpoint
);
} else if self.state == State::Closed {
net_trace!(
"tcp:{}:{}: outgoing segment will abort connection",
self.local_endpoint,
self.remote_endpoint
);
} else if self.timer.should_keep_alive(cx.now()) {
net_trace!(
"tcp:{}:{}: keep-alive timer expired",
self.local_endpoint,
self.remote_endpoint
);
} else if self.timer.should_close(cx.now()) {
net_trace!(
"tcp:{}:{}: TIME-WAIT timer expired",
self.local_endpoint,
self.remote_endpoint
);
self.reset();
return Err(Error::Exhausted);
} else {
return Err(Error::Exhausted);
}
let mut ip_repr = IpRepr::Unspecified {
src_addr: self.local_endpoint.addr,
dst_addr: self.remote_endpoint.addr,
protocol: IpProtocol::Tcp,
hop_limit: self.hop_limit.unwrap_or(64),
payload_len: 0,
}
.lower(&[])?;
let mut repr = TcpRepr {
src_port: self.local_endpoint.port,
dst_port: self.remote_endpoint.port,
control: TcpControl::None,
seq_number: self.remote_last_seq,
ack_number: Some(self.remote_seq_no + self.rx_buffer.len()),
window_len: self.scaled_window(),
window_scale: None,
max_seg_size: None,
sack_permitted: false,
sack_ranges: [None, None, None],
payload: &[],
};
match self.state {
State::Closed => {
repr.control = TcpControl::Rst;
}
State::Listen => return Err(Error::Exhausted),
State::SynSent | State::SynReceived => {
repr.control = TcpControl::Syn;
repr.window_len = self.rx_buffer.window().min((1 << 16) - 1) as u16;
if self.state == State::SynSent {
repr.ack_number = None;
repr.window_scale = Some(self.remote_win_shift);
repr.sack_permitted = true;
} else {
repr.sack_permitted = self.remote_has_sack;
repr.window_scale = self.remote_win_scale.map(|_| self.remote_win_shift);
}
}
State::Established
| State::FinWait1
| State::Closing
| State::CloseWait
| State::LastAck => {
let win_right_edge = self.local_seq_no + self.remote_win_len;
let win_limit = if win_right_edge >= self.remote_last_seq {
win_right_edge - self.remote_last_seq
} else {
0
};
let size = win_limit
.min(self.remote_mss)
.min(cx.ip_mtu() - ip_repr.buffer_len() - TCP_HEADER_LEN);
let offset = self.remote_last_seq - self.local_seq_no;
repr.payload = self.tx_buffer.get_allocated(offset, size);
if offset + repr.payload.len() == self.tx_buffer.len() {
match self.state {
State::FinWait1 | State::LastAck | State::Closing => {
repr.control = TcpControl::Fin
}
State::Established | State::CloseWait if !repr.payload.is_empty() => {
repr.control = TcpControl::Psh
}
_ => (),
}
}
}
State::FinWait2 | State::TimeWait => {}
}
let is_keep_alive;
if self.timer.should_keep_alive(cx.now()) && repr.is_empty() {
repr.seq_number = repr.seq_number - 1;
repr.payload = b"\x00"; is_keep_alive = true;
} else {
is_keep_alive = false;
}
if is_keep_alive {
net_trace!(
"tcp:{}:{}: sending a keep-alive",
self.local_endpoint,
self.remote_endpoint
);
} else if !repr.payload.is_empty() {
net_trace!(
"tcp:{}:{}: tx buffer: sending {} octets at offset {}",
self.local_endpoint,
self.remote_endpoint,
repr.payload.len(),
self.remote_last_seq - self.local_seq_no
);
}
if repr.control != TcpControl::None || repr.payload.is_empty() {
let flags = match (repr.control, repr.ack_number) {
(TcpControl::Syn, None) => "SYN",
(TcpControl::Syn, Some(_)) => "SYN|ACK",
(TcpControl::Fin, Some(_)) => "FIN|ACK",
(TcpControl::Rst, Some(_)) => "RST|ACK",
(TcpControl::Psh, Some(_)) => "PSH|ACK",
(TcpControl::None, Some(_)) => "ACK",
_ => "<unreachable>",
};
net_trace!(
"tcp:{}:{}: sending {}",
self.local_endpoint,
self.remote_endpoint,
flags
);
}
if repr.control == TcpControl::Syn {
let max_segment_size = cx.ip_mtu() - ip_repr.buffer_len() - TCP_HEADER_LEN;
repr.max_seg_size = Some(max_segment_size as u16);
}
ip_repr.set_payload_len(repr.buffer_len());
emit(cx, (ip_repr, repr))?;
self.timer.rewind_keep_alive(cx.now(), self.keep_alive);
match self.ack_delay_timer {
AckDelayTimer::Idle => {}
AckDelayTimer::Waiting(_) => {
net_trace!(
"tcp:{}:{}: stop delayed ack timer",
self.local_endpoint,
self.remote_endpoint
)
}
AckDelayTimer::Immediate => {
net_trace!(
"tcp:{}:{}: stop delayed ack timer (was force-expired)",
self.local_endpoint,
self.remote_endpoint
)
}
}
self.ack_delay_timer = AckDelayTimer::Idle;
if is_keep_alive {
return Ok(());
}
self.remote_last_seq = repr.seq_number + repr.segment_len();
self.remote_last_ack = repr.ack_number;
self.remote_last_win = repr.window_len;
if repr.segment_len() > 0 {
self.rtte
.on_send(cx.now(), repr.seq_number + repr.segment_len());
}
if !self.seq_to_transmit(cx) && repr.segment_len() > 0 {
self.timer
.set_for_retransmit(cx.now(), self.rtte.retransmission_timeout());
}
if self.state == State::Closed {
self.local_endpoint = IpEndpoint::default();
self.remote_endpoint = IpEndpoint::default();
}
Ok(())
}
#[allow(clippy::if_same_then_else)]
pub(crate) fn poll_at(&self, cx: &mut Context) -> PollAt {
if !self.remote_endpoint.is_specified() {
PollAt::Ingress
} else if self.remote_last_ts.is_none() {
PollAt::Now
} else if self.state == State::Closed {
PollAt::Now
} else if self.seq_to_transmit(cx) {
PollAt::Now
} else {
let want_ack = self.ack_to_transmit() || self.window_to_update();
let delayed_ack_poll_at = match (want_ack, self.ack_delay_timer) {
(false, _) => PollAt::Ingress,
(true, AckDelayTimer::Idle) => PollAt::Now,
(true, AckDelayTimer::Waiting(t)) => PollAt::Time(t),
(true, AckDelayTimer::Immediate) => PollAt::Now,
};
let timeout_poll_at = match (self.remote_last_ts, self.timeout) {
(Some(remote_last_ts), Some(timeout)) => PollAt::Time(remote_last_ts + timeout),
(_, _) => PollAt::Ingress,
};
*[self.timer.poll_at(), timeout_poll_at, delayed_ack_poll_at]
.iter()
.min()
.unwrap_or(&PollAt::Ingress)
}
}
}
impl<'a> fmt::Write for TcpSocket<'a> {
fn write_str(&mut self, slice: &str) -> fmt::Result {
let slice = slice.as_bytes();
if self.send_slice(slice) == Ok(slice.len()) {
Ok(())
} else {
Err(fmt::Error)
}
}
}
#[cfg(test)]
mod test {
use super::*;
use crate::wire::ip::test::{MOCK_IP_ADDR_1, MOCK_IP_ADDR_2, MOCK_IP_ADDR_3, MOCK_UNSPECIFIED};
use crate::wire::{IpAddress, IpCidr, IpRepr};
use core::i32;
use std::ops::{Deref, DerefMut};
use std::vec::Vec;
const LOCAL_PORT: u16 = 80;
const REMOTE_PORT: u16 = 49500;
const LOCAL_END: IpEndpoint = IpEndpoint {
addr: MOCK_IP_ADDR_1,
port: LOCAL_PORT,
};
const REMOTE_END: IpEndpoint = IpEndpoint {
addr: MOCK_IP_ADDR_2,
port: REMOTE_PORT,
};
const LOCAL_SEQ: TcpSeqNumber = TcpSeqNumber(10000);
const REMOTE_SEQ: TcpSeqNumber = TcpSeqNumber(-10001);
const SEND_IP_TEMPL: IpRepr = IpRepr::Unspecified {
src_addr: MOCK_IP_ADDR_1,
dst_addr: MOCK_IP_ADDR_2,
protocol: IpProtocol::Tcp,
payload_len: 20,
hop_limit: 64,
};
const SEND_TEMPL: TcpRepr<'static> = TcpRepr {
src_port: REMOTE_PORT,
dst_port: LOCAL_PORT,
control: TcpControl::None,
seq_number: TcpSeqNumber(0),
ack_number: Some(TcpSeqNumber(0)),
window_len: 256,
window_scale: None,
max_seg_size: None,
sack_permitted: false,
sack_ranges: [None, None, None],
payload: &[],
};
const _RECV_IP_TEMPL: IpRepr = IpRepr::Unspecified {
src_addr: MOCK_IP_ADDR_1,
dst_addr: MOCK_IP_ADDR_2,
protocol: IpProtocol::Tcp,
payload_len: 20,
hop_limit: 64,
};
const RECV_TEMPL: TcpRepr<'static> = TcpRepr {
src_port: LOCAL_PORT,
dst_port: REMOTE_PORT,
control: TcpControl::None,
seq_number: TcpSeqNumber(0),
ack_number: Some(TcpSeqNumber(0)),
window_len: 64,
window_scale: None,
max_seg_size: None,
sack_permitted: false,
sack_ranges: [None, None, None],
payload: &[],
};
#[cfg(feature = "proto-ipv6")]
const BASE_MSS: u16 = 1440;
#[cfg(all(feature = "proto-ipv4", not(feature = "proto-ipv6")))]
const BASE_MSS: u16 = 1460;
struct TestSocket {
socket: TcpSocket<'static>,
cx: Context<'static>,
}
impl Deref for TestSocket {
type Target = TcpSocket<'static>;
fn deref(&self) -> &Self::Target {
&self.socket
}
}
impl DerefMut for TestSocket {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.socket
}
}
fn send(
socket: &mut TestSocket,
timestamp: Instant,
repr: &TcpRepr,
) -> Result<Option<TcpRepr<'static>>> {
socket.cx.set_now(timestamp);
let ip_repr = IpRepr::Unspecified {
src_addr: MOCK_IP_ADDR_2,
dst_addr: MOCK_IP_ADDR_1,
protocol: IpProtocol::Tcp,
payload_len: repr.buffer_len(),
hop_limit: 64,
};
net_trace!("send: {}", repr);
assert!(socket.socket.accepts(&mut socket.cx, &ip_repr, repr));
match socket.socket.process(&mut socket.cx, &ip_repr, repr) {
Ok(Some((_ip_repr, repr))) => {
net_trace!("recv: {}", repr);
Ok(Some(repr))
}
Ok(None) => Ok(None),
Err(err) => Err(err),
}
}
fn recv<F>(socket: &mut TestSocket, timestamp: Instant, mut f: F)
where
F: FnMut(Result<TcpRepr>),
{
socket.cx.set_now(timestamp);
let result = socket
.socket
.dispatch(&mut socket.cx, |_, (ip_repr, tcp_repr)| {
let ip_repr = ip_repr.lower(&[IpCidr::new(LOCAL_END.addr, 24)]).unwrap();
assert_eq!(ip_repr.protocol(), IpProtocol::Tcp);
assert_eq!(ip_repr.src_addr(), MOCK_IP_ADDR_1);
assert_eq!(ip_repr.dst_addr(), MOCK_IP_ADDR_2);
assert_eq!(ip_repr.payload_len(), tcp_repr.buffer_len());
net_trace!("recv: {}", tcp_repr);
Ok(f(Ok(tcp_repr)))
});
match result {
Ok(()) => (),
Err(e) => f(Err(e)),
}
}
macro_rules! send {
($socket:ident, $repr:expr) =>
(send!($socket, time 0, $repr));
($socket:ident, $repr:expr, $result:expr) =>
(send!($socket, time 0, $repr, $result));
($socket:ident, time $time:expr, $repr:expr) =>
(send!($socket, time $time, $repr, Ok(None)));
($socket:ident, time $time:expr, $repr:expr, $result:expr) =>
(assert_eq!(send(&mut $socket, Instant::from_millis($time), &$repr), $result));
}
macro_rules! recv {
($socket:ident, [$( $repr:expr ),*]) => ({
$( recv!($socket, Ok($repr)); )*
recv!($socket, Err(Error::Exhausted))
});
($socket:ident, $result:expr) =>
(recv!($socket, time 0, $result));
($socket:ident, time $time:expr, $result:expr) =>
(recv(&mut $socket, Instant::from_millis($time), |result| {
let result = result.map(|mut repr| {
repr.control = repr.control.quash_psh();
repr
});
assert_eq!(result, $result)
}));
($socket:ident, time $time:expr, $result:expr, exact) =>
(recv(&mut $socket, Instant::from_millis($time), |repr| assert_eq!(repr, $result)));
}
macro_rules! sanity {
($socket1:expr, $socket2:expr) => {{
let (s1, s2) = ($socket1, $socket2);
assert_eq!(s1.state, s2.state, "state");
assert_eq!(s1.listen_address, s2.listen_address, "listen_address");
assert_eq!(s1.local_endpoint, s2.local_endpoint, "local_endpoint");
assert_eq!(s1.remote_endpoint, s2.remote_endpoint, "remote_endpoint");
assert_eq!(s1.local_seq_no, s2.local_seq_no, "local_seq_no");
assert_eq!(s1.remote_seq_no, s2.remote_seq_no, "remote_seq_no");
assert_eq!(s1.remote_last_seq, s2.remote_last_seq, "remote_last_seq");
assert_eq!(s1.remote_last_ack, s2.remote_last_ack, "remote_last_ack");
assert_eq!(s1.remote_last_win, s2.remote_last_win, "remote_last_win");
assert_eq!(s1.remote_win_len, s2.remote_win_len, "remote_win_len");
assert_eq!(s1.timer, s2.timer, "timer");
}};
}
fn socket() -> TestSocket {
socket_with_buffer_sizes(64, 64)
}
fn socket_with_buffer_sizes(tx_len: usize, rx_len: usize) -> TestSocket {
let rx_buffer = SocketBuffer::new(vec![0; rx_len]);
let tx_buffer = SocketBuffer::new(vec![0; tx_len]);
let mut socket = TcpSocket::new(rx_buffer, tx_buffer);
socket.set_ack_delay(None);
let cx = Context::mock();
TestSocket { socket, cx }
}
fn socket_syn_received_with_buffer_sizes(tx_len: usize, rx_len: usize) -> TestSocket {
let mut s = socket_with_buffer_sizes(tx_len, rx_len);
s.state = State::SynReceived;
s.local_endpoint = LOCAL_END;
s.remote_endpoint = REMOTE_END;
s.local_seq_no = LOCAL_SEQ;
s.remote_seq_no = REMOTE_SEQ + 1;
s.remote_last_seq = LOCAL_SEQ;
s.remote_win_len = 256;
s
}
fn socket_syn_received() -> TestSocket {
socket_syn_received_with_buffer_sizes(64, 64)
}
fn socket_syn_sent_with_buffer_sizes(tx_len: usize, rx_len: usize) -> TestSocket {
let mut s = socket_with_buffer_sizes(tx_len, rx_len);
s.state = State::SynSent;
s.local_endpoint = IpEndpoint::new(MOCK_UNSPECIFIED, LOCAL_PORT);
s.remote_endpoint = REMOTE_END;
s.local_seq_no = LOCAL_SEQ;
s.remote_last_seq = LOCAL_SEQ;
s
}
fn socket_syn_sent() -> TestSocket {
socket_syn_sent_with_buffer_sizes(64, 64)
}
fn socket_syn_sent_with_local_ipendpoint(local: IpEndpoint) -> TestSocket {
let mut s = socket();
s.state = State::SynSent;
s.local_endpoint = local;
s.remote_endpoint = REMOTE_END;
s.local_seq_no = LOCAL_SEQ;
s.remote_last_seq = LOCAL_SEQ;
s
}
fn socket_established_with_buffer_sizes(tx_len: usize, rx_len: usize) -> TestSocket {
let mut s = socket_syn_received_with_buffer_sizes(tx_len, rx_len);
s.state = State::Established;
s.local_seq_no = LOCAL_SEQ + 1;
s.remote_last_seq = LOCAL_SEQ + 1;
s.remote_last_ack = Some(REMOTE_SEQ + 1);
s.remote_last_win = 64;
s
}
fn socket_established() -> TestSocket {
socket_established_with_buffer_sizes(64, 64)
}
fn socket_fin_wait_1() -> TestSocket {
let mut s = socket_established();
s.state = State::FinWait1;
s
}
fn socket_fin_wait_2() -> TestSocket {
let mut s = socket_fin_wait_1();
s.state = State::FinWait2;
s.local_seq_no = LOCAL_SEQ + 1 + 1;
s.remote_last_seq = LOCAL_SEQ + 1 + 1;
s
}
fn socket_closing() -> TestSocket {
let mut s = socket_fin_wait_1();
s.state = State::Closing;
s.remote_last_seq = LOCAL_SEQ + 1 + 1;
s.remote_seq_no = REMOTE_SEQ + 1 + 1;
s
}
fn socket_time_wait(from_closing: bool) -> TestSocket {
let mut s = socket_fin_wait_2();
s.state = State::TimeWait;
s.remote_seq_no = REMOTE_SEQ + 1 + 1;
if from_closing {
s.remote_last_ack = Some(REMOTE_SEQ + 1 + 1);
}
s.timer = Timer::Close {
expires_at: Instant::from_secs(1) + CLOSE_DELAY,
};
s
}
fn socket_close_wait() -> TestSocket {
let mut s = socket_established();
s.state = State::CloseWait;
s.remote_seq_no = REMOTE_SEQ + 1 + 1;
s.remote_last_ack = Some(REMOTE_SEQ + 1 + 1);
s
}
fn socket_last_ack() -> TestSocket {
let mut s = socket_close_wait();
s.state = State::LastAck;
s
}
fn socket_recved() -> TestSocket {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 58,
..RECV_TEMPL
}]
);
s
}
#[test]
fn test_closed_reject() {
let mut s = socket();
assert_eq!(s.state, State::Closed);
let tcp_repr = TcpRepr {
control: TcpControl::Syn,
..SEND_TEMPL
};
assert!(!s.socket.accepts(&mut s.cx, &SEND_IP_TEMPL, &tcp_repr));
}
#[test]
fn test_closed_reject_after_listen() {
let mut s = socket();
s.listen(LOCAL_END).unwrap();
s.close();
let tcp_repr = TcpRepr {
control: TcpControl::Syn,
..SEND_TEMPL
};
assert!(!s.socket.accepts(&mut s.cx, &SEND_IP_TEMPL, &tcp_repr));
}
#[test]
fn test_closed_close() {
let mut s = socket();
s.close();
assert_eq!(s.state, State::Closed);
}
fn socket_listen() -> TestSocket {
let mut s = socket();
s.state = State::Listen;
s.local_endpoint = IpEndpoint::new(IpAddress::default(), LOCAL_PORT);
s
}
#[test]
fn test_listen_sack_option() {
let mut s = socket_listen();
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: None,
sack_permitted: false,
..SEND_TEMPL
}
);
assert!(!s.remote_has_sack);
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}]
);
let mut s = socket_listen();
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: None,
sack_permitted: true,
..SEND_TEMPL
}
);
assert!(s.remote_has_sack);
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
sack_permitted: true,
..RECV_TEMPL
}]
);
}
#[test]
fn test_listen_syn_win_scale_buffers() {
for (buffer_size, shift_amt) in &[
(64, 0),
(128, 0),
(1024, 0),
(65535, 0),
(65536, 1),
(65537, 1),
(131071, 1),
(131072, 2),
(524287, 3),
(524288, 4),
(655350, 4),
(1048576, 5),
] {
let mut s = socket_with_buffer_sizes(64, *buffer_size);
s.state = State::Listen;
s.local_endpoint = IpEndpoint::new(IpAddress::default(), LOCAL_PORT);
assert_eq!(s.remote_win_shift, *shift_amt);
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: None,
window_scale: Some(0),
..SEND_TEMPL
}
);
assert_eq!(s.remote_win_shift, *shift_amt);
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
window_scale: Some(*shift_amt),
window_len: cmp::min(*buffer_size, 65535) as u16,
..RECV_TEMPL
}]
);
}
}
#[test]
fn test_listen_sanity() {
let mut s = socket();
s.listen(LOCAL_PORT).unwrap();
sanity!(s, socket_listen());
}
#[test]
fn test_listen_validation() {
let mut s = socket();
assert_eq!(s.listen(0), Err(Error::Unaddressable));
}
#[test]
fn test_listen_twice() {
let mut s = socket();
assert_eq!(s.listen(80), Ok(()));
assert_eq!(s.listen(80), Err(Error::Illegal));
}
#[test]
fn test_listen_syn() {
let mut s = socket_listen();
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: None,
..SEND_TEMPL
}
);
sanity!(s, socket_syn_received());
}
#[test]
fn test_listen_syn_reject_ack() {
let mut s = socket_listen();
let tcp_repr = TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ),
..SEND_TEMPL
};
assert!(!s.socket.accepts(&mut s.cx, &SEND_IP_TEMPL, &tcp_repr));
assert_eq!(s.state, State::Listen);
}
#[test]
fn test_listen_rst() {
let mut s = socket_listen();
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ,
ack_number: None,
..SEND_TEMPL
},
Err(Error::Dropped)
);
}
#[test]
fn test_listen_close() {
let mut s = socket_listen();
s.close();
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_syn_received_ack() {
let mut s = socket_syn_received();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Established);
sanity!(s, socket_established());
}
#[test]
fn test_syn_received_ack_too_low() {
let mut s = socket_syn_received();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ), ..SEND_TEMPL
},
Ok(Some(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ,
ack_number: None,
window_len: 0,
..RECV_TEMPL
}))
);
assert_eq!(s.state, State::SynReceived);
}
#[test]
fn test_syn_received_ack_too_high() {
let mut s = socket_syn_received();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 2), ..SEND_TEMPL
},
Ok(Some(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ + 2,
ack_number: None,
window_len: 0,
..RECV_TEMPL
}))
);
assert_eq!(s.state, State::SynReceived);
}
#[test]
fn test_syn_received_fin() {
let mut s = socket_syn_received();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6 + 1),
window_len: 58,
..RECV_TEMPL
}]
);
assert_eq!(s.state, State::CloseWait);
let mut s2 = socket_close_wait();
s2.remote_last_ack = Some(REMOTE_SEQ + 1 + 6 + 1);
s2.remote_last_win = 58;
sanity!(s, s2);
}
#[test]
fn test_syn_received_rst() {
let mut s = socket_syn_received();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Listen);
assert_eq!(
s.local_endpoint,
IpEndpoint::new(IpAddress::Unspecified, LOCAL_END.port)
);
assert_eq!(s.remote_endpoint, IpEndpoint::default());
}
#[test]
fn test_syn_received_no_window_scaling() {
let mut s = socket_listen();
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: None,
..SEND_TEMPL
}
);
assert_eq!(s.state(), State::SynReceived);
assert_eq!(s.local_endpoint(), LOCAL_END);
assert_eq!(s.remote_endpoint(), REMOTE_END);
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
window_scale: None,
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
window_scale: None,
..SEND_TEMPL
}
);
assert_eq!(s.remote_win_shift, 0);
assert_eq!(s.remote_win_scale, None);
}
#[test]
fn test_syn_received_window_scaling() {
for scale in 0..14 {
let mut s = socket_listen();
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: None,
window_scale: Some(scale),
..SEND_TEMPL
}
);
assert_eq!(s.state(), State::SynReceived);
assert_eq!(s.local_endpoint(), LOCAL_END);
assert_eq!(s.remote_endpoint(), REMOTE_END);
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
window_scale: Some(0),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
window_scale: None,
..SEND_TEMPL
}
);
assert_eq!(s.remote_win_scale, Some(scale));
}
}
#[test]
fn test_syn_received_close() {
let mut s = socket_syn_received();
s.close();
assert_eq!(s.state, State::FinWait1);
}
#[test]
fn test_connect_validation() {
let mut s = socket();
assert_eq!(
s.socket
.connect(&mut s.cx, (IpAddress::Unspecified, 80), LOCAL_END),
Err(Error::Unaddressable)
);
assert_eq!(
s.socket
.connect(&mut s.cx, REMOTE_END, (MOCK_UNSPECIFIED, 0)),
Err(Error::Unaddressable)
);
assert_eq!(
s.socket
.connect(&mut s.cx, (MOCK_UNSPECIFIED, 0), LOCAL_END),
Err(Error::Unaddressable)
);
assert_eq!(
s.socket
.connect(&mut s.cx, (IpAddress::Unspecified, 80), LOCAL_END),
Err(Error::Unaddressable)
);
s.socket
.connect(&mut s.cx, REMOTE_END, LOCAL_END)
.expect("Connect failed with valid parameters");
assert_eq!(s.local_endpoint(), LOCAL_END);
assert_eq!(s.remote_endpoint(), REMOTE_END);
}
#[test]
fn test_connect() {
let mut s = socket();
s.local_seq_no = LOCAL_SEQ;
s.socket
.connect(&mut s.cx, REMOTE_END, LOCAL_END.port)
.unwrap();
assert_eq!(
s.local_endpoint,
IpEndpoint::new(MOCK_UNSPECIFIED, LOCAL_END.port)
);
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_scale: Some(0),
sack_permitted: true,
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ + 1),
max_seg_size: Some(BASE_MSS - 80),
window_scale: Some(0),
..SEND_TEMPL
}
);
assert_eq!(s.local_endpoint, LOCAL_END);
}
#[test]
fn test_connect_unspecified_local() {
let mut s = socket();
assert_eq!(
s.socket
.connect(&mut s.cx, REMOTE_END, (MOCK_UNSPECIFIED, 80)),
Ok(())
);
s.abort();
assert_eq!(
s.socket
.connect(&mut s.cx, REMOTE_END, (IpAddress::Unspecified, 80)),
Ok(())
);
s.abort();
}
#[test]
fn test_connect_specified_local() {
let mut s = socket();
assert_eq!(
s.socket
.connect(&mut s.cx, REMOTE_END, (MOCK_IP_ADDR_2, 80)),
Ok(())
);
}
#[test]
fn test_connect_twice() {
let mut s = socket();
assert_eq!(
s.socket
.connect(&mut s.cx, REMOTE_END, (IpAddress::Unspecified, 80)),
Ok(())
);
assert_eq!(
s.socket
.connect(&mut s.cx, REMOTE_END, (IpAddress::Unspecified, 80)),
Err(Error::Illegal)
);
}
#[test]
fn test_syn_sent_sanity() {
let mut s = socket();
s.local_seq_no = LOCAL_SEQ;
s.socket.connect(&mut s.cx, REMOTE_END, LOCAL_END).unwrap();
sanity!(s, socket_syn_sent_with_local_ipendpoint(LOCAL_END));
}
#[test]
fn test_syn_sent_syn_ack() {
let mut s = socket_syn_sent();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_scale: Some(0),
sack_permitted: true,
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ + 1),
max_seg_size: Some(BASE_MSS - 80),
window_scale: Some(0),
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}]
);
recv!(s, time 1000, Err(Error::Exhausted));
assert_eq!(s.state, State::Established);
sanity!(s, socket_established());
}
#[test]
fn test_syn_sent_syn_ack_not_incremented() {
let mut s = socket_syn_sent();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_scale: Some(0),
sack_permitted: true,
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ), max_seg_size: Some(BASE_MSS - 80),
window_scale: Some(0),
..SEND_TEMPL
},
Ok(Some(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ,
ack_number: None,
window_len: 0,
..RECV_TEMPL
}))
);
assert_eq!(s.state, State::SynSent);
}
#[test]
fn test_syn_sent_rst() {
let mut s = socket_syn_sent();
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_syn_sent_rst_no_ack() {
let mut s = socket_syn_sent();
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ,
ack_number: None,
..SEND_TEMPL
},
Err(Error::Dropped)
);
assert_eq!(s.state, State::SynSent);
}
#[test]
fn test_syn_sent_rst_bad_ack() {
let mut s = socket_syn_sent();
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ,
ack_number: Some(TcpSeqNumber(1234)),
..SEND_TEMPL
},
Err(Error::Dropped)
);
assert_eq!(s.state, State::SynSent);
}
#[test]
fn test_syn_sent_bad_ack() {
let mut s = socket_syn_sent();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_scale: Some(0),
sack_permitted: true,
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::None, seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ + 1), ..SEND_TEMPL
},
Err(Error::Dropped)
);
recv!(s, []);
assert_eq!(s.state, State::SynSent);
}
#[test]
fn test_syn_sent_bad_ack_seq_1() {
let mut s = socket_syn_sent();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_scale: Some(0),
sack_permitted: true,
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::None,
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ), ..SEND_TEMPL
},
Ok(Some(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ, ack_number: None,
window_len: 0,
..RECV_TEMPL
}))
);
assert_eq!(s.state, State::SynSent);
}
#[test]
fn test_syn_sent_bad_ack_seq_2() {
let mut s = socket_syn_sent();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_scale: Some(0),
sack_permitted: true,
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::None,
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ + 123456), ..SEND_TEMPL
},
Ok(Some(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ + 123456, ack_number: None,
window_len: 0,
..RECV_TEMPL
}))
);
assert_eq!(s.state, State::SynSent);
}
#[test]
fn test_syn_sent_close() {
let mut s = socket();
s.close();
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_syn_sent_win_scale_buffers() {
for (buffer_size, shift_amt) in &[
(64, 0),
(128, 0),
(1024, 0),
(65535, 0),
(65536, 1),
(65537, 1),
(131071, 1),
(131072, 2),
(524287, 3),
(524288, 4),
(655350, 4),
(1048576, 5),
] {
let mut s = socket_with_buffer_sizes(64, *buffer_size);
s.local_seq_no = LOCAL_SEQ;
assert_eq!(s.remote_win_shift, *shift_amt);
s.socket.connect(&mut s.cx, REMOTE_END, LOCAL_END).unwrap();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_scale: Some(*shift_amt),
window_len: cmp::min(*buffer_size, 65535) as u16,
sack_permitted: true,
..RECV_TEMPL
}]
);
}
}
#[test]
fn test_syn_sent_syn_ack_no_window_scaling() {
let mut s = socket_syn_sent_with_buffer_sizes(1048576, 1048576);
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_len: 65535,
window_scale: Some(5),
sack_permitted: true,
..RECV_TEMPL
}]
);
assert_eq!(s.remote_win_shift, 5);
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ + 1),
max_seg_size: Some(BASE_MSS - 80),
window_scale: None,
window_len: 42,
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Established);
assert_eq!(s.remote_win_shift, 0);
assert_eq!(s.remote_win_scale, None);
assert_eq!(s.remote_win_len, 42);
}
#[test]
fn test_syn_sent_syn_ack_window_scaling() {
let mut s = socket_syn_sent();
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_scale: Some(0),
sack_permitted: true,
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ + 1),
max_seg_size: Some(BASE_MSS - 80),
window_scale: Some(7),
window_len: 42,
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Established);
assert_eq!(s.remote_win_scale, Some(7));
assert_eq!(s.remote_win_len, 42);
}
#[test]
fn test_established_recv() {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 58,
..RECV_TEMPL
}]
);
assert_eq!(s.rx_buffer.dequeue_many(6), &b"abcdef"[..]);
}
fn setup_rfc2018_cases() -> (TestSocket, Vec<u8>) {
let mut s = socket_established_with_buffer_sizes(4000, 4000);
s.remote_has_sack = true;
let mut segment: Vec<u8> = Vec::with_capacity(500);
for _ in 0..50 {
segment.extend_from_slice(b"abcdefghij")
}
for offset in (0..5000).step_by(500) {
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + offset,
ack_number: Some(LOCAL_SEQ + 1),
payload: &segment,
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + offset + 500),
window_len: 3500,
..RECV_TEMPL
}]
);
s.recv(|data| {
assert_eq!(data.len(), 500);
assert_eq!(data, segment.as_slice());
(500, ())
})
.unwrap();
}
assert_eq!(s.remote_last_win, 3500);
(s, segment)
}
#[test]
fn test_established_rfc2018_cases() {
let (mut s, segment) = setup_rfc2018_cases();
for offset in (500..3500).step_by(500) {
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + offset + 5000,
ack_number: Some(LOCAL_SEQ + 1),
payload: &segment,
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 5000),
window_len: 4000,
sack_ranges: [
Some((
REMOTE_SEQ.0 as u32 + 1 + 5500,
REMOTE_SEQ.0 as u32 + 1 + 5500 + offset as u32
)),
None,
None
],
..RECV_TEMPL
}))
);
}
}
#[test]
fn test_established_sliding_window_recv() {
let mut s = socket_established();
assert_eq!(s.rx_buffer.len(), 0);
s.rx_buffer = SocketBuffer::new(vec![0; 262143]);
s.assembler = Assembler::new(s.rx_buffer.capacity());
s.remote_win_scale = Some(0);
s.remote_last_win = 65535;
s.remote_win_shift = 2;
let mut segment: Vec<u8> = Vec::with_capacity(1400);
for _ in 0..100 {
segment.extend_from_slice(b"abcdefghijklmn")
}
assert_eq!(segment.len(), 1400);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &segment,
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1400),
window_len: 65185,
..RECV_TEMPL
}]
);
}
#[test]
fn test_established_send() {
let mut s = socket_established();
s.send_slice(b"abcdef").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
);
assert_eq!(s.tx_buffer.len(), 6);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6),
..SEND_TEMPL
}
);
assert_eq!(s.tx_buffer.len(), 0);
s.send_slice(b"foobar").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"foobar"[..],
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 6),
..SEND_TEMPL
}
);
assert_eq!(s.tx_buffer.len(), 0);
}
#[test]
fn test_established_send_no_ack_send() {
let mut s = socket_established();
s.set_nagle_enabled(false);
s.send_slice(b"abcdef").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
);
s.send_slice(b"foobar").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"foobar"[..],
..RECV_TEMPL
}]
);
}
#[test]
fn test_established_send_buf_gt_win() {
let mut data = [0; 32];
for (i, elem) in data.iter_mut().enumerate() {
*elem = i as u8
}
let mut s = socket_established();
s.remote_win_len = 16;
s.send_slice(&data[..]).unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &data[0..16],
..RECV_TEMPL
}]
);
}
#[test]
fn test_established_send_window_shrink() {
let mut s = socket_established();
s.send_slice(b"abcdef").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
);
assert_eq!(s.tx_buffer.len(), 6);
println!(
"local_seq_no={} remote_win_len={} remote_last_seq={}",
s.local_seq_no, s.remote_win_len, s.remote_last_seq
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
window_len: 3,
payload: &b"xyzxyz"[..],
..SEND_TEMPL
}
);
assert_eq!(s.tx_buffer.len(), 6);
println!(
"local_seq_no={} remote_win_len={} remote_last_seq={}",
s.local_seq_no, s.remote_win_len, s.remote_last_seq
);
s.send_slice(b"foobar").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 64 - 6,
..RECV_TEMPL
}]
);
}
#[test]
fn test_established_send_wrap() {
let mut s = socket_established();
let local_seq_start = TcpSeqNumber(i32::MAX - 1);
s.local_seq_no = local_seq_start + 1;
s.remote_last_seq = local_seq_start + 1;
s.send_slice(b"abc").unwrap();
recv!(s, time 1000, Ok(TcpRepr {
seq_number: local_seq_start + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abc"[..],
..RECV_TEMPL
}));
}
#[test]
fn test_established_no_ack() {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: None,
..SEND_TEMPL
},
Err(Error::Dropped)
);
}
#[test]
fn test_established_bad_ack() {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(TcpSeqNumber(LOCAL_SEQ.0 - 1)),
..SEND_TEMPL
},
Err(Error::Dropped)
);
assert_eq!(s.local_seq_no, LOCAL_SEQ + 1);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 10),
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}))
);
assert_eq!(s.local_seq_no, LOCAL_SEQ + 1);
}
#[test]
fn test_established_bad_seq() {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 256,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}))
);
assert_eq!(s.remote_seq_no, REMOTE_SEQ + 1);
send!(
s,
time 100,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 256,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
},
Ok(None)
);
send!(
s,
time 2000,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 256,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}))
);
assert_eq!(s.remote_seq_no, REMOTE_SEQ + 1);
}
#[test]
fn test_established_fin() {
let mut s = socket_established();
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
assert_eq!(s.state, State::CloseWait);
sanity!(s, socket_close_wait());
}
#[test]
fn test_established_fin_after_missing() {
let mut s = socket_established();
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1 + 6,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"123456"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}))
);
assert_eq!(s.state, State::Established);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6 + 6),
window_len: 52,
..RECV_TEMPL
}))
);
assert_eq!(s.state, State::Established);
}
#[test]
fn test_established_send_fin() {
let mut s = socket_established();
s.send_slice(b"abcdef").unwrap();
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::CloseWait);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
);
}
#[test]
fn test_established_rst() {
let mut s = socket_established();
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_established_rst_no_ack() {
let mut s = socket_established();
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ + 1,
ack_number: None,
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_established_close() {
let mut s = socket_established();
s.close();
assert_eq!(s.state, State::FinWait1);
sanity!(s, socket_fin_wait_1());
}
#[test]
fn test_established_abort() {
let mut s = socket_established();
s.abort();
assert_eq!(s.state, State::Closed);
recv!(
s,
[TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}]
);
}
#[test]
fn test_established_rst_bad_seq() {
let mut s = socket_established();
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ, ack_number: None,
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}))
);
assert_eq!(s.state, State::Established);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1, ack_number: Some(LOCAL_SEQ + 1),
payload: &b"a"[..],
..SEND_TEMPL
}
);
send!(
s,
time 2000,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ, ack_number: None,
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 2), window_len: 63,
..RECV_TEMPL
}))
);
}
#[test]
fn test_fin_wait_1_fin_ack() {
let mut s = socket_fin_wait_1();
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::FinWait2);
sanity!(s, socket_fin_wait_2());
}
#[test]
fn test_fin_wait_1_fin_fin() {
let mut s = socket_fin_wait_1();
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closing);
sanity!(s, socket_closing());
}
#[test]
fn test_fin_wait_1_fin_with_data_queued() {
let mut s = socket_established();
s.remote_win_len = 6;
s.send_slice(b"abcdef123456").unwrap();
s.close();
recv!(
s,
Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
})
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::FinWait1);
}
#[test]
fn test_fin_wait_1_recv() {
let mut s = socket_fin_wait_1();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
assert_eq!(s.state, State::FinWait1);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
}
#[test]
fn test_fin_wait_1_close() {
let mut s = socket_fin_wait_1();
s.close();
assert_eq!(s.state, State::FinWait1);
}
#[test]
fn test_fin_wait_2_fin() {
let mut s = socket_fin_wait_2();
send!(s, time 1_000, TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
});
assert_eq!(s.state, State::TimeWait);
sanity!(s, socket_time_wait(false));
}
#[test]
fn test_fin_wait_2_recv() {
let mut s = socket_fin_wait_2();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
assert_eq!(s.state, State::FinWait2);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 3),
..RECV_TEMPL
}]
);
}
#[test]
fn test_fin_wait_2_close() {
let mut s = socket_fin_wait_2();
s.close();
assert_eq!(s.state, State::FinWait2);
}
#[test]
fn test_closing_ack_fin() {
let mut s = socket_closing();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
send!(s, time 1_000, TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
});
assert_eq!(s.state, State::TimeWait);
sanity!(s, socket_time_wait(true));
}
#[test]
fn test_closing_close() {
let mut s = socket_closing();
s.close();
assert_eq!(s.state, State::Closing);
}
#[test]
fn test_time_wait_from_fin_wait_2_ack() {
let mut s = socket_time_wait(false);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
}
#[test]
fn test_time_wait_from_closing_no_ack() {
let mut s = socket_time_wait(true);
recv!(s, []);
}
#[test]
fn test_time_wait_close() {
let mut s = socket_time_wait(false);
s.close();
assert_eq!(s.state, State::TimeWait);
}
#[test]
fn test_time_wait_retransmit() {
let mut s = socket_time_wait(false);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
send!(s, time 5_000, TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}, Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
})));
assert_eq!(
s.timer,
Timer::Close {
expires_at: Instant::from_secs(5) + CLOSE_DELAY
}
);
}
#[test]
fn test_time_wait_timeout() {
let mut s = socket_time_wait(false);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
assert_eq!(s.state, State::TimeWait);
recv!(s, time 60_000, Err(Error::Exhausted));
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_close_wait_ack() {
let mut s = socket_close_wait();
s.send_slice(b"abcdef").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6),
..SEND_TEMPL
}
);
}
#[test]
fn test_close_wait_close() {
let mut s = socket_close_wait();
s.close();
assert_eq!(s.state, State::LastAck);
sanity!(s, socket_last_ack());
}
#[test]
fn test_last_ack_fin_ack() {
let mut s = socket_last_ack();
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
assert_eq!(s.state, State::LastAck);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_last_ack_ack_not_of_fin() {
let mut s = socket_last_ack();
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
assert_eq!(s.state, State::LastAck);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::LastAck);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_last_ack_close() {
let mut s = socket_last_ack();
s.close();
assert_eq!(s.state, State::LastAck);
}
#[test]
fn test_listen() {
let mut s = socket();
s.listen(IpEndpoint::new(IpAddress::default(), LOCAL_PORT))
.unwrap();
assert_eq!(s.state, State::Listen);
}
#[test]
fn test_three_way_handshake() {
let mut s = socket_listen();
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: None,
..SEND_TEMPL
}
);
assert_eq!(s.state(), State::SynReceived);
assert_eq!(s.local_endpoint(), LOCAL_END);
assert_eq!(s.remote_endpoint(), REMOTE_END);
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state(), State::Established);
assert_eq!(s.local_seq_no, LOCAL_SEQ + 1);
assert_eq!(s.remote_seq_no, REMOTE_SEQ + 1);
}
#[test]
fn test_remote_close() {
let mut s = socket_established();
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::CloseWait);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
s.close();
assert_eq!(s.state, State::LastAck);
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_local_close() {
let mut s = socket_established();
s.close();
assert_eq!(s.state, State::FinWait1);
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::FinWait2);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::TimeWait);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
}
#[test]
fn test_simultaneous_close() {
let mut s = socket_established();
s.close();
assert_eq!(s.state, State::FinWait1);
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closing);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::TimeWait);
recv!(s, []);
}
#[test]
fn test_simultaneous_close_combined_fin_ack() {
let mut s = socket_established();
s.close();
assert_eq!(s.state, State::FinWait1);
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::TimeWait);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
}
#[test]
fn test_simultaneous_close_raced() {
let mut s = socket_established();
s.close();
assert_eq!(s.state, State::FinWait1);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closing);
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
assert_eq!(s.state, State::Closing);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::TimeWait);
recv!(s, []);
}
#[test]
fn test_simultaneous_close_raced_with_data() {
let mut s = socket_established();
s.send_slice(b"abcdef").unwrap();
s.close();
assert_eq!(s.state, State::FinWait1);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closing);
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
);
assert_eq!(s.state, State::Closing);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::TimeWait);
recv!(s, []);
}
#[test]
fn test_fin_with_data() {
let mut s = socket_established();
s.send_slice(b"abcdef").unwrap();
s.close();
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
)
}
#[test]
fn test_mutual_close_with_data_1() {
let mut s = socket_established();
s.send_slice(b"abcdef").unwrap();
s.close();
assert_eq!(s.state, State::FinWait1);
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 1),
..SEND_TEMPL
}
);
}
#[test]
fn test_mutual_close_with_data_2() {
let mut s = socket_established();
s.send_slice(b"abcdef").unwrap();
s.close();
assert_eq!(s.state, State::FinWait1);
recv!(
s,
[TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state, State::FinWait2);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 1),
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}]
);
assert_eq!(s.state, State::TimeWait);
}
#[test]
fn test_duplicate_seq_ack() {
let mut s = socket_recved();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 58,
..RECV_TEMPL
}))
);
}
#[test]
fn test_data_retransmit() {
let mut s = socket_established();
s.send_slice(b"abcdef").unwrap();
recv!(s, time 1000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}));
recv!(s, time 1050, Err(Error::Exhausted));
recv!(s, time 2000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}));
}
#[test]
fn test_data_retransmit_bursts() {
let mut s = socket_established();
s.remote_mss = 6;
s.send_slice(b"abcdef012345").unwrap();
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::None,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::Psh,
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"012345"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 0, Err(Error::Exhausted));
recv!(s, time 50, Err(Error::Exhausted));
recv!(s, time 1000, Ok(TcpRepr {
control: TcpControl::None,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 1500, Ok(TcpRepr {
control: TcpControl::Psh,
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"012345"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 1550, Err(Error::Exhausted));
}
#[test]
fn test_data_retransmit_bursts_half_ack() {
let mut s = socket_established();
s.remote_mss = 6;
s.send_slice(b"abcdef012345").unwrap();
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::None,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::Psh,
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"012345"[..],
..RECV_TEMPL
}), exact);
send!(s, time 5, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6),
window_len: 6,
..SEND_TEMPL
});
recv!(s, time 1500, Ok(TcpRepr {
control: TcpControl::Psh,
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"012345"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 1550, Err(Error::Exhausted));
}
#[test]
fn test_data_retransmit_bursts_half_ack_close() {
let mut s = socket_established();
s.remote_mss = 6;
s.send_slice(b"abcdef012345").unwrap();
s.close();
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::None,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"012345"[..],
..RECV_TEMPL
}), exact);
send!(s, time 5, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6),
window_len: 6,
..SEND_TEMPL
});
recv!(s, time 1500, Ok(TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"012345"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 1550, Err(Error::Exhausted));
}
#[test]
fn test_send_data_after_syn_ack_retransmit() {
let mut s = socket_syn_received();
recv!(s, time 50, Ok(TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}));
recv!(s, time 750, Ok(TcpRepr { control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}));
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.state(), State::Established);
s.send_slice(b"abcdef").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
)
}
#[test]
fn test_established_retransmit_for_dup_ack() {
let mut s = socket_established();
s.send_slice(b"abc").unwrap();
recv!(s, time 1000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abc"[..],
..RECV_TEMPL
}));
assert_eq!(s.tx_buffer.len(), 3);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
recv!(s, time 4000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abc"[..],
..RECV_TEMPL
}));
}
#[test]
fn test_established_retransmit_reset_after_ack() {
let mut s = socket_established();
s.remote_win_len = 6;
s.send_slice(b"abcdef").unwrap();
s.send_slice(b"123456").unwrap();
s.send_slice(b"ABCDEF").unwrap();
recv!(s, time 1000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}));
send!(s, time 1005, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6),
window_len: 6,
..SEND_TEMPL
});
recv!(s, time 1010, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"123456"[..],
..RECV_TEMPL
}));
send!(s, time 1015, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 6),
window_len: 6,
..SEND_TEMPL
});
recv!(s, time 1020, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"ABCDEF"[..],
..RECV_TEMPL
}));
}
#[test]
fn test_established_queue_during_retransmission() {
let mut s = socket_established();
s.remote_mss = 6;
s.send_slice(b"abcdef123456ABCDEF").unwrap();
recv!(s, time 1000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
})); recv!(s, time 1005, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"123456"[..],
..RECV_TEMPL
})); recv!(s, time 1010, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"ABCDEF"[..],
..RECV_TEMPL
})); recv!(s, time 2000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
})); send!(s, time 2005, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 6),
..SEND_TEMPL
}); recv!(s, time 2010, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"ABCDEF"[..],
..RECV_TEMPL
})); }
#[test]
fn test_close_wait_retransmit_reset_after_ack() {
let mut s = socket_close_wait();
s.remote_win_len = 6;
s.send_slice(b"abcdef").unwrap();
s.send_slice(b"123456").unwrap();
s.send_slice(b"ABCDEF").unwrap();
recv!(s, time 1000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}));
send!(s, time 1005, TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6),
window_len: 6,
..SEND_TEMPL
});
recv!(s, time 1010, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1 + 1),
payload: &b"123456"[..],
..RECV_TEMPL
}));
send!(s, time 1015, TcpRepr {
seq_number: REMOTE_SEQ + 1 + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 6),
window_len: 6,
..SEND_TEMPL
});
recv!(s, time 1020, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6 + 6,
ack_number: Some(REMOTE_SEQ + 1 + 1),
payload: &b"ABCDEF"[..],
..RECV_TEMPL
}));
}
#[test]
fn test_fin_wait_1_retransmit_reset_after_ack() {
let mut s = socket_established();
s.remote_win_len = 6;
s.send_slice(b"abcdef").unwrap();
s.send_slice(b"123456").unwrap();
s.send_slice(b"ABCDEF").unwrap();
s.close();
recv!(s, time 1000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}));
send!(s, time 1005, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6),
window_len: 6,
..SEND_TEMPL
});
recv!(s, time 1010, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"123456"[..],
..RECV_TEMPL
}));
send!(s, time 1015, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 6),
window_len: 6,
..SEND_TEMPL
});
recv!(s, time 1020, Ok(TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1 + 6 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"ABCDEF"[..],
..RECV_TEMPL
}));
}
#[test]
fn test_fast_retransmit_after_triple_duplicate_ack() {
let mut s = socket_established();
s.remote_mss = 6;
send!(s, time 0, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
s.send_slice(b"xxxxxxyyyyyywwwwwwzzzzzz").unwrap();
recv!(s, time 1000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"xxxxxx"[..],
..RECV_TEMPL
}));
recv!(s, time 1005, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"yyyyyy"[..],
..RECV_TEMPL
}));
recv!(s, time 1010, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + (6 * 2),
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"wwwwww"[..],
..RECV_TEMPL
}));
recv!(s, time 1015, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + (6 * 3),
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"zzzzzz"[..],
..RECV_TEMPL
}));
send!(s, time 1050, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
send!(s, time 1055, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
send!(s, time 1060, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
recv!(s, time 1100, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"xxxxxx"[..],
..RECV_TEMPL
}));
recv!(s, time 1105, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"yyyyyy"[..],
..RECV_TEMPL
}));
recv!(s, time 1110, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + (6 * 2),
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"wwwwww"[..],
..RECV_TEMPL
}));
recv!(s, time 1115, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + (6 * 3),
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"zzzzzz"[..],
..RECV_TEMPL
}));
assert!(match s.timer {
Timer::Retransmit { expires_at, .. } => expires_at > Instant::from_millis(1115),
_ => false,
});
send!(s, time 1120, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + (6 * 4)),
..SEND_TEMPL
});
}
#[test]
fn test_fast_retransmit_duplicate_detection_with_data() {
let mut s = socket_established();
s.send_slice(b"abc").unwrap(); recv!(s, time 1000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abc"[..],
..RECV_TEMPL
}));
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.local_rx_dup_acks, 2, "duplicate ACK counter is not set");
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"xxxxxx"[..],
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 3,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 58,
..RECV_TEMPL
}]
);
assert_eq!(
s.local_rx_dup_acks, 0,
"duplicate ACK counter is not reset when reciving data"
);
}
#[test]
fn test_fast_retransmit_duplicate_detection() {
let mut s = socket_established();
s.remote_mss = 6;
send!(s, time 0, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
send!(s, time 0, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
assert_eq!(
s.local_rx_dup_acks, 0,
"duplicate ACK counter is set but wound not transmit data"
);
s.send_slice(b"xxxxxxyyyyyywwwwwwzzzzzz").unwrap();
recv!(s, time 1000, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"xxxxxx"[..],
..RECV_TEMPL
}));
recv!(s, time 1005, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"yyyyyy"[..],
..RECV_TEMPL
}));
recv!(s, time 1010, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + (6 * 2),
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"wwwwww"[..],
..RECV_TEMPL
}));
recv!(s, time 1015, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + (6 * 3),
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"zzzzzz"[..],
..RECV_TEMPL
}));
send!(s, time 1050, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
send!(s, time 1055, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
send!(s, time 1060, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + (6 * 3)),
..SEND_TEMPL
});
assert_eq!(
s.local_rx_dup_acks, 0,
"duplicate ACK counter is not reset when reciving ACK which updates send window"
);
send!(s, time 1120, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + (6 * 4)),
..SEND_TEMPL
});
}
#[test]
fn test_fast_retransmit_dup_acks_counter() {
let mut s = socket_established();
s.send_slice(b"abc").unwrap(); recv!(s, time 0, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abc"[..],
..RECV_TEMPL
}));
send!(s, time 0, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
s.local_rx_dup_acks = u8::max_value() - 1;
send!(s, time 0, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
send!(s, time 0, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
send!(s, time 0, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
assert_eq!(
s.local_rx_dup_acks,
u8::max_value(),
"duplicate ACK count should not overflow but saturate"
);
}
#[test]
fn test_fast_retransmit_zero_window() {
let mut s = socket_established();
send!(s, time 1000, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
s.send_slice(b"abc").unwrap();
recv!(s, time 0, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abc"[..],
..RECV_TEMPL
}));
send!(s, time 1050, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
send!(s, time 1050, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
send!(s, time 1050, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
window_len: 0, ..SEND_TEMPL
});
recv!(s, Err(Error::Exhausted));
}
#[test]
fn test_maximum_segment_size() {
let mut s = socket_listen();
s.tx_buffer = SocketBuffer::new(vec![0; 32767]);
send!(
s,
TcpRepr {
control: TcpControl::Syn,
seq_number: REMOTE_SEQ,
ack_number: None,
max_seg_size: Some(1000),
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
max_seg_size: Some(BASE_MSS),
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
window_len: 32767,
..SEND_TEMPL
}
);
s.send_slice(&[0; 1200][..]).unwrap();
recv!(
s,
Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &[0; 1000][..],
..RECV_TEMPL
})
);
}
#[test]
fn test_close_wait_no_window_update() {
let mut s = socket_established();
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &[1, 2, 3, 4],
..SEND_TEMPL
}
);
assert_eq!(s.state, State::CloseWait);
recv!(
s,
Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 6),
window_len: 60,
..RECV_TEMPL
})
);
let rx_buf = &mut [0; 32];
assert_eq!(s.recv_slice(rx_buf), Ok(4));
recv!(s, Err(Error::Exhausted));
}
#[test]
fn test_time_wait_no_window_update() {
let mut s = socket_fin_wait_2();
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 2),
payload: &[1, 2, 3, 4],
..SEND_TEMPL
}
);
assert_eq!(s.state, State::TimeWait);
recv!(
s,
Ok(TcpRepr {
seq_number: LOCAL_SEQ + 2,
ack_number: Some(REMOTE_SEQ + 6),
window_len: 60,
..RECV_TEMPL
})
);
let rx_buf = &mut [0; 32];
assert_eq!(s.recv_slice(rx_buf), Ok(4));
recv!(s, Err(Error::Exhausted));
}
#[test]
fn test_psh_transmit() {
let mut s = socket_established();
s.remote_mss = 6;
s.send_slice(b"abcdef").unwrap();
s.send_slice(b"123456").unwrap();
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::None,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::Psh,
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"123456"[..],
..RECV_TEMPL
}), exact);
}
#[test]
fn test_psh_receive() {
let mut s = socket_established();
send!(
s,
TcpRepr {
control: TcpControl::Psh,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 58,
..RECV_TEMPL
}]
);
}
#[test]
fn test_zero_window_ack() {
let mut s = socket_established();
s.rx_buffer = SocketBuffer::new(vec![0; 6]);
s.assembler = Assembler::new(s.rx_buffer.capacity());
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 0,
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 6,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"123456"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 0,
..RECV_TEMPL
}))
);
}
#[test]
fn test_zero_window_ack_on_window_growth() {
let mut s = socket_established();
s.rx_buffer = SocketBuffer::new(vec![0; 6]);
s.assembler = Assembler::new(s.rx_buffer.capacity());
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 0,
..RECV_TEMPL
}]
);
recv!(s, time 0, Err(Error::Exhausted));
s.recv(|buffer| {
assert_eq!(&buffer[..3], b"abc");
(3, ())
})
.unwrap();
recv!(s, time 0, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 3,
..RECV_TEMPL
}));
recv!(s, time 0, Err(Error::Exhausted));
s.recv(|buffer| {
assert_eq!(buffer, b"def");
(buffer.len(), ())
})
.unwrap();
recv!(s, time 0, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 6,
..RECV_TEMPL
}));
}
#[test]
fn test_fill_peer_window() {
let mut s = socket_established();
s.remote_mss = 6;
s.send_slice(b"abcdef123456!@#$%^").unwrap();
recv!(
s,
[
TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
},
TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"123456"[..],
..RECV_TEMPL
},
TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"!@#$%^"[..],
..RECV_TEMPL
}
]
);
}
#[test]
fn test_announce_window_after_read() {
let mut s = socket_established();
s.rx_buffer = SocketBuffer::new(vec![0; 6]);
s.assembler = Assembler::new(s.rx_buffer.capacity());
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 3),
window_len: 3,
..RECV_TEMPL
}]
);
assert_eq!(s.remote_last_win, s.rx_buffer.window() as u16);
s.recv(|buffer| (buffer.len(), ())).unwrap();
assert!(s.window_to_update());
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 3),
window_len: 6,
..RECV_TEMPL
}]
);
assert_eq!(s.remote_last_win, s.rx_buffer.window() as u16);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 6,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"def"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 3),
window_len: 6,
..RECV_TEMPL
}))
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 3,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 9),
window_len: 0,
..RECV_TEMPL
}))
);
assert_eq!(s.remote_last_win, s.rx_buffer.window() as u16);
s.recv(|buffer| (buffer.len(), ())).unwrap();
assert!(s.window_to_update());
}
#[test]
fn test_listen_timeout() {
let mut s = socket_listen();
s.set_timeout(Some(Duration::from_millis(100)));
assert_eq!(s.socket.poll_at(&mut s.cx), PollAt::Ingress);
}
#[test]
fn test_connect_timeout() {
let mut s = socket();
s.local_seq_no = LOCAL_SEQ;
s.socket
.connect(&mut s.cx, REMOTE_END, LOCAL_END.port)
.unwrap();
s.set_timeout(Some(Duration::from_millis(100)));
recv!(s, time 150, Ok(TcpRepr {
control: TcpControl::Syn,
seq_number: LOCAL_SEQ,
ack_number: None,
max_seg_size: Some(BASE_MSS),
window_scale: Some(0),
sack_permitted: true,
..RECV_TEMPL
}));
assert_eq!(s.state, State::SynSent);
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(250))
);
recv!(s, time 250, Ok(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(TcpSeqNumber(0)),
window_scale: None,
..RECV_TEMPL
}));
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_established_timeout() {
let mut s = socket_established();
s.set_timeout(Some(Duration::from_millis(1000)));
recv!(s, time 250, Err(Error::Exhausted));
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(1250))
);
s.send_slice(b"abcdef").unwrap();
assert_eq!(s.socket.poll_at(&mut s.cx), PollAt::Now);
recv!(s, time 255, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}));
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(955))
);
recv!(s, time 955, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}));
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(1255))
);
recv!(s, time 1255, Ok(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}));
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_established_keep_alive_timeout() {
let mut s = socket_established();
s.set_keep_alive(Some(Duration::from_millis(50)));
s.set_timeout(Some(Duration::from_millis(100)));
recv!(s, time 100, Ok(TcpRepr {
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
payload: &[0],
..RECV_TEMPL
}));
recv!(s, time 100, Err(Error::Exhausted));
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(150))
);
send!(s, time 105, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(155))
);
recv!(s, time 155, Ok(TcpRepr {
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
payload: &[0],
..RECV_TEMPL
}));
recv!(s, time 155, Err(Error::Exhausted));
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(205))
);
recv!(s, time 200, Err(Error::Exhausted));
recv!(s, time 205, Ok(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}));
recv!(s, time 205, Err(Error::Exhausted));
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_fin_wait_1_timeout() {
let mut s = socket_fin_wait_1();
s.set_timeout(Some(Duration::from_millis(1000)));
recv!(s, time 100, Ok(TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}));
recv!(s, time 1100, Ok(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}));
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_last_ack_timeout() {
let mut s = socket_last_ack();
s.set_timeout(Some(Duration::from_millis(1000)));
recv!(s, time 100, Ok(TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}));
recv!(s, time 1100, Ok(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ + 1 + 1,
ack_number: Some(REMOTE_SEQ + 1 + 1),
..RECV_TEMPL
}));
assert_eq!(s.state, State::Closed);
}
#[test]
fn test_closed_timeout() {
let mut s = socket_established();
s.set_timeout(Some(Duration::from_millis(200)));
s.remote_last_ts = Some(Instant::from_millis(100));
s.abort();
assert_eq!(s.socket.poll_at(&mut s.cx), PollAt::Now);
recv!(s, time 100, Ok(TcpRepr {
control: TcpControl::Rst,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}));
assert_eq!(s.socket.poll_at(&mut s.cx), PollAt::Ingress);
}
#[test]
fn test_responds_to_keep_alive() {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}))
);
}
#[test]
fn test_sends_keep_alive() {
let mut s = socket_established();
s.set_keep_alive(Some(Duration::from_millis(100)));
assert_eq!(s.socket.poll_at(&mut s.cx), PollAt::Now);
recv!(s, time 0, Ok(TcpRepr {
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
payload: &[0],
..RECV_TEMPL
}));
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(100))
);
recv!(s, time 95, Err(Error::Exhausted));
recv!(s, time 100, Ok(TcpRepr {
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
payload: &[0],
..RECV_TEMPL
}));
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(200))
);
recv!(s, time 195, Err(Error::Exhausted));
recv!(s, time 200, Ok(TcpRepr {
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
payload: &[0],
..RECV_TEMPL
}));
send!(s, time 250, TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
});
assert_eq!(
s.socket.poll_at(&mut s.cx),
PollAt::Time(Instant::from_millis(350))
);
recv!(s, time 345, Err(Error::Exhausted));
recv!(s, time 350, Ok(TcpRepr {
seq_number: LOCAL_SEQ,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"\x00"[..],
..RECV_TEMPL
}));
}
#[test]
fn test_set_hop_limit() {
let mut s = socket_syn_received();
s.set_hop_limit(Some(0x2a));
assert_eq!(
s.socket.dispatch(&mut s.cx, |_, (ip_repr, _)| {
assert_eq!(ip_repr.hop_limit(), 0x2a);
Ok(())
}),
Ok(())
);
}
#[test]
#[should_panic(expected = "the time-to-live value of a packet must not be zero")]
fn test_set_hop_limit_zero() {
let mut s = socket_syn_received();
s.set_hop_limit(Some(0));
}
#[test]
fn test_out_of_order() {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 3,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"def"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
..RECV_TEMPL
}))
);
s.recv(|buffer| {
assert_eq!(buffer, b"");
(buffer.len(), ())
})
.unwrap();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 58,
..RECV_TEMPL
}))
);
s.recv(|buffer| {
assert_eq!(buffer, b"abcdef");
(buffer.len(), ())
})
.unwrap();
}
#[test]
fn test_buffer_wraparound_rx() {
let mut s = socket_established();
s.rx_buffer = SocketBuffer::new(vec![0; 6]);
s.assembler = Assembler::new(s.rx_buffer.capacity());
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
s.recv(|buffer| {
assert_eq!(buffer, b"abc");
(buffer.len(), ())
})
.unwrap();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 3,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"defghi"[..],
..SEND_TEMPL
}
);
let mut data = [0; 6];
assert_eq!(s.recv_slice(&mut data[..]), Ok(6));
assert_eq!(data, &b"defghi"[..]);
}
#[test]
fn test_buffer_wraparound_tx() {
let mut s = socket_established();
s.set_nagle_enabled(false);
s.tx_buffer = SocketBuffer::new(vec![b'.'; 9]);
assert_eq!(s.send_slice(b"xxxyyy"), Ok(6));
assert_eq!(s.tx_buffer.dequeue_many(3), &b"xxx"[..]);
assert_eq!(s.tx_buffer.len(), 3);
assert_eq!(s.send_slice(b"abcdef"), Ok(6));
recv!(
s,
Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"yyyabc"[..],
..RECV_TEMPL
})
);
recv!(
s,
Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"def"[..],
..RECV_TEMPL
})
);
}
#[test]
fn test_rx_close_fin() {
let mut s = socket_established();
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
assert_eq!(s.recv(|_| (0, ())), Err(Error::Finished));
}
#[test]
fn test_rx_close_fin_in_fin_wait_1() {
let mut s = socket_fin_wait_1();
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
assert_eq!(s.state, State::Closing);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
assert_eq!(s.recv(|_| (0, ())), Err(Error::Finished));
}
#[test]
fn test_rx_close_fin_in_fin_wait_2() {
let mut s = socket_fin_wait_2();
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
assert_eq!(s.state, State::TimeWait);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
assert_eq!(s.recv(|_| (0, ())), Err(Error::Finished));
}
#[test]
fn test_rx_close_fin_with_hole() {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
send!(
s,
TcpRepr {
control: TcpControl::Fin,
seq_number: REMOTE_SEQ + 1 + 6,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"ghi"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 3),
window_len: 61,
..RECV_TEMPL
}))
);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
s.recv(|data| {
assert_eq!(data, b"");
(0, ())
})
.unwrap();
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ + 1 + 9,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
assert_eq!(s.recv(|_| (0, ())), Err(Error::Illegal));
}
#[test]
fn test_rx_close_rst() {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ + 1 + 3,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
assert_eq!(s.recv(|_| (0, ())), Err(Error::Illegal));
}
#[test]
fn test_rx_close_rst_with_hole() {
let mut s = socket_established();
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 6,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"ghi"[..],
..SEND_TEMPL
},
Ok(Some(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 3),
window_len: 61,
..RECV_TEMPL
}))
);
send!(
s,
TcpRepr {
control: TcpControl::Rst,
seq_number: REMOTE_SEQ + 1 + 9,
ack_number: Some(LOCAL_SEQ + 1),
..SEND_TEMPL
}
);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
assert_eq!(s.recv(|_| (0, ())), Err(Error::Illegal));
}
#[test]
fn test_delayed_ack() {
let mut s = socket_established();
s.set_ack_delay(Some(ACK_DELAY_DEFAULT));
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
recv!(s, Err(Error::Exhausted));
recv!(s, time 11, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 3),
window_len: 61,
..RECV_TEMPL
}));
}
#[test]
fn test_delayed_ack_win() {
let mut s = socket_established();
s.set_ack_delay(Some(ACK_DELAY_DEFAULT));
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
recv!(s, Err(Error::Exhausted));
recv!(s, time 11, Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 3),
..RECV_TEMPL
}));
}
#[test]
fn test_delayed_ack_reply() {
let mut s = socket_established();
s.set_ack_delay(Some(ACK_DELAY_DEFAULT));
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
s.recv(|data| {
assert_eq!(data, b"abc");
(3, ())
})
.unwrap();
s.send_slice(&b"xyz"[..]).unwrap();
recv!(
s,
Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 3),
payload: &b"xyz"[..],
..RECV_TEMPL
})
);
}
#[test]
fn test_delayed_ack_every_second_packet() {
let mut s = socket_established();
s.set_ack_delay(Some(ACK_DELAY_DEFAULT));
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
recv!(s, Err(Error::Exhausted));
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 3,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"def"[..],
..SEND_TEMPL
}
);
recv!(
s,
Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 6),
window_len: 58,
..RECV_TEMPL
})
);
}
#[test]
fn test_delayed_ack_three_packets() {
let mut s = socket_established();
s.set_ack_delay(Some(ACK_DELAY_DEFAULT));
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abc"[..],
..SEND_TEMPL
}
);
recv!(s, Err(Error::Exhausted));
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 3,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"def"[..],
..SEND_TEMPL
}
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1 + 6,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"ghi"[..],
..SEND_TEMPL
}
);
recv!(
s,
Ok(TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1 + 9),
window_len: 55,
..RECV_TEMPL
})
);
}
#[test]
fn test_nagle() {
let mut s = socket_established();
s.remote_mss = 6;
s.send_slice(b"abcdef").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}]
);
s.send_slice(b"foobar").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"foobar"[..],
..RECV_TEMPL
}]
);
s.send_slice(b"aaabbbccc").unwrap();
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"aaabbb"[..],
..RECV_TEMPL
}]
);
send!(
s,
TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1 + 6 + 6 + 6),
..SEND_TEMPL
}
);
recv!(
s,
[TcpRepr {
seq_number: LOCAL_SEQ + 1 + 6 + 6 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"ccc"[..],
..RECV_TEMPL
}]
);
}
#[test]
fn test_final_packet_in_stream_doesnt_wait_for_nagle() {
let mut s = socket_established();
s.remote_mss = 6;
s.send_slice(b"abcdef0").unwrap();
s.socket.close();
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::None,
seq_number: LOCAL_SEQ + 1,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"abcdef"[..],
..RECV_TEMPL
}), exact);
recv!(s, time 0, Ok(TcpRepr {
control: TcpControl::Fin,
seq_number: LOCAL_SEQ + 1 + 6,
ack_number: Some(REMOTE_SEQ + 1),
payload: &b"0"[..],
..RECV_TEMPL
}), exact);
}
#[test]
fn test_doesnt_accept_wrong_port() {
let mut s = socket_established();
s.rx_buffer = SocketBuffer::new(vec![0; 6]);
s.assembler = Assembler::new(s.rx_buffer.capacity());
let tcp_repr = TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
dst_port: LOCAL_PORT + 1,
..SEND_TEMPL
};
assert!(!s.socket.accepts(&mut s.cx, &SEND_IP_TEMPL, &tcp_repr));
let tcp_repr = TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
src_port: REMOTE_PORT + 1,
..SEND_TEMPL
};
assert!(!s.socket.accepts(&mut s.cx, &SEND_IP_TEMPL, &tcp_repr));
}
#[test]
fn test_doesnt_accept_wrong_ip() {
let mut s = socket_established();
let tcp_repr = TcpRepr {
seq_number: REMOTE_SEQ + 1,
ack_number: Some(LOCAL_SEQ + 1),
payload: &b"abcdef"[..],
..SEND_TEMPL
};
let ip_repr = IpRepr::Unspecified {
src_addr: MOCK_IP_ADDR_2,
dst_addr: MOCK_IP_ADDR_1,
protocol: IpProtocol::Tcp,
payload_len: tcp_repr.buffer_len(),
hop_limit: 64,
};
assert!(s.socket.accepts(&mut s.cx, &ip_repr, &tcp_repr));
let ip_repr_wrong_src = IpRepr::Unspecified {
src_addr: MOCK_IP_ADDR_3,
dst_addr: MOCK_IP_ADDR_1,
protocol: IpProtocol::Tcp,
payload_len: tcp_repr.buffer_len(),
hop_limit: 64,
};
assert!(!s.socket.accepts(&mut s.cx, &ip_repr_wrong_src, &tcp_repr));
let ip_repr_wrong_dst = IpRepr::Unspecified {
src_addr: MOCK_IP_ADDR_2,
dst_addr: MOCK_IP_ADDR_3,
protocol: IpProtocol::Tcp,
payload_len: tcp_repr.buffer_len(),
hop_limit: 64,
};
assert!(!s.socket.accepts(&mut s.cx, &ip_repr_wrong_dst, &tcp_repr));
}
#[test]
fn test_timer_retransmit() {
const RTO: Duration = Duration::from_millis(100);
let mut r = Timer::new();
assert_eq!(r.should_retransmit(Instant::from_secs(1)), None);
r.set_for_retransmit(Instant::from_millis(1000), RTO);
assert_eq!(r.should_retransmit(Instant::from_millis(1000)), None);
assert_eq!(r.should_retransmit(Instant::from_millis(1050)), None);
assert_eq!(
r.should_retransmit(Instant::from_millis(1101)),
Some(Duration::from_millis(101))
);
r.set_for_retransmit(Instant::from_millis(1101), RTO);
assert_eq!(r.should_retransmit(Instant::from_millis(1101)), None);
assert_eq!(r.should_retransmit(Instant::from_millis(1150)), None);
assert_eq!(r.should_retransmit(Instant::from_millis(1200)), None);
assert_eq!(
r.should_retransmit(Instant::from_millis(1301)),
Some(Duration::from_millis(300))
);
r.set_for_idle(Instant::from_millis(1301), None);
assert_eq!(r.should_retransmit(Instant::from_millis(1350)), None);
}
#[test]
fn test_rtt_estimator() {
let mut r = RttEstimator::default();
let rtos = &[
751, 766, 755, 731, 697, 656, 613, 567, 523, 484, 445, 411, 378, 350, 322, 299, 280,
261, 243, 229, 215, 206, 197, 188,
];
for &rto in rtos {
r.sample(100);
assert_eq!(r.retransmission_timeout(), Duration::from_millis(rto));
}
}
}