aler9
4 years ago
12 changed files with 196 additions and 401 deletions
@ -1,176 +0,0 @@
@@ -1,176 +0,0 @@
|
||||
package serverudp |
||||
|
||||
import ( |
||||
"net" |
||||
"sync" |
||||
"time" |
||||
|
||||
"github.com/aler9/gortsplib" |
||||
"github.com/aler9/gortsplib/pkg/base" |
||||
"github.com/aler9/gortsplib/pkg/multibuffer" |
||||
"github.com/aler9/rtsp-simple-server/internal/logger" |
||||
) |
||||
|
||||
const ( |
||||
// use the same buffer size as gstreamer's rtspsrc
|
||||
kernelReadBufferSize = 0x80000 |
||||
|
||||
readBufferSize = 2048 |
||||
) |
||||
|
||||
// Publisher is implemented by client.Client.
|
||||
type Publisher interface { |
||||
OnUDPPublisherFrame(int, base.StreamType, []byte) |
||||
} |
||||
|
||||
type publisherData struct { |
||||
publisher Publisher |
||||
trackID int |
||||
} |
||||
|
||||
// Parent is implemented by program.
|
||||
type Parent interface { |
||||
Log(logger.Level, string, ...interface{}) |
||||
} |
||||
|
||||
type publisherAddr struct { |
||||
ip [net.IPv6len]byte // use a fixed-size array to enable the equality operator
|
||||
port int |
||||
} |
||||
|
||||
func (p *publisherAddr) fill(ip net.IP, port int) { |
||||
p.port = port |
||||
|
||||
if len(ip) == net.IPv4len { |
||||
copy(p.ip[0:], []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xff, 0xff}) // v4InV6Prefix
|
||||
copy(p.ip[12:], ip) |
||||
} else { |
||||
copy(p.ip[:], ip) |
||||
} |
||||
} |
||||
|
||||
// Server is a UDP server that can be used to send and receive both RTP and RTCP.
|
||||
type Server struct { |
||||
writeTimeout time.Duration |
||||
streamType gortsplib.StreamType |
||||
|
||||
pc *net.UDPConn |
||||
readBuf *multibuffer.MultiBuffer |
||||
publishersMutex sync.RWMutex |
||||
publishers map[publisherAddr]*publisherData |
||||
writeMutex sync.Mutex |
||||
|
||||
// out
|
||||
done chan struct{} |
||||
} |
||||
|
||||
// New allocates a Server.
|
||||
func New(writeTimeout time.Duration, |
||||
port int, |
||||
streamType gortsplib.StreamType, |
||||
parent Parent) (*Server, error) { |
||||
|
||||
pc, err := net.ListenUDP("udp", &net.UDPAddr{ |
||||
Port: port, |
||||
}) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
err = pc.SetReadBuffer(kernelReadBufferSize) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
s := &Server{ |
||||
writeTimeout: writeTimeout, |
||||
streamType: streamType, |
||||
pc: pc, |
||||
readBuf: multibuffer.New(1, readBufferSize), |
||||
publishers: make(map[publisherAddr]*publisherData), |
||||
done: make(chan struct{}), |
||||
} |
||||
|
||||
var label string |
||||
if s.streamType == gortsplib.StreamTypeRtp { |
||||
label = "RTP" |
||||
} else { |
||||
label = "RTCP" |
||||
} |
||||
parent.Log(logger.Info, "[UDP/"+label+" server] opened on :%d", port) |
||||
|
||||
go s.run() |
||||
return s, nil |
||||
} |
||||
|
||||
// Close closes a Server.
|
||||
func (s *Server) Close() { |
||||
s.pc.Close() |
||||
<-s.done |
||||
} |
||||
|
||||
func (s *Server) run() { |
||||
defer close(s.done) |
||||
|
||||
for { |
||||
buf := s.readBuf.Next() |
||||
n, addr, err := s.pc.ReadFromUDP(buf) |
||||
if err != nil { |
||||
break |
||||
} |
||||
|
||||
func() { |
||||
s.publishersMutex.RLock() |
||||
defer s.publishersMutex.RUnlock() |
||||
|
||||
// find publisher data
|
||||
var pubAddr publisherAddr |
||||
pubAddr.fill(addr.IP, addr.Port) |
||||
pubData, ok := s.publishers[pubAddr] |
||||
if !ok { |
||||
return |
||||
} |
||||
|
||||
pubData.publisher.OnUDPPublisherFrame(pubData.trackID, s.streamType, buf[:n]) |
||||
}() |
||||
} |
||||
} |
||||
|
||||
// Port returns the server local port.
|
||||
func (s *Server) Port() int { |
||||
return s.pc.LocalAddr().(*net.UDPAddr).Port |
||||
} |
||||
|
||||
// Write writes a UDP packet.
|
||||
func (s *Server) Write(buf []byte, addr *net.UDPAddr) { |
||||
s.writeMutex.Lock() |
||||
defer s.writeMutex.Unlock() |
||||
|
||||
s.pc.SetWriteDeadline(time.Now().Add(s.writeTimeout)) |
||||
s.pc.WriteTo(buf, addr) |
||||
} |
||||
|
||||
// AddPublisher adds a publisher.
|
||||
func (s *Server) AddPublisher(ip net.IP, port int, publisher Publisher, trackID int) { |
||||
s.publishersMutex.Lock() |
||||
defer s.publishersMutex.Unlock() |
||||
|
||||
var addr publisherAddr |
||||
addr.fill(ip, port) |
||||
|
||||
s.publishers[addr] = &publisherData{ |
||||
publisher: publisher, |
||||
trackID: trackID, |
||||
} |
||||
} |
||||
|
||||
// RemovePublisher removes a publisher.
|
||||
func (s *Server) RemovePublisher(ip net.IP, port int, publisher Publisher) { |
||||
s.publishersMutex.Lock() |
||||
defer s.publishersMutex.Unlock() |
||||
|
||||
var addr publisherAddr |
||||
addr.fill(ip, port) |
||||
|
||||
delete(s.publishers, addr) |
||||
} |
||||
@ -0,0 +1,34 @@
@@ -0,0 +1,34 @@
|
||||
package serverudpl |
||||
|
||||
import ( |
||||
"strconv" |
||||
|
||||
"github.com/aler9/gortsplib" |
||||
"github.com/aler9/rtsp-simple-server/internal/logger" |
||||
) |
||||
|
||||
// Parent is implemented by program.
|
||||
type Parent interface { |
||||
Log(logger.Level, string, ...interface{}) |
||||
} |
||||
|
||||
// New allocates a gortsplib.ServerUDPListener.
|
||||
func New(port int, |
||||
streamType gortsplib.StreamType, |
||||
parent Parent) (*gortsplib.ServerUDPListener, error) { |
||||
|
||||
listener, err := gortsplib.NewServerUDPListener(":" + strconv.FormatInt(int64(port), 10)) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
label := func() string { |
||||
if streamType == gortsplib.StreamTypeRTP { |
||||
return "RTP" |
||||
} |
||||
return "RTCP" |
||||
}() |
||||
parent.Log(logger.Info, "[UDP/"+label+" listener] opened on :%d", port) |
||||
|
||||
return listener, nil |
||||
} |
||||
Loading…
Reference in new issue