From f8f428c17d500364855ba922ea88f61a4dd12c9e Mon Sep 17 00:00:00 2001
From: Christian Deacon <christianmdeacon@gmail.com>
Date: Fri, 19 Apr 2024 16:41:40 -0400
Subject: [PATCH] Add support for establishing a single connection when using
 cooked sockets.

---
 src/sequence.c | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/sequence.c b/src/sequence.c
index 35bdd30..ce5f407 100644
--- a/src/sequence.c
+++ b/src/sequence.c
@@ -476,6 +476,23 @@ void *thread_hdl(void *temp)
     tcpsin.sin_port = htons(ti->seq.tcp.dst_port);
     memset(&tcpsin.sin_zero, 0, sizeof(tcpsin.sin_zero));
 
+    // If we're using one connection for a cooked socket, open and connect socket now.
+    if (ti->seq.tcp.use_socket && ti->seq.tcp.one_connection) {
+        if ((sock_fd = socket(sock_domain, sock_type, sock_proto)) < 0)
+        {
+            fprintf(stderr, "ERROR - Cannot setup TCP cook socket :: %s.\n", strerror(errno));
+
+            pthread_exit(NULL);
+        }
+
+        if (connect(sock_fd, (struct sockaddr *)&tcpsin, sizeof(tcpsin)) != 0)
+        {
+            fprintf(stderr, "ERROR - Cannot connect to destination using cooked sockets :: %s.\n", strerror(errno));
+
+            pthread_exit(NULL);
+        }
+    }
+
     // Loop.
     while (1)
     {
@@ -621,7 +638,7 @@ void *thread_hdl(void *temp)
                 tcph->check = csum_tcpudp_magic(iph->saddr, iph->daddr, (tcph->doff * 4) + data_len, IPPROTO_TCP, csum_partial(tcph, (tcph->doff * 4) + data_len, 0));   
             }
 
-            if (ti->seq.tcp.use_socket)
+            if (ti->seq.tcp.use_socket && !ti->seq.tcp.one_connection)
             {
                 if ((sock_fd = socket(sock_domain, sock_type, sock_proto)) < 0)
                 {