Added PTR record data

src/modules/dns-base/CMakeLists.txt

@@ -30,7 +30,7 @@ add_library(${PROJECT_NAME}
         src/RecordData/QDnsRecordDataRaw.cpp
         includes/dns-base/RecordData/QDnsRecordDataRaw.h
         src/RecordData/QDnsRecordDataAAAA.cpp
-        includes/dns-base/RecordData/QDnsRecordDataAAAA.h src/RecordData/QDnsRecordDataMX.cpp includes/dns-base/RecordData/QDnsRecordDataMX.h src/RecordData/QDnsRecordDataSOA.cpp includes/dns-base/RecordData/QDnsRecordDataSOA.h)
+        includes/dns-base/RecordData/QDnsRecordDataAAAA.h src/RecordData/QDnsRecordDataMX.cpp includes/dns-base/RecordData/QDnsRecordDataMX.h src/RecordData/QDnsRecordDataSOA.cpp includes/dns-base/RecordData/QDnsRecordDataSOA.h src/RecordData/QDnsRecordDataPTR.cpp includes/dns-base/RecordData/QDnsRecordDataPTR.h)
 
 target_link_libraries(${PROJECT_NAME} Qt5::Core Qt5::Network)
 

src/modules/dns-base/includes/dns-base/RecordData/QDnsRecordDataPTR.h

@@ -0,0 +1,46 @@
+//
+// Created by robin on 1/23/18.
+//
+
+#ifndef QDNSAGG_QDNSRECORDDATAPTR_H
+#define QDNSAGG_QDNSRECORDDATAPTR_H
+
+#include <qplatformdefs.h>
+#include <QByteArray>
+#include <QDataStream>
+#include <QHostAddress>
+
+#include "QDnsRecordData.h"
+
+namespace QDns
+{
+    namespace Base
+    {
+        class QDnsRecordDataPTR : public QDnsRecordData
+        {
+        public:
+            QDnsRecordDataPTR();
+
+            QDnsRecordDataPTR(const QByteArray &packetData, quint32 pos);
+
+            QDnsRecordDataPTR(const QByteArray &packetData, QDataStream &dataStream);
+
+            QByteArray serialize() const override;
+
+            void serialize(QDataStream &dataStream) const override;
+
+            virtual QDnsRecordData* clone() const override;
+
+            const QString &getHostname() const;
+
+            void setHostname(const QString &hostname);
+
+        private:
+            void parse(const QByteArray &packetData, QDataStream &dataStream);
+
+            QString m_hostname;
+        };
+    }
+}
+
+#endif //QDNSAGG_QDNSRECORDDATAPTR_H

src/modules/dns-base/src/RecordData/QDnsRecordDataPTR.cpp

@@ -0,0 +1,78 @@
+//
+// Created by robin on 1/23/18.
+//
+
+#include "dns-base/QDnsLabel.h"
+#include "dns-base/RecordData/QDnsRecordDataPTR.h"
+
+namespace QDns
+{
+    namespace Base
+    {
+        QDnsRecordDataPTR::QDnsRecordDataPTR()
+                : QDnsRecordData()
+        {
+        }
+
+        QDnsRecordDataPTR::QDnsRecordDataPTR(const QByteArray &packetData, quint32 pos)
+                : QDnsRecordDataPTR()
+        {
+            QDataStream dataStream(packetData);
+            dataStream.skipRawData(pos);
+            parse(packetData, dataStream);
+        }
+
+        QDnsRecordDataPTR::QDnsRecordDataPTR(const QByteArray &packetData, QDataStream &dataStream)
+                : QDnsRecordDataPTR()
+        {
+            parse(packetData, dataStream);
+        }
+
+        void QDnsRecordDataPTR::parse(const QByteArray &packetData, QDataStream &dataStream)
+        {
+            if (dataStream.device()->bytesAvailable() < 1)
+            {
+                m_isValid = false;
+            }
+            else
+            {
+                m_hostname = QDnsLabel::parse(packetData, dataStream);
+                if (m_hostname.isNull())
+                {
+                    m_isValid = false;
+                }
+            }
+        }
+
+        QByteArray QDnsRecordDataPTR::serialize() const
+        {
+            QByteArray byteArray;
+            QDataStream dataStream(&byteArray, QIODevice::WriteOnly);
+            serialize(dataStream);
+            return byteArray;
+        }
+
+        void QDnsRecordDataPTR::serialize(QDataStream &dataStream) const
+        {
+            QDnsLabel::serialize(m_hostname, dataStream);
+        }
+
+        QDnsRecordData *QDnsRecordDataPTR::clone() const
+        {
+            auto data = new QDnsRecordDataPTR();
+            data->m_isValid = m_isValid;
+            data->m_hostname = m_hostname;
+            return data;
+        }
+
+        const QString &QDnsRecordDataPTR::getHostname() const
+        {
+            return m_hostname;
+        }
+
+        void QDnsRecordDataPTR::setHostname(const QString &hostname)
+        {
+            m_hostname = hostname;
+        }
+    }
+}

src/modules/dns-base/tests/CMakeLists.txt

@@ -11,7 +11,8 @@ add_executable(${PROJECT_NAME}
         RecordData/DnsRDataA.cpp
         RecordData/DnsRDataAAAA.cpp
         RecordData/DnsRDataMX.cpp
-        RecordData/DnsRDataSOA.cpp)
+        RecordData/DnsRDataSOA.cpp
+        RecordData/DnsRDataPTR.cpp)
 
 target_link_libraries(${PROJECT_NAME} gtest pthread qdnsagg-dns-base)
 

src/modules/dns-base/tests/RecordData/DnsRDataPTR.cpp

@@ -0,0 +1,45 @@
+//
+// Created by robin on 1/14/18.
+//
+
+#include <gtest/gtest.h>
+#include <QByteArray>
+#include <QString>
+#include <QtNetwork/QHostAddress>
+#include <dns-base/RecordData/QDnsRecordDataPTR.h>
+
+struct DnsRDataPTRTestParams
+{
+    bool isValid;
+    const char* hexData;
+    quint32 pos;
+
+    QString hostname;
+};
+
+class DnsRDataPTRTest : public ::testing::TestWithParam<DnsRDataPTRTestParams>
+{
+};
+
+TEST_P(DnsRDataPTRTest, parseAndGetters)
+{
+    auto params = GetParam();
+    auto data = QByteArray::fromHex(params.hexData);
+    auto rdata = QDns::Base::QDnsRecordDataPTR(data, params.pos);
+    ASSERT_EQ(rdata.isValid(), params.isValid);
+
+    if (rdata.isValid())
+    {
+        ASSERT_EQ(params.hostname, rdata.getHostname());
+
+        ASSERT_EQ(rdata.serialize().toHex(), QString(params.hexData).remove(" "));
+    }
+}
+
+INSTANTIATE_TEST_CASE_P(DnsRDataPTRTestInst,
+                        DnsRDataPTRTest,
+                        ::testing::Values(
+                                DnsRDataPTRTestParams {false, "", 0},
+                                DnsRDataPTRTestParams {true, "00", 0, ""},
+                                DnsRDataPTRTestParams {true, "03777777067274686f6e6903636f6d00", 0, "www.rthoni.com"}
+                        ));