dns_handler.h

Go to the documentation of this file.

/*
 * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
 */

#ifndef vnsw_agent_dns_handler_hpp
#define vnsw_agent_dns_handler_hpp

#include "pkt/proto_handler.h"
#include "vnc_cfg_types.h"
#include "bind/bind_util.h"
#include "bind/bind_resolver.h"

#define DEFAULT_DNS_TTL 120

typedef boost::asio::ip::udp boost_udp;

class AgentDnsXmppChannel;
class VmInterface;
class Timer;

class DnsHandler : public ProtoHandler {
public:
    typedef boost::function<void(const boost::system::error_code&,
                                boost_udp::resolver::iterator)> ResolveHandler;
    typedef std::vector<boost_udp::resolver *> ResolvList;
    static const uint32_t max_items_per_xmpp_msg = 20;

    struct QueryKey {
        QueryKey(const Interface *i, uint16_t x) : itf(i), xid(x) {}
        bool operator<(const QueryKey &rhs) const {
            if (itf != rhs.itf)
                return itf < rhs.itf;
            return xid < rhs.xid;
        }

        const Interface *itf;
        uint16_t xid;
    };

    enum Action {
        NONE,
        DNS_QUERY,
        DNS_UPDATE
    };

    DnsHandler(Agent *agent, boost::shared_ptr<PktInfo> info,
               boost::asio::io_context &io);
    virtual ~DnsHandler();
    bool Run();
    bool TimerExpiry(uint16_t xid);
    void DefaultDnsResolveHandler(const boost::system::error_code &error,
                                  boost_udp::resolver::iterator it,
                                  DnsItems::iterator item);

private:
    friend class DnsTest;

    bool HandleRequest();
    bool HandleDefaultDnsRequest(const VmInterface *vmitf);
    void DefaultDnsSendResponse();
    bool HandleVirtualDnsRequest(const VmInterface *vmitf);
    bool ResolveLinkLocalRequest(DnsItems::iterator &item,
                                 DnsItems *linklocal_items) const;
    bool ResolveAllLinkLocalRequests();
    void BuildDnsResolvers();
    void BuildDefaultDnsResolvers();
    bool HandleMessage();
    bool HandleDefaultDnsResponse();
    bool HandleBindResponse();
    bool HandleUpdateResponse();
    bool HandleRetryExpiry();
    bool HandleUpdate();
    bool HandleModifyVdns();
    bool UpdateAll();
    void SendXmppUpdate(AgentDnsXmppChannel *channel, DnsUpdateData *xmpp_data);
    void ParseQuery();
    void Resolve(dns_flags flags, const DnsItems &ques, DnsItems &ans,
                 DnsItems &auth, DnsItems &add);
    void SendDnsResponse();
    void UpdateQueryNames();
    void UpdateOffsets(DnsItem &item, bool name_update_required);
    void UpdateGWAddress(DnsItem &item);
    void Update(InterTaskMsg *msg);
    void DelUpdate(InterTaskMsg *msg);
    void UpdateStats();
    void GetDomainName(const VmInterface *vm_itf, std::string *domain_name) const;
    bool GetDomainNameFromDhcp(std::vector<autogen::DhcpOptionType> &options,
                               std::string *domain_name) const;
    void GetBaseName(const std::string &name, std::string *base) const;
    std::string DnsItemsToString(DnsItems &items) const;
    void HandleInvalidBindResponse(DnsHandler *handler, dns_flags flags,
                                   const DnsItems &ques, DnsItems &ans,
                                   DnsItems &auth, DnsItems &add, uint16_t xid);
    bool NeedRetryForNextServer(uint16_t code);
    bool SendToDefaultServer();
    bool DefaultMethodInUse() { return default_method_; }
    uint8_t curr_index() { return curr_index_; }
    uint8_t last_index() { return def_dns_resolvers_.size() - 1; }
    void IncrCurrIndex() { curr_index_++; }

    dnshdr  *dns_;
    uint8_t *resp_ptr_;
    uint16_t dns_resp_size_;
    uint16_t xid_;
    Action action_;
    QueryKey *rkey_;
    struct DnsResolverInfo {
        boost::asio::ip::udp::endpoint ep_;
        uint32_t retries_;
        Timer *timer_;
    };
    std::vector<DnsResolverInfo *> dns_resolvers_;
    std::vector<DnsResolverInfo *> def_dns_resolvers_;
    std::string ipam_name_;
    std::string domain_name_;
    autogen::IpamType ipam_type_;
    autogen::VirtualDnsType vdns_type_;
    DnsItems items_;
    DnsItems linklocal_items_;
    DnsNameEncoder name_encoder_;
    bool query_name_update_;
    uint16_t query_name_update_len_;   // num bytes added in the query section
    uint16_t pend_req_;
    ResolvList resolv_list_;
    tbb::mutex mutex_;
    bool default_method_;
    uint8_t curr_index_;
    bool SendDnsQuery(DnsResolverInfo *resolver, uint16_t xid);

    DISALLOW_COPY_AND_ASSIGN(DnsHandler);
};

#endif // vnsw_agent_dns_handler_hpp