From 680b1983d0e49592b0750575de11e71b3eb53660 Mon Sep 17 00:00:00 2001
From: lumtis <lucas.bertrand.22@gmail.com>
Date: Fri, 13 Oct 2023 10:23:46 -0700
Subject: [PATCH] timeouts

---
 rpc/websockets.go | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/rpc/websockets.go b/rpc/websockets.go
index 1346aad955..84d334e952 100644
--- a/rpc/websockets.go
+++ b/rpc/websockets.go
@@ -25,6 +25,7 @@ import (
 	"net"
 	"net/http"
 	"sync"
+	"time"
 
 	"github.com/cosmos/cosmos-sdk/client"
 	"github.com/ethereum/go-ethereum/common"
@@ -47,6 +48,13 @@ import (
 
 const (
 	messageSizeLimit = 32 * 1024 * 1024 // 32MB
+
+)
+
+var (
+	readTimeout  = 15 * time.Second // Time to read the request
+	writeTimeout = 15 * time.Second // Time to write the response
+	idleTimeout  = 60 * time.Second // Max time for connections using TCP Keep-Alive
 )
 
 type WebsocketsServer interface {
@@ -111,13 +119,21 @@ func (s *websocketsServer) Start() {
 	ws := mux.NewRouter()
 	ws.Handle("/", s)
 
+	// configuring the HTTP server
+	server := &http.Server{
+		Addr:         s.wsAddr,
+		Handler:      ws,
+		ReadTimeout:  readTimeout,
+		WriteTimeout: writeTimeout,
+		IdleTimeout:  idleTimeout,
+	}
+
 	go func() {
 		var err error
-		/* #nosec G114 -- http functions have no support for timeouts */
 		if s.certFile == "" || s.keyFile == "" {
-			err = http.ListenAndServe(s.wsAddr, ws)
+			err = server.ListenAndServe()
 		} else {
-			err = http.ListenAndServeTLS(s.wsAddr, s.certFile, s.keyFile, ws)
+			err = server.ListenAndServeTLS(s.certFile, s.keyFile)
 		}
 
 		if err != nil {