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