mirror of
https://github.com/status-im/status-go.git
synced 2025-02-25 13:16:15 +00:00
eeca435064
Update vendor Integrate rendezvous into status node Add a test with failover using rendezvous Use multiple servers in client Use discovery V5 by default and test that node can be started with rendezvous discovet Fix linter Update rendezvous client to one with instrumented stream Address feedback Fix test with updated topic limits Apply several suggestions Change log to debug for request errors because we continue execution Remove web3js after rebase Update rendezvous package
116 lines
2.2 KiB
Go
116 lines
2.2 KiB
Go
package nat
|
|
|
|
import (
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/jackpal/gateway"
|
|
"github.com/jackpal/go-nat-pmp"
|
|
)
|
|
|
|
var (
|
|
_ NAT = (*natpmpNAT)(nil)
|
|
)
|
|
|
|
func discoverNATPMP() <-chan NAT {
|
|
res := make(chan NAT, 1)
|
|
|
|
ip, err := gateway.DiscoverGateway()
|
|
if err == nil {
|
|
go discoverNATPMPWithAddr(res, ip)
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func discoverNATPMPWithAddr(c chan NAT, ip net.IP) {
|
|
client := natpmp.NewClient(ip)
|
|
_, err := client.GetExternalAddress()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
c <- &natpmpNAT{client, ip, make(map[int]int)}
|
|
}
|
|
|
|
type natpmpNAT struct {
|
|
c *natpmp.Client
|
|
gateway net.IP
|
|
ports map[int]int
|
|
}
|
|
|
|
func (n *natpmpNAT) GetDeviceAddress() (addr net.IP, err error) {
|
|
return n.gateway, nil
|
|
}
|
|
|
|
func (n *natpmpNAT) GetInternalAddress() (addr net.IP, err error) {
|
|
ifaces, err := net.Interfaces()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, iface := range ifaces {
|
|
addrs, err := iface.Addrs()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, addr := range addrs {
|
|
switch x := addr.(type) {
|
|
case *net.IPNet:
|
|
if x.Contains(n.gateway) {
|
|
return x.IP, nil
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil, ErrNoInternalAddress
|
|
}
|
|
|
|
func (n *natpmpNAT) GetExternalAddress() (addr net.IP, err error) {
|
|
res, err := n.c.GetExternalAddress()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
d := res.ExternalIPAddress
|
|
return net.IPv4(d[0], d[1], d[2], d[3]), nil
|
|
}
|
|
|
|
func (n *natpmpNAT) AddPortMapping(protocol string, internalPort int, description string, timeout time.Duration) (int, error) {
|
|
var (
|
|
err error
|
|
)
|
|
|
|
timeoutInSeconds := int(timeout / time.Second)
|
|
|
|
if externalPort := n.ports[internalPort]; externalPort > 0 {
|
|
_, err = n.c.AddPortMapping(protocol, internalPort, externalPort, timeoutInSeconds)
|
|
if err == nil {
|
|
n.ports[internalPort] = externalPort
|
|
return externalPort, nil
|
|
}
|
|
}
|
|
|
|
for i := 0; i < 3; i++ {
|
|
externalPort := randomPort()
|
|
_, err = n.c.AddPortMapping(protocol, internalPort, externalPort, timeoutInSeconds)
|
|
if err == nil {
|
|
n.ports[internalPort] = externalPort
|
|
return externalPort, nil
|
|
}
|
|
}
|
|
|
|
return 0, err
|
|
}
|
|
|
|
func (n *natpmpNAT) DeletePortMapping(protocol string, internalPort int) (err error) {
|
|
delete(n.ports, internalPort)
|
|
return nil
|
|
}
|
|
|
|
func (n *natpmpNAT) Type() string {
|
|
return "NAT-PMP"
|
|
}
|