server: fix closing when the tls handshake fails

mookums · Oct 19, 2024 · b169578 · b169578
1 parent 27c4d37
commit b169578
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/core/server.zig b/src/core/server.zig
@@ -549,7 +549,17 @@ pub fn Server(
                 log.debug("processing handshake", .{});
                 handshake_job.count += 1;
 
-                if (length < 0 or handshake_job.count >= 50) {
+                if (length <= 0) {
+                    log.debug("handshake connection closed", .{});
+                    try rt.net.close(.{
+                        .fd = p.socket,
+                        .func = close_task,
+                        .ctx = p,
+                    });
+                    return error.TLSHandshakeClosed;
+                }
+
+                if (handshake_job.count >= 50) {
                     log.debug("handshake taken too many cycles", .{});
                     try rt.net.close(.{
                         .fd = p.socket,