mirror of
https://github.com/mfontanini/libtins
synced 2026-01-24 03:05:57 +01:00
82 lines
2.5 KiB
C++
82 lines
2.5 KiB
C++
#include <gtest/gtest.h>
|
|
#include <algorithm>
|
|
#include <string>
|
|
#include <stdint.h>
|
|
#include "ip.h"
|
|
#include "tcp.h"
|
|
#include "udp.h"
|
|
#include "rawpdu.h"
|
|
#include "pdu.h"
|
|
#include "packet.h"
|
|
|
|
using namespace std;
|
|
using namespace Tins;
|
|
|
|
class PDUTest : public testing::Test {
|
|
public:
|
|
};
|
|
|
|
TEST_F(PDUTest, FindPDU) {
|
|
IP ip = IP("192.168.0.1") / TCP(22, 52) / RawPDU("Test");
|
|
EXPECT_TRUE(ip.find_pdu<TCP>() != NULL);
|
|
EXPECT_TRUE(ip.find_pdu<RawPDU>() != NULL);
|
|
EXPECT_FALSE(ip.find_pdu<UDP>() != NULL);
|
|
TCP &t1 = ip.rfind_pdu<TCP>();
|
|
const TCP &t2 = ip.rfind_pdu<TCP>();
|
|
(void)t1;
|
|
(void)t2;
|
|
EXPECT_THROW(ip.rfind_pdu<UDP>(), pdu_not_found);
|
|
}
|
|
|
|
TEST_F(PDUTest, OperatorConcat) {
|
|
std::string raw_payload = "Test";
|
|
IP ip = IP("192.168.0.1") / TCP(22, 52) / RawPDU(raw_payload);
|
|
EXPECT_EQ(ip.dst_addr(), "192.168.0.1");
|
|
ASSERT_TRUE(ip.inner_pdu() != NULL);
|
|
TCP *tcp = ip.find_pdu<TCP>();
|
|
ASSERT_TRUE(tcp != NULL);
|
|
EXPECT_EQ(tcp->dport(), 22);
|
|
EXPECT_EQ(tcp->sport(), 52);
|
|
ASSERT_TRUE(tcp->inner_pdu() != NULL);
|
|
RawPDU *raw = tcp->find_pdu<RawPDU>();
|
|
ASSERT_TRUE(raw != NULL);
|
|
ASSERT_EQ(raw->payload_size(), raw_payload.size());
|
|
EXPECT_TRUE(std::equal(raw_payload.begin(), raw_payload.end(), raw->payload().begin()));
|
|
}
|
|
|
|
TEST_F(PDUTest, OperatorConcatOnPointers) {
|
|
std::string raw_payload = "Test";
|
|
IP ip = IP("192.168.0.1") / TCP(22, 52);
|
|
TCP *tcp = ip.find_pdu<TCP>();
|
|
ASSERT_TRUE(tcp != NULL);
|
|
tcp /= RawPDU(raw_payload);
|
|
RawPDU *raw = ip.find_pdu<RawPDU>();
|
|
ASSERT_TRUE(raw != NULL);
|
|
ASSERT_EQ(raw->payload_size(), raw_payload.size());
|
|
EXPECT_TRUE(std::equal(raw->payload().begin(), raw->payload().end(), raw_payload.begin()));
|
|
}
|
|
|
|
TEST_F(PDUTest, OperatorConcatOnPacket) {
|
|
std::string raw_payload = "Test";
|
|
Packet packet = IP("192.168.0.1") / TCP(22, 52);
|
|
TCP *tcp = packet.pdu()->find_pdu<TCP>();
|
|
ASSERT_TRUE(tcp != NULL);
|
|
tcp /= RawPDU(raw_payload);
|
|
RawPDU *raw = packet.pdu()->find_pdu<RawPDU>();
|
|
ASSERT_TRUE(raw != NULL);
|
|
ASSERT_EQ(raw->payload_size(), raw_payload.size());
|
|
EXPECT_TRUE(std::equal(raw->payload().begin(), raw->payload().end(), raw_payload.begin()));
|
|
}
|
|
|
|
TEST_F(PDUTest, TinsCast) {
|
|
PDU *null_pdu = 0;
|
|
TCP tcp;
|
|
PDU *pdu = &tcp;
|
|
EXPECT_EQ(tins_cast<TCP*>(pdu), &tcp);
|
|
EXPECT_EQ(tins_cast<const TCP*>(pdu), &tcp);
|
|
EXPECT_EQ(tins_cast<TCP*>(null_pdu), null_pdu);
|
|
EXPECT_EQ(tins_cast<UDP*>(pdu), null_pdu);
|
|
EXPECT_THROW(tins_cast<UDP>(*pdu), bad_tins_cast);
|
|
}
|
|
|