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

UDP now calculates the checksum when the transport layer protocol is IPv6.

Browse Source
This commit is contained in:
Matias Fontanini
2013-04-23 23:00:43 -03:00
parent 5197e7f5f1
commit cf19c8758d
1 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/udp.cpp
View File

@@ -36,6 +36,7 @@
#include "constants.h"
#include "utils.h"
#include "ip.h"
#include "ipv6.h"
#include "rawpdu.h"
#include "exceptions.h"
@@ -78,13 +79,13 @@ void UDP::write_serialization(uint8_t *buffer, uint32_t total_sz, const PDU *par
#ifdef TINS_DEBUG
assert(total_sz >= sizeof(udphdr));
#endif
const Tins::IP *ip_packet = dynamic_cast<const Tins::IP*>(parent);
_udp.check = 0;
if(inner_pdu())
length(sizeof(udphdr) + inner_pdu()->size());
else
length(sizeof(udphdr));
std::memcpy(buffer, &_udp, sizeof(udphdr));
const Tins::IP *ip_packet = dynamic_cast<const Tins::IP*>(parent);
if(ip_packet) {
uint32_t checksum = Utils::pseudoheader_checksum(
ip_packet->src_addr(),
@@ -97,6 +98,21 @@ void UDP::write_serialization(uint8_t *buffer, uint32_t total_sz, const PDU *par
_udp.check = Endian::host_to_be<uint16_t>(~checksum);
((udphdr*)buffer)->check = _udp.check;
}
else {
const Tins::IPv6 *ip6_packet = dynamic_cast<const Tins::IPv6*>(parent);
if(ip6_packet) {
uint32_t checksum = Utils::pseudoheader_checksum(
ip6_packet->src_addr(),
ip6_packet->dst_addr(),
size(),
Constants::IP::PROTO_UDP
) + Utils::do_checksum(buffer, buffer + total_sz);
while (checksum >> 16)
checksum = (checksum & 0xffff)+(checksum >> 16);
_udp.check = Endian::host_to_be<uint16_t>(~checksum);
((udphdr*)buffer)->check = _udp.check;
}
}
}
bool UDP::matches_response(const uint8_t *ptr, uint32_t total_sz) const {