2020-10-03 12:56:16 +02:00
|
|
|
package rtpconn
|
|
|
|
|
|
|
|
import (
|
|
|
|
"io"
|
|
|
|
"log"
|
|
|
|
|
|
|
|
"github.com/pion/rtp"
|
2020-10-03 12:54:17 +02:00
|
|
|
"github.com/pion/rtp/codecs"
|
2020-10-03 12:56:16 +02:00
|
|
|
"github.com/pion/webrtc/v3"
|
|
|
|
|
|
|
|
"sfu/packetcache"
|
|
|
|
"sfu/rtptime"
|
|
|
|
)
|
|
|
|
|
2020-10-03 12:54:17 +02:00
|
|
|
func isVP8Keyframe(packet *rtp.Packet) bool {
|
|
|
|
var vp8 codecs.VP8Packet
|
|
|
|
_, err := vp8.Unmarshal(packet.Payload)
|
|
|
|
if err != nil {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
return vp8.S != 0 && vp8.PID == 0 &&
|
|
|
|
len(vp8.Payload) > 0 && (vp8.Payload[0]&0x1) == 0
|
|
|
|
}
|
|
|
|
|
2020-10-03 12:56:16 +02:00
|
|
|
func readLoop(conn *rtpUpConnection, track *rtpUpTrack) {
|
|
|
|
writers := rtpWriterPool{conn: conn, track: track}
|
|
|
|
defer func() {
|
|
|
|
writers.close()
|
|
|
|
close(track.readerDone)
|
|
|
|
}()
|
|
|
|
|
|
|
|
isvideo := track.track.Kind() == webrtc.RTPCodecTypeVideo
|
2020-10-03 12:54:17 +02:00
|
|
|
codec := track.track.Codec().Name
|
2020-10-03 12:56:16 +02:00
|
|
|
buf := make([]byte, packetcache.BufSize)
|
|
|
|
var packet rtp.Packet
|
|
|
|
for {
|
|
|
|
bytes, err := track.track.Read(buf)
|
|
|
|
if err != nil {
|
|
|
|
if err != io.EOF {
|
|
|
|
log.Printf("%v", err)
|
|
|
|
}
|
|
|
|
break
|
|
|
|
}
|
|
|
|
track.rate.Accumulate(uint32(bytes))
|
|
|
|
|
|
|
|
err = packet.Unmarshal(buf[:bytes])
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("%v", err)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
track.jitter.Accumulate(packet.Timestamp)
|
|
|
|
|
2020-10-03 12:54:17 +02:00
|
|
|
kf := false
|
|
|
|
if isvideo && codec == webrtc.VP8 {
|
|
|
|
kf = isVP8Keyframe(&packet)
|
|
|
|
}
|
|
|
|
|
|
|
|
first, index := track.cache.Store(
|
|
|
|
packet.SequenceNumber, packet.Timestamp, kf, buf[:bytes],
|
|
|
|
)
|
2020-10-03 12:56:16 +02:00
|
|
|
if packet.SequenceNumber-first > 24 {
|
2020-10-04 12:51:56 +02:00
|
|
|
found, first, bitmap :=
|
|
|
|
track.cache.BitmapGet(packet.SequenceNumber - 4)
|
2020-10-03 12:56:16 +02:00
|
|
|
if found {
|
|
|
|
err := conn.sendNACK(track, first, bitmap)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("%v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
_, rate := track.rate.Estimate()
|
|
|
|
delay := uint32(rtptime.JiffiesPerSec / 1024)
|
|
|
|
if rate > 512 {
|
|
|
|
delay = rtptime.JiffiesPerSec / rate / 2
|
|
|
|
}
|
|
|
|
|
|
|
|
writers.write(packet.SequenceNumber, index, delay,
|
|
|
|
isvideo, packet.Marker)
|
|
|
|
|
|
|
|
select {
|
|
|
|
case action := <-track.localCh:
|
|
|
|
err := writers.add(action.track, action.add)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("add/remove track: %v", err)
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|