lkm/gb28181/udp_server.go

package gb28181

import (
	"github.com/pion/rtp"
	"github.com/yangjiechina/avformat/transport"
	"github.com/yangjiechina/lkm/log"
	"github.com/yangjiechina/lkm/stream"
	"net"
)

// UDPServer GB28181UDP收流
type UDPServer struct {
	stream.StreamServer[*UDPSource]
	udp    *transport.UDPServer
	filter Filter
}

func (U *UDPServer) OnNewSession(conn net.Conn) *UDPSource {
	return nil
}

func (U *UDPServer) OnCloseSession(session *UDPSource) {
	U.filter.RemoveSource(session.SSRC())
	session.Close()

	if stream.AppConfig.GB28181.IsMultiPort() {
		U.udp.Close()
		U.Handler = nil
	}
}

func (U *UDPServer) OnPacket(conn net.Conn, data []byte) []byte {
	U.StreamServer.OnPacket(conn, data)

	packet := rtp.Packet{}
	err := packet.Unmarshal(data)
	if err != nil {
		log.Sugar.Errorf("解析rtp失败 err:%s conn:%s", err.Error(), conn.RemoteAddr().String())
		return nil
	}

	source := U.filter.FindSource(packet.SSRC)
	if source == nil {
		log.Sugar.Errorf("ssrc匹配source失败 ssrc:%x conn:%s", packet.SSRC, conn.RemoteAddr().String())
		return nil
	}

	if stream.SessionStateHandshakeDone == source.State() {
		conn.(*transport.Conn).Data = source
		source.PreparePublish(conn, packet.SSRC, source)
	}

	source.InputRtp(&packet)
	return nil
}

func NewUDPServer(addr net.Addr, filter Filter) (*UDPServer, error) {
	server := &UDPServer{
		filter: filter,
	}

	udp, err := transport.NewUDPServer(addr, server)
	if err != nil {
		return nil, err
	}

	server.udp = udp
	server.StreamServer = stream.StreamServer[*UDPSource]{
		SourceType: stream.SourceType28181,
		Handler:    server,
	}
	return server, nil
}