Niko/forwardproxy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refine error user

Browse source
This commit is contained in:
Mygod 2023-07-21 13:57:03 -04:00
parent 4e71e857d9
commit 6a4bb51efc
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
forwardproxy.go
View file

@ -422,17 +422,23 @@ func (h Handler) checkCredentials(r *http.Request) error {
return nil return nil
} }
} }
repl := r.Context().Value(caddy.ReplacerCtxKey).(*caddy.Replacer)
buf := make([]byte, base64.StdEncoding.DecodedLen(len([]byte(pa[1])))) buf := make([]byte, base64.StdEncoding.DecodedLen(len([]byte(pa[1]))))
n, err := base64.StdEncoding.Decode(buf, []byte(pa[1])) n, err := base64.StdEncoding.Decode(buf, []byte(pa[1]))
if err == nil && utf8.Valid(buf[:n]) { if err != nil {
repl.Set("http.auth.user.id", "invalidbase64:"+err.Error())
return err
}
if utf8.Valid(buf[:n]) {
cred := string(buf[:n]) cred := string(buf[:n])
i := strings.IndexByte(cred, ':') i := strings.IndexByte(cred, ':')
repl := r.Context().Value(caddy.ReplacerCtxKey).(*caddy.Replacer)
if i >= 0 { if i >= 0 {
repl.Set("http.auth.user.id", "invalid:"+cred[:i]) repl.Set("http.auth.user.id", "invalid:"+cred[:i])
} else { } else {
repl.Set("http.auth.user.id", "invalidformat:"+cred) repl.Set("http.auth.user.id", "invalidformat:"+cred)
} }
} else {
repl.Set("http.auth.user.id", "invalid::")
} }
return errors.New("invalid credentials") return errors.New("invalid credentials")
} }