From 78679f7a46d62615759fc27484534a25e0ce18e9 Mon Sep 17 00:00:00 2001
From: xx <a1585346868@hotmail.com>
Date: Fri, 14 Jun 2024 19:00:33 +0800
Subject: [PATCH] =?UTF-8?q?=E5=85=BC=E5=AE=B9linux=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/net/tcp_forward.cpp | 1 +
 src/util/sqlite3.cpp    | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/src/net/tcp_forward.cpp b/src/net/tcp_forward.cpp
index 6d87059..5d3bc4d 100644
--- a/src/net/tcp_forward.cpp
+++ b/src/net/tcp_forward.cpp
@@ -29,6 +29,7 @@
 #include "net/tcp_server.h"
 #include "HPSocket/HPSocket.h"
 #include "util/strutils.h"
+#include <string.h>
 ylib::network::tcp::forward::forward()
 {
 }
diff --git a/src/util/sqlite3.cpp b/src/util/sqlite3.cpp
index 3193d1d..cc083dc 100644
--- a/src/util/sqlite3.cpp
+++ b/src/util/sqlite3.cpp
@@ -61,6 +61,8 @@ void ylib::sqlite3::close()
 
 bool ylib::sqlite3::exec(const std::string& sql)
 {
+	if (m_db == nullptr)
+		return false;
 #if DEBUG_PRINT_SQL == 1
 #ifdef _DEBUG
 	std::cout << "[SQLITE3] :\t" << codec::to_gbk(sql) << std::endl;
@@ -88,6 +90,8 @@ static int QueryResult(void* param, int column, char** argv, char** azColName)
 }
 bool ylib::sqlite3::query(const std::string& sql, std::vector<std::map<std::string, std::string>>& data)
 {
+	if (m_db == nullptr)
+		return false;
 #if DEBUG_PRINT_SQL == 1
 #ifdef _DEBUG
 	std::cout << "[SQLITE3] :\t" << codec::to_gbk(sql) << std::endl;
@@ -104,6 +108,7 @@ bool ylib::sqlite3::query(const std::string& sql, std::vector<std::map<std::stri
 
 int64 ylib::sqlite3::count(const std::string& sql)
 {
+
 	SQLITE_RESULT results;
 	if (query(sql, results) == false)
 		return 0;
@@ -113,5 +118,7 @@ int64 ylib::sqlite3::count(const std::string& sql)
 }
 int64 ylib::sqlite3::last_insert_id()
 {
+	if (m_db == nullptr)
+		return 0;
 	return sqlite3_last_insert_rowid(DB_PTR);
 }
\ No newline at end of file