26141725099

Committed 19 May 2026 08:32PM UTC coverage: 89.343% (+0.009%) from 89.334%

Build # 26141725099

Build Type

push

github

Committed by

web-flow

Commit Message

Merge pull request #5609 from randombit/jack/improve-http

Improve the HTTP 1.0 client used for OCSP/CRL

Coverage Stats

109341 of 122383 relevant lines covered (89.34%)

11264402.07 hits per line

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

87.74

/src/lib/utils/sqlite3/sqlite3.cpp

/*
* SQLite wrapper
* (C) 2012 Jack Lloyd
*
* Botan is released under the Simplified BSD License (see license.txt)
*/

#include <botan/sqlite3.h>

#include <botan/exceptn.h>
#include <botan/mem_ops.h>
#include <botan/internal/fmt.h>
#include <botan/internal/int_utils.h>
#include <sqlite3.h>

namespace Botan {

Sqlite3_Database::Sqlite3_Database(std::string_view db_filename, std::optional<int> sqlite_open_flags) {
   // SQLITE_OPEN_FULLMUTEX ensures that the database object can be used
   // concurrently from multiple threads.
   const int open_flags =
      sqlite_open_flags.value_or(SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE | SQLITE_OPEN_FULLMUTEX);
   sqlite3* db = nullptr;
   const int rc = ::sqlite3_open_v2(std::string(db_filename).c_str(), &db, open_flags, nullptr);

   if(rc != 0) [[unlikely]] {
      const std::string err_msg = (db != nullptr) ? ::sqlite3_errmsg(db) : "unknown error";
      ::sqlite3_close_v2(db);
      throw SQL_DB_Error("sqlite3_open failed - " + err_msg);
   }

   m_db = std::shared_ptr<sqlite3>(db, [](sqlite3* p) noexcept { ::sqlite3_close_v2(p); });
}

Sqlite3_Database::~Sqlite3_Database() = default;

std::shared_ptr<SQL_Database::Statement> Sqlite3_Database::new_statement(std::string_view base_sql) const {
   return std::make_shared<Sqlite3_Statement>(m_db, base_sql);
}

std::shared_ptr<SQL_Database::Statement> Sqlite3_Database::upsert(
   std::string_view table, std::initializer_list<std::string_view> columns) const {
   BOTAN_ARG_CHECK(columns.size() > 0, "upsert requires at least one column");

   std::string sql = "INSERT OR REPLACE INTO ";
   sql += table;
   sql += " (";
   bool first = true;
   for(const auto& col : columns) {
      if(!first) {
         sql += ", ";
      }
      sql += col;
      first = false;
   }
   sql += ") VALUES (";
   for(size_t i = 1; i <= columns.size(); ++i) {
      if(i > 1) {
         sql += ", ";
      }
      sql += fmt("?{}", i);
   }
   sql += ")";

   return new_statement(sql);
}

size_t Sqlite3_Database::row_count(std::string_view table_name) {
   auto stmt = new_statement(fmt("select count(*) from {}", table_name));

   if(stmt->step()) {
      return stmt->get_size_t(0);
   } else {
      throw SQL_DB_Error(fmt("Querying size of table '{}' failed", table_name));
   }
}

void Sqlite3_Database::create_table(const Table_Schema& schema) {
   BOTAN_ARG_CHECK(!schema.name().empty(), "create_table requires a table name");
   BOTAN_ARG_CHECK(!schema.columns().empty(), "create_table requires at least one column");

   std::string sql = "CREATE TABLE ";
   if(schema.is_if_not_exists()) {
      sql += "IF NOT EXISTS ";
   }
   sql += schema.name();
   sql += " (";
   bool first = true;
   for(const auto& col : schema.columns()) {
      if(!first) {
         sql += ", ";
      }
      sql += col.name();
      sql += ' ';
      switch(col.type()) {
         case Column_Type::Blob:
            sql += "BLOB";
            break;
         case Column_Type::String:
            sql += "TEXT";
            break;
         case Column_Type::Integer:
            sql += "INTEGER";
            break;
      }
      if(col.is_primary_key()) {
         sql += " PRIMARY KEY";
      }
      if(col.is_unique()) {
         sql += " UNIQUE";
      }
      if(col.is_not_null()) {
         sql += " NOT NULL";
      }
      first = false;
   }
   sql += ")";

   char* errmsg = nullptr;
   const int rc = ::sqlite3_exec(m_db.get(), sql.c_str(), nullptr, nullptr, &errmsg);

   if(rc != SQLITE_OK) {
      const std::string err_msg = (errmsg != nullptr) ? errmsg : "unknown error";
      ::sqlite3_free(errmsg);
      throw SQL_DB_Error("sqlite3_exec for create_table failed - " + err_msg, rc);
   }
}

size_t Sqlite3_Database::rows_changed_by_last_statement() {
   const auto result = ::sqlite3_changes64(m_db.get());
   BOTAN_ASSERT_NOMSG(result >= 0);
   return static_cast<size_t>(result);
}

bool Sqlite3_Database::is_threadsafe() const {
   // sqlite3_db_mutex() returns the connection's mutex if the connection is in
   // serialized mode, and nullptr otherwise. This reflects both the compile-time
   // SQLITE_THREADSAFE setting and the per-connection SQLITE_OPEN_(FULL|NO)MUTEX
   // open flags actually used.
   //
   // https://www.sqlite.org/c3ref/db_mutex.html
   return ::sqlite3_db_mutex(m_db.get()) != nullptr;
}

Sqlite3_Database::Sqlite3_Statement::Sqlite3_Statement(std::shared_ptr<sqlite3> db, std::string_view base_sql) :
      m_db(std::move(db)), m_stmt{} {
   const int rc =
      ::sqlite3_prepare_v2(m_db.get(), base_sql.data(), static_cast<int>(base_sql.size()), &m_stmt, nullptr);

   if(rc != SQLITE_OK) {
      throw SQL_DB_Error(fmt("sqlite3_prepare failed on '{}' with err {}", base_sql, rc), rc);
   }
}

void Sqlite3_Database::Sqlite3_Statement::bind(int column, std::string_view val) {
   if(val.data() == nullptr) {
      bind_null(column);
      return;
   }
   const int rc = ::sqlite3_bind_text64(m_stmt, column, val.data(), val.size(), SQLITE_TRANSIENT, SQLITE_UTF8);
   if(rc != SQLITE_OK) {
      throw SQL_DB_Error("sqlite3_bind_text failed", rc);
   }
}

void Sqlite3_Database::Sqlite3_Statement::bind(int column, size_t val) {
   const int rc = ::sqlite3_bind_int64(m_stmt, column, val);
   if(rc != SQLITE_OK) {
      throw SQL_DB_Error("sqlite3_bind_int failed", rc);
   }
}

void Sqlite3_Database::Sqlite3_Statement::bind(int column, std::chrono::system_clock::time_point time) {
   const uint64_t timeval = std::chrono::duration_cast<std::chrono::seconds>(time.time_since_epoch()).count();
   bind(column, static_cast<size_t>(timeval));
}

void Sqlite3_Database::Sqlite3_Statement::bind(int column, const std::vector<uint8_t>& val) {
   bind(column, val.data(), val.size());
}

void Sqlite3_Database::Sqlite3_Statement::bind(int column, const uint8_t* p, size_t len) {
   if(p == nullptr) {
      bind_null(column);
      return;
   }
   const int rc = ::sqlite3_bind_blob64(m_stmt, column, p, len, SQLITE_TRANSIENT);
   if(rc != SQLITE_OK) {
      throw SQL_DB_Error("sqlite3_bind_blob failed", rc);
   }
}

void Sqlite3_Database::Sqlite3_Statement::bind_null(int column) {
   const int rc = ::sqlite3_bind_null(m_stmt, column);
   if(rc != SQLITE_OK) {
      throw SQL_DB_Error("sqlite3_bind_null failed", rc);
   }
}

std::span<const uint8_t> Sqlite3_Database::Sqlite3_Statement::get_blob(int column) {
   const auto column_type = ::sqlite3_column_type(m_stmt, column);
   if(column_type == SQLITE_NULL) {
      return {};
   }

   BOTAN_ASSERT(column_type == SQLITE_BLOB, "Return value is a blob");

   const void* session_blob = ::sqlite3_column_blob(m_stmt, column);
   const int session_blob_size = ::sqlite3_column_bytes(m_stmt, column);

   BOTAN_ASSERT(session_blob_size >= 0, "Blob size is non-negative");

   return {static_cast<const uint8_t*>(session_blob), static_cast<size_t>(session_blob_size)};
}

std::optional<std::string> Sqlite3_Database::Sqlite3_Statement::get_str(int column) {
   const auto column_type = ::sqlite3_column_type(m_stmt, column);
   if(column_type == SQLITE_NULL) {
      return std::nullopt;
   }

   BOTAN_ASSERT(column_type == SQLITE_TEXT, "Return value is text");

   const unsigned char* str = ::sqlite3_column_text(m_stmt, column);
   const int len = ::sqlite3_column_bytes(m_stmt, column);
   BOTAN_ASSERT(len >= 0, "Text length is non-negative");

   return std::string(cast_uint8_ptr_to_char(str), static_cast<size_t>(len));
}

size_t Sqlite3_Database::Sqlite3_Statement::get_size_t(int column) {
   BOTAN_ASSERT(::sqlite3_column_type(m_stmt, column) == SQLITE_INTEGER, "Return count is an integer");

   return checked_cast_to<size_t>(::sqlite3_column_int64(m_stmt, column));
}

size_t Sqlite3_Database::Sqlite3_Statement::spin() {
   size_t steps = 0;
   while(step()) {
      ++steps;
   }

   return steps;
}

bool Sqlite3_Database::Sqlite3_Statement::step() {
   const int rc = ::sqlite3_step(m_stmt);
   if(rc == SQLITE_ROW) {
      return true;
   }
   if(rc == SQLITE_DONE) {
      return false;
   }
   throw SQL_DB_Error(fmt("sqlite3_step failed - {}", ::sqlite3_errmsg(::sqlite3_db_handle(m_stmt))), rc);
}

Sqlite3_Database::Sqlite3_Statement::~Sqlite3_Statement() {
   ::sqlite3_finalize(m_stmt);
}

}  // namespace Botan

1	/*
2	* SQLite wrapper
3	* (C) 2012 Jack Lloyd
4	*
5	* Botan is released under the Simplified BSD License (see license.txt)
6	*/
7
8	#include <botan/sqlite3.h>
9
10	#include <botan/exceptn.h>
11	#include <botan/mem_ops.h>
12	#include <botan/internal/fmt.h>
13	#include <botan/internal/int_utils.h>
14	#include <sqlite3.h>
15
16	namespace Botan {
17
18	Sqlite3_Database::Sqlite3_Database(std::string_view db_filename, std::optional<int> sqlite_open_flags) {	34✔
19	// SQLITE_OPEN_FULLMUTEX ensures that the database object can be used
20	// concurrently from multiple threads.
21	const int open_flags =	34✔
22	sqlite_open_flags.value_or(SQLITE_OPEN_READWRITE \| SQLITE_OPEN_CREATE \| SQLITE_OPEN_FULLMUTEX);	34✔
23	sqlite3* db = nullptr;	34✔
24	const int rc = ::sqlite3_open_v2(std::string(db_filename).c_str(), &db, open_flags, nullptr);	68✔
25
26	if(rc != 0) [[unlikely]] {	34✔
27	const std::string err_msg = (db != nullptr) ? ::sqlite3_errmsg(db) : "unknown error";	×
28	::sqlite3_close_v2(db);	×
29	throw SQL_DB_Error("sqlite3_open failed - " + err_msg);	×
30	}	×
31
32	m_db = std::shared_ptr<sqlite3>(db, [](sqlite3* p) noexcept { ::sqlite3_close_v2(p); });	102✔
33	}	34✔
34
35	Sqlite3_Database::~Sqlite3_Database() = default;	34✔
36
37	std::shared_ptr<SQL_Database::Statement> Sqlite3_Database::new_statement(std::string_view base_sql) const {	868✔
38	return std::make_shared<Sqlite3_Statement>(m_db, base_sql);	868✔
39	}
40
41	std::shared_ptr<SQL_Database::Statement> Sqlite3_Database::upsert(	287✔
42	std::string_view table, std::initializer_list<std::string_view> columns) const {
43	BOTAN_ARG_CHECK(columns.size() > 0, "upsert requires at least one column");	287✔
44
45	std::string sql = "INSERT OR REPLACE INTO ";	287✔
46	sql += table;	287✔
47	sql += " (";	287✔
48	bool first = true;	287✔
49	for(const auto& col : columns) {	1,907✔
50	if(!first) {	1,620✔
51	sql += ", ";	1,333✔
52	}
53	sql += col;	1,620✔
54	first = false;	1,620✔
55	}
56	sql += ") VALUES (";	287✔
57	for(size_t i = 1; i <= columns.size(); ++i) {	1,907✔
58	if(i > 1) {	1,620✔
59	sql += ", ";	1,333✔
60	}
61	sql += fmt("?{}", i);	3,240✔
62	}
63	sql += ")";	287✔
64
65	return new_statement(sql);	287✔
66	}	287✔
67
68	size_t Sqlite3_Database::row_count(std::string_view table_name) {	22✔
69	auto stmt = new_statement(fmt("select count(*) from {}", table_name));	43✔
70
71	if(stmt->step()) {	1✔
72	return stmt->get_size_t(0);	1✔
73	} else {
74	throw SQL_DB_Error(fmt("Querying size of table '{}' failed", table_name));	×
75	}
76	}	1✔
77
78	void Sqlite3_Database::create_table(const Table_Schema& schema) {	65✔
79	BOTAN_ARG_CHECK(!schema.name().empty(), "create_table requires a table name");	65✔
80	BOTAN_ARG_CHECK(!schema.columns().empty(), "create_table requires at least one column");	65✔
81
82	std::string sql = "CREATE TABLE ";	65✔
83	if(schema.is_if_not_exists()) {	65✔
84	sql += "IF NOT EXISTS ";	21✔
85	}
86	sql += schema.name();	65✔
87	sql += " (";	65✔
88	bool first = true;	65✔
89	for(const auto& col : schema.columns()) {	365✔
90	if(!first) {	300✔
91	sql += ", ";	235✔
92	}
93	sql += col.name();	300✔
94	sql += ' ';	300✔
95	switch(col.type()) {	300✔
96	case Column_Type::Blob:	98✔
97	sql += "BLOB";	98✔
98	break;
99	case Column_Type::String:	84✔
100	sql += "TEXT";	84✔
101	break;
102	case Column_Type::Integer:	118✔
103	sql += "INTEGER";	118✔
104	break;
105	}
106	if(col.is_primary_key()) {	300✔
107	sql += " PRIMARY KEY";	43✔
108	}
109	if(col.is_unique()) {	300✔
110	sql += " UNIQUE";	×
111	}
112	if(col.is_not_null()) {	300✔
113	sql += " NOT NULL";	144✔
114	}
115	first = false;	300✔
116	}
117	sql += ")";	65✔
118
119	char* errmsg = nullptr;	65✔
120	const int rc = ::sqlite3_exec(m_db.get(), sql.c_str(), nullptr, nullptr, &errmsg);	65✔
121
122	if(rc != SQLITE_OK) {	65✔
123	const std::string err_msg = (errmsg != nullptr) ? errmsg : "unknown error";	×
124	::sqlite3_free(errmsg);	×
125	throw SQL_DB_Error("sqlite3_exec for create_table failed - " + err_msg, rc);	×
126	}	×
127	}	65✔
128
129	size_t Sqlite3_Database::rows_changed_by_last_statement() {	22✔
130	const auto result = ::sqlite3_changes64(m_db.get());	22✔
131	BOTAN_ASSERT_NOMSG(result >= 0);	22✔
132	return static_cast<size_t>(result);	22✔
133	}
134
135	bool Sqlite3_Database::is_threadsafe() const {	375✔
136	// sqlite3_db_mutex() returns the connection's mutex if the connection is in
137	// serialized mode, and nullptr otherwise. This reflects both the compile-time
138	// SQLITE_THREADSAFE setting and the per-connection SQLITE_OPEN_(FULL\|NO)MUTEX
139	// open flags actually used.
140	//
141	// https://www.sqlite.org/c3ref/db_mutex.html
142	return ::sqlite3_db_mutex(m_db.get()) != nullptr;	375✔
143	}
144
145	Sqlite3_Database::Sqlite3_Statement::Sqlite3_Statement(std::shared_ptr<sqlite3> db, std::string_view base_sql) :	868✔
146	m_db(std::move(db)), m_stmt{} {	868✔
147	const int rc =	868✔
148	::sqlite3_prepare_v2(m_db.get(), base_sql.data(), static_cast<int>(base_sql.size()), &m_stmt, nullptr);	868✔
149
150	if(rc != SQLITE_OK) {	868✔
151	throw SQL_DB_Error(fmt("sqlite3_prepare failed on '{}' with err {}", base_sql, rc), rc);	22✔
152	}
153	}	868✔
154
155	void Sqlite3_Database::Sqlite3_Statement::bind(int column, std::string_view val) {	757✔
156	if(val.data() == nullptr) {	757✔
157	bind_null(column);	×
158	return;	×
159	}
160	const int rc = ::sqlite3_bind_text64(m_stmt, column, val.data(), val.size(), SQLITE_TRANSIENT, SQLITE_UTF8);	757✔
161	if(rc != SQLITE_OK) {	757✔
162	throw SQL_DB_Error("sqlite3_bind_text failed", rc);	×
163	}
164	}
165
166	void Sqlite3_Database::Sqlite3_Statement::bind(int column, size_t val) {	965✔
167	const int rc = ::sqlite3_bind_int64(m_stmt, column, val);	965✔
168	if(rc != SQLITE_OK) {	965✔
169	throw SQL_DB_Error("sqlite3_bind_int failed", rc);	×
170	}
171	}	965✔
172
173	void Sqlite3_Database::Sqlite3_Statement::bind(int column, std::chrono::system_clock::time_point time) {	239✔
174	const uint64_t timeval = std::chrono::duration_cast<std::chrono::seconds>(time.time_since_epoch()).count();	239✔
175	bind(column, static_cast<size_t>(timeval));	239✔
176	}	239✔
177
178	void Sqlite3_Database::Sqlite3_Statement::bind(int column, const std::vector<uint8_t>& val) {	674✔
179	bind(column, val.data(), val.size());	674✔
180	}	674✔
181
182	void Sqlite3_Database::Sqlite3_Statement::bind(int column, const uint8_t* p, size_t len) {	680✔
183	if(p == nullptr) {	680✔
184	bind_null(column);	258✔
185	return;	258✔
186	}
187	const int rc = ::sqlite3_bind_blob64(m_stmt, column, p, len, SQLITE_TRANSIENT);	422✔
188	if(rc != SQLITE_OK) {	422✔
189	throw SQL_DB_Error("sqlite3_bind_blob failed", rc);	×
190	}
191	}
192
193	void Sqlite3_Database::Sqlite3_Statement::bind_null(int column) {	261✔
194	const int rc = ::sqlite3_bind_null(m_stmt, column);	261✔
195	if(rc != SQLITE_OK) {	261✔
196	throw SQL_DB_Error("sqlite3_bind_null failed", rc);	×
197	}
198	}	261✔
199
200	std::span<const uint8_t> Sqlite3_Database::Sqlite3_Statement::get_blob(int column) {	282✔
201	const auto column_type = ::sqlite3_column_type(m_stmt, column);	282✔
202	if(column_type == SQLITE_NULL) {	282✔
203	return {};	95✔
204	}
205
206	BOTAN_ASSERT(column_type == SQLITE_BLOB, "Return value is a blob");	187✔
207
208	const void* session_blob = ::sqlite3_column_blob(m_stmt, column);	187✔
209	const int session_blob_size = ::sqlite3_column_bytes(m_stmt, column);	187✔
210
211	BOTAN_ASSERT(session_blob_size >= 0, "Blob size is non-negative");	187✔
212
213	return {static_cast<const uint8_t*>(session_blob), static_cast<size_t>(session_blob_size)};	187✔
214	}
215
216	std::optional<std::string> Sqlite3_Database::Sqlite3_Statement::get_str(int column) {	127✔
217	const auto column_type = ::sqlite3_column_type(m_stmt, column);	127✔
218	if(column_type == SQLITE_NULL) {	127✔
219	return std::nullopt;	×
220	}
221
222	BOTAN_ASSERT(column_type == SQLITE_TEXT, "Return value is text");	127✔
223
224	const unsigned char* str = ::sqlite3_column_text(m_stmt, column);	127✔
225	const int len = ::sqlite3_column_bytes(m_stmt, column);	127✔
226	BOTAN_ASSERT(len >= 0, "Text length is non-negative");	127✔
227
228	return std::string(cast_uint8_ptr_to_char(str), static_cast<size_t>(len));	254✔
229	}
230
231	size_t Sqlite3_Database::Sqlite3_Statement::get_size_t(int column) {	6✔
232	BOTAN_ASSERT(::sqlite3_column_type(m_stmt, column) == SQLITE_INTEGER, "Return count is an integer");	6✔
233
234	return checked_cast_to<size_t>(::sqlite3_column_int64(m_stmt, column));	6✔
235	}
236
237	size_t Sqlite3_Database::Sqlite3_Statement::spin() {	631✔
238	size_t steps = 0;	631✔
239	while(step()) {	631✔
240	++steps;	×
241	}
242
243	return steps;	631✔
244	}
245
246	bool Sqlite3_Database::Sqlite3_Statement::step() {	1,001✔
247	const int rc = ::sqlite3_step(m_stmt);	1,001✔
248	if(rc == SQLITE_ROW) {	1,001✔
249	return true;
250	}
251	if(rc == SQLITE_DONE) {	794✔
252	return false;
253	}
254	throw SQL_DB_Error(fmt("sqlite3_step failed - {}", ::sqlite3_errmsg(::sqlite3_db_handle(m_stmt))), rc);	×
255	}
256
257	Sqlite3_Database::Sqlite3_Statement::~Sqlite3_Statement() {	846✔
258	::sqlite3_finalize(m_stmt);	846✔
259	}	846✔
260
261	} // namespace Botan

randombit / botan / 26141725099

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