introduce AddInternetStack
authorMathieu Lacage <mathieu.lacage@sophia.inria.fr>
Mon, 24 Mar 2008 11:56:50 -0700
changeset 2707 1c353c7aef5b
parent 2706 1e5771a4796c
child 2708 83cc896e501c
introduce AddInternetStack
src/internet-node/internet-node.cc
src/internet-node/internet-node.h
src/internet-node/internet-stack.cc
src/internet-node/internet-stack.h
src/internet-node/wscript
--- a/src/internet-node/internet-node.cc	Mon Mar 24 11:50:58 2008 -0700
+++ b/src/internet-node/internet-node.cc	Mon Mar 24 11:56:50 2008 -0700
@@ -23,77 +23,16 @@
 
 #include "ns3/net-device.h"
 #include "ns3/callback.h"
+#include "internet-node.h"
+#include "internet-stack.h"
 
-#include "ipv4-l4-demux.h"
-#include "internet-node.h"
-#include "udp-l4-protocol.h"
-#include "tcp-l4-protocol.h"
-#include "ipv4-l3-protocol.h"
-#include "arp-l3-protocol.h"
-#include "udp-impl.h"
-#include "tcp-impl.h"
-#include "ipv4-impl.h"
 
 namespace ns3 {
 
 InternetNode::InternetNode()
 {
-  Construct ();
-}
-
-InternetNode::InternetNode(uint32_t systemId)
-{
-  Construct ();
+  AddInternetStack (this);
 }
 
-InternetNode::~InternetNode ()
-{}
-
-void
-InternetNode::Construct (void)
-{
-  Ptr<Ipv4L3Protocol> ipv4 = CreateObject<Ipv4L3Protocol> ();
-  Ptr<ArpL3Protocol> arp = CreateObject<ArpL3Protocol> ();
-  ipv4->SetNode (this);
-  arp->SetNode (this);
-  // XXX remove the PeekPointer below.
-  RegisterProtocolHandler (MakeCallback (&Ipv4L3Protocol::Receive, PeekPointer (ipv4)), 
-                           Ipv4L3Protocol::PROT_NUMBER, 0);
-  RegisterProtocolHandler (MakeCallback (&ArpL3Protocol::Receive, PeekPointer (arp)),
-                           ArpL3Protocol::PROT_NUMBER, 0);
-
 
-  Ptr<Ipv4L4Demux> ipv4L4Demux = CreateObject<Ipv4L4Demux> ();
-  Ptr<UdpL4Protocol> udp = CreateObject<UdpL4Protocol> ();
-  Ptr<TcpL4Protocol> tcp = CreateObject<TcpL4Protocol> ();
-
-  ipv4L4Demux->SetNode (this);
-  udp->SetNode (this);
-  tcp->SetNode (this);
-
-  ipv4L4Demux->Insert (udp);
-  ipv4L4Demux->Insert (tcp);
-
-  Ptr<UdpImpl> udpImpl = CreateObject<UdpImpl> ();
-  Ptr<TcpImpl> tcpImpl = CreateObject<TcpImpl> ();
-  Ptr<Ipv4Impl> ipv4Impl = CreateObject<Ipv4Impl> ();
-
-  udpImpl->SetUdp (udp);
-  tcpImpl->SetTcp (tcp);
-  ipv4Impl->SetIpv4 (ipv4);
-
-  Object::AggregateObject (ipv4);
-  Object::AggregateObject (arp);
-  Object::AggregateObject (ipv4Impl);
-  Object::AggregateObject (udpImpl);
-  Object::AggregateObject (tcpImpl);
-  Object::AggregateObject (ipv4L4Demux);
-}
-
-void 
-InternetNode::DoDispose()
-{
-  Node::DoDispose ();
-}
-
 }//namespace ns3
--- a/src/internet-node/internet-node.h	Mon Mar 24 11:50:58 2008 -0700
+++ b/src/internet-node/internet-node.h	Mon Mar 24 11:56:50 2008 -0700
@@ -58,13 +58,6 @@
 {
 public:
   InternetNode();
-  InternetNode(uint32_t systemId);
-  virtual ~InternetNode ();
-
-protected:
-  virtual void DoDispose(void);
-private:
-  void Construct (void);
 };
 
 }//namespace ns3
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/internet-node/internet-stack.cc	Mon Mar 24 11:56:50 2008 -0700
@@ -0,0 +1,78 @@
+// -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*-
+//
+// Copyright (c) 2006 Georgia Tech Research Corporation
+// All rights reserved.
+//
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License version 2 as
+// published by the Free Software Foundation;
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+//
+// Author: George F. Riley<riley@ece.gatech.edu>
+//
+
+#include "ns3/net-device.h"
+#include "ns3/callback.h"
+
+#include "ipv4-l4-demux.h"
+#include "internet-node.h"
+#include "udp-l4-protocol.h"
+#include "tcp-l4-protocol.h"
+#include "ipv4-l3-protocol.h"
+#include "arp-l3-protocol.h"
+#include "udp-impl.h"
+#include "tcp-impl.h"
+#include "ipv4-impl.h"
+
+namespace ns3 {
+
+void 
+AddInternetStack (Ptr<Node> node)
+{
+  Ptr<Ipv4L3Protocol> ipv4 = CreateObject<Ipv4L3Protocol> ();
+  Ptr<ArpL3Protocol> arp = CreateObject<ArpL3Protocol> ();
+  ipv4->SetNode (node);
+  arp->SetNode (node);
+  // XXX remove the PeekPointer below.
+  node->RegisterProtocolHandler (MakeCallback (&Ipv4L3Protocol::Receive, PeekPointer (ipv4)), 
+                                 Ipv4L3Protocol::PROT_NUMBER, 0);
+  node->RegisterProtocolHandler (MakeCallback (&ArpL3Protocol::Receive, PeekPointer (arp)),
+                                 ArpL3Protocol::PROT_NUMBER, 0);
+
+
+  Ptr<Ipv4L4Demux> ipv4L4Demux = CreateObject<Ipv4L4Demux> ();
+  Ptr<UdpL4Protocol> udp = CreateObject<UdpL4Protocol> ();
+  Ptr<TcpL4Protocol> tcp = CreateObject<TcpL4Protocol> ();
+
+  ipv4L4Demux->SetNode (node);
+  udp->SetNode (node);
+  tcp->SetNode (node);
+
+  ipv4L4Demux->Insert (udp);
+  ipv4L4Demux->Insert (tcp);
+
+  Ptr<UdpImpl> udpImpl = CreateObject<UdpImpl> ();
+  Ptr<TcpImpl> tcpImpl = CreateObject<TcpImpl> ();
+  Ptr<Ipv4Impl> ipv4Impl = CreateObject<Ipv4Impl> ();
+
+  udpImpl->SetUdp (udp);
+  tcpImpl->SetTcp (tcp);
+  ipv4Impl->SetIpv4 (ipv4);
+
+  node->AggregateObject (ipv4);
+  node->AggregateObject (arp);
+  node->AggregateObject (ipv4Impl);
+  node->AggregateObject (udpImpl);
+  node->AggregateObject (tcpImpl);
+  node->AggregateObject (ipv4L4Demux);
+}
+
+}//namespace ns3
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/internet-node/internet-stack.h	Mon Mar 24 11:56:50 2008 -0700
@@ -0,0 +1,34 @@
+// -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*-
+//
+// Copyright (c) 2006 Georgia Tech Research Corporation
+// All rights reserved.
+//
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License version 2 as
+// published by the Free Software Foundation;
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+//
+// Author: George F. Riley<riley@ece.gatech.edu>
+
+#ifndef INTERNET_STACK_H
+#define INTERNET_STACK_H
+
+#include "ns3/ptr.h"
+
+namespace ns3 {
+
+class Node;
+
+void AddInternetStack (Ptr<Node> node);
+
+}//namespace ns3
+
+#endif /* INTERNET_STACK_H */
--- a/src/internet-node/wscript	Mon Mar 24 11:50:58 2008 -0700
+++ b/src/internet-node/wscript	Mon Mar 24 11:56:50 2008 -0700
@@ -5,6 +5,7 @@
     obj = bld.create_ns3_module('internet-node', ['node'])
     obj.source = [
         'internet-node.cc',
+        'internet-stack.cc',
         'ipv4-l4-demux.cc',
         'ipv4-l4-protocol.cc',
         'ipv4-header.cc',
@@ -39,6 +40,7 @@
     headers.module = 'internet-node'
     headers.source = [
         'internet-node.h',
+        'internet-stack.h',
         'ascii-trace.h',
         'pcap-trace.h',
         'ipv4-header.h',