#include "sql/statement.h"
#include <stddef.h>
#include <stdint.h>
#include "base/containers/span.h"
#include "base/dcheck_is_on.h"
#include "base/logging.h"
#include "base/numerics/safe_conversions.h"
#include "base/sequence_checker.h"
#include "base/strings/string_piece.h"
#include "base/strings/string_util.h"
#include "base/strings/utf_string_conversions.h"
#include "base/time/time.h"
#include "sql/sqlite_result_code.h"
#include "sql/sqlite_result_code_values.h"
#include "third_party/sqlite/sqlite3.h"
namespace sql {
Statement::Statement()
: ref_(base::MakeRefCounted<Database::StatementRef>(nullptr,
nullptr,
false)) {}
Statement::Statement(scoped_refptr<Database::StatementRef> ref)
: ref_(std::move(ref)) {}
Statement::~Statement() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
Reset(true);
}
void Statement::Assign(scoped_refptr<Database::StatementRef> ref) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
Reset(true);
ref_ = std::move(ref);
}
void Statement::Clear() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
Assign(base::MakeRefCounted<Database::StatementRef>(nullptr, nullptr, false));
succeeded_ = false;
}
bool Statement::CheckValid() const {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
DLOG_IF(FATAL, !ref_->was_valid())
<< "Cannot call mutating statements on an invalid statement.";
return is_valid();
}
SqliteResultCode Statement::StepInternal() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
if (!CheckValid())
return SqliteResultCode::kError;
absl::optional<base::ScopedBlockingCall> scoped_blocking_call;
ref_->InitScopedBlockingCall(FROM_HERE, &scoped_blocking_call);
auto sqlite_result_code = ToSqliteResultCode(sqlite3_step(ref_->stmt()));
return CheckSqliteResultCode(sqlite_result_code);
}
bool Statement::Run() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << "Run() must be called exactly once";
run_called_ = true;
DCHECK(!step_called_) << "Run() must not be mixed with Step()";
#endif
return StepInternal() == SqliteResultCode::kDone;
}
bool Statement::Step() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << "Run() must not be mixed with Step()";
step_called_ = true;
#endif
return StepInternal() == SqliteResultCode::kRow;
}
void Statement::Reset(bool clear_bound_vars) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
absl::optional<base::ScopedBlockingCall> scoped_blocking_call;
ref_->InitScopedBlockingCall(FROM_HERE, &scoped_blocking_call);
if (is_valid()) {
if (clear_bound_vars)
sqlite3_clear_bindings(ref_->stmt());
sqlite3_reset(ref_->stmt());
}
if (ref_->database())
ref_->database()->ReleaseCacheMemoryIfNeeded(false);
succeeded_ = false;
#if DCHECK_IS_ON()
run_called_ = false;
step_called_ = false;
#endif
}
bool Statement::Succeeded() const {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
return is_valid() && succeeded_;
}
void Statement::BindNull(int param_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " must not be called after Run()";
DCHECK(!step_called_) << __func__ << " must not be called after Step()";
#endif
if (!is_valid())
return;
DCHECK_GE(param_index, 0);
DCHECK_LT(param_index, sqlite3_bind_parameter_count(ref_->stmt()))
<< "Invalid parameter index";
int sqlite_result_code = sqlite3_bind_null(ref_->stmt(), param_index + 1);
DCHECK_EQ(sqlite_result_code, SQLITE_OK);
}
void Statement::BindBool(int param_index, bool val) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
return BindInt64(param_index, val ? 1 : 0);
}
void Statement::BindInt(int param_index, int val) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " must not be called after Run()";
DCHECK(!step_called_) << __func__ << " must not be called after Step()";
#endif
if (!is_valid())
return;
DCHECK_GE(param_index, 0);
DCHECK_LT(param_index, sqlite3_bind_parameter_count(ref_->stmt()))
<< "Invalid parameter index";
int sqlite_result_code = sqlite3_bind_int(ref_->stmt(), param_index + 1, val);
DCHECK_EQ(sqlite_result_code, SQLITE_OK);
}
void Statement::BindInt64(int param_index, int64_t val) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " must not be called after Run()";
DCHECK(!step_called_) << __func__ << " must not be called after Step()";
#endif
if (!is_valid())
return;
DCHECK_GE(param_index, 0);
DCHECK_LT(param_index, sqlite3_bind_parameter_count(ref_->stmt()))
<< "Invalid parameter index";
int sqlite_result_code =
sqlite3_bind_int64(ref_->stmt(), param_index + 1, val);
DCHECK_EQ(sqlite_result_code, SQLITE_OK);
}
void Statement::BindDouble(int param_index, double val) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " must not be called after Run()";
DCHECK(!step_called_) << __func__ << " must not be called after Step()";
#endif
if (!is_valid())
return;
DCHECK_GE(param_index, 0);
DCHECK_LT(param_index, sqlite3_bind_parameter_count(ref_->stmt()))
<< "Invalid parameter index";
int sqlite_result_code =
sqlite3_bind_double(ref_->stmt(), param_index + 1, val);
DCHECK_EQ(sqlite_result_code, SQLITE_OK);
}
void Statement::BindTime(int param_index, base::Time val) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " must not be called after Run()";
DCHECK(!step_called_) << __func__ << " must not be called after Step()";
#endif
if (!is_valid())
return;
DCHECK_GE(param_index, 0);
DCHECK_LT(param_index, sqlite3_bind_parameter_count(ref_->stmt()))
<< "Invalid parameter index";
int64_t int_value = val.ToDeltaSinceWindowsEpoch().InMicroseconds();
int sqlite_result_code =
sqlite3_bind_int64(ref_->stmt(), param_index + 1, int_value);
DCHECK_EQ(sqlite_result_code, SQLITE_OK);
}
void Statement::BindTimeDelta(int param_index, base::TimeDelta delta) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " must not be called after Run()";
DCHECK(!step_called_) << __func__ << " must not be called after Step()";
#endif
if (!is_valid()) {
return;
}
DCHECK_GE(param_index, 0);
DCHECK_LT(param_index, sqlite3_bind_parameter_count(ref_->stmt()))
<< "Invalid parameter index";
int64_t int_value = delta.InMicroseconds();
int sqlite_result_code =
sqlite3_bind_int64(ref_->stmt(), param_index + 1, int_value);
DCHECK_EQ(sqlite_result_code, SQLITE_OK);
}
void Statement::BindCString(int param_index, const char* val) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " must not be called after Run()";
DCHECK(!step_called_) << __func__ << " must not be called after Step()";
#endif
DCHECK(val);
if (!is_valid())
return;
DCHECK_GE(param_index, 0);
DCHECK_LT(param_index, sqlite3_bind_parameter_count(ref_->stmt()))
<< "Invalid parameter index";
int sqlite_result_code = sqlite3_bind_text(ref_->stmt(), param_index + 1, val,
-1, SQLITE_TRANSIENT);
DCHECK_EQ(sqlite_result_code, SQLITE_OK);
}
void Statement::BindString(int param_index, base::StringPiece value) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " must not be called after Run()";
DCHECK(!step_called_) << __func__ << " must not be called after Step()";
#endif
if (!is_valid())
return;
DCHECK_GE(param_index, 0);
DCHECK_LT(param_index, sqlite3_bind_parameter_count(ref_->stmt()))
<< "Invalid parameter index";
static constexpr char kEmptyPlaceholder[] = {0x00};
const char* data = (value.size() > 0) ? value.data() : kEmptyPlaceholder;
int sqlite_result_code = sqlite3_bind_text(
ref_->stmt(), param_index + 1, data, value.size(), SQLITE_TRANSIENT);
DCHECK_EQ(sqlite_result_code, SQLITE_OK);
}
void Statement::BindString16(int param_index, base::StringPiece16 value) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
return BindString(param_index, base::UTF16ToUTF8(value));
}
void Statement::BindBlob(int param_index, base::span<const uint8_t> value) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " must not be called after Run()";
DCHECK(!step_called_) << __func__ << " must not be called after Step()";
#endif
if (!is_valid())
return;
DCHECK_GE(param_index, 0);
DCHECK_LT(param_index, sqlite3_bind_parameter_count(ref_->stmt()))
<< "Invalid parameter index";
static constexpr uint8_t kEmptyPlaceholder[] = {0x00};
const uint8_t* data = (value.size() > 0) ? value.data() : kEmptyPlaceholder;
int sqlite_result_code = sqlite3_bind_blob(
ref_->stmt(), param_index + 1, data, value.size(), SQLITE_TRANSIENT);
DCHECK_EQ(sqlite_result_code, SQLITE_OK);
}
int Statement::ColumnCount() const {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
if (!is_valid())
return 0;
return sqlite3_column_count(ref_->stmt());
}
static_assert(static_cast<int>(ColumnType::kInteger) == SQLITE_INTEGER,
"INTEGER mismatch");
static_assert(static_cast<int>(ColumnType::kFloat) == SQLITE_FLOAT,
"FLOAT mismatch");
static_assert(static_cast<int>(ColumnType::kText) == SQLITE_TEXT,
"TEXT mismatch");
static_assert(static_cast<int>(ColumnType::kBlob) == SQLITE_BLOB,
"BLOB mismatch");
static_assert(static_cast<int>(ColumnType::kNull) == SQLITE_NULL,
"NULL mismatch");
ColumnType Statement::GetColumnType(int col) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
return static_cast<enum ColumnType>(sqlite3_column_type(ref_->stmt(), col));
}
bool Statement::ColumnBool(int column_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
return static_cast<bool>(ColumnInt64(column_index));
}
int Statement::ColumnInt(int column_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid())
return 0;
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
return sqlite3_column_int(ref_->stmt(), column_index);
}
int64_t Statement::ColumnInt64(int column_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid())
return 0;
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
return sqlite3_column_int64(ref_->stmt(), column_index);
}
double Statement::ColumnDouble(int column_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid())
return 0;
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
return sqlite3_column_double(ref_->stmt(), column_index);
}
base::Time Statement::ColumnTime(int column_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid())
return base::Time();
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
int64_t int_value = sqlite3_column_int64(ref_->stmt(), column_index);
return base::Time::FromDeltaSinceWindowsEpoch(base::Microseconds(int_value));
}
base::TimeDelta Statement::ColumnTimeDelta(int column_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid()) {
return base::TimeDelta();
}
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
int64_t int_value = sqlite3_column_int64(ref_->stmt(), column_index);
return base::Microseconds(int_value);
}
std::string Statement::ColumnString(int column_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid())
return std::string();
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
const char* string_buffer = reinterpret_cast<const char*>(
sqlite3_column_text(ref_->stmt(), column_index));
int size = sqlite3_column_bytes(ref_->stmt(), column_index);
std::string result;
if (string_buffer && size > 0)
result.assign(string_buffer, size);
return result;
}
std::u16string Statement::ColumnString16(int column_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid())
return std::u16string();
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
std::string string = ColumnString(column_index);
return string.empty() ? std::u16string() : base::UTF8ToUTF16(string);
}
base::span<const uint8_t> Statement::ColumnBlob(int column_index) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid())
return base::span<const uint8_t>();
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
int result_size = sqlite3_column_bytes(ref_->stmt(), column_index);
const void* result_buffer = sqlite3_column_blob(ref_->stmt(), column_index);
DCHECK(result_size == 0 || result_buffer != nullptr)
<< "sqlite3_column_blob() returned a null buffer for a non-empty BLOB";
return base::make_span(static_cast<const uint8_t*>(result_buffer),
base::checked_cast<size_t>(result_size));
}
bool Statement::ColumnBlobAsString(int column_index, std::string* result) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid())
return false;
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
const void* result_buffer = sqlite3_column_blob(ref_->stmt(), column_index);
int size = sqlite3_column_bytes(ref_->stmt(), column_index);
if (result_buffer && size > 0) {
result->assign(reinterpret_cast<const char*>(result_buffer), size);
} else {
result->clear();
}
return true;
}
bool Statement::ColumnBlobAsString16(int column_index, std::u16string* result) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
DCHECK(result);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid()) {
return false;
}
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
const void* result_buffer = sqlite3_column_blob(ref_->stmt(), column_index);
int size = sqlite3_column_bytes(ref_->stmt(), column_index);
if (result_buffer && size > 0) {
result->assign(reinterpret_cast<const char16_t*>(result_buffer), size / 2);
} else {
result->clear();
}
return true;
}
bool Statement::ColumnBlobAsVector(int column_index,
std::vector<char>* result) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
#if DCHECK_IS_ON()
DCHECK(!run_called_) << __func__ << " can be used after Step(), not Run()";
DCHECK(step_called_) << __func__ << " can only be used after Step()";
#endif
if (!CheckValid())
return false;
DCHECK_GE(column_index, 0);
DCHECK_LT(column_index, sqlite3_data_count(ref_->stmt()))
<< "Invalid column index";
const void* result_buffer = sqlite3_column_blob(ref_->stmt(), column_index);
int size = sqlite3_column_bytes(ref_->stmt(), column_index);
if (result_buffer && size > 0) {
result->assign(static_cast<const char*>(result_buffer),
static_cast<const char*>(result_buffer) + size);
} else {
result->clear();
}
return true;
}
bool Statement::ColumnBlobAsVector(int column_index,
std::vector<uint8_t>* result) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
return ColumnBlobAsVector(column_index,
reinterpret_cast<std::vector<char>*>(result));
}
std::string Statement::GetSQLStatement() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
return sqlite3_sql(ref_->stmt());
}
SqliteResultCode Statement::CheckSqliteResultCode(
SqliteResultCode sqlite_result_code) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
succeeded_ = IsSqliteSuccessCode(sqlite_result_code);
if (!succeeded_ && ref_.get() && ref_->database()) {
auto sqlite_error_code = ToSqliteErrorCode(sqlite_result_code);
ref_->database()->OnSqliteError(sqlite_error_code, this, nullptr);
}
return sqlite_result_code;
}
}