From 93245a16d0ed35738178acca3ad09ec2f9d141da Mon Sep 17 00:00:00 2001
From: shadow1ng <shadow1ng>
Date: Thu, 13 May 2021 18:06:14 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E4=B8=80=E4=B8=AACheckErrs?=
 =?UTF-8?q?=E5=AD=97=E5=85=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Plugins/mssql.go | 2 +-
 Plugins/mysql.go | 2 +-
 common/log.go    | 1 +
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/Plugins/mssql.go b/Plugins/mssql.go
index 0220d87..c0df695 100644
--- a/Plugins/mssql.go
+++ b/Plugins/mssql.go
@@ -36,7 +36,7 @@ func MssqlScan(info *common.HostInfo) (tmperr error) {
 func MssqlConn(info *common.HostInfo, user string, pass string) (flag bool, err error) {
 	flag = false
 	Host, Port, Username, Password := info.Host, info.Ports, user, pass
-	dataSourceName := fmt.Sprintf("server=%s;user id=%s;password=%s;port=%d;encrypt=disable;timeout=%d", Host, Username, Password, Port, time.Duration(info.Timeout)*time.Second)
+	dataSourceName := fmt.Sprintf("server=%s;user id=%s;password=%s;port=%v;encrypt=disable;timeout=%v", Host, Username, Password, Port, time.Duration(info.Timeout)*time.Second)
 	db, err := sql.Open("mssql", dataSourceName)
 	if err == nil {
 		db.SetConnMaxLifetime(time.Duration(info.Timeout) * time.Second)
diff --git a/Plugins/mysql.go b/Plugins/mysql.go
index 23cb815..e545da1 100644
--- a/Plugins/mysql.go
+++ b/Plugins/mysql.go
@@ -36,7 +36,7 @@ func MysqlScan(info *common.HostInfo) (tmperr error) {
 func MysqlConn(info *common.HostInfo, user string, pass string) (flag bool, err error) {
 	flag = false
 	Host, Port, Username, Password := info.Host, info.Ports, user, pass
-	dataSourceName := fmt.Sprintf("%v:%v@tcp(%v:%v)/%v?charset=utf8", Username, Password, Host, Port, "mysql")
+	dataSourceName := fmt.Sprintf("%v:%v@tcp(%v:%v)/mysql?charset=utf8&timeout=%v", Username, Password, Host, Port, time.Duration(info.Timeout)*time.Second)
 	db, err := sql.Open("mysql", dataSourceName)
 	if err == nil {
 		db.SetConnMaxLifetime(time.Duration(info.Timeout) * time.Second)
diff --git a/common/log.go b/common/log.go
index 6bf1000..bea8198 100644
--- a/common/log.go
+++ b/common/log.go
@@ -72,6 +72,7 @@ func CheckErrs(err error) bool {
 		"no pg_hba.conf entry",
 		"no supported methods remain",
 		"No connection could be made",
+		"invalid packet size",
 	}
 	for _, key := range errs {
 		if strings.Contains(strings.ToLower(err.Error()), strings.ToLower(key)) {