6675548676

Committed 28 Oct 2023 07:19AM UTC coverage: 56.095% (-1.4%) from 57.482%

Build # 6675548676

Build Type

Pull #13226

github

Committed by

web-flow

Commit Message

Merge e90063b7e into f8c2901a6

Pull Request Pull Request #13226: Cleanup `ueberbackend.cc` a bit

Run Details

30191 of 84938 branches covered (0.0%)

Branch coverage included in aggregate %.

181 of 236 new or added lines in 1 file covered. (76.69%)

2287 existing lines in 47 files now uncovered.

103158 of 152781 relevant lines covered (67.52%)

2599861.42 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

2.74

/modules/pipebackend/pipebackend.cc

/*
 * This file is part of PowerDNS or dnsdist.
 * Copyright -- PowerDNS.COM B.V. and its contributors
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of version 2 of the GNU General Public License as
 * published by the Free Software Foundation.
 *
 * In addition, for the avoidance of any doubt, permission is granted to
 * link this program with OpenSSL and to (re)distribute the binaries
 * produced as the result of such linking.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <string>
#include <map>
#include <unistd.h>
#include <stdlib.h>
#include <sstream>
#include "coprocess.hh"

#include "pdns/namespaces.hh"

#include "pdns/dns.hh"
#include "pdns/dnsbackend.hh"
#include "pdns/dnspacket.hh"
#include "pdns/pdnsexception.hh"
#include "pdns/logger.hh"
#include "pdns/arguments.hh"
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "pipebackend.hh"

static const char* kBackendId = "[PIPEBackend]";

CoWrapper::CoWrapper(const string& command, int timeout, int abiVersion)
{
  d_command = command;
  d_timeout = timeout;
  d_abiVersion = abiVersion;
  launch(); // let exceptions fall through - if initial launch fails, we want to die
  // I think
}

CoWrapper::~CoWrapper()
{
}

void CoWrapper::launch()
{
  if (d_cp)
    return;

  if (d_command.empty())
    throw ArgException("pipe-command is not specified");

  if (isUnixSocket(d_command)) {
    d_cp = std::make_unique<UnixRemote>(d_command);
  }
  else {
    auto coprocess = std::make_unique<CoProcess>(d_command, d_timeout);
    coprocess->launch();
    d_cp = std::move(coprocess);
  }

  d_cp->send("HELO\t" + std::to_string(d_abiVersion));
  string banner;
  d_cp->receive(banner);
  g_log << Logger::Error << "Backend launched with banner: " << banner << endl;
}

void CoWrapper::send(const string& line)
{
  launch();
  try {
    d_cp->send(line);
    return;
  }
  catch (PDNSException& ae) {
    d_cp.reset();
    throw;
  }
}
void CoWrapper::receive(string& line)
{
  launch();
  try {
    d_cp->receive(line);
    return;
  }
  catch (PDNSException& ae) {
    g_log << Logger::Warning << kBackendId << " Unable to receive data from coprocess. " << ae.reason << endl;
    d_cp.reset();
    throw;
  }
}

PipeBackend::PipeBackend(const string& suffix)
{
  d_disavow = false;
  signal(SIGCHLD, SIG_IGN);
  setArgPrefix("pipe" + suffix);
  try {
    launch();
  }
  catch (const ArgException& A) {
    g_log << Logger::Error << kBackendId << " Unable to launch, fatal argument error: " << A.reason << endl;
    throw;
  }
  catch (...) {
    throw;
  }
}

void PipeBackend::launch()
{
  if (d_coproc)
    return;

  try {
    if (!getArg("regex").empty()) {
      d_regex = std::make_unique<Regex>(getArg("regex"));
    }
    d_regexstr = getArg("regex");
    d_abiVersion = getArgAsNum("abi-version");
    d_coproc = std::make_unique<CoWrapper>(getArg("command"), getArgAsNum("timeout"), getArgAsNum("abi-version"));
  }

  catch (const ArgException& A) {
    cleanup();
    throw;
  }
}

/*
 * Cleans up the co-process wrapper
 */
void PipeBackend::cleanup()
{
  d_coproc.reset(nullptr);
  d_regex.reset();
  d_regexstr = string();
  d_abiVersion = 0;
}

void PipeBackend::lookup(const QType& qtype, const DNSName& qname, int zoneId, DNSPacket* pkt_p)
{
  try {
    launch();
    d_disavow = false;
    if (d_regex && !d_regex->match(qname.toStringRootDot())) {
      if (::arg().mustDo("query-logging"))
        g_log << Logger::Error << "Query for '" << qname << "' failed regex '" << d_regexstr << "'" << endl;
      d_disavow = true; // don't pass to backend
    }
    else {
      ostringstream query;
      string localIP = "0.0.0.0";
      string remoteIP = "0.0.0.0";
      Netmask realRemote("0.0.0.0/0");
      if (pkt_p) {
        localIP = pkt_p->getLocal().toString();
        realRemote = pkt_p->getRealRemote();
        remoteIP = pkt_p->getInnerRemote().toString();
      }
      // abi-version = 1
      // type    qname           qclass  qtype   id      remote-ip-address
      query << "Q\t" << qname.toStringRootDot() << "\tIN\t" << qtype.toString() << "\t" << zoneId << "\t" << remoteIP;

      // add the local-ip-address if abi-version is set to 2
      if (d_abiVersion >= 2)
        query << "\t" << localIP;
      if (d_abiVersion >= 3)
        query << "\t" << realRemote.toString();

      if (::arg().mustDo("query-logging"))
        g_log << Logger::Error << "Query: '" << query.str() << "'" << endl;
      d_coproc->send(query.str());
    }
  }
  catch (PDNSException& pe) {
    g_log << Logger::Error << kBackendId << " Error from coprocess: " << pe.reason << endl;
    d_disavow = true;
  }
  d_qtype = qtype;
  d_qname = qname;
}

bool PipeBackend::list(const DNSName& target, int inZoneId, bool /* include_disabled */)
{
  try {
    launch();
    d_disavow = false;
    ostringstream query;
    // The question format:

    // type    qname           qclass  qtype   id      ip-address
    if (d_abiVersion >= 4)
      query << "AXFR\t" << inZoneId << "\t" << target.toStringRootDot();
    else
      query << "AXFR\t" << inZoneId;

    d_coproc->send(query.str());
  }
  catch (PDNSException& ae) {
    g_log << Logger::Error << kBackendId << " Error from coprocess: " << ae.reason << endl;
  }
  d_qname = DNSName(std::to_string(inZoneId)); // why do we store a number here??
  return true;
}

string PipeBackend::directBackendCmd(const string& query)
{
  if (d_abiVersion < 5)
    return "not supported on ABI version " + std::to_string(d_abiVersion) + " (use ABI version 5 or later)\n";

  try {
    launch();
    ostringstream oss;
    oss << "CMD\t" << query;
    d_coproc->send(oss.str());
  }
  catch (PDNSException& ae) {
    g_log << Logger::Error << kBackendId << " Error from coprocess: " << ae.reason << endl;
    cleanup();
  }

  ostringstream oss;
  while (true) {
    string line;
    d_coproc->receive(line);
    if (line == "END")
      break;
    oss << line << std::endl;
  };

  return oss.str();
}

//! For the dynamic loader
DNSBackend* PipeBackend::maker()
{
  try {
    return new PipeBackend();
  }
  catch (...) {
    g_log << Logger::Error << kBackendId << " Unable to instantiate a pipebackend!" << endl;
    return nullptr;
  }
}

PipeBackend::~PipeBackend()
{
  cleanup();
}

bool PipeBackend::get(DNSResourceRecord& r)
{
  if (d_disavow) // this query has been blocked
    return false;

  string line;

  // The answer format:
  // DATA    qname           qclass  qtype   ttl     id      content
  unsigned int extraFields = 0;
  if (d_abiVersion >= 3)
    extraFields = 2;

  try {
    launch();
    for (;;) {
      d_coproc->receive(line);
      vector<string> parts;
      stringtok(parts, line, "\t");
      if (parts.empty()) {
        g_log << Logger::Error << kBackendId << " Coprocess returned empty line in query for " << d_qname << endl;
        throw PDNSException("Format error communicating with coprocess");
      }
      else if (parts[0] == "FAIL") {
        throw DBException("coprocess returned a FAIL");
      }
      else if (parts[0] == "END") {
        return false;
      }
      else if (parts[0] == "LOG") {
        g_log << Logger::Error << "Coprocess: " << line.substr(4) << endl;
        continue;
      }
      else if (parts[0] == "DATA") { // yay
        if (parts.size() < 7 + extraFields) {
          g_log << Logger::Error << kBackendId << " Coprocess returned incomplete or empty line in data section for query for " << d_qname << endl;
          throw PDNSException("Format error communicating with coprocess in data section");
          // now what?
        }

        if (d_abiVersion >= 3) {
          r.scopeMask = std::stoi(parts[1]);
          r.auth = (parts[2] == "1");
        }
        else {
          r.scopeMask = 0;
          r.auth = true;
        }
        r.qname = DNSName(parts[1 + extraFields]);
        r.qtype = parts[3 + extraFields];
        pdns::checked_stoi_into(r.ttl, parts[4 + extraFields]);
        pdns::checked_stoi_into(r.domain_id, parts[5 + extraFields]);

        if (r.qtype.getCode() != QType::MX && r.qtype.getCode() != QType::SRV) {
          r.content.clear();
          for (unsigned int n = 6 + extraFields; n < parts.size(); ++n) {
            if (n != 6 + extraFields)
              r.content.append(1, ' ');
            r.content.append(parts[n]);
          }
        }
        else {
          if (parts.size() < 8 + extraFields) {
            g_log << Logger::Error << kBackendId << " Coprocess returned incomplete MX/SRV line in data section for query for " << d_qname << endl;
            throw PDNSException("Format error communicating with coprocess in data section of MX/SRV record");
          }

          r.content = parts[6 + extraFields] + " " + parts[7 + extraFields];
        }
        break;
      }
      else
        throw PDNSException("Coprocess backend sent incorrect response '" + line + "'");
    }
  }
  catch (DBException& dbe) {
    g_log << Logger::Error << kBackendId << " " << dbe.reason << endl;
    throw;
  }
  catch (PDNSException& pe) {
    g_log << Logger::Error << kBackendId << " " << pe.reason << endl;
    cleanup();
    throw;
  }
  return true;
}

//
// Magic class that is activated when the dynamic library is loaded
//

class PipeFactory : public BackendFactory
{
public:
  PipeFactory() :
    BackendFactory("pipe") {}

  void declareArguments(const string& suffix = "") override
  {
    declare(suffix, "command", "Command to execute for piping questions to", "");
    declare(suffix, "timeout", "Number of milliseconds to wait for an answer", "2000");
    declare(suffix, "regex", "Regular expression of queries to pass to coprocess", "");
    declare(suffix, "abi-version", "Version of the pipe backend ABI", "1");
  }

  DNSBackend* make(const string& suffix = "") override
  {
    return new PipeBackend(suffix);
  }
};

class PipeLoader
{
public:
  PipeLoader()
  {
    BackendMakers().report(new PipeFactory);
    g_log << Logger::Info << kBackendId << " This is the pipe backend version " VERSION
#ifndef REPRODUCIBLE
          << " (" __DATE__ " " __TIME__ ")"
#endif
          << " reporting" << endl;
  }
};

static PipeLoader pipeloader;

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	#ifdef HAVE_CONFIG_H
23	#include "config.h"
24	#endif
25	#include <string>
26	#include <map>
27	#include <unistd.h>
28	#include <stdlib.h>
29	#include <sstream>
30	#include "coprocess.hh"
31
32	#include "pdns/namespaces.hh"
33
34	#include "pdns/dns.hh"
35	#include "pdns/dnsbackend.hh"
36	#include "pdns/dnspacket.hh"
37	#include "pdns/pdnsexception.hh"
38	#include "pdns/logger.hh"
39	#include "pdns/arguments.hh"
40	#include <sys/socket.h>
41	#include <netinet/in.h>
42	#include <arpa/inet.h>
43	#include "pipebackend.hh"
44
45	static const char* kBackendId = "[PIPEBackend]";
46
47	CoWrapper::CoWrapper(const string& command, int timeout, int abiVersion)
UNCOV 48	{	×
UNCOV 49	d_command = command;	×
UNCOV 50	d_timeout = timeout;	×
UNCOV 51	d_abiVersion = abiVersion;	×
UNCOV 52	launch(); // let exceptions fall through - if initial launch fails, we want to die	×
53	// I think
UNCOV 54	}	×
55
56	CoWrapper::~CoWrapper()
UNCOV 57	{	×
UNCOV 58	}	×
59
60	void CoWrapper::launch()
UNCOV 61	{	×
UNCOV 62	if (d_cp)	×
UNCOV 63	return;	×
64
UNCOV 65	if (d_command.empty())	×
66	throw ArgException("pipe-command is not specified");	×
67
UNCOV 68	if (isUnixSocket(d_command)) {	×
69	d_cp = std::make_unique<UnixRemote>(d_command);	×
70	}	×
UNCOV 71	else {	×
UNCOV 72	auto coprocess = std::make_unique<CoProcess>(d_command, d_timeout);	×
UNCOV 73	coprocess->launch();	×
UNCOV 74	d_cp = std::move(coprocess);	×
UNCOV 75	}	×
76
UNCOV 77	d_cp->send("HELO\t" + std::to_string(d_abiVersion));	×
UNCOV 78	string banner;	×
UNCOV 79	d_cp->receive(banner);	×
UNCOV 80	g_log << Logger::Error << "Backend launched with banner: " << banner << endl;	×
UNCOV 81	}	×
82
83	void CoWrapper::send(const string& line)
UNCOV 84	{	×
UNCOV 85	launch();	×
UNCOV 86	try {	×
UNCOV 87	d_cp->send(line);	×
UNCOV 88	return;	×
UNCOV 89	}	×
UNCOV 90	catch (PDNSException& ae) {	×
91	d_cp.reset();	×
92	throw;	×
93	}	×
UNCOV 94	}	×
95	void CoWrapper::receive(string& line)
UNCOV 96	{	×
UNCOV 97	launch();	×
UNCOV 98	try {	×
UNCOV 99	d_cp->receive(line);	×
UNCOV 100	return;	×
UNCOV 101	}	×
UNCOV 102	catch (PDNSException& ae) {	×
103	g_log << Logger::Warning << kBackendId << " Unable to receive data from coprocess. " << ae.reason << endl;	×
104	d_cp.reset();	×
105	throw;	×
106	}	×
UNCOV 107	}	×
108
109	PipeBackend::PipeBackend(const string& suffix)
UNCOV 110	{	×
UNCOV 111	d_disavow = false;	×
UNCOV 112	signal(SIGCHLD, SIG_IGN);	×
UNCOV 113	setArgPrefix("pipe" + suffix);	×
UNCOV 114	try {	×
UNCOV 115	launch();	×
UNCOV 116	}	×
UNCOV 117	catch (const ArgException& A) {	×
118	g_log << Logger::Error << kBackendId << " Unable to launch, fatal argument error: " << A.reason << endl;	×
119	throw;	×
120	}	×
UNCOV 121	catch (...) {	×
122	throw;	×
123	}	×
UNCOV 124	}	×
125
126	void PipeBackend::launch()
UNCOV 127	{	×
UNCOV 128	if (d_coproc)	×
UNCOV 129	return;	×
130
UNCOV 131	try {	×
UNCOV 132	if (!getArg("regex").empty()) {	×
133	d_regex = std::make_unique<Regex>(getArg("regex"));	×
134	}	×
UNCOV 135	d_regexstr = getArg("regex");	×
UNCOV 136	d_abiVersion = getArgAsNum("abi-version");	×
UNCOV 137	d_coproc = std::make_unique<CoWrapper>(getArg("command"), getArgAsNum("timeout"), getArgAsNum("abi-version"));	×
UNCOV 138	}	×
139
UNCOV 140	catch (const ArgException& A) {	×
141	cleanup();	×
142	throw;	×
143	}	×
UNCOV 144	}	×
145
146	/*
147	* Cleans up the co-process wrapper
148	*/
149	void PipeBackend::cleanup()
UNCOV 150	{	×
UNCOV 151	d_coproc.reset(nullptr);	×
UNCOV 152	d_regex.reset();	×
UNCOV 153	d_regexstr = string();	×
UNCOV 154	d_abiVersion = 0;	×
UNCOV 155	}	×
156
157	void PipeBackend::lookup(const QType& qtype, const DNSName& qname, int zoneId, DNSPacket* pkt_p)
UNCOV 158	{	×
UNCOV 159	try {	×
UNCOV 160	launch();	×
UNCOV 161	d_disavow = false;	×
UNCOV 162	if (d_regex && !d_regex->match(qname.toStringRootDot())) {	×
163	if (::arg().mustDo("query-logging"))	×
164	g_log << Logger::Error << "Query for '" << qname << "' failed regex '" << d_regexstr << "'" << endl;	×
165	d_disavow = true; // don't pass to backend	×
166	}	×
UNCOV 167	else {	×
UNCOV 168	ostringstream query;	×
UNCOV 169	string localIP = "0.0.0.0";	×
UNCOV 170	string remoteIP = "0.0.0.0";	×
UNCOV 171	Netmask realRemote("0.0.0.0/0");	×
UNCOV 172	if (pkt_p) {	×
UNCOV 173	localIP = pkt_p->getLocal().toString();	×
UNCOV 174	realRemote = pkt_p->getRealRemote();	×
UNCOV 175	remoteIP = pkt_p->getInnerRemote().toString();	×
UNCOV 176	}	×
177	// abi-version = 1
178	// type qname qclass qtype id remote-ip-address
UNCOV 179	query << "Q\t" << qname.toStringRootDot() << "\tIN\t" << qtype.toString() << "\t" << zoneId << "\t" << remoteIP;	×
180
181	// add the local-ip-address if abi-version is set to 2
UNCOV 182	if (d_abiVersion >= 2)	×
UNCOV 183	query << "\t" << localIP;	×
UNCOV 184	if (d_abiVersion >= 3)	×
UNCOV 185	query << "\t" << realRemote.toString();	×
186
UNCOV 187	if (::arg().mustDo("query-logging"))	×
188	g_log << Logger::Error << "Query: '" << query.str() << "'" << endl;	×
UNCOV 189	d_coproc->send(query.str());	×
UNCOV 190	}	×
UNCOV 191	}	×
UNCOV 192	catch (PDNSException& pe) {	×
193	g_log << Logger::Error << kBackendId << " Error from coprocess: " << pe.reason << endl;	×
194	d_disavow = true;	×
195	}	×
UNCOV 196	d_qtype = qtype;	×
UNCOV 197	d_qname = qname;	×
UNCOV 198	}	×
199
200	bool PipeBackend::list(const DNSName& target, int inZoneId, bool /* include_disabled */)
201	{	×
202	try {	×
203	launch();	×
204	d_disavow = false;	×
205	ostringstream query;	×
206	// The question format:
207
208	// type qname qclass qtype id ip-address
209	if (d_abiVersion >= 4)	×
210	query << "AXFR\t" << inZoneId << "\t" << target.toStringRootDot();	×
211	else	×
212	query << "AXFR\t" << inZoneId;	×
213
214	d_coproc->send(query.str());	×
215	}	×
216	catch (PDNSException& ae) {	×
217	g_log << Logger::Error << kBackendId << " Error from coprocess: " << ae.reason << endl;	×
218	}	×
219	d_qname = DNSName(std::to_string(inZoneId)); // why do we store a number here??	×
220	return true;	×
221	}	×
222
223	string PipeBackend::directBackendCmd(const string& query)
224	{	×
225	if (d_abiVersion < 5)	×
226	return "not supported on ABI version " + std::to_string(d_abiVersion) + " (use ABI version 5 or later)\n";	×
227
228	try {	×
229	launch();	×
230	ostringstream oss;	×
231	oss << "CMD\t" << query;	×
232	d_coproc->send(oss.str());	×
233	}	×
234	catch (PDNSException& ae) {	×
235	g_log << Logger::Error << kBackendId << " Error from coprocess: " << ae.reason << endl;	×
236	cleanup();	×
237	}	×
238
239	ostringstream oss;	×
240	while (true) {	×
241	string line;	×
242	d_coproc->receive(line);	×
243	if (line == "END")	×
244	break;	×
245	oss << line << std::endl;	×
246	};	×
247
248	return oss.str();	×
249	}	×
250
251	//! For the dynamic loader
252	DNSBackend* PipeBackend::maker()
253	{	×
254	try {	×
255	return new PipeBackend();	×
256	}	×
257	catch (...) {	×
258	g_log << Logger::Error << kBackendId << " Unable to instantiate a pipebackend!" << endl;	×
259	return nullptr;	×
260	}	×
261	}	×
262
263	PipeBackend::~PipeBackend()
UNCOV 264	{	×
UNCOV 265	cleanup();	×
UNCOV 266	}	×
267
268	bool PipeBackend::get(DNSResourceRecord& r)
UNCOV 269	{	×
UNCOV 270	if (d_disavow) // this query has been blocked	×
271	return false;	×
272
UNCOV 273	string line;	×
274
275	// The answer format:
276	// DATA qname qclass qtype ttl id content
UNCOV 277	unsigned int extraFields = 0;	×
UNCOV 278	if (d_abiVersion >= 3)	×
UNCOV 279	extraFields = 2;	×
280
UNCOV 281	try {	×
UNCOV 282	launch();	×
UNCOV 283	for (;;) {	×
UNCOV 284	d_coproc->receive(line);	×
UNCOV 285	vector<string> parts;	×
UNCOV 286	stringtok(parts, line, "\t");	×
UNCOV 287	if (parts.empty()) {	×
288	g_log << Logger::Error << kBackendId << " Coprocess returned empty line in query for " << d_qname << endl;	×
289	throw PDNSException("Format error communicating with coprocess");	×
290	}	×
UNCOV 291	else if (parts[0] == "FAIL") {	×
292	throw DBException("coprocess returned a FAIL");	×
293	}	×
UNCOV 294	else if (parts[0] == "END") {	×
UNCOV 295	return false;	×
UNCOV 296	}	×
UNCOV 297	else if (parts[0] == "LOG") {	×
298	g_log << Logger::Error << "Coprocess: " << line.substr(4) << endl;	×
299	continue;	×
300	}	×
UNCOV 301	else if (parts[0] == "DATA") { // yay	×
UNCOV 302	if (parts.size() < 7 + extraFields) {	×
303	g_log << Logger::Error << kBackendId << " Coprocess returned incomplete or empty line in data section for query for " << d_qname << endl;	×
304	throw PDNSException("Format error communicating with coprocess in data section");	×
305	// now what?
306	}	×
307
UNCOV 308	if (d_abiVersion >= 3) {	×
UNCOV 309	r.scopeMask = std::stoi(parts[1]);	×
UNCOV 310	r.auth = (parts[2] == "1");	×
UNCOV 311	}	×
UNCOV 312	else {	×
UNCOV 313	r.scopeMask = 0;	×
UNCOV 314	r.auth = true;	×
UNCOV 315	}	×
UNCOV 316	r.qname = DNSName(parts[1 + extraFields]);	×
UNCOV 317	r.qtype = parts[3 + extraFields];	×
UNCOV 318	pdns::checked_stoi_into(r.ttl, parts[4 + extraFields]);	×
UNCOV 319	pdns::checked_stoi_into(r.domain_id, parts[5 + extraFields]);	×
320
UNCOV 321	if (r.qtype.getCode() != QType::MX && r.qtype.getCode() != QType::SRV) {	×
UNCOV 322	r.content.clear();	×
UNCOV 323	for (unsigned int n = 6 + extraFields; n < parts.size(); ++n) {	×
UNCOV 324	if (n != 6 + extraFields)	×
325	r.content.append(1, ' ');	×
UNCOV 326	r.content.append(parts[n]);	×
UNCOV 327	}	×
UNCOV 328	}	×
UNCOV 329	else {	×
UNCOV 330	if (parts.size() < 8 + extraFields) {	×
331	g_log << Logger::Error << kBackendId << " Coprocess returned incomplete MX/SRV line in data section for query for " << d_qname << endl;	×
332	throw PDNSException("Format error communicating with coprocess in data section of MX/SRV record");	×
333	}	×
334
UNCOV 335	r.content = parts[6 + extraFields] + " " + parts[7 + extraFields];	×
UNCOV 336	}	×
UNCOV 337	break;	×
UNCOV 338	}	×
339	else	×
340	throw PDNSException("Coprocess backend sent incorrect response '" + line + "'");	×
UNCOV 341	}	×
UNCOV 342	}	×
UNCOV 343	catch (DBException& dbe) {	×
344	g_log << Logger::Error << kBackendId << " " << dbe.reason << endl;	×
345	throw;	×
346	}	×
UNCOV 347	catch (PDNSException& pe) {	×
348	g_log << Logger::Error << kBackendId << " " << pe.reason << endl;	×
349	cleanup();	×
350	throw;	×
351	}	×
UNCOV 352	return true;	×
UNCOV 353	}	×
354
355	//
356	// Magic class that is activated when the dynamic library is loaded
357	//
358
359	class PipeFactory : public BackendFactory
360	{
361	public:
362	PipeFactory() :
363	BackendFactory("pipe") {}	1,321✔
364
365	void declareArguments(const string& suffix = "") override
UNCOV 366	{	×
UNCOV 367	declare(suffix, "command", "Command to execute for piping questions to", "");	×
UNCOV 368	declare(suffix, "timeout", "Number of milliseconds to wait for an answer", "2000");	×
UNCOV 369	declare(suffix, "regex", "Regular expression of queries to pass to coprocess", "");	×
UNCOV 370	declare(suffix, "abi-version", "Version of the pipe backend ABI", "1");	×
UNCOV 371	}	×
372
373	DNSBackend* make(const string& suffix = "") override
UNCOV 374	{	×
UNCOV 375	return new PipeBackend(suffix);	×
UNCOV 376	}	×
377	};
378
379	class PipeLoader
380	{
381	public:
382	PipeLoader()
383	{	1,321✔
384	BackendMakers().report(new PipeFactory);	1,321✔
385	g_log << Logger::Info << kBackendId << " This is the pipe backend version " VERSION	1,321✔
386	#ifndef REPRODUCIBLE	1,321✔
387	<< " (" __DATE__ " " __TIME__ ")"	1,321✔
388	#endif	1,321✔
389	<< " reporting" << endl;	1,321✔
390	}	1,321✔
391	};
392
393	static PipeLoader pipeloader;

PowerDNS / pdns / 6675548676

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous