Adds error handler for mp4 init

cmd/mp4/mp4.go

@@ -48,13 +48,17 @@ func handlerKeyframe(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	defer stream.RemoveConsumer(cons)
+
 	w.Header().Set("Content-Type", cons.MimeType())
 
-	data := cons.Init()
+	data, err := cons.Init()
+	if err != nil {
+		log.Error().Err(err).Msg("[api.keyframe] init")
+		return
+	}
 	data = append(data, <-exit...)
 
-	stream.RemoveConsumer(cons)
-
 	// Apple Safari won't show frame without length
 	w.Header().Set("Content-Length", strconv.Itoa(len(data)))
 
@@ -97,8 +101,13 @@ func handlerMP4(w http.ResponseWriter, r *http.Request) {
 
 	w.Header().Set("Content-Type", cons.MimeType())
 
-	data := cons.Init()
-	if _, err := w.Write(data); err != nil {
+	data, err := cons.Init()
+	if err != nil {
+		log.Error().Err(err).Msg("[api.mp4] init")
+		return
+	}
+
+	if _, err = w.Write(data); err != nil {
 		log.Error().Err(err).Msg("[api.mp4] write")
 		return
 	}

cmd/mp4/mse.go

@@ -41,5 +41,12 @@ func handlerWS(ctx *api.Context, msg *streamer.Message) {
 		Type: MsgTypeMSE, Value: cons.MimeType(),
 	})
 
-	ctx.Write(cons.Init())
+	data, err := cons.Init()
+	if err != nil {
+		log.Warn().Err(err).Msg("[api.mse] init")
+		ctx.Error(err)
+		return
+	}
+
+	ctx.Write(data)
 }

pkg/mp4/consumer.go

@@ -86,7 +86,7 @@ func (c *Consumer) MimeType() string {
 	return c.muxer.MimeType(c.codecs)
 }
 
-func (c *Consumer) Init() []byte {
+func (c *Consumer) Init() ([]byte, error) {
 	if c.muxer == nil {
 		c.muxer = &Muxer{}
 	}

pkg/mp4/muxer.go

@@ -2,6 +2,7 @@ package mp4
 
 import (
 	"encoding/binary"
+	"fmt"
 	"github.com/AlexxIT/go2rtc/pkg/h264"
 	"github.com/AlexxIT/go2rtc/pkg/streamer"
 	"github.com/deepch/vdk/codec/h264parser"
@@ -32,18 +33,21 @@ func (m *Muxer) MimeType(codecs []*streamer.Codec) string {
 	return s + `"`
 }
 
-func (m *Muxer) GetInit(codecs []*streamer.Codec) []byte {
+func (m *Muxer) GetInit(codecs []*streamer.Codec) ([]byte, error) {
 	moov := MOOV()
 
 	for _, codec := range codecs {
 		switch codec.Name {
 		case streamer.CodecH264:
 			sps, pps := h264.GetParameterSet(codec.FmtpLine)
+			if sps == nil {
+				return nil, fmt.Errorf("empty SPS: %#v", codec)
+			}
 
 			// TODO: remove
 			codecData, err := h264parser.NewCodecDataFromSPSAndPPS(sps, pps)
 			if err != nil {
-				return nil
+				return nil, err
 			}
 
 			width := codecData.Width()
@@ -83,7 +87,7 @@ func (m *Muxer) GetInit(codecs []*streamer.Codec) []byte {
 	data := make([]byte, moov.Len())
 	moov.Marshal(data)
 
-	return append(FTYP(), data...)
+	return append(FTYP(), data...), nil
 }
 
 func (m *Muxer) Rewind() {