Merge pull request #48 from Minecrell/local_address

Make local bind address configurable.
This commit is contained in:
Alex Bee 2014-11-17 13:05:57 +00:00
commit f92aa9a402
1 changed files with 22 additions and 2 deletions

View File

@ -31,6 +31,7 @@ type Conn struct {
stRemovers []Remover stRemovers []Remover
// I/O stuff to server // I/O stuff to server
dialer *net.Dialer
sock net.Conn sock net.Conn
io *bufio.ReadWriter io *bufio.ReadWriter
in chan *Line in chan *Line
@ -58,6 +59,9 @@ type Config struct {
SSL bool SSL bool
SSLConfig *tls.Config SSLConfig *tls.Config
// Local address to connect to the server.
LocalAddr string
// Replaceable function to customise the 433 handler's new nick // Replaceable function to customise the 433 handler's new nick
NewNick func(string) string NewNick func(string) string
@ -118,8 +122,24 @@ func Client(cfg *Config) *Conn {
cfg.Me.Ident = "goirc" cfg.Me.Ident = "goirc"
cfg.Me.Name = "Powered by GoIRC" cfg.Me.Name = "Powered by GoIRC"
} }
dialer := new(net.Dialer)
if cfg.LocalAddr != "" {
if !hasPort(cfg.LocalAddr) {
cfg.LocalAddr += ":0"
}
local, err := net.ResolveTCPAddr("tcp", cfg.LocalAddr)
if err == nil {
dialer.LocalAddr = local
} else {
logging.Error("irc.Client(): Cannot resolve local address %s: %s", cfg.LocalAddr, err)
}
}
conn := &Conn{ conn := &Conn{
cfg: cfg, cfg: cfg,
dialer: dialer,
in: make(chan *Line, 32), in: make(chan *Line, 32),
out: make(chan string, 32), out: make(chan string, 32),
intHandlers: handlerSet(), intHandlers: handlerSet(),
@ -208,7 +228,7 @@ func (conn *Conn) Connect() error {
conn.cfg.Server += ":6697" conn.cfg.Server += ":6697"
} }
logging.Info("irc.Connect(): Connecting to %s with SSL.", conn.cfg.Server) 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 { if s, err := tls.DialWithDialer(conn.dialer, "tcp", conn.cfg.Server, conn.cfg.SSLConfig); err == nil {
conn.sock = s conn.sock = s
} else { } else {
return err return err
@ -218,7 +238,7 @@ func (conn *Conn) Connect() error {
conn.cfg.Server += ":6667" conn.cfg.Server += ":6667"
} }
logging.Info("irc.Connect(): Connecting to %s without SSL.", conn.cfg.Server) 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 := conn.dialer.Dial("tcp", conn.cfg.Server); err == nil {
conn.sock = s conn.sock = s
} else { } else {
return err return err