of_config: sample test code to use ofconfig capable_switch
Signed-off-by: Isaku Yamahata <yamahata@valinux.co.jp> Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
This commit is contained in:
Isaku Yamahata
FUJITA Tomonori
parent
e2566bbd70
commit
3cfb6fa6e9
229
ryu/tests/integrated/test_of_config.py
Normal file
229
ryu/tests/integrated/test_of_config.py
Normal file
@ -0,0 +1,229 @@
|
||||
# Copyright (C) 2013 Nippon Telegraph and Telephone Corporation.
|
||||
# Copyright (C) 2013 Isaku Yamahata <yamahata at private email ne 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 gevent
|
||||
import traceback
|
||||
|
||||
import lxml.etree
|
||||
import ncclient
|
||||
|
||||
from ryu.base import app_manager
|
||||
from ryu.lib.netconf import constants as nc_consts
|
||||
from ryu.lib import of_config
|
||||
from ryu.lib.of_config import capable_switch
|
||||
from ryu.lib.of_config import constants as ofc_consts
|
||||
|
||||
|
||||
# Change those depending on switch configuration
|
||||
HOST = '127.0.0.1'
|
||||
PORT = 1830
|
||||
USERNAME = 'linc'
|
||||
PASSWORD = 'linc'
|
||||
|
||||
CAPABLE_SWITCH_ID = 'CapableSwitch0'
|
||||
LOGICAL_SWITCH = 'LogicalSwitch0'
|
||||
PORT_ID = 'LogicalSwitch0-Port2'
|
||||
CONTROLLER_ID = 'Switch0-DefaultController'
|
||||
|
||||
PORT_DICT = {
|
||||
'capable_switch': CAPABLE_SWITCH_ID,
|
||||
'port_id': PORT_ID,
|
||||
'logical_switch': LOGICAL_SWITCH,
|
||||
'controller_id': CONTROLLER_ID,
|
||||
'ip': HOST,
|
||||
}
|
||||
|
||||
SWITCH_PORT_DOWN = '''
|
||||
<nc:config xmlns:nc="urn:ietf:params:xml:ns:netconf:base:1.0">
|
||||
<capable-switch xmlns="urn:onf:of111:config:yang">
|
||||
<id>%(capable_switch)s</id>
|
||||
<resources>
|
||||
<port>
|
||||
<resource-id>%(port_id)s</resource-id>
|
||||
<configuration operation="merge">
|
||||
<admin-state>down</admin-state>
|
||||
<no-receive>false</no-receive>
|
||||
<no-forward>false</no-forward>
|
||||
<no-packet-in>false</no-packet-in>
|
||||
</configuration>
|
||||
</port>
|
||||
</resources>
|
||||
</capable-switch>
|
||||
</nc:config>
|
||||
''' % PORT_DICT
|
||||
|
||||
SWITCH_ADVERTISED = '''
|
||||
<nc:config xmlns:nc="urn:ietf:params:xml:ns:netconf:base:1.0">
|
||||
<capable-switch xmlns="urn:onf:of111:config:yang">
|
||||
<id>%(capable_switch)s</id>
|
||||
<resources>
|
||||
<port>
|
||||
<resource-id>%(port_id)s</resource-id>
|
||||
<features>
|
||||
<advertised operation="merge">
|
||||
<rate>10Mb-FD</rate>
|
||||
<auto-negotiate>true</auto-negotiate>
|
||||
<medium>copper</medium>
|
||||
<pause>unsupported</pause>
|
||||
</advertised>
|
||||
</features>
|
||||
</port>
|
||||
</resources>
|
||||
</capable-switch>
|
||||
</nc:config>
|
||||
''' % PORT_DICT
|
||||
|
||||
SWITCH_CONTROLLER = '''
|
||||
<nc:config xmlns:nc="urn:ietf:params:xml:ns:netconf:base:1.0">
|
||||
<capable-switch xmlns="urn:onf:of111:config:yang">
|
||||
<id>%(capable_switch)s</id>
|
||||
<logical-switches>
|
||||
<switch>
|
||||
<id>%(logical_switch)s</id>
|
||||
<controllers>
|
||||
<controller operation="merge">
|
||||
<id>%(controller_id)s</id>
|
||||
<role>master</role>
|
||||
<ip-address>%(ip)s</ip-address>
|
||||
<port>6633</port>
|
||||
<protocol>tcp</protocol>
|
||||
</controller>
|
||||
</controllers>
|
||||
</switch>
|
||||
</logical-switches>
|
||||
</capable-switch>
|
||||
</nc:config>
|
||||
''' % PORT_DICT
|
||||
|
||||
|
||||
def _get_schema():
|
||||
# file_name = of_config.OF_CONFIG_1_0_XSD
|
||||
# file_name = of_config.OF_CONFIG_1_1_XSD
|
||||
file_name = of_config.OF_CONFIG_1_1_1_XSD
|
||||
return lxml.etree.XMLSchema(file=file_name)
|
||||
|
||||
|
||||
class OFConfigClient(app_manager.RyuApp):
|
||||
def __init__(self, *args, **kwargs):
|
||||
super(OFConfigClient, self).__init__(*args, **kwargs)
|
||||
self.switch = capable_switch.OFCapableSwitch(
|
||||
host=HOST, port=PORT, username=USERNAME, password=PASSWORD)
|
||||
gevent.spawn(self._do_of_config)
|
||||
|
||||
def _validate(self, tree):
|
||||
xmlschema = _get_schema()
|
||||
try:
|
||||
xmlschema.assertValid(tree)
|
||||
except:
|
||||
traceback.print_exc()
|
||||
|
||||
def _do_get(self):
|
||||
data_xml = self.switch.get()
|
||||
|
||||
tree = lxml.etree.fromstring(data_xml)
|
||||
# print(lxml.etree.tostring(tree, pretty_print=True))
|
||||
self._validate(tree)
|
||||
|
||||
name_spaces = set()
|
||||
for e in tree.getiterator():
|
||||
name_spaces.add(capable_switch.get_ns_tag(e.tag)[0])
|
||||
print(name_spaces)
|
||||
|
||||
return tree
|
||||
|
||||
def _do_get_config(self, source):
|
||||
print('source = %s' % source)
|
||||
config_xml = self.switch.get_config(source)
|
||||
|
||||
tree = lxml.etree.fromstring(config_xml)
|
||||
# print(lxml.etree.tostring(tree, pretty_print=True))
|
||||
self._validate(tree)
|
||||
|
||||
def _do_edit_config(self, config):
|
||||
tree = lxml.etree.fromstring(config)
|
||||
self._validate(tree)
|
||||
self.switch.edit_config(target='running', config=config)
|
||||
|
||||
def _print_ports(self, tree, ns):
|
||||
for port in tree.findall('{%s}%s/{%s}%s' % (ns, ofc_consts.RESOURCES,
|
||||
ns, ofc_consts.PORT)):
|
||||
print(lxml.etree.tostring(port, pretty_print=True))
|
||||
|
||||
def _set_ports_down(self):
|
||||
"""try to set all ports down with etree operation"""
|
||||
tree = self._do_get()
|
||||
print lxml.etree.tostring(tree, pretty_print=True)
|
||||
|
||||
qname = lxml.etree.QName(tree.tag)
|
||||
ns = qname.namespace
|
||||
self._print_ports(tree, ns)
|
||||
|
||||
switch_id = tree.find('{%s}%s' % (ns, ofc_consts.ID))
|
||||
resources = tree.find('{%s}%s' % (ns, ofc_consts.RESOURCES))
|
||||
configuration = tree.find(
|
||||
'{%s}%s/{%s}%s/{%s}%s' % (ns, ofc_consts.RESOURCES,
|
||||
ns, ofc_consts.PORT,
|
||||
ns, ofc_consts.CONFIGURATION))
|
||||
admin_state = tree.find(
|
||||
'{%s}%s/{%s}%s/{%s}%s/{%s}%s' % (ns, ofc_consts.RESOURCES,
|
||||
ns, ofc_consts.PORT,
|
||||
ns, ofc_consts.CONFIGURATION,
|
||||
ns, ofc_consts.ADMIN_STATE))
|
||||
|
||||
config_ = lxml.etree.Element(
|
||||
'{%s}%s' % (ncclient.xml_.BASE_NS_1_0, nc_consts.CONFIG))
|
||||
capable_switch_ = lxml.etree.SubElement(config_, tree.tag)
|
||||
switch_id_ = lxml.etree.SubElement(capable_switch_, switch_id.tag)
|
||||
switch_id_.text = switch_id.text
|
||||
resources_ = lxml.etree.SubElement(capable_switch_,
|
||||
resources.tag)
|
||||
for port in tree.findall(
|
||||
'{%s}%s/{%s}%s' % (ns, ofc_consts.RESOURCES,
|
||||
ns, ofc_consts.PORT)):
|
||||
resource_id = port.find('{%s}%s' % (ns, ofc_consts.RESOURCE_ID))
|
||||
|
||||
port_ = lxml.etree.SubElement(resources_, port.tag)
|
||||
resource_id_ = lxml.etree.SubElement(port_, resource_id.tag)
|
||||
resource_id_.text = resource_id.text
|
||||
configuration_ = lxml.etree.SubElement(port_, configuration.tag)
|
||||
configuration_.set(ofc_consts.OPERATION, nc_consts.MERGE)
|
||||
admin_state_ = lxml.etree.SubElement(configuration_,
|
||||
admin_state.tag)
|
||||
admin_state_.text = ofc_consts.DOWN
|
||||
self._do_edit_config(lxml.etree.tostring(config_, pretty_print=True))
|
||||
|
||||
tree = self._do_get()
|
||||
self._print_ports(tree, ns)
|
||||
|
||||
def _do_of_config(self):
|
||||
self._do_get()
|
||||
self._do_get_config('running')
|
||||
self._do_get_config('startup')
|
||||
|
||||
# LINC doesn't support 'candidate' datastore
|
||||
try:
|
||||
self._do_get_config('candidate')
|
||||
except ncclient.NCClientError:
|
||||
traceback.print_exc()
|
||||
|
||||
# use raw XML format
|
||||
self._do_edit_config(SWITCH_PORT_DOWN)
|
||||
self._do_edit_config(SWITCH_ADVERTISED)
|
||||
self._do_edit_config(SWITCH_CONTROLLER)
|
||||
|
||||
self._set_ports_down()
|
||||
|
||||
self.switch.close_session()
|
||||
Loading…
Reference in New Issue
Block a user