X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Ftraffic_scripts%2Flisp%2Flispgpe_check.py;fp=resources%2Ftraffic_scripts%2Flisp%2Flispgpe_check.py;h=d4de8635d70f93127c1b7c83b2c3dde7ca37affe;hp=7979b6ced9562370d54be9a5f9def2876c4be80d;hb=aafa2efa63b891e85ff13255691aae4909bc503c;hpb=a295b175cd410e63829a71eb4565f4bc0f6965a4

diff --git a/resources/traffic_scripts/lisp/lispgpe_check.py b/resources/traffic_scripts/lisp/lispgpe_check.py
index 7979b6ced9..d4de8635d7 100755
--- a/resources/traffic_scripts/lisp/lispgpe_check.py
+++ b/resources/traffic_scripts/lisp/lispgpe_check.py
@@ -96,7 +96,8 @@ def main():
 
     args = TrafficScriptArg(
         ['tg_src_mac', 'tg_dst_mac', 'src_ip', 'dst_ip', 'dut_if1_mac',
-         'dut_if2_mac', 'src_rloc', 'dst_rloc'])
+         'dut_if2_mac', 'src_rloc', 'dst_rloc'],
+        ['ot_mode'])
 
     tx_src_mac = args.get_arg('tg_src_mac')
     tx_dst_mac = args.get_arg('dut_if1_mac')
@@ -108,6 +109,7 @@ def main():
     dst_rloc = args.get_arg("dst_rloc")
     tx_if = args.get_arg('tx_if')
     rx_if = args.get_arg('rx_if')
+    ot_mode = args.get_arg('ot_mode')
 
     rxq = RxQueue(rx_if)
     txq = TxQueue(tx_if)
@@ -146,9 +148,17 @@ def main():
 
     ip = ether.payload
 
-    if not isinstance(ip, ip_format):
-        raise RuntimeError(
-            "Not an IP packet received {0}".format(ip.__repr__()))
+    if ot_mode == '6to4':
+        if not isinstance(ip, IP):
+            raise RuntimeError(
+                "Not an IP packet received {0}".format(ip.__repr__()))
+    elif ot_mode == '4to6':
+        if not isinstance(ip, IPv6):
+            raise RuntimeError(
+                "Not an IP packet received {0}".format(ip.__repr__()))
+    elif not isinstance(ip, ip_format):
+            raise RuntimeError(
+                "Not an IP packet received {0}".format(ip.__repr__()))
 
     lisp = ether.getlayer(LispGPEHeader).underlayer
     if not lisp: