1 files changed, 14 insertions, 8 deletions
diff --git a/src/psock.h b/src/psock.h
index 5faf24b..da2be7b 100644
--- a/src/psock.h
+++ b/src/psock.h
@@ -1,22 +1,29 @@
 #ifndef PSOCK_H
 #define PSOCK_H 1
 
+#include <netinet/in.h>
 #include <stdint.h>
 #include <stdlib.h>
 
-
+struct icmphdr;
 struct pdesc;
+struct ppkt;
 
-struct psock
-{
+struct psock {
     int epoll_fd;
     int icmp_fd;
 
     struct {
-        size_t used;
-        size_t max;
-        uint8_t * buffer;
-    } packet;
+        struct sockaddr_storage peer;
+
+        struct {
+            size_t used;
+            size_t max;
+            uint8_t * buffer;
+            struct icmphdr * icmphdr;
+            struct ppkt * ppkt;
+        } packet;
+    } current;
 
     struct {
         size_t used;
@@ -25,7 +32,6 @@ struct psock
     } remotes;
 };
 
-
 int psock_init(struct psock *, size_t, size_t);
 
 void psock_free(struct psock *);