前言

在 Lab2 中我们实现了 TCP Receiver，负责在收到报文段之后将数据写入重组器中，并回复给发送方确认应答号。在 Lab3 中，我们将实现 TCP 连接的另一个端点——发送方，负责读取 ByteStream（由发送方上层应用程序创建并写入数据），并将字节流转换为报文段发送给接收方。

代码实现

TCP Sender 将负责：

跟踪 TCP Receiver 的窗口，处理确认应答号和窗口大小
通过从 ByteStream 中读取内容来填充发送窗口，创建新的报文段（可以包含 SYN 和 FIN 标志），并发送它们
跟踪哪些分段已发送但尚未被接收方确认——我们称之为未完成报文段（outstanding segment）
如果发送报文段后经过足够长的时间仍未得到确认，则重新发送未完成的报文段

由于涉及到超时处理，我们可以先实现一个简单的定时器 Timer，类声明如下所示：

复制class Timer {private:uint32_t _rto;          // 超时时间uint32_t _remain_time;	// 剩余时间bool _is_running;		// 是否在运行public:Timer(uint32_t rto);// 启动计时器void start();// 停止计时器void stop();// 是否超时bool is_time_out();// 设置过去了多少时间void elapse(size_t eplased);// 设置超时时间void set_time_out(uint32_t duration);
};

根据实验指导书的要求，定时器不能通过调用系统时间函数来知道过了多长时间，而是由外部传入的时长参数告知，这一点可以从 send_retx.cc 测试用例得到印证：

复制TCPSenderTestHarness test{"Retx SYN twice at the right times, then ack", cfg};
test.execute(ExpectSegment{}.with_no_flags().with_syn(true).with_payload_size(0).with_seqno(isn));
test.execute(ExpectNoSegment{});
test.execute(ExpectState{TCPSenderStateSummary::SYN_SENT});// 外部指定逝去的时间
test.execute(Tick{retx_timeout - 1u});

所以这个定时器的实现就很简单，外部通过调用 Timer::elapse() 告知定时器多久过去了，定时器只要更新一下剩余时长就好了：

复制
Timer::Timer(uint32_t rto) : _rto(rto), _remain_time(rto), _is_running(false) {}void Timer::start() {_is_running = true;_remain_time = _rto;
}void Timer::stop() { _is_running = false; }bool Timer::is_time_out() { return _remain_time == 0; }void Timer::elapse(size_t elapsed) {if (elapsed > _remain_time) {_remain_time = 0;} else {_remain_time -= elapsed;}
}void Timer::set_time_out(uint32_t duration) {_rto = duration;_remain_time = duration;
}

完成定时器之后，来看看 TCPSender 类有哪些成员：

复制class TCPSender {private://! our initial sequence number, the number for our SYN.WrappingInt32 _isn;//! outbound queue of segments that the TCPSender wants sentstd::queue<TCPSegment> _segments_out{};// 未被确认的报文段std::queue<std::pair<TCPSegment, uint64_t>> _outstand_segments{};//! retransmission timer for the connectionunsigned int _initial_retransmission_timeout;//! outgoing stream of bytes that have not yet been sentByteStream _stream;//! the (absolute) sequence number for the next byte to be sentuint64_t _next_seqno{0};// ackno checkpointuint64_t _ack_seq{0};// 连续重传次数uint32_t _consecutive_retxs{0};// 未被确认的序号长度uint64_t _outstand_bytes{0};// 接收方窗口长度uint16_t _window_size{1};// 是否同步bool _is_syned{false};// 是否结束bool _is_fin{false};// 计时器Timer _timer;public://! Initialize a TCPSenderTCPSender(const size_t capacity = TCPConfig::DEFAULT_CAPACITY,const uint16_t retx_timeout = TCPConfig::TIMEOUT_DFLT,const std::optional<WrappingInt32> fixed_isn = {});//! \name "Input" interface for the writerByteStream &stream_in() { return _stream; }const ByteStream &stream_in() const { return _stream; }//! \brief A new acknowledgment was receivedbool ack_received(const WrappingInt32 ackno, const uint16_t window_size);//! \brief Generate an empty-payload segment (useful for creating empty ACK segments)void send_empty_segment();// 发送报文段void send_segment(std::string &&data, bool syn = false, bool fin = false);//! \brief create and send segments to fill as much of the window as possiblevoid fill_window();//! \brief Notifies the TCPSender of the passage of timevoid tick(const size_t ms_since_last_tick);//! \brief How many sequence numbers are occupied by segments sent but not yet acknowledged?size_t bytes_in_flight() const;//! \brief Number of consecutive retransmissions that have occurred in a rowunsigned int consecutive_retransmissions() const;//! \brief TCPSegments that the TCPSender has enqueued for transmission.std::queue<TCPSegment> &segments_out() { return _segments_out; }//! \brief absolute seqno for the next byte to be sentuint64_t next_seqno_absolute() const { return _next_seqno; }//! \brief relative seqno for the next byte to be sentWrappingInt32 next_seqno() const { return wrap(_next_seqno, _isn); }
};

可以看到，我们 TCPSender 有以下主要成员：

queue<TCPSegment> _segments_out：待发送的报文段队列，外部程序会从这个队列里面取出报文段并发送出去
queue<pair<TCPSegment, uint64_t>> _outstand_segments：存放未被确认的报文段和它对应的绝对序列号的队列
uint64_t _ack_seq：上一次收到的绝对确认应答号
uint32_t _consecutive_retxs：最早发送的但是未被确认的报文段的重传次数，用于更新超时时间
uint64_t _outstand_bytes：所有未被确认的报文段所占序列号空间长度，SYN 和 FIN 也要占用一个序号
uint16_t _window_size：接收方窗口大小，初始值为 1，由于没有实现加性递增乘性递减（AIMD）拥塞控制机制，所以不用维护发送方的拥塞窗口大小，直接维护接收方窗口大小
bool _is_syned：是否成功同步
bool _is_fin：是否关闭连接
Timer _timer：定时器

先来实现一些比较简单的函数：

复制//! \param[in] capacity the capacity of the outgoing byte stream
//! \param[in] retx_timeout the initial amount of time to wait before retransmitting the oldest outstanding segment
//! \param[in] fixed_isn the Initial Sequence Number to use, if set (otherwise uses a random ISN)
TCPSender::TCPSender(const size_t capacity, const uint16_t retx_timeout, const std::optional<WrappingInt32> fixed_isn): _isn(fixed_isn.value_or(WrappingInt32{random_device()()})), _initial_retransmission_timeout{retx_timeout}, _stream(capacity), _timer(retx_timeout) {}uint64_t TCPSender::bytes_in_flight() const { return _outstand_bytes; }unsigned int TCPSender::consecutive_retransmissions() const { return _consecutive_retxs; }

丢包处理

根据实验指导书中的描述：

Periodically, the owner of the TCPSender will call the TCPSender’s tick method, indicating the passage of time.

外部会定期调用 TCPSender::tick() 函数来告知它过了多长时间，TCPSender 要根据传入的时间判断最早发送的包是不是超时未被确认，如果是（定时器溢出），就说明这个包丢掉了，需要重传。

同时超时也意味着网络可能比较拥挤，沿途的某个路由器内部队列满了，再次发送也有可能丢失，不仅浪费了带宽，还会进一步加剧网络的拥堵。不如耐心点，把超时时间翻倍，如果下一次成功收到确认应答号就还原成初始超时时间。这个超时时间估计机制和 CS144 第 61 集和《计算机网络：自顶而下方法》第 158 页所讲授的指数移动平均机制不太一样：

值得注意的是，实验指导书中只将超时作为重传的条件，而没有考虑三次冗余 ACK 触发快速重传情况。因此 Timer::tick() 的代码实现如下：

复制//! \param[in] ms_since_last_tick the number of milliseconds since the last call to this method
void TCPSender::tick(const size_t ms_since_last_tick) {_timer.elapse(ms_since_last_tick);if (!_timer.is_time_out())return;if (_outstand_segments.empty()) {return _timer.set_time_out(_initial_retransmission_timeout);}// 超时需要重发第一个报文段，同时将超时时间翻倍_segments_out.push(_outstand_segments.front().first);_consecutive_retxs += 1;_timer.set_time_out(_initial_retransmission_timeout * (1 << _consecutive_retxs));_timer.start();
}

这里只重传了一个报文段，而不是像回退 N 步（GBN）协议那样重传整个窗口内的报文段，这是因为 Lab2 中实现的接收方会缓存所有乱序到达的报文段，而 GBN 是直接将其丢弃掉了。如果我们重传的包被成功接收了，并且使接收方成功重组了整个发送窗口内的数据，就不需要重传后续的报文段了。如果没有成功重组，仍有部分数据缺失，接收方会回复一个它想要的报文段的序号，到时候重传这个报文段就行了。

发送报文段

发送方需要根据接收方的确认应答号和窗口大小决定需要发送哪些数据，假设当前数据接收情况如下图所示，绿色和蓝色的部分是已成功接收并重组的数据，红色部分是成功接收但是因为前方有报文没达到而未重组的数据：

假设最后一个红色矩形就是上次发送的最后一个报文段，那么 TCPSender 的各个成员的值就是图中所标注的那样，这时候调用 TCPSender::fill_window() 发送的应该是 _next_seq ~ _ack_seq + _window_size 之间的数据。不过在发送数据之前需要完成三次握手，所以需要先判断 _is_syned 是否为 true，如果为 false 就需要发送 SYN 包与接收端进行连接。所有数据都发送完成之后需要发送一个 FIN 报文段（可以携带最后一批数据或者不懈携带任何数据）说明 TCPSender 已经没有新数据要发送了，可以断开连接了。

复制
void TCPSender::fill_window() {if (!_is_syned) {// 等待 SYN 超时if (!_outstand_segments.empty())return;// 发送一个 SYN 包send_segment("", true);} else {size_t remain_size = max(_window_size, static_cast<uint16_t>(1)) + _ack_seq - _next_seqno;// 当缓冲区中有待发送数据时就发送数据报文段while (remain_size > 0 && !_stream.buffer_empty()) {auto ws = min(min(remain_size, TCPConfig::MAX_PAYLOAD_SIZE), _stream.buffer_size());remain_size -= ws;string &&data = _stream.peek_output(ws);_stream.pop_output(ws);// 置位 FIN_is_fin |= (_stream.eof() && !_is_fin && remain_size > 0);send_segment(std::move(data), false, _is_fin);}// 缓冲区输入结束时发送 FIN（缓冲区为空时不会进入循环体，需要再次发送）if (_stream.eof() && !_is_fin && remain_size > 0) {_is_fin = true;send_segment("", false, true);}}
}void TCPSender::send_segment(string &&data, bool syn, bool fin) {// 创建报文段TCPSegment segment;segment.header().syn = syn;segment.header().fin = fin;segment.header().seqno = next_seqno();segment.payload() = std::move(data);// 将报文段放到发送队列中_segments_out.push(segment);_outstand_segments.push({segment, _next_seqno});// 更新序号auto len = segment.length_in_sequence_space();_outstand_bytes += len;_next_seqno += len;
}void TCPSender::send_empty_segment() {TCPSegment seg;seg.header().seqno = next_seqno();_segments_out.push(seg);
}

这里有一个地方值得思考的问题是：把同一个报文段保存到两个队列中不会导致数据的拷贝吗？实际上不会，因为 TCPSegment::_payload 的数据类型是 Buffer，它的声明如下所示：

复制//! \brief A reference-counted read-only string that can discard bytes from the front
class Buffer {private:std::shared_ptr<std::string> _storage{};size_t _starting_offset{};public:Buffer() = default;//! \brief Construct by taking ownership of a stringBuffer(std::string &&str) noexcept : _storage(std::make_shared<std::string>(std::move(str))) {}//! \name Expose contents as a std::string_viewstd::string_view str() const {if (not _storage) {return {};}return {_storage->data() + _starting_offset, _storage->size() - _starting_offset};}operator std::string_view() const { return str(); }//! \brief Get character at location `n`uint8_t at(const size_t n) const { return str().at(n); }//! \brief Size of the stringsize_t size() const { return str().size(); }//! \brief Make a copy to a new std::stringstd::string copy() const { return std::string(str()); }//! \brief Discard the first `n` bytes of the string (does not require a copy or move)//! \note Doesn't free any memory until the whole string has been discarded in all copies of the Buffer.void remove_prefix(const size_t n);
};

可以看到 Buffer 内部使用智能指针 shared_ptr<string> _storage 共享了同一份字符串，当 queue.push(buffer) 的时候调用了 Buffer(const Buffer &) 拷贝构造函数，只对 _storage 指针进行赋值而不涉及字符串复制操作。同时 Buffer(string &&str) 构造函数接受右值，可以直接把传入的字符串偷取过来，无需拷贝，效率是很高的。

确认应答号处理

当发送方收到确认应答号时，需要判断这个应答号是否合法，如果收到的确认引导号落在发送窗口以外，就不去管它。否则需要重置超时时间为初始值，并移除 _outstand_segments 队列中绝对序列号小于绝对确认应答号的报文段。如果不存在未确认的报文段了就关闭定时器，否则得再次启动定时器，为重传下一个报文段做准备。

复制bool TCPSender::ack_received(const WrappingInt32 ackno, const uint16_t window_size) {auto ack_seq = unwrap(ackno, _isn, _ack_seq);// absolute ackno 不能落在窗口外if (ack_seq > _next_seqno)return false;_window_size = window_size;// 忽略已处理过的确认应答号if (ack_seq <= _ack_seq)return true;_ack_seq = ack_seq;_is_syned = true;// 重置超时时间为初始值_timer.set_time_out(_initial_retransmission_timeout);_consecutive_retxs = 0;// 移除已被确认的报文段while (!_outstand_segments.empty()) {auto &[segment, seqno] = _outstand_segments.front();if (seqno >= ack_seq)break;_outstand_bytes -= segment.length_in_sequence_space();_outstand_segments.pop();}// 再次填满发送窗口fill_window();// 如果还有没被确认的报文段就重启计时器if (!_outstand_segments.empty())_timer.start();else_timer.stop();return true;
}