KomodiaDNS::CDNSQuery Class Reference

#include <DNSQuery.h>

Inheritance diagram for KomodiaDNS::CDNSQuery:

[legend]

Collaboration diagram for KomodiaDNS::CDNSQuery:

[legend]

List of all members.

Public Types
enum	_DnsQueryTypes {   A = 1, NS = 2, MD = 3, MF = 4,   CNAME = 5, SOA = 6, MB = 7, MG = 8,   MR = 9, NLL = 10, WKS = 11, PTR = 12,   HINFO = 13, MINFO = 14, MX = 15, TXT = 16,   ERR = 17 }
typedef enum KomodiaDNS::CDNSQuery::_DnsQueryTypes	DNSQueryTypes
Public Member Functions
const DnsQuestionHeader &	GetQuery (int iIndex) const
std::string	GetQueryString (int iIndex) const
int	GetQueryCount () const
DNSQueryTypes	GetFirstQueryType () const
int	Parse (const char *pBuffer, int iNumber)
unsigned short	GetQuestionCount () const
void	AddQuery (const std::string &rDomain, DNSQueryTypes aType)
unsigned short	GetSerializationSize () const
unsigned short	SerializeQuery (char *pBuffer) const
CDNSQuery &	operator= (const CDNSQuery &rQuery)
	CDNSQuery ()
	CDNSQuery (const CDNSQuery &rQuery)
virtual	~CDNSQuery ()
void	SetLocalLog (CErrorLog *pLog)
Static Public Member Functions
static std::string	GetCurrentDateTime ()
static std::string	ErrorCodeToString (DWORD dwErrorCode)
static void	SetLog (CErrorLog *pLog)
Protected Member Functions
const std::string &	GetClassName () const
void	WriteMessage (const std::string &rMethod, const std::string &rMessage, LogPriority aPriority=lpMessage)
void	WriteMessage (const std::string &rMethod, const std::string &rMessage, DWORD dwAdditionalData, LogPriority aPriority=lpMessage)
virtual void	ReportError (const std::string &rMethod) const
virtual void	ReportError (const std::string &rMethod, const std::string &rMessage) const
virtual void	ReportError (const std::string &rMethod, const std::string &rMessage, DWORD dwAdditionalData) const
virtual void	ReportError (const std::string &rMethod, int iErrorCode) const
virtual void	ReportError (const std::string &rMethod, const std::string &rMessage, int iErrorCode) const
virtual void	ReportErrorOS (const std::string &rMethod, const std::string &rMessage) const
void	SetName (const std::string &rName) const
Static Protected Member Functions
static void	WriteStaticMessage (const std::string &rClass, const std::string &rMethod, const std::string &rMessage, LogPriority aPriority=lpMessage)
static void	WriteStaticMessage (const std::string &rClass, const std::string &rMethod, const std::string &rMessage, DWORD dwAdditionalData, LogPriority aPriority=lpMessage)
static void	ReportStaticError (const std::string &rClass, const std::string &rMethod)
static void	ReportStaticError (const std::string &rClass, const std::string &rMethod, const std::string &rMessage)
static void	ReportStaticError (const std::string &rClass, const std::string &rMethod, const std::string &rMessage, DWORD dwAdditionalData)

---

Detailed Description

Definition at line 62 of file DNSQuery.h.

---

Member Typedef Documentation

typedef enum KomodiaDNS::CDNSQuery::_DnsQueryTypes KomodiaDNS::CDNSQuery::DNSQueryTypes

typedef enum CErrorHandler::_LogPriority CErrorHandler::LogPriority [inherited]

---

Member Enumeration Documentation

enum KomodiaDNS::CDNSQuery::_DnsQueryTypes

Enumerator:

A
NS
MD
MF
CNAME
SOA
MB
MG
MR
NLL
WKS
PTR
HINFO
MINFO
MX
TXT
ERR

Definition at line 65 of file DNSQuery.h.

    {
        A=1,        //a host address
        NS=2,       //an authoritative name server
        MD=3,       //a mail destination (Obsolete - use MX)
        MF=4,       //a mail forwarder (Obsolete - use MX)
        CNAME=5,    //the canonical name for an alias
        SOA=6,      //marks the start of a zone of authority
        MB=7,       //a mailbox domain name (EXPERIMENTAL)
        MG=8,       //a mail group member (EXPERIMENTAL)
        MR=9,       //a mail rename domain name (EXPERIMENTAL)
        NLL=10, //a null RR (EXPERIMENTAL)
        WKS=11,     //a well known service description
        PTR=12,     //a domain name pointer
        HINFO=13,   //host information
        MINFO=14,   //mailbox or mail list information
        MX=15,      //mail exchange
        TXT=16,     //text strings,
        ERR=17      //Internal error, not part of DNS
    } DNSQueryTypes;

enum CErrorHandler::_LogPriority [inherited]

Enumerator:

lpDebug
lpMessage
lpCritical
lpError

Definition at line 53 of file ErrorHandler.h.

    {
        lpDebug,
        lpMessage,
        lpCritical,
        lpError
    } LogPriority;

---

Constructor & Destructor Documentation

KomodiaDNS::CDNSQuery::CDNSQuery

(

)

Definition at line 55 of file DNSQuery.cpp.

                     : CErrorHandler()
{
    try
    {
        //Set our name
        SetName(CDNSQuery_Class);
    }
    ERROR_HANDLER("CDNSQuery")
}

KomodiaDNS::CDNSQuery::CDNSQuery

(

const CDNSQuery &

rQuery

)

Definition at line 65 of file DNSQuery.cpp.

                                            : CErrorHandler(rQuery),
                                                m_aQueries(rQuery.m_aQueries)
{
    try
    {
        //Set our name
        SetName(CDNSQuery_Class);
    }
    ERROR_HANDLER("CDNSQuery")
}

KomodiaDNS::CDNSQuery::~CDNSQuery

(

)

[virtual]

Definition at line 76 of file DNSQuery.cpp.

{
}

---

Member Function Documentation

void KomodiaDNS::CDNSQuery::AddQuery	(	const std::string &	rDomain,
		DNSQueryTypes	aType
	)

Definition at line 90 of file DNSQuery.cpp.

{
    try
    {
        //Construct the structure
        DnsQuestionHeader aData;

        //Start putting the data
        aData.sName=CDNSManager::ParseAddress(rDomain);
        aData.usType=htons(aType);
        aData.usClass=htons(DNS_CLASS_INTERNET);

        //Add to the vector
        m_aQueries.push_back(aData);
    }
    ERROR_HANDLER("AddQuery")
}

std::string CErrorHandler::ErrorCodeToString

(

DWORD

dwErrorCode

)

[static, inherited]

Definition at line 122 of file ErrorHandler.cpp.

{
    try
    {
        //Get the error string
        LPVOID lpMsgBuf;
        FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | 
                      FORMAT_MESSAGE_FROM_SYSTEM | 
                      FORMAT_MESSAGE_IGNORE_INSERTS,
                      NULL,
                      dwErrorCode,
                      MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
                      (LPTSTR) &lpMsgBuf,
                      0,
                      NULL);

        //Save it
        std::string sMessage;
        sMessage+=(char*)lpMsgBuf;
        
        //Release the buffer
        LocalFree(lpMsgBuf);

        //Done
        return sMessage;
    }
    catch (...)
    {
        return "Unknown";
    }
}

const std::string & CErrorHandler::GetClassName

(

)

const [protected, inherited]

Reimplemented in CSocketThreadManager.

Definition at line 516 of file ErrorHandler.cpp.

{
    return m_sClassName;
}

std::string CErrorHandler::GetCurrentDateTime

(

)

[static, inherited]

Definition at line 521 of file ErrorHandler.cpp.

{
    try
    {
        //Our string
        std::string sDate;

        //Our tmp buf
        char cTmp[128];

        //Get date
        _strdate(cTmp);
        sDate=cTmp;
        sDate+=' ';

        //Get time
        _strtime(cTmp);
        sDate+=cTmp;

        //Done
        return sDate;
    }
    ERROR_HANDLER_STATIC_RETURN(CErrorHandler_Class,"GetCurrentDateTime","")
}

CDNSQuery::DNSQueryTypes KomodiaDNS::CDNSQuery::GetFirstQueryType

(

)

const

Definition at line 259 of file DNSQuery.cpp.

{
    try
    {
        //Do we have the data
        if (m_aQueries.empty())
            return ERR;
        else
            return (DNSQueryTypes)m_aQueries.front().usType;
    }
    ERROR_HANDLER_RETURN("GetFirstQueryType",ERR)
}

const DnsQuestionHeader & KomodiaDNS::CDNSQuery::GetQuery

(

int

iIndex

)

const

Definition at line 277 of file DNSQuery.cpp.

{
    return m_aQueries[iIndex];
}

int KomodiaDNS::CDNSQuery::GetQueryCount

(

)

const

Definition at line 272 of file DNSQuery.cpp.

{
    return m_aQueries.size();
}

std::string KomodiaDNS::CDNSQuery::GetQueryString

(

int

iIndex

)

const

Definition at line 282 of file DNSQuery.cpp.

{
    try
    {
        //Get the string
        std::string sName;
        sName=m_aQueries[iIndex].sName;

        //Convert it
        return CDNSParser::ConvertName(sName);
    }
    ERROR_HANDLER_RETURN("GetQueryString","")
}

unsigned short KomodiaDNS::CDNSQuery::GetQuestionCount

(

)

const

Definition at line 109 of file DNSQuery.cpp.

{
    return m_aQueries.size();
}

unsigned short KomodiaDNS::CDNSQuery::GetSerializationSize

(

)

const

Definition at line 178 of file DNSQuery.cpp.

{
    try
    {
        //Our size counter
        unsigned short usSize;
        usSize=0;

        //Iterate the vector
        QueryVector::const_iterator aIterator;
        aIterator=m_aQueries.begin();

        //Start the loop
        while (aIterator!=m_aQueries.end())
        {
            //Calculate the struct size
            usSize+=aIterator->sName.length()+1;
            usSize+=sizeof(unsigned short)*2;

            //Next data
            ++aIterator;
        }

        //Done
        return usSize;
    }
    ERROR_HANDLER_RETURN("GetSerializationSize",0)
}

CDNSQuery & KomodiaDNS::CDNSQuery::operator=

(

const CDNSQuery &

rQuery

)

Definition at line 80 of file DNSQuery.cpp.

{
    //Copy the vetor
    QueryVector aTmp(rQuery.m_aQueries);
    m_aQueries.swap(aTmp);

    //Done
    return *this;
}

int KomodiaDNS::CDNSQuery::Parse	(	const char *	pBuffer,
		int	iNumber
	)

Definition at line 114 of file DNSQuery.cpp.

{
    try
    {
        //Our backup buffer for iterations
        const char* pBackupBuffer;
        pBackupBuffer=pBuffer;

        //Start to iterate the data
        for (int iCount=0;iCount<iNumber;++iCount)
        {
            //Try to parse the name
            std::string sParsedName;

            //The parsed length
            unsigned short usLength;

            //Parse it
            sParsedName=CDNSParser::ParseName(pBuffer,
                                              pBuffer,
                                              usLength);

            //Check the length
            if (!usLength)
            {
                //Report it
                ReportError("Parse","Received zero length!");

                //Exit
                return NULL;
            }

            //Add it to our header
            DnsQuestionHeader aData;

            //Set the string name
            aData.sName=sParsedName;

            //Increase backup length
            pBackupBuffer+=usLength;

            //Get the rest of the data
            memcpy(&aData.usType,
                   pBackupBuffer,
                   DnsQuestionCopyHeader);

            //Reverse the two filds
            aData.usClass=htons(aData.usClass);
            aData.usType=htons(aData.usType);

            //Add it to the question vector
            m_aQueries.push_back(aData);

            //Increase the buffer
            pBackupBuffer+=DnsQuestionCopyHeader;
        }

        //Done
        return pBackupBuffer-pBuffer;
    }
    ERROR_HANDLER_RETURN("Parse",0)
}

void CErrorHandler::ReportError	(	const std::string &	rMethod,
		const std::string &	rMessage,
		int	iErrorCode
	)			const [protected, virtual, inherited]

Definition at line 154 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Get the log
        CErrorLog* pLog;
        pLog=GetLog();

        //Convert the error code
        char aTmp[11];
        sprintf(aTmp,"%d",iErrorCode);

        //Get the string for it
        std::string sError;
        sError=rMessage;
        sError+=", and Socket error: ";
        sError+=aTmp;
        sError+=", ";
        sError+=ErrorCodeToString(iErrorCode);

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          sError);
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportError	(	const std::string &	rMethod,
		int	iErrorCode
	)			const [protected, virtual, inherited]

Definition at line 187 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Get the log
        CErrorLog* pLog;
        pLog=GetLog();

        //Convert the error code
        char aTmp[11];
        sprintf(aTmp,"%d",iErrorCode);

        //Get the string for it
        std::string sError;
        sError="Socket error: ";
        sError+=aTmp;
        sError+=", ";
        sError+=ErrorCodeToString(iErrorCode);

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          sError);
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportError	(	const std::string &	rMethod,
		const std::string &	rMessage,
		DWORD	dwAdditionalData
	)			const [protected, virtual, inherited]

Definition at line 275 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Get the log
        CErrorLog* pLog;
        pLog=GetLog();

        //Convert the number
        char aTmp[11];
        ltoa(dwAdditionalData,aTmp,10);

        //Create the new message
        std::string sNewMessage(rMessage);
        sNewMessage+="Additional data: ";
        sNewMessage+=aTmp;

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          sNewMessage);
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportError	(	const std::string &	rMethod,
		const std::string &	rMessage
	)			const [protected, virtual, inherited]

Definition at line 218 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        CErrorLog* pLog;
        pLog=GetLog();

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          rMessage);
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportError

(

const std::string &

rMethod

)

const [protected, virtual, inherited]

Definition at line 311 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
#ifdef WIN32
        //Get the last error
        DWORD dwLastError;
        dwLastError=GetLastError();

        //Report the error
        GetLog()->ReportError(m_sClassName,
                              rMethod,
                              ErrorCodeToString(dwLastError));

#else
        GetLog()->ReportCatchError(m_sClassName,
                                   rMethod,
                                   "Unknown error!");
#endif
    }
    ERROR_UNKNOWN("ReportError")
}

void CErrorHandler::ReportErrorOS	(	const std::string &	rMethod,
		const std::string &	rMessage
	)			const [protected, virtual, inherited]

Definition at line 237 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Get the last error
        DWORD dwLastError;
        dwLastError=GetLastError();

        //Format the message
        std::string sMessage;
        sMessage=rMessage;
        sMessage+=", with error code: ";

        //Convert the error code
        char aTmp[11];
        itoa(dwLastError,aTmp,10);

        //Add it again
        sMessage+=aTmp;
        sMessage+=" ";
        sMessage+=ErrorCodeToString(dwLastError);
        
        //Get the log
        CErrorLog* pLog;
        pLog=GetLog();

        //Report to the log
        pLog->ReportError(m_sClassName,
                          rMethod,
                          sMessage);
    }
    ERROR_UNKNOWN("ReportErrorOS")
}

void CErrorHandler::ReportStaticError	(	const std::string &	rClass,
		const std::string &	rMethod,
		const std::string &	rMessage,
		DWORD	dwAdditionalData
	)			[static, protected, inherited]

Definition at line 489 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
        //Convert the number
        char aTmp[11];
        ltoa(dwAdditionalData,aTmp,10);

        //Create the new message
        std::string sNewMessage(rMessage);
        sNewMessage+="Additional data: ";
        sNewMessage+=aTmp;

        //Report to the log
        m_pLog->ReportError(rClass,
                            rMethod,
                            sNewMessage);
    }
    ERROR_UNKNOWN("ReportStaticError")
}

void CErrorHandler::ReportStaticError	(	const std::string &	rClass,
		const std::string &	rMethod,
		const std::string &	rMessage
	)			[static, protected, inherited]

Definition at line 472 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
        //Report to the log
        m_pLog->ReportError(rClass,
                            rMethod,
                            rMessage);
    }
    ERROR_UNKNOWN("ReportStaticError")
}

void CErrorHandler::ReportStaticError	(	const std::string &	rClass,
		const std::string &	rMethod
	)			[static, protected, inherited]

Definition at line 446 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
#ifdef WIN32
        //Get the last error
        DWORD dwLastError;
        dwLastError=GetLastError();

        //Report the error
        m_pLog->ReportError(rClass,
                            rMethod,
                            ErrorCodeToString(dwLastError));
#else
        m_pLog->ReportError(rClass,
                            rMethod,
                            "Unknown error!");
#endif
    }
    ERROR_UNKNOWN("ReportStaticError")
}

unsigned short KomodiaDNS::CDNSQuery::SerializeQuery

(

char *

pBuffer

)

const

Definition at line 207 of file DNSQuery.cpp.

{
    try
    {
        //Our size counter
        unsigned short usSize;
        usSize=0;

        //Iterate the vector
        QueryVector::const_iterator aIterator;
        aIterator=m_aQueries.begin();

        //Start the loop
        while (aIterator!=m_aQueries.end())
        {
            //Calculate the struct size
            usSize+=aIterator->sName.length()+1;
            usSize+=sizeof(unsigned short)*2;

            //Copy the string
            memcpy(pBuffer,
                   aIterator->sName.c_str(),
                   aIterator->sName.length());
            pBuffer+=aIterator->sName.length();

            //Terminating null
            *pBuffer=0;
            pBuffer++;

            //Copy the data
            //Type
            memcpy(pBuffer,
                   &aIterator->usType,
                   sizeof(aIterator->usType));
            pBuffer+=sizeof(unsigned short);

            //Class
            memcpy(pBuffer,
                   &aIterator->usClass,
                   sizeof(aIterator->usClass));
            pBuffer+=sizeof(unsigned short);

            //Next data
            ++aIterator;
        }

        //Done
        return usSize;
    }
    ERROR_HANDLER_RETURN("SerializeQuery",0)
}

void CErrorHandler::SetLocalLog

(

CErrorLog *

pLog

)

[inherited]

Definition at line 441 of file ErrorHandler.cpp.

{
    m_pLocalLog=pLog;
}

void CErrorHandler::SetLog

(

CErrorLog *

pLog

)

[static, inherited]

Definition at line 305 of file ErrorHandler.cpp.

{
    //Save the new log
    m_pLog=pLog;
}

void CErrorHandler::SetName

(

const std::string &

rName

)

const [protected, inherited]

Definition at line 116 of file ErrorHandler.cpp.

{
    //Save the class name
    m_sClassName=rName;
}

void CErrorHandler::WriteMessage	(	const std::string &	rMethod,
		const std::string &	rMessage,
		DWORD	dwAdditionalData,
		LogPriority	aPriority = lpMessage
	)			[protected, inherited]

Definition at line 355 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Convert the number
        char aTmp[11];
        ltoa(dwAdditionalData,aTmp,10);

        //Create the new message
        std::string sNewMessage(rMessage);
        sNewMessage+="Additional data: ";
        sNewMessage+=aTmp;

        //Delegate the call
        GetLog()->WriteMessage(m_sClassName,
                               rMethod,
                               sNewMessage,
                               aPriority);
    }
    ERROR_UNKNOWN("WriteMessage")
}

void CErrorHandler::WriteMessage	(	const std::string &	rMethod,
		const std::string &	rMessage,
		LogPriority	aPriority = lpMessage
	)			[protected, inherited]

Definition at line 337 of file ErrorHandler.cpp.

{
    if (!GetLog())
        return;

    try
    {
        //Delegate the call
        GetLog()->WriteMessage(m_sClassName,
                               rMethod,
                               rMessage,
                               aPriority);
    }
    ERROR_UNKNOWN("WriteMessage")
}

void CErrorHandler::WriteStaticMessage	(	const std::string &	rClass,
		const std::string &	rMethod,
		const std::string &	rMessage,
		DWORD	dwAdditionalData,
		LogPriority	aPriority = lpMessage
	)			[static, protected, inherited]

Definition at line 403 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
        //Convert the number
        char aTmp[11];
        ltoa(dwAdditionalData,aTmp,10);

        //Create the new message
        std::string sNewMessage(rMessage);
        sNewMessage+="Additional data: ";
        sNewMessage+=aTmp;

        //Delegate the call
        m_pLog->WriteMessage(rClass,
                             rMethod,
                             sNewMessage,
                             aPriority);
    }
    ERROR_UNKNOWN("WriteStaticMessage")
}

void CErrorHandler::WriteStaticMessage	(	const std::string &	rClass,
		const std::string &	rMethod,
		const std::string &	rMessage,
		LogPriority	aPriority = lpMessage
	)			[static, protected, inherited]

Definition at line 383 of file ErrorHandler.cpp.

{
    if (!m_pLog)
        return;

    try
    {
        //Delegate the call
        m_pLog->WriteMessage(rClass,
                             rMethod,
                             rMessage,
                             aPriority);
    }
    ERROR_UNKNOWN("WriteStaticMessage")
}

---

The documentation for this class was generated from the following files:

• DNSQuery.h
• DNSQuery.cpp