1128 lines
26 KiB
Go
1128 lines
26 KiB
Go
|
||
package rtmp
|
||
|
||
import (
|
||
"strings"
|
||
_"bytes"
|
||
"net"
|
||
"net/url"
|
||
"bufio"
|
||
"time"
|
||
"fmt"
|
||
"encoding/hex"
|
||
"io"
|
||
"github.com/nareix/pio"
|
||
"github.com/nareix/flv/flvio"
|
||
"github.com/nareix/av"
|
||
"github.com/nareix/codec/h264parser"
|
||
"github.com/nareix/codec/aacparser"
|
||
"crypto/hmac"
|
||
"crypto/sha256"
|
||
"crypto/rand"
|
||
)
|
||
|
||
func ParseURL(uri string) (host string, path string) {
|
||
var u *url.URL
|
||
if u, _ = url.Parse(uri); u == nil {
|
||
return
|
||
}
|
||
host = u.Host
|
||
if _, _, err := net.SplitHostPort(u.Host); err != nil {
|
||
host += ":1935"
|
||
}
|
||
path = u.Path
|
||
return
|
||
}
|
||
|
||
func DialTimeout(uri string, timeout time.Duration) (conn *Conn, err error) {
|
||
host, path := ParseURL(uri)
|
||
dailer := net.Dialer{Timeout: timeout}
|
||
var netconn net.Conn
|
||
if netconn, err = dailer.Dial("tcp", host); err != nil {
|
||
return
|
||
}
|
||
|
||
fmt.Println("rtmp: connected")
|
||
|
||
conn = NewConn(netconn)
|
||
conn.Host = host
|
||
conn.Path = path
|
||
return
|
||
}
|
||
|
||
type Server struct {
|
||
Addr string
|
||
HandlePublish func(*Conn)
|
||
HandlePlay func(*Conn)
|
||
}
|
||
|
||
func (self *Server) handleConn(conn *Conn) (err error) {
|
||
if err = conn.handshake(); err != nil {
|
||
return
|
||
}
|
||
|
||
if err = conn.determineType(); err != nil {
|
||
fmt.Println("rtmp: conn closed:", err)
|
||
return
|
||
}
|
||
|
||
if conn.playing {
|
||
if self.HandlePlay != nil {
|
||
self.HandlePlay(conn)
|
||
conn.Close()
|
||
}
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Server) ListenAndServe() (err error) {
|
||
addr := self.Addr
|
||
if addr == "" {
|
||
addr = ":1935"
|
||
}
|
||
var tcpaddr *net.TCPAddr
|
||
if tcpaddr, err = net.ResolveTCPAddr("tcp", addr); err != nil {
|
||
err = fmt.Errorf("rtmp: ListenAndServe: %s", err)
|
||
return
|
||
}
|
||
|
||
var listener *net.TCPListener
|
||
if listener, err = net.ListenTCP("tcp", tcpaddr); err != nil {
|
||
return
|
||
}
|
||
|
||
var netconn net.Conn
|
||
for {
|
||
if netconn, err = listener.Accept(); err != nil {
|
||
return
|
||
}
|
||
|
||
conn := NewConn(netconn)
|
||
conn.isserver = true
|
||
go self.handleConn(conn)
|
||
}
|
||
}
|
||
|
||
type Conn struct {
|
||
Host string
|
||
Path string
|
||
Debug bool
|
||
streams []av.CodecData
|
||
|
||
br *pio.Reader
|
||
bw *pio.Writer
|
||
bufr *bufio.Reader
|
||
bufw *bufio.Writer
|
||
intw *pio.Writer
|
||
netconn net.Conn
|
||
|
||
writeMaxChunkSize int
|
||
readMaxChunkSize int
|
||
|
||
lastcsid uint32
|
||
lastcs *chunkStream
|
||
csmap map[uint32]*chunkStream
|
||
|
||
isserver bool
|
||
publishing, playing bool
|
||
reading, writing bool
|
||
avmsgsid uint32
|
||
|
||
gotcommand bool
|
||
commandname string
|
||
commandtransid float64
|
||
commandobj flvio.AMFMap
|
||
commandparams []interface{}
|
||
|
||
gotmsg bool
|
||
msgdata []byte
|
||
msgtypeid uint8
|
||
|
||
eventtype uint16
|
||
}
|
||
|
||
func NewConn(netconn net.Conn) *Conn {
|
||
conn := &Conn{}
|
||
conn.netconn = netconn
|
||
conn.csmap = make(map[uint32]*chunkStream)
|
||
conn.readMaxChunkSize = 128
|
||
conn.writeMaxChunkSize = 128
|
||
conn.bufr = bufio.NewReaderSize(netconn, 4096)
|
||
conn.bufw = bufio.NewWriterSize(netconn, 4096)
|
||
conn.br = pio.NewReader(conn.bufr)
|
||
conn.bw = pio.NewWriter(conn.bufw)
|
||
conn.intw = pio.NewWriter(nil)
|
||
return conn
|
||
}
|
||
|
||
type chunkStream struct {
|
||
timenow uint32
|
||
timedelta uint32
|
||
hastimeext bool
|
||
msgsid uint32
|
||
msgtypeid uint8
|
||
msgdatalen uint32
|
||
msgdataleft uint32
|
||
msghdrtype uint8
|
||
msgdata []byte
|
||
}
|
||
|
||
func (self *chunkStream) Start() {
|
||
self.msgdataleft = self.msgdatalen
|
||
self.msgdata = make([]byte, self.msgdatalen)
|
||
}
|
||
|
||
const (
|
||
msgtypeidUserControl = 4
|
||
msgtypeidWindowAckSize = 5
|
||
msgtypeidSetPeerBandwidth = 6
|
||
msgtypeidSetChunkSize = 1
|
||
msgtypeidCommandMsgAMF0 = 20
|
||
msgtypeidCommandMsgAMF3 = 17
|
||
msgtypeidDataMsgAMF0 = 18
|
||
msgtypeidDataMsgAMF3 = 15
|
||
msgtypeidVideoMsg = 9
|
||
msgtypeidAudioMsg = 8
|
||
)
|
||
|
||
const (
|
||
eventtypeStreamBegin = 0
|
||
eventtypeSetBufferLength = 3
|
||
)
|
||
|
||
func (self *Conn) Close() (err error) {
|
||
return self.netconn.Close()
|
||
}
|
||
|
||
func (self *Conn) pollCommand() (err error) {
|
||
for {
|
||
if err = self.pollMsg(); err != nil {
|
||
return
|
||
}
|
||
if self.gotcommand {
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
func (self *Conn) pollMsg() (err error) {
|
||
self.gotmsg = false
|
||
self.gotcommand = false
|
||
for {
|
||
if err = self.readChunk(); err != nil {
|
||
return
|
||
}
|
||
if self.gotmsg {
|
||
fmt.Println("rtmp: gotmsg iscommand", self.gotcommand)
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
func (self *Conn) determineType() (err error) {
|
||
var connectpath, playpath string
|
||
|
||
// < connect("app")
|
||
if err = self.pollCommand(); err != nil {
|
||
return
|
||
}
|
||
if self.commandname != "connect" {
|
||
err = fmt.Errorf("rtmp: first command is not connect")
|
||
return
|
||
}
|
||
if self.commandobj != nil {
|
||
app := self.commandobj["app"]
|
||
connectpath, _ = app.(string)
|
||
}
|
||
|
||
// > WindowAckSize
|
||
if err = self.writeWindowAckSize(5000000); err != nil {
|
||
return
|
||
}
|
||
// > SetPeerBandwidth
|
||
if err = self.writeSetPeerBandwidth(5000000, 2); err != nil {
|
||
return
|
||
}
|
||
self.writeMaxChunkSize = 4096
|
||
// > SetChunkSize
|
||
if err = self.writeSetChunkSize(uint32(self.writeMaxChunkSize)); err != nil {
|
||
return
|
||
}
|
||
|
||
// > _result("NetConnection.Connect.Success")
|
||
w := self.writeCommandMsgStart()
|
||
flvio.WriteAMF0Val(w, "_result")
|
||
flvio.WriteAMF0Val(w, self.commandtransid)
|
||
flvio.WriteAMF0Val(w, flvio.AMFMap{
|
||
"fmtVer": "FMS/3,0,1,123",
|
||
"capabilities": 31,
|
||
})
|
||
flvio.WriteAMF0Val(w, flvio.AMFMap{
|
||
"level": "status",
|
||
"code": "NetConnection.Connect.Success",
|
||
"description": "Connection Success.",
|
||
"objectEncoding": 3,
|
||
})
|
||
self.writeCommandMsgEnd(3, 0)
|
||
|
||
for {
|
||
if err = self.pollMsg(); err != nil {
|
||
return
|
||
}
|
||
if self.gotcommand {
|
||
switch self.commandname {
|
||
|
||
// < createStream
|
||
case "createStream":
|
||
self.avmsgsid = uint32(1)
|
||
// > _result(streamid)
|
||
w := self.writeCommandMsgStart()
|
||
flvio.WriteAMF0Val(w, "_result")
|
||
flvio.WriteAMF0Val(w, self.commandtransid)
|
||
flvio.WriteAMF0Val(w, nil)
|
||
flvio.WriteAMF0Val(w, self.avmsgsid) // streamid=1
|
||
self.writeCommandMsgEnd(3, 0)
|
||
|
||
// < play("path")
|
||
case "play":
|
||
if len(self.commandparams) < 1 {
|
||
err = fmt.Errorf("rtmp: play params invalid")
|
||
return
|
||
}
|
||
playpath, _ = self.commandparams[0].(string)
|
||
|
||
// > streamBegin(streamid)
|
||
self.writeStreamBegin(self.avmsgsid)
|
||
|
||
// > onStatus()
|
||
w := self.writeCommandMsgStart()
|
||
flvio.WriteAMF0Val(w, "onStatus")
|
||
flvio.WriteAMF0Val(w, self.commandtransid)
|
||
flvio.WriteAMF0Val(w, nil)
|
||
flvio.WriteAMF0Val(w, flvio.AMFMap{
|
||
"level": "status",
|
||
"code": "NetStream.Play.Start",
|
||
"description": "Start live",
|
||
})
|
||
self.writeCommandMsgEnd(5, self.avmsgsid)
|
||
|
||
// > |RtmpSampleAccess()
|
||
w = self.writeDataMsgStart()
|
||
flvio.WriteAMF0Val(w, "|RtmpSampleAccess")
|
||
flvio.WriteAMF0Val(w, true)
|
||
flvio.WriteAMF0Val(w, true)
|
||
self.writeDataMsgEnd(5, self.avmsgsid)
|
||
|
||
fmt.Println("rtmp: playing")
|
||
|
||
self.Path = fmt.Sprintf("/%s/%s", connectpath, playpath)
|
||
self.playing = true
|
||
self.writing = true
|
||
return
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Conn) checkConnectResult() (ok bool, errmsg string) {
|
||
errmsg = "params invalid"
|
||
if len(self.commandparams) < 1 {
|
||
return
|
||
}
|
||
|
||
obj, _ := self.commandparams[0].(flvio.AMFMap)
|
||
if obj == nil {
|
||
return
|
||
}
|
||
|
||
_code, _ := obj["code"]
|
||
if _code == nil {
|
||
return
|
||
}
|
||
code, _ := _code.(string)
|
||
if code != "NetConnection.Connect.Success" {
|
||
return
|
||
}
|
||
|
||
ok = true
|
||
return
|
||
}
|
||
|
||
func (self *Conn) checkCreateStreamResult() (ok bool, avmsgsid uint32) {
|
||
if len(self.commandparams) < 1 {
|
||
return
|
||
}
|
||
|
||
ok = true
|
||
_avmsgsid, _ := self.commandparams[0].(float64)
|
||
avmsgsid = uint32(_avmsgsid)
|
||
return
|
||
}
|
||
|
||
func (self *Conn) connectPlay() (err error) {
|
||
var connectpath, playpath string
|
||
pathsegs := strings.Split(self.Path, "/")
|
||
if len(pathsegs) > 1 {
|
||
connectpath = pathsegs[1]
|
||
}
|
||
if len(pathsegs) > 2 {
|
||
playpath = pathsegs[2]
|
||
}
|
||
|
||
// > connect("app")
|
||
fmt.Printf("rtmp: > connect('%s') host=%s\n", connectpath, self.Host)
|
||
w := self.writeCommandMsgStart()
|
||
flvio.WriteAMF0Val(w, "connect")
|
||
flvio.WriteAMF0Val(w, 1)
|
||
flvio.WriteAMF0Val(w, flvio.AMFMap{
|
||
"app": connectpath,
|
||
"flashVer": "MAC 22,0,0,192",
|
||
"tcUrl": fmt.Sprintf("rtmp://%s/%s", self.Host, connectpath),
|
||
"fpad": false,
|
||
"capabilities": 15,
|
||
"audioCodecs": 4071,
|
||
"videoCodecs": 252,
|
||
"videoFunction": 1,
|
||
})
|
||
self.writeCommandMsgEnd(3, 0)
|
||
|
||
for {
|
||
if err = self.pollMsg(); err != nil {
|
||
return
|
||
}
|
||
if self.gotcommand {
|
||
// < _result("NetConnection.Connect.Success")
|
||
if self.commandname == "_result" {
|
||
var ok bool
|
||
var errmsg string
|
||
if ok, errmsg = self.checkConnectResult(); !ok {
|
||
err = fmt.Errorf("rtmp: command connect failed: %s", errmsg)
|
||
return
|
||
}
|
||
fmt.Printf("rtmp: < _result() of connect\n")
|
||
break
|
||
}
|
||
} else {
|
||
if self.msgtypeid == msgtypeidWindowAckSize {
|
||
self.writeWindowAckSize(2500000)
|
||
}
|
||
}
|
||
}
|
||
|
||
// > createStream()
|
||
fmt.Printf("rtmp: > createStream()\n")
|
||
w = self.writeCommandMsgStart()
|
||
flvio.WriteAMF0Val(w, "createStream")
|
||
flvio.WriteAMF0Val(w, 2)
|
||
flvio.WriteAMF0Val(w, nil)
|
||
self.writeCommandMsgEnd(3, 0)
|
||
|
||
// > SetBufferLength 0,3000ms
|
||
self.writeSetBufferLength(0, 3000)
|
||
|
||
for {
|
||
if err = self.pollMsg(); err != nil {
|
||
return
|
||
}
|
||
if self.gotcommand {
|
||
// < _result(avmsgsid) of createStream
|
||
if self.commandname == "_result" {
|
||
var ok bool
|
||
if ok, self.avmsgsid = self.checkCreateStreamResult(); !ok {
|
||
err = fmt.Errorf("rtmp: createStream command failed")
|
||
return
|
||
}
|
||
break
|
||
}
|
||
}
|
||
}
|
||
|
||
fmt.Println("rtmp: play", playpath)
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Conn) ReadHeader() (err error) {
|
||
if !self.reading && !self.writing {
|
||
if err = self.handshake(); err != nil {
|
||
return
|
||
}
|
||
if err = self.connectPlay(); err != nil {
|
||
return
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
func (self *Conn) Streams() (streams []av.CodecData, err error) {
|
||
streams = self.streams
|
||
return
|
||
}
|
||
|
||
func (self *Conn) WritePacket(pkt av.Packet) (err error) {
|
||
ts := uint32(pkt.Time/time.Millisecond)
|
||
stream := self.streams[pkt.Idx]
|
||
fmt.Println("rtmp: WritePacket", pkt.Idx, pkt.Time, pkt.CompositionTime, ts)
|
||
|
||
switch stream.Type() {
|
||
case av.AAC:
|
||
audiodata := self.makeAACAudiodata(stream.(av.AudioCodecData), flvio.AAC_RAW, pkt.Data)
|
||
w := self.writeAudioDataStart()
|
||
audiodata.Marshal(w)
|
||
self.writeAudioDataEnd(ts)
|
||
|
||
case av.H264:
|
||
videodata := self.makeH264Videodata(flvio.AVC_NALU, pkt.IsKeyFrame, pkt.Data)
|
||
videodata.CompositionTime = int32(pkt.CompositionTime/time.Millisecond)
|
||
w := self.writeVideoDataStart()
|
||
videodata.Marshal(w)
|
||
self.writeVideoDataEnd(ts)
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Conn) WriteHeader(streams []av.CodecData) (err error) {
|
||
metadata := flvio.AMFMap{}
|
||
|
||
metadata["server"] = "joy4 streaming server (github.com/nareix/joy4)"
|
||
metadata["duration"] = 0
|
||
|
||
for _, _stream := range streams {
|
||
typ := _stream.Type()
|
||
switch {
|
||
case typ.IsVideo():
|
||
stream := _stream.(av.VideoCodecData)
|
||
switch typ {
|
||
case av.H264:
|
||
metadata["videocodecid"] = flvio.VIDEO_H264
|
||
|
||
default:
|
||
err = fmt.Errorf("rtmp: WriteHeader: unsupported video codecType=%v", stream.Type())
|
||
return
|
||
}
|
||
|
||
metadata["width"] = stream.Width()
|
||
metadata["height"] = stream.Height()
|
||
metadata["framerate"] = 24 // TODO: make it correct
|
||
metadata["videodatarate"] = 0
|
||
|
||
case typ.IsAudio():
|
||
stream := _stream.(av.AudioCodecData)
|
||
switch typ {
|
||
case av.AAC:
|
||
metadata["audiocodecid"] = flvio.SOUND_AAC
|
||
|
||
default:
|
||
err = fmt.Errorf("rtmp: WriteHeader: unsupported audio codecType=%v", stream.Type())
|
||
return
|
||
}
|
||
|
||
metadata["audiodatarate"] = 0
|
||
}
|
||
}
|
||
|
||
// > onMetaData()
|
||
w := self.writeDataMsgStart()
|
||
flvio.WriteAMF0Val(w, "onMetaData")
|
||
flvio.WriteAMF0Val(w, metadata)
|
||
if err = self.writeDataMsgEnd(5, self.avmsgsid); err != nil {
|
||
return
|
||
}
|
||
|
||
// > Videodata(decoder config)
|
||
// > Audiodata(decoder config)
|
||
for _, stream := range streams {
|
||
switch stream.Type() {
|
||
case av.H264:
|
||
h264 := stream.(h264parser.CodecData)
|
||
videodata := self.makeH264Videodata(flvio.AVC_SEQHDR, true, h264.AVCDecoderConfRecordBytes())
|
||
w := self.writeVideoDataStart()
|
||
videodata.Marshal(w)
|
||
if err = self.writeVideoDataEnd(0); err != nil {
|
||
return
|
||
}
|
||
|
||
case av.AAC:
|
||
aac := stream.(aacparser.CodecData)
|
||
audiodata := self.makeAACAudiodata(aac, flvio.AAC_SEQHDR, aac.MPEG4AudioConfigBytes())
|
||
w := self.writeAudioDataStart()
|
||
audiodata.Marshal(w)
|
||
if err = self.writeAudioDataEnd(0); err != nil {
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
self.streams = streams
|
||
return
|
||
}
|
||
|
||
func (self *Conn) makeH264Videodata(pkttype uint8, iskeyframe bool, data []byte) flvio.Videodata {
|
||
videodata := flvio.Videodata{
|
||
CodecID: flvio.VIDEO_H264,
|
||
AVCPacketType: pkttype,
|
||
Data: data,
|
||
}
|
||
if iskeyframe {
|
||
videodata.FrameType = flvio.FRAME_KEY
|
||
} else {
|
||
videodata.FrameType = flvio.FRAME_INTER
|
||
}
|
||
return videodata
|
||
}
|
||
|
||
func (self *Conn) makeAACAudiodata(stream av.AudioCodecData, pkttype uint8, data []byte) flvio.Audiodata {
|
||
audiodata := flvio.Audiodata{
|
||
SoundFormat: flvio.SOUND_AAC,
|
||
SoundRate: flvio.SOUND_44Khz,
|
||
AACPacketType: pkttype,
|
||
Data: data,
|
||
}
|
||
switch stream.SampleFormat().BytesPerSample() {
|
||
case 1:
|
||
audiodata.SoundSize = flvio.SOUND_8BIT
|
||
default:
|
||
audiodata.SoundSize = flvio.SOUND_16BIT
|
||
}
|
||
switch stream.ChannelLayout().Count() {
|
||
case 1:
|
||
audiodata.SoundType = flvio.SOUND_MONO
|
||
case 2:
|
||
audiodata.SoundType = flvio.SOUND_STEREO
|
||
}
|
||
return audiodata
|
||
}
|
||
|
||
func (self *Conn) writeSetChunkSize(size uint32) (err error) {
|
||
w := self.writeProtoCtrlMsgStart()
|
||
w.WriteU32BE(size)
|
||
return self.writeProtoCtrlMsgEnd(msgtypeidSetChunkSize)
|
||
}
|
||
|
||
func (self *Conn) writeWindowAckSize(size uint32) (err error) {
|
||
w := self.writeProtoCtrlMsgStart()
|
||
w.WriteU32BE(size)
|
||
return self.writeProtoCtrlMsgEnd(msgtypeidWindowAckSize)
|
||
}
|
||
|
||
func (self *Conn) writeSetPeerBandwidth(acksize uint32, limittype uint8) (err error) {
|
||
w := self.writeProtoCtrlMsgStart()
|
||
w.WriteU32BE(acksize)
|
||
w.WriteU8(limittype)
|
||
return self.writeProtoCtrlMsgEnd(msgtypeidSetPeerBandwidth)
|
||
}
|
||
|
||
func (self *Conn) writeProtoCtrlMsgStart() *pio.Writer {
|
||
self.intw.SaveToVecOn()
|
||
return self.intw
|
||
}
|
||
|
||
func (self *Conn) writeProtoCtrlMsgEnd(msgtypeid uint8) (err error) {
|
||
msgdatav := self.intw.SaveToVecOff()
|
||
return self.writeChunks(2, 0, msgtypeid, 0, msgdatav)
|
||
}
|
||
|
||
func (self *Conn) writeCommandMsgStart() *pio.Writer {
|
||
self.intw.SaveToVecOn()
|
||
return self.intw
|
||
}
|
||
|
||
func (self *Conn) writeCommandMsgEnd(csid uint32, msgsid uint32) (err error) {
|
||
msgdatav := self.intw.SaveToVecOff()
|
||
return self.writeChunks(csid, 0, msgtypeidCommandMsgAMF0, msgsid, msgdatav)
|
||
}
|
||
|
||
func (self *Conn) writeDataMsgStart() *pio.Writer {
|
||
self.intw.SaveToVecOn()
|
||
return self.intw
|
||
}
|
||
|
||
func (self *Conn) writeDataMsgEnd(csid uint32, msgsid uint32) (err error) {
|
||
msgdatav := self.intw.SaveToVecOff()
|
||
return self.writeChunks(csid, 0, msgtypeidDataMsgAMF0, msgsid, msgdatav)
|
||
}
|
||
|
||
func (self *Conn) writeVideoDataStart() *pio.Writer {
|
||
self.intw.SaveToVecOn()
|
||
return self.intw
|
||
}
|
||
|
||
func (self *Conn) writeVideoDataEnd(timestamp uint32) (err error) {
|
||
msgdatav := self.intw.SaveToVecOff()
|
||
return self.writeChunks(7, timestamp, msgtypeidVideoMsg, self.avmsgsid, msgdatav)
|
||
}
|
||
|
||
func (self *Conn) writeAudioDataStart() *pio.Writer {
|
||
self.intw.SaveToVecOn()
|
||
return self.intw
|
||
}
|
||
|
||
func (self *Conn) writeAudioDataEnd(timestamp uint32) (err error) {
|
||
msgdatav := self.intw.SaveToVecOff()
|
||
return self.writeChunks(6, timestamp, msgtypeidAudioMsg, self.avmsgsid, msgdatav)
|
||
}
|
||
|
||
func (self *Conn) writeUserControlMsgStart(eventtype uint16) *pio.Writer {
|
||
self.intw.SaveToVecOn()
|
||
self.intw.WriteU16BE(eventtype)
|
||
return self.intw
|
||
}
|
||
|
||
func (self *Conn) writeUserControlMsgEnd() (err error) {
|
||
msgdatav := self.intw.SaveToVecOff()
|
||
return self.writeChunks(2, 0, msgtypeidUserControl, 0, msgdatav)
|
||
}
|
||
|
||
func (self *Conn) writeStreamBegin(msgsid uint32) (err error) {
|
||
w := self.writeUserControlMsgStart(eventtypeStreamBegin)
|
||
w.WriteU32BE(msgsid)
|
||
return self.writeUserControlMsgEnd()
|
||
}
|
||
|
||
func (self *Conn) writeSetBufferLength(msgsid uint32, timestamp uint32) (err error) {
|
||
w := self.writeUserControlMsgStart(eventtypeStreamBegin)
|
||
w.WriteU32BE(msgsid)
|
||
w.WriteU32BE(timestamp)
|
||
return self.writeUserControlMsgEnd()
|
||
}
|
||
|
||
func (self *Conn) writeChunks(csid uint32, timestamp uint32, msgtypeid uint8, msgsid uint32, msgdatav [][]byte) (err error) {
|
||
msgdatalen := pio.VecLen(msgdatav)
|
||
|
||
// [Type 0][Type 3][Type 3]....
|
||
|
||
// 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
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
// | timestamp |message length |
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
// | message length (cont) |message type id| msg stream id |
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
// | message stream id (cont) |
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
//
|
||
// Figure 9 Chunk Message Header – Type 0
|
||
if err = self.bw.WriteU8(byte(csid)&0x3f); err != nil {
|
||
return
|
||
}
|
||
if err = self.bw.WriteU24BE(timestamp); err != nil {
|
||
return
|
||
}
|
||
if err = self.bw.WriteU24BE(uint32(msgdatalen)); err != nil {
|
||
return
|
||
}
|
||
if err = self.bw.WriteU8(msgtypeid); err != nil {
|
||
return
|
||
}
|
||
if err = self.bw.WriteU32LE(msgsid); err != nil {
|
||
return
|
||
}
|
||
|
||
msgdataoff := 0
|
||
for {
|
||
size := msgdatalen - msgdataoff
|
||
if size > self.writeMaxChunkSize {
|
||
size = self.writeMaxChunkSize
|
||
}
|
||
|
||
write := pio.VecSlice(msgdatav, msgdataoff, msgdataoff+size)
|
||
for _, b := range write {
|
||
if _, err = self.bw.Write(b); err != nil {
|
||
return
|
||
}
|
||
}
|
||
|
||
msgdataoff += size
|
||
if msgdataoff == msgdatalen {
|
||
break
|
||
}
|
||
|
||
// Type 3
|
||
if err = self.bw.WriteU8(byte(csid)&0x3f|3<<6); err != nil {
|
||
return
|
||
}
|
||
}
|
||
|
||
fmt.Printf("rtmp: write chunk msgdatalen=%d msgsid=%d\n", msgdatalen, msgsid)
|
||
|
||
if err = self.bufw.Flush(); err != nil {
|
||
return
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Conn) readChunk() (err error) {
|
||
var msghdrtype uint8
|
||
var csid uint32
|
||
var header uint8
|
||
if header, err = self.br.ReadU8(); err != nil {
|
||
return
|
||
}
|
||
msghdrtype = header>>6
|
||
|
||
csid = uint32(header)&0x3f
|
||
switch csid {
|
||
default: // Chunk basic header 1
|
||
case 0: // Chunk basic header 2
|
||
var i uint8
|
||
if i, err = self.br.ReadU8(); err != nil {
|
||
return
|
||
}
|
||
csid = uint32(i)+64
|
||
case 1: // Chunk basic header 3
|
||
var i uint16
|
||
if i, err = self.br.ReadU16BE(); err != nil {
|
||
return
|
||
}
|
||
csid = uint32(i)+64
|
||
}
|
||
|
||
var cs *chunkStream
|
||
if self.lastcs != nil && self.lastcsid == csid {
|
||
cs = self.lastcs
|
||
} else {
|
||
cs = &chunkStream{}
|
||
self.csmap[csid] = cs
|
||
}
|
||
self.lastcs = cs
|
||
self.lastcsid = csid
|
||
|
||
var timestamp uint32
|
||
|
||
switch msghdrtype {
|
||
case 0:
|
||
// 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
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
// | timestamp |message length |
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
// | message length (cont) |message type id| msg stream id |
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
// | message stream id (cont) |
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
//
|
||
// Figure 9 Chunk Message Header – Type 0
|
||
if cs.msgdataleft != 0 {
|
||
err = fmt.Errorf("rtmp: chunk msgdataleft=%d invalid", cs.msgdataleft)
|
||
return
|
||
}
|
||
var h[]byte
|
||
if h, err = self.br.ReadBytes(11); err != nil {
|
||
return
|
||
}
|
||
timestamp = pio.GetU24BE(h[0:3])
|
||
cs.msghdrtype = msghdrtype
|
||
cs.msgdatalen = pio.GetU24BE(h[3:6])
|
||
cs.msgtypeid = h[6]
|
||
cs.msgsid = pio.GetU32LE(h[7:11])
|
||
if timestamp == 0xffffff {
|
||
if timestamp, err = self.br.ReadU32BE(); err != nil {
|
||
return
|
||
}
|
||
cs.hastimeext = true
|
||
} else {
|
||
cs.hastimeext = false
|
||
}
|
||
cs.timenow = timestamp
|
||
cs.Start()
|
||
|
||
case 1:
|
||
// 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
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
// | timestamp delta |message length |
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
// | message length (cont) |message type id|
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
//
|
||
// Figure 10 Chunk Message Header – Type 1
|
||
if cs.msgdataleft != 0 {
|
||
err = fmt.Errorf("rtmp: chunk msgdataleft=%d invalid", cs.msgdataleft)
|
||
return
|
||
}
|
||
var h[]byte
|
||
if h, err = self.br.ReadBytes(7); err != nil {
|
||
return
|
||
}
|
||
timestamp = pio.GetU24BE(h[0:3])
|
||
cs.msghdrtype = msghdrtype
|
||
cs.msgdatalen = pio.GetU24BE(h[3:6])
|
||
cs.msgtypeid = h[6]
|
||
if timestamp == 0xffffff {
|
||
if timestamp, err = self.br.ReadU32BE(); err != nil {
|
||
return
|
||
}
|
||
cs.hastimeext = true
|
||
} else {
|
||
cs.hastimeext = false
|
||
}
|
||
cs.timedelta = timestamp
|
||
cs.timenow += timestamp
|
||
cs.Start()
|
||
|
||
case 2:
|
||
// 0 1 2
|
||
// 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
// | timestamp delta |
|
||
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
||
//
|
||
// Figure 11 Chunk Message Header – Type 2
|
||
if cs.msgdataleft != 0 {
|
||
err = fmt.Errorf("rtmp: chunk msgdataleft=%d invalid", cs.msgdataleft)
|
||
return
|
||
}
|
||
var h[]byte
|
||
if h, err = self.br.ReadBytes(3); err != nil {
|
||
return
|
||
}
|
||
cs.msghdrtype = msghdrtype
|
||
timestamp = pio.GetU24BE(h[0:3])
|
||
if timestamp == 0xffffff {
|
||
if timestamp, err = self.br.ReadU32BE(); err != nil {
|
||
return
|
||
}
|
||
cs.hastimeext = true
|
||
} else {
|
||
cs.hastimeext = false
|
||
}
|
||
cs.timedelta = timestamp
|
||
cs.timenow += timestamp
|
||
cs.Start()
|
||
|
||
case 3:
|
||
if cs.msgdataleft == 0 {
|
||
switch cs.msghdrtype {
|
||
case 0:
|
||
if cs.hastimeext {
|
||
if timestamp, err = self.br.ReadU32BE(); err != nil {
|
||
return
|
||
}
|
||
cs.timenow = timestamp
|
||
}
|
||
case 1, 2:
|
||
if cs.hastimeext {
|
||
if timestamp, err = self.br.ReadU32BE(); err != nil {
|
||
return
|
||
}
|
||
} else {
|
||
timestamp = cs.timedelta
|
||
}
|
||
cs.timenow += timestamp
|
||
}
|
||
cs.Start()
|
||
}
|
||
|
||
default:
|
||
err = fmt.Errorf("rtmp: invalid chunk msg header type=%d", msghdrtype)
|
||
return
|
||
}
|
||
|
||
size := int(cs.msgdataleft)
|
||
if size > self.readMaxChunkSize {
|
||
size = self.readMaxChunkSize
|
||
}
|
||
off := cs.msgdatalen-cs.msgdataleft
|
||
buf := cs.msgdata[off:int(off)+size]
|
||
if _, err = io.ReadFull(self.br, buf); err != nil {
|
||
return
|
||
}
|
||
cs.msgdataleft -= uint32(size)
|
||
|
||
if true {
|
||
fmt.Printf("rtmp: chunk csid=%d msgsid=%d msgtypeid=%d msghdrtype=%d len=%d left=%d\n",
|
||
csid, cs.msgsid, cs.msgtypeid, cs.msghdrtype, cs.msgdatalen, cs.msgdataleft)
|
||
}
|
||
|
||
if cs.msgdataleft == 0 {
|
||
if true {
|
||
fmt.Println("rtmp: chunk data")
|
||
fmt.Print(hex.Dump(cs.msgdata))
|
||
fmt.Printf("%x\n", cs.msgdata)
|
||
}
|
||
|
||
if err = self.handleMsg(cs.msgsid, cs.msgtypeid, cs.msgdata); err != nil {
|
||
return
|
||
}
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Conn) handleCommandMsgAMF0(r *pio.Reader) (err error) {
|
||
commandname, _ := flvio.ReadAMF0Val(r)
|
||
commandtransid, _ := flvio.ReadAMF0Val(r)
|
||
commandobj, _ := flvio.ReadAMF0Val(r)
|
||
|
||
var ok bool
|
||
if self.commandname, ok = commandname.(string); !ok {
|
||
err = fmt.Errorf("rtmp: CommandMsgAMF0 command is not string")
|
||
return
|
||
}
|
||
|
||
self.commandobj, _ = commandobj.(flvio.AMFMap)
|
||
self.commandtransid, _ = commandtransid.(float64)
|
||
self.commandparams = []interface{}{}
|
||
for {
|
||
if val, rerr := flvio.ReadAMF0Val(r); rerr != nil {
|
||
break
|
||
} else {
|
||
self.commandparams = append(self.commandparams, val)
|
||
}
|
||
}
|
||
|
||
self.gotcommand = true
|
||
return
|
||
}
|
||
|
||
func (self *Conn) handleMsg(msgsid uint32, msgtypeid uint8, msgdata []byte) (err error) {
|
||
self.msgdata = msgdata
|
||
self.msgtypeid = msgtypeid
|
||
|
||
switch msgtypeid {
|
||
case msgtypeidCommandMsgAMF0:
|
||
r := pio.NewReaderBytes(msgdata)
|
||
if err = self.handleCommandMsgAMF0(r); err != nil {
|
||
return
|
||
}
|
||
|
||
case msgtypeidCommandMsgAMF3:
|
||
r := pio.NewReaderBytes(msgdata)
|
||
r.ReadU8() // skip first byte
|
||
if err = self.handleCommandMsgAMF0(r); err != nil {
|
||
return
|
||
}
|
||
|
||
case msgtypeidUserControl:
|
||
if len(msgdata) >= 2 {
|
||
self.eventtype = pio.GetU16BE(msgdata)
|
||
} else {
|
||
err = fmt.Errorf("rtmp: short packet of UserControl")
|
||
return
|
||
}
|
||
}
|
||
|
||
self.gotmsg = true
|
||
return
|
||
}
|
||
|
||
var (
|
||
hsClientFullKey = []byte{
|
||
'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
|
||
'F', 'l', 'a', 's', 'h', ' ', 'P', 'l', 'a', 'y', 'e', 'r', ' ',
|
||
'0', '0', '1',
|
||
0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1,
|
||
0x02, 0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB,
|
||
0x93, 0xB8, 0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE,
|
||
}
|
||
hsServerFullKey = []byte{
|
||
'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
|
||
'F', 'l', 'a', 's', 'h', ' ', 'M', 'e', 'd', 'i', 'a', ' ',
|
||
'S', 'e', 'r', 'v', 'e', 'r', ' ',
|
||
'0', '0', '1',
|
||
0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1,
|
||
0x02, 0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB,
|
||
0x93, 0xB8, 0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE,
|
||
}
|
||
hsClientPartialKey = hsClientFullKey[:30]
|
||
hsServerPartialKey = hsServerFullKey[:36]
|
||
)
|
||
|
||
func hsMakeDigest(key []byte, src []byte, gap int) (dst []byte) {
|
||
h := hmac.New(sha256.New, key)
|
||
if gap <= 0 {
|
||
h.Write(src)
|
||
} else {
|
||
h.Write(src[:gap])
|
||
h.Write(src[gap+32:])
|
||
}
|
||
return h.Sum(nil)
|
||
}
|
||
|
||
func hsCalcDigestPos(p []byte, off int, mod int, add int) (pos int) {
|
||
for i := 0; i < 4; i++ {
|
||
pos += int(p[i+off])
|
||
}
|
||
pos = pos%mod+add
|
||
return
|
||
}
|
||
|
||
func hsCreateC1(p []byte) {
|
||
p[4] = 9
|
||
p[5] = 0
|
||
p[6] = 124
|
||
p[7] = 2
|
||
rand.Read(p[8:])
|
||
gap := hsCalcDigestPos(p, 8, 728, 12)
|
||
digest := hsMakeDigest(hsClientPartialKey, p, gap)
|
||
copy(p[gap:], digest)
|
||
}
|
||
|
||
func (self *Conn) handshake() (err error) {
|
||
if self.isserver {
|
||
return self.handshakeServer()
|
||
} else {
|
||
return self.handshakeClient()
|
||
}
|
||
}
|
||
|
||
func (self *Conn) handshakeClient() (err error) {
|
||
var random [(1+1536*2)*2]byte
|
||
|
||
C0C1C2 := random[:1536*2+1]
|
||
C0 := C0C1C2[:1]
|
||
C1 := C0C1C2[1:1536+1]
|
||
C0C1 := C0C1C2[:1536+1]
|
||
C2 := C0C1C2[1536+1:]
|
||
|
||
S0S1S2 := random[1536*2+1:]
|
||
//S0 := S0S1S2[:1]
|
||
S1 := S0S1S2[1:1536+1]
|
||
//S0S1 := S0S1S2[:1536+1]
|
||
//S2 := S0S1S2[1536+1:]
|
||
|
||
C0[0] = 3
|
||
hsCreateC1(C1)
|
||
|
||
// > C0C1
|
||
if _, err = self.bw.Write(C0C1); err != nil {
|
||
return
|
||
}
|
||
if err = self.bufw.Flush(); err != nil {
|
||
return
|
||
}
|
||
|
||
// < S0S1S2
|
||
if _, err = io.ReadFull(self.br, S0S1S2); err != nil {
|
||
return
|
||
}
|
||
|
||
fmt.Println("rtmp: handshakeClient: server version", S1[4],S1[5],S1[6],S1[7])
|
||
|
||
if S1[4] >= 3 {
|
||
} else {
|
||
C2 = S1
|
||
}
|
||
|
||
// > C2
|
||
if _, err = self.bw.Write(C2); err != nil {
|
||
return
|
||
}
|
||
if err = self.bufw.Flush(); err != nil {
|
||
return
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (self *Conn) handshakeServer() (err error) {
|
||
return
|
||
}
|
||
|