chore: Fix potential slowloris attack (GO-S2112)

ikawaha · Oct 16, 2023 · ddcfb29 · ddcfb29
1 parent 08c1152
commit ddcfb29
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cmd/server/cmd.go b/cmd/server/cmd.go
@@ -11,6 +11,7 @@ import (
 	"os"
 	"os/signal"
 	"syscall"
+	"time"
 
 	"github.com/ikawaha/kagome-dict/dict"
 	"github.com/ikawaha/kagome-dict/ipa"
@@ -106,8 +107,9 @@ func command(ctx context.Context, opt *option) error {
 	mux.Handle("/", &TokenizeDemoHandler{tokenizer: t})
 	mux.Handle("/tokenize", &TokenizeHandler{tokenizer: t})
 	srv := http.Server{
-		Addr:    opt.http,
-		Handler: mux,
+		Addr:              opt.http,
+		Handler:           mux,
+		ReadHeaderTimeout: 20 * time.Second,
 	}
 	ch := make(chan error)
 	go func() {