DwmDnsMessage.hh

Go to the documentation of this file.

//===========================================================================
// @(#) $DwmPath: dwm/DwmDns/tags/DwmDns-0.1.0/classes/include/DwmDnsMessage.hh 10136 $
// @(#) $Id: DwmDnsMessage.hh 10136 2018-01-28 06:01:09Z dwm $
//===========================================================================
//  Copyright (c) Daniel W. McRobb 2000, 2016, 2018
//  All rights reserved.
//
//  Redistribution and use in source and binary forms, with or without
//  modification, are permitted provided that the following conditions
//  are met:
//
//  1. Redistributions of source code must retain the above copyright
//     notice, this list of conditions and the following disclaimer.
//  2. Redistributions in binary form must reproduce the above copyright
//     notice, this list of conditions and the following disclaimer in the
//     documentation and/or other materials provided with the distribution.
//  3. The names of the authors and copyright holders may not be used to
//     endorse or promote products derived from this software without
//     specific prior written permission.
//
//  IN NO EVENT SHALL DANIEL W. MCROBB BE LIABLE TO ANY PARTY FOR
//  DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES,
//  INCLUDING LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE,
//  EVEN IF DANIEL W. MCROBB HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH
//  DAMAGE.
//
//  THE SOFTWARE PROVIDED HEREIN IS ON AN "AS IS" BASIS, AND
//  DANIEL W. MCROBB HAS NO OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT,
//  UPDATES, ENHANCEMENTS, OR MODIFICATIONS. DANIEL W. MCROBB MAKES NO
//  REPRESENTATIONS AND EXTENDS NO WARRANTIES OF ANY KIND, EITHER
//  IMPLIED OR EXPRESS, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
//  WARRANTIES OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE,
//  OR THAT THE USE OF THIS SOFTWARE WILL NOT INFRINGE ANY PATENT,
//  TRADEMARK OR OTHER RIGHTS.
//===========================================================================

//---------------------------------------------------------------------------
//---------------------------------------------------------------------------

#ifndef _DWMDNSMESSAGE_HH_
#define _DWMDNSMESSAGE_HH_

extern "C" {
  #include <unistd.h>
}

#include <cstdint>
#include <vector>

#include "DwmDnsMessageHeader.hh"
#include "DwmDnsMessageQuestion.hh"
#include "DwmDnsResourceRecord.hh"

namespace Dwm {

  namespace Dns {

    //------------------------------------------------------------------------
    //------------------------------------------------------------------------
    
    //------------------------------------------------------------------------
    //------------------------------------------------------------------------
    class Message
    {
    public:
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      Message();
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      bool operator == (const Message & message) const;
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      void Clear();
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const MessageHeader & Header() const
      {
        return _header;
      }
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      MessageHeader & Header()
      {
        return _header;
      }

      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const std::vector<MessageQuestion> & Questions() const
      {
        return _questions;
      }

      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      std::vector<MessageQuestion> & Questions()
      {
        return _questions;
      }
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const std::vector<ResourceRecord> & Answers() const
      {
        return _answers;
      }
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      std::vector<ResourceRecord> & Answers()
      {
        return _answers;
      }
        
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const std::vector<ResourceRecord> & Authorities() const
      {
        return _authorities;
      }
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      std::vector<ResourceRecord> & Authorities()
      {
        return _authorities;
      }
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const std::vector<ResourceRecord> & Additional() const
      {
        return _additional;
      }
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      std::vector<ResourceRecord> & Additional()
      {
        return _additional;
      }
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      uint8_t *Encode(uint8_t *pkt, uint8_t *ptr, uint16_t pktlen) const;
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      const uint8_t *Decode(const uint8_t *pkt, const uint8_t *ptr,
                            uint16_t pktlen);

      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      ssize_t SendTo(int sockFd, int flags, const struct sockaddr *to,
                     int tolen) const;

      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      ssize_t Write(int fd) const;
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      ssize_t RecvFrom(int sockFd, int flags, sockaddr *from,
                       socklen_t *fromlen);

      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      ssize_t Read(int fd);
      
      //----------------------------------------------------------------------
      //----------------------------------------------------------------------
      void EnableEDNS(uint16_t pduSize, bool doDnssec);
      
    private:
      mutable MessageHeader         _header;
      std::vector<MessageQuestion>  _questions;
      std::vector<ResourceRecord>   _answers;
      std::vector<ResourceRecord>   _authorities;
      std::vector<ResourceRecord>   _additional;
    };
    
  }  // namespace Dns

}  // namespace Dwm

#endif  // _DWMDNSMESSAGE_HH_