mirror of https://github.com/q191201771/lal.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
350 lines
8.5 KiB
Go
350 lines
8.5 KiB
Go
package rtmp
|
|
|
|
import (
|
|
"bufio"
|
|
"encoding/hex"
|
|
"github.com/q191201771/nezha/pkg/bele"
|
|
"github.com/q191201771/nezha/pkg/log"
|
|
"github.com/q191201771/nezha/pkg/unique"
|
|
"net"
|
|
"strings"
|
|
"sync"
|
|
"sync/atomic"
|
|
)
|
|
|
|
// TODO chef: PubSession SubSession
|
|
|
|
// TODO chef: 没有进化成Pub Sub时的超时释放
|
|
|
|
type ServerSessionObserver interface {
|
|
NewRTMPPubSessionCB(session *ServerSession) // 上层代码应该在这个事件回调中注册音视频数据的监听
|
|
NewRTMPSubSessionCB(session *ServerSession)
|
|
//DelRTMPPubSessionCB(session *PubSession)
|
|
//DelRTMPSubSessionCB(session *SubSession)
|
|
}
|
|
|
|
type PubSessionObserver interface {
|
|
AVMsgObserver
|
|
}
|
|
|
|
func (s *ServerSession) SetPubSessionObserver(obs PubSessionObserver) {
|
|
s.avObs = obs
|
|
}
|
|
|
|
type ServerSessionType int
|
|
|
|
const (
|
|
ServerSessionTypeUnknown ServerSessionType = iota // 收到客户端的publish或者play信令之前的类型状态
|
|
ServerSessionTypePub
|
|
ServerSessionTypeSub
|
|
)
|
|
|
|
type ServerSession struct {
|
|
AppName string
|
|
StreamName string
|
|
StreamNameWithRawQuery string
|
|
UniqueKey string
|
|
|
|
obs ServerSessionObserver
|
|
t ServerSessionType
|
|
hs HandshakeServer
|
|
chunkComposer *ChunkComposer
|
|
packer *MessagePacker
|
|
|
|
conn net.Conn
|
|
rb *bufio.Reader
|
|
wb *bufio.Writer
|
|
wChan chan []byte
|
|
closeOnce sync.Once
|
|
exitChan chan struct{}
|
|
hasClosedFlag uint32
|
|
|
|
// only for PubSession
|
|
avObs PubSessionObserver
|
|
|
|
// only for SubSession
|
|
isFresh bool
|
|
waitKeyNalu bool
|
|
}
|
|
|
|
func NewServerSession(obs ServerSessionObserver, conn net.Conn) *ServerSession {
|
|
return &ServerSession{
|
|
UniqueKey: unique.GenUniqueKey("RTMPPUBSUB"),
|
|
obs: obs,
|
|
t: ServerSessionTypeUnknown,
|
|
chunkComposer: NewChunkComposer(),
|
|
packer: NewMessagePacker(),
|
|
conn: conn,
|
|
rb: bufio.NewReaderSize(conn, readBufSize),
|
|
wb: bufio.NewWriterSize(conn, writeBufSize),
|
|
wChan: make(chan []byte, wChanSize),
|
|
exitChan: make(chan struct{}),
|
|
isFresh: true,
|
|
waitKeyNalu: true,
|
|
}
|
|
}
|
|
|
|
func (s *ServerSession) RunLoop() (err error) {
|
|
if err = s.handshake(); err != nil {
|
|
s.dispose(err)
|
|
return err
|
|
}
|
|
|
|
go s.runWriteLoop()
|
|
|
|
if err = s.chunkComposer.RunLoop(s.rb, s.doMsg); err != nil {
|
|
s.dispose(err)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (s *ServerSession) Dispose() {
|
|
if atomic.LoadUint32(&s.hasClosedFlag) == 1 {
|
|
return
|
|
}
|
|
s.dispose(nil)
|
|
}
|
|
|
|
func (s *ServerSession) AsyncWrite(msg []byte) error {
|
|
if atomic.LoadUint32(&s.hasClosedFlag) == 1 {
|
|
return rtmpErr
|
|
}
|
|
|
|
s.wChan <- msg
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerSession) runReadLoop() error {
|
|
return s.chunkComposer.RunLoop(s.rb, s.doMsg)
|
|
}
|
|
|
|
func (s *ServerSession) runWriteLoop() {
|
|
for {
|
|
select {
|
|
case <-s.exitChan:
|
|
return
|
|
case msg := <-s.wChan:
|
|
if _, err := s.conn.Write(msg); err != nil {
|
|
s.dispose(err)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *ServerSession) dispose(err error) {
|
|
s.closeOnce.Do(func() {
|
|
atomic.StoreUint32(&s.hasClosedFlag, 1)
|
|
close(s.exitChan)
|
|
if err := s.conn.Close(); err != nil {
|
|
log.Errorf("conn close error. [%s] err=%v", s.UniqueKey, err)
|
|
}
|
|
})
|
|
}
|
|
|
|
func (s *ServerSession) handshake() error {
|
|
if err := s.hs.ReadC0C1(s.rb); err != nil {
|
|
return err
|
|
}
|
|
if err := s.hs.WriteS0S1S2(s.conn); err != nil {
|
|
return err
|
|
}
|
|
if err := s.hs.ReadC2(s.rb); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerSession) doMsg(stream *Stream) error {
|
|
//log.Debugf("%d %d %v", stream.header.msgTypeID, stream.msgLen, stream.header)
|
|
switch stream.header.MsgTypeID {
|
|
case typeidSetChunkSize:
|
|
// TODO chef:
|
|
case typeidCommandMessageAMF0:
|
|
return s.doCommandMessage(stream)
|
|
case TypeidDataMessageAMF0:
|
|
return s.doDataMessageAMF0(stream)
|
|
case typeidAck:
|
|
return s.doACK(stream)
|
|
case TypeidAudio:
|
|
fallthrough
|
|
case TypeidVideo:
|
|
if s.t != ServerSessionTypePub {
|
|
log.Errorf("read audio/video message but server session not pub type. [%s]", s.UniqueKey)
|
|
return rtmpErr
|
|
}
|
|
//log.Infof("t:%d ts:%d len:%d", stream.header.MsgTypeID, stream.timestampAbs, stream.msg.e - stream.msg.b)
|
|
s.avObs.ReadRTMPAVMsgCB(stream.header, stream.timestampAbs, stream.msg.buf[stream.msg.b:stream.msg.e])
|
|
default:
|
|
log.Warnf("unknown message. [%s] typeid=%d", s.UniqueKey, stream.header.MsgTypeID)
|
|
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerSession) doACK(stream *Stream) error {
|
|
seqNum := bele.BEUint32(stream.msg.buf[stream.msg.b:stream.msg.e])
|
|
log.Infof("-----> Acknowledgement. [%s] ignore. sequence number=%d.", s.UniqueKey, seqNum)
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerSession) doDataMessageAMF0(stream *Stream) error {
|
|
if s.t != ServerSessionTypePub {
|
|
log.Errorf("read audio/video message but server session not pub type. [%s]", s.UniqueKey)
|
|
return rtmpErr
|
|
}
|
|
|
|
val, err := stream.msg.peekStringWithType()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
switch val {
|
|
case "|RtmpSampleAccess": // TODO chef: handle this?
|
|
return nil
|
|
case "@setDataFrame":
|
|
// macos obs
|
|
// skip @setDataFrame
|
|
val, err = stream.msg.readStringWithType()
|
|
val, err := stream.msg.peekStringWithType()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if val != "onMetaData" {
|
|
return rtmpErr
|
|
}
|
|
case "onMetaData":
|
|
// noop
|
|
default:
|
|
// TODO chef:
|
|
log.Error(val)
|
|
log.Error(hex.Dump(stream.msg.buf[stream.msg.b:stream.msg.e]))
|
|
return nil
|
|
}
|
|
|
|
s.avObs.ReadRTMPAVMsgCB(stream.header, stream.timestampAbs, stream.msg.buf[stream.msg.b:stream.msg.e])
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerSession) doCommandMessage(stream *Stream) error {
|
|
cmd, err := stream.msg.readStringWithType()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tid, err := stream.msg.readNumberWithType()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
switch cmd {
|
|
case "connect":
|
|
return s.doConnect(tid, stream)
|
|
case "createStream":
|
|
return s.doCreateStream(tid, stream)
|
|
case "publish":
|
|
return s.doPublish(tid, stream)
|
|
case "play":
|
|
return s.doPlay(tid, stream)
|
|
case "releaseStream":
|
|
fallthrough
|
|
case "FCPublish":
|
|
fallthrough
|
|
case "FCUnpublish":
|
|
fallthrough
|
|
case "getStreamLength":
|
|
log.Warnf("read command message,ignore it. [%s] %s", s.UniqueKey, cmd)
|
|
default:
|
|
log.Errorf("unknown cmd. [%s] cmd=%s", s.UniqueKey, cmd)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerSession) doConnect(tid int, stream *Stream) error {
|
|
val, err := stream.msg.readObjectWithType()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var ok bool
|
|
s.AppName, ok = val["app"].(string)
|
|
if !ok {
|
|
return rtmpErr
|
|
}
|
|
log.Infof("-----> connect('%s'). [%s]", s.AppName, s.UniqueKey)
|
|
|
|
if err := s.packer.writeWinAckSize(s.conn, windowAcknowledgementSize); err != nil {
|
|
return err
|
|
}
|
|
if err := s.packer.writePeerBandwidth(s.conn, peerBandwidth, peerBandwidthLimitTypeDynamic); err != nil {
|
|
return err
|
|
}
|
|
if err := s.packer.writeChunkSize(s.conn, LocalChunkSize); err != nil {
|
|
return err
|
|
}
|
|
if err := s.packer.writeConnectResult(s.conn, tid); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerSession) doCreateStream(tid int, stream *Stream) error {
|
|
log.Infof("-----> createStream() [%s]", s.UniqueKey)
|
|
if err := s.packer.writeCreateStreamResult(s.conn, tid); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerSession) doPublish(tid int, stream *Stream) (err error) {
|
|
if err = stream.msg.readNull(); err != nil {
|
|
return err
|
|
}
|
|
s.StreamNameWithRawQuery, err = stream.msg.readStringWithType()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ss := strings.Split(s.StreamNameWithRawQuery, "?")
|
|
s.StreamName = ss[0]
|
|
|
|
pubType, err := stream.msg.readStringWithType()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Debug(pubType)
|
|
log.Infof("-----> publish('%s') [%s]", s.StreamName, s.UniqueKey)
|
|
// TODO chef: hardcode streamID
|
|
if err := s.packer.writeOnStatusPublish(s.conn, 1); err != nil {
|
|
return err
|
|
}
|
|
s.t = ServerSessionTypePub
|
|
newUniqueKey := strings.Replace(s.UniqueKey, "RTMPPUBSUB", "RTMPPUB", 1)
|
|
log.Infof("session unique key upgrade. %s -> %s", s.UniqueKey, newUniqueKey)
|
|
s.UniqueKey = newUniqueKey
|
|
s.obs.NewRTMPPubSessionCB(s)
|
|
return nil
|
|
}
|
|
|
|
func (s *ServerSession) doPlay(tid int, stream *Stream) (err error) {
|
|
if err = stream.msg.readNull(); err != nil {
|
|
return err
|
|
}
|
|
s.StreamNameWithRawQuery, err = stream.msg.readStringWithType()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ss := strings.Split(s.StreamNameWithRawQuery, "?")
|
|
s.StreamName = ss[0]
|
|
|
|
log.Infof("-----> play('%s') [%s]", s.StreamName, s.UniqueKey)
|
|
// TODO chef: start duration reset
|
|
|
|
if err := s.packer.writeOnStatusPlay(s.conn, 1); err != nil {
|
|
return err
|
|
}
|
|
s.t = ServerSessionTypeSub
|
|
newUniqueKey := strings.Replace(s.UniqueKey, "RTMPPUBSUB", "RTMPSUB", 1)
|
|
log.Infof("session unique key upgrade. %s -> %s", s.UniqueKey, newUniqueKey)
|
|
s.UniqueKey = newUniqueKey
|
|
s.obs.NewRTMPSubSessionCB(s)
|
|
return nil
|
|
}
|