优化小程序部分

go/pkg/verifyfiles/verifyfiles_test.go

@@ -0,0 +1,124 @@
+package verifyfiles
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+// newTestResolver 造一个临时验证目录，返回 resolver。
+func newTestResolver(t *testing.T, cfg Config) *Resolver {
+	t.Helper()
+	r, err := New(cfg)
+	if err != nil {
+		t.Fatalf("创建 Resolver 失败: %v", err)
+	}
+	if r == nil {
+		t.Fatal("Resolver 不应为 nil")
+	}
+	return r
+}
+
+// TestResolveAllowed 确认合法验证文件能被解析到磁盘路径。
+func TestResolveAllowed(t *testing.T) {
+	dir := t.TempDir()
+	if err := os.WriteFile(filepath.Join(dir, "1XXl8fPHr9.txt"), []byte("1XXl8fPHr9"), 0o644); err != nil {
+		t.Fatalf("准备测试文件失败: %v", err)
+	}
+	if err := os.WriteFile(filepath.Join(dir, "baidu_verify_codexxx.html"), []byte("<html></html>"), 0o644); err != nil {
+		t.Fatalf("准备测试文件失败: %v", err)
+	}
+
+	r := newTestResolver(t, Config{Dir: dir})
+
+	for _, name := range []string{"1XXl8fPHr9.txt", "baidu_verify_codexxx.html"} {
+		got, ok := r.Resolve("/" + name)
+		if !ok {
+			t.Fatalf("合法验证文件 /%s 未被解析出来", name)
+		}
+		want := filepath.Join(dir, name)
+		if got != want {
+			t.Fatalf("路径不对，期望 %s，实际 %s", want, got)
+		}
+	}
+}
+
+// TestResolveRejected 确认非法/敏感/不存在的路径一律不放行。
+func TestResolveRejected(t *testing.T) {
+	dir := t.TempDir()
+	for _, name := range []string{"secret.env", "notes.md", "index.html", "app.conf"} {
+		if err := os.WriteFile(filepath.Join(dir, name), []byte("x"), 0o644); err != nil {
+			t.Fatalf("准备测试文件失败: %v", err)
+		}
+	}
+	if err := os.Mkdir(filepath.Join(dir, "sub.txt"), 0o755); err != nil {
+		t.Fatalf("准备测试目录失败: %v", err)
+	}
+
+	r := newTestResolver(t, Config{Dir: dir})
+
+	cases := []string{
+		"/",
+		"",
+		"/secret.env",           // 敏感后缀不在白名单
+		"/notes.md",             // 后缀不在白名单
+		"/app.conf",             // 配置文件不得外泄
+		"/index.html",           // 会被 ServeFile 重定向
+		"/sub/dir.txt",          // 只允许根路径单段文件名
+		"/../secret.env",        // 路径穿越
+		"/1XXl8fPHr9.txt/../..", // 路径穿越
+		"/missing.txt",          // 文件不存在
+		"/sub.txt",              // 目录不是文件
+	}
+	for _, c := range cases {
+		if got, ok := r.Resolve(c); ok {
+			t.Fatalf("路径 %q 不应被放行，实际解析为 %s", c, got)
+		}
+	}
+}
+
+// TestAllowHost 确认域名白名单按「忽略端口 + 忽略大小写」匹配。
+func TestAllowHost(t *testing.T) {
+	r := newTestResolver(t, Config{Dir: t.TempDir(), Hosts: "api.yunzer.cn"})
+
+	if !r.AllowHost("api.yunzer.cn") {
+		t.Fatal("api.yunzer.cn 应被允许")
+	}
+	if !r.AllowHost("API.Yunzer.cn:443") {
+		t.Fatal("带端口与大写的 api.yunzer.cn 应被允许")
+	}
+	if r.AllowHost("other.com") {
+		t.Fatal("未在白名单里的域名不应被允许")
+	}
+
+	// 未配置 hosts 时，所有走本服务的域名都生效
+	all := newTestResolver(t, Config{Dir: t.TempDir()})
+	if !all.AllowHost("api.yunzer.cn") || !all.AllowHost("dh2.fun") {
+		t.Fatal("未配置 verify_hosts 时应允许所有域名")
+	}
+}
+
+// TestNewDisabled 确认 dir = "-" 表示关闭该功能。
+func TestNewDisabled(t *testing.T) {
+	r, err := New(Config{Dir: "-"})
+	if err != nil {
+		t.Fatalf("不应报错: %v", err)
+	}
+	if r != nil {
+		t.Fatal("dir = \"-\" 时 Resolver 应为 nil")
+	}
+}
+
+// TestDefaultExts 确认默认扩展名包含 .txt。
+func TestDefaultExts(t *testing.T) {
+	r := newTestResolver(t, Config{Dir: t.TempDir()})
+	found := false
+	for _, e := range r.Exts() {
+		if e == ".txt" {
+			found = true
+		}
+	}
+	if !found {
+		t.Fatalf("默认扩展名应包含 .txt，实际 %v", r.Exts())
+	}
+}