fourth smth

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>

1 files changed, 41 insertions, 1 deletions

diff --git a/src/ppkt.c b/src/ppkt.c
index cb01b79..9496187 100644
--- a/src/ppkt.c
+++ b/src/ppkt.c
@@ -1,6 +1,46 @@
 #include "ppkt.h"
+#include "psock.h"
 
+#include <netinet/ip.h>
+#include <netinet/ip_icmp.h>
 
-void ppkt_process_icmp(struct psock * psock)
+int ppkt_process_icmp(struct psock * psock)
 {
+    if (psock->current.peer.ss_family == AF_INET) {
+        psock->current.packet.icmphdr = (struct icmphdr *)(psock->current.packet.buffer + sizeof(struct iphdr));
+
+        psock->current.packet.icmphdr->checksum = ntohs(psock->current.packet.icmphdr->checksum);
+        psock->current.packet.icmphdr->un.echo.id = ntohs(psock->current.packet.icmphdr->un.echo.id);
+        psock->current.packet.icmphdr->un.echo.sequence = ntohs(psock->current.packet.icmphdr->un.echo.sequence);
+    } else {
+        return -1;
+    }
+
+    return 0;
+}
+
+int ppkt_process_ppkt(struct psock * psock)
+{
+    if (psock->current.peer.ss_family == AF_INET) {
+        if (psock->current.packet.used < sizeof(struct iphdr) + sizeof(*psock->current.packet.icmphdr)) {
+            return -1;
+        }
+
+        psock->current.packet.ppkt = (struct ppkt *)(psock->current.packet.buffer + sizeof(struct iphdr) +
+                                                     sizeof(*psock->current.packet.icmphdr));
+    } else {
+        return -1;
+    }
+
+    psock->current.packet.ppkt->ident = ntohl(psock->current.packet.ppkt->ident);
+    if (psock->current.packet.ppkt->ident != PTUNNEL_IDENT) {
+        return -1;
+    }
+
+    psock->current.packet.ppkt->total_size = ntohs(psock->current.packet.ppkt->total_size);
+    if (psock->current.packet.ppkt->total_size > psock->current.packet.used) {
+        return -1;
+    }
+
+    return 0;
 }