Alexey Khit
129 lines
2.6 KiB
Go
129 lines
2.6 KiB
Go
package mp4
|
|
|
|
import (
|
|
"github.com/AlexxIT/go2rtc/cmd/api"
|
|
"github.com/AlexxIT/go2rtc/cmd/streams"
|
|
"github.com/AlexxIT/go2rtc/pkg/mp4"
|
|
"github.com/AlexxIT/go2rtc/pkg/streamer"
|
|
"strings"
|
|
)
|
|
|
|
const packetSize = 8192
|
|
|
|
func handlerWS(ctx *api.Transport, msg *streamer.Message) {
|
|
src := ctx.Request.URL.Query().Get("src")
|
|
stream := streams.GetOrNew(src)
|
|
if stream == nil {
|
|
return
|
|
}
|
|
|
|
cons := &mp4.Consumer{}
|
|
cons.UserAgent = ctx.Request.UserAgent()
|
|
cons.RemoteAddr = ctx.Request.RemoteAddr
|
|
|
|
if codecs, ok := msg.Value.(string); ok {
|
|
cons.Medias = parseMedias(codecs, true)
|
|
}
|
|
|
|
cons.Listen(func(msg interface{}) {
|
|
if data, ok := msg.([]byte); ok {
|
|
for len(data) > packetSize {
|
|
ctx.Write(data[:packetSize])
|
|
data = data[packetSize:]
|
|
}
|
|
ctx.Write(data)
|
|
}
|
|
})
|
|
|
|
if err := stream.AddConsumer(cons); err != nil {
|
|
log.Warn().Err(err).Caller().Send()
|
|
ctx.Error(err)
|
|
return
|
|
}
|
|
|
|
ctx.OnClose(func() {
|
|
stream.RemoveConsumer(cons)
|
|
})
|
|
|
|
ctx.Write(&streamer.Message{Type: "mse", Value: cons.MimeType()})
|
|
|
|
data, err := cons.Init()
|
|
if err != nil {
|
|
log.Warn().Err(err).Caller().Send()
|
|
ctx.Error(err)
|
|
return
|
|
}
|
|
|
|
ctx.Write(data)
|
|
|
|
cons.Start()
|
|
}
|
|
|
|
func handlerWS4(ctx *api.Transport, msg *streamer.Message) {
|
|
src := ctx.Request.URL.Query().Get("src")
|
|
stream := streams.GetOrNew(src)
|
|
if stream == nil {
|
|
return
|
|
}
|
|
|
|
cons := &mp4.Segment{}
|
|
|
|
if codecs, ok := msg.Value.(string); ok {
|
|
cons.Medias = parseMedias(codecs, false)
|
|
}
|
|
|
|
cons.Listen(func(msg interface{}) {
|
|
if data, ok := msg.([]byte); ok {
|
|
ctx.Write(data)
|
|
}
|
|
})
|
|
|
|
if err := stream.AddConsumer(cons); err != nil {
|
|
log.Error().Err(err).Caller().Send()
|
|
return
|
|
}
|
|
|
|
ctx.OnClose(func() {
|
|
stream.RemoveConsumer(cons)
|
|
})
|
|
}
|
|
|
|
func parseMedias(codecs string, parseAudio bool) (medias []*streamer.Media) {
|
|
var videos []*streamer.Codec
|
|
var audios []*streamer.Codec
|
|
|
|
for _, name := range strings.Split(codecs, ",") {
|
|
switch name {
|
|
case "avc1.640029":
|
|
codec := &streamer.Codec{Name: streamer.CodecH264}
|
|
videos = append(videos, codec)
|
|
case "hvc1.1.6.L153.B0":
|
|
codec := &streamer.Codec{Name: streamer.CodecH265}
|
|
videos = append(videos, codec)
|
|
case "mp4a.40.2":
|
|
codec := &streamer.Codec{Name: streamer.CodecAAC}
|
|
audios = append(audios, codec)
|
|
}
|
|
}
|
|
|
|
if videos != nil {
|
|
media := &streamer.Media{
|
|
Kind: streamer.KindVideo,
|
|
Direction: streamer.DirectionRecvonly,
|
|
Codecs: videos,
|
|
}
|
|
medias = append(medias, media)
|
|
}
|
|
|
|
if audios != nil && parseAudio {
|
|
media := &streamer.Media{
|
|
Kind: streamer.KindAudio,
|
|
Direction: streamer.DirectionRecvonly,
|
|
Codecs: audios,
|
|
}
|
|
medias = append(medias, media)
|
|
}
|
|
|
|
return
|
|
}
|