apps/go2rtc
1
0
Fork 0
mirror of https://github.com/AlexxIT/go2rtc.git synced 2025-09-27 04:36:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix concurrent SRTP sessions map read and map write #1489

Browse Source
This commit is contained in:
Alex X
2025-02-24 22:04:14 +03:00
parent 45b223a2ef
commit effff6f88d
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/srtp/server.go
View File

@@ -65,6 +65,13 @@ func (s *Server) DelSession(session *Session) {
s.mu.Unlock() s.mu.Unlock()
} }
func (s *Server) GetSession(ssrc uint32) (session *Session) {
s.mu.Lock()
session = s.sessions[ssrc]
s.mu.Unlock()
return
}
func (s *Server) handle() error { func (s *Server) handle() error {
b := make([]byte, 2048) b := make([]byte, 2048)
for { for {
@@ -80,14 +87,14 @@ func (s *Server) handle() error {
case 99, 110, 0x80 | 99, 0x80 | 110: case 99, 110, 0x80 | 99, 0x80 | 110:
// this is default position for SSRC in RTP packet // this is default position for SSRC in RTP packet
ssrc := binary.BigEndian.Uint32(b[8:]) ssrc := binary.BigEndian.Uint32(b[8:])
if session, ok := s.sessions[ssrc]; ok { if session := s.GetSession(ssrc); session != nil {
session.ReadRTP(b[:n]) session.ReadRTP(b[:n])
} }
case 200, 201, 202, 203, 204, 205, 206, 207: case 200, 201, 202, 203, 204, 205, 206, 207:
// this is default position for SSRC in RTCP packet // this is default position for SSRC in RTCP packet
ssrc := binary.BigEndian.Uint32(b[4:]) ssrc := binary.BigEndian.Uint32(b[4:])
if session, ok := s.sessions[ssrc]; ok { if session := s.GetSession(ssrc); session != nil {
session.ReadRTCP(b[:n]) session.ReadRTCP(b[:n])
} }
} }