1 # Copyright (C) 2014 Nippon Telegraph and Telephone Corporation.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
20 from ryu.base import app_manager
22 from ryu.lib import hub
23 from ryu.lib.hub import StreamServer
24 from ryu.lib.packet import bmp
27 class BMPStation(app_manager.RyuApp):
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')
37 self.output_fd = open(output_file, 'w')
38 self.failed_dump_fd = open(failed_dump, 'w')
40 self.failed_pkt_count = 0
43 super(BMPStation, self).start()
44 self.logger.debug("listening on %s:%s", self.server_host,
47 return hub.spawn(StreamServer((self.server_host, self.server_port),
48 self.loop).serve_forever)
50 def loop(self, sock, addr):
51 self.logger.debug("BMP client connected, ip=%s, port=%s", addr[0],
55 required_len = bmp.BMPMessage._HDR_LEN
58 ret = sock.recv(required_len)
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)
71 if len(buf) < required_len:
75 msg, rest = bmp.BMPMessage.parser(buf)
76 except Exception as e:
78 self.failed_dump_fd.write(pkt)
79 self.failed_dump_fd.flush()
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)
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],
90 self.output_fd.flush()
93 required_len = bmp.BMPMessage._HDR_LEN
95 self.logger.debug("BMP client disconnected, ip=%s, port=%s", addr[0],