73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
# Copyright (c) 2016 Mirantis, Inc.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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 os_vif
|
|
from oslo_log import log as logging
|
|
import pyroute2
|
|
from stevedore import driver as stv_driver
|
|
|
|
_BINDING_NAMESPACE = 'kuryr_kubernetes.cni.binding'
|
|
_IPDB = {}
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def _get_binding_driver(vif):
|
|
mgr = stv_driver.DriverManager(namespace=_BINDING_NAMESPACE,
|
|
name=type(vif).__name__,
|
|
invoke_on_load=True)
|
|
return mgr.driver
|
|
|
|
|
|
def get_ipdb(netns=None):
|
|
try:
|
|
return _IPDB[netns]
|
|
except KeyError:
|
|
if netns:
|
|
ipdb = pyroute2.IPDB(nl=pyroute2.NetNS(netns))
|
|
else:
|
|
ipdb = pyroute2.IPDB()
|
|
_IPDB[netns] = ipdb
|
|
return ipdb
|
|
|
|
|
|
def _configure_l3(vif, ifname, netns):
|
|
with get_ipdb(netns).interfaces[ifname] as iface:
|
|
for subnet in vif.network.subnets.objects:
|
|
for fip in subnet.ips.objects:
|
|
iface.add_ip(str(fip.address), mask=str(subnet.cidr.netmask))
|
|
|
|
routes = get_ipdb(netns).routes
|
|
for subnet in vif.network.subnets.objects:
|
|
for route in subnet.routes.objects:
|
|
routes.add(gateway=str(route.gateway),
|
|
dst=str(route.cidr)).commit()
|
|
if subnet.gateway:
|
|
routes.add(gateway=str(subnet.gateway),
|
|
dst='default').commit()
|
|
|
|
|
|
def connect(vif, instance_info, ifname, netns=None):
|
|
driver = _get_binding_driver(vif)
|
|
os_vif.plug(vif, instance_info)
|
|
driver.connect(vif, ifname, netns)
|
|
_configure_l3(vif, ifname, netns)
|
|
|
|
|
|
def disconnect(vif, instance_info, ifname, netns=None):
|
|
driver = _get_binding_driver(vif)
|
|
driver.disconnect(vif, ifname, netns)
|
|
os_vif.unplug(vif, instance_info)
|