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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
package packetdump
import (
"bytes"
"testing"
"time"
"github.com/pion/interceptor"
"github.com/pion/interceptor/internal/test"
"github.com/pion/logging"
"github.com/pion/rtcp"
"github.com/pion/rtp"
"github.com/stretchr/testify/assert"
)
func TestSenderFilterEverythingOut(t *testing.T) {
buf := bytes.Buffer{}
factory, err := NewSenderInterceptor(
RTPWriter(&buf),
RTCPWriter(&buf),
Log(logging.NewDefaultLoggerFactory().NewLogger("test")),
RTPFilter(func(pkt *rtp.Packet) bool {
return false
}),
RTCPFilter(func(pkt []rtcp.Packet) bool {
return false
}),
)
assert.NoError(t, err)
i, err := factory.NewInterceptor("")
assert.NoError(t, err)
assert.Zero(t, buf.Len())
stream := test.NewMockStream(&interceptor.StreamInfo{
SSRC: 123456,
ClockRate: 90000,
}, i)
defer func() {
assert.NoError(t, stream.Close())
}()
err = stream.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{
SenderSSRC: 123,
MediaSSRC: 456,
}})
assert.NoError(t, err)
err = stream.WriteRTP(&rtp.Packet{Header: rtp.Header{
SequenceNumber: uint16(0),
}})
assert.NoError(t, err)
// Give time for packets to be handled and stream written to.
time.Sleep(50 * time.Millisecond)
err = i.Close()
assert.NoError(t, err)
// Every packet should have been filtered out – nothing should be written.
assert.Zero(t, buf.Len())
}
func TestSenderFilterNothing(t *testing.T) {
buf := bytes.Buffer{}
factory, err := NewSenderInterceptor(
RTPWriter(&buf),
RTCPWriter(&buf),
Log(logging.NewDefaultLoggerFactory().NewLogger("test")),
RTPFilter(func(pkt *rtp.Packet) bool {
return true
}),
RTCPFilter(func(pkt []rtcp.Packet) bool {
return true
}),
)
assert.NoError(t, err)
i, err := factory.NewInterceptor("")
assert.NoError(t, err)
assert.EqualValues(t, 0, buf.Len())
stream := test.NewMockStream(&interceptor.StreamInfo{
SSRC: 123456,
ClockRate: 90000,
}, i)
defer func() {
assert.NoError(t, stream.Close())
}()
err = stream.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{
SenderSSRC: 123,
MediaSSRC: 456,
}})
assert.NoError(t, err)
err = stream.WriteRTP(&rtp.Packet{Header: rtp.Header{
SequenceNumber: uint16(0),
}})
assert.NoError(t, err)
// Give time for packets to be handled and stream written to.
time.Sleep(50 * time.Millisecond)
err = i.Close()
assert.NoError(t, err)
assert.NotZero(t, buf.Len())
}
|
