From 39a7da0a370767a27c08a1bce705eb978b160670 Mon Sep 17 00:00:00 2001
From: kyle <kywelsh@gmail.com>
Date: Sat, 20 Dec 2014 15:23:57 +0200
Subject: [PATCH] Added timeout to deal with connecting to slow and  unreliable
 irc servers

---
 client/connection.go | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/client/connection.go b/client/connection.go
index 2775933..b216ac1 100644
--- a/client/connection.go
+++ b/client/connection.go
@@ -79,6 +79,9 @@ type Config struct {
 	// Split PRIVMSGs, NOTICEs and CTCPs longer than
 	// SplitLen characters over multiple lines.
 	SplitLen int
+
+	// Timeout, The amount of time in seconds until a timeout is triggered.
+	Timeout time.Duration
 }
 
 func NewConfig(nick string, args ...string) *Config {
@@ -88,6 +91,7 @@ func NewConfig(nick string, args ...string) *Config {
 		NewNick:  func(s string) string { return s + "_" },
 		Recover:  (*Conn).LogPanic, // in dispatch.go
 		SplitLen: 450,
+		Timeout: 60,
 	}
 	cfg.Me.Ident = "goirc"
 	if len(args) > 0 && args[0] != "" {
@@ -207,8 +211,14 @@ func (conn *Conn) Connect() error {
 		if !hasPort(conn.cfg.Server) {
 			conn.cfg.Server += ":6697"
 		}
+		if &conn.cfg.Timeout != nil{
+			conn.cfg.Timeout = (60 * time.Second)
+		}
 		logging.Info("irc.Connect(): Connecting to %s with SSL.", conn.cfg.Server)
-		if s, err := tls.Dial("tcp", conn.cfg.Server, conn.cfg.SSLConfig); err == nil {
+		dialer := &net.Dialer{
+                Timeout : conn.cfg.Timeout,
+         }
+		if s, err := tls.DialWithDialer(dialer, "tcp", conn.cfg.Server, conn.cfg.SSLConfig); err == nil {
 			conn.sock = s
 		} else {
 			return err
@@ -217,8 +227,11 @@ func (conn *Conn) Connect() error {
 		if !hasPort(conn.cfg.Server) {
 			conn.cfg.Server += ":6667"
 		}
+		if &conn.cfg.Timeout != nil{
+			conn.cfg.Timeout = (60 * time.Second)
+		}
 		logging.Info("irc.Connect(): Connecting to %s without SSL.", conn.cfg.Server)
-		if s, err := net.Dial("tcp", conn.cfg.Server); err == nil {
+		if s, err := net.DialTimeout("tcp", conn.cfg.Server, conn.cfg.Timeout); err == nil {
 			conn.sock = s
 		} else {
 			return err