package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-31 18:42:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

multicast: add readOnly flag (#423)

Browse Source
This commit is contained in:
Alessandro Ros
2023-09-16 11:03:32 +02:00
committed by GitHub
parent e6f7c4dea4
commit 99773e19af
3 changed files with 103 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/multicast/multi_conn.go
View File

@@ -25,6 +25,7 @@ type MultiConn struct {
// NewMultiConn allocates a MultiConn.
func NewMultiConn(
address string,
readOnly bool,
listenPacket func(network, address string) (net.PacketConn, error),
) (Conn, error) {
addr, err := net.ResolveUDPAddr("udp4", address)
@@ -67,8 +68,12 @@ func NewMultiConn(
return nil, fmt.Errorf("no multicast-capable interfaces found")
}
writeConns := make([]*net.UDPConn, len(enabledInterfaces))
writeConnIPs := make([]*ipv4.PacketConn, len(enabledInterfaces))
var writeConns []*net.UDPConn
var writeConnIPs []*ipv4.PacketConn
if !readOnly {
writeConns = make([]*net.UDPConn, len(enabledInterfaces))
writeConnIPs = make([]*ipv4.PacketConn, len(enabledInterfaces))
for i, intf := range enabledInterfaces {
tmp, err := listenPacket("udp4", "224.0.0.0:"+strconv.FormatInt(int64(addr.Port), 10))
@@ -104,6 +109,7 @@ func NewMultiConn(
writeConns[i] = writeConn
writeConnIPs[i] = writeConnIP
}
}
return &MultiConn{
addr: addr,

15
pkg/multicast/multi_conn_lin.go
View File

@@ -24,6 +24,7 @@ type MultiConn struct {
// NewMultiConn allocates a MultiConn.
func NewMultiConn(
address string,
readOnly bool,
_ func(network, address string) (net.PacketConn, error),
) (Conn, error) {
addr, err := net.ResolveUDPAddr("udp4", address)
@@ -84,6 +85,10 @@ func NewMultiConn(
return nil, fmt.Errorf("no multicast-capable interfaces found")
}
var writeFiles []*os.File
var writeConns []net.PacketConn
if !readOnly {
writeSocks := make([]int, len(enabledInterfaces))
for i, intf := range enabledInterfaces {
@@ -145,15 +150,17 @@ func NewMultiConn(
writeSocks[i] = writeSock
}
readFile := os.NewFile(uintptr(readSock), "")
readConn, _ := net.FilePacketConn(readFile)
writeFiles := make([]*os.File, len(writeSocks))
writeConns := make([]net.PacketConn, len(writeSocks))
writeFiles = make([]*os.File, len(writeSocks))
writeConns = make([]net.PacketConn, len(writeSocks))
for i, writeSock := range writeSocks {
writeFiles[i] = os.NewFile(uintptr(writeSock), "")
writeConns[i], _ = net.FilePacketConn(writeFiles[i])
}
}
readFile := os.NewFile(uintptr(readSock), "")
readConn, _ := net.FilePacketConn(readFile)
return &MultiConn{
addr: addr,

2
server_udp_listener.go
View File

@@ -76,7 +76,7 @@ func newServerUDPListener(
var listenIP net.IP
if multicastEnable {
var err error
pc, err = multicast.NewMultiConn(address, listenPacket)
pc, err = multicast.NewMultiConn(address, false, listenPacket)
if err != nil {
return nil, err
}