mirror of
https://github.com/shadow1ng/fscan.git
synced 2025-09-14 14:06:44 +08:00
120bd9f341
将所有29个services插件从RegisterPlugin()转换为高效的RegisterPluginWithPorts()注册方式： 核心优化： - 消除启动时1200+无用插件实例创建（25插件×47次调用） - 统一插件注册机制，移除性能较差的旧接口 - 优化插件存在性检查，使用O(1)查询替代实例化检查 技术改进： - 移除旧RegisterPlugin()函数，简化代码路径 - 所有service插件使用统一高效注册方式 - 保持业务逻辑和外部接口完全不变 性能提升： - 显著减少启动时间和内存占用 - 消除重复的"加载了175个AV产品信息"日志输出 - 插件系统响应更快，扫描启动更迅速 影响范围：29个services插件全部完成转换 向后兼容：保持所有现有功能和接口不变
122 lines
2.5 KiB
Go
122 lines
2.5 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
ldaplib "github.com/go-ldap/ldap/v3"
|
|
"github.com/shadow1ng/fscan/common"
|
|
)
|
|
|
|
type LDAPPlugin struct {
|
|
name string
|
|
ports []int
|
|
}
|
|
|
|
func NewLDAPPlugin() *LDAPPlugin {
|
|
return &LDAPPlugin{
|
|
name: "ldap",
|
|
ports: []int{389, 636, 3268, 3269},
|
|
}
|
|
}
|
|
|
|
func (p *LDAPPlugin) GetName() string {
|
|
return p.name
|
|
}
|
|
|
|
func (p *LDAPPlugin) GetPorts() []int {
|
|
return p.ports
|
|
}
|
|
|
|
func (p *LDAPPlugin) 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)
|
|
}
|
|
|
|
credentials := GenerateCredentials("ldap")
|
|
if len(credentials) == 0 {
|
|
return &ScanResult{
|
|
Success: false,
|
|
Service: "ldap",
|
|
Error: fmt.Errorf("没有可用的测试凭据"),
|
|
}
|
|
}
|
|
|
|
for _, cred := range credentials {
|
|
if p.testCredential(ctx, info, cred) {
|
|
common.LogSuccess(fmt.Sprintf("LDAP %s %s:%s", target, cred.Username, cred.Password))
|
|
return &ScanResult{
|
|
Success: true,
|
|
Service: "ldap",
|
|
Username: cred.Username,
|
|
Password: cred.Password,
|
|
}
|
|
}
|
|
}
|
|
|
|
return &ScanResult{
|
|
Success: false,
|
|
Service: "ldap",
|
|
Error: fmt.Errorf("未发现弱密码"),
|
|
}
|
|
}
|
|
|
|
|
|
func (p *LDAPPlugin) testCredential(ctx context.Context, info *common.HostInfo, cred Credential) bool {
|
|
conn, err := p.connectLDAP(ctx, info, cred)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
defer conn.Close()
|
|
|
|
// 简单的绑定测试
|
|
if err := conn.Bind(cred.Username, cred.Password); err == nil {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (p *LDAPPlugin) connectLDAP(ctx context.Context, info *common.HostInfo, creds Credential) (*ldaplib.Conn, error) {
|
|
target := fmt.Sprintf("%s:%s", info.Host, info.Ports)
|
|
|
|
if info.Ports == "636" {
|
|
return ldaplib.DialTLS("tcp", target, nil)
|
|
}
|
|
return ldaplib.Dial("tcp", target)
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
func (p *LDAPPlugin) identifyService(ctx context.Context, info *common.HostInfo) *ScanResult {
|
|
target := fmt.Sprintf("%s:%s", info.Host, info.Ports)
|
|
|
|
conn, err := p.connectLDAP(ctx, info, Credential{})
|
|
if err != nil {
|
|
return &ScanResult{
|
|
Success: false,
|
|
Service: "ldap",
|
|
Error: err,
|
|
}
|
|
}
|
|
defer conn.Close()
|
|
|
|
banner := "LDAP"
|
|
common.LogSuccess(fmt.Sprintf("LDAP %s %s", target, banner))
|
|
return &ScanResult{
|
|
Success: true,
|
|
Service: "ldap",
|
|
Banner: banner,
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
// 使用高效注册方式：直接传递端口信息，避免实例创建
|
|
RegisterPluginWithPorts("ldap", func() Plugin {
|
|
return NewLDAPPlugin()
|
|
}, []int{389, 636, 3268, 3269})
|
|
} |