1 /* 2 * This file is part of PowerDNS or dnsdist. 3 * Copyright -- PowerDNS.COM B.V. and its contributors 4 * 5 * This program is free software; you can redistribute it and/or modify 6 * it under the terms of version 2 of the GNU General Public License as 7 * published by the Free Software Foundation. 8 * 9 * In addition, for the avoidance of any doubt, permission is granted to 10 * link this program with OpenSSL and to (re)distribute the binaries 11 * produced as the result of such linking. 12 * 13 * This program is distributed in the hope that it will be useful, 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16 * GNU General Public License for more details. 17 * 18 * You should have received a copy of the GNU General Public License 19 * along with this program; if not, write to the Free Software 20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. 21 */ 22 #pragma once 23 #include <string> 24 #include <cstdio> 25 #include <stdexcept> 26 #include <stack> 27 #include <deque> 28 29 #include "namespaces.hh" 30 31 class ZoneParserTNG 32 { 33 public: 34 ZoneParserTNG(const string& fname, DNSName zname=g_rootdnsname, string reldir="", bool upgradeContent=false); 35 ZoneParserTNG(const vector<string>& zonedata, DNSName zname, bool upgradeContent=false); 36 37 ~ZoneParserTNG(); 38 bool get(DNSResourceRecord& rr, std::string* comment=0); 39 typedef runtime_error exception; 40 typedef std::deque<pair<string::size_type, string::size_type> > parts_t; 41 DNSName getZoneName(); 42 string getLineOfFile(); // for error reporting purposes 43 pair<string,int> getLineNumAndFile(); // idem disableGenerate()44 void disableGenerate() 45 { 46 d_generateEnabled = false; 47 } setMaxGenerateSteps(size_t max)48 void setMaxGenerateSteps(size_t max) 49 { 50 d_maxGenerateSteps = max; 51 } 52 private: 53 bool getLine(); 54 bool getTemplateLine(); 55 void stackFile(const std::string& fname); 56 unsigned makeTTLFromZone(const std::string& str); 57 58 struct filestate { filestateZoneParserTNG::filestate59 filestate(FILE* fp, string filename) : d_fp(fp), d_filename(filename), d_lineno(0){} 60 FILE *d_fp; 61 string d_filename; 62 int d_lineno; 63 }; 64 65 parts_t d_parts; 66 string d_reldir; 67 string d_line; 68 DNSName d_prevqname; 69 DNSName d_zonename; 70 string d_templateline; 71 vector<string> d_zonedata; 72 vector<string>::iterator d_zonedataline; 73 std::stack<filestate> d_filestates; 74 parts_t d_templateparts; 75 size_t d_maxGenerateSteps{0}; 76 int d_defaultttl; 77 uint32_t d_templatecounter, d_templatestop, d_templatestep; 78 bool d_havedollarttl; 79 bool d_fromfile; 80 bool d_generateEnabled{true}; 81 bool d_upgradeContent; 82 }; 83