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

Always calculate IP checksum

Browse Source 
Fixes #105
This commit is contained in:
Matias Fontanini
2015-11-08 10:25:49 -08:00
parent 87c4963533
commit 534bdaf30b
3 changed files with 52 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
tests/src/ip.cpp
View File

@@ -26,6 +26,7 @@ public:
fragmented_ether_ip_packet[], tot_len_zero_packet[];
void test_equals(const IP &ip1, const IP &ip2);
void test_overwrite_source_address(IP& ip);
};
const uint8_t IPTest::expected_packet[] = {
@@ -718,3 +719,42 @@ TEST_F(IPTest, RemoveOption) {
PDU::serialization_type new_buffer = ip.serialize();
EXPECT_EQ(old_buffer, new_buffer);
}
void IPTest::test_overwrite_source_address(IP& ip) {
const uint8_t expected_output[] = {
69,0,0,40,0,1,0,0,128,6,38,186,1,2,3,4,8,8,8,8,0,32,0,12,0,
0,0,0,0,0,0,0,80,0,127,166,27,253,0,0
};
// Add TCP so we can check if the timestamp is correctly calculated
ip /= TCP(12, 32);
PDU::serialization_type buffer = ip.serialize();
EXPECT_EQ(IPv4Address("1.2.3.4"), ip.src_addr());
EXPECT_EQ(0x26ba, ip.checksum());
vector<uint8_t> output_buffer(expected_output,
expected_output + sizeof(expected_output));
EXPECT_EQ(output_buffer, buffer);
}
TEST_F(IPTest, OverwriteSourceAddress) {
IP ip("8.8.8.8");
ip.src_addr("1.2.3.4");
test_overwrite_source_address(ip);
}
TEST_F(IPTest, OverwriteSourceAddressUsingConstructor) {
IP ip("8.8.8.8", "1.2.3.4");
test_overwrite_source_address(ip);
}
TEST_F(IPTest, OverwriteSourceAddressConstructingFromBuffer) {
// Only the expected packet's IP layer
const uint8_t expected_output[] = {
69,0,0,20,0,1,0,0,128,0,0,0,1,2,3,4,8,8,8,8
};
IP ip(expected_output, sizeof(expected_output));
test_overwrite_source_address(ip);
}