mirror of https://github.com/fluffle/goirc
Merge pull request #48 from Minecrell/local_address
Make local bind address configurable.
This commit is contained in:
commit
f92aa9a402
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue