中间件判断逻辑

csznet · Apr 9, 2024 · 163e85b · 163e85b
1 parent 2f0d9c6
commit 163e85b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/control/control.go b/control/control.go
@@ -274,7 +274,7 @@ func Middleware(next http.HandlerFunc) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 		// 只有当密码设置并且不为"none"时，才进行检查
 		if conf.Pass != "" && conf.Pass != "none" {
-			if cookie, err := r.Cookie("p"); err != nil || cookie.Value != conf.Pass || r.URL.Path == "/api" && r.URL.Query().Get("pass") != conf.Pass {
+			if cookie, err := r.Cookie("p"); err != nil || cookie.Value != conf.Pass || strings.HasPrefix(r.URL.Path, "/api") && r.URL.Query().Get("pass") != conf.Pass {
 				http.Redirect(w, r, "/pwd", http.StatusSeeOther)
 				return
 			}