A Discrete-Event Network Simulator
API
tcp-syn-connection-failed-test.cc
Go to the documentation of this file.
1/*
2 * Copyright (c) 2019 Alexander Krotov <krotov@iitp.ru>
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License version 2 as
6 * published by the Free Software Foundation;
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public License
14 * along with this program; if not, write to the Free Software
15 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 *
17 */
18
19#include "ns3/core-module.h"
20#include "ns3/internet-module.h"
21#include "ns3/network-module.h"
22#include "ns3/test.h"
23
24#include <iostream>
25
26using namespace ns3;
27
35{
36 public:
42 TcpSynConnectionFailedTest(std::string desc, bool useEcn);
43
49 void DoRun() override;
50
51 private:
52 bool m_connectionFailed{false};
53 bool m_useEcn{false};
54};
55
57 : TestCase(desc),
58 m_useEcn(useEcn)
59{
60}
61
62void
64{
65 m_connectionFailed = true;
66}
67
68void
70{
71 Ptr<Node> node = CreateObject<Node>();
72
73 InternetStackHelper internet;
74 internet.Install(node);
75
76 TypeId tid = TcpSocketFactory::GetTypeId();
77
78 Ptr<Socket> socket = Socket::CreateSocket(node, tid);
79 if (m_useEcn)
80 {
81 Ptr<TcpSocketBase> tcpSocket = DynamicCast<TcpSocketBase>(socket);
82 tcpSocket->SetUseEcn(TcpSocketState::On);
83 }
84 socket->Bind();
85 socket->SetConnectCallback(
88 socket->Connect(InetSocketAddress(Ipv4Address::GetLoopback(), 9));
89
90 Simulator::Run();
91 Simulator::Destroy();
92
93 NS_TEST_ASSERT_MSG_EQ(m_connectionFailed, true, "Connection failed callback was not called");
94}
95
102{
103 public:
105 : TestSuite("tcp-syn-connection-failed-test", UNIT)
106 {
107 AddTestCase(new TcpSynConnectionFailedTest("TCP SYN connection failed test no ECN", false),
108 TestCase::QUICK);
109 AddTestCase(new TcpSynConnectionFailedTest("TCP SYN connection failed test with ECN", true),
110 TestCase::QUICK);
111 }
112};
113
Test that connection failed callback is called when SYN retransmission number is exceeded.
bool m_connectionFailed
Connection failure indicator.
void HandleConnectionFailed(Ptr< Socket > socket)
Handle a connection failure.
void DoRun() override
Implementation to actually run this TestCase.
bool m_useEcn
Use ECN (true or false)
TcpSynConnectionFailedTest(std::string desc, bool useEcn)
Constructor.
an Inet address class
aggregate IP/TCP/UDP functionality to existing Nodes.
void Install(std::string nodeName) const
Aggregate implementations of the ns3::Ipv4, ns3::Ipv6, ns3::Udp, and ns3::Tcp classes onto the provid...
void SetConnectCallback(Callback< void, Ptr< Socket > > connectionSucceeded, Callback< void, Ptr< Socket > > connectionFailed)
Specify callbacks to allow the caller to determine if the connection succeeds of fails.
Definition: socket.cc:85
virtual int Connect(const Address &address)=0
Initiate a connection to a remote host.
virtual int Bind(const Address &address)=0
Allocate a local endpoint for this socket.
encapsulates test code
Definition: test.h:1060
void AddTestCase(TestCase *testCase, TestDuration duration=QUICK)
Add an individual child TestCase to this test suite.
Definition: test.cc:305
A suite of tests to run.
Definition: test.h:1256
@ UNIT
This test suite implements a Unit Test.
Definition: test.h:1265
a unique identifier for an interface.
Definition: type-id.h:59
Callback< R, Args... > MakeNullCallback()
Definition: callback.h:750
#define NS_TEST_ASSERT_MSG_EQ(actual, limit, msg)
Test that an actual and expected (limit) value are equal and report and abort if not.
Definition: test.h:144
Every class exported by the ns3 library is enclosed in the ns3 namespace.
Callback< R, Args... > MakeCallback(R(T::*memPtr)(Args...), OBJ objPtr)
Build Callbacks for class method members which take varying numbers of arguments and potentially retu...
Definition: callback.h:707
static TcpSynConnectionFailedTestSuite g_TcpSynConnectionFailedTestSuite
Static variable for test initialization.