d2_update_message.h

Go to the documentation of this file.

// Copyright (C) 2013-2021 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#ifndef D2_UPDATE_MESSAGE_H
#define D2_UPDATE_MESSAGE_H

#include <d2srv/d2_zone.h>
#include <dns/message.h>
#include <dns/name.h>
#include <dns/rcode.h>
#include <dns/rrclass.h>
#include <dns/rrset.h>
#include <dns/tsig.h>

#include <map>

namespace isc {
namespace d2 {

class InvalidZoneSection : public Exception {
public:
    InvalidZoneSection(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) {}
};

class InvalidQRFlag : public Exception {
public:
    InvalidQRFlag(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) {}
};

class NotUpdateMessage : public Exception {
public:
    NotUpdateMessage(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) {}
};

class TSIGVerifyError : public Exception {
public:
    TSIGVerifyError(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) {}
};


class D2UpdateMessage;

typedef boost::shared_ptr<D2UpdateMessage> D2UpdateMessagePtr;

class D2UpdateMessage {
public:

    enum Direction {
        INBOUND,
        OUTBOUND
    };

    enum QRFlag {
        REQUEST,
        RESPONSE
    };

    enum UpdateMsgSection {
        SECTION_ZONE,
        SECTION_PREREQUISITE,
        SECTION_UPDATE,
        SECTION_ADDITIONAL
    };

public:
    D2UpdateMessage(const Direction direction = OUTBOUND);


private:
    D2UpdateMessage(const D2UpdateMessage& source);
    D2UpdateMessage& operator=(const D2UpdateMessage& source);

public:

    QRFlag getQRFlag() const;

    uint16_t getId() const;

    void setId(const uint16_t id);

    const dns::Rcode& getRcode() const;

    void setRcode(const dns::Rcode& rcode);

    unsigned int getRRCount(const UpdateMsgSection section) const;


    const dns::RRsetIterator beginSection(const UpdateMsgSection section) const;

    const dns::RRsetIterator endSection(const UpdateMsgSection section) const;

    void setZone(const dns::Name& zone, const dns::RRClass& rrclass);

    D2ZonePtr getZone() const;

    void addRRset(const UpdateMsgSection section, const dns::RRsetPtr& rrset);


    void toWire(dns::AbstractMessageRenderer& renderer,
                dns::TSIGContext* const tsig_ctx = NULL);

    void fromWire(const void* received_data, size_t bytes_received,
                  dns::TSIGContext* const tsig_context = NULL);

private:
    static
    dns::Message::Section ddnsToDnsSection(const UpdateMsgSection section);

    void validateResponse() const;

    dns::Message message_;

    D2ZonePtr zone_;

};

} // namespace d2
} // namespace isc

#endif // D2_UPDATE_MESSAGE_H