9 "github.com/bytom/p2p/upnp"
10 log "github.com/sirupsen/logrus"
11 cmn "github.com/tendermint/tmlibs/common"
14 //Listener subset of the methods of DefaultListener
15 type Listener interface {
16 Connections() <-chan net.Conn
17 InternalAddress() *NetAddress
18 ExternalAddress() *NetAddress
23 //DefaultListener Implements bytomd server 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 //NewDefaultListener create a default listener
52 func NewDefaultListener(protocol string, lAddr string, skipUPNP bool) (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.Info("Local listener", " ip:", listenerIP, " port:", listenerPort)
77 // Determine internal address...
78 var intAddr *NetAddress
79 intAddr, err = NewNetAddressString(lAddr)
84 // Determine external address...
85 var extAddr *NetAddress
86 //skipUPNP: If true, does not try getUPNPExternalAddress()
88 // If the lAddrIP is INADDR_ANY, try UPnP
89 if lAddrIP == "" || lAddrIP == "0.0.0.0" {
90 extAddr = getUPNPExternalAddress(lAddrPort, listenerPort)
98 if address := GetIP(); address.Success == true {
99 extAddr = NewNetAddressIPPort(net.ParseIP(address.Ip), uint16(lAddrPort))
103 // Otherwise just use the local address...
105 extAddr = getNaiveExternalAddress(listenerPort, false)
108 cmn.PanicCrisis("Could not determine external address!")
111 dl := &DefaultListener{
115 connections: make(chan net.Conn, numBufferedConnections),
117 dl.BaseService = *cmn.NewBaseService(nil, "DefaultListener", dl)
118 dl.Start() // Started upon construction
120 if !listenerStatus && getExtIP {
121 conn, err := net.DialTimeout("tcp", extAddr.String(), 3*time.Second)
123 if err != nil && conn == nil {
124 log.Error("Could not open listen port")
127 if err == nil && conn != nil {
128 log.Info("Success open listen port")
129 listenerStatus = true
134 return dl, listenerStatus
137 //OnStart start listener
138 func (l *DefaultListener) OnStart() error {
139 l.BaseService.OnStart()
144 //OnStop stop listener
145 func (l *DefaultListener) OnStop() {
146 l.BaseService.OnStop()
150 //listenRoutine 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
172 //Connections a channel of inbound connections. It gets closed when the listener closes.
173 func (l *DefaultListener) Connections() <-chan net.Conn {
177 //InternalAddress listener internal address
178 func (l *DefaultListener) InternalAddress() *NetAddress {
182 //ExternalAddress listener external address for remote peer dial
183 func (l *DefaultListener) ExternalAddress() *NetAddress {
187 // NetListener the returned listener is already Accept()'ing. So it's not suitable to pass into http.Serve().
188 func (l *DefaultListener) NetListener() net.Listener {
192 //String string of default listener
193 func (l *DefaultListener) String() string {
194 return fmt.Sprintf("Listener(@%v)", l.extAddr)
197 //getUPNPExternalAddress UPNP external address discovery & port mapping
198 func getUPNPExternalAddress(externalPort, internalPort int) *NetAddress {
199 log.Info("Getting UPNP external address")
200 nat, err := upnp.Discover()
202 log.Info("Could not perform UPNP discover. error:", err)
206 ext, err := nat.GetExternalAddress()
208 log.Info("Could not perform UPNP external address. error:", err)
212 // UPnP can't seem to get the external port, so let's just be explicit.
213 if externalPort == 0 {
214 externalPort = defaultExternalPort
217 externalPort, err = nat.AddPortMapping("tcp", externalPort, internalPort, "bytomd", 0)
219 log.Info("Could not add UPNP port mapping. error:", err)
223 log.Info("Got UPNP external address ", ext)
224 return NewNetAddressIPPort(ext, uint16(externalPort))
227 func getNaiveExternalAddress(port int, settleForLocal bool) *NetAddress {
228 addrs, err := net.InterfaceAddrs()
230 cmn.PanicCrisis(cmn.Fmt("Could not fetch interface addresses: %v", err))
233 for _, a := range addrs {
234 ipnet, ok := a.(*net.IPNet)
239 if v4 == nil || (!settleForLocal && v4[0] == 127) {
242 return NewNetAddressIPPort(ipnet.IP, uint16(port))
245 // try again, but settle for local
246 log.Info("Node may not be connected to internet. Settling for local address")
247 return getNaiveExternalAddress(port, true)