From fe961811baa5fa03be56a4ab630d219551626577 Mon Sep 17 00:00:00 2001
From: dowright <dowright@nationofnobodies.com>
Date: Tue, 24 Aug 2021 18:39:03 -0700
Subject: [PATCH] work on new packet

---
 new_packet.pyx | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/new_packet.pyx b/new_packet.pyx
index 2e2ed07..32e1861 100644
--- a/new_packet.pyx
+++ b/new_packet.pyx
@@ -96,12 +96,12 @@ cdef class CPacket:
 
         self.ip_header = <iphdr*>self.data
 
-        cdef u_int8_t iphdr_len = (ip_header.tos & 15) * 4
+        cdef u_int8_t iphdr_len = (self.ip_header.tos & 15) * 4
 
         cdef u_int8_t tcphdr_len
         cdef u_int8_t udphdr_len
 
-        if (ip_header.protocol == IPPROTO_TCP):
+        if (self.ip_header.protocol == IPPROTO_TCP):
 
             self.tcp_header = <tcphdr*>self.data[iphdr_len]
 
@@ -109,7 +109,7 @@ cdef class CPacket:
 
             self.payload = self.data[iphdr_len+tcphdr_len:self.data_len]
 
-        elif (ip_header.protocol == IPPROTO_UDP):
+        elif (self.ip_header.protocol == IPPROTO_UDP):
 
             self.udp_header = <udphdr*>self.data[iphdr_len]
 
@@ -117,7 +117,7 @@ cdef class CPacket:
 
             self.payload = self.data[iphdr_len + udphdr_len:self.data_len]
 
-        elif (ip_header.protocol == IPPROTO_ICMP):
+        elif (self.ip_header.protocol == IPPROTO_ICMP):
 
             self.icmp_header = <icmphdr*>self.data[iphdr_len]