mirror of
https://github.com/mfontanini/libtins
synced 2026-01-27 12:14:26 +01:00
Moved endianness change functions to endianness.h.
This commit is contained in:
13
src/udp.cpp
13
src/udp.cpp
@@ -24,6 +24,7 @@
|
||||
#include <cstring>
|
||||
#include "udp.h"
|
||||
#include "constants.h"
|
||||
#include "utils.h"
|
||||
#include "ip.h"
|
||||
#include "rawpdu.h"
|
||||
|
||||
@@ -47,20 +48,16 @@ Tins::UDP::UDP(const uint8_t *buffer, uint32_t total_sz)
|
||||
inner_pdu(new RawPDU(buffer + sizeof(udphdr), total_sz));
|
||||
}
|
||||
|
||||
void Tins::UDP::payload(uint8_t *new_payload, uint32_t new_payload_size) {
|
||||
inner_pdu(new RawPDU(new_payload, new_payload_size));
|
||||
}
|
||||
|
||||
void Tins::UDP::dport(uint16_t new_dport) {
|
||||
_udp.dport = Utils::host_to_be(new_dport);
|
||||
_udp.dport = Endian::host_to_be(new_dport);
|
||||
}
|
||||
|
||||
void Tins::UDP::sport(uint16_t new_sport) {
|
||||
_udp.sport = Utils::host_to_be(new_sport);
|
||||
_udp.sport = Endian::host_to_be(new_sport);
|
||||
}
|
||||
|
||||
void Tins::UDP::length(uint16_t new_len) {
|
||||
_udp.len = Utils::host_to_be(new_len);
|
||||
_udp.len = Endian::host_to_be(new_len);
|
||||
}
|
||||
|
||||
uint32_t Tins::UDP::header_size() const {
|
||||
@@ -78,7 +75,7 @@ void Tins::UDP::write_serialization(uint8_t *buffer, uint32_t total_sz, const PD
|
||||
Utils::do_checksum(buffer, buffer + total_sz);
|
||||
while (checksum >> 16)
|
||||
checksum = (checksum & 0xffff)+(checksum >> 16);
|
||||
((udphdr*)buffer)->check = Utils::host_to_be<uint16_t>(~checksum);
|
||||
((udphdr*)buffer)->check = Endian::host_to_be<uint16_t>(~checksum);
|
||||
}
|
||||
_udp.check = 0;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user