在Python中計算IP校驗和

Question

我需要計算IP數據包的校驗和，如http://www.faqs.org/rfcs/rfc1071.html中所述。

我已經下面的代碼：

#!/usr/bin/python 
import struct 

data = "45 00 00 47 73 88 40 00 40 06 a2 c4 83 9f 0e 85 83 9f 0e a1" 

# a test for the checksum calculation 

def _checksum(data): 
    #calculate the header sum 
    ip_header_sum = sum(struct.unpack_from("6H", data)) 
    #add the carry 
    ip_header_sum = (ip_header_sum & 0xFFFF) + (ip_header_sum >> 16 & 0xFFFF) 
    #invert the sum, python does not support inversion (~a is -a + 1) so we have to do 
    #little trick: ~a is the same as 0xFFFF & ~a 
    ip_header_sum = ~ip_header_sum & 0xFFFF 

    return ip_header_sum #should return 0 if correct 

data = data.split() 
data = map(lambda x: int(x,16), data) 
data = struct.pack("%dB" % len(data), *data) 

print " ".join(map(lambda x: "0x%02x" % ord(x), data)) 
print "Checksum: 0x%04x" % _checksum(data) 

它與我所使用Wireshark捕獲和一個包，必須具有正確的校驗和，因此應計算爲0

不幸的是，結果是0x6524。 有趣的是，對於每個正確的數據包，結果始終爲0x6524 ...

誰發現錯誤？

編輯，以使誤差更清晰 * *編輯

Answer 1

您可以直接使用checksum udp calculation python的解決方案，這會導致預期的校驗和值爲零。

import struct 

data = "45 00 00 47 73 88 40 00 40 06 a2 c4 83 9f 0e 85 83 9f 0e a1" 

def carry_around_add(a, b): 
    c = a + b 
    return (c & 0xffff) + (c >> 16) 

def checksum(msg): 
    s = 0 
    for i in range(0, len(msg), 2): 
     w = ord(msg[i]) + (ord(msg[i+1]) << 8) 
     s = carry_around_add(s, w) 
    return ~s & 0xffff 

data = data.split() 
data = map(lambda x: int(x,16), data) 
data = struct.pack("%dB" % len(data), *data) 

print ' '.join('%02X' % ord(x) for x in data) 
print "Checksum: 0x%04x" % checksum(data) 

結果：

45 00 00 47 73 88 40 00 40 06 A2 C4 83 9F 0E 85 83 9F 0E A1 
Checksum: 0x0000 

Answer 2

你有兩個問題在這裏第二次。

首先，您撥打struct.unpack_from的電話只打開緩衝區中的4個16位值（即8個字節）。如果你想解開整個標題，你需要做一些類似struct.unpack_from("!nH")的地方，其中n是你想要解包的短褲數量。假設data只包含IP標頭，則可以使用struct.unpack_from("!%dH"%(len(data)/2), data)生成適當的格式字符串。

其次，一旦你這樣做了，你會發現校驗和現在可以達到0.這是一個已經有校驗和的數據包的正確結果，就像這樣。 （您在上面的數據包中突出顯示了A2和C4字節。）要從頭開始計算數據包的正確校驗和，需要將校驗和字節設置爲0.（請參閱RFC1071中步驟2的開頭：「生成校驗和，校驗和字段本身被清除「）。