test tool: add OpenFlowSwitch test tool

What this patch implements:

    This patch implements a tool which performs OpenFlow conformance testing.
    The tool can:
      - install various flow entries of OpenFlow 1.3 as follows:
        * all 'action' except SET_QUEUE and GROUP
        * all 'match' except IN_PHY_PORT
      - confirm whether the datapath works correctly according to the installed flow entries.

Required test environment:

    The tool requires the test environment that includes 2 switches and 2 links as follows:

                      +-----------+
           +----------| target sw | The OpenFlow switch to be tested
           |          +-----------+
    +------------+      (1)   (2)
    | controller |       |     |
    +------------+      (1)   (2)
           |          +-----------+
           +----------| tester sw | OpenFlow Switch
                      +-----------+

      (X) : port number

How does the tool work:

    The tool will:
      - install the flow entries into the target switch.
      - transmit a packet from the tester switch to the target switch.
        * According to the installed flow entries, the target switch outputs a packet towards the tester switch.
        * The tester switch sends a PacketIn message to the tool.
      - examine the packet that is included in the PacketIn message and output the result.
        * The tool supports also the table-miss examinations.

    The tool processes test files in the specified directory sequentially.

    By forming the result, you can create the list as:
    http://osrg.github.io/ryu-certification/switch/ovs.html
    http://osrg.github.io/ryu-certification/switch/LINC.html
    http://osrg.github.io/ryu-certification/switch/ofsoftswitch13.html

Required test file format:

    [
        "test name",
        {
            test pattern #01
        },
        {
            test pattern #02
        }, ...
    ]

    Each test pattern has the following internal format:

    {
        "description": description of this test pattern,
        "prerequisite": [
            FlowMod messages used by this test pattern
        ],
        "tests": [
            "ingress": inbound packet data,
            "egress": outbound packet data,
            "PACKET-IN": packet data sent by PacketIn message,
            "table-miss": table id in which table-miss occurs
        ]
    }

    The extensions of the test files have to be '.json'.

How to run:

    Do the following command:
    ryu-manager ryu/tests/switch/tester.py

    The following options can be used:
      - --test-switch-target   (target sw dp-id)
      - --test-switch-tester   (tester sw dp-id)
      - --test-switch-dir      (test files directory)

    ex)
    ryu-manager --test-switch-target 0000000000000005 --test-switch-dir /home/ryu/tests ryu/tests/switch/tester.py

Signed-off-by: WATANABE Fumitaka <watanabe.fumitaka@nttcom.co.jp>
Signed-off-by: Yuichi Ito <ito.yuichi0@gmail.com>
Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
This commit is contained in:
Yuichi Ito 2013-12-13 17:25:24 +09:00 committed by FUJITA Tomonori
parent c44f71a7e1
commit 70542380d3
2 changed files with 906 additions and 0 deletions

View File

@ -50,3 +50,11 @@ CONF.register_cli_opts([
'ovs bridge',
deprecated_name='quantum-controller-addr')
])
CONF.register_cli_opts([
# tests/switch/tester
cfg.StrOpt('target', default='0000000000000001', help='target sw dp-id'),
cfg.StrOpt('tester', default='0000000000000002', help='tester sw dp-id'),
cfg.StrOpt('dir', default='ryu/tests/switch/of13',
help='test files directory')
], group='test-switch')

898
ryu/tests/switch/tester.py Normal file
View File

@ -0,0 +1,898 @@
# Copyright (C) 2013 Nippon Telegraph and Telephone Corporation.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
# implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import inspect
import json
import logging
import os
import signal
import sys
import traceback
from oslo.config import cfg
# import all packet libraries.
PKT_LIB_PATH = 'ryu.lib.packet'
for modname, moddef in sys.modules.iteritems():
if not modname.startswith(PKT_LIB_PATH) or not moddef:
continue
for (clsname, clsdef, ) in inspect.getmembers(moddef):
if not inspect.isclass(clsdef):
continue
exec 'from %s import %s' % (modname, clsname)
from ryu.base import app_manager
from ryu.controller import handler
from ryu.controller import ofp_event
from ryu.controller.handler import set_ev_cls
from ryu.exception import RyuException
from ryu.lib import dpid as dpid_lib
from ryu.lib import hub
from ryu.lib import stringify
from ryu.lib.packet import packet
from ryu.ofproto import ofproto_v1_3
from ryu.ofproto import ofproto_v1_3_parser
""" Required test network:
+-----------+
+----------| target sw | The switch to be tested
| +-----------+
+------------+ (1) (2)
| controller | | |
+------------+ (1) (2)
| +-----------+
+----------| tester sw | OpenFlow Switch
+-----------+
(X) : port number
Tests send a packet from port 1 of the tester sw.
If the packet matched with a flow entry of the target sw,
the target sw resends the packet from port 2 (or the port which
connected with the controller), according to the flow entry.
Then the tester sw receives the packet and sends a PacketIn message.
If the packet did not match, the target sw drops the packet.
"""
CONF = cfg.CONF
# Default settings.
TESTER_SENDER_PORT = 1
TESTER_RECEIVE_PORT = 2
TARGET_SENDER_PORT = 2
TARGET_RECEIVE_PORT = 1
INTERVAL = 1 # sec
WAIT_TIMER = 3 # sec
# Test file format.
KEY_DESC = 'description'
KEY_PREREQ = 'prerequisite'
KEY_FLOW = 'OFPFlowMod'
KEY_TESTS = 'tests'
KEY_INGRESS = 'ingress'
KEY_EGRESS = 'egress'
KEY_PKT_IN = 'PACKET_IN'
KEY_TBL_MISS = 'table-miss'
# Test state.
STATE_INIT = 0
STATE_FLOW_INSTALL = 1
STATE_FLOW_EXIST_CHK = 2
STATE_TARGET_PKT_COUNT = 3
STATE_TESTER_PKT_COUNT = 4
STATE_FLOW_MATCH_CHK = 5
STATE_NO_PKTIN_REASON = 6
STATE_GET_MATCH_COUNT = 7
STATE_UNMATCH_PKT_SEND = 8
STATE_FLOW_UNMATCH_CHK = 9
# Test result.
TEST_OK = 'OK'
TEST_ERROR = 'ERROR'
RYU_INTERNAL_ERROR = '- (Ryu internal error.)'
TEST_FILE_ERROR = '%(file)s : Test file format error (%(detail)s)'
NO_TEST_FILE = 'Test file (*.json) is not found.'
INVALID_PATH = '%(path)s : No such file or directory.'
# Test result details.
FAILURE = 0
ERROR = 1
TIMEOUT = 2
RCV_ERR = 3
MSG = {STATE_INIT:
{TIMEOUT: 'Failed to initialize flow tables: barrier request timeout.',
RCV_ERR: 'Failed to initialize flow tables: %(err_msg)s'},
STATE_FLOW_INSTALL:
{TIMEOUT: 'Failed to add flows: barrier request timeout.',
RCV_ERR: 'Failed to add flows: %(err_msg)s'},
STATE_FLOW_EXIST_CHK:
{FAILURE: 'Added incorrect flows: %(flows)s',
TIMEOUT: 'Failed to add flows: flow stats request timeout.',
RCV_ERR: 'Failed to add flows: %(err_msg)s'},
STATE_TARGET_PKT_COUNT:
{TIMEOUT: 'Failed to request port stats from target: request timeout.',
RCV_ERR: 'Failed to request port stats from target: %(err_msg)s'},
STATE_TESTER_PKT_COUNT:
{TIMEOUT: 'Failed to request port stats from tester: request timeout.',
RCV_ERR: 'Failed to request port stats from tester: %(err_msg)s'},
STATE_FLOW_MATCH_CHK:
{FAILURE: 'Received incorrect %(pkt_type)s: %(detail)s',
TIMEOUT: '', # for check no packet-in reason.
RCV_ERR: 'Failed to send packet: %(err_msg)s'},
STATE_NO_PKTIN_REASON:
{FAILURE: 'Receiving timeout: %(detail)s'},
STATE_GET_MATCH_COUNT:
{TIMEOUT: 'Failed to request table stats: request timeout.',
RCV_ERR: 'Failed to request table stats: %(err_msg)s'},
STATE_UNMATCH_PKT_SEND:
{TIMEOUT: 'Faild to send packet: barrier request timeout.',
RCV_ERR: 'Faild to send packet: %(err_msg)s'},
STATE_FLOW_UNMATCH_CHK:
{FAILURE: 'Table-miss error: increment in matched_count.',
ERROR: 'Table-miss error: no change in lookup_count.',
TIMEOUT: 'Failed to request table stats: request timeout.',
RCV_ERR: 'Failed to request table stats: %(err_msg)s'}}
ERR_MSG = 'OFPErrorMsg[type=0x%02x, code=0x%02x]'
class TestMessageBase(RyuException):
def __init__(self, state, message_type, **argv):
msg = MSG[state][message_type] % argv
super(TestMessageBase, self).__init__(msg=msg)
class TestFailure(TestMessageBase):
def __init__(self, state, **argv):
super(TestFailure, self).__init__(state, FAILURE, **argv)
class TestTimeout(TestMessageBase):
def __init__(self, state):
super(TestTimeout, self).__init__(state, TIMEOUT)
class TestReceiveError(TestMessageBase):
def __init__(self, state, err_msg):
argv = {'err_msg': ERR_MSG % (err_msg.type, err_msg.code)}
super(TestReceiveError, self).__init__(state, RCV_ERR, **argv)
class TestError(TestMessageBase):
def __init__(self, state, **argv):
super(TestError, self).__init__(state, ERROR, **argv)
class OfTester(app_manager.RyuApp):
""" OpenFlow Switch Tester. """
OFP_VERSIONS = [ofproto_v1_3.OFP_VERSION]
def __init__(self):
super(OfTester, self).__init__()
self._set_logger()
self.target_dpid = self._convert_dpid(CONF['test-switch']['target'])
self.tester_dpid = self._convert_dpid(CONF['test-switch']['tester'])
self.logger.info('target_dpid=%s',
dpid_lib.dpid_to_str(self.target_dpid))
self.logger.info('tester_dpid=%s',
dpid_lib.dpid_to_str(self.tester_dpid))
test_dir = CONF['test-switch']['dir']
self.logger.info('Test files directory = %s', test_dir)
self.target_sw = None
self.tester_sw = None
self.state = STATE_INIT
self.sw_waiter = None
self.waiter = None
self.send_msg_xids = []
self.rcv_msgs = []
self.test_thread = hub.spawn(
self._test_sequential_execute, test_dir)
def _set_logger(self):
self.logger.propagate = False
s_hdlr = logging.StreamHandler()
self.logger.addHandler(s_hdlr)
if CONF.log_file:
f_hdlr = logging.handlers.WatchedFileHandler(CONF.log_file)
self.logger.addHandler(f_hdlr)
def _convert_dpid(self, dpid_str):
try:
dpid = int(dpid_str, 16)
except ValueError as err:
self.logger.error('Invarid dpid parameter. %s', err)
self._test_end()
return dpid
def close(self):
if self.test_thread is not None:
hub.kill(self.test_thread)
hub.joinall([self.test_thread])
self._test_end('--- Test terminated ---')
@set_ev_cls(ofp_event.EventOFPStateChange,
[handler.MAIN_DISPATCHER, handler.DEAD_DISPATCHER])
def dispacher_change(self, ev):
assert ev.datapath is not None
if ev.state == handler.MAIN_DISPATCHER:
self._register_sw(ev.datapath)
elif ev.state == handler.DEAD_DISPATCHER:
self._unregister_sw(ev.datapath)
def _register_sw(self, dp):
if dp.id == self.target_dpid:
self.target_sw = TargetSw(dp, self.logger)
msg = 'Join target SW.'
elif dp.id == self.tester_dpid:
self.tester_sw = TesterSw(dp, self.logger)
msg = 'Join tester SW.'
else:
msg = 'Connect unknown SW.'
if dp.id:
self.logger.info('dpid=%s : %s',
dpid_lib.dpid_to_str(dp.id), msg)
if self.target_sw and self.tester_sw:
if self.sw_waiter is not None:
self.sw_waiter.set()
def _unregister_sw(self, dp):
if dp.id == self.target_dpid:
del self.target_sw
self.target_sw = None
msg = 'Leave target SW.'
elif dp.id == self.tester_dpid:
del self.tester_sw
self.tester_sw = None
msg = 'Leave tester SW.'
else:
msg = 'Disconnect unknown SW.'
if dp.id:
self.logger.info('dpid=%s : %s',
dpid_lib.dpid_to_str(dp.id), msg)
def _test_sequential_execute(self, test_dir):
""" Execute OpenFlow Switch test. """
# Parse test pattern from test files.
tests = TestPatterns(test_dir, self.logger)
if not tests:
self.logger.warning(NO_TEST_FILE)
self._test_end()
self.logger.info('--- Test start ---')
test_keys = tests.keys()
test_keys.sort()
for file_name in test_keys:
self._test_file_execute(tests[file_name])
self._test_end(msg='--- Test end ---')
def _test_file_execute(self, testfile):
for i, test in enumerate(testfile.tests):
desc = testfile.description if i == 0 else None
self._test_execute(test, desc)
def _test_execute(self, test, description):
if not self.target_sw or not self.tester_sw:
self.logger.info('waiting for switches connection...')
self.sw_waiter = hub.Event()
self.sw_waiter.wait()
self.sw_waiter = None
if description:
self.logger.info('%s', description)
# Test execute.
try:
# 0. Initialize.
self._test(STATE_INIT)
# 1. Install flows.
for flow in test.prerequisite:
self._test(STATE_FLOW_INSTALL, flow)
self._test(STATE_FLOW_EXIST_CHK, flow)
# 2. Check flow matching.
for pkt in test.tests:
if KEY_EGRESS in pkt or KEY_PKT_IN in pkt:
target_pkt_count = [self._test(STATE_TARGET_PKT_COUNT,
True)]
tester_pkt_count = [self._test(STATE_TESTER_PKT_COUNT,
False)]
result = self._test(STATE_FLOW_MATCH_CHK, pkt)
if result == TIMEOUT:
target_pkt_count.append(self._test(
STATE_TARGET_PKT_COUNT, True))
tester_pkt_count.append(self._test(
STATE_TESTER_PKT_COUNT, False))
test_type = (KEY_EGRESS if KEY_EGRESS in pkt
else KEY_PKT_IN)
self._test(STATE_NO_PKTIN_REASON, test_type,
target_pkt_count, tester_pkt_count)
else:
before_stats = self._test(STATE_GET_MATCH_COUNT)
self._test(STATE_UNMATCH_PKT_SEND, pkt)
hub.sleep(INTERVAL)
self._test(STATE_FLOW_UNMATCH_CHK, before_stats, pkt)
result = [TEST_OK]
except (TestFailure, TestError,
TestTimeout, TestReceiveError) as err:
result = [TEST_ERROR, str(err)]
except Exception:
result = [TEST_ERROR, RYU_INTERNAL_ERROR]
# Output test result.
self.logger.info(' %-100s %s', test.description, result[0])
if 1 < len(result):
self.logger.info(' %s', result[1])
if (result[1] == RYU_INTERNAL_ERROR
or result == 'An unknown exception'):
self.logger.error(traceback.format_exc())
if result[0] != TEST_OK and self.state == STATE_INIT:
self._test_end('--- Test terminated ---')
hub.sleep(0)
def _test_end(self, msg=None):
self.test_thread = None
if msg:
self.logger.info(msg)
pid = os.getpid()
os.kill(pid, signal.SIGTERM)
def _test(self, state, *args):
test = {STATE_INIT: self._test_initialize,
STATE_FLOW_INSTALL: self._test_flow_install,
STATE_FLOW_EXIST_CHK: self._test_flow_exist_check,
STATE_TARGET_PKT_COUNT: self._test_get_packet_count,
STATE_TESTER_PKT_COUNT: self._test_get_packet_count,
STATE_FLOW_MATCH_CHK: self._test_flow_matching_check,
STATE_NO_PKTIN_REASON: self._test_no_pktin_reason_check,
STATE_GET_MATCH_COUNT: self._test_get_match_count,
STATE_UNMATCH_PKT_SEND: self._test_unmatch_packet_send,
STATE_FLOW_UNMATCH_CHK: self._test_flow_unmatching_check}
self.send_msg_xids = []
self.rcv_msgs = []
self.state = state
return test[state](*args)
def _test_initialize(self):
xid = self.target_sw.del_test_flow()
self.send_msg_xids.append(xid)
xid = self.target_sw.send_barrier_request()
self.send_msg_xids.append(xid)
self._wait()
assert len(self.rcv_msgs) == 1
msg = self.rcv_msgs[0]
assert isinstance(msg, ofproto_v1_3_parser.OFPBarrierReply)
def _test_flow_install(self, flow):
xid = self.target_sw.add_flow(flow_mod=flow)
self.send_msg_xids.append(xid)
xid = self.target_sw.send_barrier_request()
self.send_msg_xids.append(xid)
self._wait()
assert len(self.rcv_msgs) == 1
msg = self.rcv_msgs[0]
assert isinstance(msg, ofproto_v1_3_parser.OFPBarrierReply)
def _test_flow_exist_check(self, flow_mod):
xid = self.target_sw.send_flow_stats()
self.send_msg_xids.append(xid)
self._wait()
ng_stats = []
for msg in self.rcv_msgs:
assert isinstance(msg, ofproto_v1_3_parser.OFPFlowStatsReply)
for stats in msg.body:
result, stats = self._compare_flow(stats, flow_mod)
if result:
return
else:
ng_stats.append(stats)
raise TestFailure(self.state, flows=', '.join(ng_stats))
def _test_get_packet_count(self, is_target):
sw = self.target_sw if is_target else self.tester_sw
xid = sw.send_port_stats()
self.send_msg_xids.append(xid)
self._wait()
result = {}
for msg in self.rcv_msgs:
for stats in msg.body:
result[stats.port_no] = {'rx': stats.rx_packets,
'tx': stats.tx_packets}
return result
def _test_flow_matching_check(self, pkt):
self.logger.debug("send_packet:[%s]", packet.Packet(pkt[KEY_INGRESS]))
self.logger.debug("egress:[%s]", packet.Packet(pkt.get(KEY_EGRESS)))
self.logger.debug("packet_in:[%s]",
packet.Packet(pkt.get(KEY_PKT_IN)))
# 1. send a packet from the OpenFlow Switch.
xid = self.tester_sw.send_packet_out(pkt[KEY_INGRESS])
self.send_msg_xids.append(xid)
# 2. receive a PacketIn message.
try:
self._wait()
except TestTimeout:
return TIMEOUT
assert len(self.rcv_msgs) == 1
msg = self.rcv_msgs[0]
assert isinstance(msg, ofproto_v1_3_parser.OFPPacketIn)
self.logger.debug("dpid=%s : receive_packet[%s]",
dpid_lib.dpid_to_str(msg.datapath.id),
packet.Packet(msg.data))
# 3. check the SW which sended PacketIn and output packet.
pkt_in_src_model = (self.tester_sw if KEY_EGRESS in pkt
else self.target_sw)
model_pkt = (pkt[KEY_EGRESS] if KEY_EGRESS in pkt
else pkt[KEY_PKT_IN])
if msg.datapath.id != pkt_in_src_model.dp.id:
pkt_type = 'packet-in'
err_msg = 'SW[dpid=%s]' % dpid_lib.dpid_to_str(msg.datapath.id)
elif msg.reason != ofproto_v1_3.OFPR_ACTION:
pkt_type = 'packet-in'
err_msg = 'OFPPacketIn[reason=%d]' % msg.reason
elif repr(msg.data) != repr(model_pkt):
pkt_type = 'packet'
err_msg = self._diff_packets(packet.Packet(model_pkt),
packet.Packet(msg.data))
else:
return TEST_OK
raise TestFailure(self.state, pkt_type=pkt_type,
detail=err_msg)
def _test_no_pktin_reason_check(self, test_type,
target_pkt_count, tester_pkt_count):
before_target_receive = target_pkt_count[0][TARGET_RECEIVE_PORT]['rx']
before_target_send = target_pkt_count[0][TARGET_SENDER_PORT]['tx']
before_tester_receive = tester_pkt_count[0][TESTER_RECEIVE_PORT]['rx']
before_tester_send = tester_pkt_count[0][TESTER_SENDER_PORT]['tx']
after_target_receive = target_pkt_count[1][TARGET_RECEIVE_PORT]['rx']
after_target_send = target_pkt_count[1][TARGET_SENDER_PORT]['tx']
after_tester_receive = tester_pkt_count[1][TESTER_RECEIVE_PORT]['rx']
after_tester_send = tester_pkt_count[1][TESTER_SENDER_PORT]['tx']
if after_tester_send == before_tester_send:
log_msg = 'no change in tx_packets on tester.'
elif after_target_receive == before_target_receive:
log_msg = 'no change in rx_packtes on target.'
elif test_type == KEY_EGRESS:
if after_target_send == before_target_send:
log_msg = 'no change in tx_packets on target.'
elif after_tester_receive == before_tester_receive:
log_msg = 'no change in rx_packets on tester.'
else:
log_msg = 'increment in rx_packets in tester.'
else:
assert test_type == KEY_PKT_IN
log_msg = 'no packet-in.'
raise TestFailure(self.state, detail=log_msg)
def _test_get_match_count(self):
xid = self.target_sw.send_table_stats()
self.send_msg_xids.append(xid)
self._wait()
result = {}
for msg in self.rcv_msgs:
for stats in msg.body:
result[stats.table_id] = {'lookup': stats.lookup_count,
'matched': stats.matched_count}
return result
def _test_unmatch_packet_send(self, pkt):
# Send a packet from the OpenFlow Switch.
self.logger.debug("send_packet:[%s]", packet.Packet(pkt[KEY_INGRESS]))
self.tester_sw.send_packet_out(pkt[KEY_INGRESS])
# Wait OFPBarrierReply.
xid = self.tester_sw.send_barrier_request()
self.send_msg_xids.append(xid)
self._wait()
assert len(self.rcv_msgs) == 1
msg = self.rcv_msgs[0]
assert isinstance(msg, ofproto_v1_3_parser.OFPBarrierReply)
def _test_flow_unmatching_check(self, before_stats, pkt):
# Check matched packet count.
rcv_msgs = self._test_get_match_count()
lookup = False
for target_tbl_id in pkt[KEY_TBL_MISS]:
before = before_stats[target_tbl_id]
after = rcv_msgs[target_tbl_id]
if before['lookup'] < after['lookup']:
lookup = True
if before['matched'] < after['matched']:
raise TestFailure(self.state)
if not lookup:
raise TestError(self.state)
def _compare_flow(self, stats1, stats2):
attr_list = ['cookie', 'priority', 'hard_timeout', 'idle_timeout',
'table_id', 'instructions', 'match']
for attr in attr_list:
value1 = getattr(stats1, attr)
value2 = getattr(stats2, attr)
if str(value1) != str(value2):
flow_stats = []
for attr in attr_list:
flow_stats.append('%s=%s' % (attr, getattr(stats1, attr)))
return False, 'flow_stats(%s)' % ','.join(flow_stats)
return True, None
def _diff_packets(self, model_pkt, rcv_pkt):
msg = []
for rcv_p in rcv_pkt.protocols:
if type(rcv_p) != str:
model_protocols = model_pkt.get_protocols(type(rcv_p))
if len(model_protocols) == 1:
model_p = model_protocols[0]
diff = []
for attr in rcv_p.__dict__:
if attr.startswith('_'):
continue
if callable(attr):
continue
if hasattr(rcv_p.__class__, attr):
continue
rcv_attr = repr(getattr(rcv_p, attr))
model_attr = repr(getattr(model_p, attr))
if rcv_attr != model_attr:
diff.append('%s=%s' % (attr, rcv_attr))
if diff:
msg.append('%s(%s)' %
(rcv_p.__class__.__name__,
','.join(diff)))
else:
if (not model_protocols or
not str(rcv_p) in str(model_protocols)):
msg.append(str(rcv_p))
else:
model_p = ''
for p in model_pkt.protocols:
if type(p) == str:
model_p = p
break
if model_p != rcv_p:
msg.append('str(%s)' % repr(rcv_p))
if msg:
return '/'.join(msg)
else:
raise RyuException('Internal error.'
' receive packet is matching.')
def _wait(self):
""" Wait until specific OFP message received
or timer is exceeded. """
assert self.waiter is None
self.waiter = hub.Event()
self.rcv_msgs = []
timeout = False
timer = hub.Timeout(WAIT_TIMER)
try:
self.waiter.wait()
except hub.Timeout as t:
if t is not timer:
raise RyuException('Internal error. Not my timeout.')
timeout = True
finally:
timer.cancel()
self.waiter = None
if timeout:
raise TestTimeout(self.state)
if (self.rcv_msgs and isinstance(
self.rcv_msgs[0], ofproto_v1_3_parser.OFPErrorMsg)):
raise TestReceiveError(self.state, self.rcv_msgs[0])
@set_ev_cls(ofp_event.EventOFPFlowStatsReply, handler.MAIN_DISPATCHER)
def flow_stats_reply_handler(self, ev):
state_list = [STATE_FLOW_EXIST_CHK]
if self.state in state_list:
if self.waiter and ev.msg.xid in self.send_msg_xids:
self.rcv_msgs.append(ev.msg)
if not ev.msg.flags & ofproto_v1_3.OFPMPF_REPLY_MORE:
self.waiter.set()
hub.sleep(0)
@set_ev_cls(ofp_event.EventOFPTableStatsReply, handler.MAIN_DISPATCHER)
def table_stats_reply_handler(self, ev):
state_list = [STATE_GET_MATCH_COUNT,
STATE_FLOW_UNMATCH_CHK]
if self.state in state_list:
if self.waiter and ev.msg.xid in self.send_msg_xids:
self.rcv_msgs.append(ev.msg)
if not ev.msg.flags & ofproto_v1_3.OFPMPF_REPLY_MORE:
self.waiter.set()
hub.sleep(0)
@set_ev_cls(ofp_event.EventOFPPortStatsReply, handler.MAIN_DISPATCHER)
def port_stats_reply_handler(self, ev):
state_list = [STATE_TARGET_PKT_COUNT,
STATE_TESTER_PKT_COUNT]
if self.state in state_list:
if self.waiter and ev.msg.xid in self.send_msg_xids:
self.rcv_msgs.append(ev.msg)
if not ev.msg.flags & ofproto_v1_3.OFPMPF_REPLY_MORE:
self.waiter.set()
hub.sleep(0)
@set_ev_cls(ofp_event.EventOFPBarrierReply, handler.MAIN_DISPATCHER)
def barrier_reply_handler(self, ev):
state_list = [STATE_INIT,
STATE_FLOW_INSTALL,
STATE_UNMATCH_PKT_SEND]
if self.state in state_list:
if self.waiter and ev.msg.xid in self.send_msg_xids:
self.rcv_msgs.append(ev.msg)
self.waiter.set()
hub.sleep(0)
@set_ev_cls(ofp_event.EventOFPPacketIn, handler.MAIN_DISPATCHER)
def packet_in_handler(self, ev):
state_list = [STATE_FLOW_MATCH_CHK]
if self.state in state_list:
if self.waiter:
self.rcv_msgs.append(ev.msg)
self.waiter.set()
hub.sleep(0)
@set_ev_cls(ofp_event.EventOFPErrorMsg, [handler.HANDSHAKE_DISPATCHER,
handler.CONFIG_DISPATCHER,
handler.MAIN_DISPATCHER])
def error_msg_handler(self, ev):
if ev.msg.xid in self.send_msg_xids:
self.rcv_msgs.append(ev.msg)
if self.waiter:
self.waiter.set()
hub.sleep(0)
class OpenFlowSw(object):
def __init__(self, dp, logger):
super(OpenFlowSw, self).__init__()
self.dp = dp
self.logger = logger
def _send_msg(self, msg):
msg.xid = None
self.dp.set_xid(msg)
self.dp.send_msg(msg)
return msg.xid
def add_flow(self, flow_mod=None, in_port=None, out_port=None):
""" Add flow. """
ofp = self.dp.ofproto
parser = self.dp.ofproto_parser
if flow_mod:
mod = flow_mod
else:
match = parser.OFPMatch(in_port=in_port)
max_len = (0 if out_port != ofp.OFPP_CONTROLLER
else ofp.OFPCML_MAX)
actions = [parser.OFPActionOutput(out_port, max_len)]
inst = [parser.OFPInstructionActions(ofp.OFPIT_APPLY_ACTIONS,
actions)]
mod = parser.OFPFlowMod(self.dp, cookie=0,
command=ofp.OFPFC_ADD,
match=match, instructions=inst)
return self._send_msg(mod)
def send_barrier_request(self):
""" send a BARRIER_REQUEST message."""
parser = self.dp.ofproto_parser
req = parser.OFPBarrierRequest(self.dp)
return self._send_msg(req)
def send_port_stats(self):
""" Get port stats."""
ofp = self.dp.ofproto
parser = self.dp.ofproto_parser
flags = 0
req = parser.OFPPortStatsRequest(self.dp, flags, ofp.OFPP_ANY)
return self._send_msg(req)
class TargetSw(OpenFlowSw):
def __init__(self, dp, logger):
super(TargetSw, self).__init__(dp, logger)
def del_test_flow(self):
""" Delete all flow except default flow. """
ofp = self.dp.ofproto
parser = self.dp.ofproto_parser
mod = parser.OFPFlowMod(self.dp,
table_id=ofp.OFPTT_ALL,
command=ofp.OFPFC_DELETE,
out_port=ofp.OFPP_ANY,
out_group=ofp.OFPG_ANY)
self.dp.send_msg(mod)
def send_flow_stats(self):
""" Get all flow. """
ofp = self.dp.ofproto
parser = self.dp.ofproto_parser
req = parser.OFPFlowStatsRequest(self.dp, 0, ofp.OFPTT_ALL,
ofp.OFPP_ANY, ofp.OFPG_ANY,
0, 0, parser.OFPMatch())
return self._send_msg(req)
def send_table_stats(self):
""" Get table stats. """
parser = self.dp.ofproto_parser
req = parser.OFPTableStatsRequest(self.dp, 0)
return self._send_msg(req)
class TesterSw(OpenFlowSw):
def __init__(self, dp, logger):
super(TesterSw, self).__init__(dp, logger)
# Add packet in flow.
ofp = self.dp.ofproto
self.add_flow(in_port=TESTER_RECEIVE_PORT,
out_port=ofp.OFPP_CONTROLLER)
def send_packet_out(self, data):
""" send a PacketOut message."""
ofp = self.dp.ofproto
parser = self.dp.ofproto_parser
actions = [parser.OFPActionOutput(TESTER_SENDER_PORT)]
out = parser.OFPPacketOut(
datapath=self.dp, buffer_id=ofp.OFP_NO_BUFFER,
data=data, in_port=ofp.OFPP_CONTROLLER, actions=actions)
return self._send_msg(out)
class TestPatterns(dict):
""" List of Test class objects. """
def __init__(self, test_dir, logger):
super(TestPatterns, self).__init__()
self.logger = logger
# Parse test pattern from test files.
self._get_tests(test_dir)
def _get_tests(self, path):
if not os.path.exists(path):
msg = INVALID_PATH % {'path': path}
self.logger.warning(msg)
return
if os.path.isdir(path): # Directory
for test_path in os.listdir(path):
test_path = path + (test_path if path[-1:] == '/'
else '/%s' % test_path)
self._get_tests(test_path)
elif os.path.isfile(path): # File
(dummy, ext) = os.path.splitext(path)
if ext == '.json':
test = TestFile(path, self.logger)
self[test.description] = test
class TestFile(stringify.StringifyMixin):
"""Test File object include Test objects."""
def __init__(self, path, logger):
super(TestFile, self).__init__()
self.logger = logger
self.description = None
self.tests = []
self._get_tests(path)
def _get_tests(self, path):
with open(path, 'rb') as fhandle:
buf = fhandle.read()
try:
json_list = json.loads(buf)
for test_json in json_list:
if isinstance(test_json, unicode):
self.description = test_json
else:
self.tests.append(Test(test_json))
except (ValueError, TypeError) as e:
result = (TEST_FILE_ERROR %
{'file': path, 'detail': e.message})
self.logger.warning(result)
class Test(stringify.StringifyMixin):
def __init__(self, test_json):
super(Test, self).__init__()
(self.description,
self.prerequisite,
self.tests) = self._parse_test(test_json)
def _parse_test(self, buf):
def __test_pkt_from_json(test):
data = eval('/'.join(test))
data.serialize()
return str(data.data)
# parse 'description'
description = buf.get(KEY_DESC)
# parse 'prerequisite'
prerequisite = []
if not KEY_PREREQ in buf:
raise ValueError('a test requires a "%s" block' % KEY_PREREQ)
for flow in buf[KEY_PREREQ]:
cls = getattr(ofproto_v1_3_parser, KEY_FLOW)
msg = cls.from_jsondict(flow[KEY_FLOW], datapath=DummyDatapath())
msg.version = ofproto_v1_3.OFP_VERSION
msg.msg_type = msg.cls_msg_type
msg.xid = 0
prerequisite.append(msg)
# parse 'tests'
tests = []
if not KEY_TESTS in buf:
raise ValueError('a test requires a "%s" block.' % KEY_TESTS)
for test in buf[KEY_TESTS]:
if len(test) != 2:
raise ValueError(
'"%s" block requires "%s" field and one of "%s" or "%s"'
' or "%s" field.' % (KEY_TESTS, KEY_INGRESS, KEY_EGRESS,
KEY_PKT_IN, KEY_TBL_MISS))
test_pkt = {}
# parse 'ingress'
if not KEY_INGRESS in test:
raise ValueError('a test requires "%s" field.' % KEY_INGRESS)
test_pkt[KEY_INGRESS] = __test_pkt_from_json(test[KEY_INGRESS])
if len(test_pkt[KEY_INGRESS]) < 64:
print 'NG!!!'
# parse 'egress' or 'PACKET_IN' or 'table-miss'
if KEY_EGRESS in test:
test_pkt[KEY_EGRESS] = __test_pkt_from_json(test[KEY_EGRESS])
elif KEY_PKT_IN in test:
test_pkt[KEY_PKT_IN] = __test_pkt_from_json(test[KEY_PKT_IN])
elif KEY_TBL_MISS in test:
test_pkt[KEY_TBL_MISS] = test[KEY_TBL_MISS]
tests.append(test_pkt)
return (description, prerequisite, tests)
class DummyDatapath(object):
def __init__(self):
self.ofproto = ofproto_v1_3
self.ofproto_parser = ofproto_v1_3_parser