summaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorYuichi Ito <ito.yuichi0@gmail.com>2013-12-13 17:25:24 +0900
committerFUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>2013-12-13 17:27:49 +0900
commit70542380d3ee828792242bef813b752521e7a61a (patch)
treed30c535ce39ffae15b2b75829a45d3cc007a7432
parentc44f71a7e1a5463c3f78ef43942963e396062a59 (diff)
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>
-rw-r--r--ryu/flags.py8
-rw-r--r--ryu/tests/switch/tester.py898
2 files changed, 906 insertions, 0 deletions
diff --git a/ryu/flags.py b/ryu/flags.py
index 71a5a2ab..66ac1abe 100644
--- a/ryu/flags.py
+++ b/ryu/flags.py
@@ -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')
diff --git a/ryu/tests/switch/tester.py b/ryu/tests/switch/tester.py
new file mode 100644
index 00000000..519a7633
--- /dev/null
+++ b/ryu/tests/switch/tester.py
@@ -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