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
|
// Copyright 2018 Google, Inc. All rights reserved.
//
// Use of this source code is governed by a BSD-style license
// that can be found in the LICENSE file in the root of the source
// tree.
package lcmdefrag
import (
"testing"
"github.com/gopacket/gopacket"
"github.com/gopacket/gopacket/layers"
)
var (
fragmentOne = []byte{
0x4c, 0x43, 0x30, 0x33, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0d,
0x00, 0x00, 0x00, 0x2d, 0x00, 0x00, 0x00, 0x02, 0x4c, 0x43, 0x4d, 0x5f,
0x53, 0x45, 0x4c, 0x46, 0x5f, 0x54, 0x45, 0x53, 0x54, 0x00, 0x6c, 0x63,
0x6d, 0x20, 0x73, 0x65, 0x6c, 0x66, 0x20, 0x74, 0x65, 0x73, 0x74,
}
fragmentTwo = []byte{
0x4c, 0x43, 0x30, 0x33, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0d,
0x00, 0x00, 0x00, 0x2d, 0x00, 0x01, 0x00, 0x02, 0x6c, 0x63, 0x6d, 0x20,
0x73, 0x65, 0x6c, 0x66, 0x20, 0x74, 0x65, 0x73, 0x74,
}
completePacket = []byte{
0x4c, 0x43, 0x30, 0x32, 0x00, 0x00, 0x00, 0x00, 0x4c, 0x43, 0x4d, 0x5f,
0x53, 0x45, 0x4c, 0x46, 0x5f, 0x54, 0x45, 0x53, 0x54, 0x00, 0x6c, 0x63,
0x6d, 0x20, 0x73, 0x65, 0x6c, 0x66, 0x20, 0x74, 0x65, 0x73, 0x74,
}
)
func TestOrderedLCMDefrag(t *testing.T) {
defragmenter := NewLCMDefragmenter()
var err error
packet := gopacket.NewPacket(fragmentOne, layers.LayerTypeLCM, gopacket.NoCopy)
lcm := packet.Layer(layers.LayerTypeLCM).(*layers.LCM)
lcm, err = defragmenter.Defrag(lcm)
if lcm != nil {
t.Fatal("Returned incomplete LCM message.")
}
if err != nil {
t.Fatal(err)
}
packet = gopacket.NewPacket(fragmentTwo, layers.LayerTypeLCM, gopacket.NoCopy)
lcm = packet.Layer(layers.LayerTypeLCM).(*layers.LCM)
lcm, err = defragmenter.Defrag(lcm)
if lcm == nil {
t.Fatal("Did not receive defragmented LCM message.")
}
if err != nil {
t.Fatal(err)
}
}
func TestUnorderedLCMDefrag(t *testing.T) {
defragmenter := NewLCMDefragmenter()
var err error
packet := gopacket.NewPacket(fragmentTwo, layers.LayerTypeLCM, gopacket.NoCopy)
lcm := packet.Layer(layers.LayerTypeLCM).(*layers.LCM)
lcm, err = defragmenter.Defrag(lcm)
if lcm != nil {
t.Fatal("Returned incomplete LCM message.")
}
if err != nil {
t.Fatal(err)
}
packet = gopacket.NewPacket(fragmentOne, layers.LayerTypeLCM, gopacket.NoCopy)
lcm = packet.Layer(layers.LayerTypeLCM).(*layers.LCM)
lcm, err = defragmenter.Defrag(lcm)
if lcm == nil {
t.Fatal("Did not receive defragmented LCM message.")
}
if err != nil {
t.Fatal(err)
}
}
func TestNonLCMDefrag(t *testing.T) {
defragmenter := NewLCMDefragmenter()
var err error
packet := gopacket.NewPacket(completePacket, layers.LayerTypeLCM, gopacket.NoCopy)
lcm := packet.Layer(layers.LayerTypeLCM).(*layers.LCM)
lcm, err = defragmenter.Defrag(lcm)
if lcm == nil {
t.Fatal("Did not receive complete LCM message.")
}
if err != nil {
t.Fatal(err)
}
}
|
