Add p2p security module (#143)

[bytom/vapor.git] / p2p / peer_set.go

diff --git a/p2p/peer_set.go b/p2p/peer_set.go
index e26746b..c652371 100644 (file)
--- a/p2p/peer_set.go
+++ b/p2p/peer_set.go
@@ -50,6 +50,14 @@ func (ps *PeerSet) Add(peer *Peer) error {
        return nil
 }
 
+func (ps *PeerSet) DoFilter(ip string, pubKey string) error {
+       if ps.Has(pubKey) {
+               return ErrDuplicatePeer
+       }
+
+       return nil
+}
+
 // Get looks up a peer by the provided peerKey.
 func (ps *PeerSet) Get(peerKey string) *Peer {
        ps.mtx.Lock()