From 6d031c48bc69c2ed93b439de3aa61310a8b8e3c1 Mon Sep 17 00:00:00 2001 From: dowright Date: Tue, 24 Aug 2021 18:45:12 -0700 Subject: [PATCH] work on new packet --- new_packet.pyx | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/new_packet.pyx b/new_packet.pyx index 32e1861..1711ddf 100644 --- a/new_packet.pyx +++ b/new_packet.pyx @@ -100,22 +100,25 @@ cdef class CPacket: cdef u_int8_t tcphdr_len cdef u_int8_t udphdr_len + cdef u_int8_t cmbhdr_len if (self.ip_header.protocol == IPPROTO_TCP): self.tcp_header = self.data[iphdr_len] tcphdr_len = (self.tcp_header.th_off & 15) * 4 + cmbhdr_len = iphdr_len + tcphdr_len - self.payload = self.data[iphdr_len+tcphdr_len:self.data_len] + self.payload = self.data[cmbhdr_len:self.data_len] elif (self.ip_header.protocol == IPPROTO_UDP): self.udp_header = self.data[iphdr_len] udphdr_len = 8 + cmbhdr_len = iphdr_len + udphdr_len - self.payload = self.data[iphdr_len + udphdr_len:self.data_len] + self.payload = self.data[cmbhdr_len:self.data_len] elif (self.ip_header.protocol == IPPROTO_ICMP):