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
114
115
116
117
118
119
120
121
122
|
#include <gtest/gtest.h>
#include <cstring>
#include <string>
#include <stdint.h>
#include <tins/dot1q.h>
#include <tins/arp.h>
#include <tins/ip.h>
#include <tins/tcp.h>
#include <tins/rawpdu.h>
#include <tins/ethernetII.h>
using namespace std;
using namespace Tins;
class Dot1QTest : public testing::Test {
public:
static const uint8_t expected_packet[];
void test_equals(const Dot1Q& pdu1, const Dot1Q& pdu2);
};
const uint8_t Dot1QTest::expected_packet[] = {
255, 255, 255, 255, 255, 255, 0, 25, 6, 234, 184, 193, 129, 0, 176,
123, 8, 6, 0, 1, 8, 0, 6, 4, 0, 2, 0, 25, 6, 234, 184, 193, 192, 168,
123, 1, 255, 255, 255, 255, 255, 255, 192, 168, 123, 1, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
};
TEST_F(Dot1QTest, DefaultConstructor) {
Dot1Q dot1;
EXPECT_EQ(0, dot1.payload_type());
EXPECT_EQ(0, dot1.priority());
EXPECT_EQ(0, dot1.cfi());
EXPECT_EQ(0, dot1.id());
}
TEST_F(Dot1QTest, ConstructorFromBuffer) {
EthernetII eth(expected_packet, sizeof(expected_packet));
const Dot1Q* dot1 = eth.find_pdu<Dot1Q>();
ASSERT_TRUE(dot1 != NULL);
EXPECT_EQ(0x806, dot1->payload_type());
EXPECT_EQ(5, dot1->priority());
EXPECT_EQ(1, dot1->cfi());
EXPECT_EQ(123, dot1->id());
const ARP* arp = dot1->find_pdu<ARP>();
ASSERT_TRUE(arp != NULL);
// just to check it the offset's OK
EXPECT_EQ(ARP::hwaddress_type("00:19:06:ea:b8:c1"), arp->sender_hw_addr());
}
TEST_F(Dot1QTest, Serialize) {
EthernetII eth(expected_packet, sizeof(expected_packet));
PDU::serialization_type buffer = eth.serialize();
EXPECT_EQ(
PDU::serialization_type(expected_packet, expected_packet + sizeof(expected_packet)),
buffer
);
}
TEST_F(Dot1QTest, SerializeCraftedPacket)
{
EthernetII pkt = EthernetII() / Dot1Q(10) / IP("192.168.1.2") / TCP(23, 45) / RawPDU("asdasdasd");
PDU::serialization_type buffer = pkt.serialize();
EXPECT_EQ(buffer[12], 0x81);
EXPECT_EQ(buffer[13], 0x00);
EthernetII pkt2(&buffer[0], buffer.size());
const Dot1Q &q1 = pkt2.rfind_pdu<Dot1Q>();
EXPECT_EQ(10, q1.id());
}
TEST_F(Dot1QTest, PayloadType) {
Dot1Q dot1;
dot1.payload_type(0x9283);
EXPECT_EQ(0x9283, dot1.payload_type());
}
TEST_F(Dot1QTest, Priority) {
Dot1Q dot1;
dot1.priority(5);
EXPECT_EQ(5, dot1.priority());
}
TEST_F(Dot1QTest, CFI) {
Dot1Q dot1;
dot1.cfi(1);
EXPECT_EQ(1, dot1.cfi());
}
TEST_F(Dot1QTest, Id) {
Dot1Q dot1;
dot1.id(3543);
EXPECT_EQ(3543, dot1.id());
}
TEST_F(Dot1QTest, QinQ) {
EthernetII pkt = EthernetII() / Dot1Q(10) / Dot1Q(42) / IP("192.168.1.2") /
TCP(23, 45) / RawPDU("asdasdasd");
PDU::serialization_type buffer = pkt.serialize();
// First PID of the serialized packet should be Dot1AD = 0x88a8
EXPECT_EQ(buffer[12], 0x88);
EXPECT_EQ(buffer[13], 0xa8);
// Second PID of the serialized packet chouls be Dot1Q = 0x8100
EXPECT_EQ(buffer[16], 0x81);
EXPECT_EQ(buffer[17], 0x00);
EthernetII pkt2(&buffer[0], buffer.size());
const Dot1Q& q1 = pkt2.rfind_pdu<Dot1Q>();
ASSERT_TRUE(q1.inner_pdu() != NULL);
const Dot1Q& q2 = q1.inner_pdu()->rfind_pdu<Dot1Q>();
EXPECT_EQ(10, q1.id());
EXPECT_EQ(42, q2.id());
}
TEST_F(Dot1QTest, SerializeAfterInnerPduRemoved) {
EthernetII eth1 = EthernetII() / Dot1Q() / IP();
eth1.serialize();
eth1.rfind_pdu<Dot1Q>().inner_pdu(0);
PDU::serialization_type buffer = eth1.serialize();
EthernetII eth2(&buffer[0], buffer.size());
EXPECT_EQ(eth1.size(), eth2.size());
}
|
