src/nepi/resources/ns3/ns3server.py

   1 #
   2 #    NEPI, a framework to manage network experiments
   3 #    Copyright (C) 2014 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 as published by
   7 #    the Free Software Foundation, either version 3 of the License, or
   8 #    (at your option) any later version.
   9 #
  10 #    This program is distributed in the hope that it will be useful,
  11 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 #    GNU General Public License for more details.
  14 #
  15 #    You should have received a copy of the GNU General Public License
  16 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 #
  18 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
  19
  20 import base64
  21 import cPickle
  22 import errno
  23 import logging
  24 import os
  25 import socket
  26 import sys
  27
  28 from optparse import OptionParser, SUPPRESS_HELP
  29
  30 from ns3wrapper import NS3Wrapper
  31
  32 class NS3WrapperMessage:
  33     CREATE = "CREATE"
  34     FACTORY = "FACTORY"
  35     INVOKE = "INVOKE"
  36     SET = "SET"
  37     GET = "GET"
  38     FLUSH = "FLUSH"
  39     START = "START"
  40     STOP = "STOP"
  41     SHUTDOWN = "SHUTDOWN"
  42
  43 def handle_message(ns3_wrapper, msg_type, args, kwargs):
  44     if msg_type == NS3WrapperMessage.SHUTDOWN:
  45         ns3_wrapper.shutdown()
  46
  47         return "BYEBYE"
  48
  49     if msg_type == NS3WrapperMessage.STOP:
  50         time = kwargs.get("time")
  51
  52         ns3_wrapper.stop(time=time)
  53
  54         return "STOPPED"
  55
  56     if msg_type == NS3WrapperMessage.START:
  57         ns3_wrapper.start()
  58
  59         return "STARTED"
  60
  61     if msg_type == NS3WrapperMessage.CREATE:
  62         clazzname = args.pop(0)
  63
  64         return ns3_wrapper.create(clazzname, *args)
  65
  66     if msg_type == NS3WrapperMessage.FACTORY:
  67         type_name = args.pop(0)
  68
  69         return ns3_wrapper.factory(type_name, **kwargs)
  70
  71     if msg_type == NS3WrapperMessage.INVOKE:
  72         uuid = args.pop(0)
  73         operation = args.pop(0)
  74
  75         return ns3_wrapper.invoke(uuid, operation, *args, **kwargs)
  76
  77     if msg_type == NS3WrapperMessage.GET:
  78         uuid = args.pop(0)
  79         name = args.pop(0)
  80
  81         return ns3_wrapper.get(uuid, name)
  82
  83     if msg_type == NS3WrapperMessage.SET:
  84         uuid = args.pop(0)
  85         name = args.pop(0)
  86         value = args.pop(0)
  87
  88         return ns3_wrapper.set(uuid, name, value)
  89
  90     if msg_type == NS3WrapperMessage.FLUSH:
  91         # Forces flushing output and error streams.
  92         # NS-3 output will stay unflushed until the program exits or
  93         # explicit invocation flush is done
  94         sys.stdout.flush()
  95         sys.stderr.flush()
  96
  97         ns3_wrapper.logger.debug("FLUSHED")
  98
  99         return "FLUSHED"
 100
 101 def open_socket(socket_name):
 102     sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
 103     sock.bind(socket_name)
 104     return sock
 105
 106 def close_socket(sock):
 107     try:
 108         sock.close()
 109     except:
 110         pass
 111
 112 def recv_msg(conn, ns3_wrapper):
 113     msg = []
 114     chunk = ''
 115
 116     while '\n' not in chunk:
 117         try:
 118             chunk = conn.recv(1024)
 119         except (OSError, socket.error), e:
 120             import traceback
 121             err = traceback.format_exc()
 122             ns3_wrapper.logger.error(err)
 123
 124             if e[0] != errno.EINTR:
 125                 raise
 126             # Ignore eintr errors
 127             continue
 128
 129         if chunk:
 130             msg.append(chunk)
 131         else:
 132             # empty chunk = EOF
 133             break
 134
 135     msg = ''.join(msg).strip()
 136
 137     # The message is formatted as follows:
 138     #   MESSAGE_TYPE|args|kwargs
 139     #
 140     #   where MESSAGE_TYPE, args and kwargs are pickld and enoded in base64
 141
 142     def decode(item):
 143         item = base64.b64decode(item).rstrip()
 144         return cPickle.loads(item)
 145
 146     decoded = map(decode, msg.split("|"))
 147
 148     # decoded message
 149     dmsg_type = decoded.pop(0)
 150     dargs = list(decoded.pop(0)) # transforming touple into list
 151     dkwargs = decoded.pop(0)
 152
 153     return (dmsg_type, dargs, dkwargs)
 154
 155 def send_reply(conn, reply):
 156     encoded = base64.b64encode(cPickle.dumps(reply))
 157     conn.send("%s\n" % encoded)
 158
 159 def get_options():
 160     usage = ("usage: %prog -S <socket-name> -L <ns-log>  -D <enable-dump> -v ")
 161
 162     parser = OptionParser(usage = usage)
 163
 164     parser.add_option("-S", "--socket-name", dest="socket_name",
 165         help = "Name for the unix socket used to interact with this process",
 166         default = "tap.sock", type="str")
 167
 168     parser.add_option("-L", "--ns-log", dest="ns_log",
 169         help = "NS_LOG environmental variable to be set",
 170         default = "", type="str")
 171
 172     parser.add_option("-D", "--enable-dump", dest="enable_dump",
 173         help = "Enable dumping the remote executed ns-3 commands to a script "
 174             "in order to later reproduce and debug the experiment",
 175         action = "store_true",
 176         default = False)
 177
 178     parser.add_option("-v", "--verbose",
 179         help="Print debug output",
 180         action="store_true",
 181         dest="verbose", default=False)
 182
 183     (options, args) = parser.parse_args()
 184
 185     return (options.socket_name, options.verbose, options.ns_log,
 186             options.enable_dump)
 187
 188 def run_server(socket_name, level = logging.INFO, ns_log = None,
 189         enable_dump = False):
 190
 191     # Sets NS_LOG environmental variable for NS debugging
 192     if ns_log:
 193         os.environ["NS_LOG"] = ns_log
 194
 195     ###### ns-3 wrapper instantiation
 196
 197     ns3_wrapper = NS3Wrapper(loglevel=level, enable_dump = enable_dump)
 198
 199     ns3_wrapper.logger.info("STARTING...")
 200
 201     # create unix socket to receive instructions
 202     sock = open_socket(socket_name)
 203     sock.listen(0)
 204
 205     # wait for messages to arrive and process them
 206     stop = False
 207
 208     while not stop:
 209         conn, addr = sock.accept()
 210         conn.settimeout(5)
 211
 212         try:
 213             (msg_type, args, kwargs) = recv_msg(conn, ns3_wrapper)
 214         except socket.timeout, e:
 215             # Ingore time-out
 216             close_socket(conn)
 217             continue
 218
 219         if not msg_type:
 220             # Ignore - connection lost
 221             close_socket(conn)
 222             continue
 223
 224         if msg_type == NS3WrapperMessage.SHUTDOWN:
 225            stop = True
 226
 227         try:
 228             reply = handle_message(ns3_wrapper, msg_type, args, kwargs)
 229         except:
 230             import traceback
 231             err = traceback.format_exc()
 232             ns3_wrapper.logger.error(err)
 233             close_socket(conn)
 234             raise
 235
 236         try:
 237             send_reply(conn, reply)
 238         except socket.error:
 239             import traceback
 240             err = traceback.format_exc()
 241             ns3_wrapper.logger.error(err)
 242             close_socket(conn)
 243             raise
 244
 245         close_socket(conn)
 246
 247     close_socket(sock)
 248
 249     ns3_wrapper.logger.info("EXITING...")
 250
 251 if __name__ == '__main__':
 252
 253     (socket_name, verbose, ns_log, enable_dump) = get_options()
 254
 255     ## configure logging
 256     FORMAT = "%(asctime)s %(name)s %(levelname)-4s %(message)s"
 257     level = logging.DEBUG if verbose else logging.INFO
 258
 259     logging.basicConfig(format = FORMAT, level = level)
 260
 261     ## Run the server
 262     run_server(socket_name, level, ns_log, enable_dump)
 263