mirror of
https://github.com/shadow1ng/fscan.git
synced 2025-06-20 18:00:52 +00:00
98 lines
2.8 KiB
Go
98 lines
2.8 KiB
Go
package Common
|
|
|
|
import "fmt"
|
|
|
|
// 扫描模式常量 - 使用大写开头表示这是一个预设的扫描模式
|
|
const (
|
|
ModeAll = "All" // 全量扫描
|
|
ModeBasic = "Basic" // 基础扫描
|
|
ModeDatabase = "Database" // 数据库扫描
|
|
ModeWeb = "Web" // Web扫描
|
|
ModeService = "Service" // 服务扫描
|
|
ModeVul = "Vul" // 漏洞扫描
|
|
ModePort = "Port" // 端口扫描
|
|
ModeICMP = "ICMP" // ICMP探测
|
|
ModeLocal = "Local" // 本地信息收集
|
|
)
|
|
|
|
// 插件分类映射表 - 所有插件名使用小写
|
|
var pluginGroups = map[string][]string{
|
|
ModeAll: {
|
|
"web", "fcgi", // web类
|
|
"mysql", "mssql", "redis", "mongodb", "postgres", // 数据库类
|
|
"oracle", "memcached", "elasticsearch", "rabbitmq", "kafka", "activemq", "cassandra", "neo4j", // 数据库类
|
|
"ftp", "ssh", "telnet", "smb", "rdp", "vnc", "netbios", "ldap", "smtp", "imap", "pop3", "snmp", "modbus", "rsync", // 服务类
|
|
"ms17010", "smbghost", "smb2", // 漏洞类
|
|
"findnet", // 其他
|
|
},
|
|
ModeBasic: {
|
|
"web", "ftp", "ssh", "smb", "findnet",
|
|
},
|
|
ModeDatabase: {
|
|
"mysql", "mssql", "redis", "mongodb",
|
|
"postgres", "oracle", "memcached", "elasticsearch", "rabbitmq", "kafka", "activemq", "cassandra", "neo4j",
|
|
},
|
|
ModeWeb: {
|
|
"web", "fcgi",
|
|
},
|
|
ModeService: {
|
|
"ftp", "ssh", "telnet", "smb", "rdp", "vnc", "netbios", "ldap", "smtp", "imap", "pop3", "modbus", "rsync",
|
|
},
|
|
ModeVul: {
|
|
"ms17010", "smbghost", "smb2",
|
|
},
|
|
ModeLocal: {
|
|
"localinfo",
|
|
},
|
|
}
|
|
|
|
// ParseScanMode 解析扫描模式
|
|
func ParseScanMode(mode string) {
|
|
LogInfo(fmt.Sprintf("解析扫描模式: %s", mode))
|
|
|
|
// 检查是否是预设模式
|
|
presetModes := []string{
|
|
ModeAll, ModeBasic, ModeDatabase, ModeWeb,
|
|
ModeService, ModeVul, ModePort, ModeICMP, ModeLocal,
|
|
}
|
|
|
|
for _, presetMode := range presetModes {
|
|
if mode == presetMode {
|
|
ScanMode = mode
|
|
if plugins := GetPluginsForMode(mode); plugins != nil {
|
|
LogInfo(fmt.Sprintf("使用预设模式: %s, 包含插件: %v", mode, plugins))
|
|
} else {
|
|
LogInfo(fmt.Sprintf("使用预设模式: %s", mode))
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
// 检查是否是有效的插件名
|
|
if _, exists := PluginManager[mode]; exists {
|
|
ScanMode = mode
|
|
LogInfo(fmt.Sprintf("使用单个插件: %s", mode))
|
|
return
|
|
}
|
|
|
|
// 默认使用All模式
|
|
ScanMode = ModeAll
|
|
LogInfo(fmt.Sprintf("未识别的模式，使用默认模式: %s", ModeAll))
|
|
LogInfo(fmt.Sprintf("包含插件: %v", pluginGroups[ModeAll]))
|
|
}
|
|
|
|
// GetPluginsForMode 获取指定模式下的插件列表
|
|
func GetPluginsForMode(mode string) []string {
|
|
plugins, exists := pluginGroups[mode]
|
|
if exists {
|
|
return plugins
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 辅助函数
|
|
func IsPortScan() bool { return ScanMode == ModePort }
|
|
func IsICMPScan() bool { return ScanMode == ModeICMP }
|
|
func IsWebScan() bool { return ScanMode == ModeWeb }
|
|
func GetScanMode() string { return ScanMode }
|