mirror of
https://github.com/mfontanini/libtins
synced 2026-01-27 12:14:26 +01:00
129 lines
3.6 KiB
C++
129 lines
3.6 KiB
C++
#include <algorithm>
|
|
#include <gtest/gtest.h>
|
|
#include "ethernetII.h"
|
|
#include "utils.h"
|
|
#include "network_interface.h"
|
|
|
|
using namespace Tins;
|
|
|
|
typedef EthernetII::address_type address_type;
|
|
|
|
class EthernetIITest : public ::testing::Test {
|
|
public:
|
|
static const uint8_t expected_packet[];
|
|
static address_type s_addr;
|
|
static address_type d_addr;
|
|
static address_type empty_addr;
|
|
static const NetworkInterface iface;
|
|
static const uint16_t p_type;
|
|
|
|
void test_equals(const EthernetII ð1, const EthernetII ð2);
|
|
};
|
|
|
|
const uint8_t EthernetIITest::expected_packet[] = {
|
|
'\xaa', '\xbb', '\xcc', '\xdd',
|
|
'\xee', '\xff', '\x8a', '\x8b',
|
|
'\x8c', '\x8d', '\x8e', '\x8f',
|
|
'\xd0', '\xab'
|
|
};
|
|
|
|
address_type EthernetIITest::s_addr("8a:8b:8c:8d:8e:8f");
|
|
|
|
address_type EthernetIITest::d_addr("aa:bb:cc:dd:ee:ff");
|
|
|
|
address_type EthernetIITest::empty_addr;
|
|
|
|
const NetworkInterface EthernetIITest::iface("lo");
|
|
|
|
const uint16_t EthernetIITest::p_type = 0xd0ab;
|
|
|
|
void EthernetIITest::test_equals(const EthernetII ð1, const EthernetII ð2) {
|
|
EXPECT_EQ(eth1.dst_addr(), eth2.dst_addr());
|
|
EXPECT_EQ(eth1.src_addr(), eth2.src_addr());
|
|
EXPECT_EQ(eth1.payload_type(), eth2.payload_type());
|
|
EXPECT_EQ(eth1.iface(), eth2.iface());
|
|
EXPECT_EQ((bool)eth1.inner_pdu(), (bool)eth2.inner_pdu());
|
|
}
|
|
|
|
TEST_F(EthernetIITest, DefaultConstructor) {
|
|
EthernetII eth(iface);
|
|
EXPECT_EQ(eth.iface(), iface);
|
|
EXPECT_EQ(eth.dst_addr(), empty_addr);
|
|
EXPECT_EQ(eth.src_addr(), empty_addr);
|
|
EXPECT_EQ(eth.payload_type(), 0);
|
|
EXPECT_FALSE(eth.inner_pdu());
|
|
EXPECT_EQ(eth.pdu_type(), PDU::ETHERNET_II);
|
|
}
|
|
|
|
TEST_F(EthernetIITest, CopyConstructor) {
|
|
EthernetII eth1(expected_packet, sizeof(expected_packet));
|
|
EthernetII eth2(eth1);
|
|
test_equals(eth1, eth2);
|
|
}
|
|
|
|
TEST_F(EthernetIITest, CopyAssignmentOperator) {
|
|
EthernetII eth1(expected_packet, sizeof(expected_packet));
|
|
EthernetII eth2;
|
|
eth2 = eth1;
|
|
test_equals(eth1, eth2);
|
|
}
|
|
|
|
TEST_F(EthernetIITest, NestedCopy) {
|
|
EthernetII *nested = new EthernetII(expected_packet, sizeof(expected_packet));
|
|
EthernetII eth1(expected_packet, sizeof(expected_packet));
|
|
eth1.inner_pdu(nested);
|
|
EthernetII eth2(eth1);
|
|
test_equals(eth1, eth2);
|
|
}
|
|
|
|
TEST_F(EthernetIITest, SourceAddress) {
|
|
EthernetII eth;
|
|
eth.src_addr(s_addr);
|
|
ASSERT_EQ(eth.src_addr(), s_addr);
|
|
}
|
|
|
|
TEST_F(EthernetIITest, DestinationAddress) {
|
|
EthernetII eth;
|
|
eth.dst_addr(d_addr);
|
|
ASSERT_EQ(eth.dst_addr(), d_addr);
|
|
}
|
|
|
|
TEST_F(EthernetIITest, PayloadType) {
|
|
EthernetII eth;
|
|
eth.payload_type(p_type);
|
|
ASSERT_EQ(eth.payload_type(), p_type);
|
|
}
|
|
|
|
TEST_F(EthernetIITest, Interface) {
|
|
EthernetII eth;
|
|
eth.iface(iface);
|
|
ASSERT_EQ(eth.iface(), iface);
|
|
}
|
|
|
|
TEST_F(EthernetIITest, CompleteConstructor) {
|
|
EthernetII* eth2 = new EthernetII();
|
|
EthernetII eth(iface, d_addr, s_addr, eth2);
|
|
EXPECT_EQ(eth.dst_addr(), d_addr);
|
|
EXPECT_EQ(eth.src_addr(), s_addr);
|
|
EXPECT_TRUE(eth.inner_pdu() == eth2);
|
|
EXPECT_EQ(eth.payload_type(), 0);
|
|
EXPECT_EQ(eth.iface(), iface);
|
|
}
|
|
|
|
TEST_F(EthernetIITest, Serialize) {
|
|
EthernetII eth(iface, d_addr, s_addr);
|
|
eth.payload_type(p_type);
|
|
PDU::serialization_type serialized = eth.serialize();
|
|
ASSERT_EQ(serialized.size(), sizeof(expected_packet));
|
|
EXPECT_TRUE(std::equal(serialized.begin(), serialized.end(), expected_packet));
|
|
}
|
|
|
|
TEST_F(EthernetIITest, ConstructorFromBuffer) {
|
|
EthernetII eth(expected_packet, sizeof(expected_packet));
|
|
EXPECT_EQ(eth.src_addr(), s_addr);
|
|
EXPECT_EQ(eth.dst_addr(), d_addr);
|
|
EXPECT_EQ(eth.payload_type(), p_type);
|
|
}
|
|
|
|
|