added timeout.read & timeout.write flags

Author: alash3al

main.go

@@ -15,7 +15,8 @@ import (
 
 func main() {
 	cfg := smtpsrv.ServerConfig{
-		ReadTimeout:     60 * time.Second,
+		ReadTimeout:     time.Duration(*flagReadTimeout) * time.Second,
+		WriteTimeout:    time.Duration(*flagWriteTimeout) * time.Second,
 		ListenAddr:      *flagListenAddr,
 		MaxMessageBytes: int(*flagMaxMessageSize),
 		BannerDomain:    *flagServerName,

vars.go

@@ -7,6 +7,8 @@ var (
 	flagListenAddr     = flag.String("listen", ":smtp", "the smtp address to listen on")
 	flagWebhook        = flag.String("webhook", "http://localhost:8080/my/webhook", "the webhook to send the data to")
 	flagMaxMessageSize = flag.Int64("msglimit", 1024*1024*2, "maximum incoming message size")
+	flagReadTimeout    = flag.Int("timeout.read", 5, "the read timeout in seconds")
+	flagWriteTimeout   = flag.Int("timeout.write", 5, "the write timeout in seconds")
 )
 
 func init() {