9 "github.com/bytom/p2p/upnp"
10 log "github.com/sirupsen/logrus"
11 cmn "github.com/tendermint/tmlibs/common"
12 tlog "github.com/tendermint/tmlibs/log"
15 type Listener interface {
16 Connections() <-chan net.Conn
17 InternalAddress() *NetAddress
18 ExternalAddress() *NetAddress
23 // Implements Listener
24 type DefaultListener struct {
30 connections chan net.Conn
34 numBufferedConnections = 10
35 defaultExternalPort = 8770
39 func splitHostPort(addr string) (host string, port int) {
40 host, portStr, err := net.SplitHostPort(addr)
44 port, err = strconv.Atoi(portStr)
51 // skipUPNP: If true, does not try getUPNPExternalAddress()
52 func NewDefaultListener(protocol string, lAddr string, skipUPNP bool, logger tlog.Logger) (Listener, bool) {
53 // Local listen IP & port
54 lAddrIP, lAddrPort := splitHostPort(lAddr)
57 var listener net.Listener
60 for i := 0; i < tryListenSeconds; i++ {
61 listener, err = net.Listen(protocol, lAddr)
64 } else if i < tryListenSeconds-1 {
65 time.Sleep(time.Second * 1)
71 // Actual listener local IP & port
72 listenerIP, listenerPort := splitHostPort(listener.Addr().String())
73 log.WithFields(log.Fields{
76 }).Info("Local listener")
78 // Determine internal address...
79 var intAddr *NetAddress
80 intAddr, err = NewNetAddressString(lAddr)
85 // Determine external address...
86 var extAddr *NetAddress
88 // If the lAddrIP is INADDR_ANY, try UPnP
89 if lAddrIP == "" || lAddrIP == "0.0.0.0" {
90 extAddr = getUPNPExternalAddress(lAddrPort, listenerPort)
97 if address := GetIP([]string{}, time.Duration(0)); address.Success == true {
98 extAddr = NewNetAddressIPPort(net.ParseIP(address.Ip), uint16(lAddrPort))
101 // Otherwise just use the local address...
103 extAddr = getNaiveExternalAddress(listenerPort)
106 cmn.PanicCrisis("Could not determine external address!")
109 dl := &DefaultListener{
113 connections: make(chan net.Conn, numBufferedConnections),
115 dl.BaseService = *cmn.NewBaseService(logger, "DefaultListener", dl)
116 dl.Start() // Started upon construction
120 func (l *DefaultListener) OnStart() error {
121 l.BaseService.OnStart()
126 func (l *DefaultListener) OnStop() {
127 l.BaseService.OnStop()
131 // Accept connections and pass on the channel
132 func (l *DefaultListener) listenRoutine() {
134 conn, err := l.listener.Accept()
137 break // Go to cleanup
140 // listener wasn't stopped,
141 // yet we encountered an error.
146 l.connections <- conn
151 for _ = range l.connections {
156 // A channel of inbound connections.
157 // It gets closed when the listener closes.
158 func (l *DefaultListener) Connections() <-chan net.Conn {
162 func (l *DefaultListener) InternalAddress() *NetAddress {
166 func (l *DefaultListener) ExternalAddress() *NetAddress {
170 // NOTE: The returned listener is already Accept()'ing.
171 // So it's not suitable to pass into http.Serve().
172 func (l *DefaultListener) NetListener() net.Listener {
176 func (l *DefaultListener) String() string {
177 return fmt.Sprintf("Listener(@%v)", l.extAddr)
180 /* external address helpers */
182 // UPNP external address discovery & port mapping
183 func getUPNPExternalAddress(externalPort, internalPort int) *NetAddress {
184 log.Info("Getting UPNP external address")
185 nat, err := upnp.Discover()
187 log.WithField("error", err).Error("Could not perform UPNP discover")
191 ext, err := nat.GetExternalAddress()
193 log.WithField("error", err).Error("Could not perform UPNP external address")
197 // UPnP can't seem to get the external port, so let's just be explicit.
198 if externalPort == 0 {
199 externalPort = defaultExternalPort
202 externalPort, err = nat.AddPortMapping("tcp", externalPort, internalPort, "bytomd", 0)
204 log.WithField("error", err).Error("Could not add UPNP port mapping")
208 log.WithField("address", ext).Info("Got UPNP external address")
209 return NewNetAddressIPPort(ext, uint16(externalPort))
212 // TODO: use syscalls: http://pastebin.com/9exZG4rh
213 func getNaiveExternalAddress(port int) *NetAddress {
214 addrs, err := net.InterfaceAddrs()
216 cmn.PanicCrisis(cmn.Fmt("Could not fetch interface addresses: %v", err))
219 for _, a := range addrs {
220 ipnet, ok := a.(*net.IPNet)
225 if v4 == nil || v4[0] == 127 {
228 return NewNetAddressIPPort(ipnet.IP, uint16(port))