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

Add parent PDU member to PDU class

Browse Source 
Now this is a bidirectional list of PDUs
This commit is contained in:
Matias Fontanini
2017-04-29 09:27:08 -07:00
parent 35383ac359
commit 1b4d22314d
4 changed files with 48 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/src/pdu_test.cpp
View File

@@ -17,7 +17,7 @@ public:
};
TEST_F(PDUTest, FindPDU) {
IP ip = IP("192.168.0.1") / TCP(22, 52) / RawPDU("Test");
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);
@@ -28,6 +28,21 @@ TEST_F(PDUTest, FindPDU) {
EXPECT_THROW(ip.rfind_pdu<UDP>(), pdu_not_found);
}
TEST_F(PDUTest, PDURelationship) {
IP packet = IP("192.168.0.1") / TCP(22, 52) / RawPDU("Test");
IP* ip = packet.find_pdu<IP>();
TCP* tcp = packet.find_pdu<TCP>();
RawPDU* raw = packet.find_pdu<RawPDU>();
ASSERT_TRUE(ip != 0);
ASSERT_TRUE(tcp != 0);
ASSERT_TRUE(raw != 0);
EXPECT_EQ(0, ip->parent_pdu());
EXPECT_EQ(ip, tcp->parent_pdu());
EXPECT_EQ(tcp, raw->parent_pdu());
}
TEST_F(PDUTest, OperatorConcat) {
std::string raw_payload = "Test";
IP ip = IP("192.168.0.1") / TCP(22, 52) / RawPDU(raw_payload);