tomh/ns-3-netfilter: src/helper/internet-stack-helper.cc@2398826af6b4 (annotated)

2833 f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	1	/* -- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -- */
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	2	/*
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	3	* Copyright (c) 2008 INRIA
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	4	*
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	5	* This program is free software; you can redistribute it and/or modify
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	6	* it under the terms of the GNU General Public License version 2 as
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	7	* published by the Free Software Foundation;
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	8	*
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	9	* This program is distributed in the hope that it will be useful,
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	10	* but WITHOUT ANY WARRANTY; without even the implied warranty of
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	11	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	12	* GNU General Public License for more details.
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	13	*
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	14	* You should have received a copy of the GNU General Public License
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	15	* along with this program; if not, write to the Free Software
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	16	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	17	*
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	18	* Author: Mathieu Lacage <mathieu.lacage@sophia.inria.fr>
f3474fd850b2 add missing license headers Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2737 diff changeset	19	*/
2708 83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	20	#include "internet-stack-helper.h"
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	21	#include "ns3/internet-stack.h"
2737 c692287f91da aggregate PacketSocketFactory unconditionally. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2708 diff changeset	22	#include "ns3/packet-socket-factory.h"
2845 2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	23	#include "ns3/config.h"
2708 83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	24
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	25	namespace ns3 {
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	26
2845 2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	27	std::vector<InternetStackHelper::Trace> InternetStackHelper::m_traces;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	28	std::string InternetStackHelper::m_pcapBaseFilename;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	29
2708 83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	30	void
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	31	InternetStackHelper::Build (NodeContainer c)
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	32	{
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	33	for (NodeContainer::Iterator i = c.Begin (); i != c.End (); ++i)
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	34	{
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	35	Ptr<Node> node = *i;
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	36	AddInternetStack (node);
2737 c692287f91da aggregate PacketSocketFactory unconditionally. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2708 diff changeset	37	Ptr<PacketSocketFactory> factory = CreateObject<PacketSocketFactory> ();
c692287f91da aggregate PacketSocketFactory unconditionally. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: 2708 diff changeset	38	node->AggregateObject (factory);
2708 83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	39	}
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	40	}
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	41
2845 2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	42	void
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	43	InternetStackHelper::EnablePcap (std::string filename)
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	44	{
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	45	InternetStackHelper::m_pcapBaseFilename = filename;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	46	Config::Connect ("/NodeList/*/$ns3::Ipv4L3Protocol/Tx",
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	47	MakeCallback (&InternetStackHelper::LogTxIp));
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	48	Config::Connect ("/NodeList/*/$ns3::Ipv4L3Protocol/Rx",
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	49	MakeCallback (&InternetStackHelper::LogRxIp));
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	50	}
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	51
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	52	uint32_t
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	53	InternetStackHelper::GetNodeIndex (std::string context)
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	54	{
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	55	std::string::size_type pos;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	56	pos = context.find ("/NodeList/");
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	57	NS_ASSERT (pos == 0);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	58	std::string::size_type afterNodeIndex = context.find ("/", 11);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	59	NS_ASSERT (afterNodeIndex != std::string::npos);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	60	std::string index = context.substr (10, afterNodeIndex - 10);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	61	std::istringstream iss;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	62	iss.str (index);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	63	uint32_t nodeIndex;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	64	iss >> nodeIndex;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	65	return nodeIndex;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	66	}
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	67
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	68	void
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	69	InternetStackHelper::LogTxIp (std::string context, Ptr<const Packet> packet, uint32_t interfaceIndex)
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	70	{
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	71	Ptr<PcapWriter> writer = InternetStackHelper::GetStream (GetNodeIndex (context), interfaceIndex);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	72	writer->WritePacket (packet);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	73	}
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	74
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	75	void
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	76	InternetStackHelper::LogRxIp (std::string context, Ptr<const Packet> packet, uint32_t interfaceIndex)
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	77	{
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	78	Ptr<PcapWriter> writer = InternetStackHelper::GetStream (GetNodeIndex (context), interfaceIndex);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	79	writer->WritePacket (packet);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	80	}
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	81
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	82	Ptr<PcapWriter>
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	83	InternetStackHelper::GetStream (uint32_t nodeId, uint32_t interfaceId)
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	84	{
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	85	for (std::vector<Trace>::iterator i = m_traces.begin ();
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	86	i != m_traces.end (); i++)
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	87	{
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	88	if (i->nodeId == nodeId &&
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	89	i->interfaceId == interfaceId)
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	90	{
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	91	return i->writer;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	92	}
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	93	}
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	94	InternetStackHelper::Trace trace;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	95	trace.nodeId = nodeId;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	96	trace.interfaceId = interfaceId;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	97	trace.writer = Create<PcapWriter> ();
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	98	std::ostringstream oss;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	99	oss << m_pcapBaseFilename << ".pcap-" << nodeId << "-" << interfaceId;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	100	trace.writer->Open (oss.str ());
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	101	trace.writer->WriteIpHeader ();
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	102	m_traces.push_back (trace);
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	103	return trace.writer;
2398826af6b4 Add IP layer tracing helpers to InternetStackHelper Raj Bhattacharjea <raj.b@gatech.edu> parents: 2834 diff changeset	104	}
2708 83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	105
83cc896e501c InternetStackHelper. Mathieu Lacage <mathieu.lacage@sophia.inria.fr> parents: diff changeset	106	} // namespace ns3

author	Raj Bhattacharjea <raj.b@gatech.edu>
	Wed, 02 Apr 2008 14:09:36 -0400
changeset 2845	2398826af6b4
parent 2834	1aab57845b07
child 2865	d40eb18a4da0
permissions	-rw-r--r--