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 var listenerStatus = false
62 for i := 0; i < tryListenSeconds; i++ {
63 listener, err = net.Listen(protocol, lAddr)
66 } else if i < tryListenSeconds-1 {
67 time.Sleep(time.Second * 1)
73 // Actual listener local IP & port
74 listenerIP, listenerPort := splitHostPort(listener.Addr().String())
75 log.WithFields(log.Fields{
78 }).Info("Local listener")
80 // Determine internal address...
81 var intAddr *NetAddress
82 intAddr, err = NewNetAddressString(lAddr)
87 // Determine external address...
88 var extAddr *NetAddress
90 // If the lAddrIP is INADDR_ANY, try UPnP
91 if lAddrIP == "" || lAddrIP == "0.0.0.0" {
92 extAddr = getUPNPExternalAddress(lAddrPort, listenerPort)
100 if address := GetIP(); address.Success == true {
101 extAddr = NewNetAddressIPPort(net.ParseIP(address.Ip), uint16(lAddrPort))
105 // Otherwise just use the local address...
107 extAddr = getNaiveExternalAddress(listenerPort)
110 cmn.PanicCrisis("Could not determine external address!")
113 dl := &DefaultListener{
117 connections: make(chan net.Conn, numBufferedConnections),
119 dl.BaseService = *cmn.NewBaseService(logger, "DefaultListener", dl)
120 dl.Start() // Started upon construction
122 if !listenerStatus && getExtIP {
123 conn, err := net.DialTimeout("tcp", extAddr.String(), 3*time.Second)
125 if err != nil && conn == nil {
126 log.Error("Could not open listen port")
129 if err == nil && conn != nil {
130 log.Info("Success open listen port")
131 listenerStatus = true
136 return dl, listenerStatus
139 func (l *DefaultListener) OnStart() error {
140 l.BaseService.OnStart()
145 func (l *DefaultListener) OnStop() {
146 l.BaseService.OnStop()
150 // Accept connections and pass on the channel
151 func (l *DefaultListener) listenRoutine() {
153 conn, err := l.listener.Accept()
156 break // Go to cleanup
159 // listener wasn't stopped,
160 // yet we encountered an error.
165 l.connections <- conn
170 for _ = range l.connections {
175 // A channel of inbound connections.
176 // It gets closed when the listener closes.
177 func (l *DefaultListener) Connections() <-chan net.Conn {
181 func (l *DefaultListener) InternalAddress() *NetAddress {
185 func (l *DefaultListener) ExternalAddress() *NetAddress {
189 // NOTE: The returned listener is already Accept()'ing.
190 // So it's not suitable to pass into http.Serve().
191 func (l *DefaultListener) NetListener() net.Listener {
195 func (l *DefaultListener) String() string {
196 return fmt.Sprintf("Listener(@%v)", l.extAddr)
199 /* external address helpers */
201 // UPNP external address discovery & port mapping
202 func getUPNPExternalAddress(externalPort, internalPort int) *NetAddress {
203 log.Info("Getting UPNP external address")
204 nat, err := upnp.Discover()
206 log.WithField("error", err).Error("Could not perform UPNP discover")
210 ext, err := nat.GetExternalAddress()
212 log.WithField("error", err).Error("Could not perform UPNP external address")
216 // UPnP can't seem to get the external port, so let's just be explicit.
217 if externalPort == 0 {
218 externalPort = defaultExternalPort
221 externalPort, err = nat.AddPortMapping("tcp", externalPort, internalPort, "bytomd", 0)
223 log.WithField("error", err).Error("Could not add UPNP port mapping")
227 log.WithField("address", ext).Info("Got UPNP external address")
228 return NewNetAddressIPPort(ext, uint16(externalPort))
231 // TODO: use syscalls: http://pastebin.com/9exZG4rh
232 func getNaiveExternalAddress(port int) *NetAddress {
233 addrs, err := net.InterfaceAddrs()
235 cmn.PanicCrisis(cmn.Fmt("Could not fetch interface addresses: %v", err))
238 for _, a := range addrs {
239 ipnet, ok := a.(*net.IPNet)
244 if v4 == nil || v4[0] == 127 {
247 return NewNetAddressIPPort(ipnet.IP, uint16(port))