blob: c25cb1a57ccf31d302968371785b880c4ab1d8ed [file] [log] [blame]
#
# Copyright 2016-present Ciena 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 threading
import sys
import os
import time
import monotonic
from scapy.all import *
class McastTraffic(threading.Thread):
DST_MAC_DEFAULT = '01:00:5e:00:01:01'
SRC_MAC_DEFAULT = '02:88:b4:e4:90:77'
SRC_IP_DEFAULT = '1.2.3.4'
SEND_STATE = 1
RECV_STATE = 2
def __init__(self, addrs, iface = 'eth0', dst_mac = DST_MAC_DEFAULT, src_mac = SRC_MAC_DEFAULT,
src_ip = SRC_IP_DEFAULT, cb = None, arg = None):
threading.Thread.__init__(self)
self.addrs = addrs
self.iface = iface
self.dst_mac = dst_mac
self.src_mac = src_mac
self.src_ip = src_ip
self.cb = cb
self.arg = arg
self.state = self.SEND_STATE | self.RECV_STATE
def run(self):
eth = Ether(dst = self.dst_mac, src = self.src_mac)
while self.state & self.SEND_STATE:
for addr in self.addrs:
#data = repr(time.time())
data = repr(monotonic.monotonic())
ip = IP(dst = addr, src = self.src_ip)
sendp(eth/ip/data, iface = self.iface)
if self.cb:
self.cb(self.arg)
def stop(self):
self.state = 0
def stopReceives(self):
self.state &= ~self.RECV_STATE
def stopSends(self):
self.state &= ~self.SEND_STATE
def isRecvStopped(self):
return False if self.state & self.RECV_STATE else True
def isSendStopped(self):
return False if self.state & self.SEND_STATE else True