ryu/build/lib.linux-armv7l-2.7/ryu/app/bmpstation.py

   1 # Copyright (C) 2014 Nippon Telegraph and Telephone Corporation.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #    http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
  12 # implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 import os
  17 import socket
  18 import time
  19
  20 from ryu.base import app_manager
  21
  22 from ryu.lib import hub
  23 from ryu.lib.hub import StreamServer
  24 from ryu.lib.packet import bmp
  25
  26
  27 class BMPStation(app_manager.RyuApp):
  28     def __init__(self):
  29         super(BMPStation, self).__init__()
  30         self.name = 'bmpstation'
  31         self.server_host = os.environ.get('RYU_BMP_SERVER_HOST', '0.0.0.0')
  32         self.server_port = int(os.environ.get('RYU_BMP_SERVER_PORT', 11019))
  33         output_file = os.environ.get('RYU_BMP_OUTPUT_FILE', 'ryu_bmp.log')
  34         failed_dump = os.environ.get('RYU_BMP_FAILED_DUMP',
  35                                      'ryu_bmp_failed.dump')
  36
  37         self.output_fd = open(output_file, 'w')
  38         self.failed_dump_fd = open(failed_dump, 'w')
  39
  40         self.failed_pkt_count = 0
  41
  42     def start(self):
  43         super(BMPStation, self).start()
  44         self.logger.debug("listening on %s:%s", self.server_host,
  45                           self.server_port)
  46
  47         return hub.spawn(StreamServer((self.server_host, self.server_port),
  48                                       self.loop).serve_forever)
  49
  50     def loop(self, sock, addr):
  51         self.logger.debug("BMP client connected, ip=%s, port=%s", addr[0],
  52                           addr[1])
  53         is_active = True
  54         buf = bytearray()
  55         required_len = bmp.BMPMessage._HDR_LEN
  56
  57         while is_active:
  58             ret = sock.recv(required_len)
  59             if len(ret) == 0:
  60                 is_active = False
  61                 break
  62             buf += ret
  63             while len(buf) >= required_len:
  64                 version, len_, _ = bmp.BMPMessage.parse_header(buf)
  65                 if version != bmp.VERSION:
  66                     self.logger.error("unsupported bmp version: %d", version)
  67                     is_active = False
  68                     break
  69
  70                 required_len = len_
  71                 if len(buf) < required_len:
  72                     break
  73
  74                 try:
  75                     msg, rest = bmp.BMPMessage.parser(buf)
  76                 except Exception as e:
  77                     pkt = buf[:len_]
  78                     self.failed_dump_fd.write(pkt)
  79                     self.failed_dump_fd.flush()
  80                     buf = buf[len_:]
  81                     self.failed_pkt_count += 1
  82                     self.logger.error("failed to parse: %s"
  83                                       " (total fail count: %d)",
  84                                       e, self.failed_pkt_count)
  85                 else:
  86                     t = time.strftime("%Y %b %d %H:%M:%S", time.localtime())
  87                     self.logger.debug("%s | %s | %s\n", t, addr[0], msg)
  88                     self.output_fd.write("%s | %s | %s\n\n" % (t, addr[0],
  89                                                                msg))
  90                     self.output_fd.flush()
  91                     buf = rest
  92
  93                 required_len = bmp.BMPMessage._HDR_LEN
  94
  95         self.logger.debug("BMP client disconnected, ip=%s, port=%s", addr[0],
  96                           addr[1])
  97
  98         sock.close()