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

Fixes #98: Erase streams when they're reassembed on IPv4Reassembler

Browse Source
This commit is contained in:
Matias Fontanini
2015-08-22 10:57:45 -07:00
parent af71a4eca7
commit 891f4ac4d7
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/ip_reassembler.cpp
View File

@@ -102,14 +102,16 @@ IPv4Reassembler::packet_status IPv4Reassembler::process(PDU &pdu) {
if(ip && ip->inner_pdu()) {
// There's fragmentation
if(ip->is_fragmented()) {
key_type key = make_key(ip);
// Create it or look it up, it's the same
Internals::IPv4Stream &stream = streams[make_key(ip)];
Internals::IPv4Stream &stream = streams[key];
stream.add_fragment(ip);
if(stream.is_complete()) {
PDU *pdu = stream.allocate_pdu();
// Erase this stream, since it's already assembled
streams.erase(key);
// The packet is corrupt
if(!pdu) {
streams.erase(make_key(ip));
return FRAGMENTED;
}
ip->inner_pdu(pdu);