@@ -5,6 +5,7 @@ import (
 	"io"
 	"log"
 	"net/http"
+	"time"
 
 	"github.com/stackrox/rox/pkg/sync"
 )
@@ -69,14 +70,24 @@ func rootHandler(w http.ResponseWriter, r *http.Request) {
 }
 
 func tlsServer() {
-	err := http.ListenAndServeTLS(":8443", "/tmp/certs/server.crt", "/tmp/certs/server.key", http.HandlerFunc(rootHandler))
+	server := &http.Server{
+		ReadHeaderTimeout: 5 * time.Second,
+		Addr:              ":8443",
+		Handler:           http.HandlerFunc(rootHandler),
+	}
+	err := server.ListenAndServeTLS("/tmp/certs/server.crt", "/tmp/certs/server.key")
 	if err != nil {
 		log.Fatal("ListenAndServe: ", err)
 	}
 }
 
 func nonTLSServer() {
-	if err := http.ListenAndServe(":8080", http.HandlerFunc(rootHandler)); err != nil {
+	server := &http.Server{
+		ReadHeaderTimeout: 5 * time.Second,
+		Addr:              ":8080",
+		Handler:           http.HandlerFunc(rootHandler),
+	}
+	if err := server.ListenAndServe(); err != nil {
 		panic(err)
 	}
 }
Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@ import (`
`5`	`5`	`"io"`
`6`	`6`	`"log"`
`7`	`7`	`"net/http"`
	`8`	`+ "time"`
`8`	`9`
`9`	`10`	`"github.com/stackrox/rox/pkg/sync"`
`10`	`11`	`)`
`@@ -69,14 +70,24 @@ func rootHandler(w http.ResponseWriter, r *http.Request) {`
`69`	`70`	`}`
`70`	`71`
`71`	`72`	`func tlsServer() {`
`72`		`- err := http.ListenAndServeTLS(":8443", "/tmp/certs/server.crt", "/tmp/certs/server.key", http.HandlerFunc(rootHandler))`
	`73`	`+ server := &http.Server{`
	`74`	`+ ReadHeaderTimeout: 5 * time.Second,`
	`75`	`+ Addr: ":8443",`
	`76`	`+ Handler: http.HandlerFunc(rootHandler),`
	`77`	`+ }`
	`78`	`+ err := server.ListenAndServeTLS("/tmp/certs/server.crt", "/tmp/certs/server.key")`
`73`	`79`	`if err != nil {`
`74`	`80`	`log.Fatal("ListenAndServe: ", err)`
`75`	`81`	`}`
`76`	`82`	`}`
`77`	`83`
`78`	`84`	`func nonTLSServer() {`
`79`		`- if err := http.ListenAndServe(":8080", http.HandlerFunc(rootHandler)); err != nil {`
	`85`	`+ server := &http.Server{`
	`86`	`+ ReadHeaderTimeout: 5 * time.Second,`
	`87`	`+ Addr: ":8080",`
	`88`	`+ Handler: http.HandlerFunc(rootHandler),`
	`89`	`+ }`
	`90`	`+ if err := server.ListenAndServe(); err != nil {`
`80`	`91`	`panic(err)`
`81`	`92`	`}`
`82`	`93`	`}`