From: Isaku Yamahata <yamah...@valinux.co.jp>

Signed-off-by: Isaku Yamahata <yamah...@valinux.co.jp>
Signed-off-by: YAMAMOTO Takashi <yamam...@valinux.co.jp>
---
 ryu/tests/integrated/test_vrrp_linux_multi.py | 104 +++++++++++++
 ryu/tests/integrated/test_vrrp_multi.py       | 144 +++++++++++++++++
 ryu/tests/integrated/vrrp_common.py           | 216 ++++++++++++++++++++++++++
 3 files changed, 464 insertions(+)
 create mode 100644 ryu/tests/integrated/test_vrrp_linux_multi.py
 create mode 100644 ryu/tests/integrated/test_vrrp_multi.py
 create mode 100644 ryu/tests/integrated/vrrp_common.py

diff --git a/ryu/tests/integrated/test_vrrp_linux_multi.py 
b/ryu/tests/integrated/test_vrrp_linux_multi.py
new file mode 100644
index 0000000..a80323c
--- /dev/null
+++ b/ryu/tests/integrated/test_vrrp_linux_multi.py
@@ -0,0 +1,104 @@
+# Copyright (C) 2013 Nippon Telegraph and Telephone Corporation.
+# Copyright (C) 2013 Isaku Yamahata <yamahata at valinux co jp>
+#
+# 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.
+
+"""
+Usage:
+PYTHONPATH=. ./bin/ryu-manager --verbose \
+             ryu.services.protocols.vrrp.manager \
+             ryu.services.protocols.vrrp.dumper \
+             ryu.services.protocols.vrrp.sample_manager.py \
+             ryu.tests.integrated.test_vrrp_linux_multi \
+             ryu.app.rest
+
+ryu.services.protocols.vrrp.dumper is optional.
+ryu.app.rest is merely to prevent ryu-manager from exiting.
+
+                    ----------------
+      /--<--veth0-->|              |
+   Ryu              | linux bridge |<--veth2--> command to generate packets
+      \--<--veth1-->|   (vrrpbr)   |
+                    ----------------
+
+
+# ip link add veth0 type veth peer name veth0-br
+# ip link add veth1 type veth peer name veth1-br
+# ip link add veth2 type veth peer name veth2-br
+
+# brctl addbr vrrpbr
+# brctl addif vrrpbr veth0-br
+# brctl addif vrrpbr veth1-br
+# brctl addif vrrpbr veth2-br
+
+
+# ip link set veth0 up
+# ip link set veth0-br up
+# ip link set veth1 up
+# ip link set veth1-br up
+# ip link set veth2 up
+# ip link set veth2-br up
+# ip link set vrrpbr up
+
+if you like, capture packets on each interfaces like
+# tshark -i vrrpbr
+# tshark -i veth0
+# tshark -i veth1
+# tshark -i veth2
+
+virtual router mac address: 00:00:5E:00:01:{VRID} = 00:00:5E:00:01:07
+during working, send packets destined to mac address 00:00:5E:00:01:07
+from veth2 by packet generator like packeth
+
+NOTE: vrid: 7 and ip address: 10.0.0.1... are hardcoded below
+"""
+
+from ryu.base import app_manager
+from ryu.lib import hub
+from ryu.lib import mac as lib_mac
+from ryu.lib.packet import vrrp
+from ryu.services.protocols.vrrp import api as vrrp_api
+from ryu.services.protocols.vrrp import event as vrrp_event
+from ryu.services.protocols.vrrp import monitor_linux
+
+from . import vrrp_common
+
+
+class VRRPConfigApp(vrrp_common.VRRPCommon):
+    _IFNAME0 = 'veth0'
+    _IFNAME1 = 'veth1'
+
+    def __init__(self, *args, **kwargs):
+        super(VRRPConfigApp, self).__init__(*args, **kwargs)
+
+    def start(self):
+        hub.spawn(self._main)
+
+    def _configure_vrrp_router(self, vrrp_version, priority,
+                               primary_ip_address, ifname, vrid):
+        interface = vrrp_event.VRRPInterfaceNetworkDevice(
+            lib_mac.DONTCARE_STR, primary_ip_address, None, ifname)
+        self.logger.debug('%s', interface)
+
+        vip = '10.0.%d.1' % vrid
+        ip_addresses = [vip]
+        config = vrrp_event.VRRPConfig(
+            version=vrrp_version, vrid=vrid, priority=priority,
+            ip_addresses=ip_addresses)
+        self.logger.debug('%s', config)
+
+        rep = vrrp_api.vrrp_config(self, interface, config)
+        self.logger.debug('%s', rep)
+
+        return rep
diff --git a/ryu/tests/integrated/test_vrrp_multi.py 
b/ryu/tests/integrated/test_vrrp_multi.py
new file mode 100644
index 0000000..b1f2431
--- /dev/null
+++ b/ryu/tests/integrated/test_vrrp_multi.py
@@ -0,0 +1,144 @@
+# Copyright (C) 2013 Nippon Telegraph and Telephone Corporation.
+# Copyright (C) 2013 Isaku Yamahata <yamahata at valinux co jp>
+#
+# 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.
+
+"""
+Usage:
+PYTHONPATH=. ./bin/ryu-manager --verbose \
+             ryu.topology.switches \
+             ryu.services.protocols.vrrp.manager \
+             ryu.tests.integrated.test_vrrp_multi \
+             ryu.services.protocols.vrrp.dumper
+
+ryu.services.protocols.vrrp.dumper is optional.
+
+         +---+          ----------------
+      /--|OVS|<--veth-->|              |
+   Ryu   +---+          | linux bridge |<--veth--> command to generate packets
+      \--|OVS|<--veth-->|              |
+         +---+          ----------------
+
+configure OVSs to connect ryu
+example
+# brctl addbr b0
+# ip link add veth0-ovs type veth peer name veth0-br
+# ip link add veth1-ovs type veth peer name veth1-br
+# brctl addif b0 veth0-br
+# brctl addif b0 veth1-br
+# brctl show
+bridge name     bridge id               STP enabled     interfaces
+b0              8000.6642e5822497       no              veth0-br
+                                                        veth1-br
+ovs-system              0000.122038293b55       no
+
+# ovs-vsctl add-br s0
+# ovs-vsctl add-port s0 veth0-ovs
+# ovs-vsctl add-br s1
+# ovs-vsctl add-port s1 veth1-ovs
+# ovs-vsctl set-controller s0 tcp:127.0.0.1:6633
+# ovs-vsctl set bridge s0 protocols='[OpenFlow12]'
+# ovs-vsctl set-controller s1 tcp:127.0.0.1:6633
+# ovs-vsctl set bridge s1 protocols='[OpenFlow12]'
+# ovs-vsctl show
+20c2a046-ae7e-4453-a576-11034db24985
+    Manager "ptcp:6634"
+    Bridge "s0"
+        Controller "tcp:127.0.0.1:6633"
+            is_connected: true
+        Port "veth0-ovs"
+            Interface "veth0-ovs"
+        Port "s0"
+            Interface "s0"
+                type: internal
+    Bridge "s1"
+        Controller "tcp:127.0.0.1:6633"
+            is_connected: true
+        Port "veth1-ovs"
+            Interface "veth1-ovs"
+        Port "s1"
+            Interface "s1"
+                type: internal
+    ovs_version: "1.9.90"
+# ip link veth0-br set up
+# ip link veth0-ovs set up
+# ip link veth1-br set up
+# ip link veth1-ovs set up
+# ip link b0 set up
+"""
+
+from ryu.base import app_manager
+from ryu.controller import handler
+from ryu.lib import dpid as lib_dpid
+from ryu.lib import hub
+from ryu.lib.packet import vrrp
+from ryu.services.protocols.vrrp import api as vrrp_api
+from ryu.services.protocols.vrrp import event as vrrp_event
+from ryu.services.protocols.vrrp import monitor_openflow
+from ryu.topology import event as topo_event
+from ryu.topology import api as topo_api
+
+from . import vrrp_common
+
+
+class VRRPConfigApp(vrrp_common.VRRPCommon):
+    _IFNAME0 = 0
+    _IFNAME1 = 1
+
+    def __init__(self, *args, **kwargs):
+        super(VRRPConfigApp, self).__init__(*args, **kwargs)
+        self.start_main = False
+
+    @handler.set_ev_cls(topo_event.EventSwitchEnter)
+    def _switch_enter_handler(self, ev):
+        if self.start_main:
+            return
+
+        switches = topo_api.get_switch(self)
+        if len(switches) < 2:
+            return
+
+        self.start_main = True
+        app_mgr = app_manager.AppManager.get_instance()
+        self.logger.debug('%s', app_mgr.applications)
+        self.switches = app_mgr.applications['switches']
+        hub.spawn(self._main)
+
+    def _configure_vrrp_router(self, vrrp_version, priority,
+                               ip_addr, switch_index, vrid):
+        switches = self.switches
+        self.logger.debug('%s', switches.dps)
+        dpid = sorted(switches.dps.keys())[switch_index]
+        self.logger.debug('%s', lib_dpid.dpid_to_str(dpid))
+        self.logger.debug('%s', switches.port_state)
+        # hack: use the smallest port no to avoid picking OVS local port
+        port_no = sorted(switches.port_state[dpid].keys())[0]
+        self.logger.debug('%d', port_no)
+        port = switches.port_state[dpid][port_no]
+        self.logger.debug('%s', port)
+        mac = port.hw_addr
+        self.logger.debug('%s', mac)
+
+        interface = vrrp_event.VRRPInterfaceOpenFlow(
+            mac, ip_addr, None, dpid, port_no)
+        self.logger.debug('%s', interface)
+
+        config = vrrp_event.VRRPConfig(
+            version=vrrp_version, vrid=vrid, priority=priority,
+            ip_addresses=[ip_addr])
+        self.logger.debug('%s', config)
+
+        rep = vrrp_api.vrrp_config(self, interface, config)
+        self.logger.debug('%s', rep)
+        return rep
diff --git a/ryu/tests/integrated/vrrp_common.py 
b/ryu/tests/integrated/vrrp_common.py
new file mode 100644
index 0000000..6ab48e6
--- /dev/null
+++ b/ryu/tests/integrated/vrrp_common.py
@@ -0,0 +1,216 @@
+# Copyright (C) 2013 Nippon Telegraph and Telephone Corporation.
+# Copyright (C) 2013 Isaku Yamahata <yamahata at valinux co jp>
+#
+# 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 time
+import random
+
+from ryu.base import app_manager
+from ryu.lib import hub
+from ryu.lib import mac as lib_mac
+from ryu.lib.packet import vrrp
+from ryu.services.protocols.vrrp import api as vrrp_api
+from ryu.services.protocols.vrrp import event as vrrp_event
+
+
+_VRID = 7
+_PRIMARY_IP_ADDRESS0 = '10.0.0.2'
+_PRIMARY_IP_ADDRESS1 = '10.0.0.3'
+
+
+class VRRPCommon(app_manager.RyuApp):
+    _IFNAME0 = None
+    _IFNAME1 = None
+
+    def __init__(self, *args, **kwargs):
+        super(VRRPCommon, self).__init__(*args, **kwargs)
+
+    def _main(self):
+        self._main_version(vrrp.VRRP_VERSION_V3)
+        self._main_version(vrrp.VRRP_VERSION_V2)
+        print "done!"
+
+    def _main_version(self, vrrp_version):
+        self._main_version_priority(vrrp_version,
+                                    vrrp.VRRP_PRIORITY_ADDRESS_OWNER)
+        self._main_version_priority(vrrp_version,
+                                    vrrp.VRRP_PRIORITY_BACKUP_MAX)
+        self._main_version_priority(vrrp_version,
+                                    vrrp.VRRP_PRIORITY_BACKUP_DEFAULT)
+        self._main_version_priority(vrrp_version,
+                                    vrrp.VRRP_PRIORITY_BACKUP_MIN)
+
+    def _main_version_priority(self, vrrp_version, priority):
+        self._main_version_priority_sleep(vrrp_version, priority, False)
+        self._main_version_priority_sleep(vrrp_version, priority, True)
+
+    def _check(self, vrrp_api, instances):
+        while True:
+            while True:
+                rep = vrrp_api.vrrp_list(self)
+                if len(rep.instance_list) >= len(instances) * 2:
+                    if any(i.state == vrrp_event.VRRP_STATE_INITIALIZE
+                           for i in rep.instance_list):
+                        continue
+                    break
+                print len(rep.instance_list), '/', len(instances) * 2
+                time.sleep(1)
+
+#                for i in rep.instance_list:
+#                    print i.instance_name, i.monitor_name, i.config, \
+#                          i.interface, i.state
+            assert len(rep.instance_list) == len(instances) * 2
+            num_of_master = 0
+            d = dict(((i.instance_name, i) for i in rep.instance_list))
+            bad = 0
+            for i in rep.instance_list:
+                assert i.state in (vrrp_event.VRRP_STATE_MASTER,
+                                   vrrp_event.VRRP_STATE_BACKUP)
+                if i.state == vrrp_event.VRRP_STATE_MASTER:
+                    num_of_master += 1
+
+                vr = instances[i.config.vrid]
+                if (vr[0].config.priority > vr[1].config.priority and
+                        i.instance_name == vr[1].instance_name) or \
+                   (vr[0].config.priority < vr[1].config.priority and
+                        i.instance_name == vr[0].instance_name):
+                    if i.state == vrrp_event.VRRP_STATE_MASTER:
+                        print "bad master:"
+                        print d[vr[0].instance_name].state, \
+                            d[vr[0].instance_name].config.priority
+                        print d[vr[1].instance_name].state, \
+                            d[vr[1].instance_name].config.priority
+                        bad += 1
+#                       assert i.state != vrrp_event.VRRP_STATE_MASTER
+            if bad > 0:
+                # this could be a transient state
+                print bad, "bad masters"
+                time.sleep(1)
+                continue
+            if num_of_master >= len(instances):
+                assert num_of_master == len(instances)
+                break
+            print num_of_master, '/', len(instances)
+            time.sleep(1)
+            continue
+
+    def _main_version_priority_sleep(self, vrrp_version, priority, do_sleep):
+        app_mgr = app_manager.AppManager.get_instance()
+        self.logger.debug('%s', app_mgr.applications)
+        vrrp_mgr = app_mgr.applications['VRRPManager']
+
+        step = 5
+        instances = {}
+        for vrid in xrange(1, 256, step):
+            if vrid == _VRID:
+                continue
+            print "vrid", vrid
+            l = {}
+            prio = max(vrrp.VRRP_PRIORITY_BACKUP_MIN,
+                       min(vrrp.VRRP_PRIORITY_BACKUP_MAX, vrid))
+            rep0 = self._configure_vrrp_router(vrrp_version,
+                                               prio,
+                                               _PRIMARY_IP_ADDRESS0,
+                                               self._IFNAME0,
+                                               vrid)
+            assert not rep0.instance_name is None
+            l[0] = rep0
+            prio = max(vrrp.VRRP_PRIORITY_BACKUP_MIN,
+                       min(vrrp.VRRP_PRIORITY_BACKUP_MAX, 256 - vrid))
+            rep1 = self._configure_vrrp_router(vrrp_version,
+                                               prio,
+                                               _PRIMARY_IP_ADDRESS1,
+                                               self._IFNAME1,
+                                               vrid)
+            assert not rep1.instance_name is None
+            l[1] = rep1
+            instances[vrid] = l
+
+        print "vrid", _VRID
+        l = {}
+        rep0 = self._configure_vrrp_router(vrrp_version, priority,
+                                           _PRIMARY_IP_ADDRESS0,
+                                           self._IFNAME0, _VRID)
+        assert not rep0.instance_name is None
+        l[0] = rep0
+        rep1 = self._configure_vrrp_router(
+            vrrp_version, vrrp.VRRP_PRIORITY_BACKUP_DEFAULT,
+            _PRIMARY_IP_ADDRESS1, self._IFNAME1, _VRID)
+        assert not rep1.instance_name is None
+        l[1] = rep1
+        instances[_VRID] = l
+
+        self.logger.debug('%s', vrrp_mgr._instances)
+
+        if do_sleep:
+            print "priority", priority
+            print "waiting for instances starting"
+
+            self._check(vrrp_api, instances)
+
+        for vrid in instances.keys():
+            if vrid == _VRID:
+                continue
+            which = vrid & 1
+            new_priority = int(random.uniform(vrrp.VRRP_PRIORITY_BACKUP_MIN,
+                                              vrrp.VRRP_PRIORITY_BACKUP_MAX))
+            i = instances[vrid][which]
+            vrrp_api.vrrp_config_change(self, i.instance_name,
+                                        priority=new_priority)
+            i.config.priority = new_priority
+
+        if do_sleep:
+            print "priority shuffled"
+
+            self._check(vrrp_api, instances)
+
+        for vrid in instances.keys():
+            if vrid == _VRID:
+                continue
+            which = vrid & 1
+            vrrp_api.vrrp_shutdown(self, instances[vrid][which].instance_name)
+        vrrp_api.vrrp_shutdown(self, instances[_VRID][0].instance_name)
+
+        if do_sleep:
+            print "shutting down instances"
+            while True:
+                rep = vrrp_api.vrrp_list(self)
+                if len(rep.instance_list) <= len(instances):
+                    break
+                print "left", len(rep.instance_list)
+                time.sleep(1)
+            assert len(rep.instance_list) == len(instances)
+            print "waiting for the rest becoming master"
+            while True:
+                rep = vrrp_api.vrrp_list(self)
+                if all(i.state == vrrp_event.VRRP_STATE_MASTER
+                       for i in rep.instance_list):
+                    break
+                time.sleep(1)
+
+        vrrp_api.vrrp_shutdown(self, instances[_VRID][1].instance_name)
+        for vrid in instances.keys():
+            if vrid == _VRID:
+                continue
+            which = 1 - (vrid & 1)
+            vrrp_api.vrrp_shutdown(self, instances[vrid][which].instance_name)
+
+        print "waiting for instances shutting down"
+        while True:
+            rep = vrrp_api.vrrp_list(self)
+            if not rep.instance_list:
+                break
+            print "left", len(rep.instance_list)
+            time.sleep(1)
-- 
1.8.3.1


------------------------------------------------------------------------------
Shape the Mobile Experience: Free Subscription
Software experts and developers: Be at the forefront of tech innovation.
Intel(R) Software Adrenaline delivers strategic insight and game-changing 
conversations that shape the rapidly evolving mobile landscape. Sign up now. 
http://pubads.g.doubleclick.net/gampad/clk?id=63431311&iu=/4140/ostg.clktrk
_______________________________________________
Ryu-devel mailing list
Ryu-devel@lists.sourceforge.net
https://lists.sourceforge.net/lists/listinfo/ryu-devel

Reply via email to