TCP: Add back support for listening on a socket

1 files changed, 14 insertions, 2 deletions

diff --git a/kernel/network/tcp.c b/kernel/network/tcp.c
index 6bec93e..a2df3cd 100644
--- a/kernel/network/tcp.c
+++ b/kernel/network/tcp.c
@@ -224,9 +224,15 @@ void handle_tcp(ipv4_t src_ip, ipv4_t dst_ip, const u8 *payload,
   u32 seq_num = htonl(n_seq_num);
   u32 ack_num = htonl(n_ack_num);
   u16 window_size = htons(n_window_size);
-  struct TcpConnection *con =
-      tcp_find_connection(src_ip, src_port, dst_ip, dst_port);
+
+  struct TcpConnection *con = NULL;
+  if (SYN == flags) {
+    con = tcp_connect_to_listen(src_ip, src_port, dst_ip, dst_port);
+  } else {
+    con = tcp_find_connection(src_ip, src_port, dst_ip, dst_port);
+  }
   if (!con) {
+    kprintf("Unable to find connection for port: %d\n", dst_port);
     return;
   }
 
@@ -321,6 +327,12 @@ void handle_tcp(ipv4_t src_ip, ipv4_t dst_ip, const u8 *payload,
     }
     break;
   }
+  case TCP_STATE_FIN_WAIT2: {
+    if (FIN & flags) {
+      tcp_send_empty_payload(con, ACK);
+    }
+    break;
+  }
   case TCP_STATE_CLOSE_WAIT: {
     // Waiting for this machine to close the connection. There is
     // nothing to respond with.