commit 4b853008dff47170dcfb3bcd9aeba266d81a9f29
author Chetan Gaonker <chetang@calsoftlabs.com> Mon Mar 28 15:17:28 2016 -0700
committer Chetan Gaonker <chetang@calsoftlabs.com> Mon Mar 28 15:17:28 2016 -0700
tree 1b0fd25db6c17f3d40e8a72edfa88a6d83561b17
parent 4eb1207648d53d161c17fe5d7779546e5c07fbf1

Add src_ip/src_mac support to mcast traffic generator class

src/test/utils/McastTraffic.py

diff --git a/src/test/utils/McastTraffic.py b/src/test/utils/McastTraffic.py
index 2bd28e2..7ab79a7 100644
--- a/src/test/utils/McastTraffic.py
+++ b/src/test/utils/McastTraffic.py
@@ -6,16 +6,20 @@
 from scapy.all import *
 
 class McastTraffic(threading.Thread):
-
-    dst_mac = '01:00:5e:00:01:01'
-    src_mac = '02:88:b4:e4:90:77'
-    src_ip = '1.2.3.4'
+    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', cb = None, arg = None):
+
+    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
@@ -46,4 +50,3 @@
     def isSendStopped(self):
         return False if self.state & self.SEND_STATE else True
 
-