undo not adding session if its not a SYN packet, so we get correct RS…

…T response
compscidr · Nov 7, 2024 · ecff78c · ecff78c
1 parent e5a177d
commit ecff78c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/src/main/kotlin/com/jasonernst/kanonproxy/KAnonProxy.kt b/src/main/kotlin/com/jasonernst/kanonproxy/KAnonProxy.kt
@@ -171,14 +171,14 @@ class KAnonProxy(
                 transportHeader.destinationPort,
                 ipHeader.protocol,
             )
-        if (!sessionTableBySessionKey.containsKey(key)) {
-            if (transportHeader is TcpHeader) {
-                if (transportHeader.isSyn().not()) {
-                    logger.warn("non-syn packet for new session: {}, ignoring", key)
-                    return
-                }
-            }
-        }
+//        if (!sessionTableBySessionKey.containsKey(key)) {
+//            if (transportHeader is TcpHeader) {
+//                if (transportHeader.isSyn().not()) {
+//                    logger.warn("non-syn packet for new session: {}, ignoring", key)
+//                    return
+//                }
+//            }
+//        }
         val session =
             sessionTableBySessionKey.getOrPut(key) {
                 isNewSession = true

diff --git a/src/test/kotlin/com/jasonernst/kanonproxy/tcp/TcpHandlingTest.kt b/src/test/kotlin/com/jasonernst/kanonproxy/tcp/TcpHandlingTest.kt
@@ -23,7 +23,7 @@ import java.net.SocketException
 import java.nio.ByteBuffer
 
 // this needs to be set to 250 if we want to test the TIME_WAIT state
-@Timeout(60)
+@Timeout(20)
 class TcpHandlingTest {
     private val logger = LoggerFactory.getLogger(javaClass)
     private val kAnonProxy = KAnonProxy(IcmpLinux, mockk(relaxed = true))