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行） 功能影响：零破坏性，保持所有业务逻辑不变 这是基于业务需求分析的精准重构，消除真正多余的部分， 保持系统架构合理性和向后兼容性。
142 lines
2.9 KiB
Go
142 lines
2.9 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/shadow1ng/fscan/common"
|
|
)
|
|
|
|
type MemcachedPlugin struct {
|
|
name string
|
|
ports []int
|
|
}
|
|
|
|
func NewMemcachedPlugin() *MemcachedPlugin {
|
|
return &MemcachedPlugin{
|
|
name: "memcached",
|
|
ports: []int{11211, 11212, 11213},
|
|
}
|
|
}
|
|
|
|
func (p *MemcachedPlugin) Name() string {
|
|
return p.name
|
|
}
|
|
|
|
func (p *MemcachedPlugin) GetPorts() []int {
|
|
return p.ports
|
|
}
|
|
|
|
func (p *MemcachedPlugin) Scan(ctx context.Context, info *common.HostInfo) *ScanResult {
|
|
target := fmt.Sprintf("%s:%s", info.Host, info.Ports)
|
|
|
|
if common.DisableBrute {
|
|
return p.identifyService(ctx, info)
|
|
}
|
|
|
|
conn := p.connectToMemcached(ctx, info)
|
|
if conn == nil {
|
|
return &ScanResult{
|
|
Success: false,
|
|
Service: "memcached",
|
|
Error: fmt.Errorf("Memcached服务连接失败"),
|
|
}
|
|
}
|
|
defer conn.Close()
|
|
|
|
if p.testBasicCommand(conn) {
|
|
common.LogSuccess(fmt.Sprintf("Memcached %s 未授权访问", target))
|
|
return &ScanResult{
|
|
Success: true,
|
|
Service: "memcached",
|
|
Banner: "未授权访问",
|
|
}
|
|
}
|
|
|
|
return &ScanResult{
|
|
Success: false,
|
|
Service: "memcached",
|
|
Error: fmt.Errorf("Memcached服务连接失败"),
|
|
}
|
|
}
|
|
|
|
|
|
func (p *MemcachedPlugin) connectToMemcached(ctx context.Context, info *common.HostInfo) net.Conn {
|
|
target := fmt.Sprintf("%s:%s", info.Host, info.Ports)
|
|
timeout := time.Duration(common.Timeout) * time.Second
|
|
|
|
conn, err := net.DialTimeout("tcp", target, timeout)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
conn.SetDeadline(time.Now().Add(timeout))
|
|
|
|
return conn
|
|
}
|
|
|
|
func (p *MemcachedPlugin) testBasicCommand(conn net.Conn) bool {
|
|
timeout := time.Duration(common.Timeout) * time.Second
|
|
|
|
conn.SetWriteDeadline(time.Now().Add(timeout))
|
|
if _, err := conn.Write([]byte("version\r\n")); err != nil {
|
|
return false
|
|
}
|
|
|
|
conn.SetReadDeadline(time.Now().Add(timeout))
|
|
response := make([]byte, 1024)
|
|
n, err := conn.Read(response)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
responseStr := string(response[:n])
|
|
return strings.Contains(responseStr, "VERSION") || strings.Contains(responseStr, "memcached")
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
func (p *MemcachedPlugin) identifyService(ctx context.Context, info *common.HostInfo) *ScanResult {
|
|
target := fmt.Sprintf("%s:%s", info.Host, info.Ports)
|
|
|
|
conn := p.connectToMemcached(ctx, info)
|
|
if conn == nil {
|
|
return &ScanResult{
|
|
Success: false,
|
|
Service: "memcached",
|
|
Error: fmt.Errorf("无法连接到Memcached服务"),
|
|
}
|
|
}
|
|
defer conn.Close()
|
|
|
|
banner := "Memcached"
|
|
if p.testBasicCommand(conn) {
|
|
common.LogSuccess(fmt.Sprintf("Memcached %s %s", target, banner))
|
|
return &ScanResult{
|
|
Success: true,
|
|
Service: "memcached",
|
|
Banner: banner,
|
|
}
|
|
}
|
|
|
|
return &ScanResult{
|
|
Success: false,
|
|
Service: "memcached",
|
|
Error: fmt.Errorf("无法识别为Memcached服务"),
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
// 使用高效注册方式：直接传递端口信息，避免实例创建
|
|
RegisterPluginWithPorts("memcached", func() Plugin {
|
|
return NewMemcachedPlugin()
|
|
}, []int{11211, 11212, 11213})
|
|
} |