X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=src%2Fnepi%2Fresources%2Flinux%2Fns3%2Fns3simulation.py;h=0f94c4cb93a7f5d273e85f5b4849ee2a737977bf;hb=64fdda7bf665776fb043b355e989502e6c367a0c;hp=25b9535188ae88098c557d0ce050e4e6ba1e8878;hpb=e405b1997e739c7aa8de43a5f7639e6670964b92;p=nepi.git diff --git a/src/nepi/resources/linux/ns3/ns3simulation.py b/src/nepi/resources/linux/ns3/ns3simulation.py index 25b95351..0f94c4cb 100644 --- a/src/nepi/resources/linux/ns3/ns3simulation.py +++ b/src/nepi/resources/linux/ns3/ns3simulation.py @@ -20,7 +20,7 @@ from nepi.execution.attribute import Attribute, Flags, Types from nepi.execution.trace import Trace, TraceAttr from nepi.execution.resource import ResourceManager, clsinit_copy, \ - ResourceState, reschedule_delay + ResourceState, ResourceFactory, reschedule_delay from nepi.resources.linux.application import LinuxApplication from nepi.util.timefuncs import tnow, tdiffsec from nepi.resources.ns3.ns3simulation import NS3Simulation @@ -32,6 +32,9 @@ import os import time import threading +## TODO: Clean up DCE part. All that is DCE specific should go +## in the linux ns3dceapplication.py + @clsinit_copy class LinuxNS3Simulation(LinuxApplication, NS3Simulation): _rtype = "LinuxNS3Simulation" @@ -73,6 +76,13 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): type = Types.Bool, flags = Flags.Design) + enable_dump = Attribute("enableDump", + "Enable dumping the remote executed ns-3 commands to a script " + "in order to later reproduce and debug the experiment", + type = Types.Bool, + default = False, + flags = Flags.Design) + build_mode = Attribute("buildMode", "Mode used to build ns-3 with waf. One if: debug, release, oprimized ", default = "optimized", @@ -86,14 +96,9 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): default = "ns-3-dev", flags = Flags.Design) - enable_dce = Attribute("enableDCE", - "Install DCE source code", - default = False, - type = Types.Bool, - flags = Flags.Design) - pybindgen_version = Attribute("pybindgenVersion", "Version of pybindgen to install from bazar repo", + #default = "864", default = "868", flags = Flags.Design) @@ -107,12 +112,12 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): cls._register_attribute(sched_type) cls._register_attribute(check_sum) cls._register_attribute(ns_log) + cls._register_attribute(enable_dump) cls._register_attribute(verbose) cls._register_attribute(build_mode) cls._register_attribute(ns3_version) cls._register_attribute(pybindgen_version) cls._register_attribute(populate_routing_tables) - cls._register_attribute(enable_dce) def __init__(self, ec, guid): LinuxApplication.__init__(self, ec, guid) @@ -123,11 +128,7 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): self._socket_name = "ns3-%s.sock" % os.urandom(4).encode('hex') self._dce_manager_helper_uuid = None self._dce_application_helper_uuid = None - - # Lock used to synchronize usage of DceManagerHelper - self.dce_manager_lock = threading.Lock() - # Lock used to synchronize usage of DceApplicationHelper - self.dce_application_lock = threading.Lock() + self._enable_dce = None @property def socket_name(self): @@ -137,14 +138,6 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): def remote_socket(self): return os.path.join(self.run_home, self.socket_name) - @property - def dce_manager_helper_uuid(self): - return self._dce_manager_helper_uuid - - @property - def dce_application_helper_uuid(self): - return self._dce_application_helper_uuid - @property def ns3_build_home(self): return os.path.join(self.node.bin_dir, "ns-3", self.get("ns3Version"), @@ -165,6 +158,14 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): os.path.join(self.node.src_dir, "ns3wrapper", "ns3wrapper.py"), overwrite = False) + # upload ns3 wrapper debug python script + ns3_wrapper_debug = os.path.join(os.path.dirname(__file__), "..", "..", "ns3", + "ns3wrapper_debug.py") + + self.node.upload(ns3_wrapper_debug, + os.path.join(self.node.src_dir, "ns3wrapper", "ns3wrapper_debug.py"), + overwrite = False) + # upload ns3_server python script ns3_server = os.path.join(os.path.dirname(__file__), "..", "..", "ns3", "ns3server.py") @@ -187,6 +188,11 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): src_dir = os.path.join(self.node.src_dir, "ns-3") super(LinuxNS3Simulation, self).upload_sources(src_dir = src_dir) + + def upload_extra_sources(self, sources = None, src_dir = None): + return super(LinuxNS3Simulation, self).upload_sources( + sources = sources, + src_dir = src_dir) def upload_start_command(self): command = self.get("command") @@ -225,10 +231,6 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): stype = self.create("StringValue", sched_type) self.invoke(GLOBAL_VALUE_UUID, "Bind", "SchedulerType", btrue) - if self.get("enableDCE"): - self._dce_manager_helper_uuid = self.create("DceManagerHelper") - self._dce_application_helper_uuid = self.create("DceApplicationHelper") - def do_deploy(self): if not self.node or self.node.state < ResourceState.READY: self.debug("---- RESCHEDULING DEPLOY ---- node state %s " % self.node.state ) @@ -283,7 +285,26 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): if self.get("populateRoutingTables") == True: self.invoke(IPV4_GLOBAL_ROUTING_HELPER_UUID, "PopulateRoutingTables") - self._client.start() + self._client.start() + + """ + # XXX: IS THIS REALLY NEEDED?? + # Wait until the Simulation is actually started... + + is_running = False + for i in xrange(1000): + is_running = self.invoke(SIMULATOR_UUID, "isRunning") + + if is_running: + break + else: + time.sleep(1) + else: + if not is_running: + msg = " Simulation did not start" + self.error(msg) + raise RuntimeError + """ self.set_started() else: @@ -312,6 +333,22 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): super(LinuxApplication, self).do_release() + @property + def enable_dce(self): + if self._enable_dce is None: + from nepi.resources.ns3.ns3dceapplication import NS3BaseDceApplication + rclass = ResourceFactory.get_resource_type( + NS3BaseDceApplication.get_rtype()) + + self._enable_dce = False + for guid in self.ec.resources: + rm = self.ec.get_resource(guid) + if isinstance(rm, rclass): + self._enable_dce = True + break + + return self._enable_dce + @property def _start_command(self): command = [] @@ -325,6 +362,9 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): if ns_log: command.append("-L '%s'" % ns_log) + if self.get("enableDump"): + command.append("-D") + if self.get("verbose"): command.append("-v") @@ -349,8 +389,8 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): @property def dce_repo(self): - #return "http://code.nsnam.org/ns-3-dce" - return "http://code.nsnam.org/epmancini/ns-3-dce" + return "http://code.nsnam.org/ns-3-dce" + #eturn "http://code.nsnam.org/epmancini/ns-3-dce" @property def _build(self): @@ -396,7 +436,7 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): } clone_dce_cmd = " echo 'DCE will not be built' " - if self.get("enableDCE"): + if self.enable_dce: clone_dce_cmd = ( # DCE installation # Test if dce is alredy installed @@ -416,7 +456,6 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): 'dce_repo': self.dce_repo } - return ( # NS3 installation "( " @@ -483,7 +522,7 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): @property def _install(self): install_dce_cmd = " echo 'DCE will not be installed' " - if self.get("enableDCE"): + if self.enable_dce: install_dce_cmd = ( " ( " " ((test -d %(ns3_build_home)s/bin_dce ) && " @@ -493,6 +532,7 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): " ( " # If not, copy ns-3 build to bin " cd ${SRC}/dce/ns-3-dce && " + " rm -rf ${SRC}/dce/ns-3-dce/build && " " ./waf configure %(enable_opt)s --with-pybindgen=${SRC}/pybindgen/%(pybindgen_version)s " " --prefix=%(ns3_build_home)s --with-ns3=%(ns3_build_home)s && " " ./waf build && " @@ -520,6 +560,7 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): # If not, copy ns-3 build to bin " mkdir -p %(ns3_build_home)s && " " cd ${SRC}/ns-3/%(ns3_version)s && " + " rm -rf ${SRC}/ns-3/%(ns3_version)s/build && " " ./waf configure -d %(build_mode)s --with-pybindgen=${SRC}/pybindgen/%(pybindgen_version)s " " --prefix=%(ns3_build_home)s && " " ./waf build && " @@ -545,14 +586,38 @@ class LinuxNS3Simulation(LinuxApplication, NS3Simulation): env.append("PYTHONPATH=$PYTHONPATH:${NS3BINDINGS:=%(ns3_build_home)s/lib/python/site-packages}" % { 'ns3_build_home': self.ns3_build_home }) - env.append("LD_LIBRARY_PATH=${NS3LIBRARIES:=%(ns3_build_home)s/lib/}" % { + # If NS3LIBRARIES is defined and not empty, assign its value, + # if not assign ns3_build_home/lib/ to NS3LIBRARIES and LD_LIBARY_PATH + env.append("LD_LIBRARY_PATH=${NS3LIBRARIES:=%(ns3_build_home)s/lib}" % { 'ns3_build_home': self.ns3_build_home }) - env.append("DCE_PATH=$PATH:$NS3LIBRARIES/../bin_dce") - env.append("DCE_ROOT=$PATH:$NS3LIBRARIES/..") + env.append("DCE_PATH=$NS3LIBRARIES/../bin_dce") + env.append("DCE_ROOT=$NS3LIBRARIES/..") return " ".join(env) + def replace_paths(self, command): + """ + Replace all special path tags with shell-escaped actual paths. + """ + return ( command + .replace("${USR}", self.node.usr_dir) + .replace("${LIB}", self.node.lib_dir) + .replace("${BIN}", self.node.bin_dir) + .replace("${SRC}", self.node.src_dir) + .replace("${SHARE}", self.node.share_dir) + .replace("${EXP}", self.node.exp_dir) + .replace("${EXP_HOME}", self.node.exp_home) + .replace("${APP_HOME}", self.app_home) + .replace("${RUN_HOME}", self.run_home) + .replace("${NODE_HOME}", self.node.node_home) + .replace("${HOME}", self.node.home_dir) + # If NS3LIBRARIES is defined and not empty, use that value, + # if not use ns3_build_home/lib/ + .replace("${BIN_DCE}", "${NS3LIBRARIES-%s/lib}/../bin_dce" % \ + self.ns3_build_home) + ) + def valid_connection(self, guid): # TODO: Validate! return True