X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=GPL%2Ftraffic_scripts%2Fsend_ip_check_headers.py;h=a13b1b73b89097b1afe89bfc59731010a0564b7b;hb=0ac6c584b2ff05aa123080c20f7b8071d4a28b1b;hp=0ecebbe3984c577fd16d741c51964517693ce846;hpb=b67b82c15d7d0da402df180b0aaab3daa4cfc76c;p=csit.git diff --git a/GPL/traffic_scripts/send_ip_check_headers.py b/GPL/traffic_scripts/send_ip_check_headers.py index 0ecebbe398..a13b1b73b8 100644 --- a/GPL/traffic_scripts/send_ip_check_headers.py +++ b/GPL/traffic_scripts/send_ip_check_headers.py @@ -30,8 +30,6 @@ MAC addresses are checked in received packet. import sys -import ipaddress - from robot.api import logger from scapy.layers.inet import IP from scapy.layers.inet6 import IPv6, ICMPv6ND_NS, ICMPv6MLReport2, ICMPv6ND_RA @@ -40,39 +38,7 @@ from scapy.packet import Raw from .PacketVerifier import RxQueue, TxQueue from .TrafficScriptArg import TrafficScriptArg - - -def valid_ipv4(ip_address): - """Check IPv4 address. - - :param ip_address: IPv4 address to check. - :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) - return True - except (AttributeError, ipaddress.AddressValueError): - return False - - -def valid_ipv6(ip_address): - """Check IPv6 address. - - :param ip_address: IPv6 address to check. - :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) - return True - except (AttributeError, ipaddress.AddressValueError): - return False - +from .ValidIp import valid_ipv4, valid_ipv6 def main(): """Send IP/IPv6 packet from one traffic generator interface to the other."""