X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Ftraffic_scripts%2Fipv4_ping_ttl_check.py;h=99a7b0a5f05c5eff026d2cc0b94da9bbe499f537;hp=e6f5637cf93c90a4fcbbef047d7cbf12d1807e33;hb=84c4bb50e13685176751e90bd67ce02730061032;hpb=bc4d98a34af3f26a1c9264758186189f15513a7e;ds=sidebyside diff --git a/resources/traffic_scripts/ipv4_ping_ttl_check.py b/resources/traffic_scripts/ipv4_ping_ttl_check.py index e6f5637cf9..99a7b0a5f0 100755 --- a/resources/traffic_scripts/ipv4_ping_ttl_check.py +++ b/resources/traffic_scripts/ipv4_ping_ttl_check.py @@ -15,7 +15,7 @@ from scapy.all import Ether, IP, ICMP from resources.libraries.python.PacketVerifier \ - import Interface, create_gratuitous_arp_request, auto_pad + import Interface, create_gratuitous_arp_request, auto_pad, checksum_equal from resources.libraries.python.TrafficScriptArg import TrafficScriptArg @@ -117,7 +117,7 @@ def main(): chksum = copy[IP].chksum del copy[IP].chksum tmp = IP(str(copy[IP])) - if tmp.chksum != chksum: + if not checksum_equal(tmp.chksum, chksum): raise RuntimeError('Received IPv4 packet contains invalid checksum, ' '{} instead of {}'.format(chksum, tmp.chksum)) @@ -131,7 +131,7 @@ def main(): chksum = copy[IP][ICMP].chksum del copy[IP][ICMP].chksum tmp = ICMP(str(copy[IP][ICMP])) - if tmp.chksum != chksum: + if not checksum_equal(tmp.chksum, chksum): raise RuntimeError('Received ICMP packet contains invalid checksum, ' '{} instead of {}'.format(chksum, tmp.chksum))