blob: cd52ff5a8d5cbd32d266b97e84768b904e59610a [file] [log] [blame]
A R Karthick36cfcef2016-08-18 15:20:07 -07001#
Chetan Gaonkercfcce782016-05-10 10:10:42 -07002# Copyright 2016-present Ciena Corporation
3#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
A R Karthick36cfcef2016-08-18 15:20:07 -07007#
Chetan Gaonkercfcce782016-05-10 10:10:42 -07008# http://www.apache.org/licenses/LICENSE-2.0
A R Karthick36cfcef2016-08-18 15:20:07 -07009#
Chetan Gaonkercfcce782016-05-10 10:10:42 -070010# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15#
Chetan Gaonker7997bb42016-03-28 09:46:15 -070016import os
17import json
18##load the olt config
19
20class OltConfig:
21 def __init__(self, olt_conf_file = ''):
22 if not olt_conf_file:
23 self.olt_conf_file = os.getenv('OLT_CONFIG')
24 else:
25 self.olt_conf_file = olt_conf_file
26 try:
27 self.olt_handle = open(self.olt_conf_file, 'r')
28 self.olt_conf = json.load(self.olt_handle)
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070029 self.olt_conf['olt'] = True
Chetan Gaonker7997bb42016-03-28 09:46:15 -070030 except:
31 self.olt_handle = None
32 self.olt_conf = {}
33 self.olt_conf['olt'] = False
A R Karthick7c864552016-07-27 10:47:04 -070034
Chetan Gaonker7997bb42016-03-28 09:46:15 -070035 def on_olt(self):
36 return self.olt_conf['olt'] is True
37
38 def olt_port_map(self):
Chetan Gaonker5209fe82016-04-19 10:09:53 -070039 if self.on_olt() and self.olt_conf.has_key('port_map'):
Chetan Gaonker7997bb42016-03-28 09:46:15 -070040 port_map = {}
A.R Karthick88e80b92016-12-05 20:23:45 -080041 if self.olt_conf['port_map'].has_key('switches'):
42 port_map['switches'] = self.olt_conf['port_map']['switches']
43 else:
44 port_map['switches'] = []
45 nr_switches = 1
46 if self.olt_conf['port_map'].has_key('nr_switches'):
47 nr_switches = int(self.olt_conf['port_map']['nr_switches'])
48 for sw in xrange(nr_switches):
49 switch = 'br-int{}'.format(sw+1) if sw > 0 else 'br-int'
50 port_map['switches'].append(switch)
51 #if we have a host interface enabled, invalidate the switches config
52 if self.olt_conf['port_map'].has_key('host'):
53 #if host interface is specified, then use the host instead of ovs switch
54 port_map['host'] = self.olt_conf['port_map']['host']
55 port_map['switches'] = [ port_map['host'] ]
56 else:
57 port_map['host'] = port_map['switches'][0]
58 nr_switches = len(port_map['switches'])
59 port_map['switch_port_list'] = []
A R Karthickb7e80902016-05-17 09:38:31 -070060 if self.olt_conf['port_map'].has_key('ports'):
61 port_map['ports'] = self.olt_conf['port_map']['ports']
A R Karthick36cfcef2016-08-18 15:20:07 -070062 num_ports = len(port_map['ports'])
A.R Karthick88e80b92016-12-05 20:23:45 -080063 port_map['switch_port_list'].append( (port_map['switches'][0], port_map['ports']) )
A R Karthickb7e80902016-05-17 09:38:31 -070064 else:
65 port_map['ports'] = []
66 num_ports = int(self.olt_conf['port_map']['num_ports'])
A.R Karthick88e80b92016-12-05 20:23:45 -080067 for sw in xrange(nr_switches):
68 port_list = []
69 switch = port_map['switches'][sw]
70 port_start = sw * num_ports * 2
71 port_end = port_start + num_ports * 2
72 for port in xrange(port_start, port_end, 2):
73 port_name = 'veth{}'.format(port)
74 port_map['ports'].append(port_name)
75 port_list.append(port_name)
76 port_map['switch_port_list'].append( (switch, port_list) )
A R Karthick36cfcef2016-08-18 15:20:07 -070077 ##also add dhcprelay ports. We add as many relay ports as subscriber ports
A.R Karthick88e80b92016-12-05 20:23:45 -080078 port_map['num_ports'] = num_ports
A R Karthick36cfcef2016-08-18 15:20:07 -070079 relay_ports = num_ports
80 port_map['relay_ports'] = []
A.R Karthick88e80b92016-12-05 20:23:45 -080081 port_map['switch_relay_port_list'] = []
82 for sw in xrange(nr_switches):
83 port_list = []
84 switch = port_map['switches'][sw]
85 port_start = (nr_switches + sw) * relay_ports * 2
86 port_end = port_start + relay_ports * 2
87 for port in xrange(port_start, port_end, 2):
88 port_name = 'veth{}'.format(port)
89 port_map['relay_ports'].append(port_name)
90 port_list.append(port_name)
91 port_map['switch_relay_port_list'].append( (switch, port_list) )
A R Karthickb7e80902016-05-17 09:38:31 -070092 port_num = 1
93 port_map['uplink'] = int(self.olt_conf['uplink'])
A R Karthick07769362016-07-28 17:36:15 -070094 port_map['wan'] = None
95 if self.olt_conf.has_key('wan'):
96 port_map['wan'] = self.olt_conf['wan']
A R Karthickb7e80902016-05-17 09:38:31 -070097 port_list = []
98 ##build the port map and inverse port map
A.R Karthick88e80b92016-12-05 20:23:45 -080099 for sw in xrange(nr_switches):
100 sw_portnum = 1
101 switch, ports = port_map['switch_port_list'][sw]
102 uplink = sw * num_ports + port_map['uplink']
103 port_map[switch] = {}
104 port_map[switch]['uplink'] = uplink
105 for p in ports:
106 port_map[port_num] = p
107 port_map[p] = port_num
108 if sw_portnum != port_map['uplink']:
109 #create tx, rx map
110 port_list.append( (uplink, port_num) )
111 port_num += 1
112 sw_portnum += 1
A R Karthick36cfcef2016-08-18 15:20:07 -0700113 ##build the port and inverse map for relay ports
114 for port in port_map['relay_ports']:
115 port_map[port_num] = port
116 port_map[port] = port_num
117 port_num += 1
A R Karthickb7e80902016-05-17 09:38:31 -0700118 port_map['start_vlan'] = 0
A R Karthickb7e80902016-05-17 09:38:31 -0700119 if self.olt_conf['port_map'].has_key('start_vlan'):
120 port_map['start_vlan'] = int(self.olt_conf['port_map']['start_vlan'])
121
122 return port_map, port_list
123 else:
124 return None, None
125
Chetan Gaonkera2b87df2016-03-31 15:41:31 -0700126 def olt_device_data(self):
127 if self.on_olt():
128 accessDeviceDict = {}
129 accessDeviceDict['uplink'] = str(self.olt_conf['uplink'])
130 accessDeviceDict['vlan'] = str(self.olt_conf['vlan'])
131 return accessDeviceDict
132 return None
A.R Karthick369f89e2017-03-02 15:22:45 -0800133
134 def get_vcpes(self):
135 if self.on_olt():
136 if self.olt_conf.has_key('vcpe'):
137 return self.olt_conf['vcpe']
138 return []
A R Karthick03f40aa2017-03-20 19:33:55 -0700139
140 def get_vcpes_by_type(self, service):
141 return filter(lambda vcpe: vcpe['type'].lower() == service.lower(), self.get_vcpes())