69 lines
2.4 KiB
C++
69 lines
2.4 KiB
C++
#include "fsm_retx.hh"
|
|
|
|
#include "tcp_config.hh"
|
|
#include "tcp_expectation.hh"
|
|
#include "util.hh"
|
|
|
|
#include <algorithm>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
|
|
using namespace std;
|
|
using State = TCPTestHarness::State;
|
|
|
|
int main() {
|
|
try {
|
|
TCPConfig cfg{};
|
|
cfg.recv_capacity = 65000;
|
|
auto rd = get_random_generator();
|
|
|
|
// NOTE: the timeouts in this test are carefully adjusted to work whether the tcp_state_machine sends
|
|
// immediately upon write() or only after the next tick(). If you edit this test, make sure that
|
|
// it passes both ways (i.e., with and without calling _try_send() in TCPConnection::write())
|
|
|
|
// NOTE 2: ACK -- I think I was successful, although given unrelated
|
|
// refactor to template code it will be more challenging now
|
|
// to wait until tick() to send an outgoing segment.
|
|
|
|
// single segment re-transmit
|
|
{
|
|
const WrappingInt32 tx_ackno(rd());
|
|
TCPTestHarness test_1 = TCPTestHarness::in_established(cfg, tx_ackno - 1, tx_ackno - 1);
|
|
|
|
string data = "asdf";
|
|
test_1.execute(Write{data});
|
|
test_1.execute(Tick(1));
|
|
|
|
check_segment(test_1, data, false, __LINE__);
|
|
|
|
test_1.execute(Tick(cfg.rt_timeout - 2));
|
|
test_1.execute(ExpectNoSegment{}, "test 1 failed: re-tx too fast");
|
|
|
|
test_1.execute(Tick(2));
|
|
check_segment(test_1, data, false, __LINE__);
|
|
|
|
test_1.execute(Tick(10 * cfg.rt_timeout + 100));
|
|
check_segment(test_1, data, false, __LINE__);
|
|
|
|
for (unsigned i = 2; i < TCPConfig::MAX_RETX_ATTEMPTS; ++i) {
|
|
test_1.execute(Tick((cfg.rt_timeout << i) - i)); // exponentially increasing delay length
|
|
test_1.execute(ExpectNoSegment{}, "test 1 failed: re-tx too fast after timeout");
|
|
test_1.execute(Tick(i));
|
|
check_segment(test_1, data, false, __LINE__);
|
|
}
|
|
|
|
test_1.execute(ExpectState{State::ESTABLISHED});
|
|
|
|
test_1.execute(Tick(1 + (cfg.rt_timeout << TCPConfig::MAX_RETX_ATTEMPTS)));
|
|
test_1.execute(ExpectState{State::RESET});
|
|
test_1.execute(ExpectOneSegment{}.with_rst(true).with_ack(false).with_seqno(tx_ackno),
|
|
"test 1 failed: RST on re-tx failure was malformed");
|
|
}
|
|
} catch (const exception &e) {
|
|
cerr << e.what() << endl;
|
|
return 1;
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|