sun
b92d3dbc00
1. 修复了变量缩进的格式问题 2. 添加初始化时的白名单来源打印 3. 优化origin校验逻辑，兼容大小写的协议前缀 4. 新增各阶段的CORS请求日志 5. 修正日志中的客户端IP字段名 6. 修复文件末尾缺失换行符的问题
122 lines
3.3 KiB
Go
122 lines
3.3 KiB
Go
package middleware
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
allowedOrigins []string
|
|
allowAllOrigins bool
|
|
corsMaxAgeHeader = "86400"
|
|
)
|
|
|
|
func normalizeOrigin(origin string) string {
|
|
origin = strings.TrimSpace(origin)
|
|
origin = strings.TrimRight(origin, "/")
|
|
return strings.ToLower(origin)
|
|
}
|
|
|
|
func InitCORSConfig() {
|
|
origins := os.Getenv("ALLOWED_ORIGINS")
|
|
if origins == "" {
|
|
log.Println("警告: ALLOWED_ORIGINS 环境变量未设置")
|
|
log.Println("出于安全考虑，跨域请求将被拒绝。如需开放跨域请配置 ALLOWED_ORIGINS")
|
|
log.Println("开发环境可设置 ALLOWED_ORIGINS=* 允许所有来源（不可与凭据共用）")
|
|
return
|
|
}
|
|
|
|
parts := strings.Split(origins, ",")
|
|
for _, p := range parts {
|
|
o := strings.TrimSpace(p)
|
|
if o == "" {
|
|
continue
|
|
}
|
|
if o == "*" {
|
|
allowAllOrigins = true
|
|
continue
|
|
}
|
|
allowedOrigins = append(allowedOrigins, normalizeOrigin(o))
|
|
}
|
|
|
|
if allowAllOrigins {
|
|
log.Println("警告: ALLOWED_ORIGINS=*，将允许所有来源跨域请求（不携带凭据）")
|
|
}
|
|
if len(allowedOrigins) > 0 {
|
|
log.Printf("已配置 %d 个允许的跨域来源白名单", len(allowedOrigins))
|
|
for _, o := range allowedOrigins {
|
|
log.Printf(" - %s", o)
|
|
}
|
|
}
|
|
}
|
|
|
|
func isValidOrigin(origin string) bool {
|
|
if origin == "" || origin == "null" || origin == "nil" {
|
|
return false
|
|
}
|
|
// 使用小写比较，避免大小写问题
|
|
lowerOrigin := strings.ToLower(origin)
|
|
if !strings.HasPrefix(lowerOrigin, "http://") && !strings.HasPrefix(lowerOrigin, "https://") {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func matchOrigin(origin string) (string, bool) {
|
|
if !isValidOrigin(origin) {
|
|
log.Printf("[CORS] Origin %q 验证失败", origin)
|
|
return "", false
|
|
}
|
|
if allowAllOrigins {
|
|
log.Printf("[CORS] Origin %q 匹配 allowAllOrigins", origin)
|
|
return "*", true
|
|
}
|
|
normalized := normalizeOrigin(origin)
|
|
log.Printf("[CORS] 检查 origin %q (normalized: %q) 对比白名单: %v", origin, normalized, allowedOrigins)
|
|
for _, allowed := range allowedOrigins {
|
|
if allowed == normalized {
|
|
log.Printf("[CORS] Origin %q 匹配白名单 %q", origin, allowed)
|
|
return origin, true
|
|
}
|
|
}
|
|
log.Printf("[CORS] Origin %q 未匹配任何白名单", origin)
|
|
return "", false
|
|
}
|
|
|
|
func CORS(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
origin := r.Header.Get("Origin")
|
|
|
|
if origin != "" {
|
|
allowOrigin, matched := matchOrigin(origin)
|
|
if matched {
|
|
w.Header().Set("Access-Control-Allow-Origin", allowOrigin)
|
|
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
|
|
w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization")
|
|
w.Header().Set("Access-Control-Expose-Headers", "X-Request-Id")
|
|
w.Header().Set("Access-Control-Max-Age", corsMaxAgeHeader)
|
|
if allowOrigin != "*" {
|
|
w.Header().Set("Access-Control-Allow-Credentials", "true")
|
|
}
|
|
vary := w.Header().Get("Vary")
|
|
if vary == "" {
|
|
w.Header().Set("Vary", "Origin")
|
|
} else if !strings.Contains(vary, "Origin") {
|
|
w.Header().Set("Vary", vary+", Origin")
|
|
}
|
|
} else {
|
|
log.Printf("CORS拦截: 来源=%q 路径=%s 方法=%s 客户端IP=%s",
|
|
origin, r.URL.Path, r.Method, GetClientIP(r))
|
|
}
|
|
}
|
|
|
|
if r.Method == http.MethodOptions {
|
|
w.WriteHeader(http.StatusNoContent)
|
|
return
|
|
}
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
} |