mirror of
https://github.com/osrg/gobgp.git
synced 2024-05-11 05:55:10 +00:00
Franza Cavalcante
8c90684b27
This PR removes dependencies on old Fabric version, as it's not supported by Python3. The current Fabric versions don't support the colors and indent used previously, so we found substitute methods from other libraries and defined these in the library files. The local function from fabric is now just a wrapper to invoke's run function. All the files were processed through 2to3 command. All the tests were executed and we don't see any difference on the outputs when running Python2 or Python3. The creation of gobgp container is removed from base.py into fabfile.py, in order to comply with Fabric2 changes and simplify dependencies.
127 lines
4.0 KiB
Python
127 lines
4.0 KiB
Python
# Copyright (C) 2015 Nippon Telegraph and Telephone Corporation.
|
|
#
|
|
# 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
|
|
|
|
from lib.base import (
|
|
BGPContainer,
|
|
CmdBuffer,
|
|
try_several_times,
|
|
wait_for_completion,
|
|
yellow,
|
|
indent,
|
|
local,
|
|
)
|
|
|
|
|
|
class BirdContainer(BGPContainer):
|
|
|
|
WAIT_FOR_BOOT = 1
|
|
SHARED_VOLUME = '/etc/bird'
|
|
|
|
def __init__(self, name, asn, router_id, ctn_image_name='osrg/bird'):
|
|
super(BirdContainer, self).__init__(name, asn, router_id,
|
|
ctn_image_name)
|
|
self.shared_volumes.append((self.config_dir, self.SHARED_VOLUME))
|
|
|
|
def _start_bird(self):
|
|
c = CmdBuffer()
|
|
c << '#!/bin/sh'
|
|
c << 'bird'
|
|
cmd = 'echo "{0:s}" > {1}/start.sh'.format(c, self.config_dir)
|
|
local(cmd)
|
|
cmd = 'chmod 755 {0}/start.sh'.format(self.config_dir)
|
|
local(cmd)
|
|
self.local('{0}/start.sh'.format(self.SHARED_VOLUME))
|
|
|
|
def _wait_for_boot(self):
|
|
def _f():
|
|
ret = self.local('birdc show status > /dev/null 2>&1; echo $?', capture=True)
|
|
return ret == '0'
|
|
|
|
return wait_for_completion(_f)
|
|
|
|
def run(self):
|
|
super(BirdContainer, self).run()
|
|
self.reload_config()
|
|
return self.WAIT_FOR_BOOT
|
|
|
|
def create_config(self):
|
|
c = CmdBuffer()
|
|
c << 'router id {0};'.format(self.router_id)
|
|
for peer, info in self.peers.items():
|
|
c << 'protocol bgp {'
|
|
c << ' local as {0};'.format(self.asn)
|
|
n_addr = info['neigh_addr'].split('/')[0]
|
|
c << ' neighbor {0} as {1};'.format(n_addr, peer.asn)
|
|
c << ' multihop;'
|
|
c << '}'
|
|
|
|
with open('{0}/bird.conf'.format(self.config_dir), 'w') as f:
|
|
print(yellow('[{0}\'s new bird.conf]'.format(self.name)))
|
|
print(yellow(indent(str(c))))
|
|
f.writelines(str(c))
|
|
|
|
def reload_config(self):
|
|
if len(self.peers) == 0:
|
|
return
|
|
|
|
def _reload():
|
|
def _is_running():
|
|
ps = self.local('ps', capture=True)
|
|
running = False
|
|
for line in ps.split('\n')[1:]:
|
|
if 'bird' in line:
|
|
running = True
|
|
return running
|
|
|
|
if _is_running():
|
|
self.local('birdc configure')
|
|
else:
|
|
self._start_bird()
|
|
|
|
self._wait_for_boot()
|
|
if not _is_running():
|
|
raise RuntimeError()
|
|
|
|
try_several_times(_reload)
|
|
|
|
|
|
class RawBirdContainer(BirdContainer):
|
|
def __init__(self, name, config, ctn_image_name='osrg/bird'):
|
|
asn = None
|
|
router_id = None
|
|
for line in config.split('\n'):
|
|
line = line.strip()
|
|
if line.startswith('local as'):
|
|
asn = int(line[len('local as'):].strip('; '))
|
|
if line.startswith('router id'):
|
|
router_id = line[len('router id'):].strip('; ')
|
|
if not asn:
|
|
raise Exception('asn not in bird config')
|
|
if not router_id:
|
|
raise Exception('router-id not in bird config')
|
|
self.config = config
|
|
super(RawBirdContainer, self).__init__(name, asn, router_id,
|
|
ctn_image_name)
|
|
|
|
def create_config(self):
|
|
with open('{0}/bird.conf'.format(self.config_dir), 'w') as f:
|
|
print(yellow('[{0}\'s new bird.conf]'.format(self.name)))
|
|
print(yellow(indent(self.config)))
|
|
f.writelines(self.config)
|