Add --p2p-host-dns flag to specify p2p external DNS (#4608)

Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>

beacon-chain/main.go

@@ -51,6 +51,7 @@ var appFlags = []cli.Flag{
 	cmd.P2PUDPPort,
 	cmd.P2PTCPPort,
 	cmd.P2PHost,
+	cmd.P2PHostDNS,
 	cmd.P2PMaxPeers,
 	cmd.P2PPrivKey,
 	cmd.P2PWhitelist,

beacon-chain/node/node.go

@@ -269,6 +269,7 @@ func (b *BeaconNode) registerP2P(ctx *cli.Context) error {
 		RelayNodeAddr:     ctx.GlobalString(cmd.RelayNode.Name),
 		DataDir:           ctx.GlobalString(cmd.DataDirFlag.Name),
 		HostAddress:       ctx.GlobalString(cmd.P2PHost.Name),
+		HostDNS:           ctx.GlobalString(cmd.P2PHostDNS.Name),
 		PrivateKey:        ctx.GlobalString(cmd.P2PPrivKey.Name),
 		TCPPort:           ctx.GlobalUint(cmd.P2PTCPPort.Name),
 		UDPPort:           ctx.GlobalUint(cmd.P2PUDPPort.Name),

beacon-chain/p2p/config.go

@@ -10,6 +10,7 @@ type Config struct {
 	Discv5BootStrapAddr   []string
 	RelayNodeAddr         string
 	HostAddress           string
+	HostDNS               string
 	PrivateKey            string
 	DataDir               string
 	TCPPort               uint

beacon-chain/p2p/options.go

@@ -45,6 +45,17 @@ func buildOptions(cfg *Config, ip net.IP, priKey *ecdsa.PrivateKey) []libp2p.Opt
 			return addrs
 		}))
 	}
+	if cfg.HostDNS != "" {
+		options = append(options, libp2p.AddrsFactory(func(addrs []multiaddr.Multiaddr) []multiaddr.Multiaddr {
+			external, err := multiaddr.NewMultiaddr(fmt.Sprintf("/dns4/%s/tcp/%d", cfg.HostDNS, cfg.TCPPort))
+			if err != nil {
+				log.WithError(err).Error("Unable to create external multiaddress")
+			} else {
+				addrs = append(addrs, external)
+			}
+			return addrs
+		}))
+	}
 	return options
 }
 

beacon-chain/p2p/service.go

@@ -217,7 +217,15 @@ func (s *Service) Start() {
 
 	p2pHostAddress := s.cfg.HostAddress
 	p2pTCPPort := s.cfg.TCPPort
-	logExternalIP4Addr(s.host.ID(), p2pHostAddress, p2pTCPPort)
+
+	if p2pHostAddress != "" {
+		logExternalIP4Addr(s.host.ID(), p2pHostAddress, p2pTCPPort)
+	}
+
+	p2pHostDNS := s.cfg.HostDNS
+	if p2pHostDNS != "" {
+		logExternalDNSAddr(s.host.ID(), p2pHostDNS, p2pTCPPort)
+	}
 }
 
 // Stop the p2p service and terminate all peer connections.
@@ -380,3 +388,14 @@ func logExternalIP4Addr(id peer.ID, addr string, port uint) {
 		).Info("Node started external p2p server")
 	}
 }
+
+func logExternalDNSAddr(id peer.ID, addr string, port uint) {
+	if addr != "" {
+		p := strconv.FormatUint(uint64(port), 10)
+
+		log.WithField(
+			"multiAddr",
+			"/dns4/"+addr+"/tcp/"+p+"/p2p/"+id.String(),
+		).Info("Node started external p2p server")
+	}
+}

beacon-chain/usage.go

@@ -95,6 +95,7 @@ var appHelpFlagGroups = []flagGroup{
 		Name: "p2p",
 		Flags: []cli.Flag{
 			cmd.P2PHost,
+			cmd.P2PHostDNS,
 			cmd.P2PMaxPeers,
 			cmd.P2PPrivKey,
 			cmd.P2PWhitelist,

shared/cmd/flags.go

@@ -94,6 +94,12 @@ var (
 		Usage: "The IP address advertised by libp2p. This may be used to advertise an external IP.",
 		Value: "",
 	}
+	// P2PHostDNS defines the host DNS to be used by libp2p.
+	P2PHostDNS = cli.StringFlag{
+		Name:  "p2p-host-dns",
+		Usage: "The DNS address advertised by libp2p. This may be used to advertise an external DNS.",
+		Value: "",
+	}
 	// P2PPrivKey defines a flag to specify the location of the private key file for libp2p.
 	P2PPrivKey = cli.StringFlag{
 		Name:  "p2p-priv-key",