QT sqlite加密数据库引擎

#include "qsql_sqliteex_p.h" #include <qcoreapplication.h> #include <qdatetime.h> #include <qvariant.h> #include <qsqlerror.h> #include <qsqlfield.h> #include <qsqlindex.h> #include <qsqlquery.h> #include <qstringlist.h> #include <qvector.h> #include <qdebug.h> #ifndef QT_NO_REGULAREXPRESSION #include <qcache.h> #include <qregularexpression.h> #endif #include <QTimeZone> #if defined Q_OS_WIN # include <qt_windows.h> #else # include <unistd.h> #endif #include "sqlite3.h" #include <functional> #include <QtSql/private/qsqlcachedresult_p.h> #include <QtSql/private/qsqldriver_p.h> Q_DECLARE_OPAQUE_POINTER(sqlite3*) Q_DECLARE_METATYPE(sqlite3*) Q_DECLARE_OPAQUE_POINTER(sqlite3_stmt*) Q_DECLARE_METATYPE(sqlite3_stmt*) QT_BEGIN_NAMESPACE static QString _q_escapeIdentifier(const QString &identifier) { QString res = identifier; if (!identifier.isEmpty() && !identifier.startsWith(QLatin1Char('"')) && !identifier.endsWith(QLatin1Char('"'))) { res.replace(QLatin1Char('"'), QLatin1String("\"\"")); res.prepend(QLatin1Char('"')).append(QLatin1Char('"')); res.replace(QLatin1Char('.'), QLatin1String("\".\"")); } return res; } static QVariant::Type qGetColumnType(const QString &tpName) { const QString typeName = tpName.toLower(); if (typeName == QLatin1String("integer") || typeName == QLatin1String("int")) return QVariant::Int; if (typeName == QLatin1String("double") || typeName == QLatin1String("float") || typeName == QLatin1String("real") || typeName.startsWith(QLatin1String("numeric"))) return QVariant::Double; if (typeName == QLatin1String("blob")) return QVariant::ByteArray; if (typeName == QLatin1String("boolean") || typeName == QLatin1String("bool")) return QVariant::Bool; return QVariant::String; } static QSqlError qMakeError(sqlite3 *access, const QString &descr, QSqlError::ErrorType type, int errorCode = -1) { return QSqlError(descr, QString(reinterpret_cast<const QChar *>(sqlite3_errmsg16(access))), type, QString::number(errorCode)); } class QSQLiteExResultPrivate; class QSQLiteExResult : public QSqlCachedResult { Q_DECLARE_PRIVATE(QSQLiteExResult) friend class QSQLiteExDriver; public: explicit QSQLiteExResult(const QSQLiteExDriver* db); ~QSQLiteExResult(); QVariant handle() const Q_DECL_OVERRIDE; protected: bool gotoNext(QSqlCachedResult::ValueCache& row, int idx) Q_DECL_OVERRIDE; bool reset(const QString &query) Q_DECL_OVERRIDE; bool prepare(const QString &query) Q_DECL_OVERRIDE; bool exec() Q_DECL_OVERRIDE; int size() Q_DECL_OVERRIDE; int numRowsAffected() Q_DECL_OVERRIDE; QVariant lastInsertId() const Q_DECL_OVERRIDE; QSqlRecord record() const Q_DECL_OVERRIDE; void detachFromResultSet() Q_DECL_OVERRIDE; void virtual_hook(int id, void *data) Q_DECL_OVERRIDE; }; class QSQLiteExDriverPrivate : public QSqlDriverPrivate { Q_DECLARE_PUBLIC(QSQLiteExDriver) public: inline QSQLiteExDriverPrivate() : QSqlDriverPrivate(), access(0) { dbmsType = QSqlDriver::SQLite; } sqlite3 *access; QList <QSQLiteExResult *> results; QStringList notificationid; }; class QSQLiteExResultPrivate: public QSqlCachedResultPrivate { Q_DECLARE_PUBLIC(QSQLiteExResult) public: Q_DECLARE_SQLDRIVER_PRIVATE(QSQLiteExDriver) QSQLiteExResultPrivate(QSQLiteExResult *q, const QSQLiteExDriver *drv); void cleanup(); bool fetchNext(QSqlCachedResult::ValueCache &values, int idx, bool initialFetch); // initializes the recordInfo and the cache void initColumns(bool emptyResultset); void finalize(); sqlite3_stmt *stmt; bool skippedStatus; // the status of the fetchNext() that's skipped bool skipRow; // skip the next fetchNext()? QSqlRecord rInf; QVector<QVariant> firstRow; }; QSQLiteExResultPrivate::QSQLiteExResultPrivate(QSQLiteExResult *q, const QSQLiteExDriver *drv) : QSqlCachedResultPrivate(q, drv), stmt(0), skippedStatus(false), skipRow(false) { } void QSQLiteExResultPrivate::cleanup() { Q_Q(QSQLiteExResult); finalize(); rInf.clear(); skippedStatus = false; skipRow = false; q->setAt(QSql::BeforeFirstRow); q->setActive(false); q->cleanup(); } void QSQLiteExResultPrivate::finalize() { if (!stmt) return; sqlite3_finalize(stmt); stmt = 0; } void QSQLiteExResultPrivate::initColumns(bool emptyResultset) { Q_Q(QSQLiteExResult); int nCols = sqlite3_column_count(stmt); if (nCols <= 0) return; q->init(nCols); for (int i = 0; i < nCols; ++i) { QString colName = QString(reinterpret_cast<const QChar *>( sqlite3_column_name16(stmt, i)) ).remove(QLatin1Char('"')); const QString tableName = QString(reinterpret_cast<const QChar *>( sqlite3_column_table_name16(stmt, i)) ).remove(QLatin1Char('"')); // must use typeName for resolving the type to match QSqliteDriver::record QString typeName = QString(reinterpret_cast<const QChar *>( sqlite3_column_decltype16(stmt, i))); // sqlite3_column_type is documented to have undefined behavior if the result set is empty int stp = emptyResultset ? -1 : sqlite3_column_type(stmt, i); QVariant::Type fieldType; if (!typeName.isEmpty()) { fieldType = qGetColumnType(typeName); } else { // Get the proper type for the field based on stp value switch (stp) { case SQLITE_INTEGER: fieldType = QVariant::Int; break; case SQLITE_FLOAT: fieldType = QVariant::Double; break; case SQLITE_BLOB: fieldType = QVariant::ByteArray; break; case SQLITE_TEXT: fieldType = QVariant::String; break; case SQLITE_NULL: default: fieldType = QVariant::Invalid; break; } } QSqlField fld(colName, fieldType, tableName); fld.setSqlType(stp); rInf.append(fld); } } bool QSQLiteExResultPrivate::fetchNext(QSqlCachedResult::ValueCache &values, int idx, bool initialFetch) { Q_Q(QSQLiteExResult); int res; int i; if (skipRow) { // already fetched Q_ASSERT(!initialFetch); skipRow = false; for(int i=0;i<firstRow.count();i++) values[i]=firstRow[i]; return skippedStatus; } skipRow = initialFetch; if(initialFetch) { firstRow.clear(); firstRow.resize(sqlite3_column_count(stmt)); } if (!stmt) { q->setLastError(QSqlError(QCoreApplication::translate("QSQLiteResult", "Unable to fetch row"), QCoreApplication::translate("QSQLiteResult", "No query"), QSqlError::ConnectionError)); q->setAt(QSql::AfterLastRow); return false; } res = sqlite3_step(stmt); switch(res) { case SQLITE_ROW: // check to see if should fill out columns if (rInf.isEmpty()) // must be first call. initColumns(false); if (idx < 0 && !initialFetch) return true; for (i = 0; i < rInf.count(); ++i) { switch (sqlite3_column_type(stmt, i)) { case SQLITE_BLOB: values[i + idx] = QByteArray(static_cast<const char *>( sqlite3_column_blob(stmt, i)),