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
|
// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
// SPDX-License-Identifier: MIT
// Package ciphersuite provides the crypto operations needed for a DTLS CipherSuite
package ciphersuite
import (
"testing"
"github.com/pion/dtls/v3/pkg/protocol"
"github.com/pion/dtls/v3/pkg/protocol/recordlayer"
"github.com/stretchr/testify/assert"
)
func TestGenerateAEADAdditionalDataCID(t *testing.T) {
cases := map[string]struct {
reason string
header *recordlayer.Header
payloadLen int
expected []byte
}{
"WithConnectionID": {
reason: "Should successfully generate additional data with valid header",
header: &recordlayer.Header{
ContentType: protocol.ContentTypeConnectionID,
ConnectionID: []byte{1, 2, 3, 4, 5, 6, 7, 8},
Version: protocol.Version1_2,
Epoch: 2,
SequenceNumber: 277,
},
payloadLen: 1784,
expected: []byte{
255, 255, 255, 255, 255, 255, 255, 255, 25, 8, 25, 254, 253,
0, 2, 0, 0, 0, 0, 1, 21, 1, 2, 3, 4, 5, 6, 7, 8, 6, 248,
},
},
"IgnoreContentType": {
reason: "Should use Connection ID content type regardless of header content type.",
header: &recordlayer.Header{
ContentType: protocol.ContentTypeAlert,
ConnectionID: []byte{1, 2, 3, 4, 5, 6, 7, 8},
Version: protocol.Version1_2,
Epoch: 2,
SequenceNumber: 277,
},
payloadLen: 1784,
expected: []byte{
255, 255, 255, 255, 255, 255, 255, 255, 25, 8, 25, 254, 253,
0, 2, 0, 0, 0, 0, 1, 21, 1, 2, 3, 4, 5, 6, 7, 8, 6, 248,
},
},
}
for name, tc := range cases {
t.Run(name, func(t *testing.T) {
data := generateAEADAdditionalDataCID(tc.header, tc.payloadLen)
assert.Equal(t, tc.expected, data)
})
}
}
|
