mirror of
https://github.com/mfontanini/libtins
synced 2026-01-24 19:21:35 +01:00
123 lines
3.5 KiB
C++
123 lines
3.5 KiB
C++
#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());
|
|
}
|