sql/statement.cc - chromium/src - Git at Google

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.

 #include "sql/statement.h"

 #include <stddef.h>
 #include <stdint.h>

 #include "base/logging.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "third_party/sqlite/sqlite3.h"

 namespace sql {

 // This empty constructor initializes our reference with an empty one so that
 // we don't have to NULL-check the ref_ to see if the statement is valid: we
 // only have to check the ref's validity bit.
 Statement::Statement()
     : ref_(new Connection::StatementRef(NULL, NULL, false)),
       stepped_(false),
       succeeded_(false) {
 }

 Statement::Statement(scoped_refptr<Connection::StatementRef> ref)
     : ref_(ref),
       stepped_(false),
       succeeded_(false) {
 }

 Statement::~Statement() {
   // Free the resources associated with this statement. We assume there's only
   // one statement active for a given sqlite3_stmt at any time, so this won't
   // mess with anything.
   Reset(true);
 }

 void Statement::Assign(scoped_refptr<Connection::StatementRef> ref) {
   Reset(true);
   ref_ = ref;
 }

 void Statement::Clear() {
   Assign(new Connection::StatementRef(NULL, NULL, false));
   succeeded_ = false;
 }

 bool Statement::CheckValid() const {
   // Allow operations to fail silently if a statement was invalidated
   // because the database was closed by an error handler.
   DLOG_IF(FATAL, !ref_->was_valid())
       << "Cannot call mutating statements on an invalid statement.";
   return is_valid();
 }

 int Statement::StepInternal(bool timer_flag) {
   ref_->AssertIOAllowed();
   if (!CheckValid())
     return SQLITE_ERROR;

   const bool was_stepped = stepped_;
   stepped_ = true;
   int ret = SQLITE_ERROR;
   if (!ref_->connection()) {
     ret = sqlite3_step(ref_->stmt());
   } else {
     if (!timer_flag) {
       ret = sqlite3_step(ref_->stmt());
     } else {
       const base::TimeTicks before = ref_->connection()->Now();
       ret = sqlite3_step(ref_->stmt());
       const base::TimeTicks after = ref_->connection()->Now();
       const bool read_only = !!sqlite3_stmt_readonly(ref_->stmt());
       ref_->connection()->RecordTimeAndChanges(after - before, read_only);
     }

     if (!was_stepped)
       ref_->connection()->RecordOneEvent(Connection::EVENT_STATEMENT_RUN);

     if (ret == SQLITE_ROW)
       ref_->connection()->RecordOneEvent(Connection::EVENT_STATEMENT_ROWS);
   }
   return CheckError(ret);
 }

 bool Statement::Run() {
   DCHECK(!stepped_);
   return StepInternal(true) == SQLITE_DONE;
 }

 bool Statement::RunWithoutTimers() {
   DCHECK(!stepped_);
   return StepInternal(false) == SQLITE_DONE;
 }

 bool Statement::Step() {
   return StepInternal(true) == SQLITE_ROW;
 }

 void Statement::Reset(bool clear_bound_vars) {
   ref_->AssertIOAllowed();
   if (is_valid()) {
     if (clear_bound_vars)
       sqlite3_clear_bindings(ref_->stmt());

     // StepInternal() cannot track success because statements may be reset
     // before reaching SQLITE_DONE.  Don't call CheckError() because
     // sqlite3_reset() returns the last step error, which StepInternal() already
     // checked.
     const int rc =sqlite3_reset(ref_->stmt());
     if (rc == SQLITE_OK && ref_->connection())
       ref_->connection()->RecordOneEvent(Connection::EVENT_STATEMENT_SUCCESS);
   }

   // Potentially release dirty cache pages if an autocommit statement made
   // changes.
   if (ref_->connection())
     ref_->connection()->ReleaseCacheMemoryIfNeeded(false);

   succeeded_ = false;
   stepped_ = false;
 }

 bool Statement::Succeeded() const {
   if (!is_valid())
     return false;

   return succeeded_;
 }

 bool Statement::BindNull(int col) {
   DCHECK(!stepped_);
   if (!is_valid())
     return false;

   return CheckOk(sqlite3_bind_null(ref_->stmt(), col + 1));
 }

 bool Statement::BindBool(int col, bool val) {
   return BindInt(col, val ? 1 : 0);
 }

 bool Statement::BindInt(int col, int val) {
   DCHECK(!stepped_);
   if (!is_valid())
     return false;

   return CheckOk(sqlite3_bind_int(ref_->stmt(), col + 1, val));
 }

 bool Statement::BindInt64(int col, int64_t val) {
   DCHECK(!stepped_);
   if (!is_valid())
     return false;

   return CheckOk(sqlite3_bind_int64(ref_->stmt(), col + 1, val));
 }

 bool Statement::BindDouble(int col, double val) {
   DCHECK(!stepped_);
   if (!is_valid())
     return false;

   return CheckOk(sqlite3_bind_double(ref_->stmt(), col + 1, val));
 }

 bool Statement::BindCString(int col, const char* val) {
   DCHECK(!stepped_);
   if (!is_valid())
     return false;

   return CheckOk(
       sqlite3_bind_text(ref_->stmt(), col + 1, val, -1, SQLITE_TRANSIENT));
 }

 bool Statement::BindString(int col, const std::string& val) {
   DCHECK(!stepped_);
   if (!is_valid())
     return false;

   return CheckOk(sqlite3_bind_text(ref_->stmt(),
                                    col + 1,
                                    val.data(),
                                    val.size(),
                                    SQLITE_TRANSIENT));
 }

 bool Statement::BindString16(int col, const base::string16& value) {
   return BindString(col, base::UTF16ToUTF8(value));
 }

 bool Statement::BindBlob(int col, const void* val, int val_len) {
   DCHECK(!stepped_);
   if (!is_valid())
     return false;

   return CheckOk(
       sqlite3_bind_blob(ref_->stmt(), col + 1, val, val_len, SQLITE_TRANSIENT));
 }

 int Statement::ColumnCount() const {
   if (!is_valid())
     return 0;

   return sqlite3_column_count(ref_->stmt());
 }

 ColType Statement::ColumnType(int col) const {
   // Verify that our enum matches sqlite's values.
   static_assert(COLUMN_TYPE_INTEGER == SQLITE_INTEGER, "integer no match");
   static_assert(COLUMN_TYPE_FLOAT == SQLITE_FLOAT, "float no match");
   static_assert(COLUMN_TYPE_TEXT == SQLITE_TEXT, "integer no match");
   static_assert(COLUMN_TYPE_BLOB == SQLITE_BLOB, "blob no match");
   static_assert(COLUMN_TYPE_NULL == SQLITE_NULL, "null no match");

   return static_cast<ColType>(sqlite3_column_type(ref_->stmt(), col));
 }

 ColType Statement::DeclaredColumnType(int col) const {
   std::string column_type = base::ToLowerASCII(
       sqlite3_column_decltype(ref_->stmt(), col));

   if (column_type == "integer")
     return COLUMN_TYPE_INTEGER;
   else if (column_type == "float")
     return COLUMN_TYPE_FLOAT;
   else if (column_type == "text")
     return COLUMN_TYPE_TEXT;
   else if (column_type == "blob")
     return COLUMN_TYPE_BLOB;

   return COLUMN_TYPE_NULL;
 }

 bool Statement::ColumnBool(int col) const {
   return !!ColumnInt(col);
 }

 int Statement::ColumnInt(int col) const {
   if (!CheckValid())
     return 0;

   return sqlite3_column_int(ref_->stmt(), col);
 }

 int64_t Statement::ColumnInt64(int col) const {
   if (!CheckValid())
     return 0;

   return sqlite3_column_int64(ref_->stmt(), col);
 }

 double Statement::ColumnDouble(int col) const {
   if (!CheckValid())
     return 0;

   return sqlite3_column_double(ref_->stmt(), col);
 }

 std::string Statement::ColumnString(int col) const {
   if (!CheckValid())
     return std::string();

   const char* str = reinterpret_cast<const char*>(
       sqlite3_column_text(ref_->stmt(), col));
   int len = sqlite3_column_bytes(ref_->stmt(), col);

   std::string result;
   if (str && len > 0)
     result.assign(str, len);
   return result;
 }

 base::string16 Statement::ColumnString16(int col) const {
   if (!CheckValid())
     return base::string16();

   std::string s = ColumnString(col);
   return !s.empty() ? base::UTF8ToUTF16(s) : base::string16();
 }

 int Statement::ColumnByteLength(int col) const {
   if (!CheckValid())
     return 0;

   return sqlite3_column_bytes(ref_->stmt(), col);
 }

 const void* Statement::ColumnBlob(int col) const {
   if (!CheckValid())
     return NULL;

   return sqlite3_column_blob(ref_->stmt(), col);
 }

 bool Statement::ColumnBlobAsString(int col, std::string* blob) {
   if (!CheckValid())
     return false;

   const void* p = ColumnBlob(col);
   size_t len = ColumnByteLength(col);
   blob->resize(len);
   if (blob->size() != len) {
     return false;
   }
   blob->assign(reinterpret_cast<const char*>(p), len);
   return true;
 }

 bool Statement::ColumnBlobAsString16(int col, base::string16* val) const {
   if (!CheckValid())
     return false;

   const void* data = ColumnBlob(col);
   size_t len = ColumnByteLength(col) / sizeof(base::char16);
   val->resize(len);
   if (val->size() != len)
     return false;
   val->assign(reinterpret_cast<const base::char16*>(data), len);
   return true;
 }

 bool Statement::ColumnBlobAsVector(int col, std::vector<char>* val) const {
   val->clear();

   if (!CheckValid())
     return false;

   const void* data = sqlite3_column_blob(ref_->stmt(), col);
   int len = sqlite3_column_bytes(ref_->stmt(), col);
   if (data && len > 0) {
     val->resize(len);
     memcpy(&(*val)[0], data, len);
   }
   return true;
 }

 bool Statement::ColumnBlobAsVector(
     int col,
     std::vector<unsigned char>* val) const {
   return ColumnBlobAsVector(col, reinterpret_cast< std::vector<char>* >(val));
 }

 const char* Statement::GetSQLStatement() {
   return sqlite3_sql(ref_->stmt());
 }

 bool Statement::CheckOk(int err) const {
   // Binding to a non-existent variable is evidence of a serious error.
   // TODO(gbillock,shess): make this invalidate the statement so it
   // can't wreak havoc.
   if (err == SQLITE_RANGE)
     DLOG(FATAL) << "Bind value out of range";
   return err == SQLITE_OK;
 }

 int Statement::CheckError(int err) {
   // Please don't add DCHECKs here, OnSqliteError() already has them.
   succeeded_ = (err == SQLITE_OK || err == SQLITE_ROW || err == SQLITE_DONE);
   if (!succeeded_ && ref_.get() && ref_->connection())
     return ref_->connection()->OnSqliteError(err, this, NULL);
   return err;
 }

 }  // namespace sql
	// Copyright (c) 2012 The Chromium Authors. All rights reserved.
	// Use of this source code is governed by a BSD-style license that can be
	// found in the LICENSE file.

	#include "sql/statement.h"

	#include <stddef.h>
	#include <stdint.h>

	#include "base/logging.h"
	#include "base/strings/string_util.h"
	#include "base/strings/utf_string_conversions.h"
	#include "third_party/sqlite/sqlite3.h"

	namespace sql {

	// This empty constructor initializes our reference with an empty one so that
	// we don't have to NULL-check the ref_ to see if the statement is valid: we
	// only have to check the ref's validity bit.
	Statement::Statement()
	: ref_(new Connection::StatementRef(NULL, NULL, false)),
	stepped_(false),
	succeeded_(false) {
	}

	Statement::Statement(scoped_refptr<Connection::StatementRef> ref)
	: ref_(ref),
	stepped_(false),
	succeeded_(false) {
	}

	Statement::~Statement() {
	// Free the resources associated with this statement. We assume there's only
	// one statement active for a given sqlite3_stmt at any time, so this won't
	// mess with anything.
	Reset(true);
	}

	void Statement::Assign(scoped_refptr<Connection::StatementRef> ref) {
	Reset(true);
	ref_ = ref;
	}

	void Statement::Clear() {
	Assign(new Connection::StatementRef(NULL, NULL, false));
	succeeded_ = false;
	}

	bool Statement::CheckValid() const {
	// Allow operations to fail silently if a statement was invalidated
	// because the database was closed by an error handler.
	DLOG_IF(FATAL, !ref_->was_valid())
	<< "Cannot call mutating statements on an invalid statement.";
	return is_valid();
	}

	int Statement::StepInternal(bool timer_flag) {
	ref_->AssertIOAllowed();
	if (!CheckValid())
	return SQLITE_ERROR;

	const bool was_stepped = stepped_;
	stepped_ = true;
	int ret = SQLITE_ERROR;
	if (!ref_->connection()) {
	ret = sqlite3_step(ref_->stmt());
	} else {
	if (!timer_flag) {
	ret = sqlite3_step(ref_->stmt());
	} else {
	const base::TimeTicks before = ref_->connection()->Now();
	ret = sqlite3_step(ref_->stmt());
	const base::TimeTicks after = ref_->connection()->Now();
	const bool read_only = !!sqlite3_stmt_readonly(ref_->stmt());
	ref_->connection()->RecordTimeAndChanges(after - before, read_only);
	}

	if (!was_stepped)
	ref_->connection()->RecordOneEvent(Connection::EVENT_STATEMENT_RUN);

	if (ret == SQLITE_ROW)
	ref_->connection()->RecordOneEvent(Connection::EVENT_STATEMENT_ROWS);
	}
	return CheckError(ret);
	}

	bool Statement::Run() {
	DCHECK(!stepped_);
	return StepInternal(true) == SQLITE_DONE;
	}

	bool Statement::RunWithoutTimers() {
	DCHECK(!stepped_);
	return StepInternal(false) == SQLITE_DONE;
	}

	bool Statement::Step() {
	return StepInternal(true) == SQLITE_ROW;
	}

	void Statement::Reset(bool clear_bound_vars) {
	ref_->AssertIOAllowed();
	if (is_valid()) {
	if (clear_bound_vars)
	sqlite3_clear_bindings(ref_->stmt());

	// StepInternal() cannot track success because statements may be reset
	// before reaching SQLITE_DONE. Don't call CheckError() because
	// sqlite3_reset() returns the last step error, which StepInternal() already
	// checked.
	const int rc =sqlite3_reset(ref_->stmt());
	if (rc == SQLITE_OK && ref_->connection())
	ref_->connection()->RecordOneEvent(Connection::EVENT_STATEMENT_SUCCESS);
	}

	// Potentially release dirty cache pages if an autocommit statement made
	// changes.
	if (ref_->connection())
	ref_->connection()->ReleaseCacheMemoryIfNeeded(false);

	succeeded_ = false;
	stepped_ = false;
	}

	bool Statement::Succeeded() const {
	if (!is_valid())
	return false;

	return succeeded_;
	}

	bool Statement::BindNull(int col) {
	DCHECK(!stepped_);
	if (!is_valid())
	return false;

	return CheckOk(sqlite3_bind_null(ref_->stmt(), col + 1));
	}

	bool Statement::BindBool(int col, bool val) {
	return BindInt(col, val ? 1 : 0);
	}

	bool Statement::BindInt(int col, int val) {
	DCHECK(!stepped_);
	if (!is_valid())
	return false;

	return CheckOk(sqlite3_bind_int(ref_->stmt(), col + 1, val));
	}

	bool Statement::BindInt64(int col, int64_t val) {
	DCHECK(!stepped_);
	if (!is_valid())
	return false;

	return CheckOk(sqlite3_bind_int64(ref_->stmt(), col + 1, val));
	}

	bool Statement::BindDouble(int col, double val) {
	DCHECK(!stepped_);
	if (!is_valid())
	return false;

	return CheckOk(sqlite3_bind_double(ref_->stmt(), col + 1, val));
	}

	bool Statement::BindCString(int col, const char* val) {
	DCHECK(!stepped_);
	if (!is_valid())
	return false;

	return CheckOk(
	sqlite3_bind_text(ref_->stmt(), col + 1, val, -1, SQLITE_TRANSIENT));
	}

	bool Statement::BindString(int col, const std::string& val) {
	DCHECK(!stepped_);
	if (!is_valid())
	return false;

	return CheckOk(sqlite3_bind_text(ref_->stmt(),
	col + 1,
	val.data(),
	val.size(),
	SQLITE_TRANSIENT));
	}

	bool Statement::BindString16(int col, const base::string16& value) {
	return BindString(col, base::UTF16ToUTF8(value));
	}

	bool Statement::BindBlob(int col, const void* val, int val_len) {
	DCHECK(!stepped_);
	if (!is_valid())
	return false;

	return CheckOk(
	sqlite3_bind_blob(ref_->stmt(), col + 1, val, val_len, SQLITE_TRANSIENT));
	}

	int Statement::ColumnCount() const {
	if (!is_valid())
	return 0;

	return sqlite3_column_count(ref_->stmt());
	}

	ColType Statement::ColumnType(int col) const {
	// Verify that our enum matches sqlite's values.
	static_assert(COLUMN_TYPE_INTEGER == SQLITE_INTEGER, "integer no match");
	static_assert(COLUMN_TYPE_FLOAT == SQLITE_FLOAT, "float no match");
	static_assert(COLUMN_TYPE_TEXT == SQLITE_TEXT, "integer no match");
	static_assert(COLUMN_TYPE_BLOB == SQLITE_BLOB, "blob no match");
	static_assert(COLUMN_TYPE_NULL == SQLITE_NULL, "null no match");

	return static_cast<ColType>(sqlite3_column_type(ref_->stmt(), col));
	}

	ColType Statement::DeclaredColumnType(int col) const {
	std::string column_type = base::ToLowerASCII(
	sqlite3_column_decltype(ref_->stmt(), col));

	if (column_type == "integer")
	return COLUMN_TYPE_INTEGER;
	else if (column_type == "float")
	return COLUMN_TYPE_FLOAT;
	else if (column_type == "text")
	return COLUMN_TYPE_TEXT;
	else if (column_type == "blob")
	return COLUMN_TYPE_BLOB;

	return COLUMN_TYPE_NULL;
	}

	bool Statement::ColumnBool(int col) const {
	return !!ColumnInt(col);
	}

	int Statement::ColumnInt(int col) const {
	if (!CheckValid())
	return 0;

	return sqlite3_column_int(ref_->stmt(), col);
	}

	int64_t Statement::ColumnInt64(int col) const {
	if (!CheckValid())
	return 0;

	return sqlite3_column_int64(ref_->stmt(), col);
	}

	double Statement::ColumnDouble(int col) const {
	if (!CheckValid())
	return 0;

	return sqlite3_column_double(ref_->stmt(), col);
	}

	std::string Statement::ColumnString(int col) const {
	if (!CheckValid())
	return std::string();

	const char* str = reinterpret_cast<const char*>(
	sqlite3_column_text(ref_->stmt(), col));
	int len = sqlite3_column_bytes(ref_->stmt(), col);

	std::string result;
	if (str && len > 0)
	result.assign(str, len);
	return result;
	}

	base::string16 Statement::ColumnString16(int col) const {
	if (!CheckValid())
	return base::string16();

	std::string s = ColumnString(col);
	return !s.empty() ? base::UTF8ToUTF16(s) : base::string16();
	}

	int Statement::ColumnByteLength(int col) const {
	if (!CheckValid())
	return 0;

	return sqlite3_column_bytes(ref_->stmt(), col);
	}

	const void* Statement::ColumnBlob(int col) const {
	if (!CheckValid())
	return NULL;

	return sqlite3_column_blob(ref_->stmt(), col);
	}

	bool Statement::ColumnBlobAsString(int col, std::string* blob) {
	if (!CheckValid())
	return false;

	const void* p = ColumnBlob(col);
	size_t len = ColumnByteLength(col);
	blob->resize(len);
	if (blob->size() != len) {
	return false;
	}
	blob->assign(reinterpret_cast<const char*>(p), len);
	return true;
	}

	bool Statement::ColumnBlobAsString16(int col, base::string16* val) const {
	if (!CheckValid())
	return false;

	const void* data = ColumnBlob(col);
	size_t len = ColumnByteLength(col) / sizeof(base::char16);
	val->resize(len);
	if (val->size() != len)
	return false;
	val->assign(reinterpret_cast<const base::char16*>(data), len);
	return true;
	}

	bool Statement::ColumnBlobAsVector(int col, std::vector<char>* val) const {
	val->clear();

	if (!CheckValid())
	return false;

	const void* data = sqlite3_column_blob(ref_->stmt(), col);
	int len = sqlite3_column_bytes(ref_->stmt(), col);
	if (data && len > 0) {
	val->resize(len);
	memcpy(&(*val)[0], data, len);
	}
	return true;
	}

	bool Statement::ColumnBlobAsVector(
	int col,
	std::vector<unsigned char>* val) const {
	return ColumnBlobAsVector(col, reinterpret_cast< std::vector<char>* >(val));
	}

	const char* Statement::GetSQLStatement() {
	return sqlite3_sql(ref_->stmt());
	}

	bool Statement::CheckOk(int err) const {
	// Binding to a non-existent variable is evidence of a serious error.
	// TODO(gbillock,shess): make this invalidate the statement so it
	// can't wreak havoc.
	if (err == SQLITE_RANGE)
	DLOG(FATAL) << "Bind value out of range";
	return err == SQLITE_OK;
	}

	int Statement::CheckError(int err) {
	// Please don't add DCHECKs here, OnSqliteError() already has them.
	succeeded_ = (err == SQLITE_OK \|\| err == SQLITE_ROW \|\| err == SQLITE_DONE);
	if (!succeeded_ && ref_.get() && ref_->connection())
	return ref_->connection()->OnSqliteError(err, this, NULL);
	return err;
	}

	} // namespace sql