diff options
Diffstat (limited to 'GPL/traffic_scripts/policer.py')
-rw-r--r-- | GPL/traffic_scripts/policer.py | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/GPL/traffic_scripts/policer.py b/GPL/traffic_scripts/policer.py index 262d7878b6..76b7812687 100644 --- a/GPL/traffic_scripts/policer.py +++ b/GPL/traffic_scripts/policer.py @@ -26,12 +26,11 @@ """Traffic script for IPsec verification.""" import sys -import logging from ipaddress import ip_address from scapy.layers.l2 import Ether from scapy.layers.inet import IP, TCP -from scapy.layers.inet6 import IPv6, ICMPv6ND_NS +from scapy.layers.inet6 import IPv6, ICMPv6ND_NS, ICMPv6MLReport2, ICMPv6ND_RA from scapy.packet import Raw from .TrafficScriptArg import TrafficScriptArg @@ -113,9 +112,17 @@ def main(): if pkt_recv.haslayer(ICMPv6ND_NS): # read another packet in the queue if the current one is ICMPv6ND_NS continue - else: - # otherwise process the current packet - break + elif pkt_recv.haslayer(ICMPv6MLReport2): + # read another packet in the queue if the current one is + # ICMPv6MLReport2 + continue + elif pkt_recv.haslayer(ICMPv6ND_RA): + # read another packet in the queue if the current one is + # ICMPv6ND_RA + continue + + # otherwise process the current packet + break if pkt_recv is None: raise RuntimeError(u"Rx timeout") |