3rd-party/libtins
1
0
Fork 0
mirror of https://github.com/mfontanini/libtins synced 2026-01-23 02:35:57 +01:00
Code Issues Releases Wiki Activity

Added ARPTest.

Browse Source
This commit is contained in:
Matias Fontanini
2011-09-15 11:42:26 -03:00
parent 7bc143e290
commit 13e260b825
1 changed files with 140 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

140
tests/src/arp.cpp Normal file
View File

@@ -0,0 +1,140 @@
#include <gtest/gtest.h>
#include <cstring>
#include <string>
#include <stdint.h>
#include "arp.h"
#include "utils.h"
using namespace std;
using namespace Tins;
class ARPTest : public testing::Test {
public:
static const uint8_t empty_addr[];
static const uint8_t hw_addr1[];
static const uint8_t hw_addr2[];
static const string ip_addr1;
static const uint8_t expected_packet[];
};
const uint8_t ARPTest::empty_addr[] = {'\x00', '\x00', '\x00', '\x00', '\x00', '\x00'};
const uint8_t ARPTest::hw_addr1[] = {'\x13', '\xda', '\xde', '\xf1', '\x01', '\x85'};
const uint8_t ARPTest::hw_addr2[] = {'\x7a', '\x1f', '\xf4', '\x39', '\xab', '\x0d'};
const string ARPTest::ip_addr1("192.168.0.154");
const uint8_t ARPTest::expected_packet[] = {'\x00', '\x01', '\x08', '\x00', '\x06', '\x04', '\x00', '\x02', '\x03', '\xde', '\xf5', '\x12', '\t', '\xfa', '\xc0', '\xa8', '-', '\xe7', '\xf5', '\x12', '\xda', 'g', '\xbd', '\r', ' ', '\x9b', 'Q', '\xfe'};
TEST_F(ARPTest, DefaultContructor) {
ARP arp;
EXPECT_EQ(arp.target_ip_addr(), 0);
EXPECT_EQ(arp.sender_ip_addr(), 0);
EXPECT_TRUE(memcmp(arp.target_hw_addr(), empty_addr, sizeof(empty_addr)) == 0);
EXPECT_TRUE(memcmp(arp.target_hw_addr(), empty_addr, sizeof(empty_addr)) == 0);
EXPECT_EQ(arp.pdu_type(), PDU::ARP);
}
TEST_F(ARPTest, CompleteContructor) {
ARP arp(0x1234, 0xa3f1, hw_addr1, hw_addr2);
EXPECT_TRUE(memcmp(arp.target_hw_addr(), hw_addr1, sizeof(hw_addr1)) == 0);
EXPECT_TRUE(memcmp(arp.sender_hw_addr(), hw_addr2, sizeof(hw_addr2)) == 0);
EXPECT_EQ(arp.target_ip_addr(), 0x1234);
EXPECT_EQ(arp.sender_ip_addr(), 0xa3f1);
}
TEST_F(ARPTest, SenderIPAddrString) {
ARP arp;
arp.sender_ip_addr(ip_addr1);
EXPECT_EQ(arp.sender_ip_addr(), Utils::ip_to_int(ip_addr1));
}
TEST_F(ARPTest, SenderIPAddrInt) {
ARP arp;
arp.sender_ip_addr(Utils::ip_to_int(ip_addr1));
EXPECT_EQ(arp.sender_ip_addr(), Utils::ip_to_int(ip_addr1));
}
TEST_F(ARPTest, TargetIPAddrString) {
ARP arp;
arp.target_ip_addr(ip_addr1);
EXPECT_EQ(arp.target_ip_addr(), Utils::ip_to_int(ip_addr1));
}
TEST_F(ARPTest, TargetIPAddrInt) {
ARP arp;
arp.target_ip_addr(Utils::ip_to_int(ip_addr1));
EXPECT_EQ(arp.target_ip_addr(), Utils::ip_to_int(ip_addr1));
}
TEST_F(ARPTest, TargetHWAddr) {
ARP arp;
arp.target_hw_addr(hw_addr1);
EXPECT_TRUE(memcmp(arp.target_hw_addr(), hw_addr1, sizeof(hw_addr1)) == 0);
}
TEST_F(ARPTest, SenderHWAddr) {
ARP arp;
arp.sender_hw_addr(hw_addr1);
EXPECT_TRUE(memcmp(arp.sender_hw_addr(), hw_addr1, sizeof(hw_addr1)) == 0);
}
TEST_F(ARPTest, ProtAddrFormat) {
ARP arp;
arp.prot_addr_format(0x45fa);
EXPECT_EQ(arp.prot_addr_format(), 0x45fa);
}
TEST_F(ARPTest, ProtAddrLength) {
ARP arp;
arp.prot_addr_length(0x4f);
EXPECT_EQ(arp.prot_addr_length(), 0x4f);
}
TEST_F(ARPTest, HWAddrFormat) {
ARP arp;
arp.hw_addr_format(0x45fa);
EXPECT_EQ(arp.hw_addr_format(), 0x45fa);
}
TEST_F(ARPTest, HWAddrLength) {
ARP arp;
arp.hw_addr_length(0xd1);
EXPECT_EQ(arp.hw_addr_length(), 0xd1);
}
TEST_F(ARPTest, Opcode) {
ARP arp;
arp.opcode(ARP::REQUEST);
EXPECT_EQ(arp.opcode(), ARP::REQUEST);
}
TEST_F(ARPTest, Serialize) {
ARP arp1(0x1234, 0xa3f1, hw_addr1, hw_addr2);
uint32_t size;
uint8_t *buffer = arp1.serialize(size);
ASSERT_TRUE(buffer);
ARP arp2(arp1);
uint32_t size2;
uint8_t *buffer2 = arp2.serialize(size2);
ASSERT_EQ(size, size2);
EXPECT_TRUE(memcmp(buffer, buffer2, size) == 0);
delete[] buffer;
}
TEST_F(ARPTest, ConstructorFromBuffer) {
ARP arp1(expected_packet, sizeof(expected_packet));
uint32_t size;
uint8_t *buffer = arp1.serialize(size);
ARP arp2(buffer, size);
EXPECT_EQ(arp1.opcode(), arp2.opcode());
ASSERT_EQ(arp1.hw_addr_length(), arp2.hw_addr_length());
EXPECT_EQ(arp1.hw_addr_format(), arp2.hw_addr_format());
ASSERT_EQ(arp1.prot_addr_length(), arp2.prot_addr_length());
EXPECT_EQ(arp1.prot_addr_format(), arp2.prot_addr_format());
EXPECT_EQ(arp1.sender_ip_addr(), arp2.sender_ip_addr());
EXPECT_EQ(arp1.target_ip_addr(), arp2.target_ip_addr());
EXPECT_TRUE(memcmp(arp1.sender_hw_addr(), arp2.sender_hw_addr(), arp2.hw_addr_length()) == 0);
EXPECT_TRUE(memcmp(arp1.target_hw_addr(), arp2.target_hw_addr(), arp2.hw_addr_length()) == 0);
}