diff --git a/src/applications/onoff/onoff-application.cc b/src/applications/onoff/onoff-application.cc index 0940a5518..8424fda65 100644 --- a/src/applications/onoff/onoff-application.cc +++ b/src/applications/onoff/onoff-application.cc @@ -111,6 +111,12 @@ OnOffApplication::SetMaxBytes(uint32_t maxBytes) m_maxBytes = maxBytes; } +Ptr +OnOffApplication::GetSocket (void) const +{ + NS_LOG_FUNCTION (this); + return m_socket; +} void OnOffApplication::DoDispose (void) diff --git a/src/applications/onoff/onoff-application.h b/src/applications/onoff/onoff-application.h index c7e43ec33..6858b0de2 100644 --- a/src/applications/onoff/onoff-application.h +++ b/src/applications/onoff/onoff-application.h @@ -102,6 +102,11 @@ public: */ void SetMaxBytes (uint32_t maxBytes); + /** + * \return pointer to associated socket + */ + Ptr GetSocket (void) const; + protected: virtual void DoDispose (void); private: diff --git a/src/applications/packet-sink/packet-sink.cc b/src/applications/packet-sink/packet-sink.cc index 1ddf50c33..c0b1bfb70 100644 --- a/src/applications/packet-sink/packet-sink.cc +++ b/src/applications/packet-sink/packet-sink.cc @@ -75,6 +75,20 @@ uint32_t PacketSink::GetTotalRx() const return m_totalRx; } +Ptr +PacketSink::GetListeningSocket (void) const +{ + NS_LOG_FUNCTION (this); + return m_socket; +} + +std::list > +PacketSink::GetAcceptedSockets (void) const +{ + NS_LOG_FUNCTION (this); + return m_socketList; +} + void PacketSink::DoDispose (void) { NS_LOG_FUNCTION (this); diff --git a/src/applications/packet-sink/packet-sink.h b/src/applications/packet-sink/packet-sink.h index 26338692f..f5a890579 100644 --- a/src/applications/packet-sink/packet-sink.h +++ b/src/applications/packet-sink/packet-sink.h @@ -77,7 +77,17 @@ public: * \return the total bytes received in this sink app */ uint32_t GetTotalRx () const; - + + /** + * \return pointer to listening socket + */ + Ptr GetListeningSocket (void) const; + + /** + * \return list of pointers to accepted sockets + */ + std::list > GetAcceptedSockets (void) const; + protected: virtual void DoDispose (void); private: