28176940609

Committed 25 Jun 2026 02:20PM UTC coverage: 71.035% (-0.03%) from 71.067%

Build # 28176940609

Build Type

push

github

Committed by

web-flow

Commit Message

Merge pull request #17603 from rgacogne/ddist-469

dnsdist: Fix out-of-bounds read in SetMacAddrAction

Coverage Stats

46473 of 81902 branches covered (56.74%)

Branch coverage included in aggregate %.

0 of 3 new or added lines in 1 file covered. (0.0%)

70 existing lines in 11 files now uncovered.

133430 of 171356 relevant lines covered (77.87%)

6048322.14 hits per line

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

64.0

/modules/gpgsqlbackend/spgsql.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 "spgsql.hh"
#include <sys/time.h>
#include <iostream>
#include "pdns/logger.hh"
#include "pdns/dns.hh"
#include "pdns/namespaces.hh"
#include <algorithm>

class SPgSQLStatement : public SSqlStatement
{
public:
  SPgSQLStatement(Logr::log_t log, const string& query, bool dolog, int nparams, SPgSQL* db, unsigned int nstatement)
  {
    d_slog = log;
    d_query = query;
    d_dolog = dolog;
    d_parent = db;
    d_nparams = nparams;
    d_nstatement = nstatement;
  }

  SSqlStatement* bind(const string& name, bool value) override { return bind(name, string(value ? "t" : "f")); }
  SSqlStatement* bind(const string& name, int value) override { return bind(name, std::to_string(value)); }
  SSqlStatement* bind(const string& name, uint32_t value) override { return bind(name, std::to_string(value)); }
  SSqlStatement* bind(const string& name, long value) override { return bind(name, std::to_string(value)); }
  SSqlStatement* bind(const string& name, unsigned long value) override { return bind(name, std::to_string(value)); }
  SSqlStatement* bind(const string& name, long long value) override { return bind(name, std::to_string(value)); }
  SSqlStatement* bind(const string& name, unsigned long long value) override { return bind(name, std::to_string(value)); }
  SSqlStatement* bind(const string& /* name */, const std::string& value) override
  {
    prepareStatement();
    allocate();
    if (d_paridx >= d_nparams) {
      releaseStatement();
      throw SSqlException("Attempt to bind more parameters than query has: " + d_query);
    }
    paramValues[d_paridx] = new char[value.size() + 1];
    memset(paramValues[d_paridx], 0, sizeof(char) * (value.size() + 1));
    value.copy(paramValues[d_paridx], value.size());
    paramLengths[d_paridx] = value.size();
    d_paridx++;
    return this;
  }
  SSqlStatement* bindNull(const string& /* name */) override
  {
    prepareStatement();
    d_paridx++;
    return this;
  } // these are set null in allocate()
  SSqlStatement* execute() override
  {
    prepareStatement();
    if (d_dolog) {
      if (g_slogStructured) {
        if (d_paridx) {
          // This is ugly, but will do until paramValues is converted to a
          // std::array.
          std::vector<char*> vecparam;
          vecparam.reserve(d_paridx);
          for (auto i = 0; i < d_paridx; ++i) {
            vecparam[i] = paramValues[i];
          }
          d_slog->info(Logr::Warning, "execute SQL query", "query", Logging::Loggable(d_query), "arguments", Logging::IterLoggable(vecparam.cbegin(), vecparam.cend()));
        }
        else {
          d_slog->info(Logr::Warning, "execute SQL query", "query", Logging::Loggable(d_query));
        }
      }
      else {
        g_log << Logger::Warning << "Query " << ((long)(void*)this) << ": Statement: " << d_query << endl;
        if (d_paridx) {
          std::stringstream log_message;
          // Log message is similar, but not exactly the same as the postgres server log.
          log_message << "Query " << ((long)(void*)this) << ": Parameters: ";
          for (int i = 0; i < d_paridx; i++) {
            if (i != 0) {
              log_message << ", ";
            }
            log_message << "$" << (i + 1) << " = ";
            if (paramValues[i] == nullptr) {
              log_message << "NULL";
            }
            else {
              log_message << "'" << paramValues[i] << "'";
            }
          }
          g_log << Logger::Warning << log_message.str() << endl;
        }
      }
      d_dtime.set();
    }
    if (!d_stmt.empty()) {
      d_res_set = PQexecPrepared(d_db(), d_stmt.c_str(), d_nparams, paramValues, paramLengths, nullptr, 0);
    }
    else {
      d_res_set = PQexecParams(d_db(), d_query.c_str(), d_nparams, nullptr, paramValues, paramLengths, nullptr, 0);
    }
    ExecStatusType status = PQresultStatus(d_res_set);
    if (status != PGRES_COMMAND_OK && status != PGRES_TUPLES_OK && status != PGRES_NONFATAL_ERROR) {
      string errmsg(PQresultErrorMessage(d_res_set));
      releaseStatement();
      throw SSqlException("Fatal error during query: " + d_query + string(": ") + errmsg);
    }
    d_cur_set = 0;
    if (d_dolog) {
      auto diff = d_dtime.udiffNoReset();
      SLOG(g_log << Logger::Warning << "Query " << ((long)(void*)this) << ": " << diff << " us to execute" << endl,
           d_slog->info(Logr::Warning, "query completed", "microseconds", Logging::Loggable(diff)));
    }

    nextResult();
    return this;
  }

  void nextResult()
  {
    if (d_res_set == nullptr)
      return;
    if (d_cur_set >= PQntuples(d_res_set)) {
      PQclear(d_res_set);
      d_res_set = nullptr;
      return;
    }
    if (PQftype(d_res_set, 0) == 1790) { // REFCURSOR
      SLOG(g_log << Logger::Error << "Postgres query returned a REFCURSOR and we do not support those - see https://github.com/PowerDNS/pdns/pull/10259" << endl,
           d_slog->info(Logr::Error, "query returned a REFCURSOR which is not supported by PowerDNS", "more information", Logging::Loggable("https://github.com/PowerDNS/pdns/pull/10259")));
      PQclear(d_res_set);
      d_res_set = nullptr;
    }
    else {
      d_res = d_res_set;
      d_res_set = nullptr;
      d_resnum = PQntuples(d_res);
    }
  }

  bool hasNextRow() override
  {
    if (d_dolog && d_residx == d_resnum) {
      SLOG(g_log << Logger::Warning << "Query " << ((long)(void*)this) << ": " << d_dtime.udiff() << " us total to last row" << endl,
           d_slog->info(Logr::Warning, "all query results procesed", "microseconds", Logging::Loggable(d_dtime.udiff())));
    }

    return d_residx < d_resnum;
  }

  SSqlStatement* nextRow(row_t& row) override
  {
    int i;
    row.clear();
    if (d_residx >= d_resnum || !d_res)
      return this;
    row.reserve(PQnfields(d_res));
    for (i = 0; i < PQnfields(d_res); i++) {
      if (PQgetisnull(d_res, d_residx, i)) {
        row.emplace_back("");
      }
      else if (PQftype(d_res, i) == 16) { // BOOLEAN
        char* val = PQgetvalue(d_res, d_residx, i);
        row.emplace_back(val[0] == 't' ? "1" : "0");
      }
      else {
        row.emplace_back(PQgetvalue(d_res, d_residx, i));
      }
    }
    d_residx++;
    if (d_residx >= d_resnum) {
      PQclear(d_res);
      d_res = nullptr;
      nextResult();
    }
    return this;
  }

  SSqlStatement* getResult(result_t& result) override
  {
    result.clear();
    if (d_res == nullptr)
      return this;
    result.reserve(d_resnum);
    row_t row;
    while (hasNextRow()) {
      nextRow(row);
      result.push_back(std::move(row));
    }
    return this;
  }

  SSqlStatement* reset() override
  {
    int i;
    if (d_res) {
      PQclear(d_res);
    }
    if (d_res_set) {
      PQclear(d_res_set);
    }
    d_res_set = nullptr;
    d_res = nullptr;
    d_paridx = d_residx = d_resnum = 0;
    if (paramValues) {
      for (i = 0; i < d_nparams; i++) {
        if (paramValues[i]) {
          delete[] paramValues[i];
        }
      }
    }
    delete[] paramValues;
    paramValues = nullptr;
    delete[] paramLengths;
    paramLengths = nullptr;
    return this;
  }

  const std::string& getQuery() override { return d_query; }

  ~SPgSQLStatement() override
  {
    releaseStatement();
  }

private:
  PGconn* d_db()
  {
    return d_parent->db();
  }

  void releaseStatement()
  {
    d_prepared = false;
    reset();
    if (!d_stmt.empty()) {
      string cmd = string("DEALLOCATE " + d_stmt);
      PGresult* res = PQexec(d_db(), cmd.c_str());
      PQclear(res);
      d_stmt.clear();
    }
  }

  void prepareStatement()
  {
    if (d_prepared)
      return;
    if (d_parent->usePrepared()) {
      // prepare a statement; name must be unique per session (using d_nstatement to ensure this).
      this->d_stmt = string("stmt") + std::to_string(d_nstatement);
      PGresult* res = PQprepare(d_db(), d_stmt.c_str(), d_query.c_str(), d_nparams, nullptr);
      ExecStatusType status = PQresultStatus(res);
      string errmsg(PQresultErrorMessage(res));
      PQclear(res);
      if (status != PGRES_COMMAND_OK && status != PGRES_TUPLES_OK && status != PGRES_NONFATAL_ERROR) {
        releaseStatement();
        throw SSqlException("Fatal error during prePQpreparepare: " + d_query + string(": ") + errmsg);
      }
    }
    paramValues = nullptr;
    paramLengths = nullptr;
    d_cur_set = d_paridx = d_residx = d_resnum = 0;
    d_res = nullptr;
    d_res_set = nullptr;
    d_prepared = true;
  }

  void allocate()
  {
    if (paramValues != nullptr)
      return;
    paramValues = new char*[d_nparams];
    paramLengths = new int[d_nparams];
    memset(paramValues, 0, sizeof(char*) * d_nparams);
    memset(paramLengths, 0, sizeof(int) * d_nparams);
  }

  string d_query;
  string d_stmt;
  SPgSQL* d_parent;
  PGresult* d_res_set{nullptr};
  PGresult* d_res{nullptr};
  bool d_dolog;
  std::shared_ptr<Logr::Logger> d_slog;
  DTime d_dtime; // only used if d_dolog is set
  bool d_prepared{false};
  int d_nparams;
  int d_paridx{0};
  char** paramValues{nullptr};
  int* paramLengths{nullptr};
  int d_residx{0};
  int d_resnum{0};
  int d_cur_set{0};
  unsigned int d_nstatement;
};

bool SPgSQL::s_dolog;

static string escapeForPQparam(const string& v)
{
  string ret = v;
  boost::replace_all(ret, "\\", "\\\\");
  boost::replace_all(ret, "'", "\\'");

  return string("'") + ret + string("'");
}

SPgSQL::SPgSQL(Logr::log_t log, const string& database, const string& host, const string& port, const string& user,
               const string& password, const string& extra_connection_parameters, const bool use_prepared)
{
  d_slog = log;
  d_db = nullptr;
  d_in_trx = false;
  d_connectstr = "";
  d_nstatements = 0;

  if (!database.empty())
    d_connectstr += "dbname=" + escapeForPQparam(database);

  if (!user.empty())
    d_connectstr += " user=" + escapeForPQparam(user);

  if (!host.empty())
    d_connectstr += " host=" + escapeForPQparam(host);

  if (!port.empty())
    d_connectstr += " port=" + escapeForPQparam(port);

  if (!extra_connection_parameters.empty())
    d_connectstr += " " + extra_connection_parameters;

  d_connectlogstr = d_connectstr;

  if (!password.empty()) {
    d_connectlogstr += " password=<HIDDEN>";
    d_connectstr += " password=" + escapeForPQparam(password);
  }

  d_use_prepared = use_prepared;

  d_db = PQconnectdb(d_connectstr.c_str());

  if (!d_db || PQstatus(d_db) == CONNECTION_BAD) {
    try {
      throw sPerrorException("Unable to connect to database, connect string: " + d_connectlogstr);
    }
    catch (...) {
      if (d_db)
        PQfinish(d_db);
      d_db = 0;
      throw;
    }
  }
}

void SPgSQL::setLog(bool state)
{
  s_dolog = state;
}

SPgSQL::~SPgSQL()
{
  PQfinish(d_db);
}

SSqlException SPgSQL::sPerrorException(const string& reason)
{
  return SSqlException(reason + string(": ") + (d_db ? PQerrorMessage(d_db) : "no connection"));
}

void SPgSQL::execute(const string& query)
{
  PGresult* res = PQexec(d_db, query.c_str());
  ExecStatusType status = PQresultStatus(res);
  string errmsg(PQresultErrorMessage(res));
  PQclear(res);
  if (status != PGRES_COMMAND_OK && status != PGRES_TUPLES_OK && status != PGRES_NONFATAL_ERROR) {
    throw sPerrorException("Fatal error during query: " + errmsg);
  }
}

std::unique_ptr<SSqlStatement> SPgSQL::prepare(const string& query, int nparams)
{
  d_nstatements++;
  return std::make_unique<SPgSQLStatement>(d_slog, query, s_dolog, nparams, this, d_nstatements);
}

void SPgSQL::startTransaction()
{
  execute("begin");
  d_in_trx = true;
}

void SPgSQL::commit()
{
  execute("commit");
  d_in_trx = false;
}

void SPgSQL::rollback()
{
  execute("rollback");
  d_in_trx = false;
}

bool SPgSQL::isConnectionUsable()
{
  if (PQstatus(d_db) != CONNECTION_OK) {
    return false;
  }

  bool usable = false;
  int sd = PQsocket(d_db);
  bool wasNonBlocking = isNonBlocking(sd);

  if (!wasNonBlocking) {
    if (!setNonBlocking(sd)) {
      return usable;
    }
  }

  usable = isTCPSocketUsable(sd);

  if (!wasNonBlocking) {
    if (!setBlocking(sd)) {
      usable = false;
    }
  }

  return usable;
}

void SPgSQL::reconnect()
{
  PQreset(d_db);
}

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
23	#ifdef HAVE_CONFIG_H
24	#include "config.h"
25	#endif
26	#include <string>
27	#include "spgsql.hh"
28	#include <sys/time.h>
29	#include <iostream>
30	#include "pdns/logger.hh"
31	#include "pdns/dns.hh"
32	#include "pdns/namespaces.hh"
33	#include <algorithm>
34
35	class SPgSQLStatement : public SSqlStatement
36	{
37	public:
38	SPgSQLStatement(Logr::log_t log, const string& query, bool dolog, int nparams, SPgSQL* db, unsigned int nstatement)
39	{	131,497✔
40	d_slog = log;	131,497✔
41	d_query = query;	131,497✔
42	d_dolog = dolog;	131,497✔
43	d_parent = db;	131,497✔
44	d_nparams = nparams;	131,497✔
45	d_nstatement = nstatement;	131,497✔
46	}	131,497✔
47
48	SSqlStatement* bind(const string& name, bool value) override { return bind(name, string(value ? "t" : "f")); }	529,658✔
49	SSqlStatement* bind(const string& name, int value) override { return bind(name, std::to_string(value)); }	548,146✔
50	SSqlStatement* bind(const string& name, uint32_t value) override { return bind(name, std::to_string(value)); }	223,907✔
51	SSqlStatement* bind(const string& name, long value) override { return bind(name, std::to_string(value)); }	82✔
52	SSqlStatement* bind(const string& name, unsigned long value) override { return bind(name, std::to_string(value)); }	11✔
53	SSqlStatement* bind(const string& name, long long value) override { return bind(name, std::to_string(value)); }	×
54	SSqlStatement* bind(const string& name, unsigned long long value) override { return bind(name, std::to_string(value)); }	×
55	SSqlStatement* bind(const string& /* name */, const std::string& value) override
56	{	2,229,349✔
57	prepareStatement();	2,229,349✔
58	allocate();	2,229,349✔
59	if (d_paridx >= d_nparams) {	2,229,349!
60	releaseStatement();	×
61	throw SSqlException("Attempt to bind more parameters than query has: " + d_query);	×
62	}	×
63	paramValues[d_paridx] = new char[value.size() + 1];	2,229,349✔
64	memset(paramValues[d_paridx], 0, sizeof(char) * (value.size() + 1));	2,229,349✔
65	value.copy(paramValues[d_paridx], value.size());	2,229,349✔
66	paramLengths[d_paridx] = value.size();	2,229,349✔
67	d_paridx++;	2,229,349✔
68	return this;	2,229,349✔
69	}	2,229,349✔
70	SSqlStatement* bindNull(const string& /* name */) override
71	{	162,999✔
72	prepareStatement();	162,999✔
73	d_paridx++;	162,999✔
74	return this;	162,999✔
75	} // these are set null in allocate()	162,999✔
76	SSqlStatement* execute() override
77	{	332,370✔
78	prepareStatement();	332,370✔
79	if (d_dolog) {	332,370!
80	if (g_slogStructured) {	×
81	if (d_paridx) {	×
82	// This is ugly, but will do until paramValues is converted to a
83	// std::array.
84	std::vector<char*> vecparam;	×
85	vecparam.reserve(d_paridx);	×
86	for (auto i = 0; i < d_paridx; ++i) {	×
87	vecparam[i] = paramValues[i];	×
88	}	×
89	d_slog->info(Logr::Warning, "execute SQL query", "query", Logging::Loggable(d_query), "arguments", Logging::IterLoggable(vecparam.cbegin(), vecparam.cend()));	×
90	}	×
91	else {	×
92	d_slog->info(Logr::Warning, "execute SQL query", "query", Logging::Loggable(d_query));	×
93	}	×
94	}	×
95	else {	×
96	g_log << Logger::Warning << "Query " << ((long)(void*)this) << ": Statement: " << d_query << endl;	×
97	if (d_paridx) {	×
98	std::stringstream log_message;	×
99	// Log message is similar, but not exactly the same as the postgres server log.
100	log_message << "Query " << ((long)(void*)this) << ": Parameters: ";	×
101	for (int i = 0; i < d_paridx; i++) {	×
102	if (i != 0) {	×
103	log_message << ", ";	×
104	}	×
105	log_message << "$" << (i + 1) << " = ";	×
106	if (paramValues[i] == nullptr) {	×
107	log_message << "NULL";	×
108	}	×
109	else {	×
110	log_message << "'" << paramValues[i] << "'";	×
111	}	×
112	}	×
113	g_log << Logger::Warning << log_message.str() << endl;	×
114	}	×
115	}	×
116	d_dtime.set();	×
117	}	×
118	if (!d_stmt.empty()) {	332,370!
119	d_res_set = PQexecPrepared(d_db(), d_stmt.c_str(), d_nparams, paramValues, paramLengths, nullptr, 0);	332,370✔
120	}	332,370✔
UNCOV 121	else {	×
UNCOV 122	d_res_set = PQexecParams(d_db(), d_query.c_str(), d_nparams, nullptr, paramValues, paramLengths, nullptr, 0);	×
UNCOV 123	}	×
124	ExecStatusType status = PQresultStatus(d_res_set);	332,370✔
125	if (status != PGRES_COMMAND_OK && status != PGRES_TUPLES_OK && status != PGRES_NONFATAL_ERROR) {	332,370!
126	string errmsg(PQresultErrorMessage(d_res_set));	×
127	releaseStatement();	×
128	throw SSqlException("Fatal error during query: " + d_query + string(": ") + errmsg);	×
129	}	×
130	d_cur_set = 0;	332,370✔
131	if (d_dolog) {	332,370!
132	auto diff = d_dtime.udiffNoReset();	×
133	SLOG(g_log << Logger::Warning << "Query " << ((long)(void*)this) << ": " << diff << " us to execute" << endl,	×
134	d_slog->info(Logr::Warning, "query completed", "microseconds", Logging::Loggable(diff)));	×
135	}	×
136
137	nextResult();	332,370✔
138	return this;	332,370✔
139	}	332,370✔
140
141	void nextResult()
142	{	350,907✔
143	if (d_res_set == nullptr)	350,907✔
144	return;	18,539✔
145	if (d_cur_set >= PQntuples(d_res_set)) {	332,368✔
146	PQclear(d_res_set);	313,831✔
147	d_res_set = nullptr;	313,831✔
148	return;	313,831✔
149	}	313,831✔
150	if (PQftype(d_res_set, 0) == 1790) { // REFCURSOR	18,537!
151	SLOG(g_log << Logger::Error << "Postgres query returned a REFCURSOR and we do not support those - see https://github.com/PowerDNS/pdns/pull/10259" << endl,	×
152	d_slog->info(Logr::Error, "query returned a REFCURSOR which is not supported by PowerDNS", "more information", Logging::Loggable("https://github.com/PowerDNS/pdns/pull/10259")));	×
153	PQclear(d_res_set);	×
154	d_res_set = nullptr;	×
155	}	×
156	else {	18,537✔
157	d_res = d_res_set;	18,537✔
158	d_res_set = nullptr;	18,537✔
159	d_resnum = PQntuples(d_res);	18,537✔
160	}	18,537✔
161	}	18,537✔
162
163	bool hasNextRow() override
164	{	296,716✔
165	if (d_dolog && d_residx == d_resnum) {	296,716!
166	SLOG(g_log << Logger::Warning << "Query " << ((long)(void*)this) << ": " << d_dtime.udiff() << " us total to last row" << endl,	×
167	d_slog->info(Logr::Warning, "all query results procesed", "microseconds", Logging::Loggable(d_dtime.udiff())));	×
168	}	×
169
170	return d_residx < d_resnum;	296,716✔
171	}	296,716✔
172
173	SSqlStatement* nextRow(row_t& row) override
174	{	272,287✔
175	int i;	272,287✔
176	row.clear();	272,287✔
177	if (d_residx >= d_resnum \|\| !d_res)	272,287!
178	return this;	×
179	row.reserve(PQnfields(d_res));	272,287✔
180	for (i = 0; i < PQnfields(d_res); i++) {	2,616,222✔
181	if (PQgetisnull(d_res, d_residx, i)) {	2,343,935✔
182	row.emplace_back("");	199,020✔
183	}	199,020✔
184	else if (PQftype(d_res, i) == 16) { // BOOLEAN	2,144,915✔
185	char* val = PQgetvalue(d_res, d_residx, i);	166✔
186	row.emplace_back(val[0] == 't' ? "1" : "0");	166✔
187	}	166✔
188	else {	2,144,749✔
189	row.emplace_back(PQgetvalue(d_res, d_residx, i));	2,144,749✔
190	}	2,144,749✔
191	}	2,343,935✔
192	d_residx++;	272,287✔
193	if (d_residx >= d_resnum) {	272,287✔
194	PQclear(d_res);	18,539✔
195	d_res = nullptr;	18,539✔
196	nextResult();	18,539✔
197	}	18,539✔
198	return this;	272,287✔
199	}	272,287✔
200
201	SSqlStatement* getResult(result_t& result) override
202	{	1,288✔
203	result.clear();	1,288✔
204	if (d_res == nullptr)	1,288✔
205	return this;	179✔
206	result.reserve(d_resnum);	1,109✔
207	row_t row;	1,109✔
208	while (hasNextRow()) {	2,283✔
209	nextRow(row);	1,174✔
210	result.push_back(std::move(row));	1,174✔
211	}	1,174✔
212	return this;	1,109✔
213	}	1,288✔
214
215	SSqlStatement* reset() override
216	{	456,528✔
217	int i;	456,528✔
218	if (d_res) {	456,528!
219	PQclear(d_res);	×
220	}	×
221	if (d_res_set) {	456,528!
222	PQclear(d_res_set);	×
223	}	×
224	d_res_set = nullptr;	456,528✔
225	d_res = nullptr;	456,528✔
226	d_paridx = d_residx = d_resnum = 0;	456,528✔
227	if (paramValues) {	456,528✔
228	for (i = 0; i < d_nparams; i++) {	2,724,691✔
229	if (paramValues[i]) {	2,392,346✔
230	delete[] paramValues[i];	2,229,350✔
231	}	2,229,350✔
232	}	2,392,346✔
233	}	332,345✔
234	delete[] paramValues;	456,528✔
235	paramValues = nullptr;	456,528✔
236	delete[] paramLengths;	456,528✔
237	paramLengths = nullptr;	456,528✔
238	return this;	456,528✔
239	}	456,528✔
240
241	const std::string& getQuery() override { return d_query; }	×
242
243	~SPgSQLStatement() override
244	{	124,258✔
245	releaseStatement();	124,258✔
246	}	124,258✔
247
248	private:
249	PGconn* d_db()
250	{	347,043✔
251	return d_parent->db();	347,043✔
252	}	347,043✔
253
254	void releaseStatement()
255	{	124,148✔
256	d_prepared = false;	124,148✔
257	reset();	124,148✔
258	if (!d_stmt.empty()) {	124,148✔
259	string cmd = string("DEALLOCATE " + d_stmt);	7,134✔
260	PGresult* res = PQexec(d_db(), cmd.c_str());	7,134✔
261	PQclear(res);	7,134✔
262	d_stmt.clear();	7,134✔
263	}	7,134✔
264	}	124,148✔
265
266	void prepareStatement()
267	{	2,724,689✔
268	if (d_prepared)	2,724,689✔
269	return;	2,717,153✔
270	if (d_parent->usePrepared()) {	7,539✔
271	// prepare a statement; name must be unique per session (using d_nstatement to ensure this).
272	this->d_stmt = string("stmt") + std::to_string(d_nstatement);	7,539✔
273	PGresult* res = PQprepare(d_db(), d_stmt.c_str(), d_query.c_str(), d_nparams, nullptr);	7,539✔
274	ExecStatusType status = PQresultStatus(res);	7,539✔
275	string errmsg(PQresultErrorMessage(res));	7,539✔
276	PQclear(res);	7,539✔
277	if (status != PGRES_COMMAND_OK && status != PGRES_TUPLES_OK && status != PGRES_NONFATAL_ERROR) {	7,539!
278	releaseStatement();	×
279	throw SSqlException("Fatal error during prePQpreparepare: " + d_query + string(": ") + errmsg);	×
280	}	×
281	}	7,539✔
282	paramValues = nullptr;	7,536✔
283	paramLengths = nullptr;	7,536✔
284	d_cur_set = d_paridx = d_residx = d_resnum = 0;	7,536✔
285	d_res = nullptr;	7,536✔
286	d_res_set = nullptr;	7,536✔
287	d_prepared = true;	7,536✔
288	}	7,536✔
289
290	void allocate()
291	{	2,229,344✔
292	if (paramValues != nullptr)	2,229,344✔
293	return;	1,897,000✔
294	paramValues = new char*[d_nparams];	332,344✔
295	paramLengths = new int[d_nparams];	332,344✔
296	memset(paramValues, 0, sizeof(char) d_nparams);	332,344✔
297	memset(paramLengths, 0, sizeof(int) * d_nparams);	332,344✔
298	}	332,344✔
299
300	string d_query;
301	string d_stmt;
302	SPgSQL* d_parent;
303	PGresult* d_res_set{nullptr};
304	PGresult* d_res{nullptr};
305	bool d_dolog;
306	std::shared_ptr<Logr::Logger> d_slog;
307	DTime d_dtime; // only used if d_dolog is set
308	bool d_prepared{false};
309	int d_nparams;
310	int d_paridx{0};
311	char** paramValues{nullptr};
312	int* paramLengths{nullptr};
313	int d_residx{0};
314	int d_resnum{0};
315	int d_cur_set{0};
316	unsigned int d_nstatement;
317	};
318
319	bool SPgSQL::s_dolog;
320
321	static string escapeForPQparam(const string& v)
322	{	3,339✔
323	string ret = v;	3,339✔
324	boost::replace_all(ret, "\\", "\\\\");	3,339✔
325	boost::replace_all(ret, "'", "\\'");	3,339✔
326
327	return string("'") + ret + string("'");	3,339✔
328	}	3,339✔
329
330	SPgSQL::SPgSQL(Logr::log_t log, const string& database, const string& host, const string& port, const string& user,
331	const string& password, const string& extra_connection_parameters, const bool use_prepared)
332	{	1,934✔
333	d_slog = log;	1,934✔
334	d_db = nullptr;	1,934✔
335	d_in_trx = false;	1,934✔
336	d_connectstr = "";	1,934✔
337	d_nstatements = 0;	1,934✔
338
339	if (!database.empty())	1,934✔
340	d_connectstr += "dbname=" + escapeForPQparam(database);	1,931✔
341
342	if (!user.empty())	1,934✔
343	d_connectstr += " user=" + escapeForPQparam(user);	1,414✔
344
345	if (!host.empty())	1,934!
346	d_connectstr += " host=" + escapeForPQparam(host);	×
347
348	if (!port.empty())	1,934!
349	d_connectstr += " port=" + escapeForPQparam(port);	×
350
351	if (!extra_connection_parameters.empty())	1,934!
352	d_connectstr += " " + extra_connection_parameters;	×
353
354	d_connectlogstr = d_connectstr;	1,934✔
355
356	if (!password.empty()) {	1,934!
357	d_connectlogstr += " password=<HIDDEN>";	×
358	d_connectstr += " password=" + escapeForPQparam(password);	×
359	}	×
360
361	d_use_prepared = use_prepared;	1,934✔
362
363	d_db = PQconnectdb(d_connectstr.c_str());	1,934✔
364
365	if (!d_db \|\| PQstatus(d_db) == CONNECTION_BAD) {	1,934✔
366	try {	×
367	throw sPerrorException("Unable to connect to database, connect string: " + d_connectlogstr);	×
368	}	×
369	catch (...) {	×
370	if (d_db)	×
371	PQfinish(d_db);	×
372	d_db = 0;	×
373	throw;	×
374	}	×
375	}	×
376	}	1,934✔
377
378	void SPgSQL::setLog(bool state)
379	{	1,934✔
380	s_dolog = state;	1,934✔
381	}	1,934✔
382
383	SPgSQL::~SPgSQL()
384	{	1,830✔
385	PQfinish(d_db);	1,830✔
386	}	1,830✔
387
388	SSqlException SPgSQL::sPerrorException(const string& reason)
389	{	×
390	return SSqlException(reason + string(": ") + (d_db ? PQerrorMessage(d_db) : "no connection"));	×
391	}	×
392
393	void SPgSQL::execute(const string& query)
394	{	1,155✔
395	PGresult* res = PQexec(d_db, query.c_str());	1,155✔
396	ExecStatusType status = PQresultStatus(res);	1,155✔
397	string errmsg(PQresultErrorMessage(res));	1,155✔
398	PQclear(res);	1,155✔
399	if (status != PGRES_COMMAND_OK && status != PGRES_TUPLES_OK && status != PGRES_NONFATAL_ERROR) {	1,155!
400	throw sPerrorException("Fatal error during query: " + errmsg);	×
401	}	×
402	}	1,155✔
403
404	std::unique_ptr<SSqlStatement> SPgSQL::prepare(const string& query, int nparams)
405	{	131,502✔
406	d_nstatements++;	131,502✔
407	return std::make_unique<SPgSQLStatement>(d_slog, query, s_dolog, nparams, this, d_nstatements);	131,502✔
408	}	131,502✔
409
410	void SPgSQL::startTransaction()
411	{	585✔
412	execute("begin");	585✔
413	d_in_trx = true;	585✔
414	}	585✔
415
416	void SPgSQL::commit()
417	{	544✔
418	execute("commit");	544✔
419	d_in_trx = false;	544✔
420	}	544✔
421
422	void SPgSQL::rollback()
423	{	26✔
424	execute("rollback");	26✔
425	d_in_trx = false;	26✔
426	}	26✔
427
428	bool SPgSQL::isConnectionUsable()
429	{	23,490✔
430	if (PQstatus(d_db) != CONNECTION_OK) {	23,490!
431	return false;	×
432	}	×
433
434	bool usable = false;	23,490✔
435	int sd = PQsocket(d_db);	23,490✔
436	bool wasNonBlocking = isNonBlocking(sd);	23,490✔
437
438	if (!wasNonBlocking) {	23,490!
439	if (!setNonBlocking(sd)) {	×
440	return usable;	×
441	}	×
442	}	×
443
444	usable = isTCPSocketUsable(sd);	23,490✔
445
446	if (!wasNonBlocking) {	23,490!
447	if (!setBlocking(sd)) {	×
448	usable = false;	×
449	}	×
450	}	×
451
452	return usable;	23,490✔
453	}	23,490✔
454
455	void SPgSQL::reconnect()
456	{	×
457	PQreset(d_db);	×
458	}	×

PowerDNS / pdns / 28176940609

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