A Discrete-Event Network Simulator
API
ns3tcp-socket-writer.cc
Go to the documentation of this file.
1 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2 /*
3  * Copyright (c) 2010 University of Washington
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 #include "ns3tcp-socket-writer.h"
20 #include "ns3/tcp-socket-factory.h"
21 #include "ns3/packet.h"
22 
23 namespace ns3
24 {
25 
26 SocketWriter::SocketWriter () : m_node (0), m_socket (0), m_isSetup (false), m_isConnected (false)
27 {
28 }
29 
31 {
32  m_socket = 0;
33  m_node = 0;
34 }
35 
36 /* static */
37 TypeId
39 {
40  static TypeId tid = TypeId ("ns3::SocketWriter")
42  .SetGroupName ("Stats")
43  .AddConstructor<SocketWriter> ()
44  ;
45  return tid;
46 }
47 
48 void
50 {
52  m_socket->Bind ();
53 }
54 
55 void
57 {
58 }
59 
60 void
62 {
63  m_peer = peer;
64  m_node = node;
65  m_isSetup = true;
66 }
67 
68 void
70 {
71  if (!m_isSetup)
72  {
73  NS_FATAL_ERROR ("Forgot to call Setup() first");
74  }
76  m_isConnected = true;
77 }
78 
79 void
80 SocketWriter::Write (uint32_t numBytes)
81 {
82  if (!m_isConnected)
83  {
84  Connect ();
85  }
86  Ptr<Packet> packet = Create<Packet> (numBytes);
87  m_socket->Send (packet);
88 }
89 
90 void
92 {
93  m_socket->Close ();
94 }
95 }
virtual void StartApplication(void)
Application specific startup code.
virtual void StopApplication(void)
Application specific shutdown code.
void Write(uint32_t numBytes)
#define NS_FATAL_ERROR(msg)
Report a fatal error with a message and terminate.
Definition: fatal-error.h:162
a polymophic address class
Definition: address.h:90
void Setup(Ptr< Node > node, Address peer)
The base class for all ns3 applications.
Definition: application.h:60
static Ptr< Socket > CreateSocket(Ptr< Node > node, TypeId tid)
This method wraps the creation of sockets that is performed on a given node by a SocketFactory specif...
Definition: socket.cc:71
static TypeId GetTypeId(void)
Get the type ID.
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.
Every class exported by the ns3 library is enclosed in the ns3 namespace.
static TypeId GetTypeId(void)
Register this type.
virtual int Send(Ptr< Packet > p, uint32_t flags)=0
Send data (or dummy data) to the remote host.
virtual int Close(void)=0
Close a socket.
a unique identifier for an interface.
Definition: type-id.h:58
TypeId SetParent(TypeId tid)
Set the parent TypeId.
Definition: type-id.cc:915