mirror of
https://github.com/aler9/gortsplib
synced 2025-10-05 07:06:58 +08:00
a1396206b5
* split tracks from medias * move tracks into dedicated package * move media into dedicated package * edit Medias.Marshal() in order to return SDP * add medias.Find() and simplify examples * improve coverage * fix rebase errors * replace TrackIDs with MediaIDs * implement media-specific and track-specific callbacks for reading RTCP and RTP packets * rename publish into record, read into play * add v2 tag * rename tracks into formats
151 lines
3.0 KiB
Go
151 lines
3.0 KiB
Go
// Package base contains the primitives of the RTSP protocol.
|
|
package base
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/aler9/gortsplib/v2/pkg/url"
|
|
)
|
|
|
|
const (
|
|
rtspProtocol10 = "RTSP/1.0"
|
|
requestMaxMethodLength = 64
|
|
requestMaxURLLength = 2048
|
|
requestMaxProtocolLength = 64
|
|
)
|
|
|
|
// Method is the method of a RTSP request.
|
|
type Method string
|
|
|
|
// methods.
|
|
const (
|
|
Announce Method = "ANNOUNCE"
|
|
Describe Method = "DESCRIBE"
|
|
GetParameter Method = "GET_PARAMETER"
|
|
Options Method = "OPTIONS"
|
|
Pause Method = "PAUSE"
|
|
Play Method = "PLAY"
|
|
Record Method = "RECORD"
|
|
Setup Method = "SETUP"
|
|
SetParameter Method = "SET_PARAMETER"
|
|
Teardown Method = "TEARDOWN"
|
|
)
|
|
|
|
// Request is a RTSP request.
|
|
type Request struct {
|
|
// request method
|
|
Method Method
|
|
|
|
// request url
|
|
URL *url.URL
|
|
|
|
// map of header values
|
|
Header Header
|
|
|
|
// optional body
|
|
Body []byte
|
|
}
|
|
|
|
// Read reads a request.
|
|
func (req *Request) Read(rb *bufio.Reader) error {
|
|
byts, err := readBytesLimited(rb, ' ', requestMaxMethodLength)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
req.Method = Method(byts[:len(byts)-1])
|
|
|
|
if req.Method == "" {
|
|
return fmt.Errorf("empty method")
|
|
}
|
|
|
|
byts, err = readBytesLimited(rb, ' ', requestMaxURLLength)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
rawURL := string(byts[:len(byts)-1])
|
|
|
|
ur, err := url.Parse(rawURL)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid URL (%v)", rawURL)
|
|
}
|
|
req.URL = ur
|
|
|
|
byts, err = readBytesLimited(rb, '\r', requestMaxProtocolLength)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
proto := byts[:len(byts)-1]
|
|
|
|
if string(proto) != rtspProtocol10 {
|
|
return fmt.Errorf("expected '%s', got %v", rtspProtocol10, proto)
|
|
}
|
|
|
|
err = readByteEqual(rb, '\n')
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = req.Header.read(rb)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = (*body)(&req.Body).read(req.Header, rb)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// MarshalSize returns the size of a Request.
|
|
func (req Request) MarshalSize() int {
|
|
n := 0
|
|
|
|
urStr := req.URL.CloneWithoutCredentials().String()
|
|
n += len([]byte(string(req.Method) + " " + urStr + " " + rtspProtocol10 + "\r\n"))
|
|
|
|
if len(req.Body) != 0 {
|
|
req.Header["Content-Length"] = HeaderValue{strconv.FormatInt(int64(len(req.Body)), 10)}
|
|
}
|
|
|
|
n += req.Header.marshalSize()
|
|
|
|
n += body(req.Body).marshalSize()
|
|
|
|
return n
|
|
}
|
|
|
|
// MarshalTo writes a Request.
|
|
func (req Request) MarshalTo(buf []byte) (int, error) {
|
|
pos := 0
|
|
|
|
urStr := req.URL.CloneWithoutCredentials().String()
|
|
pos += copy(buf[pos:], []byte(string(req.Method)+" "+urStr+" "+rtspProtocol10+"\r\n"))
|
|
|
|
if len(req.Body) != 0 {
|
|
req.Header["Content-Length"] = HeaderValue{strconv.FormatInt(int64(len(req.Body)), 10)}
|
|
}
|
|
|
|
pos += req.Header.marshalTo(buf[pos:])
|
|
|
|
pos += body(req.Body).marshalTo(buf[pos:])
|
|
|
|
return pos, nil
|
|
}
|
|
|
|
// Marshal writes a Request.
|
|
func (req Request) Marshal() ([]byte, error) {
|
|
buf := make([]byte, req.MarshalSize())
|
|
_, err := req.MarshalTo(buf)
|
|
return buf, err
|
|
}
|
|
|
|
// String implements fmt.Stringer.
|
|
func (req Request) String() string {
|
|
buf, _ := req.Marshal()
|
|
return string(buf)
|
|
}
|