fix: 彻底解决UDP插件阻塞导致扫描无法结束的问题

根因分析(通过 goroutine dump 定位):
1. UDP conn.Write 在 WSL2 上可能永久阻塞(SetDeadline 对 Write 不生效)
2. SNMP community 爆破混入通用密码字典(57个),串行 × 10s超时 = 10分钟

修复:
- 提取 udpProbe() 公共函数,用 context timeout + conn.Close 双保险
  超时后强制关闭连接,中断阻塞的 Write/Read
- BACnet/DNS/IPMI/TFTP 统一使用 udpProbe()
- SNMP probe 使用 goroutine + context select 保护
- SNMP community 列表不再混入通用密码字典(8个专用 community 足够)
- SNMP 后续 community 爆破用 3s 短超时 + 连续失败 3 次快速退出

效果: 同样的扫描从无限卡死 → 9秒完成
This commit is contained in:
ZacharyZcR
2026-06-14 22:23:50 +08:00
parent 4169eb6ee0
commit 04cae2e42d
7 changed files with 113 additions and 118 deletions
+3 -16
View File
@@ -28,25 +28,12 @@ func (p *DNSPlugin) Scan(ctx context.Context, info *common.HostInfo, session *co
queryID := randomUint16()
query := buildDNSRootNSQuery(queryID)
conn, err := session.DialUDP(ctx, target, timeout)
if err != nil {
return &ScanResult{Success: false, Service: "dns"}
}
defer conn.Close()
_ = conn.SetDeadline(time.Now().Add(timeout))
if _, err := conn.Write(query); err != nil {
data, n := udpProbe(ctx, session, target, timeout, query, 1500)
if data == nil || n < 12 {
return &ScanResult{Success: false, Service: "dns"}
}
buf := make([]byte, 1500)
n, err := conn.Read(buf)
if err != nil || n < 12 {
return &ScanResult{Success: false, Service: "dns"}
}
banner, ok := parseDNSResponse(buf[:n], queryID)
banner, ok := parseDNSResponse(data[:n], queryID)
if !ok {
return &ScanResult{Success: false, Service: "dns"}
}