Add startup experiments for NLSR and current testbed topology
refs: #4785
Change-Id: I957b8c229ed0696b2f3fca9445f9f27274b0e197
diff --git a/ndn/experiments/nlsr/mcn_failure_experiment.py b/ndn/experiments/nlsr/mcn_failure_experiment.py
new file mode 100644
index 0000000..cd448cc
--- /dev/null
+++ b/ndn/experiments/nlsr/mcn_failure_experiment.py
@@ -0,0 +1,79 @@
+# -*- Mode:python; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
+#
+# Copyright (C) 2015-2018, The University of Memphis,
+# Arizona Board of Regents,
+# Regents of the University of California.
+#
+# This file is part of Mini-NDN.
+# See AUTHORS.md for a complete list of Mini-NDN authors and contributors.
+#
+# Mini-NDN is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Mini-NDN 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 Mini-NDN, e.g., in COPYING.md file.
+# If not, see <http://www.gnu.org/licenses/>.
+
+from ndn.experiments.experiment import Experiment
+from ndn.apps.ndn_ping_client import NDNPingClient
+
+import time
+
+class MCNFailureExperiment(Experiment):
+
+ def __init__(self, args):
+ Experiment.__init__(self, args)
+
+ self.PING_COLLECTION_TIME_BEFORE_FAILURE = 60
+ self.PING_COLLECTION_TIME_AFTER_RECOVERY = 120
+
+ def getMostConnectedNode(self):
+ mcn = max(self.net.hosts, key=lambda host: len(host.intfNames()))
+ print "The most connected node is: {}".format(mcn.name)
+ return mcn
+
+ def setup(self):
+ if self.options.nPings != 0:
+ Experiment.setup(self)
+
+ def run(self):
+ mostConnectedNode = self.getMostConnectedNode()
+
+ if self.options.nPings != 0:
+ self.startPctPings()
+
+ # After the pings are scheduled, collect pings for 1 minute
+ time.sleep(self.PING_COLLECTION_TIME_BEFORE_FAILURE)
+
+ # Bring down MCN
+ self.failNode(mostConnectedNode)
+
+ # MCN is down for 2 minutes
+ time.sleep(int(self.options.arguments.waitTime))
+
+ # Bring MCN back up
+ self.recoverNode(mostConnectedNode)
+
+ # Restart pings
+ if self.options.nPings != 0:
+ for nodeToPing in self.pingedDict[mostConnectedNode]:
+ NDNPingClient.ping(mostConnectedNode, nodeToPing, self.PING_COLLECTION_TIME_AFTER_RECOVERY)
+
+ # Collect pings for more seconds after MCN is up
+ time.sleep(self.PING_COLLECTION_TIME_AFTER_RECOVERY)
+ else:
+ self.checkConvergence()
+
+ @staticmethod
+ def parseArguments(parser):
+ parser.add_argument("--wait-time", dest="waitTime", default="120",
+ help="[Experiment] Generic wait time for experiment use")
+
+Experiment.register("mcn-failure", MCNFailureExperiment)