lib/packet/tcp: use checksum_ip()

Signed-off-by: Isaku Yamahata <yamahata@valinux.co.jp>
Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
This commit is contained in:
Isaku Yamahata 2013-03-29 12:51:02 +09:00 committed by FUJITA Tomonori
parent 4053cda285
commit e1414b7fbe

View File

@ -17,7 +17,6 @@ import struct
from . import packet_base
from . import packet_utils
import ipv4
class tcp(packet_base.PacketBase):
@ -66,12 +65,6 @@ class tcp(packet_base.PacketBase):
if self.csum == 0:
length = self.length + len(payload)
if prev.version == 4:
ph = struct.pack('!IIBBH', prev.src, prev.dst, 0, 6, length)
elif prev.version == 6:
ph = struct.pack('!16s16sBBH', prev.src, prev.dst, 0, 6,
length)
f = ph + h + payload
self.csum = packet_utils.checksum(f)
self.csum = packet_utils.checksum_ip(prev, length, h + payload)
struct.pack_into('!H', h, 16, self.csum)
return h