From 81b39e4b9aa5a8b2cd694e828cbac71b40bfddaf Mon Sep 17 00:00:00 2001
From: Bert Hubert <bert.hubert@netherlabs.nl>
Date: Sat, 8 May 2010 22:13:13 +0000
Subject: [PATCH] add in-place reordering infrastructure

git-svn-id: svn://svn.powerdns.com/pdns/trunk/pdns@1604 d19b8d6e-7fed-0310-83ef-9ca221ded41b
---
 pdns/backends/gsql/gsqlbackend.cc | 14 ++++++++++++++
 pdns/backends/gsql/gsqlbackend.hh |  2 ++
 pdns/dnsbackend.hh                |  5 +++++
 pdns/pdnssec.cc                   | 30 ++++++++++++++++++++----------
 4 files changed, 41 insertions(+), 10 deletions(-)

diff --git a/pdns/backends/gsql/gsqlbackend.cc b/pdns/backends/gsql/gsqlbackend.cc
index d672d000f..350532228 100644
--- a/pdns/backends/gsql/gsqlbackend.cc
+++ b/pdns/backends/gsql/gsqlbackend.cc
@@ -243,6 +243,20 @@ GSQLBackend::GSQLBackend(const string &mode, const string &suffix)
   d_setOrderAuthQuery = getArg("set-order-and-auth-query");
 }
 
+bool GSQLBackend::updateDNSSECOrderAndAuth(uint32_t domain_id, const std::string& zonename, const std::string& qname, bool auth)
+{
+  char output[1024];
+  // ordername='%s',auth=%d where name='%s' and domain_id='%d'
+  
+  string ins=toLower(labelReverse(makeRelative(qname, zonename)));
+  snprintf(output, sizeof(output)-1, d_setOrderAuthQuery.c_str(), sqlEscape(ins).c_str(), auth, sqlEscape(qname).c_str(), domain_id);
+  cerr<<"sql: '"<<output<<"'\n";
+  
+  d_db->doCommand(output);
+
+  return true;
+}
+
 bool GSQLBackend::getBeforeAndAfterNames(uint32_t id, const std::string& zonename, const std::string& qname, std::string& before, std::string& after)
 {
   cerr<<"gsql before/after called for id="<<id<<", qname="<<qname<<endl;
diff --git a/pdns/backends/gsql/gsqlbackend.hh b/pdns/backends/gsql/gsqlbackend.hh
index 8d6d1c000..55b6836cf 100644
--- a/pdns/backends/gsql/gsqlbackend.hh
+++ b/pdns/backends/gsql/gsqlbackend.hh
@@ -40,6 +40,8 @@ public:
   bool getDomainInfo(const string &domain, DomainInfo &di);
   void setNotified(uint32_t domain_id, uint32_t serial);
   bool getBeforeAndAfterNames(uint32_t id, const std::string& zname, const std::string& qname, std::string& before, std::string& after);
+  bool updateDNSSECOrderAndAuth(uint32_t domain_id, const std::string& zonename, const std::string& qname, bool auth);
+
 private:
   string d_qname;
   QType d_qtype;
diff --git a/pdns/dnsbackend.hh b/pdns/dnsbackend.hh
index e046ae10d..edebb3d4d 100644
--- a/pdns/dnsbackend.hh
+++ b/pdns/dnsbackend.hh
@@ -80,6 +80,11 @@ public:
     return false;
   }
 
+  virtual bool updateDNSSECOrderAndAuth(uint32_t domain_id, const std::string& zonename, const std::string& qname, bool auth)
+  {
+    return false;
+  }
+
   //! Initiates a list of the specified domain
   /** Once initiated, DNSResourceRecord objects can be retrieved using get(). Should return false
       if the backend does not consider itself responsible for the id passed.
diff --git a/pdns/pdnssec.cc b/pdns/pdnssec.cc
index dd5955551..97f05aa79 100644
--- a/pdns/pdnssec.cc
+++ b/pdns/pdnssec.cc
@@ -88,24 +88,30 @@ void loadMainConfig()
   UeberBackend::go();
 }
 
-void listZones()
+void orderZone(const std::string& zone)
 {
   loadMainConfig();
-  cerr<<"hier1 launcheable: "<<BackendMakers().numLauncheable()<<endl;
-  
-  cerr<<"new"<<endl;
+    
   UeberBackend* B = new UeberBackend("default");
   SOAData sd;
-  cerr<<"hier2 "<<(void*)B<<endl;
-  if(!B->getSOA("powerdnssec.org", sd)) {
+  
+  if(!B->getSOA(zone, sd)) {
     cerr<<"No SOA!"<<endl;
   } 
   cerr<<"ID: "<<sd.domain_id<<endl;
-  sd.db->list("powerdnssec.org", sd.domain_id);
+  sd.db->list(zone, sd.domain_id);
   DNSResourceRecord rr;
+
+  set<string> qnames;
   
   while(sd.db->get(rr)) {
-    cerr<<rr.qname<<endl;
+  //  cerr<<rr.qname<<endl;
+    qnames.insert(rr.qname);
+  }
+  
+  BOOST_FOREACH(const string& qname, qnames)
+  {
+    sd.db->updateDNSSECOrderAndAuth(sd.domain_id, zone, qname, true);
   }
   cerr<<"Done listing"<<endl;
 }
@@ -139,8 +145,12 @@ try
 
   DNSSECKeeper dk(g_vm["key-repository"].as<string>());
 
-  if(cmds[0] == "list-zones") {
-    listZones();
+  if(cmds[0] == "order-zone") {
+    if(cmds.size() != 2) {
+      cerr << "Error: "<<cmds[0]<<" takes exactly 1 parameter"<<endl;
+      return 0;
+    }
+    orderZone(cmds[1]);
   }
   else if(cmds[0] == "update-zone-keys") {
     if(cmds.size() != 2) {
-- 
2.50.1