A Discrete-Event Network Simulator
API
tcp-datasentcb-test.cc
Go to the documentation of this file.
1 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2 /*
3  * Copyright (c) 2015 Natale Patriciello <natale.patriciello@gmail.com>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation;
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17  *
18  */
19 
20 #include "tcp-general-test.h"
21 #include "ns3/node.h"
22 #include "ns3/log.h"
23 
24 using namespace ns3;
25 
26 NS_LOG_COMPONENT_DEFINE ("TcpDatSentCbTest");
27 
36 {
37 public:
42  static TypeId GetTypeId (void);
43 
45  {
46  }
47 
48 protected:
49  virtual Ptr<TcpSocketBase> Fork ();
50  virtual void ReceivedData (Ptr<Packet> packet, const TcpHeader& tcpHeader);
51 };
52 
54 
55 TypeId
57 {
58  static TypeId tid = TypeId ("ns3::TcpSocketHalfAck")
60  .SetGroupName ("Internet")
61  .AddConstructor<TcpSocketHalfAck> ()
62  ;
63  return tid;
64 }
65 
68 {
69  return CopyObject<TcpSocketHalfAck> (this);
70 }
71 
72 void
74 {
75  NS_LOG_FUNCTION (this << packet << tcpHeader);
76  static uint32_t times = 1;
77 
78  Ptr<Packet> halved = packet->Copy ();
79 
80  if (times % 2 == 0)
81  halved->RemoveAtEnd (packet->GetSize () / 2);
82 
83  times++;
84 
85  TcpSocketMsgBase::ReceivedData (halved, tcpHeader);
86 }
87 
88 
102 {
103 public:
104 
111  TcpDataSentCbTestCase (const std::string &desc, uint32_t size, uint32_t packets) :
112  TcpGeneralTest (desc),
113  m_pktSize (size),
114  m_pktCount (packets),
115  m_notifiedData (0)
116  { }
117 
118 protected:
119  virtual Ptr<TcpSocketMsgBase> CreateReceiverSocket (Ptr<Node> node);
120 
121  virtual void DataSent (uint32_t size, SocketWho who);
122  virtual void ConfigureEnvironment ();
123  virtual void FinalChecks ();
124 
125 private:
126  uint32_t m_pktSize;
127  uint32_t m_pktCount;
128  uint32_t m_notifiedData;
129 };
130 
131 void
133 {
135  SetAppPktCount (m_pktCount);
136  SetAppPktSize (m_pktSize);
137 }
138 
139 void
141 {
142  NS_LOG_FUNCTION (this << who << size);
143 
144  m_notifiedData += size;
145 }
146 
147 void
149 {
150  NS_TEST_ASSERT_MSG_EQ (m_notifiedData, GetPktSize () * GetPktCount (),
151  "Notified more data than application sent");
152 }
153 
156 {
157  NS_LOG_FUNCTION (this);
158 
159  return CreateSocket (node, TcpSocketHalfAck::GetTypeId (), m_congControlTypeId);
160 }
161 
169 {
170 public:
172  : TestSuite ("tcp-datasentcb", UNIT)
173  {
174  AddTestCase (new TcpDataSentCbTestCase ("Check the data sent callback", 500, 10), TestCase::QUICK);
175  AddTestCase (new TcpDataSentCbTestCase ("Check the data sent callback", 100, 100), TestCase::QUICK);
176  AddTestCase (new TcpDataSentCbTestCase ("Check the data sent callback", 1000, 50), TestCase::QUICK);
177  AddTestCase (new TcpDataSentCbTestCase ("Check the data sent callback", 855, 18), TestCase::QUICK);
178  AddTestCase (new TcpDataSentCbTestCase ("Check the data sent callback", 1243, 59), TestCase::QUICK);
179  }
180 
181 };
182 
Smart pointer class similar to boost::intrusive_ptr.
Definition: ptr.h:73
#define NS_LOG_FUNCTION(parameters)
If log level LOG_FUNCTION is enabled, this macro will output all input parameters separated by "...
Class for inserting callbacks special points of the flow of TCP sockets.
#define NS_OBJECT_ENSURE_REGISTERED(type)
Register an Object subclass with the TypeId system.
Definition: object-base.h:45
TcpDataSentCbTestCase(const std::string &desc, uint32_t size, uint32_t packets)
Constructor.
Fast test.
Definition: test.h:1160
A suite of tests to run.
Definition: test.h:1342
#define NS_LOG_COMPONENT_DEFINE(name)
Define a Log component with a specific name.
Definition: log.h:201
uint32_t GetSize(void) const
Returns the the size in bytes of the packet (including the zero-filled initial payload).
Definition: packet.h:796
uint32_t m_notifiedData
Amount of data notified.
static TypeId GetTypeId(void)
Get the type ID.
virtual void FinalChecks()
Performs the (eventual) final checks through test asserts.
uint32_t m_pktSize
Packet size.
uint32_t m_pktCount
Number of packets sent.
virtual void DataSent(uint32_t size, SocketWho who)
Notifying application for sent data.
Data Sent callback test.
Socket that the 50% of the times saves the entire packet in the buffer, while in the other 50% saves ...
SocketWho
Used as parameter of methods, specifies on what node the caller is interested (e.g.
#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:168
TestSuite: Data Sent callback.
Ptr< Packet > Copy(void) const
performs a COW copy of the packet.
Definition: packet.cc:121
Every class exported by the ns3 library is enclosed in the ns3 namespace.
Header for the Transmission Control Protocol.
Definition: tcp-header.h:44
virtual Ptr< TcpSocketMsgBase > CreateReceiverSocket(Ptr< Node > node)
Create and install the socket to install on the receiver.
void RemoveAtEnd(uint32_t size)
Remove size bytes from the end of the current packet.
Definition: packet.cc:332
virtual void ConfigureEnvironment(void)
Change the configuration of the evironment.
virtual Ptr< TcpSocketBase > Fork()
Call CopyObject<> to clone me.
General infrastructure for TCP testing.
static TcpDataSentCbTestSuite g_tcpDataSentCbTestSuite
Static variable for test initialization.
virtual void ReceivedData(Ptr< Packet > packet, const TcpHeader &tcpHeader)
Recv of a data, put into buffer, call L7 to get it if necessary.
virtual void ConfigureEnvironment()
Change the configuration of the evironment.
a unique identifier for an interface.
Definition: type-id.h:58
virtual void ReceivedData(Ptr< Packet > packet, const TcpHeader &tcpHeader)
Recv of a data, put into buffer, call L7 to get it if necessary.
TypeId SetParent(TypeId tid)
Set the parent TypeId.
Definition: type-id.cc:914