938 lines
22 KiB
Go
938 lines
22 KiB
Go
package rtsp
|
||
|
||
import (
|
||
"bufio"
|
||
"bytes"
|
||
"crypto/md5"
|
||
"encoding/base64"
|
||
"encoding/binary"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"github.com/nareix/av"
|
||
"github.com/nareix/av/pktqueue"
|
||
"github.com/nareix/codec"
|
||
"github.com/nareix/codec/aacparser"
|
||
"github.com/nareix/codec/h264parser"
|
||
"github.com/nareix/rtsp/sdp"
|
||
"html"
|
||
"io"
|
||
"net"
|
||
"net/textproto"
|
||
"net/url"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
var ErrCodecDataChange = fmt.Errorf("rtsp: codec data change, please call HandleCodecDataChange()")
|
||
|
||
type Client struct {
|
||
DebugRtsp bool
|
||
DebugRtp bool
|
||
Headers []string
|
||
|
||
RtspTimeout time.Duration
|
||
RtpTimeout time.Duration
|
||
RtpKeepAliveTimeout time.Duration
|
||
rtpKeepaliveTimer time.Time
|
||
rtpKeepaliveEnterCnt int
|
||
|
||
setupCalled bool
|
||
setupIdx []int
|
||
setupMap []int
|
||
playCalled bool
|
||
|
||
authHeaders func(method string) []string
|
||
|
||
url *url.URL
|
||
conn *connWithTimeout
|
||
rconn io.Reader
|
||
requestUri string
|
||
cseq uint
|
||
streams []*Stream
|
||
session string
|
||
body io.Reader
|
||
pktque *pktqueue.Queue
|
||
}
|
||
|
||
type Request struct {
|
||
Header []string
|
||
Uri string
|
||
Method string
|
||
}
|
||
|
||
type Response struct {
|
||
BlockLength int
|
||
Block []byte
|
||
BlockNo int
|
||
|
||
StatusCode int
|
||
Header textproto.MIMEHeader
|
||
ContentLength int
|
||
Body []byte
|
||
}
|
||
|
||
func DialTimeout(uri string, timeout time.Duration) (self *Client, err error) {
|
||
var URL *url.URL
|
||
if URL, err = url.Parse(html.UnescapeString(uri)); err != nil {
|
||
return
|
||
}
|
||
|
||
if _, _, err := net.SplitHostPort(URL.Host); err != nil {
|
||
URL.Host = URL.Host + ":554"
|
||
}
|
||
|
||
dailer := net.Dialer{Timeout: timeout}
|
||
var conn net.Conn
|
||
if conn, err = dailer.Dial("tcp", URL.Host); err != nil {
|
||
return
|
||
}
|
||
|
||
u2 := *URL
|
||
u2.User = nil
|
||
|
||
connt := &connWithTimeout{Conn: conn}
|
||
|
||
self = &Client{
|
||
conn: connt,
|
||
rconn: connt,
|
||
url: URL,
|
||
requestUri: u2.String(),
|
||
}
|
||
return
|
||
}
|
||
|
||
func Dial(uri string) (self *Client, err error) {
|
||
return DialTimeout(uri, 0)
|
||
}
|
||
|
||
func (self *Client) Streams() (streams []av.CodecData, err error) {
|
||
if self.setupCalled {
|
||
for _, i := range self.setupIdx {
|
||
streams = append(streams, self.streams[i].CodecData)
|
||
}
|
||
} else {
|
||
err = fmt.Errorf("rtsp: no streams")
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
func (self *Client) sendRtpKeepalive() (err error) {
|
||
if self.RtpKeepAliveTimeout > 0 && self.rtpKeepaliveEnterCnt == 0 {
|
||
self.rtpKeepaliveEnterCnt++
|
||
defer func() {
|
||
self.rtpKeepaliveEnterCnt--
|
||
}()
|
||
if self.rtpKeepaliveTimer.IsZero() {
|
||
self.rtpKeepaliveTimer = time.Now()
|
||
} else if time.Now().Sub(self.rtpKeepaliveTimer) > self.RtpKeepAliveTimeout {
|
||
self.rtpKeepaliveTimer = time.Now()
|
||
if self.DebugRtsp {
|
||
fmt.Println("rtp: keep alive")
|
||
}
|
||
if err = self.Options(); err != nil {
|
||
return
|
||
}
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
func (self *Client) WriteRequest(req Request) (err error) {
|
||
self.conn.Timeout = self.RtspTimeout
|
||
self.cseq++
|
||
|
||
buf := &bytes.Buffer{}
|
||
|
||
fmt.Fprintf(buf, "%s %s RTSP/1.0\r\n", req.Method, req.Uri)
|
||
fmt.Fprintf(buf, "CSeq: %d\r\n", self.cseq)
|
||
|
||
if self.authHeaders != nil {
|
||
headers := self.authHeaders(req.Method)
|
||
for _, s := range headers {
|
||
io.WriteString(buf, s)
|
||
io.WriteString(buf, "\r\n")
|
||
}
|
||
}
|
||
for _, s := range req.Header {
|
||
io.WriteString(buf, s)
|
||
io.WriteString(buf, "\r\n")
|
||
}
|
||
for _, s := range self.Headers {
|
||
io.WriteString(buf, s)
|
||
io.WriteString(buf, "\r\n")
|
||
}
|
||
io.WriteString(buf, "\r\n")
|
||
|
||
bufout := buf.Bytes()
|
||
|
||
if self.DebugRtsp {
|
||
fmt.Print("> ", string(bufout))
|
||
}
|
||
|
||
if _, err = self.conn.Write(bufout); err != nil {
|
||
return
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Client) ReadResponse() (res Response, err error) {
|
||
var br *bufio.Reader
|
||
|
||
defer func() {
|
||
if br != nil {
|
||
buf, _ := br.Peek(br.Buffered())
|
||
self.rconn = io.MultiReader(bytes.NewReader(buf), self.rconn)
|
||
}
|
||
if res.StatusCode == 200 {
|
||
self.conn.Timeout = self.RtspTimeout
|
||
|
||
if res.ContentLength > 0 {
|
||
res.Body = make([]byte, res.ContentLength)
|
||
if _, err = io.ReadFull(self.rconn, res.Body); err != nil {
|
||
return
|
||
}
|
||
}
|
||
} else if res.BlockLength > 0 {
|
||
self.conn.Timeout = self.RtpTimeout
|
||
res.Block = make([]byte, res.BlockLength)
|
||
if _, err = io.ReadFull(self.rconn, res.Block); err != nil {
|
||
return
|
||
}
|
||
if err = self.sendRtpKeepalive(); err != nil {
|
||
return
|
||
}
|
||
}
|
||
}()
|
||
|
||
self.conn.Timeout = self.RtspTimeout
|
||
var h [4]byte
|
||
if _, err = io.ReadFull(self.rconn, h[:]); err != nil {
|
||
return
|
||
}
|
||
|
||
if h[0] == 36 {
|
||
// $
|
||
res.BlockLength = int(h[2])<<8 + int(h[3])
|
||
res.BlockNo = int(h[1])
|
||
if self.DebugRtsp {
|
||
fmt.Println("block: len", res.BlockLength, "no", res.BlockNo)
|
||
}
|
||
return
|
||
} else if h[0] == 82 && h[1] == 84 && h[2] == 83 && h[3] == 80 {
|
||
// RTSP 200 OK
|
||
self.rconn = io.MultiReader(bytes.NewReader(h[:]), self.rconn)
|
||
} else {
|
||
self.conn.Timeout = self.RtpTimeout
|
||
|
||
for {
|
||
if self.DebugRtsp {
|
||
fmt.Println("block: relocate try")
|
||
}
|
||
|
||
for {
|
||
var b [1]byte
|
||
if _, err = self.rconn.Read(b[:]); err != nil {
|
||
return
|
||
}
|
||
if b[0] == 36 {
|
||
break
|
||
}
|
||
}
|
||
if _, err = io.ReadFull(self.rconn, h[1:4]); err != nil {
|
||
return
|
||
}
|
||
|
||
res.BlockLength = int(h[2])<<8 + int(h[3])
|
||
res.BlockNo = int(h[1])
|
||
if res.BlockNo/2 < len(self.streams) {
|
||
break
|
||
}
|
||
}
|
||
|
||
if self.DebugRtsp {
|
||
fmt.Println("block: relocate done")
|
||
fmt.Println("block: len", res.BlockLength, "no", res.BlockNo)
|
||
}
|
||
return
|
||
}
|
||
|
||
br = bufio.NewReader(self.rconn)
|
||
tp := textproto.NewReader(br)
|
||
|
||
var line string
|
||
if line, err = tp.ReadLine(); err != nil {
|
||
return
|
||
}
|
||
if self.DebugRtsp {
|
||
fmt.Println("<", line)
|
||
}
|
||
|
||
fline := strings.SplitN(line, " ", 3)
|
||
if len(fline) < 2 {
|
||
err = fmt.Errorf("rtsp: malformed response line")
|
||
return
|
||
}
|
||
|
||
if res.StatusCode, err = strconv.Atoi(fline[1]); err != nil {
|
||
return
|
||
}
|
||
var header textproto.MIMEHeader
|
||
if header, err = tp.ReadMIMEHeader(); err != nil {
|
||
return
|
||
}
|
||
|
||
if self.DebugRtsp {
|
||
for k, s := range header {
|
||
fmt.Println(k, s)
|
||
}
|
||
fmt.Println()
|
||
}
|
||
|
||
if res.StatusCode != 200 && res.StatusCode != 401 {
|
||
err = fmt.Errorf("rtsp: StatusCode=%d invalid", res.StatusCode)
|
||
return
|
||
}
|
||
|
||
if res.StatusCode == 401 {
|
||
/*
|
||
RTSP/1.0 401 Unauthorized
|
||
CSeq: 2
|
||
Date: Wed, May 04 2016 10:10:51 GMT
|
||
WWW-Authenticate: Digest realm="LIVE555 Streaming Media", nonce="c633aaf8b83127633cbe98fac1d20d87"
|
||
*/
|
||
authval := header.Get("WWW-Authenticate")
|
||
hdrval := strings.SplitN(authval, " ", 2)
|
||
var realm, nonce string
|
||
|
||
if len(hdrval) == 2 {
|
||
for _, field := range strings.Split(hdrval[1], ",") {
|
||
field = strings.Trim(field, ", ")
|
||
if keyval := strings.Split(field, "="); len(keyval) == 2 {
|
||
key := keyval[0]
|
||
val := strings.Trim(keyval[1], `"`)
|
||
switch key {
|
||
case "realm":
|
||
realm = val
|
||
case "nonce":
|
||
nonce = val
|
||
}
|
||
}
|
||
}
|
||
|
||
if realm != "" && nonce != "" {
|
||
if self.url.User == nil {
|
||
err = fmt.Errorf("rtsp: please provide username and password")
|
||
return
|
||
}
|
||
var username string
|
||
var password string
|
||
var ok bool
|
||
username = self.url.User.Username()
|
||
if password, ok = self.url.User.Password(); !ok {
|
||
err = fmt.Errorf("rtsp: please provide password")
|
||
return
|
||
}
|
||
hs1 := md5hash(username + ":" + realm + ":" + password)
|
||
|
||
self.authHeaders = func(method string) []string {
|
||
hs2 := md5hash(method + ":" + self.requestUri)
|
||
response := md5hash(hs1 + ":" + nonce + ":" + hs2)
|
||
return []string{
|
||
fmt.Sprintf(`Authorization: Digest username="%s", realm="%s", nonce="%s", uri="%s", response="%s"`,
|
||
username, realm, nonce, self.requestUri, response),
|
||
fmt.Sprintf(`Authorization: Basic %s`, base64.StdEncoding.EncodeToString([]byte(username+":"+password))),
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
if sess := header.Get("Session"); sess != "" && self.session == "" {
|
||
if fields := strings.Split(sess, ";"); len(fields) > 0 {
|
||
self.session = fields[0]
|
||
}
|
||
}
|
||
|
||
res.ContentLength, _ = strconv.Atoi(header.Get("Content-Length"))
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Client) setupAll() (err error) {
|
||
idx := []int{}
|
||
for i := range self.streams {
|
||
idx = append(idx, i)
|
||
}
|
||
return self.Setup(idx)
|
||
}
|
||
|
||
func (self *Client) Setup(idx []int) (err error) {
|
||
if self.setupCalled {
|
||
err = fmt.Errorf("rtsp: Setup() called twice")
|
||
return
|
||
}
|
||
|
||
if len(self.streams) == 0 {
|
||
err = fmt.Errorf("rtsp: no streams, please call Describe() first")
|
||
return
|
||
}
|
||
|
||
self.setupMap = make([]int, len(self.streams))
|
||
for i := range self.setupMap {
|
||
self.setupMap[i] = -1
|
||
}
|
||
self.setupIdx = idx
|
||
|
||
for i, si := range idx {
|
||
self.setupMap[si] = i
|
||
|
||
uri := ""
|
||
control := self.streams[si].Sdp.Control
|
||
if strings.HasPrefix(control, "rtsp://") {
|
||
uri = control
|
||
} else {
|
||
uri = self.requestUri + "/" + control
|
||
}
|
||
req := Request{Method: "SETUP", Uri: uri}
|
||
req.Header = append(req.Header, fmt.Sprintf("Transport: RTP/AVP/TCP;unicast;interleaved=%d-%d", si*2, si*2+1))
|
||
if self.session != "" {
|
||
req.Header = append(req.Header, "Session: "+self.session)
|
||
}
|
||
if err = self.WriteRequest(req); err != nil {
|
||
return
|
||
}
|
||
if _, err = self.ReadResponse(); err != nil {
|
||
return
|
||
}
|
||
}
|
||
|
||
self.setupCalled = true
|
||
return
|
||
}
|
||
|
||
func md5hash(s string) string {
|
||
h := md5.Sum([]byte(s))
|
||
return hex.EncodeToString(h[:])
|
||
}
|
||
|
||
func (self *Client) Describe() (streams []av.CodecData, err error) {
|
||
var res Response
|
||
|
||
for i := 0; i < 2; i++ {
|
||
req := Request{
|
||
Method: "DESCRIBE",
|
||
Uri: self.requestUri,
|
||
Header: []string{"Accept: application/sdp"},
|
||
}
|
||
if err = self.WriteRequest(req); err != nil {
|
||
return
|
||
}
|
||
if res, err = self.ReadResponse(); err != nil {
|
||
return
|
||
}
|
||
if res.StatusCode == 200 {
|
||
break
|
||
}
|
||
}
|
||
if res.ContentLength == 0 {
|
||
err = fmt.Errorf("rtsp: Describe failed, StatusCode=%d", res.StatusCode)
|
||
return
|
||
}
|
||
|
||
body := string(res.Body)
|
||
|
||
if self.DebugRtsp {
|
||
fmt.Println("<", body)
|
||
}
|
||
|
||
self.streams = []*Stream{}
|
||
_, medias := sdp.Parse(body)
|
||
|
||
for _, media := range medias {
|
||
stream := &Stream{Sdp: media, client: self}
|
||
if err = stream.makeCodecData(); err != nil {
|
||
return
|
||
}
|
||
self.streams = append(self.streams, stream)
|
||
}
|
||
|
||
for _, stream := range self.streams {
|
||
streams = append(streams, stream)
|
||
}
|
||
self.pktque = &pktqueue.Queue{
|
||
Poll: self.poll,
|
||
}
|
||
self.pktque.Alloc(streams)
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Client) Options() (err error) {
|
||
req := Request{
|
||
Method: "OPTIONS",
|
||
Uri: self.requestUri,
|
||
}
|
||
if self.session != "" {
|
||
req.Header = append(req.Header, "Session: "+self.session)
|
||
}
|
||
if err = self.WriteRequest(req); err != nil {
|
||
return
|
||
}
|
||
if _, err = self.ReadResponse(); err != nil {
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
func (self *Client) HandleCodecDataChange() (_newcli *Client, err error) {
|
||
newcli := &Client{}
|
||
*newcli = *self
|
||
|
||
newcli.streams = []*Stream{}
|
||
for _, stream := range self.streams {
|
||
newstream := &Stream{}
|
||
*newstream = *stream
|
||
newstream.client = newcli
|
||
|
||
if newstream.isCodecDataChange() {
|
||
if err = newstream.makeCodecData(); err != nil {
|
||
return
|
||
}
|
||
newstream.clearCodecDataChange()
|
||
}
|
||
newcli.streams = append(newcli.streams, newstream)
|
||
}
|
||
|
||
_newcli = newcli
|
||
return
|
||
}
|
||
|
||
func (self *Stream) clearCodecDataChange() {
|
||
self.spsChanged = false
|
||
self.ppsChanged = false
|
||
}
|
||
|
||
func (self *Stream) isCodecDataChange() bool {
|
||
if self.spsChanged && self.ppsChanged {
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
|
||
func (self *Stream) makeCodecData() (err error) {
|
||
media := self.Sdp
|
||
|
||
if media.PayloadType >= 96 && media.PayloadType <= 127 {
|
||
switch media.Type {
|
||
case av.H264:
|
||
for _, nalu := range media.SpropParameterSets {
|
||
if len(nalu) > 0 {
|
||
switch nalu[0] & 0x1f {
|
||
case 7:
|
||
if len(self.sps) == 0 {
|
||
self.sps = nalu
|
||
}
|
||
case 8:
|
||
if len(self.pps) == 0 {
|
||
self.pps = nalu
|
||
}
|
||
}
|
||
}
|
||
}
|
||
if len(self.sps) > 0 && len(self.pps) > 0 {
|
||
if self.CodecData, err = h264parser.NewCodecDataFromSPSAndPPS(self.sps, self.pps); err != nil {
|
||
err = fmt.Errorf("rtsp: h264 sps/pps invalid: %s", err)
|
||
return
|
||
}
|
||
} else {
|
||
err = fmt.Errorf("rtsp: missing h264 sps or pps")
|
||
return
|
||
}
|
||
|
||
case av.AAC:
|
||
if len(media.Config) == 0 {
|
||
err = fmt.Errorf("rtsp: aac sdp config missing")
|
||
return
|
||
}
|
||
if self.CodecData, err = aacparser.NewCodecDataFromMPEG4AudioConfigBytes(media.Config); err != nil {
|
||
err = fmt.Errorf("rtsp: aac sdp config invalid: %s", err)
|
||
return
|
||
}
|
||
}
|
||
} else {
|
||
switch media.PayloadType {
|
||
case 0:
|
||
self.CodecData = codec.NewPCMMulawCodecData()
|
||
|
||
case 8:
|
||
self.CodecData = codec.NewPCMAlawCodecData()
|
||
|
||
default:
|
||
err = fmt.Errorf("rtsp: PayloadType=%d unsupported", media.PayloadType)
|
||
return
|
||
}
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Stream) handleH264Payload(naluType byte, timestamp uint32, packet []byte) (err error) {
|
||
/*
|
||
Table 7-1 – NAL unit type codes
|
||
1 Coded slice of a non-IDR picture
|
||
5 Coded slice of an IDR picture
|
||
6 Supplemental enhancement information (SEI)
|
||
7 Sequence parameter set
|
||
8 Picture parameter set
|
||
*/
|
||
switch naluType {
|
||
case 6: // SEI ignored
|
||
|
||
case 7: // sps
|
||
if self.client != nil && self.client.DebugRtp {
|
||
fmt.Println("rtsp: got sps")
|
||
}
|
||
if bytes.Compare(self.sps, packet) != 0 {
|
||
self.spsChanged = true
|
||
self.sps = packet
|
||
if self.client != nil && self.client.DebugRtp {
|
||
fmt.Println("rtsp: sps changed")
|
||
}
|
||
}
|
||
|
||
case 8: // pps
|
||
if self.client != nil && self.client.DebugRtp {
|
||
fmt.Println("rtsp: got pps")
|
||
}
|
||
if bytes.Compare(self.pps, packet) != 0 {
|
||
self.ppsChanged = true
|
||
self.pps = packet
|
||
if self.client != nil && self.client.DebugRtp {
|
||
fmt.Println("rtsp: pps changed")
|
||
}
|
||
}
|
||
|
||
default:
|
||
if naluType == 5 {
|
||
self.pkt.IsKeyFrame = true
|
||
}
|
||
self.gotpkt = true
|
||
self.pkt.Data = packet
|
||
self.timestamp = timestamp
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Stream) handlePacket(timestamp uint32, packet []byte) (err error) {
|
||
if self.isCodecDataChange() {
|
||
err = ErrCodecDataChange
|
||
return
|
||
}
|
||
|
||
switch self.Type() {
|
||
case av.H264:
|
||
/*
|
||
+---------------+
|
||
|0|1|2|3|4|5|6|7|
|
||
+-+-+-+-+-+-+-+-+
|
||
|F|NRI| Type |
|
||
+---------------+
|
||
*/
|
||
naluType := packet[0] & 0x1f
|
||
|
||
/*
|
||
NAL Unit Packet Packet Type Name Section
|
||
Type Type
|
||
-------------------------------------------------------------
|
||
0 reserved -
|
||
1-23 NAL unit Single NAL unit packet 5.6
|
||
24 STAP-A Single-time aggregation packet 5.7.1
|
||
25 STAP-B Single-time aggregation packet 5.7.1
|
||
26 MTAP16 Multi-time aggregation packet 5.7.2
|
||
27 MTAP24 Multi-time aggregation packet 5.7.2
|
||
28 FU-A Fragmentation unit 5.8
|
||
29 FU-B Fragmentation unit 5.8
|
||
30-31 reserved -
|
||
*/
|
||
|
||
switch {
|
||
|
||
case naluType >= 1 && naluType <= 23:
|
||
if err = self.handleH264Payload(naluType, timestamp, packet); err != nil {
|
||
return
|
||
}
|
||
|
||
case naluType == 28: // FU-A
|
||
/*
|
||
0 1 2 3
|
||
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
|
||
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
| FU indicator | FU header | |
|
||
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
|
||
| |
|
||
| FU payload |
|
||
| |
|
||
| +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
| :...OPTIONAL RTP padding |
|
||
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
Figure 14. RTP payload format for FU-A
|
||
|
||
The FU indicator octet has the following format:
|
||
+---------------+
|
||
|0|1|2|3|4|5|6|7|
|
||
+-+-+-+-+-+-+-+-+
|
||
|F|NRI| Type |
|
||
+---------------+
|
||
|
||
|
||
The FU header has the following format:
|
||
+---------------+
|
||
|0|1|2|3|4|5|6|7|
|
||
+-+-+-+-+-+-+-+-+
|
||
|S|E|R| Type |
|
||
+---------------+
|
||
|
||
S: 1 bit
|
||
When set to one, the Start bit indicates the start of a fragmented
|
||
NAL unit. When the following FU payload is not the start of a
|
||
fragmented NAL unit payload, the Start bit is set to zero.
|
||
|
||
E: 1 bit
|
||
When set to one, the End bit indicates the end of a fragmented NAL
|
||
unit, i.e., the last byte of the payload is also the last byte of
|
||
the fragmented NAL unit. When the following FU payload is not the
|
||
last fragment of a fragmented NAL unit, the End bit is set to
|
||
zero.
|
||
|
||
R: 1 bit
|
||
The Reserved bit MUST be equal to 0 and MUST be ignored by the
|
||
receiver.
|
||
|
||
Type: 5 bits
|
||
The NAL unit payload type as defined in table 7-1 of [1].
|
||
*/
|
||
fuIndicator := packet[0]
|
||
fuHeader := packet[1]
|
||
isStart := fuHeader&0x80 != 0
|
||
isEnd := fuHeader&0x40 != 0
|
||
naluType := fuHeader & 0x1f
|
||
if isStart {
|
||
self.fuBuffer = []byte{fuIndicator&0xe0 | fuHeader&0x1f}
|
||
}
|
||
self.fuBuffer = append(self.fuBuffer, packet[2:]...)
|
||
if isEnd {
|
||
if err = self.handleH264Payload(naluType, timestamp, self.fuBuffer); err != nil {
|
||
return
|
||
}
|
||
}
|
||
|
||
case naluType == 24:
|
||
err = fmt.Errorf("rtsp: unsupported H264 STAP-A")
|
||
return
|
||
|
||
default:
|
||
err = fmt.Errorf("rtsp: unsupported H264 naluType=%d", naluType)
|
||
return
|
||
}
|
||
|
||
case av.AAC:
|
||
self.gotpkt = true
|
||
self.pkt.Data = packet[4:]
|
||
self.timestamp = timestamp
|
||
|
||
default:
|
||
self.gotpkt = true
|
||
self.pkt.Data = packet
|
||
self.timestamp = timestamp
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Client) parseBlock(blockNo int, packet []byte) (streamIndex int, err error) {
|
||
if blockNo%2 != 0 {
|
||
// rtcp block
|
||
return
|
||
}
|
||
|
||
streamIndex = blockNo / 2
|
||
if streamIndex >= len(self.streams) {
|
||
err = fmt.Errorf("rtsp: parseBlock: streamIndex=%d invalid", streamIndex)
|
||
return
|
||
}
|
||
stream := self.streams[streamIndex]
|
||
|
||
/*
|
||
0 1 2 3
|
||
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
|
||
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
|V=2|P|X| CC |M| PT | sequence number |
|
||
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
| timestamp |
|
||
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
| synchronization source (SSRC) identifier |
|
||
+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+
|
||
| contributing source (CSRC) identifiers |
|
||
| .... |
|
||
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
*/
|
||
|
||
if len(packet) < 8 {
|
||
err = fmt.Errorf("rtp: packet too short")
|
||
return
|
||
}
|
||
payloadOffset := 12 + int(packet[0]&0xf)*4
|
||
if payloadOffset+2 > len(packet) {
|
||
err = fmt.Errorf("rtp: packet too short")
|
||
return
|
||
}
|
||
|
||
timestamp := binary.BigEndian.Uint32(packet[4:8])
|
||
payload := packet[payloadOffset:]
|
||
|
||
/*
|
||
PT Encoding Name Audio/Video (A/V) Clock Rate (Hz) Channels Reference
|
||
0 PCMU A 8000 1 [RFC3551]
|
||
1 Reserved
|
||
2 Reserved
|
||
3 GSM A 8000 1 [RFC3551]
|
||
4 G723 A 8000 1 [Vineet_Kumar][RFC3551]
|
||
5 DVI4 A 8000 1 [RFC3551]
|
||
6 DVI4 A 16000 1 [RFC3551]
|
||
7 LPC A 8000 1 [RFC3551]
|
||
8 PCMA A 8000 1 [RFC3551]
|
||
9 G722 A 8000 1 [RFC3551]
|
||
10 L16 A 44100 2 [RFC3551]
|
||
11 L16 A 44100 1 [RFC3551]
|
||
12 QCELP A 8000 1 [RFC3551]
|
||
13 CN A 8000 1 [RFC3389]
|
||
14 MPA A 90000 [RFC3551][RFC2250]
|
||
15 G728 A 8000 1 [RFC3551]
|
||
16 DVI4 A 11025 1 [Joseph_Di_Pol]
|
||
17 DVI4 A 22050 1 [Joseph_Di_Pol]
|
||
18 G729 A 8000 1 [RFC3551]
|
||
19 Reserved A
|
||
20 Unassigned A
|
||
21 Unassigned A
|
||
22 Unassigned A
|
||
23 Unassigned A
|
||
24 Unassigned V
|
||
25 CelB V 90000 [RFC2029]
|
||
26 JPEG V 90000 [RFC2435]
|
||
27 Unassigned V
|
||
28 nv V 90000 [RFC3551]
|
||
29 Unassigned V
|
||
30 Unassigned V
|
||
31 H261 V 90000 [RFC4587]
|
||
32 MPV V 90000 [RFC2250]
|
||
33 MP2T AV 90000 [RFC2250]
|
||
34 H263 V 90000 [Chunrong_Zhu]
|
||
35-71 Unassigned ?
|
||
72-76 Reserved for RTCP conflict avoidance [RFC3551]
|
||
77-95 Unassigned ?
|
||
96-127 dynamic ? [RFC3551]
|
||
*/
|
||
//payloadType := packet[1]&0x7f
|
||
|
||
if self.DebugRtsp {
|
||
//fmt.Println("packet:", stream.Type(), "offset", payloadOffset, "pt", payloadType)
|
||
if len(packet) > 24 {
|
||
fmt.Println(hex.Dump(packet[:24]))
|
||
}
|
||
}
|
||
|
||
if err = stream.handlePacket(timestamp, payload); err != nil {
|
||
return
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Client) Play() (err error) {
|
||
req := Request{
|
||
Method: "PLAY",
|
||
Uri: self.requestUri,
|
||
}
|
||
req.Header = append(req.Header, "Session: "+self.session)
|
||
if err = self.WriteRequest(req); err != nil {
|
||
return
|
||
}
|
||
self.playCalled = true
|
||
return
|
||
}
|
||
|
||
func (self *Client) poll() (err error) {
|
||
for {
|
||
var res Response
|
||
if res, err = self.ReadResponse(); err != nil {
|
||
return
|
||
}
|
||
if res.BlockLength > 0 {
|
||
var i int
|
||
if i, err = self.parseBlock(res.BlockNo, res.Block); err != nil {
|
||
return
|
||
}
|
||
stream := self.streams[i]
|
||
if stream.gotpkt {
|
||
time := float64(stream.timestamp) / float64(stream.Sdp.TimeScale)
|
||
if false {
|
||
fmt.Printf("rtsp: #%d %d/%d %d\n", i, stream.timestamp, stream.Sdp.TimeScale, len(stream.pkt.Data))
|
||
}
|
||
self.pktque.WriteTimePacket(self.setupMap[i], time, stream.pkt)
|
||
stream.pkt = av.Packet{}
|
||
stream.gotpkt = false
|
||
return
|
||
}
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
func (self *Client) ReadPacket() (i int, pkt av.Packet, err error) {
|
||
if !self.setupCalled {
|
||
if err = self.setupAll(); err != nil {
|
||
return
|
||
}
|
||
}
|
||
if !self.playCalled {
|
||
if err = self.Play(); err != nil {
|
||
return
|
||
}
|
||
}
|
||
return self.pktque.ReadPacket()
|
||
}
|
||
|
||
func (self *Client) ReadHeader() (err error) {
|
||
if _, err = self.Options(); err != nil {
|
||
return
|
||
}
|
||
if _, err = self.Describe(); err != nil {
|
||
return
|
||
}
|
||
if err = self.setupAll(); err != nil {
|
||
return
|
||
}
|
||
if err = self.Play(); err != nil {
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
func Open(uri string) (cli *Client, err error) {
|
||
var _cli *Client
|
||
if _cli, err = Dial(uri); err != nil {
|
||
return
|
||
}
|
||
if err = _cli.ReadHeader(); err != nil {
|
||
return
|
||
}
|
||
_cli.rtpKeepaliveTimer = time.Now()
|
||
_cli.RtpKeepAliveTimeout = 20 * time.Second
|
||
cli = _cli
|
||
return
|
||
}
|