zishuo/webrtc
1
0
Fork 0
mirror of https://github.com/pion/webrtc.git synced 2025-10-06 15:46:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add support for RTP header extensions, and IVF videos now fully work!

Browse Source
This commit is contained in:
Sean DuBois
2018-06-11 01:42:41 -07:00
parent 882b87b346
commit d1608eda58
3 changed files with 47 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
rtp/unmarshal.go
View File

@@ -2,6 +2,7 @@ package rtp
import (
"encoding/binary"
"github.com/pkg/errors"
)
@@ -37,9 +38,9 @@ func (p *Packet) Unmarshal(rawPacket []byte) error {
p.Timestamp = binary.BigEndian.Uint32(rawPacket[timestampOffset : timestampOffset+timestampLength])
p.SSRC = binary.BigEndian.Uint32(rawPacket[ssrcOffset : ssrcOffset+ssrcLength])
headerPlusCsrcs := headerLength + timestampLength + ssrcLength + (len(p.CSRC) * csrcLength)
if len(rawPacket) < headerPlusCsrcs {
return errors.Errorf("RTP header size insufficient; %d < %d", len(rawPacket), headerPlusCsrcs)
currOffset := headerLength + timestampLength + ssrcLength + (len(p.CSRC) * csrcLength)
if len(rawPacket) < currOffset {
return errors.Errorf("RTP header size insufficient; %d < %d", len(rawPacket), currOffset)
}
for i := range p.CSRC {
@@ -47,7 +48,15 @@ func (p *Packet) Unmarshal(rawPacket []byte) error {
p.CSRC[i] = binary.BigEndian.Uint32(rawPacket[offset:offset])
}
p.Payload = rawPacket[headerPlusCsrcs:]
if p.Extension {
currOffset += extensionHeaderIdLength
extensionLength := binary.BigEndian.Uint16(rawPacket[currOffset:])
currOffset += extensionLengthFieldLength
currOffset += (int(extensionLength) * extensionHeaderAssumedLength)
}
p.Payload = rawPacket[currOffset:]
return nil
}