1、最近用python修改完ip之后数据流出现问题,发现为修改完ip,未修改checksum的值,编写次方法留作后续使用
出入修改后ip的数据流,返回正确checksum的数据流
def checksum(new_pkt):
"""
:param new_pkt:pcap 包一条修改完 source_ip,dest_ip 的数据流
:return: 计算好checksum的一条数据流
"""
ip_header = new_pkt[14:34].hex()
checksum = 0
# 反码求和
for i in range(0,37,4):
if i == 20:continue
checksum = checksum + int(ip_header[i:i+4], 16)
checksum = "{:x}".format(checksum)
# 将进位加到低16位上
checksum = '{:016b}'.format(int(checksum[0:1], 16) + int(checksum[1:5], 16))
# 取反码
checksum_fanma = ''
for i in checksum:
if i == '1':
checksum_fanma = checksum_fanma + str(0)
else:
checksum_fanma = checksum_fanma + str(1)
checksum = bytes.fromhex("{:04x}".format(int(checksum_fanma, 2)))
# 拼接成新的数据包返回
new_pkt = new_pkt[0:24] + checksum + new_pkt[26:]
return new_pkt