mirrors/fscan
1
0
Fork 0
mirror of https://github.com/shadow1ng/fscan.git synced 2025-07-13 21:02:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: 重构扫描模式逻辑

Browse Source
This commit is contained in:
ZacharyZcR 2024-12-22 02:58:55 +08:00
parent e70a1a7bd2
commit eb1b0f32a6
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
Core/Scanner.go
View File

@ -110,14 +110,16 @@ func prepareTargetInfos(alivePorts []string, baseInfo Common.HostInfo) []Common.
func executeScans(targets []Common.HostInfo, ch *chan struct{}, wg *sync.WaitGroup) { func executeScans(targets []Common.HostInfo, ch *chan struct{}, wg *sync.WaitGroup) {
mode := Common.GetScanMode() mode := Common.GetScanMode()
var pluginsToRun []string var pluginsToRun []string
isSinglePlugin := false
// 获取要执行的插件列表 // 获取要执行的插件列表
if plugins := Common.GetPluginsForMode(mode); plugins != nil { if plugins := Common.GetPluginsForMode(mode); plugins != nil {
// 预设模式下使用配置的插件组 // 预设模式下使用配置的插件组
pluginsToRun = plugins pluginsToRun = plugins
} else { } else {
// 单插件模式下只包含指定的插件 // 单插件模式
pluginsToRun = []string{mode} pluginsToRun = []string{mode}
isSinglePlugin = true
} }
// 统一处理所有目标和插件 // 统一处理所有目标和插件
@ -133,14 +135,19 @@ func executeScans(targets []Common.HostInfo, ch *chan struct{}, wg *sync.WaitGro
// 本地扫描模式的特殊处理 // 本地扫描模式的特殊处理
if Common.LocalScan { if Common.LocalScan {
// 只执行没有端口配置的插件
if len(plugin.Ports) == 0 { if len(plugin.Ports) == 0 {
AddScan(pluginName, target, ch, wg) AddScan(pluginName, target, ch, wg)
} }
continue continue
} }
// 非本地扫描模式的常规处理 // 单插件模式直接执行，不检查端口
if isSinglePlugin {
AddScan(pluginName, target, ch, wg)
continue
}
// 预设模式下的常规处理
if len(plugin.Ports) > 0 { if len(plugin.Ports) > 0 {
if plugin.HasPort(targetPort) { if plugin.HasPort(targetPort) {
AddScan(pluginName, target, ch, wg) AddScan(pluginName, target, ch, wg)