CS144-Lab2-TCPReceiver

lab 地址 :lab2-doc
代码实现:lab2-code
完整目录:
0. ByteStream
1. StreamReassembler
2. TCPReceiver
3. TCPSender
4. TCPConnection
5. ARP
6. IP-Router

1. 目标

lab0 实现了一个读写字节流 ByteStream,lab1 实现了可靠有序不重复的字节流组装器 StreamReassembler ,接下来要实现更上层的封装,TCPSenderTCPReceiver,分别负责 TCPConnection 的收发包功能。
首先要实现的是 TCPReceiver

image

除了写入传入流之外,TCPReceiver 负责告诉 TCPSender 两件事:
1.“第一个未组装”字节的索引,称为“确认号”(ackno),这是接收方从发送方需要的第一个字节。
2.“第一个未组装”索引和“第一个不可接受”索引之间的距离。这称为“窗口大小”。
ackno 和窗口大小一起描述了接收方的窗口:允许 TCPSender 发送的一系列索引。使用该窗口,接收方可以控制传入数据的流量,使发送方限制其发送量,直到接收方准备好接收更多数据。我们有时将 ackno 称为窗口的“左边缘”(TCPReceiver 感兴趣的最小索引),将 ackno + 窗口大小称为“右边缘”(刚好超出 TCPReceiver所感兴趣的最大索引)。

2. 实现

正常情况下,TCPReceiver 会收到 3 种报文:

  • SYN 报文,带着初始 ISN ,用来标记字节流的起始位置,通常 ISN 是随机值,防止被攻击
  • FIN 报文,表明通信结束
  • 普通的数据报文,只需要写入 payload 到 ByteStream 即可。
    TCP 报文头部的 seqno 标识了 payload 字节流在完整字节流中的起始位置,然而这个字段只有 32 位,也就是说最多只能支持 4gb 的字节流,这显然是远远不够的,为了解决上面的问题,引入了 absolute sequence number (简称 abs_seqno)的概念,abs_seqno 定义为 uin64_t 类型,这样可以支持最高 2^64 - 1 长度的字节流。两者的区别如下:

image

stream index 其实际上就是 ByteStream 的字节流索引,只是少了 FIN 和 SYN 各自在字节流中的 1 个字节占用,它同样也是 uint64_t 类型。
abs_seqno 的起始位置永远是 0,这意味着它对于 seqno 会有 isn 长度的偏移,每次写入时都不断对其递增,由于其长度更长,即便 seqno 溢出了,abs_seqno 也能正常记录正确的长度。

image

2.1 seqno 和 abs_seqno 转换

由于 seqno 不是真正的 字节流 起始位置,因此接受报文时,需要对其转换成 abs_seqno,才可以方便 TCPReceiver 中计算窗口大小。
转换的接口如下:

//! Transform a 64-bit absolute sequence number (zero-indexed) into a 32-bit relative sequence number
//! \param n the absolute sequence number
//! \param isn the initial sequence number
//! \returns the relative sequence number
WrappingInt32 wrap(uint64_t n, WrappingInt32 isn);

//! Transform a 32-bit relative sequence number into a 64-bit absolute sequence number (zero-indexed)
//! \param n The relative sequence number
//! \param isn The initial sequence number
//! \param checkpoint A recent absolute sequence number
//! \returns the absolute sequence number that wraps to `n` and is closest to `checkpoint`
//! \note Each of the two streams of the TCP connection has its own ISN. One stream
//! runs from the local TCPSender to the remote TCPReceiver and has one ISN,
//! and the other stream runs from the remote TCPSender to the local TCPReceiver and
//! has a different ISN.
uint64_t unwrap(WrappingInt32 n, WrappingInt32 isn, uint64_t checkpoint);

abs_seqno 转换成 seqno 比较简单,只需要把 abs_seqno 加上 isn 初始偏移量,然后取 abs_seqno 的低 32 位值即可。
seqno 转换成 abs_seqno 则稍微麻烦些,因为 seqno=17 可以表示多种 abs_seqno,如 2^32 + 17, or 2^33 + 17, 2^34 + 17 等等,这里引入一个 checkpoint 的概念,在 TCPReceiver 中 checkpoint 是当前写入的总字节数。
假定要将 n 从 seqno 转换成 abs_seqno,这里先将 chekpointabs_seqno 转换成 seqno,然后计算 n(seqno 版本) 和 checkpoint(seqno 版本) 的偏移量,最后加到 checkpointabs_seqno 版本)上面即可得出 n(abs_seqno 版本),参考下图:

image

实现如下:

WrappingInt32 wrap(uint64_t n, WrappingInt32 isn) {
	uint64_t mask = (1ul << 32) - 1;
	return WrappingInt32{static_cast<uint32_t>((n + isn.raw_value()) & mask)};
}

uint64_t unwrap(WrappingInt32 n, WrappingInt32 isn, uint64_t checkpoint) {
	int32_t offset = n - wrap(checkpoint, isn);
	int64_t result = checkpoint + offset;
	result = result < 0 ? result + (1ul << 32) : result;
	return result;
}

2.2 接收报文

处理报文比较简单,主要关注前面说的 SYN 和 FIN 报文即可,及时更新最新的 abs_seqno

void TCPReceiver::segment_received(const TCPSegment &seg) {
	// 1. process syn flag
	if (seg.header().syn) {
		if (!_received_syn) {
			_received_syn = true;
			_isn = seg.header().seqno;
		}
	}
	
	// 2. process fin flag
	bool eof = false;
	if (seg.header().fin) {
		eof = true;
	}
	
	// 3. push payload
	if (ackno().has_value() && !_reassembler.stream_out().input_ended()) {
		// relative seqno to stream index
		uint64_t stream_index = unwrap(seg.header().seqno, _isn, _checkpoint);
		// stream index should in the window
		uint64_t abs_ackno = unwrap(ackno().value(), _isn, _checkpoint);
		if (stream_index + seg.length_in_sequence_space() <= abs_ackno || stream_index >= abs_ackno + window_size()) {
			return;
		}
		
		if (!seg.header().syn) {
			stream_index -= 1; // ignore syn flag;
		}
		_reassembler.push_substring(seg.payload().copy(), stream_index, eof);
		_checkpoint = _reassembler.stream_out().bytes_written();
	}
}

2.3 窗口大小和 ackno

窗口大小用于通知对端当前可以接收的字节流大小,ackno 用于通知对端当前接收的字节流进度。两者实现都比较简单,如下:

optional<WrappingInt32> TCPReceiver::ackno() const {
	// next_write + 1 ,because syn flag will not push in stream
	size_t next_write = _reassembler.stream_out().bytes_written() + 1;
	next_write = _reassembler.stream_out().input_ended() ? next_write + 1 : next_write;
	return !_received_syn ? optional<WrappingInt32>() : wrap(next_write, _isn);
}

size_t TCPReceiver::window_size() const {
	return _reassembler.stream_out().remaining_capacity();
}

3. 测试

image

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章