rrset.h

Go to the documentation of this file.

// Copyright (C) 2010-2015 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 RRSET_H
#define RRSET_H 1

#include <iostream>
#include <string>

#include <boost/shared_ptr.hpp>

#include <dns/exceptions.h>

#include <dns/rdata.h>
#include <dns/rrtype.h>

namespace isc {
namespace util {
class OututBuffer;
}

namespace dns {

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

// forward declarations
class Name;
class RRType;
class RRClass;
class RRTTL;
class AbstractMessageRenderer;
class AbstractRRset;
class BasicRRset;
class RdataIterator;
class BasicRRsetImpl;
class RRset;

typedef boost::shared_ptr<AbstractRRset> RRsetPtr;

typedef boost::shared_ptr<const AbstractRRset> ConstRRsetPtr;

typedef boost::shared_ptr<RdataIterator> RdataIteratorPtr;


class AbstractRRset {

private:
    AbstractRRset(const AbstractRRset& source);
    AbstractRRset& operator=(const AbstractRRset& source);
protected:
    AbstractRRset() {}
public:
    virtual ~AbstractRRset() {}


    virtual unsigned int getRdataCount() const = 0;

    virtual uint16_t getLength() const = 0;

    virtual const Name& getName() const = 0;

    virtual const RRClass& getClass() const = 0;

    virtual const RRType& getType() const = 0;

    virtual const RRTTL& getTTL() const = 0;

    virtual void setTTL(const RRTTL& ttl) = 0;


    virtual std::string toText() const = 0;

    virtual unsigned int toWire(AbstractMessageRenderer& renderer) const = 0;

    virtual unsigned int toWire(isc::util::OutputBuffer& buffer) const = 0;


    virtual void addRdata(rdata::ConstRdataPtr rdata) = 0;

    virtual void addRdata(const rdata::Rdata& rdata) = 0;

    virtual void addRdata(const std::string& rdata_str) = 0;

    virtual RdataIteratorPtr getRdataIterator() const = 0;


    virtual RRsetPtr getRRsig() const = 0;

    virtual unsigned int getRRsigDataCount() const = 0;

    virtual void addRRsig(const rdata::ConstRdataPtr& rdata) = 0;

    virtual void addRRsig(const rdata::RdataPtr& rdata) = 0;

    virtual void addRRsig(const AbstractRRset& sigs) = 0;

    virtual void addRRsig(const ConstRRsetPtr& sigs) = 0;

    virtual void addRRsig(const RRsetPtr& sigs) = 0;

    virtual void removeRRsig() = 0;

    virtual bool isSameKind(const AbstractRRset& other) const;

};

class RdataIterator {

protected:
    RdataIterator() {}
public:
    virtual ~RdataIterator() {}
private:
    RdataIterator(const RdataIterator& source);
    RdataIterator& operator=(const RdataIterator& source);

public:
    virtual void first() = 0;

    virtual void next() = 0;

    virtual const rdata::Rdata& getCurrent() const = 0;

    virtual bool isLast() const = 0;
};

class BasicRRset : public AbstractRRset {

private:
    BasicRRset(const BasicRRset& source);
    BasicRRset& operator=(const BasicRRset& source);
public:
    BasicRRset(const Name& name, const RRClass& rrclass,
               const RRType& rrtype, const RRTTL& ttl);
    virtual ~BasicRRset();


    virtual unsigned int getRdataCount() const;

    virtual uint16_t getLength() const;

    virtual const Name& getName() const;

    virtual const RRClass& getClass() const;

    virtual const RRType& getType() const;

    virtual const RRTTL& getTTL() const;

    virtual void setTTL(const RRTTL& ttl);


    virtual std::string toText() const;

    virtual unsigned int toWire(AbstractMessageRenderer& renderer) const;

    virtual unsigned int toWire(isc::util::OutputBuffer& buffer) const;


    virtual void addRdata(rdata::ConstRdataPtr rdata);

    virtual void addRdata(const rdata::Rdata& rdata);

    virtual void addRdata(const std::string& rdata_str);

    virtual RdataIteratorPtr getRdataIterator() const;


    virtual RRsetPtr getRRsig() const {
        return (RRsetPtr());
    }

    virtual unsigned int getRRsigDataCount() const {
        return (0);
    }

    virtual void addRRsig(const rdata::ConstRdataPtr&) {
        isc_throw(NotImplemented,
                  "BasicRRset does not implement the addRRsig() method");
    }

    virtual void addRRsig(const rdata::RdataPtr&) {
        isc_throw(NotImplemented,
                  "BasicRRset does not implement the addRRsig() method");
    }

    virtual void addRRsig(const AbstractRRset&) {
        isc_throw(NotImplemented,
                  "BasicRRset does not implement the addRRsig() method");
    }

    virtual void addRRsig(const ConstRRsetPtr&) {
        isc_throw(NotImplemented,
                  "BasicRRset does not implement the addRRsig() method");
    }

    virtual void addRRsig(const RRsetPtr&) {
        isc_throw(NotImplemented,
                  "BasicRRset does not implement the addRRsig() method");
    }

    virtual void removeRRsig() {
        isc_throw(NotImplemented,
                  "BasicRRset does not implement the removeRRsig() method");
    }
private:
    BasicRRsetImpl* impl_;
};

class RRset : public BasicRRset {
public:
    RRset(const Name& name, const RRClass& rrclass,
          const RRType& rrtype, const RRTTL& ttl);

    virtual ~RRset();

    virtual uint16_t getLength() const;

    virtual unsigned int toWire(AbstractMessageRenderer& renderer) const;

    virtual unsigned int toWire(isc::util::OutputBuffer& buffer) const;

    virtual void setTTL(const RRTTL& ttl) {
        BasicRRset::setTTL(ttl);
        if (rrsig_) {
            rrsig_->setTTL(ttl);
        }
    }

    virtual void addRRsig(const rdata::ConstRdataPtr& rdata) {
        if (!rrsig_) {
            rrsig_ = RRsetPtr(new RRset(getName(), getClass(),
                                        RRType::RRSIG(), getTTL()));
        }
        rrsig_->addRdata(rdata);
    }

    // Workaround for older versions of boost: some don't support implicit
    // conversion from shared_ptr<X> to shared_ptr<const X>.  Note: we should
    // revisit the interface of managing RRset signatures, at which point this
    // problem may go away.
    virtual void addRRsig(const rdata::RdataPtr& rdata) {
        // Don't try to convert as a reference here.  SunStudio will reject it.
        addRRsig(static_cast<const rdata::ConstRdataPtr>(rdata));
    }

    virtual void addRRsig(const AbstractRRset& sigs) {
        RdataIteratorPtr it = sigs.getRdataIterator();

        if (!rrsig_) {
            rrsig_ = RRsetPtr(new RRset(getName(), getClass(),
                                        RRType::RRSIG(), getTTL()));
        }

        for (it->first(); !it->isLast(); it->next()) {
            rrsig_->addRdata(it->getCurrent());
        }
    }

    virtual void addRRsig(const ConstRRsetPtr& sigs) { addRRsig(*sigs); }

    // Another workaround for older boost (see above)
    virtual void addRRsig(const RRsetPtr& sigs) { addRRsig(*sigs); }

    virtual void removeRRsig() { rrsig_ = RRsetPtr(); }

    virtual RRsetPtr getRRsig() const { return (rrsig_); }

    virtual unsigned int getRRsigDataCount() const;

private:
    RRsetPtr rrsig_;
};


std::ostream& operator<<(std::ostream& os, const AbstractRRset& rrset);
} // end of namespace dns
} // end of namespace isc
#endif  // RRSET_H

// Local Variables: 
// mode: c++
// End: