mirror of
https://github.com/shadow1ng/fscan.git
synced 2025-09-14 05:56:46 +08:00
678d750c8a
将复杂的三文件插件架构(connector/exploiter/plugin)重构为简化的单文件插件架构， 大幅减少代码重复和维护成本，提升插件开发效率。 主要改进: • 将每个服务插件从3个文件简化为1个文件 • 删除过度设计的工厂模式、适配器模式等抽象层 • 消除plugins/services/、plugins/adapters/、plugins/base/复杂目录结构 • 实现直接的插件注册机制，提升系统简洁性 • 保持完全向后兼容，所有扫描功能和输出格式不变 重构统计: • 删除文件：100+个复杂架构文件 • 新增文件：20个简化的单文件插件 • 代码减少：每个插件减少60-80%代码量 • 功能增强：所有插件包含完整扫描和利用功能 已重构插件: MySQL, SSH, Redis, MongoDB, PostgreSQL, MSSQL, Oracle, Neo4j, Memcached, RabbitMQ, ActiveMQ, Cassandra, FTP, Kafka, LDAP, Rsync, SMTP, SNMP, Telnet, VNC 验证通过: 新系统编译运行正常，所有插件功能验证通过
66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"os"
|
||
|
||
"github.com/shadow1ng/fscan/app"
|
||
"github.com/shadow1ng/fscan/common"
|
||
|
||
// 导入新的单文件插件架构
|
||
_ "github.com/shadow1ng/fscan/plugins"
|
||
)
|
||
|
||
func main() {
|
||
// 创建应用容器
|
||
container := app.NewContainer()
|
||
|
||
// 第一阶段：基础初始化（插件系统）
|
||
if err := container.Initialize(); err != nil {
|
||
handleError("基础初始化失败", err)
|
||
}
|
||
defer container.Cleanup()
|
||
|
||
// 第二阶段：解析配置
|
||
var info common.HostInfo
|
||
common.Flag(&info)
|
||
|
||
// 第三阶段：日志初始化（依赖于flag解析）
|
||
logInit := &app.LoggerInitializer{}
|
||
if err := logInit.Initialize(); err != nil {
|
||
handleError("日志初始化失败", err)
|
||
}
|
||
|
||
// 第四阶段：参数解析和验证
|
||
if err := common.Parse(&info); err != nil {
|
||
handleError("参数解析失败", err)
|
||
}
|
||
|
||
// 第五阶段：输出系统初始化
|
||
outputInit := &app.OutputInitializer{}
|
||
if err := outputInit.Initialize(); err != nil {
|
||
handleError("输出初始化失败", err)
|
||
}
|
||
|
||
// 第六阶段：执行扫描
|
||
ctx := context.Background()
|
||
if err := container.RunScan(ctx, info); err != nil {
|
||
handleError("扫描失败", err)
|
||
}
|
||
}
|
||
|
||
func handleError(msg string, err error) {
|
||
// 检查是否是应用程序错误
|
||
if appErr, ok := err.(*app.AppError); ok {
|
||
common.LogError(fmt.Sprintf("%s: %s", msg, appErr.Message))
|
||
if appErr.Cause != nil {
|
||
common.LogError(fmt.Sprintf("详细错误: %v", appErr.Cause))
|
||
}
|
||
os.Exit(appErr.Code)
|
||
} else {
|
||
common.LogError(fmt.Sprintf("%s: %v", msg, err))
|
||
os.Exit(1)
|
||
}
|
||
}
|