refactor: 输出格式重构，重构SMB、SMB2、FTP的一些验证逻辑

Core/Scanner.go

@@ -7,12 +7,12 @@ import (
 	"strconv"
 	"strings"
 	"sync"
+	"sync/atomic"
 )
 
 // Scan 执行扫描主流程
 func Scan(info Common.HostInfo) {
-	fmt.Println("[*] 开始信息扫描...")
-
+	Common.LogInfo("开始信息扫描")
 	Common.ParseScanMode(Common.ScanMode)
 
 	ch := make(chan struct{}, Common.ThreadNum)
@@ -28,7 +28,7 @@ func Scan(info Common.HostInfo) {
 	// 初始化并解析目标
 	hosts, err := Common.ParseIP(info.Host, Common.HostsFile, Common.ExcludeHosts)
 	if err != nil {
-		fmt.Printf("[-] 解析主机错误: %v\n", err)
+		Common.LogError(fmt.Sprintf("解析主机错误: %v", err))
 		return
 	}
 	lib.Inithttp()
@@ -42,50 +42,44 @@ func Scan(info Common.HostInfo) {
 func executeScan(hosts []string, info Common.HostInfo, ch *chan struct{}, wg *sync.WaitGroup) {
 	var targetInfos []Common.HostInfo
 
-	// 处理主机和端口扫描
 	if len(hosts) > 0 || len(Common.HostPort) > 0 {
-		// ICMP存活性检测
 		if (Common.DisablePing == false && len(hosts) > 1) || Common.IsICMPScan() {
 			hosts = CheckLive(hosts, Common.UsePing)
-			fmt.Printf("[+] ICMP存活主机数量: %d\n", len(hosts))
+			Common.LogInfo(fmt.Sprintf("存活主机数量: %d", len(hosts)))
 			if Common.IsICMPScan() {
 				return
 			}
 		}
 
-		// 获取存活端口
 		var alivePorts []string
 		if Common.IsWebScan() {
 			alivePorts = NoPortScan(hosts, Common.Ports)
 		} else if len(hosts) > 0 {
 			alivePorts = PortScan(hosts, Common.Ports, Common.Timeout)
-			fmt.Printf("[+] 存活端口数量: %d\n", len(alivePorts))
+			Common.LogInfo(fmt.Sprintf("存活端口数量: %d", len(alivePorts)))
 			if Common.IsPortScan() {
 				return
 			}
 		}
 
-		// 处理自定义端口
 		if len(Common.HostPort) > 0 {
 			alivePorts = append(alivePorts, Common.HostPort...)
 			alivePorts = Common.RemoveDuplicate(alivePorts)
 			Common.HostPort = nil
-			fmt.Printf("[+] 存活端口数量: %d\n", len(alivePorts))
+			Common.LogInfo(fmt.Sprintf("存活端口数量: %d", len(alivePorts)))
 		}
 
 		targetInfos = prepareTargetInfos(alivePorts, info)
 	}
 
-	// 准备URL扫描目标
 	for _, url := range Common.URLs {
 		urlInfo := info
 		urlInfo.Url = url
 		targetInfos = append(targetInfos, urlInfo)
 	}
 
-	// 执行扫描任务
 	if len(targetInfos) > 0 {
-		fmt.Println("[*] 开始漏洞扫描...")
+		Common.LogInfo("开始漏洞扫描")
 		executeScans(targetInfos, ch, wg)
 	}
 }
@@ -96,7 +90,7 @@ func prepareTargetInfos(alivePorts []string, baseInfo Common.HostInfo) []Common.
 	for _, targetIP := range alivePorts {
 		hostParts := strings.Split(targetIP, ":")
 		if len(hostParts) != 2 {
-			fmt.Printf("[-] 无效的目标地址格式: %s\n", targetIP)
+			Common.LogError(fmt.Sprintf("无效的目标地址格式: %s", targetIP))
 			continue
 		}
 		info := baseInfo
@@ -112,54 +106,42 @@ func executeScans(targets []Common.HostInfo, ch *chan struct{}, wg *sync.WaitGro
 	var pluginsToRun []string
 	isSinglePlugin := false
 
-	// 获取要执行的插件列表
 	if plugins := Common.GetPluginsForMode(mode); plugins != nil {
-		// 预设模式下使用配置的插件组
 		pluginsToRun = plugins
-		fmt.Printf("[*] 正在加载插件组: %s\n", mode)
+		Common.LogInfo(fmt.Sprintf("加载插件组: %s", mode))
 	} else {
-		// 单插件模式
 		pluginsToRun = []string{mode}
 		isSinglePlugin = true
-		fmt.Printf("[*] 正在加载单插件: %s\n", mode)
+		Common.LogInfo(fmt.Sprintf("使用单个插件: %s", mode))
 	}
 
-	// 统一处理所有目标和插件
 	for _, target := range targets {
 		targetPort, _ := strconv.Atoi(target.Ports)
 
 		for _, pluginName := range pluginsToRun {
-			// 获取插件信息
 			plugin, exists := Common.PluginManager[pluginName]
 			if !exists {
-				fmt.Printf("[-] 插件 %s 不存在\n", pluginName)
+				Common.LogError(fmt.Sprintf("插件 %s 不存在", pluginName))
 				continue
 			}
 
-			// 本地扫描模式的特殊处理
 			if Common.LocalScan {
 				if len(plugin.Ports) == 0 {
-					fmt.Printf("[+] 载入插件: %s\n", pluginName)
 					AddScan(pluginName, target, ch, wg)
 				}
 				continue
 			}
 
-			// 单插件模式直接执行，不检查端口
 			if isSinglePlugin {
-				fmt.Printf("[+] 载入插件: %s\n", pluginName)
 				AddScan(pluginName, target, ch, wg)
 				continue
 			}
 
-			// 预设模式下的常规处理
 			if len(plugin.Ports) > 0 {
 				if plugin.HasPort(targetPort) {
-					fmt.Printf("[+] 载入插件: %s (端口: %d)\n", pluginName, targetPort)
 					AddScan(pluginName, target, ch, wg)
 				}
 			} else {
-				fmt.Printf("[+] 载入插件: %s\n", pluginName)
 				AddScan(pluginName, target, ch, wg)
 			}
 		}
@@ -169,9 +151,11 @@ func executeScans(targets []Common.HostInfo, ch *chan struct{}, wg *sync.WaitGro
 // finishScan 完成扫描任务
 func finishScan(wg *sync.WaitGroup) {
 	wg.Wait()
+	// 先发送最后的成功消息
+	Common.LogSuccess(fmt.Sprintf("扫描已完成: %v/%v", Common.End, Common.Num))
+	// 等待日志处理完成后再关闭通道
 	Common.LogWG.Wait()
 	close(Common.Results)
-	fmt.Printf("[+] 扫描已完成: %v/%v\n", Common.End, Common.Num)
 }
 
 // Mutex用于保护共享资源的并发访问
@@ -193,7 +177,7 @@ func AddScan(plugin string, info Common.HostInfo, ch *chan struct{}, wg *sync.Wa
 
 		// 增加总任务数
 		Mutex.Lock()
-		Common.Num += 1
+		atomic.AddInt64(&Common.Num, 1)
 		Mutex.Unlock()
 
 		// 执行扫描
@@ -201,7 +185,7 @@ func AddScan(plugin string, info Common.HostInfo, ch *chan struct{}, wg *sync.Wa
 
 		// 增加已完成任务数
 		Mutex.Lock()
-		Common.End += 1
+		atomic.AddInt64(&Common.End, 1)
 		Mutex.Unlock()
 	}()
 }
@@ -210,20 +194,18 @@ func AddScan(plugin string, info Common.HostInfo, ch *chan struct{}, wg *sync.Wa
 func ScanFunc(name *string, info *Common.HostInfo) {
 	defer func() {
 		if err := recover(); err != nil {
-			fmt.Printf("[-] 扫描错误 %v:%v - %v\n", info.Host, info.Ports, err)
+			Common.LogError(fmt.Sprintf("扫描错误 %v:%v - %v", info.Host, info.Ports, err))
 		}
 	}()
 
-	// 检查插件是否存在
 	plugin, exists := Common.PluginManager[*name]
 	if !exists {
-		fmt.Printf("[*] 扫描类型 %v 无对应插件，已跳过\n", *name)
+		Common.LogInfo(fmt.Sprintf("扫描类型 %v 无对应插件，已跳过", *name))
 		return
 	}
 
-	// 直接调用扫描函数
 	if err := plugin.ScanFunc(info); err != nil {
-		fmt.Printf("[-] 扫描错误 %v:%v - %v\n", info.Host, info.Ports, err)
+		Common.LogError(fmt.Sprintf("扫描错误 %v:%v - %v", info.Host, info.Ports, err))
 	}
 }