RTMP Server 转发到 ffmpeg

问题描述

我正在寻找一种解决方案来接受 rtmp 连接并在 demux 后转发到 ffmpeg 管道。 这是我尝试过的解决方案，我只是把 Handeling Connection 的一部分放在了一个基于以下的长代码中： https://github.com/gwuhaolin/livego

func (s *Server) handleConn(conn *core.Conn) error {
    if err := conn.HandshakeServer(); err != nil {
        conn.Close()
        log.Println("handleConn HandshakeServer err: ",err)
        return err
    }
    connServer := core.NewConnServer(conn)

    if err := connServer.ReadMsg(); err != nil {
        conn.Close()
        log.Println("handleConn read msg err: ",err)
        return err
    }
    appname,name,url := connServer.GetInfo()
        if connServer.IsPublisher() {
        //todo: check token if is valid
        connServer.PublishInfo.Name = appname
        reader := NewVirReader(connServer)
        s.handler.HandleReader(reader)
        log.Println("new publisher: %+v",reader.Info())

        if s.getter != nil {
            writeType := reflect.TypeOf(s.getter)
            log.Println("handleConn:writeType=%v",writeType)
            writer := s.getter.GetWriter(reader.Info())
            s.handler.HandleWriter(writer)
        }
        flvWriter := new(flvBus.Bus)
        flvWriter.SetBackend(q)
        s.handler.HandleWriter(flvWriter.GetWriter(reader.Info()))
       } else {
        writer := NewVirWriter(connServer)
        log.Println("new player: %+v",writer.Info())
        s.handler.HandleWriter(writer)
    }

    return nil
}

然后在 flv 总线上： 假设ctx是ffmpeg的pipe0。

package flvBus

import (
    "github.com/gwuhaolin/livego/av"
    "github.com/gwuhaolin/livego/protocol/amf"
    "github.com/gwuhaolin/livego/utils/pio"
    "github.com/gwuhaolin/livego/utils/uid"
    "time"
)

var (
    flvHeader = []byte{0x46,0x4c,0x56,0x01,0x05,0x00,0x09}
)

const (
    headerLen = 11
)

type FLVWriter struct {
    Uid string
    av.RWBaser
    app,title,url string
    buf             []byte
    closed          chan struct{}
    ctx             Writer
}

func NewFLVWriter(app,url string,ctx Writer) *FLVWriter {
    ret := &FLVWriter{
        Uid:     uid.NewId(),app:     app,title:   title,url:     url,ctx:     ctx,RWBaser: av.NewRWBaser(time.Second * 10),closed:  make(chan struct{}),buf:     make([]byte,headerLen),}

    ret.ctx.Write(flvHeader)
    pio.PutI32BE(ret.buf[:4],0)
    ret.ctx.Write(ret.buf[:4])
    return ret
}

func (writer *FLVWriter) Write(p *av.Packet) error {
    writer.RWBaser.SetPreTime()
    h := writer.buf[:headerLen]

    typeID := av.TAG_VIDEO
    if !p.IsVideo {
        if p.IsMetadata {
            var err error
            typeID = av.TAG_SCRIPTDATAAMF0
            p.Data,err = amf.MetaDataReform(p.Data,amf.DEL)
            if err != nil {
                return err
            }
        } else {
            typeID = av.TAG_AUdio
        }
    }
    dataLen := len(p.Data)
    timestamp := p.TimeStamp
    timestamp += writer.BaseTimeStamp()
    writer.RWBaser.RecTimeStamp(timestamp,uint32(typeID))

    preDataLen := dataLen + headerLen
    timestampbase := timestamp & 0xffffff
    timestampExt := timestamp >> 24 & 0xff

    pio.PutU8(h[0:1],uint8(typeID))
    pio.PutI24BE(h[1:4],int32(dataLen))
    pio.PutI24BE(h[4:7],int32(timestampbase))
    pio.PutU8(h[7:8],uint8(timestampExt))

    if _,err := writer.ctx.Write(h); err != nil {
        return err
    }

    if _,err := writer.ctx.Write(p.Data); err != nil {
        return err
    }

    pio.PutI32BE(h[:4],int32(preDataLen))
    if _,err := writer.ctx.Write(h[:4]); err != nil {
        return err
    }

    return nil
}

func (writer *FLVWriter) Wait() {
    select {
    case <-writer.closed:
        return
    }
}

func (writer *FLVWriter) Close(error) {
    //writer.ctx.Close()
    close(writer.closed)
}

func (writer *FLVWriter) Info() (ret av.Info) {
    ret.UID = writer.Uid
    ret.URL = writer.url
    ret.Key = writer.app + "/" + writer.title
    return
}

但最后它返回给我 pipe:0: Invalid data found when processing input.

如何将捕获的帧转发到 ffmpeg？ 也欢迎任何其他解决方案。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）