diff options
author | Guillaume Ballet <gballet@gmail.com> | 2018-02-10 21:35:32 +0800 |
---|---|---|
committer | Péter Szilágyi <peterke@gmail.com> | 2018-02-10 21:35:32 +0800 |
commit | 5cf75a30c1ceb0ab35cd6b0532520d556996b21c (patch) | |
tree | 9a5d6f0fdf27967f38aed2dd5cc651106c00eaa2 /cmd | |
parent | 2f849ade8204a4b417202d90c66e2f5bef4e965d (diff) | |
download | dexon-5cf75a30c1ceb0ab35cd6b0532520d556996b21c.tar dexon-5cf75a30c1ceb0ab35cd6b0532520d556996b21c.tar.gz dexon-5cf75a30c1ceb0ab35cd6b0532520d556996b21c.tar.bz2 dexon-5cf75a30c1ceb0ab35cd6b0532520d556996b21c.tar.lz dexon-5cf75a30c1ceb0ab35cd6b0532520d556996b21c.tar.xz dexon-5cf75a30c1ceb0ab35cd6b0532520d556996b21c.tar.zst dexon-5cf75a30c1ceb0ab35cd6b0532520d556996b21c.zip |
whisper: get wnode to work with v6 (#16051)
The bulk of the issue was to adapt to the new requirement
that a v6 filter has to either contain a symmertric key or
an asymmetric one.
This commits revert one of the fixes that I made to remove
a linter warning: unexporting NewSentMessage. This is not
really a problem as I have a cleanup in the pipe that will
solve this issue.
Diffstat (limited to 'cmd')
-rw-r--r-- | cmd/wnode/main.go | 57 |
1 files changed, 41 insertions, 16 deletions
diff --git a/cmd/wnode/main.go b/cmd/wnode/main.go index e69b57d69..68e6971da 100644 --- a/cmd/wnode/main.go +++ b/cmd/wnode/main.go @@ -43,7 +43,7 @@ import ( "github.com/ethereum/go-ethereum/p2p/discover" "github.com/ethereum/go-ethereum/p2p/nat" "github.com/ethereum/go-ethereum/whisper/mailserver" - whisper "github.com/ethereum/go-ethereum/whisper/whisperv5" + whisper "github.com/ethereum/go-ethereum/whisper/whisperv6" "golang.org/x/crypto/pbkdf2" ) @@ -61,15 +61,17 @@ var ( // encryption var ( - symKey []byte - pub *ecdsa.PublicKey - asymKey *ecdsa.PrivateKey - nodeid *ecdsa.PrivateKey - topic whisper.TopicType - asymKeyID string - filterID string - symPass string - msPassword string + symKey []byte + pub *ecdsa.PublicKey + asymKey *ecdsa.PrivateKey + nodeid *ecdsa.PrivateKey + topic whisper.TopicType + + asymKeyID string + asymFilterID string + symFilterID string + symPass string + msPassword string ) // cmd arguments @@ -363,13 +365,22 @@ func configureNode() { } } - filter := whisper.Filter{ + symFilter := whisper.Filter{ KeySym: symKey, + Topics: [][]byte{topic[:]}, + AllowP2P: p2pAccept, + } + symFilterID, err = shh.Subscribe(&symFilter) + if err != nil { + utils.Fatalf("Failed to install filter: %s", err) + } + + asymFilter := whisper.Filter{ KeyAsym: asymKey, Topics: [][]byte{topic[:]}, AllowP2P: p2pAccept, } - filterID, err = shh.Subscribe(&filter) + asymFilterID, err = shh.Subscribe(&asymFilter) if err != nil { utils.Fatalf("Failed to install filter: %s", err) } @@ -522,9 +533,14 @@ func sendMsg(payload []byte) common.Hash { } func messageLoop() { - f := shh.GetFilter(filterID) - if f == nil { - utils.Fatalf("filter is not installed") + sf := shh.GetFilter(symFilterID) + if sf == nil { + utils.Fatalf("symmetric filter is not installed") + } + + af := shh.GetFilter(asymFilterID) + if af == nil { + utils.Fatalf("asymmetric filter is not installed") } ticker := time.NewTicker(time.Millisecond * 50) @@ -532,7 +548,16 @@ func messageLoop() { for { select { case <-ticker.C: - messages := f.Retrieve() + messages := sf.Retrieve() + for _, msg := range messages { + if *fileExMode || len(msg.Payload) > 2048 { + writeMessageToFile(*argSaveDir, msg) + } else { + printMessageInfo(msg) + } + } + + messages = af.Retrieve() for _, msg := range messages { if *fileExMode || len(msg.Payload) > 2048 { writeMessageToFile(*argSaveDir, msg) |