client: Add read timeout for initial greeting

emersion · Jul 5, 2021 · 26eb481 · 26eb481
1 parent c11900f
commit 26eb481
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/client.go b/client.go
@@ -91,6 +91,10 @@ func NewClient(conn net.Conn, host string) (*Client, error) {
 
 	c.setConn(conn)
 
+	// Initial greeting timeout. RFC 5321 recommends 5 minutes.
+	c.conn.SetDeadline(time.Now().Add(5 * time.Minute))
+	defer c.conn.SetDeadline(time.Time{})
+
 	_, _, err := c.Text.ReadResponse(220)
 	if err != nil {
 		c.Text.Close()