82 lines
2.1 KiB
C++
82 lines
2.1 KiB
C++
#include "dns-storage.hh"
|
|
using namespace std;
|
|
|
|
const DNSNode* DNSNode::find(dnsname& name, dnsname& last, bool* passedZonecut) const
|
|
{
|
|
cout<<"find for '"<<name<<"', last is now '"<<last<<"'"<<endl;
|
|
if(!last.empty() && passedZonecut && rrsets.count(DNSType::NS)) {
|
|
*passedZonecut=true;
|
|
}
|
|
|
|
if(name.empty()) {
|
|
cout<<"Empty lookup, returning this node or 0"<<endl;
|
|
if(!zone && rrsets.empty()) // only root zone can have this
|
|
return 0;
|
|
else
|
|
return this;
|
|
}
|
|
cout<<"Children at this node: ";
|
|
for(const auto& c: children) cout <<"'"<<c.first<<"' ";
|
|
cout<<endl;
|
|
auto iter = children.find(name.back());
|
|
cout<<"Looked for child called '"<<name.back()<<"'"<<endl;
|
|
if(iter == children.end()) {
|
|
cout<<"Found nothing, trying wildcard"<<endl;
|
|
iter = children.find("*");
|
|
if(iter == children.end()) {
|
|
cout<<"Still nothing, returning leaf"<<endl;
|
|
return this;
|
|
}
|
|
else {
|
|
cout<<"Had wildcard match, following"<<endl;
|
|
}
|
|
}
|
|
cout<<"Had match, continuing to child '"<<iter->first<<"'"<<endl;
|
|
last.push_front(name.back());
|
|
name.pop_back();
|
|
return iter->second.find(name, last, passedZonecut);
|
|
}
|
|
|
|
DNSNode* DNSNode::add(dnsname name)
|
|
{
|
|
cout<<"Add for '"<<name<<"'"<<endl;
|
|
if(name.size() == 1) {
|
|
cout<<"Last label, adding "<<name.front()<<endl;
|
|
return &children[name.front()];
|
|
}
|
|
|
|
auto back = name.back();
|
|
name.pop_back();
|
|
auto iter = children.find(back);
|
|
|
|
if(iter == children.end()) {
|
|
cout<<"Inserting new child for "<<back<<endl;
|
|
return children[back].add(name);
|
|
}
|
|
return iter->second.add(name);
|
|
}
|
|
|
|
dnsname operator+(const dnsname& a, const dnsname& b)
|
|
{
|
|
dnsname ret=a;
|
|
for(const auto& l : b.d_name)
|
|
ret.d_name.push_back(l);
|
|
return ret;
|
|
}
|
|
|
|
void DNSNode::visit(std::function<void(const dnsname& name, const DNSNode*)> visitor, dnsname name) const
|
|
{
|
|
visitor(name, this);
|
|
for(const auto& c : children)
|
|
c.second.visit(visitor, dnsname{c.first}+name);
|
|
}
|
|
|
|
// this should perform escaping rules!
|
|
std::ostream & operator<<(std::ostream &os, const dnsname& d)
|
|
{
|
|
for(const auto& l : d.d_name) {
|
|
os<<l<<".";
|
|
}
|
|
return os;
|
|
}
|