修改了 nvswitch-node.cc

Author: hhhhtw

simulation/src/point-to-point/model/nvswitch-node.cc

@@ -12,6 +12,7 @@
 #include "ns3/int-header.h"
 #include "ns3/simulator.h"
 #include <cmath>
+#include <unordered_set>
 
 namespace ns3 {
 
@@ -48,6 +49,7 @@ NVSwitchNode::NVSwitchNode(){
 		m_u[i] = 0;
 }
 
+//std::unordered_set<uint64_t> printedFlows;
 int NVSwitchNode::GetOutDev(Ptr<const Packet> p, CustomHeader &ch){
 	// look up entries
 	auto entry = m_rtTable.find(ch.dip);
@@ -66,6 +68,9 @@ int NVSwitchNode::GetOutDev(Ptr<const Packet> p, CustomHeader &ch){
 	} buf;
 	buf.u32[0] = ch.sip;
 	buf.u32[1] = ch.dip;
+	//uint32_t srcIp = ch.sip;
+	//uint32_t dstIp = ch.dip;
+	//uint64_t flowId = ((uint64_t)srcIp << 32) | dstIp;
 	if (ch.l3Prot == 0x6)
 		buf.u32[2] = ch.tcp.sport | ((uint32_t)ch.tcp.dport << 16);
 	else if (ch.l3Prot == 0x11)
@@ -74,6 +79,10 @@ int NVSwitchNode::GetOutDev(Ptr<const Packet> p, CustomHeader &ch){
 		buf.u32[2] = ch.ack.sport | ((uint32_t)ch.ack.dport << 16);
 
 	uint32_t idx = EcmpHash(buf.u8, 12, m_ecmpSeed) % nexthops.size();
+	//if (nexthops.size() > 0 && printedFlows.find(flowId) == printedFlows.end()) {
+	  //std::cout << "src IP: "<< ((srcIp >> 24) & 0xFF) << "."   << ((srcIp >> 16) & 0xFF) << "."   << ((srcIp >> 8) & 0xFF) << "."    << (srcIp & 0xFF)<<">"<<"Destination IP: "<< ((dstIp >> 24) & 0xFF) << "."   << ((dstIp >> 16) & 0xFF) << "."   << ((dstIp >> 8) & 0xFF) << "."    << (dstIp & 0xFF)<<" in nvswitch: "<<m_id<<" ECMP(header): "<<idx<< " selected interface: " << nexthops[idx]<<" selected node: "<<nexthops[idx]<<std::endl;
+		//printedFlows.insert(flowId);  //记录该流已经打印
+	//}
 	return nexthops[idx];
 }