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