fajiao
/
plugin-webrtc-plus
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity
Browse Source

播放成功

v4
langhuihui 5 years ago
parent
9990b2b536
commit
457c8ff185
1 changed files with 106 additions and 59 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 165
      main.go

165
main.go
View File

@ -18,6 +18,7 @@ import (
"github.com/pion/rtp" "github.com/pion/rtp"
"github.com/pion/rtp/codecs" "github.com/pion/rtp/codecs"
. "github.com/pion/webrtc/v2" . "github.com/pion/webrtc/v2"
"github.com/pion/webrtc/v2/pkg/media"
) )
var config struct { var config struct {
@ -111,19 +112,8 @@ func (rtc *WebRTC) Play(streamPath string) bool {
var pps []byte var pps []byte
sub.ID = rtc.RemoteAddr sub.ID = rtc.RemoteAddr
sub.Type = "WebRTC" sub.Type = "WebRTC"
stapA := func(naul ...[]byte) []byte { var lastTimeStamp uint32
var buffer bytes.Buffer var dataBuilder bytes.Buffer
buffer.WriteByte((naul[0][0] & 224) | 24)
for _, n := range naul {
l := len(n)
buffer.WriteByte(byte(l >> 8))
buffer.WriteByte(byte(l))
buffer.Write(n)
}
return buffer.Bytes()
}
//rtc.File, _ = os.OpenFile("webrtc.h264", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0666)
aud := []byte{0x09, 0x30}
sub.OnData = func(packet *avformat.SendPacket) error { sub.OnData = func(packet *avformat.SendPacket) error {
if packet.Type == avformat.FLV_TAG_TYPE_AUDIO { if packet.Type == avformat.FLV_TAG_TYPE_AUDIO {
return nil return nil
@ -131,68 +121,125 @@ func (rtc *WebRTC) Play(streamPath string) bool {
if packet.IsSequence { if packet.IsSequence {
payload := packet.Payload[11:] payload := packet.Payload[11:]
spsLen := int(payload[0])<<8 + int(payload[1]) spsLen := int(payload[0])<<8 + int(payload[1])
sps = payload[2:spsLen] payload = payload[2:]
payload = payload[3+spsLen:] sps = payload[:spsLen]
payload = payload[1+spsLen:]
ppsLen := int(payload[0])<<8 + int(payload[1]) ppsLen := int(payload[0])<<8 + int(payload[1])
pps = payload[2:ppsLen] payload = payload[2:]
pps = payload[:ppsLen]
} else { } else {
var s uint32
if lastTimeStamp > 0 {
s = packet.Timestamp - lastTimeStamp
}
if packet.IsKeyFrame { if packet.IsKeyFrame {
if err := rtc.WriteVideo(packet.Timestamp*90, true, stapA([]byte{0x9, 0x10}, sps, pps)); err != nil { dataBuilder.Write(avformat.NALU_Delimiter2)
return err dataBuilder.Write(sps)
} dataBuilder.Write(avformat.NALU_Delimiter2)
} else { dataBuilder.Write(pps)
if err := rtc.WriteVideo(packet.Timestamp*90, true, aud); err != nil {
return err
}
} }
payload := packet.Payload[5:] payload := packet.Payload[5:]
for { for {
var naulLen = int(util.BigEndian.Uint32(payload)) var naulLen = int(util.BigEndian.Uint32(payload))
payload = payload[4:] payload = payload[4:]
_payload := payload[:naulLen] dataBuilder.Write(avformat.NALU_Delimiter2)
if naulLen > 1000 { dataBuilder.Write(payload[:naulLen])
indicator := (_payload[0] & 224) | 28 rtc.videoTrack.WriteSample(media.Sample{
nalutype := _payload[0] & 31 Data: dataBuilder.Bytes(),
header := 128 | nalutype Samples: s * 90,
part := _payload[1:1000] })
marker := false dataBuilder.Reset()
for {
if err := rtc.WriteVideo(packet.Timestamp*90, marker, append([]byte{indicator, header}, part...)); err != nil {
return err
}
if _payload == nil {
break
}
_payload = _payload[1000:]
if len(_payload) <= 1000 {
header = 64 | nalutype
part = _payload
_payload = nil
marker = true
} else {
header = nalutype
part = _payload[:1000]
}
}
} else {
if err := rtc.WriteVideo(packet.Timestamp*90, true, _payload); err != nil {
return err
}
}
if len(payload) < naulLen+4 { if len(payload) < naulLen+4 {
break break
} }
payload = payload[naulLen:] payload = payload[naulLen:]
} }
// if err := videoTrack.WriteRTP(&rtp.Packet{
// Header: nextHeader(packet.Timestamp * 90),
// Payload: aud,
// }); err != nil {
// return err
// }
} }
lastTimeStamp = packet.Timestamp
return nil return nil
} }
// stapA := func(naul ...[]byte) []byte {
// var buffer bytes.Buffer
// buffer.WriteByte((naul[0][0] & 224) | 24)
// for _, n := range naul {
// l := len(n)
// buffer.WriteByte(byte(l >> 8))
// buffer.WriteByte(byte(l))
// buffer.Write(n)
// }
// return buffer.Bytes()
// }
// //rtc.File, _ = os.OpenFile("webrtc.h264", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0666)
// aud := []byte{0x09, 0x30}
// sub.OnData = func(packet *avformat.SendPacket) error {
// if packet.Type == avformat.FLV_TAG_TYPE_AUDIO {
// return nil
// }
// if packet.IsSequence {
// payload := packet.Payload[11:]
// spsLen := int(payload[0])<<8 + int(payload[1])
// sps = payload[2:spsLen]
// payload = payload[3+spsLen:]
// ppsLen := int(payload[0])<<8 + int(payload[1])
// pps = payload[2:ppsLen]
// } else {
// if packet.IsKeyFrame {
// if err := rtc.WriteVideo(packet.Timestamp*90, true, stapA([]byte{0x9, 0x10}, sps, pps)); err != nil {
// return err
// }
// } else {
// if err := rtc.WriteVideo(packet.Timestamp*90, true, aud); err != nil {
// return err
// }
// }
// payload := packet.Payload[5:]
// for {
// var naulLen = int(util.BigEndian.Uint32(payload))
// payload = payload[4:]
// _payload := payload[:naulLen]
// if naulLen > 1000 {
// indicator := (_payload[0] & 224) | 28
// nalutype := _payload[0] & 31
// header := 128 | nalutype
// part := _payload[1:1000]
// marker := false
// for {
// if err := rtc.WriteVideo(packet.Timestamp*90, marker, append([]byte{indicator, header}, part...)); err != nil {
// return err
// }
// if _payload == nil {
// break
// }
// _payload = _payload[1000:]
// if len(_payload) <= 1000 {
// header = 64 | nalutype
// part = _payload
// _payload = nil
// marker = true
// } else {
// header = nalutype
// part = _payload[:1000]
// }
// }
// } else {
// if err := rtc.WriteVideo(packet.Timestamp*90, true, _payload); err != nil {
// return err
// }
// }
// if len(payload) < naulLen+4 {
// break
// }
// payload = payload[naulLen:]
// }
// // if err := videoTrack.WriteRTP(&rtp.Packet{
// // Header: nextHeader(packet.Timestamp * 90),
// // Payload: aud,
// // }); err != nil {
// // return err
// // }
// }
// return nil
// }
go sub.Subscribe(streamPath) go sub.Subscribe(streamPath)
} }
}) })

Write Preview
Loading…
Cancel
Save