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

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

diff --git a/new_packet.pyx b/new_packet.pyx
index 61f238b..29f56c6 100644
--- a/new_packet.pyx
+++ b/new_packet.pyx
@@ -48,7 +48,7 @@ cdef class CPacket:
         self._verdict_is_set = False
         self._mark = 0
 
-        self.payload = 0
+        # self.payload = 0
 
     # def __str__(self):
     #     cdef iphdr *hdr = <iphdr*>self.payload
@@ -109,7 +109,7 @@ cdef class CPacket:
 
             self.payload = self.data[iphdr_len+tcphdr_len:self.data_len]
 
-        if (ip_header.protocol == IPPROTO_UDP):
+        elif (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]
 
-        if (ip_header.protocol == IPPROTO_ICMP):
+        elif (ip_header.protocol == IPPROTO_ICMP):
 
             self.icmp_header = <icmphdr*>self.data[iphdr_len]
 
@@ -200,6 +200,11 @@ cdef class CPacket:
                 self.icmp_header.type,
             )
 
+        else:
+            proto_header = (
+                cdef int 0,
+            )
+
         return proto_header
 
     def get_payload(self):