archive/fscan
1
0
Fork 0
mirror of https://github.com/shadow1ng/fscan.git synced 2026-09-26 05:01:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

添加一个CheckErrs字典

Browse Source
This commit is contained in:
shadow1ng
2021-05-13 18:06:14 +08:00
parent 79aa24fc8f
commit 93245a16d0
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Plugins/mssql.go
+1 -1
View File
@@ -36,7 +36,7 @@ func MssqlScan(info *common.HostInfo) (tmperr error) {
func MssqlConn(info *common.HostInfo, user string, pass string) (flag bool, err error) { func MssqlConn(info *common.HostInfo, user string, pass string) (flag bool, err error) {
flag = false flag = false
Host, Port, Username, Password := info.Host, info.Ports, user, pass 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) db, err := sql.Open("mssql", dataSourceName)
if err == nil { if err == nil {
db.SetConnMaxLifetime(time.Duration(info.Timeout) * time.Second) db.SetConnMaxLifetime(time.Duration(info.Timeout) * time.Second)
Plugins/mysql.go
+1 -1
View File
@@ -36,7 +36,7 @@ func MysqlScan(info *common.HostInfo) (tmperr error) {
func MysqlConn(info *common.HostInfo, user string, pass string) (flag bool, err error) { func MysqlConn(info *common.HostInfo, user string, pass string) (flag bool, err error) {
flag = false flag = false
Host, Port, Username, Password := info.Host, info.Ports, user, pass 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) db, err := sql.Open("mysql", dataSourceName)
if err == nil { if err == nil {
db.SetConnMaxLifetime(time.Duration(info.Timeout) * time.Second) db.SetConnMaxLifetime(time.Duration(info.Timeout) * time.Second)
common/log.go
+1
View File
@@ -72,6 +72,7 @@ func CheckErrs(err error) bool {
"no pg_hba.conf entry", "no pg_hba.conf entry",
"no supported methods remain", "no supported methods remain",
"No connection could be made", "No connection could be made",
"invalid packet size",
} }
for _, key := range errs { for _, key := range errs {
if strings.Contains(strings.ToLower(err.Error()), strings.ToLower(key)) { if strings.Contains(strings.ToLower(err.Error()), strings.ToLower(key)) {