From 54ba490246f9785ae54292832ba8546a0766906a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BD=B1=E8=88=9E=E8=80=85?= <shadow1ng@qq.com>
Date: Mon, 13 Nov 2023 10:51:41 +0800
Subject: [PATCH] Update mssql.go

---
 Plugins/mssql.go | 23 +++++++++++------------
 1 file changed, 11 insertions(+), 12 deletions(-)

diff --git a/Plugins/mssql.go b/Plugins/mssql.go
index 45167f6..6f64c67 100644
--- a/Plugins/mssql.go
+++ b/Plugins/mssql.go
@@ -3,23 +3,22 @@ package Plugins
 import (
 	"database/sql"
 	"fmt"
-	"strings"
-	"time"
-
 	_ "github.com/denisenkom/go-mssqldb"
 	"github.com/shadow1ng/fscan/common"
+	"strings"
+	"time"
 )
 
-func MssqlScan(info common.HostInfo, flags common.Flags) (tmperr error) {
-	if flags.IsBrute {
+func MssqlScan(info *common.HostInfo) (tmperr error) {
+	if common.IsBrute {
 		return
 	}
 	starttime := time.Now().Unix()
 	for _, user := range common.Userdict["mssql"] {
 		for _, pass := range common.Passwords {
 			pass = strings.Replace(pass, "{user}", user, -1)
-			flag, err := MssqlConn(info, user, pass, flags.Timeout)
-			if flag && err == nil {
+			flag, err := MssqlConn(info, user, pass)
+			if flag == true && err == nil {
 				return err
 			} else {
 				errlog := fmt.Sprintf("[-] mssql %v:%v %v %v %v", info.Host, info.Ports, user, pass, err)
@@ -28,7 +27,7 @@ func MssqlScan(info common.HostInfo, flags common.Flags) (tmperr error) {
 				if common.CheckErrs(err) {
 					return err
 				}
-				if time.Now().Unix()-starttime > (int64(len(common.Userdict["mssql"])*len(common.Passwords)) * flags.Timeout) {
+				if time.Now().Unix()-starttime > (int64(len(common.Userdict["mssql"])*len(common.Passwords)) * common.Timeout) {
 					return err
 				}
 			}
@@ -37,14 +36,14 @@ func MssqlScan(info common.HostInfo, flags common.Flags) (tmperr error) {
 	return tmperr
 }
 
-func MssqlConn(info common.HostInfo, user string, pass string, timeout int64) (flag bool, err 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=%v;encrypt=disable;timeout=%v", Host, Username, Password, Port, time.Duration(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(common.Timeout)*time.Second)
 	db, err := sql.Open("mssql", dataSourceName)
 	if err == nil {
-		db.SetConnMaxLifetime(time.Duration(timeout) * time.Second)
-		db.SetConnMaxIdleTime(time.Duration(timeout) * time.Second)
+		db.SetConnMaxLifetime(time.Duration(common.Timeout) * time.Second)
+		db.SetConnMaxIdleTime(time.Duration(common.Timeout) * time.Second)
 		db.SetMaxIdleConns(0)
 		defer db.Close()
 		err = db.Ping()