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 {
53 // Local listen IP & port
54 lAddrIP, lAddrPort := splitHostPort(lAddr)
57 var listener net.Listener
59 for i := 0; i < tryListenSeconds; i++ {
60 listener, err = net.Listen(protocol, lAddr)
63 } else if i < tryListenSeconds-1 {
64 time.Sleep(time.Second * 1)
70 // Actual listener local IP & port
71 listenerIP, listenerPort := splitHostPort(listener.Addr().String())
72 log.WithFields(log.Fields{
75 }).Info("Local listener")
77 // Determine internal address...
78 var intAddr *NetAddress
79 intAddr, err = NewNetAddressString(lAddr)
84 // Determine external address...
85 var extAddr *NetAddress
87 // If the lAddrIP is INADDR_ANY, try UPnP
88 if lAddrIP == "" || lAddrIP == "0.0.0.0" {
89 extAddr = getUPNPExternalAddress(lAddrPort, listenerPort)
92 // Otherwise just use the local address...
94 extAddr = getNaiveExternalAddress(listenerPort)
97 cmn.PanicCrisis("Could not determine external address!")
100 dl := &DefaultListener{
104 connections: make(chan net.Conn, numBufferedConnections),
106 dl.BaseService = *cmn.NewBaseService(logger, "DefaultListener", dl)
107 dl.Start() // Started upon construction
111 func (l *DefaultListener) OnStart() error {
112 l.BaseService.OnStart()
117 func (l *DefaultListener) OnStop() {
118 l.BaseService.OnStop()
122 // Accept connections and pass on the channel
123 func (l *DefaultListener) listenRoutine() {
125 conn, err := l.listener.Accept()
128 break // Go to cleanup
131 // listener wasn't stopped,
132 // yet we encountered an error.
137 l.connections <- conn
142 for _ = range l.connections {
147 // A channel of inbound connections.
148 // It gets closed when the listener closes.
149 func (l *DefaultListener) Connections() <-chan net.Conn {
153 func (l *DefaultListener) InternalAddress() *NetAddress {
157 func (l *DefaultListener) ExternalAddress() *NetAddress {
161 // NOTE: The returned listener is already Accept()'ing.
162 // So it's not suitable to pass into http.Serve().
163 func (l *DefaultListener) NetListener() net.Listener {
167 func (l *DefaultListener) String() string {
168 return fmt.Sprintf("Listener(@%v)", l.extAddr)
171 /* external address helpers */
173 // UPNP external address discovery & port mapping
174 func getUPNPExternalAddress(externalPort, internalPort int) *NetAddress {
175 log.Info("Getting UPNP external address")
176 nat, err := upnp.Discover()
178 log.WithField("error", err).Error("Could not perform UPNP discover")
182 ext, err := nat.GetExternalAddress()
184 log.WithField("error", err).Error("Could not perform UPNP external address")
188 // UPnP can't seem to get the external port, so let's just be explicit.
189 if externalPort == 0 {
190 externalPort = defaultExternalPort
193 externalPort, err = nat.AddPortMapping("tcp", externalPort, internalPort, "bytomd", 0)
195 log.WithField("error", err).Error("Could not add UPNP port mapping")
199 log.WithField("address", ext).Info("Got UPNP external address")
200 return NewNetAddressIPPort(ext, uint16(externalPort))
203 // TODO: use syscalls: http://pastebin.com/9exZG4rh
204 func getNaiveExternalAddress(port int) *NetAddress {
205 addrs, err := net.InterfaceAddrs()
207 cmn.PanicCrisis(cmn.Fmt("Could not fetch interface addresses: %v", err))
210 for _, a := range addrs {
211 ipnet, ok := a.(*net.IPNet)
216 if v4 == nil || v4[0] == 127 {
219 return NewNetAddressIPPort(ipnet.IP, uint16(port))