A Discrete-Event Network Simulator
API
tcp-datasentcb-test.cc
Go to the documentation of this file.
1/*
2 * Copyright (c) 2015 Natale Patriciello <natale.patriciello@gmail.com>
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 "tcp-general-test.h"
20
21#include "ns3/log.h"
22#include "ns3/node.h"
23#include "ns3/tcp-header.h"
24
25using namespace ns3;
26
27NS_LOG_COMPONENT_DEFINE("TcpDatSentCbTest");
28
37{
38 public:
43 static TypeId GetTypeId();
44
47 {
48 }
49
50 protected:
51 Ptr<TcpSocketBase> Fork() override;
52 void ReceivedData(Ptr<Packet> packet, const TcpHeader& tcpHeader) override;
53};
54
56
59{
60 static TypeId tid = TypeId("ns3::TcpSocketHalfAck")
62 .SetGroupName("Internet")
63 .AddConstructor<TcpSocketHalfAck>();
64 return tid;
65}
66
69{
70 return CopyObject<TcpSocketHalfAck>(this);
71}
72
73void
75{
76 NS_LOG_FUNCTION(this << packet << tcpHeader);
77 static uint32_t times = 1;
78
79 Ptr<Packet> halved = packet->Copy();
80
81 if (times % 2 == 0)
82 {
83 halved->RemoveAtEnd(packet->GetSize() / 2);
84 }
85
86 times++;
87
88 TcpSocketMsgBase::ReceivedData(halved, tcpHeader);
89}
90
104{
105 public:
112 TcpDataSentCbTestCase(const std::string& desc, uint32_t size, uint32_t packets)
113 : TcpGeneralTest(desc),
114 m_pktSize(size),
115 m_pktCount(packets),
117 {
118 }
119
120 protected:
122
123 void DataSent(uint32_t size, SocketWho who) override;
124 void ConfigureEnvironment() override;
125 void FinalChecks() override;
126
127 private:
131};
132
133void
135{
136 TcpGeneralTest::ConfigureEnvironment();
139}
140
141void
143{
144 NS_LOG_FUNCTION(this << who << size);
145
146 m_notifiedData += size;
147}
148
149void
151{
154 "Notified more data than application sent");
155}
156
159{
160 NS_LOG_FUNCTION(this);
161
163}
164
172{
173 public:
175 : TestSuite("tcp-datasentcb", UNIT)
176 {
177 AddTestCase(new TcpDataSentCbTestCase("Check the data sent callback", 500, 10),
178 TestCase::QUICK);
179 AddTestCase(new TcpDataSentCbTestCase("Check the data sent callback", 100, 100),
180 TestCase::QUICK);
181 AddTestCase(new TcpDataSentCbTestCase("Check the data sent callback", 1000, 50),
182 TestCase::QUICK);
183 AddTestCase(new TcpDataSentCbTestCase("Check the data sent callback", 855, 18),
184 TestCase::QUICK);
185 AddTestCase(new TcpDataSentCbTestCase("Check the data sent callback", 1243, 59),
186 TestCase::QUICK);
187 }
188};
189
Data Sent callback test.
Ptr< TcpSocketMsgBase > CreateReceiverSocket(Ptr< Node > node) override
Create and install the socket to install on the receiver.
void DataSent(uint32_t size, SocketWho who) override
Notifying application for sent data.
void ConfigureEnvironment() override
Change the configuration of the environment.
uint32_t m_notifiedData
Amount of data notified.
TcpDataSentCbTestCase(const std::string &desc, uint32_t size, uint32_t packets)
Constructor.
uint32_t m_pktCount
Number of packets sent.
uint32_t m_pktSize
Packet size.
void FinalChecks() override
Performs the (eventual) final checks through test asserts.
TestSuite: Data Sent callback.
Socket that the 50% of the times saves the entire packet in the buffer, while in the other 50% saves ...
Ptr< TcpSocketBase > Fork() override
Call CopyObject<> to clone me.
static TypeId GetTypeId()
Get the type ID.
void ReceivedData(Ptr< Packet > packet, const TcpHeader &tcpHeader) override
Recv of a data, put into buffer, call L7 to get it if necessary.
uint32_t GetSize() const
Returns the the size in bytes of the packet (including the zero-filled initial payload).
Definition: packet.h:863
void RemoveAtEnd(uint32_t size)
Remove size bytes from the end of the current packet.
Definition: packet.cc:376
Ptr< Packet > Copy() const
performs a COW copy of the packet.
Definition: packet.cc:131
Smart pointer class similar to boost::intrusive_ptr.
Definition: ptr.h:78
General infrastructure for TCP testing.
uint32_t GetPktCount() const
Get the number of application packets.
void SetAppPktCount(uint32_t pktCount)
Set app packet count.
SocketWho
Used as parameter of methods, specifies on what node the caller is interested (e.g.
void SetAppPktSize(uint32_t pktSize)
Set app packet size.
virtual Ptr< TcpSocketMsgBase > CreateSocket(Ptr< Node > node, TypeId socketType, TypeId congControl)
Create a socket.
TypeId m_congControlTypeId
Congestion control.
uint32_t GetPktSize() const
Get the application packet size.
Header for the Transmission Control Protocol.
Definition: tcp-header.h:46
Class for inserting callbacks special points of the flow of TCP sockets.
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:60
TypeId SetParent(TypeId tid)
Set the parent TypeId.
Definition: type-id.cc:935
#define NS_LOG_COMPONENT_DEFINE(name)
Define a Log component with a specific name.
Definition: log.h:202
#define NS_LOG_FUNCTION(parameters)
If log level LOG_FUNCTION is enabled, this macro will output all input parameters separated by ",...
#define NS_OBJECT_ENSURE_REGISTERED(type)
Register an Object subclass with the TypeId system.
Definition: object-base.h:45
#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.
static TcpDataSentCbTestSuite g_tcpDataSentCbTestSuite
Static variable for test initialization.