Fix server open connection leaks

elastic · Jan 4, 2021 · a53f084 · a53f084
1 parent 58f5edc
commit a53f084
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/server/server.go b/server/server.go
@@ -236,11 +236,13 @@ func (s *server) handle(client net.Conn) {
 	sig := make(chan struct{})
 
 	go func() {
+		defer close(sig)
+
 		var buf [1]byte
 		if _, err := io.ReadFull(client, buf[:]); err != nil {
+			client.Close()
 			return
 		}
-		close(sig)
 
 		for _, m := range s.mux {
 			if m.mux != buf[0] {