Merge pull request #76 from libp2p/feat/protected-query

Implement IsProtected interface
libp2p · Jun 3, 2020 · c66854c · c66854c
2 parents 9fba560 + f237862
commit c66854c
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/p2p/net/connmgr/connmgr.go b/p2p/net/connmgr/connmgr.go
@@ -176,6 +176,23 @@ func (cm *BasicConnMgr) Unprotect(id peer.ID, tag string) (protected bool) {
 	return true
 }
 
+func (cm *BasicConnMgr) IsProtected(id peer.ID, tag string) (protected bool) {
+	cm.plk.Lock()
+	defer cm.plk.Unlock()
+
+	tags, ok := cm.protected[id]
+	if !ok {
+		return false
+	}
+
+	if tag == "" {
+		return true
+	}
+
+	_, protected = tags[tag]
+	return protected
+}
+
 // peerInfo stores metadata for a given peer.
 type peerInfo struct {
 	id       peer.ID