-
Notifications
You must be signed in to change notification settings - Fork 390
/
dns_mgr.h
119 lines (106 loc) · 4.5 KB
/
dns_mgr.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
/*
* Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
*/
#ifndef __dns_manager_h__
#define __dns_manager_h__
#include <tbb/mutex.h>
#include <mgr/dns_oper.h>
#include <bind/named_config.h>
#include <cfg/dns_config.h>
class DB;
class DBGraph;
struct VirtualDnsConfig;
struct VirtualDnsRecordConfig;
class DnsManager {
public:
static const int max_records_per_sandesh = 200;
static const uint32_t kPendingRecordRetransmitTime = 3000; // milliseconds
static const uint32_t kMaxRetransmitCount = 32;
struct PendingList {
uint16_t xid;
std::string view;
std::string zone;
DnsItems items;
BindUtil::Operation op;
uint32_t retransmit_count;
PendingList(uint16_t id, const std::string &v, const std::string &z,
const DnsItems &it, BindUtil::Operation o) {
// xid(id), view(v), zone(z), items(it), op(o), retransmit_count(0) {}
xid = id;
view = v;
zone = z;
items = it;
op = o;
retransmit_count = 0;
}
};
typedef std::map<uint16_t, PendingList> PendingListMap;
typedef std::pair<uint16_t, PendingList> PendingListPair;
DnsManager();
virtual ~DnsManager();
void Initialize(DB *config_db, DBGraph *config_graph,
const std::string& named_config_dir,
const std::string& named_config_file,
const std::string& named_log_file,
const std::string& rndc_config_file,
const std::string& rndc_secret,
const std::string& named_max_cache_size);
void Shutdown();
void DnsView(const DnsConfig *config, DnsConfig::DnsConfigEvent ev);
void DnsPtrZone(const Subnet &subnet, const VirtualDnsConfig *vdns,
DnsConfig::DnsConfigEvent ev);
void DnsRecord(const DnsConfig *config, DnsConfig::DnsConfigEvent ev);
void HandleUpdateResponse(uint8_t *pkt, std::size_t length);
DnsConfigManager &GetConfigManager() { return config_mgr_; }
void SendUpdate(BindUtil::Operation op, const std::string &view,
const std::string &zone, DnsItems &items);
void SendRetransmit(uint16_t xid, BindUtil::Operation op,
const std::string &view, const std::string &zone,
DnsItems &items);
void UpdateAll();
void BindEventHandler(BindStatus::Event ev);
template <typename ConfigType>
void ProcessConfig(IFMapNodeProxy *proxy, const std::string &name,
DnsConfigManager::EventType event);
void ProcessAgentUpdate(BindUtil::Operation event, const std::string &name,
const std::string &vdns_name, const DnsItem &item);
bool IsBindStatusUp() { return bind_status_.IsUp(); }
private:
friend class DnsBindTest;
friend class DnsManagerTest;
bool SendRecordUpdate(BindUtil::Operation op,
const VirtualDnsRecordConfig *config);
bool PendingDone(uint16_t xid);
void ResendRecord(uint16_t xid);
void ResendAllRecords();
void AddPendingList(uint16_t xid, const std::string &view,
const std::string &zone, const DnsItems &items,
BindUtil::Operation op);
void UpdatePendingList(const std::string &view,
const std::string &zone,
const DnsItems &items);
void DeletePendingList(uint16_t xid);
void ClearPendingList();
void PendingListViewDelete(const VirtualDnsConfig *config);
bool CheckZoneDelete(ZoneList &zones, PendingList &pend);
void PendingListZoneDelete(const Subnet &subnet,
const VirtualDnsConfig *config);
void StartPendingTimer();
void CancelPendingTimer();
bool PendingTimerExpiry();
void NotifyAllDnsRecords(const VirtualDnsConfig *config,
DnsConfig::DnsConfigEvent ev);
void NotifyReverseDnsRecords(const VirtualDnsConfig *config,
DnsConfig::DnsConfigEvent ev, bool notify);
inline uint16_t GetTransId();
inline bool CheckName(std::string rec_name, std::string name);
tbb::mutex mutex_;
BindStatus bind_status_;
DnsConfigManager config_mgr_;
static uint16_t g_trans_id_;
PendingListMap pending_map_;
Timer *pending_timer_;
WorkQueue<uint16_t> pending_done_queue_;
DISALLOW_COPY_AND_ASSIGN(DnsManager);
};
#endif // __dns_manager_h__