mirror of
https://github.com/aler9/gortsplib
synced 2025-10-04 23:02:45 +08:00
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package gortsplib
|
|
|
|
import (
|
|
"bufio"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/aler9/gortsplib/pkg/base"
|
|
"github.com/aler9/gortsplib/pkg/multibuffer"
|
|
)
|
|
|
|
const (
|
|
serverReadBufferSize = 4096
|
|
serverWriteBufferSize = 4096
|
|
)
|
|
|
|
// ConnServerConf allows to configure a ConnServer.
|
|
type ConnServerConf struct {
|
|
// pre-existing TCP connection to wrap
|
|
Conn net.Conn
|
|
|
|
// (optional) timeout of read operations.
|
|
// It defaults to 10 seconds
|
|
ReadTimeout time.Duration
|
|
|
|
// (optional) timeout of write operations.
|
|
// It defaults to 10 seconds
|
|
WriteTimeout time.Duration
|
|
|
|
// (optional) read buffer count.
|
|
// If greater than 1, allows to pass buffers to routines different than the one
|
|
// that is reading frames.
|
|
// It defaults to 1
|
|
ReadBufferCount int
|
|
}
|
|
|
|
// ConnServer is a server-side RTSP connection.
|
|
type ConnServer struct {
|
|
conf ConnServerConf
|
|
br *bufio.Reader
|
|
bw *bufio.Writer
|
|
request *base.Request
|
|
frame *base.InterleavedFrame
|
|
tcpFrameBuffer *multibuffer.MultiBuffer
|
|
}
|
|
|
|
// NewConnServer allocates a ConnServer.
|
|
func NewConnServer(conf ConnServerConf) *ConnServer {
|
|
if conf.ReadTimeout == 0 {
|
|
conf.ReadTimeout = 10 * time.Second
|
|
}
|
|
if conf.WriteTimeout == 0 {
|
|
conf.WriteTimeout = 10 * time.Second
|
|
}
|
|
if conf.ReadBufferCount == 0 {
|
|
conf.ReadBufferCount = 1
|
|
}
|
|
|
|
return &ConnServer{
|
|
conf: conf,
|
|
br: bufio.NewReaderSize(conf.Conn, serverReadBufferSize),
|
|
bw: bufio.NewWriterSize(conf.Conn, serverWriteBufferSize),
|
|
request: &base.Request{},
|
|
frame: &base.InterleavedFrame{},
|
|
tcpFrameBuffer: multibuffer.New(conf.ReadBufferCount, clientTCPFrameReadBufferSize),
|
|
}
|
|
}
|
|
|
|
// Close closes all the ConnServer resources.
|
|
func (s *ConnServer) Close() error {
|
|
return s.conf.Conn.Close()
|
|
}
|
|
|
|
// NetConn returns the underlying net.Conn.
|
|
func (s *ConnServer) NetConn() net.Conn {
|
|
return s.conf.Conn
|
|
}
|
|
|
|
// ReadRequest reads a Request.
|
|
func (s *ConnServer) ReadRequest() (*base.Request, error) {
|
|
s.conf.Conn.SetReadDeadline(time.Time{}) // disable deadline
|
|
err := s.request.Read(s.br)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return s.request, nil
|
|
}
|
|
|
|
// ReadFrameTCPOrRequest reads an InterleavedFrame or a Request.
|
|
func (s *ConnServer) ReadFrameTCPOrRequest(timeout bool) (interface{}, error) {
|
|
s.frame.Content = s.tcpFrameBuffer.Next()
|
|
|
|
if timeout {
|
|
s.conf.Conn.SetReadDeadline(time.Now().Add(s.conf.ReadTimeout))
|
|
}
|
|
|
|
return base.ReadInterleavedFrameOrRequest(s.frame, s.request, s.br)
|
|
}
|
|
|
|
// WriteResponse writes a Response.
|
|
func (s *ConnServer) WriteResponse(res *base.Response) error {
|
|
s.conf.Conn.SetWriteDeadline(time.Now().Add(s.conf.WriteTimeout))
|
|
return res.Write(s.bw)
|
|
}
|
|
|
|
// WriteFrameTCP writes an InterleavedFrame.
|
|
func (s *ConnServer) WriteFrameTCP(trackId int, streamType StreamType, content []byte) error {
|
|
frame := base.InterleavedFrame{
|
|
TrackId: trackId,
|
|
StreamType: streamType,
|
|
Content: content,
|
|
}
|
|
|
|
s.conf.Conn.SetWriteDeadline(time.Now().Add(s.conf.WriteTimeout))
|
|
return frame.Write(s.bw)
|
|
}
|