1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
# 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.
from base import *
class ExaBGPContainer(BGPContainer):
SHARED_VOLUME = '/root/shared_volume'
def __init__(self, name, asn, router_id, ctn_image_name='osrg/exabgp',
exabgp_path=''):
super(ExaBGPContainer, self).__init__(name, asn, router_id,
ctn_image_name)
self.shared_volumes.append((self.config_dir, self.SHARED_VOLUME))
self.exabgp_path = exabgp_path
def _start_exabgp(self):
cmd = CmdBuffer(' ')
cmd << 'docker exec -d {0}'.format(self.name)
cmd << 'env exabgp.log.destination={0}/exabgpd.log'.format(self.SHARED_VOLUME)
cmd << './exabgp/sbin/exabgp {0}/exabgpd.conf'.format(self.SHARED_VOLUME)
local(str(cmd), capture=True)
def _update_exabgp(self):
c = CmdBuffer()
c << '#!/bin/bash'
remotepath = '/root/exabgp'
localpath = self.exabgp_path
if localpath != '':
local('cp -r {0} {1}'.format(localpath, self.config_dir))
c << 'cp {0}/etc/exabgp/exabgp.env {1}'.format(remotepath, self.SHARED_VOLUME)
c << 'sed -i -e \'s/all = false/all = true/g\' {0}/exabgp.env'.format(self.SHARED_VOLUME)
c << 'cp -r {0}/exabgp {1}'.format(self.SHARED_VOLUME,
remotepath[:-1*len('exabgp')])
c << 'cp {0}/exabgp.env {1}/etc/exabgp/'.format(self.SHARED_VOLUME, remotepath)
cmd = 'echo "{0:s}" > {1}/update.sh'.format(c, self.config_dir)
local(cmd, capture=True)
cmd = 'chmod 755 {0}/update.sh'.format(self.config_dir)
local(cmd, capture=True)
cmd = 'docker exec {0} {1}/update.sh'.format(self.name,
self.SHARED_VOLUME)
local(cmd, capture=True)
def run(self):
super(ExaBGPContainer, self).run()
self._update_exabgp()
self._start_exabgp()
return self.WAIT_FOR_BOOT
def create_config(self):
cmd = CmdBuffer()
for peer, info in self.peers.iteritems():
cmd << 'neighbor {0} {{'.format(info['neigh_addr'].split('/')[0])
cmd << ' router-id {0};'.format(self.router_id)
local_addr = ''
for me, you in itertools.product(self.ip_addrs, peer.ip_addrs):
if me[2] == you[2]:
local_addr = me[1]
if local_addr == '':
raise Exception('local_addr not found')
local_addr = local_addr.split('/')[0]
cmd << ' local-address {0};'.format(local_addr)
cmd << ' local-as {0};'.format(self.asn)
cmd << ' peer-as {0};'.format(peer.asn)
cmd << ' static {'
for route, attr in self.routes.iteritems():
if attr == '':
cmd << ' route {0} next-hop {1};'.format(route, local_addr)
else:
cmd << ' route {0} next-hop {1} attribute {2};'.format(route, local_addr, attr)
cmd << ' }'
cmd << '}'
with open('{0}/exabgpd.conf'.format(self.config_dir), 'w') as f:
print colors.yellow(str(cmd))
f.write(str(cmd))
def reload_config(self):
cmd = 'docker exec {0} /usr/bin/pkill exabgp -SIGALRM'.format(self.name)
try:
local(cmd, capture=True)
except:
self._start_exabgp()
|
