commit 2a269f7ffa917d75fbff216fa35bbc6390a35b1b
author Alexander Afanasyev <alexander.afanasyev@ucla.edu> Thu Jun 06 22:59:33 2013 -0700
committer Alexander Afanasyev <alexander.afanasyev@ucla.edu> Thu Jun 06 22:59:33 2013 -0700
tree 83c0bea1ceab903968e5930f5d0a447f110bdaba
parent b70a6c5823bdddc33253fd66e02ec7f742c41168

helper: Extending customization of Face creating in ndn::StackHelper

helper/ndn-stack-helper.cc

diff --git a/helper/ndn-stack-helper.cc b/helper/ndn-stack-helper.cc
index c6cf1c3..e0bf401 100644
--- a/helper/ndn-stack-helper.cc
+++ b/helper/ndn-stack-helper.cc
@@ -284,6 +284,31 @@
   m_netDeviceCallbacks.push_back (std::make_pair (netDeviceType, callback));
 }
 
+void
+StackHelper::UpdateNetDeviceFaceCreateCallback (TypeId netDeviceType, NetDeviceFaceCreateCallback callback)
+{
+  for (NetDeviceCallbackList::iterator i = m_netDeviceCallbacks.begin (); i != m_netDeviceCallbacks.end (); i++)
+    {
+      if (i->first == netDeviceType)
+        {
+          i->second = callback;
+          return;
+        }
+    }
+}
+
+void
+StackHelper::RemoveNetDeviceFaceCreateCallback (TypeId netDeviceType, NetDeviceFaceCreateCallback callback)
+{
+  for (NetDeviceCallbackList::iterator i = m_netDeviceCallbacks.begin (); i != m_netDeviceCallbacks.end (); i++)
+    {
+      if (i->first == netDeviceType)
+        {
+          m_netDeviceCallbacks.erase (i);
+          return;
+        }
+    }
+}
 
 Ptr<NetDeviceFace>
 StackHelper::DefaultNetDeviceCallback (Ptr<Node> node, Ptr<L3Protocol> ndn, Ptr<NetDevice> netDevice) const