From 8439c4e3ca924026ef5e566e40773935e6f67c7f Mon Sep 17 00:00:00 2001
From: bert hubert <bert.hubert@netherlabs.nl>
Date: Wed, 23 Apr 2014 14:59:56 +0200
Subject: [PATCH] fix how we emit 'disabled' for 't' and 'f' SQL, plus
 implement feature to supress duplicate SOAs and turn it on by default

---
 pdns/zone2sql.cc | 28 +++++++++++++++++++++-------
 1 file changed, 21 insertions(+), 7 deletions(-)

diff --git a/pdns/zone2sql.cc b/pdns/zone2sql.cc
index 15d183210..87dd76d35 100644
--- a/pdns/zone2sql.cc
+++ b/pdns/zone2sql.cc
@@ -208,16 +208,16 @@ static void emitRecord(const string& zoneName, const string &qname, const string
       cout<<"insert into records (domain_id, name,type,content,ttl,prio,disabled) select id ,"<<
         sqlstr(toLower(stripDot(qname)))<<", "<<
         sqlstr(qtype)<<", "<<
-        sqlstr(stripDotContent(content))<<", "<<ttl<<", "<<prio<<", "<< (disabled ? 't': 'f')<<
-        " from domains where name="<<toLower(sqlstr(zoneName))<<";\n";
+        sqlstr(stripDotContent(content))<<", "<<ttl<<", "<<prio<<", '"<< (disabled ? 't': 'f')<<
+        "' from domains where name="<<toLower(sqlstr(zoneName))<<";\n";
     } else
     {
       cout<<"insert into records (domain_id, name, ordername, auth, type,content,ttl,prio,disabled) select id ,"<<
         sqlstr(toLower(stripDot(qname)))<<", "<<
         sqlstr(toLower(labelReverse(makeRelative(stripDot(qname), zoneName))))<<", '"<< (auth  ? 't' : 'f') <<"', "<<
         sqlstr(qtype)<<", "<<
-        sqlstr(stripDotContent(content))<<", "<<ttl<<", "<<prio<<", "<<(disabled ? 't': 'f') <<
-        " from domains where name="<<toLower(sqlstr(zoneName))<<";\n";
+        sqlstr(stripDotContent(content))<<", "<<ttl<<", "<<prio<<", '"<<(disabled ? 't': 'f') <<
+        "' from domains where name="<<toLower(sqlstr(zoneName))<<";\n";
     }
   }
   else if(g_mode==GORACLE) {
@@ -295,6 +295,7 @@ try
     ::arg().setSwitch("slave","Keep BIND slaves as slaves. Only works with named-conf.")="no";
     ::arg().setSwitch("transactions","If target SQL supports it, use transactions")="no";
     ::arg().setSwitch("on-error-resume-next","Continue after errors")="no";
+    ::arg().setSwitch("filter-duplicate-soa","Filter second SOA in zone")="yes";
     ::arg().set("zone","Zonefile to parse")="";
     ::arg().set("zone-name","Specify an $ORIGIN in case it is not present")="";
     ::arg().set("named-conf","Bind 8/9 named.conf to parse")="";
@@ -319,6 +320,9 @@ try
       exit(1);
     }
   
+    bool filterDupSOA = ::arg().mustDo("filter-duplicate-soa");
+      
+
     if(::arg().mustDo("gmysql")) 
       g_mode=MYSQL;
     else if(::arg().mustDo("gpgsql"))
@@ -382,8 +386,14 @@ try
             
             ZoneParserTNG zpt(i->filename, i->name, BP.getDirectory());
             DNSResourceRecord rr;
-            while(zpt.get(rr)) 
+	    bool seenSOA=false;
+            while(zpt.get(rr)) {
+	      if(filterDupSOA && seenSOA && rr.qtype.getCode() == QType::SOA)
+		continue;
+	      if(rr.qtype.getCode() == QType::SOA)
+		seenSOA=true;
               emitRecord(i->name, rr.qname, rr.qtype.getName(), rr.content, rr.ttl, rr.priority);
+	    }
             num_domainsdone++;
           }
           catch(std::exception &ae) {
@@ -412,9 +422,13 @@ try
       startNewTransaction();
       emitDomain(zonename);
       string comment;
-      
+      bool seenSOA=false;
       while(zpt.get(rr, &comment))  {
-	
+	if(filterDupSOA && seenSOA && rr.qtype.getCode() == QType::SOA)
+	  continue;
+	if(rr.qtype.getCode() == QType::SOA)
+	  seenSOA=true;
+
         emitRecord(zonename, rr.qname, rr.qtype.getName(), rr.content, rr.ttl, rr.priority, comment);
       }
       num_domainsdone=1;
-- 
2.40.0