¡@
OpenStack Study: type_vxlan.py**** CubicPower OpenStack Study ****
# Copyright (c) 2013 OpenStack Foundation # 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. # @author: Kyle Mestery, Cisco Systems, Inc. from oslo.config import cfg import sqlalchemy as sa from sqlalchemy.orm import exc as sa_exc from neutron.common import exceptions as exc from neutron.db import api as db_api from neutron.db import model_base from neutron.openstack.common import log from neutron.plugins.common import constants as p_const from neutron.plugins.ml2 import driver_api as api from neutron.plugins.ml2.drivers import type_tunnel LOG = log.getLogger(__name__) VXLAN_UDP_PORT = 4789 MAX_VXLAN_VNI = 16777215 vxlan_opts = [ cfg.ListOpt('vni_ranges', default=[], help=_("Comma-separated list of "enumerating ranges of VXLAN VNI IDs that are " "available for tenant network allocation")), cfg.StrOpt('vxlan_group', default=None, help=_("Multicast group for VXLAN. If unset, disables VXLAN " "multicast mode.")), ] cfg.CONF.register_opts(vxlan_opts, "ml2_type_vxlan")
**** CubicPower OpenStack Study **** class VxlanAllocation(model_base.BASEV2): __tablename__ = 'ml2_vxlan_allocations' vxlan_vni = sa.Column(sa.Integer, nullable=False, primary_key=True, autoincrement=False) allocated = sa.Column(sa.Boolean, nullable=False,
**** CubicPower OpenStack Study **** class VxlanEndpoints(model_base.BASEV2): """Represents tunnel endpoint in RPC mode.""" __tablename__ = 'ml2_vxlan_endpoints' ip_address = sa.Column(sa.String(64), primary_key=True) udp_port = sa.Column(sa.Integer, primary_key=True, nullable=False, autoincrement=False)
**** CubicPower OpenStack Study **** def __repr__(self): return "
**** CubicPower OpenStack Study **** class VxlanTypeDriver(type_tunnel.TunnelTypeDriver):
**** CubicPower OpenStack Study **** def get_type(self): return p_const.TYPE_VXLAN
**** CubicPower OpenStack Study **** def initialize(self): self.vxlan_vni_ranges = [] self._parse_tunnel_ranges( cfg.CONF.ml2_type_vxlan.vni_ranges, self.vxlan_vni_ranges, p_const.TYPE_VXLAN ) self._sync_vxlan_allocations()
**** CubicPower OpenStack Study **** def reserve_provider_segment(self, session, segment): segmentation_id = segment.get(api.SEGMENTATION_ID) with session.begin(subtransactions=True): try: alloc = (session.query(VxlanAllocation). filter_by(vxlan_vni=segmentation_id). with_lockmode('update'). one()) if alloc.allocated: raise exc.TunnelIdInUse(tunnel_id=segmentation_id) LOG.debug(_("Reserving specific vxlan tunnel %s from pool"), segmentation_id) alloc.allocated = True except sa_exc.NoResultFound: LOG.debug(_("Reserving specific vxlan tunnel %s outside pool"), segmentation_id) alloc = VxlanAllocation(vxlan_vni=segmentation_id) alloc.allocated = True session.add(alloc)
**** CubicPower OpenStack Study **** def allocate_tenant_segment(self, session): with session.begin(subtransactions=True): alloc = (session.query(VxlanAllocation). filter_by(allocated=False). with_lockmode('update'). first()) if alloc: LOG.debug(_("Allocating vxlan tunnel vni %(vxlan_vni)s"), {'vxlan_vni': alloc.vxlan_vni}) alloc.allocated = True return {api.NETWORK_TYPE: p_const.TYPE_VXLAN, api.PHYSICAL_NETWORK: None, api.SEGMENTATION_ID: alloc.vxlan_vni}
**** CubicPower OpenStack Study **** def release_segment(self, session, segment): vxlan_vni = segment[api.SEGMENTATION_ID] with session.begin(subtransactions=True): try: alloc = (session.query(VxlanAllocation). filter_by(vxlan_vni=vxlan_vni). with_lockmode('update'). one()) alloc.allocated = False for low, high in self.vxlan_vni_ranges: if low <= vxlan_vni <= high: LOG.debug(_("Releasing vxlan tunnel %s to pool"), vxlan_vni) break else: session.delete(alloc) LOG.debug(_("Releasing vxlan tunnel %s outside pool"), vxlan_vni) except sa_exc.NoResultFound: LOG.warning(_("vxlan_vni %s not found"), vxlan_vni)
**** CubicPower OpenStack Study **** def _sync_vxlan_allocations(self): """ Synchronize vxlan_allocations table with configured tunnel ranges. """ # determine current configured allocatable vnis vxlan_vnis = set() for tun_min, tun_max in self.vxlan_vni_ranges: if tun_max + 1 - tun_min > MAX_VXLAN_VNI: LOG.error(_("Skipping unreasonable VXLAN VNI range " "%(tun_min)s:%(tun_max)s"), {'tun_min': tun_min, 'tun_max': tun_max}) else: vxlan_vnis |= set(xrange(tun_min, tun_max + 1)) session = db_api.get_session() with session.begin(subtransactions=True): # remove from table unallocated tunnels not currently allocatable allocs = session.query(VxlanAllocation) for alloc in allocs: try: # see if tunnel is allocatable vxlan_vnis.remove(alloc.vxlan_vni) except KeyError: # it's not allocatable, so check if its allocated if not alloc.allocated: # it's not, so remove it from table LOG.debug(_("Removing tunnel %s from pool"), alloc.vxlan_vni) session.delete(alloc) # add missing allocatable tunnels to table for vxlan_vni in sorted(vxlan_vnis): alloc = VxlanAllocation(vxlan_vni=vxlan_vni) session.add(alloc)
**** CubicPower OpenStack Study **** def get_vxlan_allocation(self, session, vxlan_vni): with session.begin(subtransactions=True): return session.query(VxlanAllocation).filter_by( vxlan_vni=vxlan_vni).first()
**** CubicPower OpenStack Study **** def get_endpoints(self): """Get every vxlan endpoints from database.""" LOG.debug(_("get_vxlan_endpoints() called")) session = db_api.get_session() with session.begin(subtransactions=True): vxlan_endpoints = session.query(VxlanEndpoints) return [{'ip_address': vxlan_endpoint.ip_address, 'udp_port': vxlan_endpoint.udp_port} for vxlan_endpoint in vxlan_endpoints]
**** CubicPower OpenStack Study **** def add_endpoint(self, ip, udp_port=VXLAN_UDP_PORT): LOG.debug(_("add_vxlan_endpoint() called for ip %s"), ip) session = db_api.get_session() with session.begin(subtransactions=True): try: vxlan_endpoint = (session.query(VxlanEndpoints). filter_by(ip_address=ip). with_lockmode('update').one()) except sa_exc.NoResultFound: vxlan_endpoint = VxlanEndpoints(ip_address=ip, udp_port=udp_port) session.add(vxlan_endpoint) return vxlan_endpoint
|