v2.1.2 核心优化与多架构发布 (#561)

* feat: v2.1.0 核心重构与功能增强

## 架构重构
- 全局变量消除,迁移至 Config/State 对象
- SMB 插件融合(smb/smb2/smbghost/smbinfo)
- 服务探测重构,实现 Nmap 风格 fallback 机制
- 输出系统重构,TXT 实时刷盘 + 双写机制
- i18n 框架升级至 go-i18n

## 性能优化
- 正则表达式预编译
- 内存优化 map[string]struct{}
- 并发指纹匹配
- SOCKS5 连接复用
- 滑动窗口调度 + 自适应线程池

## 新功能
- Web 管理界面
- 多格式 POC 适配(xray/afrog)
- 增强指纹库(3139条)
- Favicon hash 指纹识别
- 插件选择性编译(Build Tags)
- fscan-lab 靶场环境
- 默认端口扩展(62→133)

## 构建系统
- 添加 no_local tag 支持排除本地插件
- 多版本构建:fscan/fscan-nolocal/fscan-web
- CI 添加 snapshot 模式支持仅测试构建

## Bug 修复
- 修复 120+ 个问题,包括 RDP panic、批量扫描漏报、
  JSON 输出格式、Redis 检测、Context 超时等

## 测试增强
- 单元测试覆盖率 74-100%
- 并发安全测试
- 集成测试(Web/端口/服务/SSH/ICMP)

* fix(ci): 移除 PR 对 Project 自动化的触发

* fix: Elasticsearch未授权检测优先于爆破 (#554)

* fix: 修复RDP爆破高误报率问题 (#555)

- 移除 screen.go 中错误的认证结果覆盖逻辑
- 启用 NLA 协议的 ErrorCode 字段检测
- 添加 PubKeyAuth 验证确保认证真正成功
- 修复 io.go 中错误被静默忽略的问题
- 修复 socket.go/io.go 中可能导致 panic 的代码
- 修复 screen.go 中文件句柄泄漏和 log.Panic

* fix: 修复-user/-pwd凭据参数不生效的问题

问题原因:
- Parse()解析凭据后更新globalConfig
- 但BuildConfigFromFlags()创建新Config时使用默认字典
- 导致解析的UserPassPairs等凭据信息被丢弃

修复内容:
1. initialize.go: 将Parse解析的凭据结果应用到新Config
2. credential.go: 单用户密码对时创建UserPassPairs
3. rdp.go: 单凭据测试时跳过指纹识别,减少连接次数

* feat: RDP使用NLA仅验证模式,避免挤掉已登录用户

- 添加ErrNLAAuthSuccess标志用于NLA验证成功信号
- tpkt层支持nlaAuthOnly模式,验证成功后不建立完整会话
- x224层正确传播NLA验证结果
- rdpCrack改用NlaAuth进行凭据验证

* fix: 修复进度条在Windows终端满屏重复输出的问题

- 添加终端宽度检测,动态调整进度条长度
- 使用空格覆盖清除旧内容,避免残留
- 简化进度条格式,确保不超过终端宽度

* feat: 优化日志颜色方案,区分漏洞和普通信息

- 新增 LogVuln 级别(红色),用于漏洞和重要发现
- 密码爆破成功、未授权访问、POC漏洞等改用红色显示
- 普通信息(扫描统计等)改为白色
- Web指纹保持绿色

* refactor: 精简化输出,移除冗余启动信息

- 移除showParseSummary开局配置输出
- 移除LogPluginInfo/LogPluginInfoWithPort插件信息输出
- 移除alive_scanner冗余统计输出
- 移除port_scan_start扫描开始提示
- 移除handleUDPPorts SNMP死代码
- 移除相关i18n条目

* chore: 版本号更新为2.1.1

* fix: 降级依赖版本以保持Go 1.20兼容性

* feat(ldap): 添加NTLM Hash认证支持 (#433)

* chore: 清理无用的 replace 指令

* fix(ping): 修复 TTL expired 导致主机误判为存活的问题

在 ExecCommandPing 中增加错误关键词检测,当 ping 输出包含
TTL expired、Destination unreachable 等错误信息时,不再将
目标主机标记为存活。

Fixes #454

* fix(proxy): 修复透明代理导致输出全端口的问题

在代理初始化时主动探测代理行为,通过连接 RFC 5737 保留的
测试地址来检测是否存在"全回显"问题。如果探测到代理不可靠,
则在端口扫描时跳过所有端口,避免误报。

- 新增 proxyReliable 标志位标记代理可靠性
- 新增 ProbeProxyBehavior 函数探测代理行为
- 端口扫描前检查代理可靠性并输出警告

Fixes #495

* refactor: 移动debug模块到common/debug子包

* fix(web): 修复-u模式下Web插件未执行的问题

* fix: 优化输出格式和颜色显示

- 网段统计格式改为 10.253.0.0/16 网段存活: 26
- WebTitle基础信息改为白色,指纹识别单独绿色输出
- 移除重复的端口数量输出

* fix: URL解析自动补全协议头

-uf 文件中 192.168.1.1:8080 自动转为 http://192.168.1.1:8080

* fix: 修复-u/-uf模式下URLs丢失导致0目标扫描的问题

Parse阶段将URLs设置到全局状态,但Initialize随后创建新状态
并覆盖了全局状态,导致URLs数据丢失。现在在创建新状态前
先保存并迁移Parse阶段设置的URLs和HostPorts数据。

* fix: 智能检测HTTP/HTTPS协议并优化URL显示

- 修复-u/-uf模式URLs丢失导致0目标扫描问题
- detectProtocol改为主动TLS握手检测,不依赖服务名
- WebTitle输出显示完整协议(http/https)
- 隐藏标准端口(80/443)使输出更简洁

* refactor: 精简parsers包,统一配置构建入口

- 删除冗余的中间层(XXXInput、XXXParser类)
- 新增 config_builder.go 统一配置构建
- parsers包从3000+行精简至~540行
- 保留核心函数:ParseIP、ParsePort、文件读取、凭据解析

* test: 扩展parsers单元测试覆盖边缘情况

- 新增内网简写解析测试(192/172/10)
- 新增完整IP范围和无效CIDR测试
- 新增Windows行尾(CRLF)处理测试
- 新增凭据和哈希文件解析测试
- 新增端口解析边缘情况测试
- 测试覆盖率达到94.2%

* refactor: 优化控制台输出格式

- 去掉时间戳,保留[*][+]前缀
- Web输出合并WebTitle和WebFinger为一行
- 有指纹显示绿色[+],无指纹显示白色[*]
- 格式: code:xxx len:xxx title:xxx server:xxx [指纹]
- 服务探测格式: [Product:xxx ||Version:xxx] Banner:(xxx)
- 字段对齐,输出更清爽

* feat: 添加凭据测试未发现弱密码的提示

- credential_tester.go: 失败时设置 Type=ResultTypeCredential
- scanner.go: 根据结果类型在 error 级别输出'未发现弱密码'提示
- 新增 i18n 翻译 brute_no_weak_pass

使用 -log all 或 -log error 可看到此提示

* refactor(logging): 重构日志级别为层级过滤设计

- LogLevel 从 string 改为 int 类型,支持层级比较
- 层级设计:Debug(0) < Base(1) < Info(2) < Success(3) < Vuln(4) < Error(5)
- 设置一个级别后,显示该级别及以上的日志
- Error 级别始终显示,不会被配置过滤掉
- 保留向后兼容别名(LevelAll, LevelInfoSuccess 等)
- 更新测试以匹配新的层级过滤行为

* style(logging): Error级别日志改为黄色显示

* style(findnet): NetInfo输出改为每行一个IP

* refactor(ms17010): 优化错误提示,明确指出SMBv1不支持等情况

* fix(credential): 修复凭据测试结果不一致的问题

问题原因:
1. 未知错误类型不重试,导致服务端限流时跳过正确密码
2. SSH 错误分类不够准确,某些临时错误未被识别

修复内容:
1. 未知错误改为可重试(可能是临时问题)
2. 增加 SSH 特有的网络错误识别(handshake failed, disconnect 等)

* fix(portfinger): 修复SMB2服务指纹识别和NetInfo输出问题

- 添加SMB2ProgNeg探针支持现代Windows的SMB2协议
- 修复Go regexp对高位字节的UTF-8兼容问题,使用Latin-1转换
- 修复探针失败后连接重建逻辑
- 修复vendor_product字段名不匹配问题
- 修复NetInfo多行输出被其他日志打断的问题

* fix(config): 从默认端口移除9100,避免触发打印机打印 (#517)

* feat(proxy): 增强代理端口扫描的深度验证机制

- 新增4阶段深度验证:Banner读取→探测发送→响应等待→最终判定
- 新增SOCKS5错误码和代理错误文本检测
- 优化ProbeProxyBehavior探测逻辑,发送数据验证连接可达性
- 解决透明代理/全回显代理导致的假阳性问题

* fix(proxy): 修复代理深度验证的若干问题

- detector.go: 修复 AutoConfigureProxy 覆盖探测结果的问题
  只有未探测过时才设置默认 proxyReliable 值

- port_scan.go: 改进深度验证机制
  - 使用带 Host header 的 HTTP GET 请求替代 OPTIONS
  - 延长响应等待超时至 2s 以适配慢速服务器
  - 正确重置连接 deadline 避免影响后续操作

* refactor: 统一 common 包文件命名风格

Flag.go -> flag.go

* refactor(proxy): 删除自定义 contains() 函数,改用标准库

- 用 strings.Contains() 替代手写的 contains()
- 删除过时的注释

* fix(parsers): 修复带横杠域名被误识别为IP范围的问题

如 111-555.sss.com 这类域名因包含 - 被错误解析为 IP 范围,
添加 looksLikeIPRange() 检查,只有 - 前是有效 IP 才走范围解析

* fix(proxy): 修复代理模式下服务识别错误和端口漏扫问题

- port_scan.go: 验证通过后重建干净连接,避免HTTP GET探测污染服务识别
- port_scan.go: 优化验证策略,用轻量CRLF探测替代HTTP GET,超时从2.2s降至0.6s
- manager.go: 修正ProbeProxyBehavior判断逻辑,超时应视为代理正常转发

* fix(pool): 移除线程池预分配,优化大规模扫描内存占用

WithPreAlloc(true) 会预先创建所有 worker goroutine,
在大规模扫描(如 25域名×65535端口)时可能导致内存问题

* refactor(logging): 统一日志前缀,删除废弃的 LogBase

- 删除 LogBase 函数,所有调用迁移到 LogInfo/LogError
- 新增 PrefixDebug ([.]) 前缀,所有日志级别现在都有前缀
- 修复日志输出缩进不一致的问题
- 删除未使用的 PrefixDefault 常量

* perf(icmp): 实现自适应等待算法优化存活检测性能

- 新增 waitAdaptive 函数,监控响应增量实现智能提前结束
- 算法保守原则:最小等待1s + 连续500ms无新响应才提前结束
- 添加100ms检查间隔避免CPU空转
- 保留原有最大等待时间(3s/6s)作为兜底
- 添加完整单元测试覆盖各种场景

优化效果:
- 全部响应:~100ms (原3s)
- 无响应:~1s (原3s)
- 部分响应后稳定:~1.5s (原3s)

* perf(scan): 实现启发式优化提升扫描体验

1. 端口优先级排序:高价值端口(80,443,22,3389等)优先扫描
   - 用户能更快看到有意义的结果
   - 不影响端口喷洒策略

2. TCP 补充探测:ICMP 响应率<10%时自动启用
   - 对未响应主机用 TCP 80/443/22/445 补充探测
   - 解决防火墙过滤 ICMP 导致漏检的问题

* refactor(grdp): 精简RDP库,删除认证检测不需要的代码

- 删除 VNC 协议支持 (protocol/rfb, client/rfb.go)
- 删除完整客户端框架 (client/)
- 删除 RemoteApp 等插件 (plugin/)
- 删除 RLE 图形解压 (core/rle.go)
- 删除绘图指令处理 (pdu/orders.go, pdu/gdi.go)
- 精简 screen.go,移除截图和完整会话功能
- 移除未使用的 RGB 转换函数

grdp 代码从 13,044 行精简至 7,581 行,削减 42%

* refactor(common): 删除死代码,优化代码风格

- 删除未使用的 joinStrings/joinInts 函数
- 删除未使用的 memStats 字段和 getMemoryInfo 方法
- 简化 parsePasswords 中的循环为 append(...) 形式

* refactor(services): 统一数据库插件的DBWrapper

4个数据库插件(MySQL、PostgreSQL、MSSQL、Oracle)都有相同的sql.DB包装代码,
合并为通用的SQLDBWrapper,减少重复。

* refactor(core,grdp): 删除未使用的死代码

- 移除 BaseScanStrategy.LogPluginInfoWithPort 方法(无调用者)
- 移除 mcs.go 中被注释的旧 connect 函数实现

* refactor: 删除 deadcode 检测出的未使用函数

- proxy/detector.go: 删除 IsSOCKS5Standard, IsProxyInitialized
- findnet.go: 删除 NetworkInfo.OneLine, TreeFormat 方法
- port_scan.go: 删除 estimateScanTime 函数
- web_scanner.go: 删除 GetFingerprints 函数
- 清理相关测试代码

* refactor: 删除更多未使用的死代码

- parse.go: 删除 RemoveDuplicate 函数及其测试
- parsers.go: 删除 excludeHosts, removeDuplicates 别名函数
- 更新测试使用真正的函数名

* fix(test): 修复 TestParseIP_InvalidIPRange 测试用例

- 删除不合理的测试用例(无效IP被当作普通主机名处理是设计行为)
- 修复测试逻辑,只在真正通过时输出"正确"

* fix(scan): 移除域名预解析,保留原始域名进行扫描

域名预解析会将域名转换为IP,导致虚拟主机场景下HTTP访问失败
(Host头变成IP而非域名,无法正确路由)

* fix(scan): 修复 -hf 参数无法单独使用的问题

* fix(proxy): 修复透明代理环境下 SOCKS5 代理全端口误报问题

问题:在透明代理(TUN模式)环境下使用 SOCKS5 代理扫描时,
会出现全端口开放的误报,因为代理可靠性检测被透明代理污染。

修复方案(参考 fscanx):
1. 将探针从 CRLF 改为 HTTP GET,更有效检测真实连接状态
2. 删除 "uncertain" 状态,无响应一律判定为端口关闭
3. 调整超时时间以适应代理链路延迟

Fixes #524

* feat(telnet): 新增 telnetd RCE 命令执行验证,修复未授权访问日志级别

* fix: 修复 i18n.Tr vet 报错、Unicode 测试用例,移除过期域名

- 移除 i18n.Tr 中错误的 fmt.Sprintf fallback,消除 go vet 误报
- 修复 match_engine_test Unicode 测试用例与 Latin-1 转换逻辑不匹配
- README 移除过期的 fscan.club 域名
- 添加 .gitattributes 统一换行符为 LF

* refactor: 统一控制台输出风格,使用统一的日志函数

手动合并 PR #558 的改动,适配重构后的代码路径

* fix(ci): 修复版本注入和CI触发配置

- goreleaser ldflags 指向正确的包路径 common.version/commit/date
- version 改为 var 支持 ldflags 注入,banner 显示 commit 和构建日期
- test-build 触发分支增加 dev-* 通配

* fix(ci): 修复 Windows 产物 .exe.exe 双后缀问题

* feat(ci): 扩展构建架构支持 MIPS/ARM/FreeBSD/Solaris
This commit is contained in:
ZacharyZcR
2026-04-25 17:39:16 +08:00
committed by GitHub
parent 594f567650
commit 760c8ea502
927 changed files with 82001 additions and 24995 deletions
+765
View File
@@ -0,0 +1,765 @@
package core
import (
"crypto/tls"
"fmt"
"net"
"net/http"
"net/http/httptest"
"strconv"
"sync"
"testing"
"time"
"github.com/shadow1ng/fscan/common"
)
/*
web_scanner_test.go - WebScanner核心逻辑测试
注意:web_scanner.go 包含网络IO和缓存管理。
本测试文件专注于可测试的纯逻辑和算法正确性:
1. IsWebServiceByFingerprint - Web服务识别逻辑
2. createTargetFromURL - URL解析和HostInfo构建
3. 缓存操作 - MarkAsWebService, GetWebServiceInfo, IsMarkedWebService
4. 指纹缓存 - SetFingerprints, GetFingerprints
不测试的部分(需要集成测试):
- createHTTPClient - 依赖全局配置
- tryHTTP, DetectHTTPServiceOnly - 网络IO
- Execute - 完整流程
"服务识别和URL解析是纯逻辑,应该测试。
缓存操作需要验证并发安全性。"
*/
// =============================================================================
// 核心逻辑测试:Web服务识别
// =============================================================================
// TestIsWebServiceByFingerprint 测试Web服务识别逻辑
func TestIsWebServiceByFingerprint(t *testing.T) {
tests := []struct {
name string
serviceInfo *ServiceInfo
expected bool
}{
{
name: "nil服务信息",
serviceInfo: nil,
expected: false,
},
{
name: "空服务名",
serviceInfo: &ServiceInfo{
Name: "",
},
expected: false,
},
{
name: "HTTP服务",
serviceInfo: &ServiceInfo{
Name: "http",
},
expected: true,
},
{
name: "HTTPS服务",
serviceInfo: &ServiceInfo{
Name: "https",
},
expected: true,
},
{
name: "Nginx服务",
serviceInfo: &ServiceInfo{
Name: "nginx",
},
expected: true,
},
{
name: "Apache服务",
serviceInfo: &ServiceInfo{
Name: "apache",
},
expected: true,
},
{
name: "IIS服务",
serviceInfo: &ServiceInfo{
Name: "iis",
},
expected: true,
},
{
name: "Tomcat服务",
serviceInfo: &ServiceInfo{
Name: "tomcat",
},
expected: true,
},
{
name: "MySQL服务-非Web",
serviceInfo: &ServiceInfo{
Name: "mysql",
},
expected: false,
},
{
name: "Redis服务-非Web",
serviceInfo: &ServiceInfo{
Name: "redis",
},
expected: false,
},
{
name: "SSH服务-非Web",
serviceInfo: &ServiceInfo{
Name: "ssh",
},
expected: false,
},
{
name: "FTP服务-非Web",
serviceInfo: &ServiceInfo{
Name: "ftp",
},
expected: false,
},
{
name: "大小写混合-HTTP",
serviceInfo: &ServiceInfo{
Name: "HTTP/1.1",
},
expected: true,
},
{
name: "包含Web关键字-http-server",
serviceInfo: &ServiceInfo{
Name: "custom-http-server",
},
expected: true,
},
{
name: "Banner包含Server头",
serviceInfo: &ServiceInfo{
Name: "unknown",
Banner: "Server: Apache/2.4.41",
},
expected: true,
},
{
name: "Banner包含HTTP协议",
serviceInfo: &ServiceInfo{
Name: "unknown",
Banner: "HTTP/1.1 200 OK",
},
expected: true,
},
{
name: "Banner包含Content-Type",
serviceInfo: &ServiceInfo{
Name: "unknown",
Banner: "Content-Type: text/html",
},
expected: true,
},
{
name: "Banner大写-SERVER",
serviceInfo: &ServiceInfo{
Name: "unknown",
Banner: "SERVER: NGINX/1.18.0",
},
expected: true,
},
{
name: "非Web服务名+非Web Banner",
serviceInfo: &ServiceInfo{
Name: "telnet",
Banner: "Telnet Server Ready",
},
expected: false,
},
{
name: "未知服务+无Banner",
serviceInfo: &ServiceInfo{
Name: "unknown",
Banner: "",
},
expected: false,
},
{
name: "PHP服务",
serviceInfo: &ServiceInfo{
Name: "php",
},
expected: true,
},
{
name: "JSP服务",
serviceInfo: &ServiceInfo{
Name: "jsp",
},
expected: true,
},
{
name: "ASP服务",
serviceInfo: &ServiceInfo{
Name: "asp",
},
expected: true,
},
{
name: "SSL/TLS服务",
serviceInfo: &ServiceInfo{
Name: "ssl",
},
expected: true,
},
{
name: "包含非Web关键字-postgresql",
serviceInfo: &ServiceInfo{
Name: "postgresql-server",
},
expected: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
result := IsWebServiceByFingerprint(tt.serviceInfo)
if result != tt.expected {
t.Errorf("IsWebServiceByFingerprint() = %v, 期望 %v (Name=%q, Banner=%q)",
result, tt.expected, tt.serviceInfo.Name, tt.serviceInfo.Banner)
}
})
}
}
// =============================================================================
// URL解析测试
// =============================================================================
// TestCreateTargetFromURL 测试URL解析和HostInfo构建
func TestCreateTargetFromURL(t *testing.T) {
strategy := NewWebScanStrategy()
tests := []struct {
name string
baseInfo common.HostInfo
urlStr string
expectNil bool
expectedHost string
expectedPort int
expectedURL string
}{
{
name: "完整HTTP URL",
baseInfo: common.HostInfo{},
urlStr: "http://example.com",
expectNil: false,
expectedHost: "example.com",
expectedPort: 80,
expectedURL: "http://example.com",
},
{
name: "完整HTTPS URL",
baseInfo: common.HostInfo{},
urlStr: "https://example.com",
expectNil: false,
expectedHost: "example.com",
expectedPort: 443,
expectedURL: "https://example.com",
},
{
name: "HTTP+自定义端口",
baseInfo: common.HostInfo{},
urlStr: "http://example.com:8080",
expectNil: false,
expectedHost: "example.com",
expectedPort: 8080,
expectedURL: "http://example.com:8080",
},
{
name: "HTTPS+自定义端口",
baseInfo: common.HostInfo{},
urlStr: "https://example.com:8443",
expectNil: false,
expectedHost: "example.com",
expectedPort: 8443,
expectedURL: "https://example.com:8443",
},
{
name: "无协议头-自动添加http",
baseInfo: common.HostInfo{},
urlStr: "example.com",
expectNil: false,
expectedHost: "example.com",
expectedPort: 80,
expectedURL: "http://example.com",
},
{
name: "无协议头+端口",
baseInfo: common.HostInfo{},
urlStr: "example.com:8080",
expectNil: false,
expectedHost: "example.com",
expectedPort: 8080,
expectedURL: "http://example.com:8080",
},
{
name: "IP地址",
baseInfo: common.HostInfo{},
urlStr: "http://192.168.1.1",
expectNil: false,
expectedHost: "192.168.1.1",
expectedPort: 80,
expectedURL: "http://192.168.1.1",
},
{
name: "IP地址+端口",
baseInfo: common.HostInfo{},
urlStr: "http://192.168.1.1:8080",
expectNil: false,
expectedHost: "192.168.1.1",
expectedPort: 8080,
expectedURL: "http://192.168.1.1:8080",
},
{
name: "带路径的URL",
baseInfo: common.HostInfo{},
urlStr: "http://example.com/path/to/page",
expectNil: false,
expectedHost: "example.com",
expectedPort: 80,
expectedURL: "http://example.com/path/to/page",
},
{
name: "带查询参数的URL",
baseInfo: common.HostInfo{},
urlStr: "http://example.com/?key=value",
expectNil: false,
expectedHost: "example.com",
expectedPort: 80,
expectedURL: "http://example.com/?key=value",
},
{
name: "继承baseInfo属性",
baseInfo: common.HostInfo{
Info: []string{"info1", "info2"},
},
urlStr: "http://example.com",
expectNil: false,
expectedHost: "example.com",
expectedPort: 80,
expectedURL: "http://example.com",
},
{
name: "非法URL-无效字符",
baseInfo: common.HostInfo{},
urlStr: "http://example.com:abc",
expectNil: true, // 端口非法,解析失败
},
{
name: "localhost",
baseInfo: common.HostInfo{},
urlStr: "http://localhost:8080",
expectNil: false,
expectedHost: "localhost",
expectedPort: 8080,
expectedURL: "http://localhost:8080",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
result := strategy.createTargetFromURL(tt.baseInfo, tt.urlStr)
// 验证是否为nil
if tt.expectNil {
if result != nil {
t.Errorf("期望返回nil, 实际返回 %+v", result)
}
return
}
if result == nil {
t.Fatal("不应返回nil")
}
// 验证Host
if result.Host != tt.expectedHost {
t.Errorf("Host = %q, 期望 %q", result.Host, tt.expectedHost)
}
// 验证Ports
if result.Port != tt.expectedPort {
t.Errorf("Port = %d, 期望 %d", result.Port, tt.expectedPort)
}
// 验证Url
if result.URL != tt.expectedURL {
t.Errorf("URL = %q, 期望 %q", result.URL, tt.expectedURL)
}
// 验证baseInfo属性继承
if len(tt.baseInfo.Info) > 0 {
if len(result.Info) != len(tt.baseInfo.Info) {
t.Errorf("Infostr未继承, 长度 = %d, 期望 %d",
len(result.Info), len(tt.baseInfo.Info))
}
}
})
}
}
// =============================================================================
// 缓存管理测试
// =============================================================================
// TestWebServiceCache 测试Web服务缓存操作
func TestWebServiceCache(t *testing.T) {
// 清空缓存
webCacheMutex.Lock()
webServiceCache = make(map[string]*ServiceInfo)
webCacheMutex.Unlock()
t.Run("存储和读取", func(t *testing.T) {
serviceInfo := &ServiceInfo{
Name: "http",
Banner: "Apache/2.4.41",
}
// 标记Web服务
MarkAsWebService("192.168.1.1", 80, serviceInfo)
// 验证IsMarkedWebService
if !IsMarkedWebService("192.168.1.1", 80) {
t.Error("IsMarkedWebService应返回true")
}
// 验证GetWebServiceInfo
info, exists := GetWebServiceInfo("192.168.1.1", 80)
if !exists {
t.Error("GetWebServiceInfo应返回exists=true")
}
if info.Name != "http" {
t.Errorf("Name = %q, 期望 'http'", info.Name)
}
})
t.Run("不存在的服务", func(t *testing.T) {
if IsMarkedWebService("192.168.1.2", 80) {
t.Error("不存在的服务应返回false")
}
info, exists := GetWebServiceInfo("192.168.1.2", 80)
if exists {
t.Error("不存在的服务应返回exists=false")
}
if info != nil {
t.Error("不存在的服务应返回nil info")
}
})
t.Run("覆盖写入", func(t *testing.T) {
serviceInfo1 := &ServiceInfo{Name: "http"}
serviceInfo2 := &ServiceInfo{Name: "https"}
MarkAsWebService("192.168.1.3", 80, serviceInfo1)
MarkAsWebService("192.168.1.3", 80, serviceInfo2)
info, _ := GetWebServiceInfo("192.168.1.3", 80)
if info.Name != "https" {
t.Errorf("覆盖后Name = %q, 期望 'https'", info.Name)
}
})
t.Run("不同端口独立存储", func(t *testing.T) {
serviceInfo80 := &ServiceInfo{Name: "http"}
serviceInfo443 := &ServiceInfo{Name: "https"}
MarkAsWebService("192.168.1.4", 80, serviceInfo80)
MarkAsWebService("192.168.1.4", 443, serviceInfo443)
info80, _ := GetWebServiceInfo("192.168.1.4", 80)
info443, _ := GetWebServiceInfo("192.168.1.4", 443)
if info80.Name != "http" {
t.Errorf("端口80的Name = %q, 期望 'http'", info80.Name)
}
if info443.Name != "https" {
t.Errorf("端口443的Name = %q, 期望 'https'", info443.Name)
}
})
}
// TestWebServiceCache_Concurrent 测试并发安全性
func TestWebServiceCache_Concurrent(t *testing.T) {
// 清空缓存
webCacheMutex.Lock()
webServiceCache = make(map[string]*ServiceInfo)
webCacheMutex.Unlock()
t.Run("不同key并发写入", func(t *testing.T) {
var wg sync.WaitGroup
numGoroutines := 100
// 并发写入不同端口
for i := 0; i < numGoroutines; i++ {
wg.Add(1)
go func(id int) {
defer wg.Done()
serviceInfo := &ServiceInfo{
Name: "http",
}
MarkAsWebService("192.168.1.1", id, serviceInfo)
}(i)
}
wg.Wait()
// 验证数据完整性
for i := 0; i < numGoroutines; i++ {
if !IsMarkedWebService("192.168.1.1", i) {
t.Errorf("端口 %d 应被标记", i)
}
}
})
t.Run("同一key并发读写", func(t *testing.T) {
// 这才是真正的race condition测试
var wg sync.WaitGroup
numGoroutines := 100
const testHost = "192.168.1.100"
const testPort = 80
// 同时读写同一个key
for i := 0; i < numGoroutines; i++ {
wg.Add(2)
// 写goroutine
go func(id int) {
defer wg.Done()
serviceInfo := &ServiceInfo{
Name: "http",
Banner: fmt.Sprintf("writer-%d", id),
}
MarkAsWebService(testHost, testPort, serviceInfo)
}(i)
// 读goroutine
go func() {
defer wg.Done()
info, exists := GetWebServiceInfo(testHost, testPort)
// 不验证具体内容(因为写入顺序不确定)
// 只验证不会panic或返回不一致的exists/info
if exists && info == nil {
t.Error("exists=true但info=nil,数据不一致")
}
}()
}
wg.Wait()
// 验证最终状态一致
info, exists := GetWebServiceInfo(testHost, testPort)
if !exists {
t.Error("应该至少有一次写入成功")
}
if info == nil {
t.Error("exists=true但info=nil")
}
})
}
// =============================================================================
// 指纹缓存测试
// =============================================================================
// =============================================================================
// 边界情况测试
// =============================================================================
// TestCreateTargetFromURL_EdgeCases 测试URL解析边界情况
func TestCreateTargetFromURL_EdgeCases(t *testing.T) {
strategy := NewWebScanStrategy()
t.Run("空URL", func(t *testing.T) {
result := strategy.createTargetFromURL(common.HostInfo{}, "")
// url.Parse("")会成功,但Hostname()返回空
if result == nil {
t.Skip("空URL解析行为依赖于url.Parse实现")
}
})
t.Run("只有协议", func(t *testing.T) {
result := strategy.createTargetFromURL(common.HostInfo{}, "http://")
// url.Parse("http://")会成功,但Host为空
if result != nil && result.Host == "" {
t.Log("Empty host check passed as expected")
}
})
t.Run("特殊字符URL", func(t *testing.T) {
result := strategy.createTargetFromURL(common.HostInfo{}, "http://例子.com")
// 中文域名可能成功解析(IDN
if result == nil {
t.Log("中文域名解析失败(预期行为)")
}
})
t.Run("IPv6地址", func(t *testing.T) {
result := strategy.createTargetFromURL(common.HostInfo{}, "http://[::1]:8080")
if result == nil {
t.Error("IPv6地址应能正确解析")
} else {
if result.Host != "::1" {
t.Errorf("IPv6 Host = %q, 期望 '::1'", result.Host)
}
if result.Port != 8080 {
t.Errorf("IPv6 Ports = %q, 期望 '8080'", result.Port)
}
}
})
}
// TestIsWebServiceByFingerprint_Priority 测试识别优先级
func TestIsWebServiceByFingerprint_Priority(t *testing.T) {
t.Run("非Web服务名优先级高于Web Banner", func(t *testing.T) {
// 服务名是mysql,但Banner包含Web特征
serviceInfo := &ServiceInfo{
Name: "mysql",
Banner: "Server: Apache",
}
result := IsWebServiceByFingerprint(serviceInfo)
if result {
t.Error("非Web服务名应优先,即使Banner包含Web特征")
}
})
t.Run("Web服务名优先级高于非Web Banner", func(t *testing.T) {
serviceInfo := &ServiceInfo{
Name: "http",
Banner: "MySQL Server Ready",
}
result := IsWebServiceByFingerprint(serviceInfo)
if !result {
t.Error("Web服务名应优先,即使Banner包含非Web特征")
}
})
}
// =============================================================================
// 协议检测测试
// =============================================================================
// TestDetectHTTPScheme 测试HTTP/HTTPS协议智能检测
func TestDetectHTTPScheme(t *testing.T) {
// 设置WebTimeout避免测试超时
cfg := common.GetGlobalConfig()
oldTimeout := cfg.Network.WebTimeout
cfg.Network.WebTimeout = 2 * time.Second
defer func() { cfg.Network.WebTimeout = oldTimeout }()
t.Run("HTTPS服务器检测", func(t *testing.T) {
// 创建HTTPS测试服务器
server := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
}))
defer server.Close()
// 解析服务器地址
host, portStr, err := net.SplitHostPort(server.Listener.Addr().String())
if err != nil {
t.Fatalf("解析服务器地址失败: %v", err)
}
port, _ := strconv.Atoi(portStr)
// 测试检测
result := DetectHTTPScheme(host, port, cfg)
if result != "https" {
t.Errorf("DetectHTTPScheme() = %q, 期望 'https'", result)
}
})
t.Run("HTTP服务器检测", func(t *testing.T) {
// 创建HTTP测试服务器
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
}))
defer server.Close()
// 解析服务器地址
host, portStr, err := net.SplitHostPort(server.Listener.Addr().String())
if err != nil {
t.Fatalf("解析服务器地址失败: %v", err)
}
port, _ := strconv.Atoi(portStr)
// 测试检测
result := DetectHTTPScheme(host, port, cfg)
if result != "http" {
t.Errorf("DetectHTTPScheme() = %q, 期望 'http'", result)
}
})
t.Run("不存在的服务", func(t *testing.T) {
// 使用127.0.0.1的一个未使用端口
result := DetectHTTPScheme("127.0.0.1", 65534, cfg)
if result != "" {
t.Errorf("不存在的服务应返回空字符串, 实际 %q", result)
}
})
t.Run("非Web服务端口", func(t *testing.T) {
// 创建一个TCP监听器但不响应HTTP
listener, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Skipf("无法创建监听器: %v", err)
}
defer func() { _ = listener.Close() }()
// 启动一个接受连接但立即关闭的goroutine
go func() {
for {
conn, err := listener.Accept()
if err != nil {
return
}
conn.Close()
}
}()
// 解析端口
_, portStr, _ := net.SplitHostPort(listener.Addr().String())
port, _ := strconv.Atoi(portStr)
// 测试检测
result := DetectHTTPScheme("127.0.0.1", port, cfg)
if result != "" {
t.Logf("非Web服务检测返回: %q (预期空字符串,但立即关闭连接可能被误判)", result)
}
})
t.Run("TLS版本兼容性", func(t *testing.T) {
// 测试TLS 1.0兼容性(DetectHTTPScheme设置MinVersion为TLS 1.0
server := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
}))
server.TLS = &tls.Config{
MinVersion: tls.VersionTLS10,
MaxVersion: tls.VersionTLS10,
}
server.StartTLS()
defer server.Close()
host, portStr, _ := net.SplitHostPort(server.Listener.Addr().String())
port, _ := strconv.Atoi(portStr)
result := DetectHTTPScheme(host, port, cfg)
if result != "https" {
t.Errorf("TLS 1.0服务器应被检测为https, 实际 %q", result)
}
})
}