From 577408a050cd9e76aa9dc544e1702a25558384ee Mon Sep 17 00:00:00 2001
From: Quentin Dufour <quentin@deuxfleurs.fr>
Date: Fri, 9 Aug 2019 18:09:15 +0200
Subject: [PATCH] WIP algo

---
 src/algo_dup2.c | 61 +++++++++++++++++++++++++++++++++----------------
 1 file changed, 41 insertions(+), 20 deletions(-)

diff --git a/src/algo_dup2.c b/src/algo_dup2.c
index e179f79..647a978 100644
--- a/src/algo_dup2.c
+++ b/src/algo_dup2.c
@@ -1,5 +1,6 @@
 #include "proxy.h"
 #include "algo_utils.h"
+#include "packet.h"
 
 struct dup2_ctx {
   uint16_t recv_id;
@@ -19,30 +20,50 @@ int algo_dup2_on_stream(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo
   char url[256];
   struct evt_core_fdinfo *to_fdinfo = NULL;
   struct algo_ctx* app_ctx = fdinfo->cat->app_ctx;
-  union abstract_packet* ap = (union abstract_packet*) &bp->ip;
-
-  // Check that we didn't already received the packet
+  union abstract_packet *ap = (union abstract_packet*) &bp->ip, *ap_prev = NULL;
   struct dup2_ctx* dup2c = app_ctx->misc;
-  if (ring_ge(dup2c->recv_id, ap->fmt.content.clear.id)) {
+  uint16_t id = 0;
+
+  switch (ap->fmt.headers.cmd) {
+  case CMD_UDP_METADATA_THUNDER:
+    id = ap->fmt.content.udp_metadata_thunder.id;
     mv_buffer_rtof(&app_ctx->br, fdinfo);
-    return 0;
+
+    // Check that received identifier has not been delivered
+    if (ring_ge(dup2c->recv_id, id)) {
+      mv_buffer_atof(&app_ctx->br, fdinfo);
+      return 0;
+    }
+
+    // Update delivered identifier
+    dup2c->recv_id = id;
+
+    // 1. Find destination
+    ap_prev = (union abstract_packet*) get_app_buffer(&app_ctx->br, fdinfo)->ip;
+    sprintf(url, "udp:write:127.0.0.1:%d", ap_prev->fmt.content.udp_encapsulated.port);
+    to_fdinfo = evt_core_get_from_url (ctx, url);
+    if (to_fdinfo == NULL) {
+      fprintf(stderr, "No fd for URL %s in tcp-read. Dropping packet :( \n", url);
+      mv_buffer_atof (&app_ctx->br, fdinfo);
+      return 1;
+    }
+
+    // 2. Move buffer
+    mv_buffer_atow (&app_ctx->br, fdinfo, to_fdinfo);
+    main_on_udp_write(ctx, to_fdinfo);
+
+    break;
+
+  case CMD_UDP_ENCAPSULATED:
+    mv_buffer_rtoa(&app_ctx->br, fdinfo, fdinfo);
+    break;
+
+  default:
+    fprintf(stderr, "Unknown packet type\n");
+    break;
   }
 
-  dup2c->recv_id = bp->ip.ap.fmt.content.clear.id;
-
-  // 1. Find destination
-  sprintf(url, "udp:write:127.0.0.1:%d", bp->ip.ap.fmt.content.clear.port);
-  to_fdinfo = evt_core_get_from_url (ctx, url);
-  if (to_fdinfo == NULL) {
-    fprintf(stderr, "No fd for URL %s in tcp-read. Dropping packet :( \n", url);
-    mv_buffer_wtof (&app_ctx->br, fdinfo);
-    return 1;
-  }
-
-  // 2. Move buffer
-  mv_buffer_rtow (&app_ctx->br, fdinfo, to_fdinfo);
-  main_on_udp_write(ctx, to_fdinfo);
-  return 0;
+  return 1;
 }
 
 int algo_dup2_on_datagram(struct evt_core_ctx* ctx, struct evt_core_fdinfo* fdinfo, struct buffer_packet* bp) {