forked from pion/webrtc
-
Notifications
You must be signed in to change notification settings - Fork 1
/
reader.go
41 lines (33 loc) · 1.02 KB
/
reader.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package rtcp
import "io"
// A Reader reads packets from an RTCP combined packet.
type Reader struct {
r io.Reader
}
// NewReader creates a new Reader reading from r.
func NewReader(r io.Reader) *Reader {
return &Reader{r}
}
// ReadPacket reads one packet from r.
//
// It returns the parsed packet Header and a byte slice containing the encoded
// packet data (including the header). How the packet data is parsed depends on
// the Type field contained in the Header.
func (r *Reader) ReadPacket() (header Header, data []byte, err error) {
// First grab the header
headerBuf := make([]byte, headerLength)
if _, err := io.ReadFull(r.r, headerBuf); err != nil {
return header, data, err
}
if err := header.Unmarshal(headerBuf); err != nil {
return header, data, err
}
packetLen := (header.Length + 1) * 4
// Then grab the rest
bodyBuf := make([]byte, packetLen-headerLength)
if _, err := io.ReadFull(r.r, bodyBuf); err != nil {
return header, data, err
}
data = append(headerBuf, bodyBuf...)
return header, data, nil
}