src/nepi/resources/netns/netnswrapper.py

   1 #
   2 #    NEPI, a framework to manage network experiments
   3 #    Copyright (C) 2013 INRIA
   4 #
   5 #    This program is free software: you can redistribute it and/or modify
   6 #    it under the terms of the GNU General Public License version 2 as
   7 #    published by the Free Software Foundation;
   8 #
   9 #    This program is distributed in the hope that it will be useful,
  10 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 #    GNU General Public License for more details.
  13 #
  14 #    You should have received a copy of the GNU General Public License
  15 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 #
  17 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
  18
  19 import logging
  20 import time
  21 import os
  22 import sys
  23 import uuid
  24
  25 class NetNSWrapper(object):
  26     def __init__(self, loglevel = logging.INFO, enable_dump = False):
  27         super(NetNSWrapper, self).__init__()
  28         # holds reference to all C++ objects and variables in the simulation
  29         self._objects = dict()
  30
  31         # Logging
  32         self._logger = logging.getLogger("netnswrapper")
  33         self._logger.setLevel(loglevel)
  34
  35         # Object to dump instructions to reproduce and debug experiment
  36         from netnswrapper_debug import NetNSWrapperDebuger
  37         self._debuger = NetNSWrapperDebuger(enabled = enable_dump)
  38
  39     @property
  40     def debuger(self):
  41         return self._debuger
  42
  43     @property
  44     def logger(self):
  45         return self._logger
  46
  47     def make_uuid(self):
  48         return "uuid%s" % uuid.uuid4()
  49
  50     def get_object(self, uuid):
  51         return self._objects.get(uuid)
  52
  53     def create(self, clazzname, *args):
  54         """ This method should be used to construct netns objects """
  55         import netns
  56
  57         if clazzname not in ['open'] and not hasattr(netns, clazzname):
  58             msg = "Type %s not supported" % (clazzname)
  59             self.logger.error(msg)
  60
  61         uuid = self.make_uuid()
  62
  63         ### DEBUG
  64         self.logger.debug("CREATE %s( %s )" % (clazzname, str(args)))
  65
  66         self.debuger.dump_create(uuid, clazzname, args)
  67         ########
  68
  69         if clazzname == "open":
  70             path = args[0]
  71             mode = args[1]
  72             obj = open(path, mode)
  73         else:
  74             clazz = getattr(netns, clazzname)
  75
  76             # arguments starting with 'uuid' identify ns-3 C++
  77             # objects and must be replaced by the actual object
  78             realargs = self.replace_args(args)
  79
  80             obj = clazz(*realargs)
  81
  82         self._objects[uuid] = obj
  83
  84         ### DEBUG
  85         self.logger.debug("RET CREATE ( uuid %s ) %s = %s( %s )" % (str(uuid),
  86             str(obj), clazzname, str(args)))
  87         ########
  88
  89         return uuid
  90
  91     def invoke(self, uuid, operation, *args, **kwargs):
  92         newuuid = self.make_uuid()
  93
  94         ### DEBUG
  95         self.logger.debug("INVOKE %s -> %s( %s, %s ) " % (
  96             uuid, operation, str(args), str(kwargs)))
  97
  98         self.debuger.dump_invoke(newuuid, uuid, operation, args, kwargs)
  99         ########
 100
 101         obj = self.get_object(uuid)
 102
 103         method = getattr(obj, operation)
 104
 105         # arguments starting with 'uuid' identify netns
 106         # objects and must be replaced by the actual object
 107         realargs = self.replace_args(args)
 108         realkwargs = self.replace_kwargs(kwargs)
 109
 110         result = method(*realargs, **realkwargs)
 111
 112         # If the result is an object (not a base value),
 113         # then keep track of the object a return the object
 114         # reference (newuuid)
 115         if not (result is None or type(result) in [
 116                 bool, float, long, str, int]):
 117             self._objects[newuuid] = result
 118             result = newuuid
 119
 120         ### DEBUG
 121         self.logger.debug("RET INVOKE %s%s = %s -> %s(%s, %s) " % (
 122             "(uuid %s) " % str(newuuid) if newuuid else "", str(result), uuid,
 123             operation, str(args), str(kwargs)))
 124         ########
 125
 126         return result
 127
 128     def set(self, uuid, name, value):
 129         ### DEBUG
 130         self.logger.debug("SET %s %s %s" % (uuid, name, str(value)))
 131
 132         self.debuger.dump_set(uuid, name, value)
 133         ########
 134
 135         obj = self.get_object(uuid)
 136         setattr(obj, name, value)
 137
 138         ### DEBUG
 139         self.logger.debug("RET SET %s = %s -> set(%s, %s)" % (str(value), uuid, name,
 140             str(value)))
 141         ########
 142
 143         return value
 144
 145     def get(self, uuid, name):
 146         ### DEBUG
 147         self.logger.debug("GET %s %s" % (uuid, name))
 148
 149         self.debuger.dump_get(uuid, name)
 150         ########
 151
 152         obj = self.get_object(uuid)
 153         result = getattr(obj, name)
 154
 155         ### DEBUG
 156         self.logger.debug("RET GET %s = %s -> get(%s)" % (str(result), uuid, name))
 157         ########
 158
 159         return result
 160
 161     def shutdown(self):
 162         ### DEBUG
 163         self.debuger.dump_shutdown()
 164         ########
 165
 166         ### FLUSH PIPES
 167         sys.stdout.flush()
 168         sys.stderr.flush()
 169
 170         ### RELEASE OBJECTS
 171         del self._objects
 172
 173         ### DEBUG
 174         self.logger.debug("SHUTDOWN")
 175         ########
 176
 177     def replace_args(self, args):
 178         realargs = [self.get_object(arg) if \
 179                 str(arg).startswith("uuid") else arg for arg in args]
 180
 181         return realargs
 182
 183     def replace_kwargs(self, kwargs):
 184         realkwargs = dict([(k, self.get_object(v) \
 185                 if str(v).startswith("uuid") else v) \
 186                 for k,v in kwargs.iteritems()])
 187
 188         return realkwargs
 189