webrtcprivate: fix rcmgr bugs on listener

libp2p · Oct 16, 2023 · 6ac54f1 · 6ac54f1
1 parent 07903ee
commit 6ac54f1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/p2p/transport/webrtcprivate/listener.go b/p2p/transport/webrtcprivate/listener.go
@@ -81,12 +81,16 @@ func (l *listener) handleSignalingStream(s network.Stream) {
 	defer cancel()
 	defer s.Close()
 
-	scope, err := l.transport.rcmgr.OpenConnection(network.DirInbound, false, ma.StringCast("/webrtc")) // we don't have a better remote adress right now
+	scope, err := l.transport.rcmgr.OpenConnection(network.DirInbound, true, ma.StringCast("/webrtc")) // we don't have a better remote adress right now
 	if err != nil {
 		s.Reset()
 		log.Debug("failed to create connection scope:", err)
 		return
 	}
+	if err := scope.SetPeer(s.Conn().RemotePeer()); err != nil {
+		log.Debugf("resource manager blocked incoming conn from peer %s: %s", s.Conn().RemotePeer(), err)
+		return
+	}
 
 	if err := s.Scope().SetService(name); err != nil {
 		log.Debugf("error attaching stream to /webrtc listener: %s", err)