mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-22 03:10:42 +08:00
ZacharyZcR
5ad914a1bb
将 webServiceCache 扩展为通用 serviceCache，所有指纹识别结果 统一缓存，插件匹配时端口不命中则回退到服务名称匹配。 删除多余的 service_cache.go，复用已有的 ServiceInfo 体系。 补充 nil 防御、Explicit 标记、大量单元/集成/回归测试。
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package web
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/shadow1ng/fscan/webscan/lib"
|
|
)
|
|
|
|
type faviconRoundTripper struct {
|
|
called bool
|
|
}
|
|
|
|
func (rt *faviconRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
rt.called = true
|
|
<-req.Context().Done()
|
|
return nil, req.Context().Err()
|
|
}
|
|
|
|
func TestFetchFaviconHashHonorsContext(t *testing.T) {
|
|
previous := lib.Client
|
|
rt := &faviconRoundTripper{}
|
|
lib.Client = &http.Client{Transport: rt}
|
|
defer func() { lib.Client = previous }()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
|
|
hashes := NewWebTitlePlugin().fetchFaviconHash(ctx, "http://example.com")
|
|
if !rt.called {
|
|
t.Fatal("favicon client was not called")
|
|
}
|
|
if len(hashes.MMH3) != 0 || len(hashes.MD5) != 0 {
|
|
t.Fatalf("fetchFaviconHash returned hashes for canceled context: %#v", hashes)
|
|
}
|
|
}
|
|
|
|
func TestWebTitleURLUsesJoinHostPort(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
got string
|
|
want string
|
|
}{
|
|
{"ipv4", webTitleURL("http", "127.0.0.1", 8080), "http://127.0.0.1:8080"},
|
|
{"ipv6", webTitleURL("http", "::1", 8080), "http://[::1]:8080"},
|
|
{"ipv6 display with port", webTitleDisplayURL("https", "2001:db8::1", 8443, false), "https://[2001:db8::1]:8443"},
|
|
{"ipv6 display omit port", webTitleDisplayURL("https", "2001:db8::1", 443, true), "https://[2001:db8::1]"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if tt.got != tt.want {
|
|
t.Fatalf("got %q, want %q", tt.got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|