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();

    uint16_t getSourceAdressCount();

    IPv4Address getSoruceAddressAtIndex(int index);

    size_t getRecordLen();
};


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, NULL, 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);
public:

    virtual ~IgmpLayer() {}

    inline igmp_header* getIgmpHeader() { return (igmp_header*)m_Data; }

    inline IPv4Address getGroupAddress() { return IPv4Address(getIgmpHeader()->groupAddress); }

    void setGroupAddress(const IPv4Address& groupAddr);

    IgmpType getType();

    void setType(IgmpType type);

    static ProtocolType getIGMPVerFromData(uint8_t* data, size_t dataLen, bool& isQuery);


    // implement abstract methods

    void parseNextLayer() {}

    inline size_t getHeaderLen() { return sizeof(igmp_header); }

    std::string toString();

    OsiModelLayer getOsiModelLayer() { return OsiModelNetworkLayer; }
};


class IgmpV1Layer : public IgmpLayer
{
public:
    IgmpV1Layer(uint8_t* data, size_t dataLen, Layer* prevLayer, Packet* packet);

    IgmpV1Layer(IgmpType type, const IPv4Address& groupAddr = IPv4Address::Zero);

    ~IgmpV1Layer() {}


    // implement abstract methods

    void computeCalculateFields();

};


class IgmpV2Layer : public IgmpLayer
{
public:
    IgmpV2Layer(uint8_t* data, size_t dataLen, Layer* prevLayer, Packet* packet);

    IgmpV2Layer(IgmpType type, const IPv4Address& groupAddr = IPv4Address::Zero, uint8_t maxResponseTime = 0);

    ~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::Zero, uint8_t maxResponseTime = 0, uint8_t s_qrv = 0);

    inline igmpv3_query_header* getIgmpV3QueryHeader() { return (igmpv3_query_header*)m_Data; }

    uint16_t getSourceAddressCount();

    IPv4Address getSourceAddressAtIndex(int index);

    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();
};


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);

    IgmpV3ReportLayer();

    inline igmpv3_report_header* getReportHeader() { return (igmpv3_report_header*)m_Data; }

    uint16_t getGroupRecordCount();

    igmpv3_group_record* getFirstGroupRecord();

    igmpv3_group_record* getNextGroupRecord(igmpv3_group_record* groupRecord);

    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();
};

}

#endif // PACKETPP_IGMP_LAYER