forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-26 22:56:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

whisper: allowed filtering for variable topic size

Browse source
This commit is contained in:
Vlad 2017-03-20 17:34:58 +01:00
parent 58c8121a22
commit 51cfbba9e0
2 changed files with 18 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
whisper/whisperv5/api_test.go
View file

@ -17,6 +17,7 @@
package whisperv5
import (
"bytes"
"encoding/json"
"testing"
"time"

22
whisper/whisperv5/filter.go
View file

@ -89,12 +89,12 @@ func (fs *Filters) Get(id string) *Filter {
}
func (fs *Filters) NotifyWatchers(env *Envelope, p2pMessage bool) {
var j int
var msg *ReceivedMessage
fs.mutex.RLock()
defer fs.mutex.RUnlock()
j := -1
for _, watcher := range fs.watchers {
j++
if p2pMessage && !watcher.AllowP2P {
@ -118,6 +118,7 @@ func (fs *Filters) NotifyWatchers(env *Envelope, p2pMessage bool) {
}
if match && msg != nil {
log.Trace(fmt.Sprintf("message decrypted [%x]", env.Hash()))
watcher.Trigger(msg)
}
}
@ -202,10 +203,21 @@ func (f *Filter) MatchTopic(topic TopicType) bool {
}
for _, bt := range f.Topics {
for j, b := range bt {
if topic[j] != b {
return false
}
if MatchSingleTopic(topic, bt) {
return true
}
}
return false
}
func MatchSingleTopic(topic TopicType, bt []byte) bool {
if len(bt) > 4 {
bt = bt[0:4]
}
for j, b := range bt {
if topic[j] != b {
return false
}
}
return true