blob: 6119e8a868147505b6af9a7e74a22548ff493c60 [file] [log] [blame]
A R Karthick338268f2016-06-21 17:12:13 -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 Karthick338268f2016-06-21 17:12:13 -07007#
Chetan Gaonkercfcce782016-05-10 10:10:42 -07008# http://www.apache.org/licenses/LICENSE-2.0
A R Karthick338268f2016-06-21 17:12:13 -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 Gaonkerb424ff82016-03-08 12:11:12 -080016import threading
17import sys
18import os
19import time
20import monotonic
21import random
A.R Karthick2e99c472017-03-22 19:13:51 -070022import logging
23logging.getLogger('scapy.runtime').setLevel(logging.ERROR)
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080024from scapy.all import *
25from McastTraffic import *
26from IGMP import *
27from OnosCtrl import OnosCtrl
A R Karthick76a497a2017-04-12 10:59:39 -070028from CordTestUtils import log_test
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080029from nose.tools import *
A R Karthick76a497a2017-04-12 10:59:39 -070030log_test.setLevel('DEBUG')
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080031
32conf.verb = 0
33
34class IgmpChannel:
35
36 IGMP_DST_MAC = "01:00:5e:00:01:01"
37 IGMP_SRC_MAC = "5a:e1:ac:ec:4d:a1"
38 IP_SRC = '1.2.3.4'
39 IP_DST = '224.0.1.1'
40 igmp_eth = Ether(dst = IGMP_DST_MAC, src = IGMP_SRC_MAC, type = ETH_P_IP)
41 igmp_ip = IP(dst = IP_DST, src = IP_SRC)
A R Karthick338268f2016-06-21 17:12:13 -070042 ssm_list = []
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080043
ChetanGaonker689b3862016-10-17 16:25:01 -070044 def __init__(self, iface = 'veth0', ssm_list = [], src_list = ['1.2.3.4'], delay = 2,controller=None):
45 self.controller=controller
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080046 self.iface = iface
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -070047 self.ssm_list += ssm_list
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080048 self.src_list = src_list
49 self.delay = delay
A.R Karthick401a1ed2017-05-18 11:08:27 -070050 self.onos_ctrl = OnosCtrl('org.opencord.igmp',controller=self.controller)
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080051 self.onos_ctrl.activate()
A R Karthick338268f2016-06-21 17:12:13 -070052
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -070053 def igmp_load_ssm_config(self, ssm_list = []):
54 if not ssm_list:
55 ssm_list = self.ssm_list
56 self.ssm_table_load(ssm_list)
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080057
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -070058 def igmp_join(self, groups):
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080059 igmp = IGMPv3(type = IGMP_TYPE_V3_MEMBERSHIP_REPORT, max_resp_code=30,
60 gaddr='224.0.1.1')
61 for g in groups:
Chetan Gaonker38737f82016-05-11 17:44:17 -070062 gr = IGMPv3gr(rtype=IGMP_V3_GR_TYPE_INCLUDE, mcaddr=g)
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080063 gr.sources = self.src_list
64 igmp.grps.append(gr)
65
66 pkt = self.igmp_eth/self.igmp_ip/igmp
67 IGMPv3.fixup(pkt)
68 sendp(pkt, iface=self.iface)
69 if self.delay != 0:
70 time.sleep(self.delay)
71
72 def igmp_leave(self, groups):
73 igmp = IGMPv3(type = IGMP_TYPE_V3_MEMBERSHIP_REPORT, max_resp_code=30,
74 gaddr='224.0.1.1')
75 for g in groups:
Chetan Gaonker38737f82016-05-11 17:44:17 -070076 gr = IGMPv3gr(rtype=IGMP_V3_GR_TYPE_EXCLUDE, mcaddr=g)
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080077 gr.sources = self.src_list
78 igmp.grps.append(gr)
79
80 pkt = self.igmp_eth/self.igmp_ip/igmp
81 IGMPv3.fixup(pkt)
82 sendp(pkt, iface = self.iface)
83 if self.delay != 0:
84 time.sleep(self.delay)
85
86 def onos_load_config(self, config):
ChetanGaonker689b3862016-10-17 16:25:01 -070087 status, code = OnosCtrl.config(config,controller=self.controller)
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080088 if status is False:
A R Karthick76a497a2017-04-12 10:59:39 -070089 log_test.info('JSON config request returned status %d' %code)
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080090 time.sleep(2)
91
92 def ssm_table_load(self, groups):
A.R Karthick401a1ed2017-05-18 11:08:27 -070093 ssm_dict = {'apps' : { 'org.opencord.igmp' : { 'ssmTranslate' : [] } } }
94 ssm_xlate_list = ssm_dict['apps']['org.opencord.igmp']['ssmTranslate']
Chetan Gaonkerb424ff82016-03-08 12:11:12 -080095 for g in groups:
96 for s in self.src_list:
97 d = {}
98 d['source'] = s
99 d['group'] = g
100 ssm_xlate_list.append(d)
101 self.onos_load_config(ssm_dict)
102
Chetan Gaonkera58ab6e2016-03-23 15:04:20 -0700103 def cord_port_table_load(self, cord_port_map):
104 cord_group_dict = {'apps' : { 'org.ciena.cordigmp' : { 'cordIgmpTranslate' : [] } } }
105 cord_group_xlate_list = cord_group_dict['apps']['org.ciena.cordigmp']['cordIgmpTranslate']
106 for group, ports in cord_port_map.items():
107 d = {}
108 d['group'] = group
109 d['inputPort'] = ports[0]
110 d['outputPort'] = ports[1]
111 cord_group_xlate_list.append(d)
112 self.onos_load_config(cord_group_dict)
113
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800114class Channels(IgmpChannel):
115 Stopped = 0
116 Started = 1
117 Idle = 0
118 Joined = 1
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -0700119 def __init__(self, num, channel_start = 0, iface = 'veth0', iface_mcast = 'veth2', mcast_cb = None):
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800120 self.num = num
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -0700121 self.channel_start = channel_start
122 self.channels = self.generate(self.num, self.channel_start)
Chetan Gaonkercbe79642016-03-09 17:45:58 -0800123 self.group_channel_map = {}
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -0700124 #assert_equal(len(self.channels), self.num)
Chetan Gaonkercbe79642016-03-09 17:45:58 -0800125 for i in range(self.num):
126 self.group_channel_map[self.channels[i]] = i
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800127 self.state = self.Stopped
128 self.streams = None
129 self.channel_states = {}
130 self.last_chan = None
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800131 self.iface_mcast = iface_mcast
132 self.mcast_cb = mcast_cb
133 for c in range(self.num):
134 self.channel_states[c] = [self.Idle]
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -0700135 IgmpChannel.__init__(self, ssm_list = self.channels, iface=iface)
A R Karthick338268f2016-06-21 17:12:13 -0700136
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -0700137 def generate(self, num, channel_start = 0):
Chetan Gaonker38737f82016-05-11 17:44:17 -0700138 start = (225 << 24) | ( ( (channel_start >> 16) & 0xff) << 16 ) | \
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -0700139 ( ( (channel_start >> 8) & 0xff ) << 8 ) | (channel_start) & 0xff
140 start += channel_start/256 + 1
141 end = start + num
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800142 group_addrs = []
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -0700143 count = 0
144 while count != num:
145 for i in range(start, end):
146 if i&255:
147 g = '%s.%s.%s.%s' %((i>>24) &0xff, (i>>16)&0xff, (i>>8)&0xff, i&0xff)
A R Karthick76a497a2017-04-12 10:59:39 -0700148 log_test.debug('Adding group %s' %g)
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -0700149 group_addrs.append(g)
150 count += 1
151 start = end
152 end = start + 1
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800153 return group_addrs
154
155 def start(self):
156 if self.state == self.Stopped:
157 if self.streams:
158 self.streams.stop()
159 self.streams = McastTraffic(self.channels, iface=self.iface_mcast, cb = self.mcast_cb)
160 self.streams.start()
161 self.state = self.Started
162
163 def join(self, chan = None):
164 if chan is None:
165 chan = random.randint(0, self.num)
166 else:
167 if chan >= self.num:
168 chan = 0
169
170 if self.get_state(chan) == self.Joined:
Chetan Gaonkercbe79642016-03-09 17:45:58 -0800171 return chan, 0
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800172
173 groups = [self.channels[chan]]
Chetan Gaonkercbe79642016-03-09 17:45:58 -0800174 join_start = monotonic.monotonic()
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800175 self.igmp_join(groups)
176 self.set_state(chan, self.Joined)
177 self.last_chan = chan
Chetan Gaonkercbe79642016-03-09 17:45:58 -0800178 return chan, join_start
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800179
180 def leave(self, chan):
181 if chan is None:
182 chan = self.last_chan
183 if chan is None or chan >= self.num:
184 return False
185 if self.get_state(chan) != self.Joined:
186 return False
187 groups = [self.channels[chan]]
188 self.igmp_leave(groups)
189 self.set_state(chan, self.Idle)
190 if chan == self.last_chan:
191 self.last_chan = None
192 return True
A R Karthick338268f2016-06-21 17:12:13 -0700193
A.R Karthick9d914552017-05-18 11:22:57 -0700194 def join_next(self, chan = None, leave_flag = True):
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800195 if chan is None:
196 chan = self.last_chan
197 if chan is None:
198 return None
199 leave = chan
200 join = chan+1
201 else:
202 leave = chan - 1
203 join = chan
A R Karthick338268f2016-06-21 17:12:13 -0700204
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800205 if join >= self.num:
206 join = 0
207
208 if leave >= 0 and leave != join:
A.R Karthick9d914552017-05-18 11:22:57 -0700209 if leave_flag is True:
210 self.leave(leave)
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800211
212 return self.join(join)
213
214 def jump(self):
215 chan = self.last_chan
216 if chan is not None:
217 self.leave(chan)
218 s_next = chan
219 else:
220 s_next = 0
Chetan Gaonkercbe79642016-03-09 17:45:58 -0800221 if self.num - s_next < 2:
222 s_next = 0
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800223 chan = random.randint(s_next, self.num)
224 return self.join(chan)
225
226 def gaddr(self, chan):
Chetan Gaonkercbe79642016-03-09 17:45:58 -0800227 '''Return the group address for a channel'''
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800228 if chan >= self.num:
229 return None
230 return self.channels[chan]
231
Chetan Gaonkercbe79642016-03-09 17:45:58 -0800232 def caddr(self, group):
233 '''Return a channel given a group addr'''
234 if self.group_channel_map.has_key(group):
235 return self.group_channel_map[group]
236 return None
237
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800238 def recv_cb(self, pkt):
239 '''Default channel receive callback'''
A R Karthick76a497a2017-04-12 10:59:39 -0700240 log_test.debug('Received packet from source %s, destination %s' %(pkt[IP].src, pkt[IP].dst))
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800241 send_time = float(pkt[IP].payload.load)
242 recv_time = monotonic.monotonic()
A R Karthick76a497a2017-04-12 10:59:39 -0700243 log_test.debug('Packet received in %.3f usecs' %(recv_time - send_time))
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800244
A R Karthick338268f2016-06-21 17:12:13 -0700245 def recv(self, chan, cb = None, count = 1, timeout = 5):
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800246 if chan is None:
247 return None
248 if type(chan) == type([]) or type(chan) == type(()):
249 channel_list=filter(lambda c: c < self.num, chan)
250 groups = map(lambda c: self.gaddr(c), channel_list)
251 else:
252 groups = (self.gaddr(chan),)
253 if cb is None:
254 cb = self.recv_cb
A R Karthick338268f2016-06-21 17:12:13 -0700255 return sniff(prn = cb, count=count, timeout = timeout,
A.R Karthick3f260212017-05-17 14:37:46 -0700256 lfilter = lambda p: IP in p and p[IP].dst in groups, iface = bytes(self.iface[:15]))
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800257
258 def stop(self):
259 if self.streams:
260 self.streams.stop()
261 self.state = self.Stopped
262
263 def get_state(self, chan):
264 return self.channel_states[chan][0]
265
266 def set_state(self, chan, state):
267 self.channel_states[chan][0] = state
268
269if __name__ == '__main__':
Chetan Gaonkercd86bdd2016-03-17 00:08:12 -0700270 num = 5
271 start = 0
272 ssm_list = []
273 for i in xrange(2):
274 channels = Channels(num, start)
275 ssm_list += channels.channels
276 start += num
277 igmpChannel = IgmpChannel()
278 igmpChannel.igmp_load_ssm_config(ssm_list)
Chetan Gaonkerb424ff82016-03-08 12:11:12 -0800279 channels.start()
280 for i in range(num):
281 channels.join(i)
282 for i in range(num):
283 channels.recv(i)
284 for i in range(num):
285 channels.leave(i)
286 channels.stop()