PacketExtractor.h

Go to the documentation of this file.

 
#pragma once
#include "IoTypes.h"
#include "Packet.h"
#include <stddef.h>
 
namespace catapult { namespace ionet {
 
#define PACKET_EXTRACT_RESULT_LIST \
    /* A packet was succesfully extracted. */ \
    ENUM_VALUE(Success) \
    \
    /* A packet was not extracted due to insufficient data. */ \
    ENUM_VALUE(Insufficient_Data) \
    \
    /* A packet was not extracted due to a packet error. */ \
    ENUM_VALUE(Packet_Error)
 
#define ENUM_VALUE(LABEL) LABEL,
    enum class PacketExtractResult {
        PACKET_EXTRACT_RESULT_LIST
    };
#undef ENUM_VALUE
 
    std::ostream& operator<<(std::ostream& out, PacketExtractResult value);
 
    class PacketExtractor {
    public:
        PacketExtractor(ByteBuffer& data, size_t maxPacketDataSize);
 
    public:
        PacketExtractResult tryExtractNextPacket(const Packet*& pExtractedPacket);
 
        void consume();
 
    private:
        ByteBuffer& m_data;
        size_t m_maxPacketDataSize;
        size_t m_consumedBytes;
    };
}}