inet::ieee80211::BasicReassembly Class Reference

#include <BasicReassembly.h>

Inheritance diagram for inet::ieee80211::BasicReassembly:

Classes
struct	Key
struct	Value

Public Member Functions
virtual	~BasicReassembly ()
virtual Packet *	addFragment (Packet *packet) override
	Add a fragment to the reassembly buffer. More...
virtual void	purge (const MacAddress &address, int tid, int startSeqNumber, int endSeqNumber) override
	Discard fragments from the reassembly buffer. More...
Public Member Functions inherited from inet::ieee80211::IReassembly
virtual	~IReassembly ()

Protected Types
typedef std::map< Key, Value >	FragmentsMap

Protected Attributes
FragmentsMap	fragmentsMap

Member Typedef Documentation

FragmentsMap

typedef std::map<Key, Value> inet::ieee80211::BasicReassembly::FragmentsMap

protected

Constructor & Destructor Documentation

~BasicReassembly()

inet::ieee80211::BasicReassembly::~BasicReassembly ( )

virtual

{
    for (auto it = fragmentsMap.begin(); it != fragmentsMap.end(); ++it)
        for (auto fragment : it->second.fragments)
            delete fragment;
}

Member Function Documentation

addFragment()

Packet * inet::ieee80211::BasicReassembly::addFragment ( Packet *  frame )

overridevirtual

Add a fragment to the reassembly buffer.

If the new fragment completes a frame, then the reassembled frame is returned (and fragments are removed from the buffer), otherwise the function returns nullptr.

Implements inet::ieee80211::IReassembly.

{
    const auto& header = packet->peekAtFront<Ieee80211DataOrMgmtHeader>();
    // Frame is not fragmented
    if (!header->getMoreFragments() && header->getFragmentNumber() == 0)
        return packet;
    // FIXME temporary fix for mgmt frames
    if (dynamicPtrCast<const Ieee80211MgmtHeader>(header))
        return packet;
    // find entry for this frame
    Key key;
    key.macAddress = header->getTransmitterAddress();
    key.tid = -1;
    if (header->getType() == ST_DATA_WITH_QOS)
        if (const Ptr<const Ieee80211DataHeader>& qosDataHeader = dynamicPtrCast<const Ieee80211DataHeader>(header))
            key.tid = qosDataHeader->getTid();
    key.seqNum = header->getSequenceNumber().get();
    short fragNum = header->getFragmentNumber();
    ASSERT(fragNum >= 0 && fragNum < MAX_NUM_FRAGMENTS);
    auto& value = fragmentsMap[key];
    value.fragments.resize(16);
 
    // update entry
    uint16_t fragmentBit = 1 << fragNum;
    value.receivedFragments |= fragmentBit;
    if (!header->getMoreFragments())
        value.allFragments = (fragmentBit << 1) - 1;
    if (!value.fragments[fragNum])
        value.fragments[fragNum] = packet;
    else
        delete packet;
 
//    MacAddress txAddress = header->getTransmitterAddress();
 
    // if all fragments arrived, return assembled frame
    if (value.allFragments != 0 && value.allFragments == value.receivedFragments) {
        Defragmentation defragmentation;
        value.fragments.erase(std::remove(value.fragments.begin(), value.fragments.end(), nullptr), value.fragments.end());
        auto defragmentedFrame = defragmentation.defragmentFrames(&value.fragments);
        // We need to restore some data from the carrying frame's header like TX address
        // TODO Maybe we need to restore the fromDs, toDs fields as well when traveling through multiple APs
        // TODO Are there any other fields that we need to restore?
        for (auto fragment : value.fragments)
            delete fragment;
        fragmentsMap.erase(key);
        return defragmentedFrame;
    }
    else
        return nullptr;
}

purge()

void inet::ieee80211::BasicReassembly::purge ( const MacAddress &  address, int  tid, int  startSeqNumber, int  endSeqNumber  )

overridevirtual

Discard fragments from the reassembly buffer.

Frames are identified by the transmitter address, the TID, and the sequence number range [startSeqNumber, endSeqNumber[. Set tid=-1 for non-QoS frames.

Implements inet::ieee80211::IReassembly.

{
    Key key;
    key.macAddress = address;
    key.tid = tid;
    key.seqNum = startSeqNumber;
    auto itStart = fragmentsMap.lower_bound(key);
    key.seqNum = endSeqNumber;
    auto itEnd = fragmentsMap.upper_bound(key);
 
    if (endSeqNumber < startSeqNumber) {
        for (auto it = itStart; it != fragmentsMap.end();) {
            for (auto fragment : it->second.fragments)
                delete fragment;
            it = fragmentsMap.erase(it);
        }
        itStart = fragmentsMap.begin();
    }
    for (auto it = itStart; it != itEnd;) {
        for (auto fragment : it->second.fragments)
            delete fragment;
        it = fragmentsMap.erase(it);
    }
}

Member Data Documentation

fragmentsMap

FragmentsMap inet::ieee80211::BasicReassembly::fragmentsMap

protected

Referenced by addFragment(), purge(), and ~BasicReassembly().

---

The documentation for this class was generated from the following files:

• BasicReassembly.h
• BasicReassembly.cc