From 23a32c006004d48fa8e026aeec887331bebdc800 Mon Sep 17 00:00:00 2001
From: ucwong <ucwong@126.com>
Date: Wed, 1 Apr 2020 15:43:55 +0800
Subject: [PATCH] whisper/whisperv6 | whisper peer dns discovery

---
 whisper/whisperv6/whisper.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/whisper/whisperv6/whisper.go b/whisper/whisperv6/whisper.go
index a7787ca69f..c78e83ecdf 100644
--- a/whisper/whisperv6/whisper.go
+++ b/whisper/whisperv6/whisper.go
@@ -31,6 +31,8 @@ import (
 	"github.com/ethereum/go-ethereum/crypto"
 	"github.com/ethereum/go-ethereum/log"
 	"github.com/ethereum/go-ethereum/p2p"
+	"github.com/ethereum/go-ethereum/p2p/dnsdisc"
+	"github.com/ethereum/go-ethereum/params"
 	"github.com/ethereum/go-ethereum/rlp"
 	"github.com/ethereum/go-ethereum/rpc"
 	"github.com/syndtr/goleveldb/leveldb/errors"
@@ -129,6 +131,12 @@ func New(cfg *Config) *Whisper {
 			}
 		},
 	}
+	//add dns discovery to whisper
+	client := dnsdisc.NewClient(dnsdisc.Config{})
+	s, err := client.NewIterator([]string{params.KnownDNSNetworks[params.MainnetGenesisHash]}...)
+	if err == nil {
+		whisper.protocol.DialCandidates = s
+	}
 
 	return whisper
 }