RawSocketDevice.h

Go to the documentation of this file.

#pragma once
 
 
#include <memory>
#include "IpAddress.h"
#include "Device.h"
 
namespace pcpp
{
    class RawSocketDevice : public IDevice
    {
    public:
        enum RecvPacketResult
        {
            RecvSuccess = 0,
            RecvTimeout = 1,
            RecvWouldBlock = 2,
            RecvError = 3
        };
 
        explicit RawSocketDevice(const IPAddress& interfaceIP);
 
        ~RawSocketDevice();
 
        RecvPacketResult receivePacket(RawPacket& rawPacket, bool blocking = true, double timeout = -1);
 
        int receivePackets(RawPacketVector& packetVec, double timeout, int& failedRecv);
 
        bool sendPacket(const RawPacket* rawPacket);
 
        int sendPackets(const RawPacketVector& packetVec);
 
        // overridden methods
 
        bool open() override;
 
        void close() override;
 
        bool isOpened() const override
        {
            return m_DeviceOpened;
        }
 
    private:
        struct SocketContainer;
 
        enum SocketFamily
        {
            Ethernet = 0,
            IPv4 = 1,
            IPv6 = 2
        };
 
        bool m_DeviceOpened = false;
 
        SocketFamily m_SockFamily;
        std::unique_ptr<SocketContainer> m_Socket;
        IPAddress m_InterfaceIP;
 
        RecvPacketResult getError(int& errorCode) const;
    };
}  // namespace pcpp