From 0023b1e7a9cb3f87491b40b37dcd60379b787b45 Mon Sep 17 00:00:00 2001 From: qydysky Date: Sun, 10 Mar 2024 11:23:57 +0800 Subject: [PATCH] 1 --- main.go | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/main.go b/main.go index 5f0e1d0..bfa54a5 100755 --- a/main.go +++ b/main.go @@ -134,22 +134,22 @@ func copyHeader(s, t http.Header, app []Header) error { for _, v := range app { switch v.Action { case `deny`: - if va, ok := tm[v.Key]; ok && len(va) != 0 { - if exp, e := regexp.Compile(v.MatchExp); e == nil && exp.MatchString(va[0]) { + if va := t.Get(v.Key); va != "" { + if exp, e := regexp.Compile(v.MatchExp); e == nil && exp.MatchString(va) { return ErrHeaderCheckFail } } case `access`: - if va, ok := tm[v.Key]; ok && len(va) != 0 { - if exp, e := regexp.Compile(v.MatchExp); e != nil || !exp.MatchString(va[0]) { + if va := t.Get(v.Key); va != "" { + if exp, e := regexp.Compile(v.MatchExp); e != nil || !exp.MatchString(va) { return ErrHeaderCheckFail } } else { return ErrHeaderCheckFail } case `replace`: - if va, ok := tm[v.Key]; ok && len(va) != 0 { - tm[v.Key][0] = regexp.MustCompile(v.MatchExp).ReplaceAllString(va[0], v.Value) + if va := t.Get(v.Key); va != "" { + t.Set(v.Key, regexp.MustCompile(v.MatchExp).ReplaceAllString(va, v.Value)) } case `set`: t.Set(v.Key, v.Value) -- 2.39.2