Traffic scripts: Fix docstrings

[csit.git] / GPL / traffic_scripts / lisp / lisp_check.py

diff --git a/GPL/traffic_scripts/lisp/lisp_check.py b/GPL/traffic_scripts/lisp/lisp_check.py
index f7b4c6a..bfe8990 100644 (file)
--- a/GPL/traffic_scripts/lisp/lisp_check.py
+++ b/GPL/traffic_scripts/lisp/lisp_check.py
@@ -34,7 +34,7 @@ from scapy.all import bind_layers, Packet
 from scapy.fields import FlagsField, BitField, IntField
 from scapy.layers.inet import ICMP, IP, UDP
 from scapy.layers.inet6 import ICMPv6EchoRequest
 from scapy.fields import FlagsField, BitField, IntField
 from scapy.layers.inet import ICMP, IP, UDP
 from scapy.layers.inet6 import ICMPv6EchoRequest

-from scapy.layers.inet6 import IPv6, ICMPv6ND_NS, ICMPv6MLReport2
+from scapy.layers.inet6 import IPv6, ICMPv6ND_NS, ICMPv6MLReport2, ICMPv6ND_RA

 from scapy.layers.l2 import Ether
 from scapy.packet import Raw
 
 from scapy.layers.l2 import Ether
 from scapy.packet import Raw
 


@@ -73,7 +73,6 @@ def valid_ipv4(ip_address):
     :type ip_address: str
     :returns: True if IP address is correct.
     :rtype: bool
     :type ip_address: str
     :returns: True if IP address is correct.
     :rtype: bool

-    :raises AttributeError, AddressValueError: If IP address is not valid.

     """
     try:
         ipaddress.IPv4Address(ip_address)
     """
     try:
         ipaddress.IPv4Address(ip_address)


@@ -89,7 +88,6 @@ def valid_ipv6(ip_address):
     :type ip_address: str
     :returns: True if IP address is correct.
     :rtype: bool
     :type ip_address: str
     :returns: True if IP address is correct.
     :rtype: bool

-    :raises AttributeError, AddressValueError: If IP address is not valid.

     """
     try:
         ipaddress.IPv6Address(ip_address)
     """
     try:
         ipaddress.IPv6Address(ip_address)


@@ -166,6 +164,10 @@ def main():
             # read another packet in the queue if the current one is
             # ICMPv6MLReport2
             continue
             # read another packet in the queue if the current one is
             # ICMPv6MLReport2
             continue

+        elif ether.haslayer(ICMPv6ND_RA):
+            # read another packet in the queue if the current one is
+            # ICMPv6ND_RA
+            continue

 
         # otherwise process the current packet
         break
 
         # otherwise process the current packet
         break