fix: protocol match bug

wemixarchive · Aug 12, 2024 · 5a5a6dc · 5a5a6dc
1 parent 0bc582c
commit 5a5a6dc
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/p2p/peer.go b/p2p/peer.go
@@ -365,7 +365,12 @@ func countMatchingProtocols(protocols []Protocol, caps []Cap) int {
 	n := 0
 	for _, cap := range caps {
 		for _, proto := range protocols {
-			if proto.Name == cap.Name && proto.Version == cap.Version {
+			if proto.Match == nil {
+				proto.Match = func(cap Cap) bool {
+					return proto.Name == cap.Name && proto.Version == cap.Version
+				}
+			}
+			if proto.Match(cap) {
 				n++
 			}
 		}