9396932538

Committed 06 Jun 2024 07:35AM UTC coverage: 64.602% (+0.01%) from 64.591%

Build # 9396932538

Build Type

push

github

Committed by

web-flow

Commit Message

Merge pull request #14255 from rgacogne/ddist-add-proxy-protocol-values-from-ffi

dnsdist: Add a Lua FFI function to set proxy protocol values

Run Details

37082 of 88168 branches covered (42.06%)

Branch coverage included in aggregate %.

52 of 52 new or added lines in 2 files covered. (100.0%)

28 existing lines in 11 files now uncovered.

124349 of 161718 relevant lines covered (76.89%)

4981297.23 hits per line

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

66.77

/pdns/ssqlite3.cc

/*  SQLite backend for PowerDNS
 *  Copyright (C) 2003, Michel Stol <michel@powerdns.com>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 2
 *  as published by the Free Software Foundation.
 *
 *  Additionally, the license of this program contains a special
 *  exception which allows to distribute the program in binary form when
 *  it is linked against OpenSSL.
 *
 *  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 St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <string>
#include <sstream>
#include "ssqlite3.hh"
#include <iostream>
#include <fstream>
#include <utility>
#include "pdns/logger.hh"
#include "misc.hh"
#include "utility.hh"
#include <unistd.h>

/*
** Set all the parameters in the compiled SQL statement to NULL.
*
* copied from sqlite 3.3.6 // cmouse
*/
#if SQLITE_VERSION_NUMBER < 3003009
static int pdns_sqlite3_clear_bindings(sqlite3_stmt* pStmt)
{
  int i;
  int rc = SQLITE_OK;
  for (i = 1; rc == SQLITE_OK && i <= sqlite3_bind_parameter_count(pStmt); i++) {
    rc = sqlite3_bind_null(pStmt, i);
  }
  return rc;
}
#endif

static string SSQLite3ErrorString(sqlite3* database)
{
  return string(sqlite3_errmsg(database) + string(" (") + std::to_string(sqlite3_extended_errcode(database)) + string(")"));
}

class SSQLite3Statement : public SSqlStatement
{
public:
  SSQLite3Statement(SSQLite3* database, bool dolog, string query) :
    d_query(std::move(query)),
    d_db(database),
    d_dolog(dolog)
  {
  }

  SSQLite3Statement(const SSQLite3Statement&) = delete;
  SSQLite3Statement(SSQLite3Statement&&) = delete;
  SSQLite3Statement& operator=(const SSQLite3Statement&) = delete;
  SSQLite3Statement& operator=(SSQLite3Statement&&) = delete;

  int name2idx(const string& name)
  {
    string zName = string(":") + name;
    prepareStatement();
    return sqlite3_bind_parameter_index(d_stmt, zName.c_str());
    // XXX: support @ and $?
  }

  SSqlStatement* bind(const string& name, bool value) override
  {
    int idx = name2idx(name);
    if (idx > 0) {
      sqlite3_bind_int(d_stmt, idx, value ? 1 : 0);
    };
    return this;
  }

  SSqlStatement* bind(const string& name, int value) override
  {
    int idx = name2idx(name);
    if (idx > 0) {
      sqlite3_bind_int(d_stmt, idx, value);
    };
    return this;
  }

  SSqlStatement* bind(const string& name, uint32_t value) override
  {
    int idx = name2idx(name);
    if (idx > 0) {
      sqlite3_bind_int64(d_stmt, idx, value);
    };
    return this;
  }

  SSqlStatement* bind(const string& name, long value) override
  {
    int idx = name2idx(name);
    if (idx > 0) {
      sqlite3_bind_int64(d_stmt, idx, value);
    };
    return this;
  }

  SSqlStatement* bind(const string& name, unsigned long value) override
  {
    int idx = name2idx(name);
    if (idx > 0) {
      sqlite3_bind_int64(d_stmt, idx, static_cast<sqlite3_int64>(value));
    };
    return this;
  }

  SSqlStatement* bind(const string& name, long long value) override
  {
    int idx = name2idx(name);
    if (idx > 0) {
      sqlite3_bind_int64(d_stmt, idx, value);
    };
    return this;
  };

  SSqlStatement* bind(const string& name, unsigned long long value) override
  {
    int idx = name2idx(name);
    if (idx > 0) {
      sqlite3_bind_int64(d_stmt, idx, static_cast<sqlite3_int64>(value));
    };
    return this;
  }

  SSqlStatement* bind(const string& name, const std::string& value) override
  {
    int idx = name2idx(name);
    if (idx > 0) {
      sqlite3_bind_text(d_stmt, idx, value.c_str(), static_cast<int>(value.size()), SQLITE_TRANSIENT);
    };
    return this;
  }

  SSqlStatement* bindNull(const string& name) override
  {
    int idx = name2idx(name);
    if (idx > 0) {
      sqlite3_bind_null(d_stmt, idx);
    };
    return this;
  }

  SSqlStatement* execute() override
  {
    prepareStatement();
    if (d_dolog) {
      g_log << Logger::Warning << "Query " << this << ": " << d_query << endl;
      d_dtime.set();
    }

    int attempts = d_db->inTransaction() ? 1 : 0; // try only once
    while (attempts < 2 && (d_rc = sqlite3_step(d_stmt)) == SQLITE_BUSY) {
      attempts++;
    }

    if (d_rc != SQLITE_ROW && d_rc != SQLITE_DONE) {
      // failed.
      releaseStatement();
      if (d_rc == SQLITE_CANTOPEN) {
        throw SSqlException(string("CANTOPEN error in sqlite3, often caused by unwritable sqlite3 db *directory*: ") + SSQLite3ErrorString(d_db->db()));
      }
      throw SSqlException(string("Error while retrieving SQLite query results: ") + SSQLite3ErrorString(d_db->db()));
    }
    if (d_dolog) {
      g_log << Logger::Warning << "Query " << this << ": " << d_dtime.udiffNoReset() << " us to execute" << endl;
    }
    return this;
  }

  bool hasNextRow() override
  {
    if (d_dolog && d_rc != SQLITE_ROW) {
      g_log << Logger::Warning << "Query " << this << ": " << d_dtime.udiffNoReset() << " us total to last row" << endl;
    }
    return d_rc == SQLITE_ROW;
  }

  SSqlStatement* nextRow(row_t& row) override
  {
    row.clear();
    int numCols = sqlite3_column_count(d_stmt);
    row.reserve(numCols); // preallocate memory
    // Another row received, process it.
    for (int i = 0; i < numCols; i++) {
      if (sqlite3_column_type(d_stmt, i) == SQLITE_NULL) {
        row.emplace_back("");
      }
      else {
        // NOLINTNEXTLINE(cppcoreguidelines-pro-type-cstyle-cast): SQLite API returns unsigned char strings and std::strings are signed char.
        const char* pData = (const char*)sqlite3_column_text(d_stmt, i);
        row.emplace_back(pData, sqlite3_column_bytes(d_stmt, i));
      }
    }
    d_rc = sqlite3_step(d_stmt);
    return this;
  }

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

  SSqlStatement* reset() override
  {
    sqlite3_reset(d_stmt);
#if SQLITE_VERSION_NUMBER >= 3003009
    sqlite3_clear_bindings(d_stmt);
#else
    pdns_sqlite3_clear_bindings(d_stmt);
#endif
    return this;
  }

  ~SSQLite3Statement() override
  {
    // deallocate if necessary
    releaseStatement();
  }

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

private:
  string d_query;
  DTime d_dtime;
  sqlite3_stmt* d_stmt{nullptr};
  SSQLite3* d_db{nullptr};
  int d_rc{0};
  bool d_dolog;
  bool d_prepared{false};

  void prepareStatement()
  {
    const char* pTail = nullptr;

    if (d_prepared) {
      return;
    }
#if SQLITE_VERSION_NUMBER >= 3003009
    if (sqlite3_prepare_v2(d_db->db(), d_query.c_str(), -1, &d_stmt, &pTail) != SQLITE_OK)
#else
    if (sqlite3_prepare(d_db->db(), d_query.c_str(), -1, &d_stmt, &pTail) != SQLITE_OK)
#endif
    {
      releaseStatement();
      throw SSqlException(string("Unable to compile SQLite statement : '") + d_query + "': " + SSQLite3ErrorString(d_db->db()));
    }
    if ((pTail != nullptr) && strlen(pTail) > 0) {
      g_log << Logger::Warning << "Sqlite3 command partially processed. Unprocessed part: " << pTail << endl;
    }
    d_prepared = true;
  }

  void releaseStatement()
  {
    if (d_stmt != nullptr) {
      sqlite3_finalize(d_stmt);
    }
    d_stmt = nullptr;
    d_prepared = false;
  }
};

// Constructor.
SSQLite3::SSQLite3(const std::string& database, const std::string& journalmode, bool creat)
{
  if (access(database.c_str(), F_OK) == -1) {
    if (!creat) {
      throw SSqlException("SQLite database '" + database + "' does not exist yet");
    }
  }
  else {
    if (creat) {
      throw SSqlException("SQLite database '" + database + "' already exists");
    }
  }

  if (sqlite3_open(database.c_str(), &m_pDB) != SQLITE_OK) {
    throw SSqlException("Could not connect to the SQLite database '" + database + "'");
  }
  m_dolog = false;
  m_in_transaction = false;
  sqlite3_busy_handler(m_pDB, busyHandler, nullptr);

  if (journalmode.length() != 0) {
    executeImpl("PRAGMA journal_mode=" + journalmode);
  }
}

void SSQLite3::setLog(bool state)
{
  m_dolog = state;
}

// Destructor.
SSQLite3::~SSQLite3()
{
  for (int tried = 0;; ++tried) {
    int ret = sqlite3_close(m_pDB);
    if (ret == SQLITE_OK) {
      break;
    }
    cerr << "SQLite3 error state while tearing down database: " << SSQLite3ErrorString(m_pDB) << endl;
    if (tried == 0 && ret == SQLITE_BUSY) {
      continue;
    }
    cerr << "Unable to close down sqlite connection: " << ret << endl;
    abort();
  }
}

std::unique_ptr<SSqlStatement> SSQLite3::prepare(const string& query, int nparams __attribute__((unused)))
{
  return std::make_unique<SSQLite3Statement>(this, m_dolog, query);
}

void SSQLite3::executeImpl(const string& query)
{
  char* errmsg = nullptr;
  std::string errstr1;
  int execRet = sqlite3_exec(m_pDB, query.c_str(), nullptr, nullptr, &errmsg);
  if (execRet != SQLITE_OK) {
    errstr1 = errmsg;
    sqlite3_free(errmsg);
  }
  if (execRet == SQLITE_BUSY) {
    if (m_in_transaction) {
      throw SSqlException("Failed to execute query: " + errstr1);
    }
    execRet = sqlite3_exec(m_pDB, query.c_str(), nullptr, nullptr, &errmsg);
    std::string errstr2;
    if (execRet != SQLITE_OK) {
      errstr2 = errmsg;
      sqlite3_free(errmsg);
      throw SSqlException("Failed to execute query: " + errstr2);
    }
  }
  else if (execRet != SQLITE_OK) {
    throw SSqlException("Failed to execute query: " + errstr1);
  }
}

void SSQLite3::execute(const string& query)
{
  executeImpl(query);
}

int SSQLite3::busyHandler([[maybe_unused]] void* userData, [[maybe_unused]] int invocationsSoFar)
{
  Utility::usleep(1000);
  return 1;
}

void SSQLite3::startTransaction()
{
  execute("begin");
  m_in_transaction = true;
}

void SSQLite3::rollback()
{
  execute("rollback");
  m_in_transaction = false;
}

void SSQLite3::commit()
{
  execute("commit");
  m_in_transaction = false;
}

// Constructs a SSqlException object.
SSqlException SSQLite3::sPerrorException(const std::string& reason)
{
  return {reason};
}

1	/* SQLite backend for PowerDNS
2	* Copyright (C) 2003, Michel Stol <michel@powerdns.com>
3	*
4	* This program is free software; you can redistribute it and/or modify
5	* it under the terms of the GNU General Public License version 2
6	* as published by the Free Software Foundation.
7	*
8	* Additionally, the license of this program contains a special
9	* exception which allows to distribute the program in binary form when
10	* it is linked against OpenSSL.
11	*
12	* This program is distributed in the hope that it will be useful,
13	* but WITHOUT ANY WARRANTY; without even the implied warranty of
14	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15	* GNU General Public License for more details.
16	*
17	* You should have received a copy of the GNU General Public License
18	* along with this program; if not, write to the Free Software
19	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
20	*/
21
22	#ifdef HAVE_CONFIG_H
23	#include "config.h"
24	#endif
25	#include <string>
26	#include <sstream>
27	#include "ssqlite3.hh"
28	#include <iostream>
29	#include <fstream>
30	#include <utility>
31	#include "pdns/logger.hh"
32	#include "misc.hh"
33	#include "utility.hh"
34	#include <unistd.h>
35
36	/*
37	** Set all the parameters in the compiled SQL statement to NULL.
38	*
39	* copied from sqlite 3.3.6 // cmouse
40	*/
41	#if SQLITE_VERSION_NUMBER < 3003009
42	static int pdns_sqlite3_clear_bindings(sqlite3_stmt* pStmt)
43	{
44	int i;
45	int rc = SQLITE_OK;
46	for (i = 1; rc == SQLITE_OK && i <= sqlite3_bind_parameter_count(pStmt); i++) {
47	rc = sqlite3_bind_null(pStmt, i);
48	}
49	return rc;
50	}
51	#endif
52
53	static string SSQLite3ErrorString(sqlite3* database)
54	{	×
55	return string(sqlite3_errmsg(database) + string(" (") + std::to_string(sqlite3_extended_errcode(database)) + string(")"));	×
56	}	×
57
58	class SSQLite3Statement : public SSqlStatement
59	{
60	public:
61	SSQLite3Statement(SSQLite3* database, bool dolog, string query) :
62	d_query(std::move(query)),
63	d_db(database),
64	d_dolog(dolog)
65	{	143,632✔
66	}	143,632✔
67
68	SSQLite3Statement(const SSQLite3Statement&) = delete;
69	SSQLite3Statement(SSQLite3Statement&&) = delete;
70	SSQLite3Statement& operator=(const SSQLite3Statement&) = delete;
71	SSQLite3Statement& operator=(SSQLite3Statement&&) = delete;
72
73	int name2idx(const string& name)
74	{	1,759,072✔
75	string zName = string(":") + name;	1,759,072✔
76	prepareStatement();	1,759,072✔
77	return sqlite3_bind_parameter_index(d_stmt, zName.c_str());	1,759,072✔
78	// XXX: support @ and $?
79	}	1,759,072✔
80
81	SSqlStatement* bind(const string& name, bool value) override
82	{	387,525✔
83	int idx = name2idx(name);	387,525✔
84	if (idx > 0) {	387,525!
85	sqlite3_bind_int(d_stmt, idx, value ? 1 : 0);	387,525✔
86	};	387,525✔
87	return this;	387,525✔
88	}	387,525✔
89
90	SSqlStatement* bind(const string& name, int value) override
91	{	329,838✔
92	int idx = name2idx(name);	329,838✔
93	if (idx > 0) {	329,838!
94	sqlite3_bind_int(d_stmt, idx, value);	329,838✔
95	};	329,838✔
96	return this;	329,838✔
97	}	329,838✔
98
99	SSqlStatement* bind(const string& name, uint32_t value) override
100	{	233,471✔
101	int idx = name2idx(name);	233,471✔
102	if (idx > 0) {	233,471!
103	sqlite3_bind_int64(d_stmt, idx, value);	233,471✔
104	};	233,471✔
105	return this;	233,471✔
106	}	233,471✔
107
108	SSqlStatement* bind(const string& name, long value) override
109	{	66✔
110	int idx = name2idx(name);	66✔
111	if (idx > 0) {	66!
112	sqlite3_bind_int64(d_stmt, idx, value);	66✔
113	};	66✔
114	return this;	66✔
115	}	66✔
116
117	SSqlStatement* bind(const string& name, unsigned long value) override
118	{	11✔
119	int idx = name2idx(name);	11✔
120	if (idx > 0) {	11!
121	sqlite3_bind_int64(d_stmt, idx, static_cast<sqlite3_int64>(value));	11✔
122	};	11✔
123	return this;	11✔
124	}	11✔
125
126	SSqlStatement* bind(const string& name, long long value) override
127	{	×
128	int idx = name2idx(name);	×
129	if (idx > 0) {	×
130	sqlite3_bind_int64(d_stmt, idx, value);	×
131	};	×
132	return this;	×
133	};	×
134
135	SSqlStatement* bind(const string& name, unsigned long long value) override
136	{	×
137	int idx = name2idx(name);	×
138	if (idx > 0) {	×
139	sqlite3_bind_int64(d_stmt, idx, static_cast<sqlite3_int64>(value));	×
140	};	×
141	return this;	×
142	}	×
143
144	SSqlStatement* bind(const string& name, const std::string& value) override
145	{	685,840✔
146	int idx = name2idx(name);	685,840✔
147	if (idx > 0) {	685,840✔
148	sqlite3_bind_text(d_stmt, idx, value.c_str(), static_cast<int>(value.size()), SQLITE_TRANSIENT);	685,838✔
149	};	685,838✔
150	return this;	685,840✔
151	}	685,840✔
152
153	SSqlStatement* bindNull(const string& name) override
154	{	122,321✔
155	int idx = name2idx(name);	122,321✔
156	if (idx > 0) {	122,321!
157	sqlite3_bind_null(d_stmt, idx);	122,321✔
158	};	122,321✔
159	return this;	122,321✔
160	}	122,321✔
161
162	SSqlStatement* execute() override
163	{	250,228✔
164	prepareStatement();	250,228✔
165	if (d_dolog) {	250,228!
166	g_log << Logger::Warning << "Query " << this << ": " << d_query << endl;	×
167	d_dtime.set();	×
168	}	×
169
170	int attempts = d_db->inTransaction() ? 1 : 0; // try only once	250,228✔
171	while (attempts < 2 && (d_rc = sqlite3_step(d_stmt)) == SQLITE_BUSY) {	250,228!
172	attempts++;	×
173	}	×
174
175	if (d_rc != SQLITE_ROW && d_rc != SQLITE_DONE) {	250,228!
176	// failed.
177	releaseStatement();	×
178	if (d_rc == SQLITE_CANTOPEN) {	×
179	throw SSqlException(string("CANTOPEN error in sqlite3, often caused by unwritable sqlite3 db directory: ") + SSQLite3ErrorString(d_db->db()));	×
180	}	×
181	throw SSqlException(string("Error while retrieving SQLite query results: ") + SSQLite3ErrorString(d_db->db()));	×
182	}	×
183	if (d_dolog) {	250,228!
184	g_log << Logger::Warning << "Query " << this << ": " << d_dtime.udiffNoReset() << " us to execute" << endl;	×
185	}	×
186	return this;	250,228✔
187	}	250,228✔
188
189	bool hasNextRow() override
190	{	230,593✔
191	if (d_dolog && d_rc != SQLITE_ROW) {	230,593!
192	g_log << Logger::Warning << "Query " << this << ": " << d_dtime.udiffNoReset() << " us total to last row" << endl;	×
193	}	×
194	return d_rc == SQLITE_ROW;	230,593✔
195	}	230,593✔
196
197	SSqlStatement* nextRow(row_t& row) override
198	{	207,201✔
199	row.clear();	207,201✔
200	int numCols = sqlite3_column_count(d_stmt);	207,201✔
201	row.reserve(numCols); // preallocate memory	207,201✔
202	// Another row received, process it.
203	for (int i = 0; i < numCols; i++) {	1,977,579✔
204	if (sqlite3_column_type(d_stmt, i) == SQLITE_NULL) {	1,770,378✔
205	row.emplace_back("");	158,414✔
206	}	158,414✔
207	else {	1,611,964✔
208	// NOLINTNEXTLINE(cppcoreguidelines-pro-type-cstyle-cast): SQLite API returns unsigned char strings and std::strings are signed char.
209	const char* pData = (const char*)sqlite3_column_text(d_stmt, i);	1,611,964✔
210	row.emplace_back(pData, sqlite3_column_bytes(d_stmt, i));	1,611,964✔
211	}	1,611,964✔
212	}	1,770,378✔
213	d_rc = sqlite3_step(d_stmt);	207,201✔
214	return this;	207,201✔
215	}	207,201✔
216
217	SSqlStatement* getResult(result_t& result) override
218	{	1,138✔
219	result.clear();	1,138✔
220	while (hasNextRow()) {	2,146✔
221	row_t row;	1,008✔
222	nextRow(row);	1,008✔
223	result.push_back(std::move(row));	1,008✔
224	}	1,008✔
225	return this;	1,138✔
226	}	1,138✔
227
228	SSqlStatement* reset() override
229	{	250,228✔
230	sqlite3_reset(d_stmt);	250,228✔
231	#if SQLITE_VERSION_NUMBER >= 3003009	250,228✔
232	sqlite3_clear_bindings(d_stmt);	250,228✔
233	#else
234	pdns_sqlite3_clear_bindings(d_stmt);
235	#endif
236	return this;	250,228✔
237	}	250,228✔
238
239	~SSQLite3Statement() override
240	{	135,878✔
241	// deallocate if necessary
242	releaseStatement();	135,878✔
243	}	135,878✔
244
245	const string& getQuery() override { return d_query; };	×
246
247	private:
248	string d_query;
249	DTime d_dtime;
250	sqlite3_stmt* d_stmt{nullptr};
251	SSQLite3* d_db{nullptr};
252	int d_rc{0};
253	bool d_dolog;
254	bool d_prepared{false};
255
256	void prepareStatement()
257	{	2,009,299✔
258	const char* pTail = nullptr;	2,009,299✔
259
260	if (d_prepared) {	2,009,299✔
261	return;	2,001,249✔
262	}	2,001,249✔
263	#if SQLITE_VERSION_NUMBER >= 3003009	8,050✔
264	if (sqlite3_prepare_v2(d_db->db(), d_query.c_str(), -1, &d_stmt, &pTail) != SQLITE_OK)	8,050!
265	#else
266	if (sqlite3_prepare(d_db->db(), d_query.c_str(), -1, &d_stmt, &pTail) != SQLITE_OK)
267	#endif
268	{	×
269	releaseStatement();	×
270	throw SSqlException(string("Unable to compile SQLite statement : '") + d_query + "': " + SSQLite3ErrorString(d_db->db()));	×
271	}	×
272	if ((pTail != nullptr) && strlen(pTail) > 0) {	8,050!
273	g_log << Logger::Warning << "Sqlite3 command partially processed. Unprocessed part: " << pTail << endl;	×
274	}	×
275	d_prepared = true;	8,050✔
276	}	8,050✔
277
278	void releaseStatement()
279	{	135,878✔
280	if (d_stmt != nullptr) {	135,878✔
281	sqlite3_finalize(d_stmt);	7,590✔
282	}	7,590✔
283	d_stmt = nullptr;	135,878✔
284	d_prepared = false;	135,878✔
285	}	135,878✔
286	};
287
288	// Constructor.
289	SSQLite3::SSQLite3(const std::string& database, const std::string& journalmode, bool creat)
290	{	3,228✔
291	if (access(database.c_str(), F_OK) == -1) {	3,228✔
292	if (!creat) {	38!
293	throw SSqlException("SQLite database '" + database + "' does not exist yet");	×
294	}	×
295	}	38✔
296	else {	3,190✔
297	if (creat) {	3,190!
298	throw SSqlException("SQLite database '" + database + "' already exists");	×
299	}	×
300	}	3,190✔
301
302	if (sqlite3_open(database.c_str(), &m_pDB) != SQLITE_OK) {	3,228!
303	throw SSqlException("Could not connect to the SQLite database '" + database + "'");	×
304	}	×
305	m_dolog = false;	3,228✔
306	m_in_transaction = false;	3,228✔
307	sqlite3_busy_handler(m_pDB, busyHandler, nullptr);	3,228✔
308
309	if (journalmode.length() != 0) {	3,228✔
310	executeImpl("PRAGMA journal_mode=" + journalmode);	3,190✔
311	}	3,190✔
312	}	3,228✔
313
314	void SSQLite3::setLog(bool state)
315	{	3,189✔
316	m_dolog = state;	3,189✔
317	}	3,189✔
318
319	// Destructor.
320	SSQLite3::~SSQLite3()
321	{	3,046✔
322	for (int tried = 0;; ++tried) {	3,046✔
323	int ret = sqlite3_close(m_pDB);	3,046✔
324	if (ret == SQLITE_OK) {	3,046!
325	break;	3,046✔
326	}	3,046✔
UNCOV 327	cerr << "SQLite3 error state while tearing down database: " << SSQLite3ErrorString(m_pDB) << endl;	×
UNCOV 328	if (tried == 0 && ret == SQLITE_BUSY) {	×
329	continue;	×
330	}	×
UNCOV 331	cerr << "Unable to close down sqlite connection: " << ret << endl;	×
UNCOV 332	abort();	×
UNCOV 333	}	×
334	}	3,046✔
335
336	std::unique_ptr<SSqlStatement> SSQLite3::prepare(const string& query, int nparams __attribute__((unused)))
337	{	143,661✔
338	return std::make_unique<SSQLite3Statement>(this, m_dolog, query);	143,661✔
339	}	143,661✔
340
341	void SSQLite3::executeImpl(const string& query)
342	{	4,615✔
343	char* errmsg = nullptr;	4,615✔
344	std::string errstr1;	4,615✔
345	int execRet = sqlite3_exec(m_pDB, query.c_str(), nullptr, nullptr, &errmsg);	4,615✔
346	if (execRet != SQLITE_OK) {	4,615!
347	errstr1 = errmsg;	×
348	sqlite3_free(errmsg);	×
349	}	×
350	if (execRet == SQLITE_BUSY) {	4,615!
351	if (m_in_transaction) {	×
352	throw SSqlException("Failed to execute query: " + errstr1);	×
353	}	×
354	execRet = sqlite3_exec(m_pDB, query.c_str(), nullptr, nullptr, &errmsg);	×
355	std::string errstr2;	×
356	if (execRet != SQLITE_OK) {	×
357	errstr2 = errmsg;	×
358	sqlite3_free(errmsg);	×
359	throw SSqlException("Failed to execute query: " + errstr2);	×
360	}	×
361	}	×
362	else if (execRet != SQLITE_OK) {	4,615!
363	throw SSqlException("Failed to execute query: " + errstr1);	×
364	}	×
365	}	4,615✔
366
367	void SSQLite3::execute(const string& query)
368	{	1,425✔
369	executeImpl(query);	1,425✔
370	}	1,425✔
371
372	int SSQLite3::busyHandler([[maybe_unused]] void* userData, [[maybe_unused]] int invocationsSoFar)
373	{	17✔
374	Utility::usleep(1000);	17✔
375	return 1;	17✔
376	}	17✔
377
378	void SSQLite3::startTransaction()
379	{	435✔
380	execute("begin");	435✔
381	m_in_transaction = true;	435✔
382	}	435✔
383
384	void SSQLite3::rollback()
385	{	17✔
386	execute("rollback");	17✔
387	m_in_transaction = false;	17✔
388	}	17✔
389
390	void SSQLite3::commit()
391	{	415✔
392	execute("commit");	415✔
393	m_in_transaction = false;	415✔
394	}	415✔
395
396	// Constructs a SSqlException object.
397	SSqlException SSQLite3::sPerrorException(const std::string& reason)
398	{	×
399	return {reason};	×
400	}	×

PowerDNS / pdns / 9396932538

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