From 3e4fd24df14d78c8153651105a3a647999c6f684 Mon Sep 17 00:00:00 2001 From: sukun Date: Thu, 12 Oct 2023 15:17:13 +0530 Subject: [PATCH] webrtcprivate: fix rcmgr bugs on listener --- p2p/transport/webrtcprivate/listener.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/p2p/transport/webrtcprivate/listener.go b/p2p/transport/webrtcprivate/listener.go index f356746d85..97807e3ef5 100644 --- 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)