parser.hh

Go to the documentation of this file.

#ifndef SPONGE_LIBSPONGE_PARSER_HH
#define SPONGE_LIBSPONGE_PARSER_HH
 
#include "buffer.hh"
 
#include <cstdint>
#include <cstdlib>
#include <string>
#include <utility>
 
enum class ParseResult {
    NoError = 0,      
    BadChecksum,      
    PacketTooShort,   
    WrongIPVersion,   
    HeaderTooShort,   
    TruncatedPacket,  
    Unsupported       
};
 
std::string as_string(const ParseResult r);
 
class NetParser {
  private:
    Buffer _buffer;
    ParseResult _error = ParseResult::NoError;  
 
    void _check_size(const size_t size);
 
    template <typename T>
    T _parse_int();
 
  public:
    NetParser(Buffer buffer) : _buffer(buffer) {}
 
    Buffer buffer() const { return _buffer; }
 
    ParseResult get_error() const { return _error; }
 
    void set_error(ParseResult res) { _error = res; }
 
    bool error() const { return get_error() != ParseResult::NoError; }
 
    uint32_t u32();
 
    uint16_t u16();
 
    uint8_t u8();
 
    void remove_prefix(const size_t n);
};
 
struct NetUnparser {
    template <typename T>
    static void _unparse_int(std::string &s, T val);
 
    static void u32(std::string &s, const uint32_t val);
 
    static void u16(std::string &s, const uint16_t val);
 
    static void u8(std::string &s, const uint8_t val);
};
 
#endif  // SPONGE_LIBSPONGE_PARSER_HH