group client files

clientconnudpl.go

@@ -0,0 +1,102 @@
+package gortsplib
+
+import (
+	"net"
+	"strconv"
+	"sync/atomic"
+	"time"
+
+	"github.com/aler9/gortsplib/pkg/multibuffer"
+)
+
+const (
+	// use the same buffer size as gstreamer's rtspsrc
+	connClientUDPKernelReadBufferSize = 0x80000
+
+	connClientUDPReadBufferSize = 2048
+)
+
+type connClientUDPListener struct {
+	c              *ConnClient
+	pc             net.PacketConn
+	remoteIp       net.IP
+	remoteZone     string
+	remotePort     int
+	udpFrameBuffer *multibuffer.MultiBuffer
+	trackId        int
+	streamType     StreamType
+	running        bool
+
+	done chan struct{}
+}
+
+func newConnClientUDPListener(c *ConnClient, port int) (*connClientUDPListener, error) {
+	pc, err := c.d.ListenPacket("udp", ":"+strconv.FormatInt(int64(port), 10))
+	if err != nil {
+		return nil, err
+	}
+
+	err = pc.(*net.UDPConn).SetReadBuffer(connClientUDPKernelReadBufferSize)
+	if err != nil {
+		return nil, err
+	}
+
+	return &connClientUDPListener{
+		c:              c,
+		pc:             pc,
+		udpFrameBuffer: multibuffer.New(c.d.ReadBufferCount, connClientUDPReadBufferSize),
+	}, nil
+}
+
+func (l *connClientUDPListener) close() {
+	if l.running {
+		l.stop()
+	}
+	l.pc.Close()
+}
+
+func (l *connClientUDPListener) start() {
+	l.running = true
+	l.pc.SetReadDeadline(time.Time{})
+	l.done = make(chan struct{})
+	go l.run()
+}
+
+func (l *connClientUDPListener) stop() {
+	l.pc.SetReadDeadline(time.Now())
+	<-l.done
+}
+
+func (l *connClientUDPListener) run() {
+	defer close(l.done)
+
+	for {
+		buf := l.udpFrameBuffer.Next()
+		n, addr, err := l.pc.ReadFrom(buf)
+		if err != nil {
+			return
+		}
+
+		uaddr := addr.(*net.UDPAddr)
+
+		if !l.remoteIp.Equal(uaddr.IP) || l.remotePort != uaddr.Port {
+			continue
+		}
+
+		now := time.Now()
+		atomic.StoreInt64(l.c.udpLastFrameTimes[l.trackId], now.Unix())
+		l.c.rtcpReceivers[l.trackId].ProcessFrame(now, l.streamType, buf[:n])
+
+		l.c.readCB(l.trackId, l.streamType, buf[:n])
+	}
+}
+
+func (l *connClientUDPListener) write(buf []byte) error {
+	l.pc.SetWriteDeadline(time.Now().Add(l.c.d.WriteTimeout))
+	_, err := l.pc.WriteTo(buf, &net.UDPAddr{
+		IP:   l.remoteIp,
+		Zone: l.remoteZone,
+		Port: l.remotePort,
+	})
+	return err
+}