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

UDP PDU is now working.

Browse Source
This commit is contained in:
Matias
2011-08-12 21:17:44 -03:00
parent 81bcdeb402
commit 398ba31111
8 changed files with 121 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/pdu.cpp
View File

@@ -20,8 +20,10 @@
*/
#include <cassert>
#include "utils.h"
#include "pdu.h"
Tins::PDU::PDU(uint32_t flag, PDU *next_pdu) : _flag(flag), _inner_pdu(next_pdu) {
}
@@ -56,7 +58,7 @@ uint8_t *Tins::PDU::serialize(uint32_t &sz) {
return buffer;
}
void Tins::PDU::serialize(uint8_t *buffer, uint32_t total_sz, PDU *parent) {
void Tins::PDU::serialize(uint8_t *buffer, uint32_t total_sz, const PDU *parent) {
uint32_t sz = header_size() + trailer_size();
/* Must not happen... */
assert(total_sz >= sz);
@@ -65,3 +67,29 @@ void Tins::PDU::serialize(uint8_t *buffer, uint32_t total_sz, PDU *parent) {
write_serialization(buffer, total_sz, parent);
}
/* Static methods */
uint32_t Tins::PDU::do_checksum(uint8_t *start, uint8_t *end) {
uint32_t checksum(0);
uint16_t *ptr = (uint16_t*)start, *last = (uint16_t*)end, padding(0);
if(((end - start) & 1) == 1) {
last = (uint16_t*)end - 1;
padding = *(end - 1) << 8;
}
while(ptr < last)
checksum += Utils::net_to_host_s(*(ptr++));
return checksum + padding;
}
uint32_t Tins::PDU::pseudoheader_checksum(uint32_t source_ip, uint32_t dest_ip, uint32_t len, uint32_t flag) {
uint32_t checksum(0);
source_ip = Utils::net_to_host_l(source_ip);
dest_ip = Utils::net_to_host_l(dest_ip);
uint16_t *ptr = (uint16_t*)&source_ip;
checksum += *ptr + ptr[1];
ptr = (uint16_t*)&dest_ip;
checksum += *ptr + ptr[1];
checksum += flag + len;
return checksum;
}