fix(libsqlite3): 修复 double free 与移动语义缺陷
- 为 Sqlite3 与 Sqlite3_stmt 显式 delete 拷贝、实现 noexcept 移动构造/赋值 - 析构函数判空后释放,防止重复释放与 use-after-free - ~Sqlite3() 改用 sqlite3_close_v2,确保所有 statement 释放后再关闭 - stmt() 参数改为 const std::string&,支持字符串字面量 - 成员指针显式初始化为 nullptr,errMsg 初始化为 nullptr
This commit is contained in:
1 parent
b781a8d2fe
commit
985ff47c84
1 file changed
+44
-7
@@ -15,7 +15,26 @@ namespace qing {
|
|||||||
}
|
}
|
||||||
|
|
||||||
~Sqlite3_stmt() {
|
~Sqlite3_stmt() {
|
||||||
sqlite3_finalize(stmt);
|
if (stmt) {
|
||||||
|
sqlite3_finalize(stmt);
|
||||||
|
stmt = nullptr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Sqlite3_stmt(const Sqlite3_stmt&) = delete;
|
||||||
|
Sqlite3_stmt& operator=(const Sqlite3_stmt&) = delete;
|
||||||
|
|
||||||
|
Sqlite3_stmt(Sqlite3_stmt&& other) noexcept : stmt(other.stmt) {
|
||||||
|
other.stmt = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
Sqlite3_stmt& operator=(Sqlite3_stmt&& other) noexcept {
|
||||||
|
if (this != &other) {
|
||||||
|
if (stmt) sqlite3_finalize(stmt);
|
||||||
|
stmt = other.stmt;
|
||||||
|
other.stmt = nullptr;
|
||||||
|
}
|
||||||
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
void bind_int(int idx, int data) {
|
void bind_int(int idx, int data) {
|
||||||
@@ -56,7 +75,7 @@ namespace qing {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
sqlite3_stmt* stmt;
|
sqlite3_stmt* stmt = nullptr;
|
||||||
};
|
};
|
||||||
|
|
||||||
class Sqlite3 {
|
class Sqlite3 {
|
||||||
@@ -72,27 +91,45 @@ namespace qing {
|
|||||||
}
|
}
|
||||||
|
|
||||||
~Sqlite3() {
|
~Sqlite3() {
|
||||||
sqlite3_close(db);
|
if (db) {
|
||||||
|
sqlite3_close_v2(db);
|
||||||
|
db = nullptr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Sqlite3(const Sqlite3&) = delete;
|
||||||
|
Sqlite3& operator=(const Sqlite3&) = delete;
|
||||||
|
|
||||||
|
Sqlite3(Sqlite3&& other) noexcept : db(other.db) {
|
||||||
|
other.db = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
Sqlite3& operator=(Sqlite3&& other) noexcept {
|
||||||
|
if (this != &other) {
|
||||||
|
if (db) sqlite3_close_v2(db);
|
||||||
|
db = other.db;
|
||||||
|
other.db = nullptr;
|
||||||
|
}
|
||||||
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool exec(const std::string& sql) {
|
bool exec(const std::string& sql) {
|
||||||
char *errMsg;
|
char *errMsg = nullptr;
|
||||||
int rc = sqlite3_exec(db, sql.c_str(), 0, 0, &errMsg);
|
int rc = sqlite3_exec(db, sql.c_str(), 0, 0, &errMsg);
|
||||||
if (rc != SQLITE_OK) {
|
if (rc != SQLITE_OK) {
|
||||||
std::cerr << "Failed to execute at sqlite3: " << errMsg << std::endl;
|
std::cerr << "Failed to execute at sqlite3: " << errMsg << std::endl;
|
||||||
sqlite3_free(errMsg);
|
sqlite3_free(errMsg);
|
||||||
//sqlite3_close(db);
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
Sqlite3_stmt stmt(std::string& sql) {
|
Sqlite3_stmt stmt(const std::string& sql) {
|
||||||
return Sqlite3_stmt(db, sql.c_str());
|
return Sqlite3_stmt(db, sql.c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
sqlite3* db;
|
sqlite3* db = nullptr;
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
Reference in new issue
Block a user