diff --git a/src/internet/helper/internet-stack-helper.cc b/src/internet/helper/internet-stack-helper.cc index f1f80b210..ba261bf38 100644 --- a/src/internet/helper/internet-stack-helper.cc +++ b/src/internet/helper/internet-stack-helper.cc @@ -398,10 +398,6 @@ InternetStackHelper::Install (Ptr node) const CreateAndAggregateObjectFromTypeId (node, "ns3::ArpL3Protocol"); CreateAndAggregateObjectFromTypeId (node, "ns3::Ipv4L3Protocol"); CreateAndAggregateObjectFromTypeId (node, "ns3::Icmpv4L4Protocol"); - CreateAndAggregateObjectFromTypeId (node, "ns3::UdpL4Protocol"); - node->AggregateObject (m_tcpFactory.Create ()); - Ptr factory = CreateObject (); - node->AggregateObject (factory); // Set routing Ptr ipv4 = node->GetObject (); Ptr ipv4Routing = m_routing->Create (node); @@ -420,7 +416,7 @@ InternetStackHelper::Install (Ptr node) const CreateAndAggregateObjectFromTypeId (node, "ns3::Ipv6L3Protocol"); CreateAndAggregateObjectFromTypeId (node, "ns3::Icmpv6L4Protocol"); - /* TODO add UdpL4Protocol/TcpL4Protocol for IPv6 */ + // Set routing Ptr ipv6 = node->GetObject (); Ptr ipv6Routing = m_routingv6->Create (node); ipv6->SetRoutingProtocol (ipv6Routing); @@ -429,6 +425,14 @@ InternetStackHelper::Install (Ptr node) const ipv6->RegisterExtensions (); ipv6->RegisterOptions (); } + + if (m_ipv4Enabled || m_ipv6Enabled) + { + CreateAndAggregateObjectFromTypeId (node, "ns3::UdpL4Protocol"); + node->AggregateObject (m_tcpFactory.Create ()); + Ptr factory = CreateObject (); + node->AggregateObject (factory); + } } void