使用 Go 的 FileServer 安全地提供静态文件

package main

import (
    "log"
    "net/http"
    "path/filepath"
)

// GlobFilterHandler 是一个中间件，用于过滤文件，只允许访问符合指定模式的文件。
func GlobFilterHandler(h http.Handler, pattern string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        path := r.URL.Path
        fileName := filepath.Base(path)

        ok, err := filepath.Match(pattern, fileName)
        if err != nil {
            log.Println("Error in pattern match:", err)
            http.Error(w, "Internal Server Error", http.StatusInternalServerError) // 更加友好的错误提示
            return
        }
        if !ok {
            http.NotFound(w, r)
            return
        }

        h.ServeHTTP(w, r)
    })
}

func main() {
    // 假设静态文件位于 /tmp/dtest 目录
    fileHandler := http.FileServer(http.Dir("/tmp/dtest"))

    // 只允许访问 *.js 文件
    wrappedHandler := GlobFilterHandler(fileHandler, "*.js")

    // 将处理程序注册到服务器
    http.Handle("/", wrappedHandler)  // 注意这里需要指定路由

    // 启动服务器
    log.Fatal(http.ListenAndServe(":8080", nil))
}

package main

import (
    "fmt"
    "log"
    "net/http"
    "path/filepath"
)

// GlobDir 是一个自定义的 http.FileSystem 实现，用于过滤文件，只允许访问符合指定模式的文件。
type GlobDir struct {
    Dir     http.Dir
    Pattern string
}

// Open 方法实现了 http.FileSystem 接口，用于打开文件。
func (d GlobDir) Open(name string) (http.File, error) {
    baseName := filepath.Base(name)

    ok, err := filepath.Match(d.Pattern, baseName)
    if err != nil {
        return nil, err
    }
    if !ok {
        return nil, fmt.Errorf("%s not match GlobDir pattern.", baseName)
    }

    return d.Dir.Open(name)
}

func main() {
    // 假设静态文件位于 /tmp/dtest 目录
    fileHandler := http.FileServer(GlobDir{
        Dir:     http.Dir("/tmp/dtest"),
        Pattern: "*.js",
    })

    // 将处理程序注册到服务器
    http.Handle("/", fileHandler) // 注意这里需要指定路由

    // 启动服务器
    log.Fatal(http.ListenAndServe(":8080", nil))
}