Merge "adds iptools driver for ip commands"
This commit is contained in:
Zuul
Gerrit Code Review
commit
87e0edde4d
|
|
@ -21,15 +21,18 @@ LOG = logging.getLogger(__name__)
|
|||
|
||||
|
||||
impl_map = {
|
||||
'pyroute2': 'os_vif.internal.command.ip.impl_pyroute2',
|
||||
'pyroute2': 'os_vif.internal.command.ip.impl_pyroute2.PyRoute2',
|
||||
'IPTools': 'os_vif.internal.command.ip.impl_shell.IPTools',
|
||||
}
|
||||
|
||||
|
||||
def _get_impl():
|
||||
# NOTE(ralonsoh): currently there is only one implementation. No config
|
||||
# options are exposed to the user.
|
||||
pyroute2 = importutils.import_module(impl_map['pyroute2'])
|
||||
return pyroute2.PyRoute2()
|
||||
# NOTE(sean-k-mooney): currently pyroute2 has a file handle leak. An
|
||||
# iptools driver has been added as a workaround but No config options are
|
||||
# exposed to the user. The iptools driver is considered deprecated and
|
||||
# will be removed when a new release of pyroute2 is available.
|
||||
driver = 'IPTools'
|
||||
return importutils.import_object(impl_map[driver])
|
||||
|
||||
|
||||
@six.add_metaclass(abc.ABCMeta)
|
||||
|
|
|
|||
|
|
@ -0,0 +1,138 @@
|
|||
# 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 re
|
||||
|
||||
from oslo_concurrency import processutils
|
||||
from oslo_utils import excutils
|
||||
|
||||
from os_vif.internal.command.ip import api
|
||||
|
||||
|
||||
def _execute_command(*args):
|
||||
return processutils.execute(*args)
|
||||
|
||||
|
||||
class ShellIpCommands(object):
|
||||
|
||||
def __init__(self, cmd=_execute_command):
|
||||
self._execute_command = cmd
|
||||
|
||||
def add_device(self, device, dev_type, peer=None, link=None,
|
||||
vlan_id=None):
|
||||
ret = None
|
||||
if 'vlan' == dev_type:
|
||||
ret = self._execute_command('ip', 'link', 'add', 'link', link,
|
||||
'name', device, 'type', dev_type,
|
||||
'id', vlan_id)
|
||||
elif 'veth' == dev_type:
|
||||
ret = self._execute_command('ip', 'link', 'add', device, 'type',
|
||||
dev_type, 'peer', 'name', peer)
|
||||
elif 'dummy' == dev_type:
|
||||
ret = self._execute_command('ip', 'link', 'add', device,
|
||||
'type', dev_type)
|
||||
return ret
|
||||
|
||||
def del_device(self, device):
|
||||
ret = None
|
||||
if self.exist_device(device):
|
||||
ret = self._execute_command('ip', 'link', 'del', device)
|
||||
return ret
|
||||
|
||||
def set(self, device, status=None, **kwargs):
|
||||
args = ['ip', 'link', 'set', device]
|
||||
if status is not None:
|
||||
args.append(status)
|
||||
temp = [x for x in kwargs.items()]
|
||||
for x in temp:
|
||||
args += x
|
||||
self._execute_command(*args)
|
||||
|
||||
def set_status_up(self, device):
|
||||
self.set(device, status='up')
|
||||
|
||||
def set_status_down(self, device):
|
||||
self.set(device, status='down')
|
||||
|
||||
def set_device_mtu(self, device, mtu):
|
||||
args = {'mtu': mtu}
|
||||
self.set(device, args)
|
||||
|
||||
def show_device(self, device):
|
||||
val, err = _execute_command('ip', 'link', 'show', device)
|
||||
return val.splitlines()
|
||||
|
||||
def exist_device(self, device):
|
||||
try:
|
||||
self._execute_command('ip', 'link', 'show', device)
|
||||
return True
|
||||
except processutils.ProcessExecutionError as e:
|
||||
with excutils.save_and_reraise_exception() as saved_exception:
|
||||
if e.exit_code == 1:
|
||||
saved_exception.reraise = False
|
||||
return False
|
||||
|
||||
def show_state(self, device):
|
||||
regex = re.compile(r".*state (?P<state>\w+)")
|
||||
match = regex.match(self.show_device(device)[0])
|
||||
if match is None:
|
||||
return
|
||||
return match.group('state')
|
||||
|
||||
def show_promisc(self, device):
|
||||
regex = re.compile(r".*(PROMISC)")
|
||||
match = regex.match(self.show_device(device)[0])
|
||||
return True if match else False
|
||||
|
||||
def show_mac(self, device):
|
||||
exp = r".*link/ether (?P<mac>([0-9A-Fa-f]{2}[:]){5}[0-9A-Fa-f]{2})"
|
||||
regex = re.compile(exp)
|
||||
match = regex.match(self.show_device(device)[1])
|
||||
if match is None:
|
||||
return
|
||||
return match.group('mac')
|
||||
|
||||
def show_mtu(self, device):
|
||||
regex = re.compile(r".*mtu (?P<mtu>\d+)")
|
||||
match = regex.match(self.show_device(device)[0])
|
||||
if match is None:
|
||||
return
|
||||
return int(match.group('mtu'))
|
||||
|
||||
|
||||
class IPTools(api.IpCommand):
|
||||
|
||||
def __init__(self, cmd=_execute_command):
|
||||
self.ip = ShellIpCommands(cmd=cmd)
|
||||
|
||||
def set(self, device, check_exit_code=None, state=None, mtu=None,
|
||||
address=None, promisc=None):
|
||||
args = {}
|
||||
if mtu is not None:
|
||||
args['mtu'] = mtu
|
||||
if address is not None:
|
||||
args['address'] = address
|
||||
if promisc is not None:
|
||||
args['promisc'] = 'on' if promisc else 'off'
|
||||
|
||||
if isinstance(check_exit_code, int):
|
||||
check_exit_code = [check_exit_code]
|
||||
return self.ip.set(device, status=state, **args)
|
||||
|
||||
def add(self, device, dev_type, check_exit_code=None, peer=None, link=None,
|
||||
vlan_id=None):
|
||||
|
||||
return self.ip.add_device(device, dev_type, peer=peer,
|
||||
link=link, vlan_id=vlan_id)
|
||||
|
||||
def delete(self, device, check_exit_code=None):
|
||||
return self.ip.del_device(device)
|
||||
|
|
@ -0,0 +1,180 @@
|
|||
# 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 re
|
||||
|
||||
from oslo_concurrency import processutils
|
||||
from oslo_utils import excutils
|
||||
|
||||
from os_vif.internal.command.ip import impl_shell
|
||||
from os_vif.tests.functional import base
|
||||
from os_vif.tests.functional import privsep
|
||||
|
||||
|
||||
@privsep.os_vif_pctxt.entrypoint
|
||||
def _execute_command(*args):
|
||||
return processutils.execute(*args)
|
||||
|
||||
|
||||
class ShellIpCommands(object):
|
||||
|
||||
def add_device(self, device, dev_type, peer=None, link=None,
|
||||
vlan_id=None):
|
||||
if 'vlan' == dev_type:
|
||||
_execute_command('ip', 'link', 'add', 'link', link,
|
||||
'name', device, 'type', dev_type, 'vlan', 'id',
|
||||
vlan_id)
|
||||
elif 'veth' == dev_type:
|
||||
_execute_command('ip', 'link', 'add', device, 'type', dev_type,
|
||||
'peer', 'name', peer)
|
||||
elif 'dummy' == dev_type:
|
||||
_execute_command('ip', 'link', 'add', device, 'type', dev_type)
|
||||
|
||||
def del_device(self, device):
|
||||
if self.exist_device(device):
|
||||
_execute_command('ip', 'link', 'del', device)
|
||||
|
||||
def set_status_up(self, device):
|
||||
_execute_command('ip', 'link', 'set', device, 'up')
|
||||
|
||||
def set_status_down(self, device):
|
||||
_execute_command('ip', 'link', 'set', device, 'down')
|
||||
|
||||
def set_device_mtu(self, device, mtu):
|
||||
_execute_command('ip', 'link', 'set', device, 'mtu', mtu)
|
||||
|
||||
def show_device(self, device):
|
||||
val, err = _execute_command('ip', 'link', 'show', device)
|
||||
return val.splitlines()
|
||||
|
||||
def exist_device(self, device):
|
||||
try:
|
||||
_execute_command('ip', 'link', 'show', device)
|
||||
return True
|
||||
except processutils.ProcessExecutionError as e:
|
||||
with excutils.save_and_reraise_exception() as saved_exception:
|
||||
if e.exit_code == 1:
|
||||
saved_exception.reraise = False
|
||||
return False
|
||||
|
||||
def show_state(self, device):
|
||||
regex = re.compile(r".*state (?P<state>\w+)")
|
||||
match = regex.match(self.show_device(device)[0])
|
||||
if match is None:
|
||||
return
|
||||
return match.group('state')
|
||||
|
||||
def show_promisc(self, device):
|
||||
regex = re.compile(r".*(PROMISC)")
|
||||
match = regex.match(self.show_device(device)[0])
|
||||
return True if match else False
|
||||
|
||||
def show_mac(self, device):
|
||||
exp = r".*link/ether (?P<mac>([0-9A-Fa-f]{2}[:]){5}[0-9A-Fa-f]{2})"
|
||||
regex = re.compile(exp)
|
||||
match = regex.match(self.show_device(device)[1])
|
||||
if match is None:
|
||||
return
|
||||
return match.group('mac')
|
||||
|
||||
def show_mtu(self, device):
|
||||
regex = re.compile(r".*mtu (?P<mtu>\d+)")
|
||||
match = regex.match(self.show_device(device)[0])
|
||||
if match is None:
|
||||
return
|
||||
return int(match.group('mtu'))
|
||||
|
||||
|
||||
def _ip_cmd_set(*args, **kwargs):
|
||||
impl_shell.IPTools(cmd=_execute_command).set(*args, **kwargs)
|
||||
|
||||
|
||||
def _ip_cmd_add(*args, **kwargs):
|
||||
impl_shell.IPTools(cmd=_execute_command).add(*args, **kwargs)
|
||||
|
||||
|
||||
def _ip_cmd_delete(*args, **kwargs):
|
||||
impl_shell.IPTools(cmd=_execute_command).delete(*args, **kwargs)
|
||||
|
||||
|
||||
class TestIpCommand(ShellIpCommands, base.BaseFunctionalTestCase):
|
||||
|
||||
def setUp(self):
|
||||
super(TestIpCommand, self).setUp()
|
||||
|
||||
def test_set_state(self):
|
||||
device1 = "iptools_dev_1"
|
||||
device2 = "iptools_dev_2"
|
||||
self.addCleanup(self.del_device, device1)
|
||||
self.add_device(device1, 'veth', peer=device2)
|
||||
_ip_cmd_set(device1, state='up')
|
||||
_ip_cmd_set(device2, state='up')
|
||||
self.assertEqual('UP', self.show_state(device1))
|
||||
self.assertEqual('UP', self.show_state(device2))
|
||||
_ip_cmd_set(device1, state='down')
|
||||
_ip_cmd_set(device2, state='down')
|
||||
self.assertEqual('DOWN', self.show_state(device1))
|
||||
self.assertEqual('DOWN', self.show_state(device2))
|
||||
|
||||
def test_set_mtu(self):
|
||||
device = "iptools_dev_3"
|
||||
self.addCleanup(self.del_device, device)
|
||||
self.add_device(device, 'dummy')
|
||||
_ip_cmd_set(device, mtu=1200)
|
||||
self.assertEqual(1200, self.show_mtu(device))
|
||||
_ip_cmd_set(device, mtu=900)
|
||||
self.assertEqual(900, self.show_mtu(device))
|
||||
|
||||
def test_set_address(self):
|
||||
device = "iptools_dev_4"
|
||||
address1 = "36:a7:e4:f9:01:01"
|
||||
address2 = "36:a7:e4:f9:01:01"
|
||||
self.addCleanup(self.del_device, device)
|
||||
self.add_device(device, 'dummy')
|
||||
_ip_cmd_set(device, address=address1)
|
||||
self.assertEqual(address1, self.show_mac(device))
|
||||
_ip_cmd_set(device, address=address2)
|
||||
self.assertEqual(address2, self.show_mac(device))
|
||||
|
||||
def test_set_promisc(self):
|
||||
device = "iptools_dev_5"
|
||||
self.addCleanup(self.del_device, device)
|
||||
self.add_device(device, 'dummy')
|
||||
_ip_cmd_set(device, promisc=True)
|
||||
self.assertTrue(self.show_promisc(device))
|
||||
_ip_cmd_set(device, promisc=False)
|
||||
self.assertFalse(self.show_promisc(device))
|
||||
|
||||
def test_add_vlan(self):
|
||||
device = "iptools_dev_6"
|
||||
link = "iptools_devlink"
|
||||
self.addCleanup(self.del_device, device)
|
||||
self.addCleanup(self.del_device, link)
|
||||
self.add_device(link, 'dummy')
|
||||
_ip_cmd_add(device, 'vlan', link=link, vlan_id=100)
|
||||
self.assertTrue(self.exist_device(device))
|
||||
|
||||
def test_add_veth(self):
|
||||
device = "iptools_dev_7"
|
||||
peer = "iptools_devpeer"
|
||||
self.addCleanup(self.del_device, device)
|
||||
_ip_cmd_add(device, 'veth', peer=peer)
|
||||
self.assertTrue(self.exist_device(device))
|
||||
self.assertTrue(self.exist_device(peer))
|
||||
|
||||
def test_delete(self):
|
||||
device = "iptools_dev_8"
|
||||
self.addCleanup(self.del_device, device)
|
||||
self.add_device(device, 'dummy')
|
||||
self.assertTrue(self.exist_device(device))
|
||||
_ip_cmd_delete(device)
|
||||
self.assertFalse(self.exist_device(device))
|
||||
|
|
@ -16,7 +16,7 @@ from pyroute2.netlink import exceptions as ipexc
|
|||
from pyroute2.netlink.rtnl import ifinfmsg
|
||||
|
||||
from os_vif import exception
|
||||
from os_vif.internal.command.ip import api as ip_api
|
||||
from os_vif.internal.command.ip import impl_pyroute2
|
||||
from os_vif.tests.unit import base
|
||||
|
||||
|
||||
|
|
@ -35,7 +35,7 @@ class TestIpCommand(base.TestCase):
|
|||
|
||||
def setUp(self):
|
||||
super(TestIpCommand, self).setUp()
|
||||
self.ip = ip_api._get_impl()
|
||||
self.ip = impl_pyroute2.PyRoute2()
|
||||
self.ip_link_p = mock.patch.object(iproute.IPRoute, 'link')
|
||||
self.ip_link = self.ip_link_p.start()
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue