DNSResourceRecord.hpp

Go to the documentation of this file.

#pragma once
 
#include "protocolCraft/NetworkType.hpp"
 
using namespace ProtocolCraft;
 
namespace Botcraft
{
    class DNSResourceRecord : public NetworkType
    {
    public:
        const std::vector<std::string>& GetNameLabels() const
        {
            return name_labels;
        }
 
        const unsigned short GetTypeCode() const
        {
            return type_code;
        }
 
        const unsigned short GetClassCode() const
        {
            return class_code;
        }
 
        const unsigned int GetTTL() const
        {
            return ttl;
        }
 
        const unsigned short GetRDLength() const
        {
            return rd_length;
        }
 
        const std::vector<unsigned char>& GetRData() const
        {
            return rdata;
        }
 
 
        void SetNameLabels(const std::vector<std::string>& name_labels_)
        {
            name_labels = name_labels_;
        }
 
        void SetTypeCode(const unsigned short type_code_)
        {
            type_code = type_code_;
        }
 
        void SetClassCode(const unsigned short class_code_)
        {
            class_code = class_code_;
        }
 
        void SetTTL(const unsigned int ttl_)
        {
            ttl = ttl_;
        }
 
        void SetRDLength(const unsigned short rd_length_)
        {
            rd_length = rd_length_;
        }
 
        void SetRData(const std::vector<unsigned char>& rdata_)
        {
            rdata = rdata_;
        }
 
    protected:
        virtual void ReadImpl(ReadIterator& iter, size_t& length) override
        {
            name_labels = std::vector<std::string>();
            bool done = false;
 
            while (!done)
            {
                unsigned char first_byte = ReadData<unsigned char>(iter, length);
                // Check if this is a pointer
                if (first_byte >> 6 == 0x03)
                {
                    unsigned char second_byte = ReadData<unsigned char>(iter, length);
                    // we actually don't need the value pointed in this application
                    name_labels.push_back("unresolved pointer");
                    done = true;
                }
                else if (first_byte == 0x00)
                {
                    done = true;
                }
                else
                {
                    name_labels.push_back(ReadRawString(iter, length, first_byte));
                }
            }
 
            type_code = ReadData<unsigned short>(iter, length);
            class_code = ReadData<unsigned short>(iter, length);
            ttl = ReadData<unsigned int>(iter, length);
            rd_length = ReadData<unsigned short>(iter, length);
            rdata = ReadByteArray(iter, length, rd_length);
        }
 
        virtual void WriteImpl(WriteContainer& container) const override
        {
            // We don't support compression for write (it shouldn't been called in this program anyway)
            for (int i = 0; i < name_labels.size(); ++i)
            {
                WriteData<unsigned char>(static_cast<unsigned char>(name_labels[i].size()), container);
                WriteRawString(name_labels[i], container);
            }
            WriteData<unsigned char>(0, container);
 
            WriteData<unsigned short>(type_code, container);
            WriteData<unsigned short>(class_code, container);
            WriteData<unsigned int>(ttl, container);
            WriteData<unsigned short>(rd_length, container);
            WriteByteArray(rdata, container);
        }
 
        virtual Json::Value SerializeImpl() const override
        {
            Json::Value output;
 
            std::string name = "";
            for (int i = 0; i < name_labels.size(); ++i)
            {
                name += name_labels[i] + ".";
            }
            output["identification"] = name;
            output["type_code"] = type_code;
            output["class_code"] = class_code;
            output["ttl"] = ttl;
            output["rd_length"] = rd_length;
            output["rdata"] = rdata;
 
            return output;
        }
 
    private:
        std::vector<std::string> name_labels;
        unsigned short type_code;
        unsigned short class_code;
        unsigned int ttl;
        unsigned short rd_length;
        std::vector<unsigned char> rdata;
    };
}