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         ns3_wrapper.logger.debug("SHUTDOWN")
  48
  49         return "BYEBYE"
  50
  51     if msg_type == NS3WrapperMessage.STOP:
  52         time = kwargs.get("time")
  53
  54         ns3_wrapper.logger.debug("STOP time=%s" % str(time))
  55
  56         ns3_wrapper.stop(time=time)
  57         return "STOPPED"
  58
  59     if msg_type == NS3WrapperMessage.START:
  60         ns3_wrapper.logger.debug("START")
  61
  62         ns3_wrapper.start()
  63         return "STARTED"
  64
  65     if msg_type == NS3WrapperMessage.CREATE:
  66         clazzname = args.pop(0)
  67
  68         ns3_wrapper.logger.debug("CREATE %s %s" % (clazzname, str(args)))
  69
  70         uuid = ns3_wrapper.create(clazzname, *args)
  71
  72         #ns3_wrapper.logger.debug("%s = CREATE " % str(uuid))
  73         return uuid
  74
  75     if msg_type == NS3WrapperMessage.FACTORY:
  76         type_name = args.pop(0)
  77
  78         ns3_wrapper.logger.debug("FACTORY %s %s" % (type_name, str(kwargs)))
  79
  80         uuid = ns3_wrapper.factory(type_name, **kwargs)
  81
  82         #ns3_wrapper.logger.debug("%s = FACTORY " % str(uuid))
  83         return uuid
  84
  85     if msg_type == NS3WrapperMessage.INVOKE:
  86         uuid = args.pop(0)
  87         operation = args.pop(0)
  88
  89         ns3_wrapper.logger.debug("INVOKE %s %s %s %s " % (uuid, operation,
  90             str(args), str(kwargs)))
  91
  92         uuid = ns3_wrapper.invoke(uuid, operation, *args, **kwargs)
  93         return uuid
  94
  95     if msg_type == NS3WrapperMessage.GET:
  96         uuid = args.pop(0)
  97         name = args.pop(0)
  98
  99         ns3_wrapper.logger.debug("GET %s %s" % (uuid, name))
 100
 101         value = ns3_wrapper.get(uuid, name)
 102         return value
 103
 104     if msg_type == NS3WrapperMessage.SET:
 105         uuid = args.pop(0)
 106         name = args.pop(0)
 107         value = args.pop(0)
 108
 109         ns3_wrapper.logger.debug("SET %s %s %s" % (uuid, name, str(value)))
 110
 111         value = ns3_wrapper.set(uuid, name, value)
 112         return value
 113
 114     if msg_type == NS3WrapperMessage.FLUSH:
 115         # Forces flushing output and error streams.
 116         # NS-3 output will stay unflushed until the program exits or
 117         # explicit invocation flush is done
 118         sys.stdout.flush()
 119         sys.stderr.flush()
 120
 121         ns3_wrapper.logger.debug("FLUSHED")
 122
 123         return "FLUSHED"
 124
 125 def create_socket(socket_name):
 126     sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
 127     sock.bind(socket_name)
 128     return sock
 129
 130 def recv_msg(conn):
 131     msg = []
 132     chunk = ''
 133
 134     while '\n' not in chunk:
 135         try:
 136             chunk = conn.recv(1024)
 137         except (OSError, socket.error), e:
 138             if e[0] != errno.EINTR:
 139                 raise
 140             # Ignore eintr errors
 141             continue
 142
 143         if chunk:
 144             msg.append(chunk)
 145         else:
 146             # empty chunk = EOF
 147             break
 148
 149     msg = ''.join(msg).strip()
 150
 151     # The message is formatted as follows:
 152     #   MESSAGE_TYPE|args|kwargs
 153     #
 154     #   where MESSAGE_TYPE, args and kwargs are pickld and enoded in base64
 155
 156     def decode(item):
 157         item = base64.b64decode(item).rstrip()
 158         return cPickle.loads(item)
 159
 160     decoded = map(decode, msg.split("|"))
 161
 162     # decoded message
 163     dmsg_type = decoded.pop(0)
 164     dargs = list(decoded.pop(0)) # transforming touple into list
 165     dkwargs = decoded.pop(0)
 166
 167     return (dmsg_type, dargs, dkwargs)
 168
 169 def send_reply(conn, reply):
 170     encoded = base64.b64encode(cPickle.dumps(reply))
 171     conn.send("%s\n" % encoded)
 172
 173 def get_options():
 174     usage = ("usage: %prog -S <socket-name> -L <NS_LOG> -v ")
 175
 176     parser = OptionParser(usage = usage)
 177
 178     parser.add_option("-S", "--socket-name", dest="socket_name",
 179         help = "Name for the unix socket used to interact with this process",
 180         default = "tap.sock", type="str")
 181
 182     parser.add_option("-L", "--ns-log", dest="ns_log",
 183         help = "NS_LOG environmental variable to be set",
 184         default = "", type="str")
 185
 186     parser.add_option("-v", "--verbose",
 187         help="Print debug output",
 188         action="store_true",
 189         dest="verbose", default=False)
 190
 191     (options, args) = parser.parse_args()
 192
 193     return (options.socket_name, options.verbose, options.ns_log)
 194
 195 def run_server(socket_name, level = logging.INFO, ns_log = None):
 196
 197     # Sets NS_LOG environmental variable for NS debugging
 198     if ns_log:
 199         os.environ["NS_LOG"] = ns_log
 200
 201     ###### ns-3 wrapper instantiation
 202
 203     ns3_wrapper = NS3Wrapper(loglevel=level)
 204
 205     ns3_wrapper.logger.info("STARTING...")
 206
 207     # create unix socket to receive instructions
 208     sock = create_socket(socket_name)
 209     sock.listen(0)
 210
 211     # wait for messages to arrive and process them
 212     stop = False
 213
 214     while not stop:
 215         conn, addr = sock.accept()
 216         conn.settimeout(5)
 217
 218         try:
 219             (msg_type, args, kwargs) = recv_msg(conn)
 220         except socket.timeout, e:
 221             # Ingore time-out
 222             continue
 223
 224         if not msg_type:
 225             # Ignore - connection lost
 226             break
 227
 228         if msg_type == NS3WrapperMessage.SHUTDOWN:
 229            stop = True
 230
 231         try:
 232             reply = handle_message(ns3_wrapper, msg_type, args, kwargs)
 233         except:
 234             import traceback
 235             err = traceback.format_exc()
 236             ns3_wrapper.logger.error(err)
 237             raise
 238
 239         try:
 240             send_reply(conn, reply)
 241         except socket.error:
 242             break
 243
 244     ns3_wrapper.logger.info("EXITING...")
 245
 246 if __name__ == '__main__':
 247
 248     (socket_name, verbose, ns_log) = get_options()
 249
 250     ## configure logging
 251     FORMAT = "%(asctime)s %(name)s %(levelname)-4s %(message)s"
 252     level = logging.DEBUG if verbose else logging.INFO
 253
 254     logging.basicConfig(format = FORMAT, level = level)
 255
 256     ## Run the server
 257     run_server(socket_name, level, ns_log)
 258