--- a/src/internet-node/ipv4-l4-demux.cc Thu May 10 07:51:59 2007 +0200
+++ b/src/internet-node/ipv4-l4-demux.cc Thu May 10 07:58:18 2007 +0200
@@ -32,12 +32,10 @@
const Iid Ipv4L4Demux::iid ("Ipv4L4Demux");
-Ipv4L4Demux::Ipv4L4Demux (Node *node)
+Ipv4L4Demux::Ipv4L4Demux (Ptr<Node> node)
: NsUnknown (Ipv4L4Demux::iid),
m_node (node)
-{
- m_node->Ref ();
-}
+{}
Ipv4L4Demux::~Ipv4L4Demux()
{}
@@ -51,11 +49,7 @@
(*i)->Unref ();
}
m_protocols.clear ();
- if (m_node != 0)
- {
- m_node->Unref ();
- m_node = 0;
- }
+ m_node = 0;
NsUnknown::DoDispose ();
}
--- a/src/internet-node/ipv4-l4-demux.h Thu May 10 07:51:59 2007 +0200
+++ b/src/internet-node/ipv4-l4-demux.h Thu May 10 07:58:18 2007 +0200
@@ -27,6 +27,7 @@
#include <list>
#include "ns3/ns-unknown.h"
+#include "ns3/ptr.h"
namespace ns3 {
@@ -43,7 +44,7 @@
public:
static const Iid iid;
typedef int Ipv4L4ProtocolTraceType;
- Ipv4L4Demux (Node *node);
+ Ipv4L4Demux (Ptr<Node> node);
virtual ~Ipv4L4Demux();
/**
@@ -86,7 +87,7 @@
virtual void DoDispose (void);
typedef std::list<Ipv4L4Protocol*> L4List_t;
L4List_t m_protocols;
- Node *m_node;
+ Ptr<Node> m_node;
};
} //namespace ns3