Correctly calculate payload length

This commit is contained in:
Neale Pickett 2007-12-12 12:13:11 -07:00
parent 3232991bc8
commit 5282669973
1 changed files with 9 additions and 8 deletions

View File

@ -47,11 +47,11 @@ class Frame:
self.th_sum,
self.th_urp,
p) = unpack("!HHLLBBHHH", p)
(th_off, th_x2) = unpack_nybbles(x2off)
opt_length = th_off * 4
(self.th_off, th_x2) = unpack_nybbles(x2off)
opt_length = self.th_off * 4
self.th_options = p[20:opt_length]
payload = p[opt_length:self.tot_len - 40]
self.th_options, p = p[:opt_length - 20], p[opt_length - 20:]
payload = p[:self.tot_len - opt_length - 20]
self.src = (self.saddr, self.th_sport)
self.dst = (self.daddr, self.th_dport)
@ -120,11 +120,11 @@ class TCP_Session:
while True:
p = self.pc.read()
if not p:
return
raise EOFError()
try:
return Frame(p[1])
except ValueError:
raise
pass
def read_handshake(self):
# Read SYN
@ -154,8 +154,9 @@ class TCP_Session:
def __iter__(self):
while True:
try:
pkt = self.read_packet()
if not pkt:
except EOFError:
return
self.frames += 1