1+ # noqa: INP001
2+
3+
14def icmp_checksum (message : bytes , ** kwargs : object ) -> int :
25 first_arg = kwargs .get ("Tag'First .. Checksum'First - 1" )
36 assert isinstance (first_arg , tuple )
47 tag_first , checksum_first_minus_one = first_arg
5- assert tag_first == 0 and checksum_first_minus_one == 15
8+ assert tag_first == 0
9+ assert checksum_first_minus_one == 15
610 second_arg = kwargs .get ("Checksum'Last + 1 .. Message'Last" )
711 assert isinstance (second_arg , tuple )
812 checksum_last_plus_one , data_last = second_arg
9- assert checksum_last_plus_one == 32 and data_last == 511
13+ assert checksum_last_plus_one == 32
14+ assert data_last == 511
1015
1116 checksum_bytes = message [tag_first : (checksum_first_minus_one + 1 ) // 8 ]
1217 checksum_bytes += b"\x00 " * 2
@@ -18,7 +23,8 @@ def ip_header_checksum(message: bytes, **kwargs: object) -> int:
1823 first_arg = kwargs .get ("Version'First .. Header_Checksum'First - 1" )
1924 assert isinstance (first_arg , tuple )
2025 version_first , checksum_first_minus_one = first_arg
21- assert version_first == 0 and checksum_first_minus_one == 79
26+ assert version_first == 0
27+ assert checksum_first_minus_one == 79
2228 second_arg = kwargs .get ("Header_Checksum'Last + 1 .. Payload'First - 1" )
2329 assert isinstance (second_arg , tuple )
2430 checksum_last_plus_one , options_last = second_arg
@@ -42,7 +48,8 @@ def add_ones_complement(num1: int, num2: int) -> int:
4248 intermediary_result = message_in_sixteen_bit_chunks [0 ]
4349 for i in range (1 , len (message_in_sixteen_bit_chunks )):
4450 intermediary_result = add_ones_complement (
45- intermediary_result , message_in_sixteen_bit_chunks [i ]
51+ intermediary_result ,
52+ message_in_sixteen_bit_chunks [i ],
4653 )
4754
4855 return intermediary_result ^ 0xFFFF
0 commit comments