IgmpLayer.h

Go to the documentation of this file.

#ifndef PACKETPP_IGMP_LAYER
#define PACKETPP_IGMP_LAYER
 
#include "Layer.h"
#include "IpAddress.h"
#include <vector>
 
 
namespace pcpp
{
 
struct igmp_header
{
    uint8_t type;
    uint8_t maxResponseTime;
    uint16_t checksum;
    uint32_t groupAddress;
};
 
 
struct igmpv3_query_header
{
    uint8_t type;
    uint8_t maxResponseTime;
    uint16_t checksum;
    uint32_t groupAddress;
    uint8_t s_qrv;
    uint8_t qqic;
    uint16_t numOfSources;
};
 
 
struct igmpv3_report_header
{
    uint8_t type;
    uint8_t reserved1;
    uint16_t checksum;
    uint16_t reserved2;
    uint16_t numOfGroupRecords;
};
 
 
struct igmpv3_group_record
{
    uint8_t recordType;
    uint8_t auxDataLen;
    uint16_t numOfSources;
    uint32_t multicastAddress;
    uint8_t sourceAddresses[];
 
    IPv4Address getMulticastAddress() const { return multicastAddress; }
 
    uint16_t getSourceAddressCount() const;
 
    IPv4Address getSourceAddressAtIndex(int index) const;
 
    size_t getRecordLen() const;
};
 
 
enum IgmpType
{
    IgmpType_Unknown = 0,
    IgmpType_MembershipQuery = 0x11,
    IgmpType_MembershipReportV1 = 0x12,
    IgmpType_DVMRP = 0x13,
    IgmpType_P1Mv1 = 0x14,
    IgmpType_CiscoTrace = 0x15,
    IgmpType_MembershipReportV2 = 0x16,
    IgmpType_LeaveGroup = 0x17,
    IgmpType_MulticastTracerouteResponse = 0x1e,
    IgmpType_MulticastTraceroute = 0x1f,
    IgmpType_MembershipReportV3 = 0x22,
    IgmpType_MulticastRouterAdvertisement = 0x30,
    IgmpType_MulticastRouterSolicitation = 0x31,
    IgmpType_MulticastRouterTermination = 0x32,
};
 
 
class IgmpLayer : public Layer
{
protected:
 
    IgmpLayer(uint8_t* data, size_t dataLen, Layer* prevLayer, Packet* packet, ProtocolType igmpVer) : Layer(data, dataLen, prevLayer, packet) { m_Protocol = igmpVer; }
 
    IgmpLayer(IgmpType type, const IPv4Address& groupAddr, uint8_t maxResponseTime, ProtocolType igmpVer);
 
    uint16_t calculateChecksum();
 
    size_t getHeaderSizeByVerAndType(ProtocolType igmpVer, IgmpType igmpType) const;
public:
 
    virtual ~IgmpLayer() {}
 
    igmp_header* getIgmpHeader() const { return (igmp_header*)m_Data; }
 
    IPv4Address getGroupAddress() const { return getIgmpHeader()->groupAddress; }
 
    void setGroupAddress(const IPv4Address& groupAddr);
 
    IgmpType getType() const;
 
    void setType(IgmpType type);
 
    static ProtocolType getIGMPVerFromData(uint8_t* data, size_t dataLen, bool& isQuery);
 
 
    // implement abstract methods
 
    void parseNextLayer() {}
 
    size_t getHeaderLen() const { return sizeof(igmp_header); }
 
    std::string toString() const;
 
    OsiModelLayer getOsiModelLayer() const { return OsiModelNetworkLayer; }
};
 
 
class IgmpV1Layer : public IgmpLayer
{
public:
    IgmpV1Layer(uint8_t* data, size_t dataLen, Layer* prevLayer, Packet* packet)
        : IgmpLayer(data, dataLen, prevLayer, packet, IGMPv1) {}
 
    IgmpV1Layer(IgmpType type, const IPv4Address& groupAddr = IPv4Address())
        : IgmpLayer(type, groupAddr, 0, IGMPv1) {}
 
    ~IgmpV1Layer() {}
 
 
    // implement abstract methods
 
    void computeCalculateFields();
 
};
 
 
class IgmpV2Layer : public IgmpLayer
{
public:
    IgmpV2Layer(uint8_t* data, size_t dataLen, Layer* prevLayer, Packet* packet)
        : IgmpLayer(data, dataLen, prevLayer, packet, IGMPv2) {}
 
    IgmpV2Layer(IgmpType type, const IPv4Address& groupAddr = IPv4Address(), uint8_t maxResponseTime = 0)
        : IgmpLayer(type, groupAddr, maxResponseTime, IGMPv2) {}
 
    ~IgmpV2Layer() {}
 
 
    // implement abstract methods
 
    void computeCalculateFields();
};
 
 
class IgmpV3QueryLayer : public IgmpLayer
{
public:
 
    IgmpV3QueryLayer(uint8_t* data, size_t dataLen, Layer* prevLayer, Packet* packet);
 
    IgmpV3QueryLayer(const IPv4Address& multicastAddr = IPv4Address(), uint8_t maxResponseTime = 0, uint8_t s_qrv = 0);
 
    igmpv3_query_header* getIgmpV3QueryHeader() const { return (igmpv3_query_header*)m_Data; }
 
    uint16_t getSourceAddressCount() const;
 
    IPv4Address getSourceAddressAtIndex(int index) const;
 
    bool addSourceAddress(const IPv4Address& addr);
 
    bool addSourceAddressAtIndex(const IPv4Address& addr, int index);
 
    bool removeSourceAddressAtIndex(int index);
 
    bool removeAllSourceAddresses();
 
    // implement abstract methods
 
    void computeCalculateFields();
 
    size_t getHeaderLen() const;
};
 
 
class IgmpV3ReportLayer : public IgmpLayer
{
private:
    igmpv3_group_record* addGroupRecordAt(uint8_t recordType, const IPv4Address& multicastAddress, const std::vector<IPv4Address>& sourceAddresses, int offset);
 
public:
 
    IgmpV3ReportLayer(uint8_t* data, size_t dataLen, Layer* prevLayer, Packet* packet)
        : IgmpLayer(data, dataLen, prevLayer, packet, IGMPv3) {}
 
    IgmpV3ReportLayer() : IgmpLayer(IgmpType_MembershipReportV3, IPv4Address(), 0, IGMPv3) {}
 
    igmpv3_report_header* getReportHeader() const { return (igmpv3_report_header*)m_Data; }
 
    uint16_t getGroupRecordCount() const;
 
    igmpv3_group_record* getFirstGroupRecord() const;
 
    igmpv3_group_record* getNextGroupRecord(igmpv3_group_record* groupRecord) const;
 
    igmpv3_group_record* addGroupRecord(uint8_t recordType, const IPv4Address& multicastAddress, const std::vector<IPv4Address>& sourceAddresses);
 
    igmpv3_group_record* addGroupRecordAtIndex(uint8_t recordType, const IPv4Address& multicastAddress, const std::vector<IPv4Address>& sourceAddresses, int index);
 
    bool removeGroupRecordAtIndex(int index);
 
    bool removeAllGroupRecords();
 
    // implement abstract methods
 
    void computeCalculateFields();
 
    size_t getHeaderLen() const { return m_DataLen; }
};
 
}
 
#endif // PACKETPP_IGMP_LAYER