communication_dsoftbus/core/common/utils/sqlite3_utils.c
jeosif 7d99f26aa1 build heartbeat filter by saving udid after each networking
Signed-off-by: jeosif <liubao6@huawei.com>
Change-Id: Ife5c4ba4e091e2a3dd73e7f40be1125fc2480dfd
2022-09-24 15:31:11 +08:00

704 lines
24 KiB
C

/*
* Copyright (c) 2022 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "sqlite3_utils.h"
#include <securec.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "softbus_adapter_mem.h"
#include "softbus_errcode.h"
#include "softbus_log.h"
#define SQL_DEFAULT_LEN 256
/* The index of database context state */
#define DB_STATE_QUERYING (0x1)
#define DB_STATE_TRANSACTION (0x1 << 1)
typedef int32_t (*BindParaCb)(DbContext *ctx, int32_t paraNum, uint8_t *data);
typedef int32_t (*QueryDataCb)(DbContext *ctx, uint8_t *data, int32_t idx);
typedef struct {
const char *tableName;
const char *sqlForCreate;
const char *sqlForInsert;
const char *sqlForSearchByKey;
const char *sqlForRemoveByKey;
BindParaCb insertCb;
BindParaCb searchCb;
BindParaCb removeCb;
QueryDataCb queryDataCb;
} SqliteManager;
/* The default SQL statement */
#define SQL_DROP_TABLE "DROP TABLE "
#define SQL_REMOVE_ALL_RECORD "DELETE FROM "
#define SQL_BEGIN_TRANSACTION "BEGIN TRANSACTION"
#define SQL_COMMIT_TRANSACTION "COMMIT TRANSACTION"
#define SQL_ROLLBACK_TRANSACTION "ROLLBACK TRANSACTION"
#define SQL_SEARCH_IF_TABLE_EXIST "SELECT * FROM sqlite_master WHERE type ='table' AND name = '%s'"
/**
* @brief The SQL statement of TrustedDeviceInfo table.
*
* This table is used to store the trusted relationship, and its name is TrustedDeviceInfo in {@link DATABASE_NAME}.
* After each networking, record the udid value according to the device account.
*/
#define TABLE_NAME_OF_TRUSTED_DEV_INFO "TrustedDeviceInfo"
#define SQL_CREATE_TRUSTED_DEV_INFO_TABLE "CREATE TABLE IF NOT EXISTS "TABLE_NAME_OF_TRUSTED_DEV_INFO" \
(accountHash TEXT NOT NULL, \
udid TEXT NOT NULL, \
primary key(accountHash, udid));"
#define SQL_INSERT_TRUSTED_DEV_INFO "INSERT INTO "TABLE_NAME_OF_TRUSTED_DEV_INFO" \
(accountHash, udid) VALUES (?, ?)"
#define SQL_SEARCH_TRUSTED_DEV_INFO_BY_ID "SELECT udid FROM "TABLE_NAME_OF_TRUSTED_DEV_INFO" \
WHERE accountHash = ?"
#define SQL_REMOVE_TRUSTED_DEV_INFO_BY_ID "DELETE FROM "TABLE_NAME_OF_TRUSTED_DEV_INFO" \
WHERE accountHash = ? AND udid = ?"
static int32_t BindInsertTrustedDevInfoCb(DbContext *ctx, int32_t paraNum, uint8_t *data);
static int32_t BindSelectTrustedDevInfoCb(DbContext *ctx, int32_t paraNum, uint8_t *data);
static int32_t GetTrustedDevInfoByIdCb(DbContext *ctx, uint8_t *data, int32_t idx);
static SqliteManager g_sqliteMgr[TABLE_NAME_ID_MAX] = {
[TABLE_TRUSTED_DEV_INFO] = {
.tableName = TABLE_NAME_OF_TRUSTED_DEV_INFO,
.sqlForCreate = SQL_CREATE_TRUSTED_DEV_INFO_TABLE,
.sqlForInsert = SQL_INSERT_TRUSTED_DEV_INFO,
.sqlForSearchByKey = SQL_SEARCH_TRUSTED_DEV_INFO_BY_ID,
.sqlForRemoveByKey = SQL_REMOVE_TRUSTED_DEV_INFO_BY_ID,
.insertCb = BindInsertTrustedDevInfoCb,
.searchCb = BindSelectTrustedDevInfoCb,
.removeCb = BindInsertTrustedDevInfoCb,
.queryDataCb = GetTrustedDevInfoByIdCb,
},
};
static int32_t GetTrustedDevInfoByIdCb(DbContext *ctx, uint8_t *data, int32_t idx)
{
int32_t i = 0;
char *info = (char *)data + idx * UDID_BUF_LEN;
if (GetQueryResultColText(ctx, i, info, UDID_BUF_LEN) != SOFTBUS_OK) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "get query result failed");
return SOFTBUS_ERR;
}
return SOFTBUS_OK;
}
static int32_t BindInsertTrustedDevInfoCb(DbContext *ctx, int32_t paraNum, uint8_t *data)
{
int32_t rc;
int32_t idx = 1;
if (data == NULL) {
return SQLITE_ERROR;
}
const TrustedDevInfoRecord *record = (TrustedDevInfoRecord *)data;
rc = BindParaText(ctx, idx, record->accountHexHash, strlen(record->accountHexHash));
if (rc != SQLITE_OK) {
return rc;
}
return BindParaText(ctx, ++idx, record->udid, strlen(record->udid));
}
static int32_t BindSelectTrustedDevInfoCb(DbContext *ctx, int32_t paraNum, uint8_t *data)
{
int32_t idx = 1;
if (data == NULL) {
return SQLITE_ERROR;
}
return BindParaText(ctx, idx, (char *)data, strlen((char *)data));
}
static int32_t ExecuteSql(DbContext *ctx, const char *sql, uint32_t len, BindParaCb cb, uint8_t *data)
{
int32_t paraNum;
int32_t rc;
if (sql == NULL || sql[0] == '\0') {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "execute sql get invalid param");
return SQLITE_ERROR;
}
rc = sqlite3_prepare_v2(ctx->db, sql, len, &ctx->stmt, NULL);
if (rc != SQLITE_OK || ctx->stmt == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sqlite3_prepare_v2 failed, %s", sqlite3_errmsg(ctx->db));
return sqlite3_errcode(ctx->db);
}
paraNum = sqlite3_bind_parameter_count(ctx->stmt);
if (paraNum <= 0) {
rc = sqlite3_step(ctx->stmt);
if (rc != SQLITE_ROW && rc != SQLITE_DONE) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sqlite3_step <= 0 failed, %s", sqlite3_errmsg(ctx->db));
}
return rc;
}
if (paraNum > 0 && cb == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "need cd for binding parameter");
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
return SQLITE_ERROR;
}
rc = cb(ctx, paraNum, data);
if (rc != SQLITE_OK) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "binding parameter cd fail");
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
return sqlite3_errcode(ctx->db);
}
rc = sqlite3_step(ctx->stmt);
if (rc != SQLITE_ROW && rc != SQLITE_DONE) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sqlite3_step > 0 failed, %s", sqlite3_errmsg(ctx->db));
}
return rc;
}
static int32_t QueryData(DbContext *ctx, const char *sql, uint32_t len, BindParaCb cb, uint8_t *data)
{
int32_t rc;
rc = ExecuteSql(ctx, sql, len, cb, data);
if (rc != SQLITE_ROW) {
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
} else {
ctx->state |= DB_STATE_QUERYING;
}
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_DBG, "QueryData done, state: %d", ctx->state);
return rc;
}
static int32_t QueryDataNext(DbContext *ctx)
{
int32_t rc;
rc = sqlite3_step(ctx->stmt);
if (rc != SQLITE_ROW) {
ctx->state &= ~DB_STATE_QUERYING;
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
}
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_DBG, "QueryDataNext done, state: %d", ctx->state);
return rc;
}
static bool CheckDbContextParam(const DbContext *ctx)
{
if (ctx == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return false;
}
if (ctx->db == NULL || ctx->stmt != NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid db context state");
return false;
}
return true;
}
static bool CheckBindOrQueryParam(const DbContext *ctx)
{
if (ctx == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid db context parameters");
return false;
}
if (ctx->db == NULL || ctx->stmt == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid db context state");
return false;
}
return true;
}
int32_t OpenDatabase(DbContext **ctx)
{
int32_t rc;
sqlite3 *sqlite = NULL;
if (ctx == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
rc = sqlite3_open_v2(DATABASE_NAME, &sqlite, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE |
SQLITE_OPEN_NOMUTEX, NULL);
if (rc != SQLITE_OK || sqlite == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sqlite3_open_v2 fail: %s", sqlite3_errmsg(sqlite));
(void)sqlite3_close_v2(sqlite);
return SOFTBUS_ERR;
}
*ctx = (DbContext *)SoftBusCalloc(sizeof(DbContext));
if (*ctx == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "malloc DbContext fail");
(void)sqlite3_close_v2(sqlite);
return SOFTBUS_MALLOC_ERR;
} else {
(*ctx)->db = sqlite;
}
return SOFTBUS_OK;
}
int32_t CloseDatabase(DbContext *ctx)
{
if (!CheckDbContextParam(ctx)) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
(void)sqlite3_close_v2(ctx->db);
SoftBusFree(ctx);
return SOFTBUS_OK;
}
int32_t CreateTable(DbContext *ctx, TableNameID id)
{
int32_t rc;
char *errMsg = NULL;
if (!CheckDbContextParam(ctx)) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
const char *sql = g_sqliteMgr[id].sqlForCreate;
if (sql == NULL || sql[0] == '\0') {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "createsql is not impl");
return SOFTBUS_ERR;
}
rc = sqlite3_exec(ctx->db, sql, NULL, NULL, &errMsg);
if (rc != SQLITE_OK && errMsg != NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sqlite_exec fail: %s", errMsg);
sqlite3_free(errMsg);
}
return rc == SQLITE_OK ? SOFTBUS_OK : SOFTBUS_ERR;
}
int32_t DeleteTable(DbContext *ctx, TableNameID id)
{
int32_t rc;
char sql[SQL_DEFAULT_LEN] = {0};
if (!CheckDbContextParam(ctx)) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
rc = sprintf_s(sql, SQL_DEFAULT_LEN, "%s%s", SQL_DROP_TABLE, g_sqliteMgr[id].tableName);
if (rc < 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sprintf_s sql fail");
return SOFTBUS_ERR;
}
rc = ExecuteSql(ctx, sql, strlen(sql), NULL, NULL);
if (rc != SQLITE_DONE) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "delete table fail");
rc = SOFTBUS_ERR;
} else {
rc = SOFTBUS_OK;
}
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
return rc;
}
int32_t CheckTableExist(DbContext *ctx, TableNameID id, bool *isExist)
{
int32_t rc;
char sql[SQL_DEFAULT_LEN] = {0};
if (!CheckDbContextParam(ctx) || isExist == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
rc = sprintf_s(sql, SQL_DEFAULT_LEN, SQL_SEARCH_IF_TABLE_EXIST, g_sqliteMgr[id].tableName);
if (rc < 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sprintf_s sql fail");
return SOFTBUS_ERR;
}
*isExist = false;
rc = ExecuteSql(ctx, sql, strlen(sql), NULL, NULL);
if (rc == SQLITE_ROW && sqlite3_column_count(ctx->stmt) != 0) {
*isExist = true;
}
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
return SOFTBUS_OK;
}
int32_t InsertRecord(DbContext *ctx, TableNameID id, uint8_t *data)
{
int32_t rc;
if (!CheckDbContextParam(ctx) || data == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
rc = ExecuteSql(ctx, g_sqliteMgr[id].sqlForInsert, strlen(g_sqliteMgr[id].sqlForInsert),
g_sqliteMgr[id].insertCb, data);
if (rc != SQLITE_DONE) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "insert data failed");
rc = SOFTBUS_ERR;
} else {
rc = SOFTBUS_OK;
}
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_DBG, "insert data done");
return rc;
}
int32_t RemoveRecordByKey(DbContext *ctx, TableNameID id, uint8_t *data)
{
int32_t rc;
if (!CheckDbContextParam(ctx) || data == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
rc = ExecuteSql(ctx, g_sqliteMgr[id].sqlForRemoveByKey, strlen(g_sqliteMgr[id].sqlForRemoveByKey),
g_sqliteMgr[id].removeCb, data);
if (rc != SQLITE_DONE) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "remove data failed");
rc = SOFTBUS_ERR;
} else {
rc = SOFTBUS_OK;
}
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_DBG, "remove data done");
return rc;
}
int32_t RemoveAllRecord(DbContext *ctx, TableNameID id)
{
int32_t rc;
char sql[SQL_DEFAULT_LEN] = {0};
if (!CheckDbContextParam(ctx)) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
rc = sprintf_s(sql, SQL_DEFAULT_LEN, "%s%s", SQL_REMOVE_ALL_RECORD, g_sqliteMgr[id].tableName);
if (rc < 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sprintf_s sql fail");
return SOFTBUS_ERR;
}
rc = ExecuteSql(ctx, sql, strlen(sql), NULL, NULL);
if (rc != SQLITE_DONE) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "remove data failed");
rc = SOFTBUS_ERR;
} else {
rc = SOFTBUS_OK;
}
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_DBG, "remove data done");
return rc;
}
int32_t GetRecordNumByKey(DbContext *ctx, TableNameID id, uint8_t *data)
{
int32_t rc;
int32_t num = 0;
if (!CheckDbContextParam(ctx)) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return 0;
}
rc = QueryData(ctx, g_sqliteMgr[id].sqlForSearchByKey, strlen(g_sqliteMgr[id].sqlForSearchByKey),
g_sqliteMgr[id].searchCb, data);
if (rc != SQLITE_ROW) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "find no match data");
return 0;
}
do {
num++;
rc = QueryDataNext(ctx);
} while (rc == SQLITE_ROW);
if (rc != SQLITE_DONE) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "GetQueryDataNum failed");
return 0;
}
return num;
}
int32_t QueryRecordByKey(DbContext *ctx, TableNameID id, uint8_t *data, uint8_t **replyInfo, int infoNum)
{
int32_t rc;
int32_t num = 0;
if (!CheckDbContextParam(ctx) || replyInfo == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
rc = QueryData(ctx, g_sqliteMgr[id].sqlForSearchByKey, strlen(g_sqliteMgr[id].sqlForSearchByKey),
g_sqliteMgr[id].searchCb, data);
if (rc != SQLITE_ROW) {
return SOFTBUS_ERR;
}
do {
if (g_sqliteMgr[id].queryDataCb != NULL) {
g_sqliteMgr[id].queryDataCb(ctx, *replyInfo, num);
}
rc = QueryDataNext(ctx);
num++;
} while (rc == SQLITE_ROW && num < infoNum);
if (rc != SQLITE_DONE) {
if (rc == SQLITE_ROW) {
ctx->state &= ~DB_STATE_QUERYING;
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
}
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "QueryData failed");
return SOFTBUS_ERR;
}
return SOFTBUS_OK;
}
int32_t OpenTransaction(DbContext *ctx)
{
int32_t rc;
if (!CheckDbContextParam(ctx)) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
if ((ctx->state & DB_STATE_TRANSACTION) != 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "already open the transaction: %d", ctx->state);
return SOFTBUS_OK;
}
rc = ExecuteSql(ctx, SQL_BEGIN_TRANSACTION, strlen(SQL_BEGIN_TRANSACTION), NULL, NULL);
if (rc != SQLITE_DONE) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "open transaction failed");
rc = SOFTBUS_ERR;
} else {
ctx->state |= DB_STATE_TRANSACTION;
rc = SOFTBUS_OK;
}
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
return rc;
}
int32_t CloseTransaction(DbContext *ctx, CloseTransactionType type)
{
int32_t rc;
const char *sql = SQL_COMMIT_TRANSACTION;
if (!CheckDbContextParam(ctx)) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
if ((ctx->state & DB_STATE_TRANSACTION) == 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "the transaction already closed: %d", ctx->state);
return SOFTBUS_OK;
}
if (type == CLOSE_TRANS_ROLLBACK) {
sql = SQL_ROLLBACK_TRANSACTION;
}
rc = ExecuteSql(ctx, sql, strlen(sql), NULL, NULL);
if (rc != SQLITE_DONE) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "close transaction failed");
rc = SOFTBUS_ERR;
} else {
rc = SOFTBUS_OK;
}
ctx->state &= ~DB_STATE_TRANSACTION;
(void)sqlite3_finalize(ctx->stmt);
ctx->stmt = NULL;
return rc;
}
int32_t EncryptedDb(DbContext *ctx, const char *password, uint32_t len)
{
int32_t rc;
if (!CheckDbContextParam(ctx) || password == NULL || password[0] == '\0' || strlen(password) != len) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
rc = sqlite3_key(ctx->db, password, len);
if (rc != SQLITE_OK) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "config key failed: %s", sqlite3_errmsg(ctx->db));
return SOFTBUS_ERR;
}
return SOFTBUS_OK;
}
int32_t UpdateDbPassword(DbContext *ctx, const char *password, uint32_t len)
{
int32_t rc;
if (!CheckDbContextParam(ctx) || password == NULL || password[0] == '\0' || strlen(password) != len) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
rc = sqlite3_rekey(ctx->db, password, len);
if (rc != SQLITE_OK) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "update key failed: %s", sqlite3_errmsg(ctx->db));
return SOFTBUS_ERR;
}
return SOFTBUS_OK;
}
int32_t BindParaInt(DbContext *ctx, int32_t idx, int32_t value)
{
int32_t rc;
if (!CheckBindOrQueryParam(ctx) || idx <= 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SQLITE_ERROR;
}
rc = sqlite3_bind_int(ctx->stmt, idx, value);
if (rc != SQLITE_OK) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sqlite3_bind_int failed: %s", sqlite3_errmsg(ctx->db));
}
return rc;
}
int32_t BindParaInt64(DbContext *ctx, int32_t idx, int64_t value)
{
int32_t rc;
if (!CheckBindOrQueryParam(ctx) || idx <= 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SQLITE_ERROR;
}
rc = sqlite3_bind_int64(ctx->stmt, idx, value);
if (rc != SQLITE_OK) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sqlite3_bind_int64 failed: %s", sqlite3_errmsg(ctx->db));
}
return rc;
}
int32_t BindParaText(DbContext *ctx, int32_t idx, const char *value, uint32_t valueLen)
{
int32_t rc;
if (!CheckBindOrQueryParam(ctx) || idx <= 0 || value == NULL || value[0] == '\0' || strlen(value) != valueLen) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SQLITE_ERROR;
}
rc = sqlite3_bind_text(ctx->stmt, idx, value, valueLen, SQLITE_STATIC);
if (rc != SQLITE_OK) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sqlite3_bind_text failed: %s", sqlite3_errmsg(ctx->db));
}
return rc;
}
int32_t BindParaDouble(DbContext *ctx, int32_t idx, double value)
{
int32_t rc;
if (!CheckBindOrQueryParam(ctx) || idx <= 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SQLITE_ERROR;
}
rc = sqlite3_bind_double(ctx->stmt, idx, value);
if (rc != SQLITE_OK) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "sqlite3_bind_double failed: %s", sqlite3_errmsg(ctx->db));
}
return rc;
}
int32_t GetQueryResultColCount(DbContext *ctx, int32_t *count)
{
if (!CheckBindOrQueryParam(ctx)) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
if ((ctx->state & DB_STATE_QUERYING) == 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "the query already closed: %d", ctx->state);
return SOFTBUS_ERR;
}
*count = sqlite3_column_count(ctx->stmt);
return SOFTBUS_OK;
}
int32_t GetQueryResultColText(DbContext *ctx, int32_t iCol, char *text, uint32_t len)
{
const unsigned char *result;
if (!CheckBindOrQueryParam(ctx) || iCol < 0 || text == NULL) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
if ((ctx->state & DB_STATE_QUERYING) == 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "the query already closed: %d", ctx->state);
return SOFTBUS_ERR;
}
if (sqlite3_column_type(ctx->stmt, iCol) != SQLITE_TEXT) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "column type not match");
return SOFTBUS_ERR;
}
result = sqlite3_column_text(ctx->stmt, iCol);
if (strcpy_s(text, len, (const char *)result) != EOK) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "strcpy_s fail");
return SOFTBUS_ERR;
}
return SOFTBUS_OK;
}
int32_t GetQueryResultColInt(DbContext *ctx, int32_t iCol, int32_t *value)
{
if (!CheckBindOrQueryParam(ctx) || iCol < 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
if ((ctx->state & DB_STATE_QUERYING) == 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "the query already closed: %d", ctx->state);
return SOFTBUS_ERR;
}
if (sqlite3_column_type(ctx->stmt, iCol) != SQLITE_INTEGER) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "column type not match");
return SOFTBUS_ERR;
}
*value = sqlite3_column_int(ctx->stmt, iCol);
return SOFTBUS_OK;
}
int32_t GetQueryResultColInt64(DbContext *ctx, int32_t iCol, int64_t *value)
{
if (!CheckBindOrQueryParam(ctx) || iCol < 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
if ((ctx->state & DB_STATE_QUERYING) == 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "the query already closed: %d", ctx->state);
return SOFTBUS_ERR;
}
if (sqlite3_column_type(ctx->stmt, iCol) != SQLITE_INTEGER) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "column type not match");
return SOFTBUS_ERR;
}
*value = sqlite3_column_int64(ctx->stmt, iCol);
return SOFTBUS_OK;
}
int32_t GetQueryResultColDouble(DbContext *ctx, int32_t iCol, double *value)
{
if (!CheckBindOrQueryParam(ctx) || iCol < 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "invalid parameters");
return SOFTBUS_INVALID_PARAM;
}
if ((ctx->state & DB_STATE_QUERYING) == 0) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "the query already closed: %d", ctx->state);
return SOFTBUS_ERR;
}
if (sqlite3_column_type(ctx->stmt, iCol) != SQLITE_FLOAT) {
SoftBusLog(SOFTBUS_LOG_COMM, SOFTBUS_LOG_ERROR, "column type not match");
return SOFTBUS_ERR;
}
*value = sqlite3_column_double(ctx->stmt, iCol);
return SOFTBUS_OK;
}